Home
last modified time | relevance | path

Searched refs:CountBatchMatMulOperations (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/grappler/costs/
H A Dop_level_cost_estimator_test.cc525 int64_t CountBatchMatMulOperations(const OpInfo& op_info, in CountBatchMatMulOperations() function in tensorflow::grappler::OpLevelCostEstimatorTest
527 return estimator_.CountBatchMatMulOperations(op_info, found_unknown_shapes); in CountBatchMatMulOperations()
530 int64_t CountBatchMatMulOperations(const OpInfo& op_info, in CountBatchMatMulOperations() function in tensorflow::grappler::OpLevelCostEstimatorTest
533 return estimator_.CountBatchMatMulOperations(op_info, batch_mat_mul, in CountBatchMatMulOperations()
637 int64_t CountBatchMatMulOperations(const OpInfo& op_info, in CountBatchMatMulOperations() function in tensorflow::grappler::OpLevelBatchMatMulCostEstimatorTest
639 return OpLevelCostEstimatorTest::CountBatchMatMulOperations( in CountBatchMatMulOperations()
651 OpLevelCostEstimatorTest::CountBatchMatMulOperations( in CountBatchMatMulDimProduct()
1241 CountBatchMatMulOperations(DescribeBatchMatMul({2, 4}, {4, 2}).op_info, in TEST_P()
1246 CountBatchMatMulOperations( in TEST_P()
1252 CountBatchMatMulOperations( in TEST_P()
H A Dop_level_cost_estimator.h243 static int64_t CountBatchMatMulOperations(const OpInfo& op_info,
245 static int64_t CountBatchMatMulOperations(
H A Dop_level_cost_estimator.cc1246 int64_t OpLevelCostEstimator::CountBatchMatMulOperations( in CountBatchMatMulOperations() function in tensorflow::grappler::OpLevelCostEstimator
1248 return CountBatchMatMulOperations(op_info, nullptr, found_unknown_shapes); in CountBatchMatMulOperations()
1251 int64_t OpLevelCostEstimator::CountBatchMatMulOperations( in CountBatchMatMulOperations() function in tensorflow::grappler::OpLevelCostEstimator
1963 CountBatchMatMulOperations(op_info, &found_unknown_shapes); in PredictBatchMatMul()