Searched defs:bmm (Results 1 – 14 of 14) sorted by relevance
| /aosp_15_r20/external/executorch/backends/qualcomm/_passes/ |
| H A D | convert_bmm_to_matmul.py | 25 bmm = exir_ops.edge.aten.bmm.default variable in ConvertBmmToMatmul
|
| H A D | convert_to_linear.py | 38 bmm = exir_ops.edge.aten.bmm.default variable in ConvertToLinear
|
| /aosp_15_r20/external/armnn/src/backends/reference/workloads/ |
| H A D | RefBatchMatMulWorkload.cpp | 47 auto bmm = BatchMatMul(m_Data.m_Parameters, in Execute() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/mkl/ |
| H A D | mkl_batch_matmul_op.cc | 137 MklBatchMatMulHelper bmm; in Compute() local
|
| H A D | mkl_einsum_op.cc | 108 MklBatchMatMulHelper bmm; in MKLContractOperands() local
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/mkldnn/xpu/ |
| H A D | Blas.cpp | 339 Tensor bmm(const Tensor& self, const Tensor& batch2) { in bmm() function
|
| /aosp_15_r20/external/pytorch/test/inductor/ |
| H A D | test_pad_mm.py | 387 def bmm(mat1, mat2): function
|
| H A D | test_cutlass_backend.py | 386 def bmm(a, b): function
|
| /aosp_15_r20/external/pytorch/torch/onnx/ |
| H A D | symbolic_opset8.py | 211 def bmm(g: jit_utils.GraphContext, self, other): function
|
| H A D | symbolic_opset9.py | 602 def bmm(g: jit_utils.GraphContext, self, other): function
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/vulkan/ops/ |
| H A D | Mm.cpp | 906 Tensor bmm(const Tensor& mat1_arg, const Tensor& mat2_arg) { in bmm() function
|
| /aosp_15_r20/external/pytorch/torch/_inductor/ |
| H A D | decomposition.py | 230 def bmm( function
|
| /aosp_15_r20/external/pytorch/torch/jit/ |
| H A D | _shape_functions.py | 1078 def bmm(self: List[int], mat2: List[int]) -> List[int]: function
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/ |
| H A D | LinearAlgebra.cpp | 330 TORCH_META_FUNC(bmm)(const Tensor& self, const Tensor& mat2) { in TORCH_META_FUNC() argument
|