Home
last modified time | relevance | path

Searched defs:matmul_prim (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/mkl/
Dmkl_batch_matmul_op.cc142 MklMatMulPrimitive<Scalar>* matmul_prim = in Compute() local
Dmkl_matmul_op_fused.cc126 MklDnnMatMulFwdPrimitive<T, T, T, T, T>* matmul_prim = in Compute() local
Dmkl_matmul_ops_common.h640 MklMatMulPrimitive<T>* matmul_prim = nullptr; in Get() local
716 MklMatMulPrimitive<T>* matmul_prim = variable