Searched refs:MklDnnMatMulFwdPrimitiveFactory (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/mkl/ |
| D | mkl_matmul_ops_common.h | 292 class MklDnnMatMulFwdPrimitiveFactory : public MklPrimitiveFactory<T> {
308 MklDnnMatMulFwdPrimitiveFactory<T, Tinput, Tweight, Tbias, in Get()
315 MklDnnMatMulFwdPrimitiveFactory<T, Tinput, Tweight, Tbias, in Get()
324 MklDnnMatMulFwdPrimitiveFactory() {} in MklDnnMatMulFwdPrimitiveFactory() function
325 ~MklDnnMatMulFwdPrimitiveFactory() {} in ~MklDnnMatMulFwdPrimitiveFactory()
327 static MklDnnMatMulFwdPrimitiveFactory& GetInstance() { in GetInstance()
328 static MklDnnMatMulFwdPrimitiveFactory instance_; in GetInstance()
|
| D | mkl_matmul_op_fused.cc | 127 MklDnnMatMulFwdPrimitiveFactory<T, T, T, T, T>::Get(matmul_params, 0); in Compute()
|
| D | mkl_qmatmul_op.cc | 235 MklDnnMatMulFwdPrimitiveFactory<float, Tinput, Tweight, Tbias, in Compute()
|