| /external/tensorflow/tensorflow/core/grappler/costs/ |
| 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()
147 gtl::FlatMap<string, CostGraphDef::Node*> name_to_cost_node; in PredictCosts()
148 CostGraphDef* cost_graph = nullptr; in PredictCosts()
|
| D | utils.h | 43 const std::unordered_map<string, const CostGraphDef::Node*>& name_to_cost,
58 DeviceProperties GetDeviceInfo(const CostGraphDef::Node& node);
72 OpPerformanceList CostGraphToOpPerformanceData(const CostGraphDef& cost_graph,
|
| D | utils.cc | 164 const std::unordered_map<string, const CostGraphDef::Node*>& name_to_cost, in FindInputFeatures()
182 const CostGraphDef::Node* input_cost = it->second; in FindInputFeatures()
186 const CostGraphDef::Node::OutputInfo& output = in FindInputFeatures()
258 DeviceProperties GetDeviceInfo(const CostGraphDef::Node& node) { in GetDeviceInfo()
287 OpPerformanceList CostGraphToOpPerformanceData(const CostGraphDef& cost_graph, in CostGraphToOpPerformanceData()
290 std::unordered_map<string, const CostGraphDef::Node*> name_to_cost; in CostGraphToOpPerformanceData()
309 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 | graph_properties.h | 103 Status InferFromCostGraph(const CostGraphDef& cost_graph);
|
| D | measuring_cost_estimator.cc | 56 CostGraphDef* cost_graph = nullptr; in PredictCosts()
|
| /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 | 198 CostGraphDef queue_costs; in Run()
383 void SingleMachine::MergeCosts(CostGraphDef* graph_costs, in MergeCosts()
384 const CostGraphDef& init_costs, in MergeCosts()
385 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_ 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/distributed_runtime/ |
| D | graph_mgr.h | 169 CostGraphDef* cost_graph,
178 CostGraphDef* cost_graph);
|
| D | message_wrappers.h | 480 virtual CostGraphDef* mutable_cost_graph() = 0;
515 CostGraphDef* mutable_cost_graph() override;
531 CostGraphDef cost_graph_;
548 CostGraphDef* mutable_cost_graph() override;
575 CostGraphDef* mutable_cost_graph() override;
|
| D | graph_mgr.cc | 414 CostGraphDef* cost_graph = nullptr; in ExecuteAsync()
472 CostGraphDef* cost_graph, in StartParallelExecutors()
521 CostGraphDef* cost_graph) { in BuildCostModel()
|
| D | message_wrappers.cc | 620 CostGraphDef* InMemoryRunGraphResponse::mutable_cost_graph() { in mutable_cost_graph()
688 CostGraphDef* OwnedProtoRunGraphResponse::mutable_cost_graph() { in mutable_cost_graph()
759 CostGraphDef* NonOwnedProtoRunGraphResponse::mutable_cost_graph() { in mutable_cost_graph()
|
| /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/graph/ |
| D | costmodel.cc | 479 CostGraphDef* cost_graph) const { in AddToCostGraphDef()
483 CostGraphDef::Node* cnode = cost_graph->add_node(); in AddToCostGraphDef()
504 CostGraphDef::Node::InputInfo* input_info = cnode->add_input_info(); in AddToCostGraphDef()
510 CostGraphDef::Node::OutputInfo* output_info = cnode->add_output_info(); in AddToCostGraphDef()
|
| D | costmodel.h | 169 void AddToCostGraphDef(const Graph* graph, CostGraphDef* cost_graph) const;
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.-run-metadata.pbtxt | 17 type_name: ".tensorflow.CostGraphDef"
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | cost_graph.proto | 12 message CostGraphDef { message
|