| /external/tensorflow/tensorflow/core/grappler/costs/ |
| D | utils.h | 44 const std::unordered_map<string, const CostGraphDef::Node*>& name_to_cost,
59 DeviceProperties GetDeviceInfo(const CostGraphDef::Node& node);
73 OpPerformanceList CostGraphToOpPerformanceData(const CostGraphDef& cost_graph,
|
| D | analytical_cost_estimator.cc | 41 gtl::FlatMap<string, CostGraphDef::Node*>* name_to_cost_node, in AddCostNode()
43 CostGraphDef* cost_graph) { in AddCostNode()
46 CostGraphDef::Node* node; in AddCostNode()
175 gtl::FlatMap<string, CostGraphDef::Node*> name_to_cost_node; in PredictCosts()
176 CostGraphDef* cost_graph = nullptr; in PredictCosts()
|
| D | utils.cc | 165 const std::unordered_map<string, const CostGraphDef::Node*>& name_to_cost, in FindInputFeatures()
183 const CostGraphDef::Node* input_cost = it->second; in FindInputFeatures()
187 const CostGraphDef::Node::OutputInfo& output = in FindInputFeatures()
259 DeviceProperties GetDeviceInfo(const CostGraphDef::Node& node) { in GetDeviceInfo()
288 OpPerformanceList CostGraphToOpPerformanceData(const CostGraphDef& cost_graph, in CostGraphToOpPerformanceData()
291 std::unordered_map<string, const CostGraphDef::Node*> name_to_cost; in CostGraphToOpPerformanceData()
310 const CostGraphDef::Node* cost_node = it->second; in CostGraphToOpPerformanceData()
|
| D | measuring_cost_estimator.h | 30 class CostGraphDef; variable
|
| D | analytical_cost_estimator.h | 26 class CostGraphDef; variable
|
| D | measuring_cost_estimator.cc | 56 CostGraphDef* cost_graph = nullptr; in PredictCosts()
|
| D | graph_properties.h | 122 Status InferFromCostGraph(const CostGraphDef& cost_graph);
|
| /external/tensorflow/tensorflow/core/grappler/clusters/ |
| D | single_machine.h | 64 void MergeCosts(CostGraphDef* graph_costs, const CostGraphDef& init_costs,
65 const CostGraphDef& queue_costs);
|
| D | single_machine.cc | 192 CostGraphDef queue_costs; in Run()
380 void SingleMachine::MergeCosts(CostGraphDef* graph_costs, in MergeCosts()
381 const CostGraphDef& init_costs, in MergeCosts()
382 const CostGraphDef& queue_costs) { in MergeCosts()
|
| /external/tensorflow/tensorflow/cc/training/ |
| D | coordinator.h | 39 virtual Status ExportCostGraph(CostGraphDef* cost_graph) const { in ExportCostGraph()
110 Status ExportCostGraph(CostGraphDef* cost_graph) const;
|
| D | queue_runner.h | 86 Status ExportCostGraph(CostGraphDef* cost_graph) const override;
134 std::unique_ptr<CostGraphDef> cost_graph_ TF_GUARDED_BY(cg_mu_);
|
| D | queue_runner.cc | 207 Status QueueRunner::ExportCostGraph(CostGraphDef* cost_graph) const { in ExportCostGraph()
221 cost_graph_.reset(new CostGraphDef()); in SetRunArgumentsAndCostGraph()
|
| D | coordinator.cc | 118 Status Coordinator::ExportCostGraph(CostGraphDef* cost_graph) const { in ExportCostGraph()
|
| /external/tensorflow/tensorflow/python/grappler/ |
| D | cost_analyzer.h | 31 class CostGraphDef; variable
57 void PredictCosts(CostEstimator* cost_estimator, CostGraphDef* cost_graph,
|
| D | cost_analyzer.cc | 44 CostGraphDef* cost_graph, int64* total_time) { in PredictCosts()
62 CostGraphDef cost_graph_measured; in GatherCosts()
68 CostGraphDef cost_graph_analytical; in GatherCosts()
74 CostGraphDef cost_graph_analytical_filtered; in GatherCosts()
75 CostGraphDef cost_graph_measured_filtered; in GatherCosts()
|
| /external/tensorflow/tensorflow/core/graph/ |
| D | costmodel_test.cc | 89 CostGraphDef cost_graph_def; in TEST()
95 absl::flat_hash_map<int32, const CostGraphDef::Node> ids; in TEST()
|
| D | costmodel.cc | 481 CostGraphDef* cost_graph) const { in AddToCostGraphDef()
486 CostGraphDef::Node* cnode = cost_graph->add_node(); in AddToCostGraphDef()
507 CostGraphDef::Node::InputInfo* input_info = cnode->add_input_info(); in AddToCostGraphDef()
513 CostGraphDef::Node::OutputInfo* output_info = cnode->add_output_info(); in AddToCostGraphDef()
|
| D | costmodel.h | 177 void AddToCostGraphDef(const Graph* graph, CostGraphDef* cost_graph) const;
|
| /external/tensorflow/tensorflow/core/distributed_runtime/ |
| D | graph_mgr.h | 170 CostGraphDef* cost_graph,
179 CostGraphDef* cost_graph);
|
| D | message_wrappers.h | 492 virtual CostGraphDef* mutable_cost_graph() = 0;
527 CostGraphDef* mutable_cost_graph() override;
543 CostGraphDef cost_graph_;
560 CostGraphDef* mutable_cost_graph() override;
587 CostGraphDef* mutable_cost_graph() override;
|
| D | graph_mgr.cc | 446 CostGraphDef* cost_graph = nullptr; in ExecuteAsync()
514 CostGraphDef* cost_graph, CancellationManager* cancellation_manager, in StartParallelExecutors()
562 CostGraphDef* cost_graph) { in BuildCostModel()
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | costmodel_manager.h | 46 Status AddToCostGraphDef(const Graph* graph, CostGraphDef* cost_graph);
|
| D | costmodel_manager.cc | 57 CostGraphDef* cost_graph) { in AddToCostGraphDef()
|
| /external/tensorflow/tensorflow/core/util/ |
| D | dump_graph.h | 46 string DumpCostGraphDefToFile(const string& name, CostGraphDef const& graph_def,
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.-run-metadata.pbtxt | 17 type_name: ".tensorflow.CostGraphDef"
|