Searched defs:b_matrix_shape (Results 1 – 1 of 1) sorted by relevance
1071 auto b_matrix_shape = in CountMatMulOperations() local1181 TensorShapeProto* b_matrix_shape = b_matrix->mutable_shape(); in GenerateBatchMatmulContextFromEinsum() local1337 TensorShapeProto* b_matrix_shape = b_matrix->mutable_shape(); in CountBatchMatMulOperations() local1838 auto b_matrix_shape = in PredictSparseTensorDenseMatMul() local