Home
last modified time | relevance | path

Searched refs:CreateMatMulParams (Results 1 – 1 of 1) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/mkl/
Dmkl_batch_matmul_op.cc140 auto params = CreateMatMulParams(lhs.shape(), rhs.shape(), out_shape); in Compute()
177 std::unique_ptr<MklMatMulParams> CreateMatMulParams( in CreateMatMulParams() function in tensorflow::BatchMatMulMkl