Home
last modified time | relevance | path

Searched defs:CountBatchMatMulOperations (Results 1 – 2 of 2) 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
530 int64_t CountBatchMatMulOperations(const OpInfo& op_info, in CountBatchMatMulOperations() function in tensorflow::grappler::OpLevelCostEstimatorTest
637 int64_t CountBatchMatMulOperations(const OpInfo& op_info, in CountBatchMatMulOperations() function in tensorflow::grappler::OpLevelBatchMatMulCostEstimatorTest
H A Dop_level_cost_estimator.cc1246 int64_t OpLevelCostEstimator::CountBatchMatMulOperations( in CountBatchMatMulOperations() function in tensorflow::grappler::OpLevelCostEstimator
1251 int64_t OpLevelCostEstimator::CountBatchMatMulOperations( in CountBatchMatMulOperations() function in tensorflow::grappler::OpLevelCostEstimator