Searched defs:a_matrix_shape (Results 1 – 1 of 1) sorted by relevance
1069 auto a_matrix_shape = in CountMatMulOperations() local1176 TensorShapeProto* a_matrix_shape = a_matrix->mutable_shape(); in GenerateBatchMatmulContextFromEinsum() local1329 TensorShapeProto* a_matrix_shape = a_matrix->mutable_shape(); in CountBatchMatMulOperations() local