Searched defs:matmul_prim (Results 1 – 4 of 4) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/mkl/ |
| H A D | mkl_matmul_ops_common.h | 813 MklMatMulPrimitive<Tlhs, Trhs, Toutput>* matmul_prim = nullptr; in Get() local
908 MklMatMulPrimitive<T, T, T>* matmul_prim = variable
|
| H A D | mkl_batch_matmul_op.cc | 154 MklMatMulPrimitive<Tlhs, Trhs, Toutput>* matmul_prim = in Compute() local
|
| H A D | mkl_matmul_op_fused.cc | 140 MklDnnMatMulFwdPrimitive<T, T, T, T, T>* matmul_prim = in Compute() local
|
| H A D | mkl_einsum_op.cc | 114 MklMatMulPrimitive<T, T, T>* matmul_prim = in MKLContractOperands() local
|