| /external/tensorflow/tensorflow/core/grappler/costs/ |
| D | op_level_cost_estimator_test.cc | 453 Costs PredictCosts(const OpContext& op_context) const { in PredictCosts() function in tensorflow::grappler::OpLevelCostEstimatorTest
454 return estimator_.PredictCosts(op_context); in PredictCosts()
528 auto cost = estimator_.PredictCosts(op_context); in TEST_F()
548 auto cost = estimator_.PredictCosts(op_context); in TEST_F()
566 auto cost = estimator_.PredictCosts(op_context); in TEST_F()
586 auto cost = estimator_.PredictCosts(op_context); in TEST_F()
596 auto cost = PredictCosts(DescribeBiasAdd(1000, 10)); in TEST_F()
606 auto cost = PredictCosts(DescribeConvolution(16, 19, 19, 48, 48, 5, 5, 256)); in TEST_F()
617 PredictCosts(DescribeDepthwiseConv2dNative(16, 19, 19, 48, 48, 5, 5, 3)); in TEST_F()
627 auto cost = PredictCosts(DescribeBinaryOp("Dummy", 1000, 1)); in TEST_F()
[all …]
|
| D | analytical_cost_estimator.cc | 135 Status AnalyticalCostEstimator::PredictCosts(const GraphDef& optimized_graph, in PredictCosts() function in tensorflow::grappler::AnalyticalCostEstimator
167 node_costs = node_estimator_->PredictCosts(op_context); in PredictCosts()
|
| D | measuring_cost_estimator.h | 61 Status PredictCosts(const GraphDef& optimized_graph,
|
| D | analytical_cost_estimator.h | 60 Status PredictCosts(const GraphDef& optimized_graph,
|
| D | measuring_cost_estimator.cc | 53 Status MeasuringCostEstimator::PredictCosts(const GraphDef& optimized_graph, in PredictCosts() function in tensorflow::grappler::MeasuringCostEstimator
|
| D | analytical_cost_estimator_test.cc | 103 TF_ASSERT_OK(estimator.PredictCosts(item.graph, &run_metadata, &summary)); in TEST_F()
|
| D | op_level_cost_estimator.h | 37 virtual Costs PredictCosts(const OpContext& op_context) const;
|
| D | cost_estimator.h | 231 virtual Status PredictCosts(const GraphDef& optimized_graph,
|
| D | op_level_cost_estimator.cc | 373 Costs OpLevelCostEstimator::PredictCosts(const OpContext& op_context) const { in PredictCosts() function in tensorflow::grappler::OpLevelCostEstimator
1387 auto op_cost = PredictCosts(fused_op); in PredictFusedOp()
|
| /external/tensorflow/tensorflow/python/grappler/ |
| D | cost_analyzer.cc | 43 void CostAnalyzer::PredictCosts(CostEstimator* cost_estimator, in PredictCosts() function in tensorflow::grappler::CostAnalyzer
49 cost_estimator->PredictCosts(item_->graph, &run_metadata, &costs); in PredictCosts()
63 PredictCosts(&measure_estimator_, &cost_graph_measured, in GatherCosts()
69 PredictCosts(&analytical_estimator_, &cost_graph_analytical, in GatherCosts()
|
| D | cost_analyzer.h | 57 void PredictCosts(CostEstimator* cost_estimator, CostGraphDef* cost_graph,
|
| D | cluster.i | 181 cost_measure->PredictCosts(item.graph, &run_metadata, costs)); in _GetOpPerformanceDataAndRunTime()
|
| /external/tensorflow/tensorflow/core/grappler/clusters/ |
| D | virtual_cluster.cc | 87 estimator_->PredictCosts(item.graph, metadata, &ignored_costs)); in Run()
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | static_schedule.cc | 53 estimator.PredictCosts(op_context).execution_time; in PredictExecutionTime()
|