Home
last modified time | relevance | path

Searched refs:BuildNodeNameIndex (Results 1 – 18 of 18) sorted by relevance

/external/tensorflow/tensorflow/compiler/jit/
Dextract_outside_compilation_pass_test.cc54 auto node_name_image = g->BuildNodeNameIndex(); in TEST()
66 node_name_image = g->BuildNodeNameIndex(); in TEST()
137 auto node_name_image = g->BuildNodeNameIndex(); in TEST()
163 auto node_name_image = g->BuildNodeNameIndex(); in TEST()
187 auto node_name_image = g->BuildNodeNameIndex(); in TEST()
206 auto node_name_image = g->BuildNodeNameIndex(); in TEST()
219 node_name_image = g->BuildNodeNameIndex(); in TEST()
279 auto node_name_image = g->BuildNodeNameIndex(); in TEST_F()
307 auto node_name_index = xla_fbody->graph->BuildNodeNameIndex(); in TEST_F()
439 auto node_name_image = g->BuildNodeNameIndex(); in TEST_F()
[all …]
Dencapsulate_util_test.cc43 auto node_index = g.BuildNodeNameIndex(); in TEST()
Dencapsulate_util.cc211 auto node_name_index = g->BuildNodeNameIndex(); in PostprocessDataEdgesBetweenOutsideCompilations()
277 auto node_name_index = g->BuildNodeNameIndex(); in PostprocessControlEdgesBetweenOutsideCompilations()
331 auto node_name_index = g->BuildNodeNameIndex(); in PerformStaticShapeInferenceBeforeEncapsulation()
Dencapsulate_xla_computations_pass_test.cc258 std::unordered_map<string, Node*> index = graph->BuildNodeNameIndex(); in TEST()
294 graph_copy->BuildNodeNameIndex(); in TEST()
Drearrange_function_argument_pass_test.cc136 auto node_name_index = g->BuildNodeNameIndex(); in TEST()
Dextract_outside_compilation_pass.cc2510 auto node_name_index = g->BuildNodeNameIndex(); in ExtractOutsideCompilation()
/external/tensorflow/tensorflow/core/common_runtime/
Dconstant_folding_test.cc125 std::unordered_map<string, Node*> index = g.BuildNodeNameIndex(); in TEST_F()
180 auto index = g2.BuildNodeNameIndex(); in TEST_F()
203 std::unordered_map<string, Node*> index = g.BuildNodeNameIndex(); in TEST_F()
232 std::unordered_map<string, Node*> index = g.BuildNodeNameIndex(); in TEST_F()
260 std::unordered_map<string, Node*> index = g.BuildNodeNameIndex(); in TEST_F()
292 std::unordered_map<string, Node*> index = g.BuildNodeNameIndex(); in TEST_F()
427 std::unordered_map<string, Node*> index = g.BuildNodeNameIndex(); in TEST_F()
469 std::unordered_map<string, Node*> orig_index = g.BuildNodeNameIndex(); in TEST_F()
488 std::unordered_map<string, Node*> index = g.BuildNodeNameIndex(); in TEST_F()
548 std::unordered_map<string, Node*> orig_index = g.BuildNodeNameIndex(); in TEST_F()
[all …]
Dplacer_inspection_required_ops_utils_test.cc43 std::unordered_map<string, Node*> node_map = graph.BuildNodeNameIndex(); in VerifyPlacerInspectionRequiredOps()
Dpartitioning_utils_test.cc214 std::unordered_map<string, Node*> nodes = graph->BuildNodeNameIndex(); in TEST_F()
/external/tensorflow/tensorflow/compiler/tf2xla/
Dtf2xla_util_test.cc406 auto node_name_index = graph.BuildNodeNameIndex(); in TEST()
497 auto node_name_index = bwd_fbody->graph->BuildNodeNameIndex(); in TEST()
Dresource_util.cc143 fbody.graph->BuildNodeNameIndex(); in UpdateResourceUsageFromFunctionBodyAnalysis()
Dxla_compiler.cc645 auto node_name_index = graph->BuildNodeNameIndex(); in GetGraph()
668 auto node_name_index = graph->BuildNodeNameIndex(); in GetGraph()
Dxla_compiler_test.cc1762 auto node_name_index = graph->BuildNodeNameIndex(); in TEST_F()
/external/tensorflow/tensorflow/core/graph/
Dgraph_test.cc648 TEST_F(GraphTest, BuildNodeNameIndex) { in TEST_F() argument
654 auto node_name_index = graph_.BuildNodeNameIndex(); in TEST_F()
Dgraph.h703 std::unordered_map<string, Node*> BuildNodeNameIndex() const;
Dgraph.cc870 std::unordered_map<string, Node*> Graph::BuildNodeNameIndex() const { in BuildNodeNameIndex() function in tensorflow::Graph
/external/tensorflow/tensorflow/core/tpu/graph_rewrite/
Dhost_training_loop_optimization_util.cc442 const auto node_name_map = graph->BuildNodeNameIndex(); in AddReshardOp()
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/translate/
Dimport_model.cc592 std::unordered_map<string, Node*> node_map = from.BuildNodeNameIndex(); in CopyStackTraces()
1399 auto node_name_map = graph_->BuildNodeNameIndex(); in PrepareConvert()