Searched refs:CountBatchMatMulOperations (Results 1 – 3 of 3) sorted by relevance
87 int64 CountBatchMatMulOperations(const OpInfo& op_info,
462 int64 CountBatchMatMulOperations(const OpInfo& op_info, in CountBatchMatMulOperations() function in tensorflow::grappler::OpLevelCostEstimatorTest464 return estimator_.CountBatchMatMulOperations(op_info, found_unknown_shapes); in CountBatchMatMulOperations()856 CountBatchMatMulOperations(DescribeBatchMatMul({2, 4}, {4, 2}).op_info, in TEST_F()861 CountBatchMatMulOperations( in TEST_F()867 CountBatchMatMulOperations( in TEST_F()
757 int64 OpLevelCostEstimator::CountBatchMatMulOperations( in CountBatchMatMulOperations() function in tensorflow::grappler::OpLevelCostEstimator1313 CountBatchMatMulOperations(op_info, &found_unknown_shapes), op_info); in PredictBatchMatMul()