Searched refs:b_input_shape (Results 1 – 1 of 1) sorted by relevance
793 TensorShapeProto b_input_shape = MaybeGetMinimumShape( in CountBatchMatMulOperations() local805 TensorShapeProto* smaller_rank_shape = &b_input_shape; in CountBatchMatMulOperations()806 if (b_input_shape.dim_size() > a_input_shape.dim_size()) { in CountBatchMatMulOperations()807 bigger_rank_shape = &b_input_shape; in CountBatchMatMulOperations()852 for (int i = std::max(0, b_input_shape.dim_size() - matrix_rank); in CountBatchMatMulOperations()853 i < b_input_shape.dim_size(); ++i) { in CountBatchMatMulOperations()854 *(b_matrix_shape->add_dim()) = b_input_shape.dim(i); in CountBatchMatMulOperations()