Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/grappler/costs/
Dop_level_cost_estimator.h87 int64 CountBatchMatMulOperations(const OpInfo& op_info,
Dop_level_cost_estimator_test.cc462 int64 CountBatchMatMulOperations(const OpInfo& op_info, in CountBatchMatMulOperations() function in tensorflow::grappler::OpLevelCostEstimatorTest
464 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()
Dop_level_cost_estimator.cc757 int64 OpLevelCostEstimator::CountBatchMatMulOperations( in CountBatchMatMulOperations() function in tensorflow::grappler::OpLevelCostEstimator
1313 CountBatchMatMulOperations(op_info, &found_unknown_shapes), op_info); in PredictBatchMatMul()