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.h243 static int64 CountBatchMatMulOperations(const OpInfo& op_info,
245 static int64 CountBatchMatMulOperations(const OpInfo& op_info,
Dop_level_cost_estimator_test.cc524 int64 CountBatchMatMulOperations(const OpInfo& op_info, in CountBatchMatMulOperations() function in tensorflow::grappler::OpLevelCostEstimatorTest
526 return estimator_.CountBatchMatMulOperations(op_info, found_unknown_shapes); in CountBatchMatMulOperations()
529 int64 CountBatchMatMulOperations(const OpInfo& op_info, in CountBatchMatMulOperations() function in tensorflow::grappler::OpLevelCostEstimatorTest
532 return estimator_.CountBatchMatMulOperations(op_info, batch_mat_mul, in CountBatchMatMulOperations()
603 int64 CountBatchMatMulOperations(const OpInfo& op_info, in CountBatchMatMulOperations() function in tensorflow::grappler::OpLevelBatchMatMulCostEstimatorTest
605 return OpLevelCostEstimatorTest::CountBatchMatMulOperations( in CountBatchMatMulOperations()
617 OpLevelCostEstimatorTest::CountBatchMatMulOperations( in CountBatchMatMulDimProduct()
1207 CountBatchMatMulOperations(DescribeBatchMatMul({2, 4}, {4, 2}).op_info, in TEST_P()
1212 CountBatchMatMulOperations( in TEST_P()
1218 CountBatchMatMulOperations( in TEST_P()
Dop_level_cost_estimator.cc1234 int64 OpLevelCostEstimator::CountBatchMatMulOperations( in CountBatchMatMulOperations() function in tensorflow::grappler::OpLevelCostEstimator
1236 return CountBatchMatMulOperations(op_info, nullptr, found_unknown_shapes); in CountBatchMatMulOperations()
1239 int64 OpLevelCostEstimator::CountBatchMatMulOperations( in CountBatchMatMulOperations() function in tensorflow::grappler::OpLevelCostEstimator
1929 CountBatchMatMulOperations(op_info, &found_unknown_shapes); in PredictBatchMatMul()