Searched refs:b_matrix_shape (Results 1 – 1 of 1) sorted by relevance
1071 auto b_matrix_shape = in CountMatMulOperations() local1081 k_dim_b = b_matrix_shape.dim(1).size(); in CountMatMulOperations()1082 n_dim = b_matrix_shape.dim(0).size(); in CountMatMulOperations()1084 k_dim_b = b_matrix_shape.dim(0).size(); in CountMatMulOperations()1085 n_dim = b_matrix_shape.dim(1).size(); in CountMatMulOperations()1181 TensorShapeProto* b_matrix_shape = b_matrix->mutable_shape(); in GenerateBatchMatmulContextFromEinsum() local1210 *(b_matrix_shape->add_dim()) = a_input_shape.dim(i_idx); in GenerateBatchMatmulContextFromEinsum()1226 *(b_matrix_shape->add_dim()) = k_dim; in GenerateBatchMatmulContextFromEinsum()1227 *(b_matrix_shape->add_dim()) = n_dim; in GenerateBatchMatmulContextFromEinsum()1337 TensorShapeProto* b_matrix_shape = b_matrix->mutable_shape(); in CountBatchMatMulOperations() local[all …]