Searched refs:MklDnnMatMulFwdPrimitive (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/mkl/ |
| D | mkl_matmul_ops_common.h | 83 class MklDnnMatMulFwdPrimitive : public MklPrimitive {
85 explicit MklDnnMatMulFwdPrimitive( in MklDnnMatMulFwdPrimitive() function
94 ~MklDnnMatMulFwdPrimitive() {} in ~MklDnnMatMulFwdPrimitive()
294 static MklDnnMatMulFwdPrimitive<T, Tinput, Tweight, Tbias, Toutput>* Get( in Get()
296 MklDnnMatMulFwdPrimitive<T, Tinput, Tweight, Tbias, Toutput>* matmul_fwd = in Get()
302 new MklDnnMatMulFwdPrimitive<T, Tinput, Tweight, Tbias, Toutput>( in Get()
307 MklDnnMatMulFwdPrimitive<T, Tinput, Tweight, Tbias, Toutput>*>( in Get()
313 new MklDnnMatMulFwdPrimitive<T, Tinput, Tweight, Tbias, Toutput>( in Get()
|
| D | mkl_matmul_op_fused.cc | 126 MklDnnMatMulFwdPrimitive<T, T, T, T, T>* matmul_prim = in Compute()
|
| D | mkl_qmatmul_op.cc | 223 MklDnnMatMulFwdPrimitive<float, Tinput, Tweight, Tbias, Toutput>* in Compute()
|