Searched refs:MklMatMulParams (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/mkl/ |
| D | mkl_matmul_ops_common.h | 511 struct MklMatMulParams { struct
519 MklMatMulParams(memory::dims a_dims, memory::dims b_dims, memory::dims c_dims, in MklMatMulParams() argument
533 explicit MklMatMulPrimitive(const MklMatMulParams& params) in MklMatMulPrimitive()
595 void Setup(const MklMatMulParams& params) { in Setup()
638 static MklMatMulPrimitive<T>* Get(const MklMatMulParams& params, in Get()
668 static string CreateKey(const MklMatMulParams& params) { in CreateKey()
683 MklPrimitive* GetMklMatMul(const MklMatMulParams& params) { in GetMklMatMul()
688 void SetMklMatMul(const MklMatMulParams& params, MklPrimitive* op) { in SetMklMatMul()
714 MklMatMulParams params(a_dims, b_dims, c_dims, a_strides, b_strides,
|
| D | mkl_batch_matmul_op.cc | 177 std::unique_ptr<MklMatMulParams> CreateMatMulParams( in CreateMatMulParams()
226 return std::make_unique<MklMatMulParams>( in CreateMatMulParams()
|