Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/kernels/mkl/
Dmkl_batch_matmul_op.cc163 void ExpandInputDimsToOutputShape(const TensorShape& input_shape, in ExpandInputDimsToOutputShape() function in tensorflow::BatchMatMulMkl
193 ExpandInputDimsToOutputShape(lhs_shape, out_shape, &lhs_dims); in CreateMatMulParams()
196 ExpandInputDimsToOutputShape(rhs_shape, out_shape, &rhs_dims); in CreateMatMulParams()