Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/grappler/costs/
Dop_level_cost_estimator.cc716 auto a_matrix_shape = in CountMatMulOperations() local
721 m_dim = a_matrix_shape.dim(1).size(); in CountMatMulOperations()
722 k_dim = a_matrix_shape.dim(0).size(); in CountMatMulOperations()
724 m_dim = a_matrix_shape.dim(0).size(); in CountMatMulOperations()
725 k_dim = a_matrix_shape.dim(1).size(); in CountMatMulOperations()
843 TensorShapeProto* a_matrix_shape = a_matrix->mutable_shape(); in CountBatchMatMulOperations() local
846 *(a_matrix_shape->add_dim()) = a_input_shape.dim(i); in CountBatchMatMulOperations()