Searched refs:b_matrix_shape (Results 1 – 1 of 1) sorted by relevance
718 auto b_matrix_shape = in CountMatMulOperations() local728 k_dim_b = b_matrix_shape.dim(1).size(); in CountMatMulOperations()729 n_dim = b_matrix_shape.dim(0).size(); in CountMatMulOperations()731 k_dim_b = b_matrix_shape.dim(0).size(); in CountMatMulOperations()732 n_dim = b_matrix_shape.dim(1).size(); in CountMatMulOperations()851 TensorShapeProto* b_matrix_shape = b_matrix->mutable_shape(); in CountBatchMatMulOperations() local854 *(b_matrix_shape->add_dim()) = b_input_shape.dim(i); in CountBatchMatMulOperations()1247 auto b_matrix_shape = in PredictSparseTensorDenseMatMul() local1249 int64 n_dim = b_matrix_shape.dim(1).size(); in PredictSparseTensorDenseMatMul()