Searched refs:OpLevelCostEstimator (Results 1 – 9 of 9) sorted by relevance
| /external/tensorflow/tensorflow/core/grappler/costs/ |
| D | op_level_cost_estimator.cc | 218 OpLevelCostEstimator::OpLevelCostEstimator() { in OpLevelCostEstimator() function in tensorflow::grappler::OpLevelCostEstimator
221 typedef Costs (OpLevelCostEstimator::*CostImpl)(const OpContext& op_context) in OpLevelCostEstimator()
230 {kConv2d, wrap(&OpLevelCostEstimator::PredictConv2D)}, in OpLevelCostEstimator()
232 wrap(&OpLevelCostEstimator::PredictConv2DBackpropFilter)}, in OpLevelCostEstimator()
234 wrap(&OpLevelCostEstimator::PredictConv2DBackpropInput)}, in OpLevelCostEstimator()
236 wrap(&OpLevelCostEstimator::PredictFusedConv2DBiasActivation)}, in OpLevelCostEstimator()
240 {kDepthwiseConv2dNative, wrap(&OpLevelCostEstimator::PredictConv2D)}, in OpLevelCostEstimator()
242 wrap(&OpLevelCostEstimator::PredictConv2DBackpropFilter)}, in OpLevelCostEstimator()
244 wrap(&OpLevelCostEstimator::PredictConv2DBackpropInput)}, in OpLevelCostEstimator()
245 {kMatMul, wrap(&OpLevelCostEstimator::PredictMatMul)}, in OpLevelCostEstimator()
[all …]
|
| D | analytical_cost_estimator.h | 47 std::unique_ptr<OpLevelCostEstimator> node_estimator,
68 std::unique_ptr<OpLevelCostEstimator> node_estimator_;
|
| D | op_level_cost_estimator.h | 32 class OpLevelCostEstimator {
34 OpLevelCostEstimator();
35 virtual ~OpLevelCostEstimator() {} in ~OpLevelCostEstimator()
|
| D | analytical_cost_estimator.cc | 112 cluster, absl::make_unique<OpLevelCostEstimator>(), in AnalyticalCostEstimator()
117 Cluster* cluster, std::unique_ptr<OpLevelCostEstimator> node_estimator, in AnalyticalCostEstimator()
|
| D | op_level_cost_estimator_test.cc | 34 class TestOpLevelCostEstimator : public OpLevelCostEstimator {
492 auto dims = OpLevelCostEstimator::OpDimensionsFromInputs( in ValidateOpDimensionsFromImputs()
515 OpLevelCostEstimator estimator_;
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | static_schedule.cc | 31 const GraphProperties& properties, const OpLevelCostEstimator& estimator, in PredictExecutionTime()
96 OpLevelCostEstimator estimator; in EstimateEarliestExecutionTimes()
164 OpLevelCostEstimator estimator; in EstimateRequiredTimes()
|
| /external/tensorflow/tensorflow/core/grappler/clusters/ |
| D | virtual_cluster.cc | 29 : VirtualCluster(devices, absl::make_unique<OpLevelCostEstimator>(), in VirtualCluster()
34 std::unique_ptr<OpLevelCostEstimator> node_estimator, in VirtualCluster()
|
| D | virtual_cluster.h | 38 std::unique_ptr<OpLevelCostEstimator> node_estimator,
|
| /external/tensorflow/tensorflow/python/grappler/ |
| D | cluster.i | 310 tensorflow::grappler::OpLevelCostEstimator estimator; in TF_EstimatePerformance()
|