| /external/tensorflow/tensorflow/python/grappler/ |
| D | auto_mixed_precision_test.py | 368 return output_val_ref, output_val, metadata.cost_graph
414 output_val_ref, output_val, cost_graph = self._run(mode, a)
415 node_map = _build_node_map(cost_graph.node)
445 output_val_ref, output_val, cost_graph = self._run(mode, output)
446 node_map = _build_node_map(cost_graph.node)
447 num_to_f16, num_to_fp32 = _count_casts(mode, cost_graph.node)
479 output_val_ref, output_val, cost_graph = self._run(mode, output)
480 node_map = _build_node_map(cost_graph.node)
481 num_to_fp16, num_to_fp32 = _count_casts(mode, cost_graph.node)
508 output_val_ref, output_val, cost_graph = self._run(mode, output)
[all …]
|
| D | layout_optimizer_test.py | 294 for node in metadata.cost_graph.node:
332 for node in metadata.cost_graph.node:
365 for node in metadata.cost_graph.node:
400 for node in metadata.cost_graph.node:
432 for node in metadata.cost_graph.node:
462 for node in metadata.cost_graph.node:
494 for node in metadata.cost_graph.node:
525 for node in metadata.cost_graph.node:
556 for node in metadata.cost_graph.node:
586 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/grappler/costs/ |
| D | analytical_cost_estimator.cc | 43 CostGraphDef* cost_graph) { in AddCostNode() argument
52 node = cost_graph->add_node(); in AddCostNode()
176 CostGraphDef* cost_graph = nullptr; in PredictCosts() local
178 cost_graph = run_metadata->mutable_cost_graph(); in PredictCosts()
182 for (auto& node : *cost_graph->mutable_node()) { in PredictCosts()
205 if (cost_graph) { in PredictCosts()
207 &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 | 73 OpPerformanceList CostGraphToOpPerformanceData(const CostGraphDef& cost_graph,
|
| D | graph_properties.h | 122 Status InferFromCostGraph(const CostGraphDef& cost_graph);
|
| D | utils.cc | 288 OpPerformanceList CostGraphToOpPerformanceData(const CostGraphDef& cost_graph, in CostGraphToOpPerformanceData() argument
293 for (auto& node : cost_graph.node()) { in CostGraphToOpPerformanceData()
|
| /external/tensorflow/tensorflow/core/distributed_runtime/ |
| D | graph_mgr.cc | 446 CostGraphDef* cost_graph = nullptr; in ExecuteAsync() local
448 cost_graph = response->mutable_cost_graph(); in ExecuteAsync()
490 handle, step_id, item, rendezvous, ce_handle, collector, cost_graph, in ExecuteAsync()
514 CostGraphDef* cost_graph, CancellationManager* cancellation_manager, in StartParallelExecutors() argument
524 [this, item, collector, cost_graph, step_container, in StartParallelExecutors()
526 BuildCostModel(item, collector, cost_graph); in StartParallelExecutors()
562 CostGraphDef* cost_graph) { in BuildCostModel() argument
573 if (cost_graph != nullptr) { in BuildCostModel()
575 cost_model_manager_.AddToCostGraphDef(unit.graph.get(), cost_graph) in BuildCostModel()
|
| D | graph_mgr.h | 170 CostGraphDef* cost_graph,
179 CostGraphDef* cost_graph);
|
| D | master_session.h | 170 CostGraphDef cost_graph; member
|
| /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 | 293 EXPECT_LE(2, run_metadata.cost_graph().node_size()); in TEST()
294 for (const auto& node : run_metadata.cost_graph().node()) { in TEST()
|
| /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/grappler/clusters/ |
| D | single_machine_test.cc | 88 EXPECT_LE(4, metadata.cost_graph().node_size()); in TEST_F()
89 for (const auto& node : metadata.cost_graph().node()) { in TEST_F()
137 EXPECT_LE(6, metadata1.cost_graph().node_size()); in TEST_F()
138 for (const auto& node : metadata1.cost_graph().node()) { in TEST_F()
152 for (int i = 0; i < metadata1.cost_graph().node_size(); ++i) { in TEST_F()
156 for (int i = 0; i < metadata2.cost_graph().node_size(); ++i) { in TEST_F()
211 for (const auto& node : metadata.cost_graph().node()) { in TEST_F()
384 for (const auto& node : metadata.cost_graph().node()) { in TEST_F()
488 for (const auto& node : metadata.cost_graph().node()) { in TEST_F()
|
| D | virtual_cluster_test.cc | 77 EXPECT_LE(4, metadata.cost_graph().node_size()); in TEST_F()
78 for (const auto& node : metadata.cost_graph().node()) { in TEST_F()
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.-run-metadata.pbtxt | 13 name: "cost_graph"
|
| /external/tensorflow/tensorflow/core/graph/ |
| D | costmodel.cc | 481 CostGraphDef* cost_graph) const { in AddToCostGraphDef()
484 int offset = cost_graph->node_size(); in AddToCostGraphDef()
486 CostGraphDef::Node* cnode = cost_graph->add_node(); in AddToCostGraphDef()
|
| D | costmodel.h | 177 void AddToCostGraphDef(const Graph* graph, CostGraphDef* cost_graph) const;
|