Searched refs:a_matrix_shape (Results 1 – 1 of 1) sorted by relevance
1069 auto a_matrix_shape = in CountMatMulOperations() local1074 m_dim = a_matrix_shape.dim(1).size(); in CountMatMulOperations()1075 k_dim = a_matrix_shape.dim(0).size(); in CountMatMulOperations()1077 m_dim = a_matrix_shape.dim(0).size(); in CountMatMulOperations()1078 k_dim = a_matrix_shape.dim(1).size(); in CountMatMulOperations()1176 TensorShapeProto* a_matrix_shape = a_matrix->mutable_shape(); in GenerateBatchMatmulContextFromEinsum() local1209 *(a_matrix_shape->add_dim()) = a_input_shape.dim(i_idx); in GenerateBatchMatmulContextFromEinsum()1224 *(a_matrix_shape->add_dim()) = m_dim; in GenerateBatchMatmulContextFromEinsum()1225 *(a_matrix_shape->add_dim()) = k_dim; in GenerateBatchMatmulContextFromEinsum()1329 TensorShapeProto* a_matrix_shape = a_matrix->mutable_shape(); in CountBatchMatMulOperations() local[all …]