| /external/tensorflow/tensorflow/core/grappler/costs/ |
| D | analytical_cost_estimator.cc | 43 CostGraphDef* cost_graph) { in AddCostNode() argument
52 node = cost_graph->add_node(); in AddCostNode()
148 CostGraphDef* cost_graph = nullptr; in PredictCosts() local
150 cost_graph = run_metadata->mutable_cost_graph(); in PredictCosts()
154 for (auto& node : *cost_graph->mutable_node()) { in PredictCosts()
177 if (cost_graph) { in PredictCosts()
179 &name_to_cost_node, &name_to_id, cost_graph); in PredictCosts()
|
| D | measuring_cost_estimator.cc | 56 CostGraphDef* cost_graph = nullptr; in PredictCosts() local
58 cost_graph = run_metadata->mutable_cost_graph(); in PredictCosts()
107 if (cost_graph && (step + 1 == measurement_steps_)) { in PredictCosts()
108 metadata.mutable_cost_graph()->Swap(cost_graph); in PredictCosts()
|
| D | utils.h | 72 OpPerformanceList CostGraphToOpPerformanceData(const CostGraphDef& cost_graph,
|
| D | graph_properties.h | 103 Status InferFromCostGraph(const CostGraphDef& cost_graph);
|
| /external/tensorflow/tensorflow/python/grappler/ |
| D | layout_optimizer_test.py | 273 for node in metadata.cost_graph.node:
311 for node in metadata.cost_graph.node:
344 for node in metadata.cost_graph.node:
379 for node in metadata.cost_graph.node:
411 for node in metadata.cost_graph.node:
441 for node in metadata.cost_graph.node:
473 for node in metadata.cost_graph.node:
504 for node in metadata.cost_graph.node:
535 for node in metadata.cost_graph.node:
565 for node in metadata.cost_graph.node:
[all …]
|
| D | cost_analyzer.cc | 44 CostGraphDef* cost_graph, int64* total_time) { in PredictCosts() argument
50 if (cost_graph) { in PredictCosts()
51 cost_graph->Swap(run_metadata.mutable_cost_graph()); in PredictCosts()
|
| D | cost_analyzer.h | 57 void PredictCosts(CostEstimator* cost_estimator, CostGraphDef* cost_graph,
|
| /external/tensorflow/tensorflow/core/distributed_runtime/ |
| D | graph_mgr.cc | 414 CostGraphDef* cost_graph = nullptr; in ExecuteAsync() local
416 cost_graph = response->mutable_cost_graph(); in ExecuteAsync()
456 handle, step_id, item, rendezvous, ce_handle, collector, cost_graph, in ExecuteAsync()
472 CostGraphDef* cost_graph, in StartParallelExecutors() argument
483 [this, item, collector, cost_graph, step_container, in StartParallelExecutors()
485 BuildCostModel(item, collector, cost_graph); in StartParallelExecutors()
521 CostGraphDef* cost_graph) { in BuildCostModel() argument
532 if (cost_graph != nullptr) { in BuildCostModel()
534 cost_model_manager_.AddToCostGraphDef(unit.graph, cost_graph) in BuildCostModel()
|
| D | graph_mgr.h | 169 CostGraphDef* cost_graph,
178 CostGraphDef* cost_graph);
|
| /external/tensorflow/tensorflow/core/grappler/clusters/ |
| D | single_machine_test.cc | 82 EXPECT_LE(4, metadata.cost_graph().node_size()); in TEST_F()
83 for (const auto& node : metadata.cost_graph().node()) { in TEST_F()
126 EXPECT_LE(6, metadata1.cost_graph().node_size()); in TEST_F()
127 for (const auto& node : metadata1.cost_graph().node()) { in TEST_F()
139 for (int i = 0; i < metadata1.cost_graph().node_size(); ++i) { in TEST_F()
143 for (int i = 0; i < metadata2.cost_graph().node_size(); ++i) { in TEST_F()
198 for (const auto& node : metadata.cost_graph().node()) { in TEST_F()
370 for (const auto& node : metadata.cost_graph().node()) { in TEST_F()
474 for (const auto& node : metadata.cost_graph().node()) { in TEST_F()
|
| D | virtual_cluster_test.cc | 76 EXPECT_LE(4, metadata.cost_graph().node_size()); in TEST_F()
77 for (const auto& node : metadata.cost_graph().node()) { in TEST_F()
|
| /external/tensorflow/tensorflow/cc/training/ |
| D | coordinator.h | 39 virtual Status ExportCostGraph(CostGraphDef* cost_graph) const { in ExportCostGraph() argument
110 Status ExportCostGraph(CostGraphDef* cost_graph) const;
|
| D | coordinator.cc | 118 Status Coordinator::ExportCostGraph(CostGraphDef* cost_graph) const { in ExportCostGraph()
121 Status s = t->ExportCostGraph(cost_graph); in ExportCostGraph()
|
| D | queue_runner_test.cc | 386 CostGraphDef cost_graph; in TEST() local
387 TF_CHECK_OK(qr->ExportCostGraph(&cost_graph)); in TEST()
388 EXPECT_TRUE(cost_graph.node_size() > 0); in TEST()
404 CostGraphDef cost_graph; in TEST() local
405 EXPECT_EQ(qr->ExportCostGraph(&cost_graph).code(), in TEST()
|
| D | queue_runner.cc | 207 Status QueueRunner::ExportCostGraph(CostGraphDef* cost_graph) const { in ExportCostGraph()
213 cost_graph->MergeFrom(*cost_graph_); in ExportCostGraph()
|
| D | queue_runner.h | 86 Status ExportCostGraph(CostGraphDef* cost_graph) const override;
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | costmodel_manager.cc | 57 CostGraphDef* cost_graph) { in AddToCostGraphDef() argument
65 cost_model->AddToCostGraphDef(graph, cost_graph); in AddToCostGraphDef()
|
| D | costmodel_manager.h | 46 Status AddToCostGraphDef(const Graph* graph, CostGraphDef* cost_graph);
|
| D | direct_session_with_tracking_alloc_test.cc | 302 EXPECT_LE(2, run_metadata.cost_graph().node_size()); in TEST()
303 for (const auto& node : run_metadata.cost_graph().node()) { in TEST()
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.-run-metadata.pbtxt | 13 name: "cost_graph"
|
| /external/tensorflow/tensorflow/contrib/makefile/ |
| D | tf_pb_text_files.txt | 6 tensorflow/core/framework/cost_graph.pb_text.cc
|
| D | tf_proto_files.txt | 10 tensorflow/core/framework/cost_graph.proto
|
| D | proto_text_pb_cc_files.txt | 6 tensorflow/core/framework/cost_graph.pb.cc
|
| D | proto_text_pb_h_files.txt | 6 tensorflow/core/framework/cost_graph.pb.h
|
| /external/tensorflow/tensorflow/core/graph/ |
| D | costmodel.h | 169 void AddToCostGraphDef(const Graph* graph, CostGraphDef* cost_graph) const;
|