MkldnnLinearBackwardWeights

Computes backward pass for linear weights (MKLDNN).

Abstract Signature:

MkldnnLinearBackwardWeights(grad_output: Tensor, input: Tensor, weight: Tensor, bias_defined: bool)

PyTorchJAX (Core)Apple MLX

PyTorch

API: torch.mkldnn_linear_backward_weights

Strategy: Direct Mapping

Official Docs ↗

JAX (Core)

API: —

Strategy: Custom / Partial

Apple MLX

API: —

Strategy: Custom / Partial