Searched refs:OpLevelCostEstimator (Results 1 – 11 of 11) sorted by relevance
| /external/tensorflow/tensorflow/core/grappler/costs/ |
| D | op_level_cost_estimator.cc | 357 OpLevelCostEstimator::OpLevelCostEstimator() { in OpLevelCostEstimator() function in tensorflow::grappler::OpLevelCostEstimator
360 typedef Status (OpLevelCostEstimator::*CostImpl)(const OpContext& op_context, in OpLevelCostEstimator()
370 wrap(&OpLevelCostEstimator::PredictConv2D)); in OpLevelCostEstimator()
373 wrap(&OpLevelCostEstimator::PredictConv2DBackpropFilter)); in OpLevelCostEstimator()
376 wrap(&OpLevelCostEstimator::PredictConv2DBackpropInput)); in OpLevelCostEstimator()
379 wrap(&OpLevelCostEstimator::PredictFusedConv2DBiasActivation)); in OpLevelCostEstimator()
384 wrap(&OpLevelCostEstimator::PredictConv2D)); in OpLevelCostEstimator()
387 wrap(&OpLevelCostEstimator::PredictConv2DBackpropFilter)); in OpLevelCostEstimator()
390 wrap(&OpLevelCostEstimator::PredictConv2DBackpropInput)); in OpLevelCostEstimator()
392 wrap(&OpLevelCostEstimator::PredictMatMul)); in OpLevelCostEstimator()
[all …]
|
| D | analytical_cost_estimator.h | 46 std::unique_ptr<OpLevelCostEstimator> node_estimator,
51 std::unique_ptr<OpLevelCostEstimator> node_estimator,
71 std::unique_ptr<OpLevelCostEstimator> node_estimator_;
|
| D | analytical_cost_estimator.cc | 114 cluster, absl::make_unique<OpLevelCostEstimator>(), in AnalyticalCostEstimator()
119 Cluster* cluster, std::unique_ptr<OpLevelCostEstimator> node_estimator, in AnalyticalCostEstimator()
133 Cluster* cluster, std::unique_ptr<OpLevelCostEstimator> node_estimator, in AnalyticalCostEstimator()
|
| D | op_level_cost_estimator.h | 91 class OpLevelCostEstimator {
93 OpLevelCostEstimator();
94 virtual ~OpLevelCostEstimator() {} in ~OpLevelCostEstimator()
|
| D | op_level_cost_estimator_test.cc | 37 class TestOpLevelCostEstimator : public OpLevelCostEstimator {
513 using BatchMatMulDimensions = OpLevelCostEstimator::BatchMatMulDimensions;
561 auto dims = OpLevelCostEstimator::OpDimensionsFromInputs( in ValidateOpDimensionsFromInputs()
584 OpLevelCostEstimator estimator_;
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | static_schedule.cc | 33 const GraphProperties& properties, const OpLevelCostEstimator& estimator, in PredictExecutionTime()
101 OpLevelCostEstimator estimator; in EstimateEarliestExecutionTimes()
172 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 | 39 std::unique_ptr<OpLevelCostEstimator> node_estimator,
|
| /external/tensorflow/tensorflow/core/profiler/utils/ |
| D | cost_utils.h | 34 : public tensorflow::grappler::OpLevelCostEstimator {
|
| /external/tensorflow/tensorflow/tools/def_file_filter/ |
| D | symbols_pybind.txt | 300 tensorflow::grappler::OpLevelCostEstimator::OpLevelCostEstimator
301 tensorflow::grappler::OpLevelCostEstimator::PredictCosts
302 tensorflow::grappler::OpLevelCostEstimator::GetDeviceInfo
|
| /external/tensorflow/tensorflow/python/grappler/ |
| D | cluster_wrapper.cc | 249 tensorflow::grappler::OpLevelCostEstimator estimator; in PYBIND11_MODULE()
|