Searched refs:matmul_prim (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/mkl/ |
| D | mkl_matmul_ops_common.h | 640 MklMatMulPrimitive<T>* matmul_prim = nullptr; in Get() local
644 matmul_prim = new MklMatMulPrimitive<T>(params); in Get()
647 matmul_prim = dynamic_cast<MklMatMulPrimitive<T>*>( in Get()
649 if (matmul_prim == nullptr) { in Get()
650 matmul_prim = new MklMatMulPrimitive<T>(params); in Get()
652 matmul_prim); in Get()
656 return matmul_prim; in Get()
716 MklMatMulPrimitive<T>* matmul_prim = variable
721 cpu_stream.reset(CreateStream(ctx, matmul_prim->GetEngine()));
722 matmul_prim->Execute(a, b, c, cpu_stream);
|
| D | mkl_matmul_op_fused.cc | 126 MklDnnMatMulFwdPrimitive<T, T, T, T, T>* matmul_prim = in Compute() local
132 matmul_prim->GetPrimitiveDesc(); in Compute()
252 cpu_stream.reset(CreateStream(ctx, matmul_prim->GetEngine())); in Compute()
254 matmul_prim->Execute(src_data, weight_data, bias_data, dst_data, in Compute()
|
| D | mkl_batch_matmul_op.cc | 142 MklMatMulPrimitive<Scalar>* matmul_prim = in Compute() local
147 cpu_stream.reset(CreateStream(ctx, matmul_prim->GetEngine())); in Compute()
148 matmul_prim->Execute(lhs.flat<Scalar>().data(), rhs.flat<Scalar>().data(), in Compute()
|