Home
last modified time | relevance | path

Searched refs:FindNodeByName (Results 1 – 9 of 9) sorted by relevance

/external/tensorflow/tensorflow/compiler/jit/
Dclone_constants_for_better_clustering_test.cc28 using ::tensorflow::testing::FindNodeByName;
82 TF_ASSERT_OK(FindNodeByName(result.get(), "tr0")->input_tensor(1, &tr0_perm)); in TEST()
85 TF_ASSERT_OK(FindNodeByName(result.get(), "tr1")->input_tensor(1, &tr1_perm)); in TEST()
108 TF_ASSERT_OK(FindNodeByName(result.get(), "tr0")->input_tensor(1, &tr0_perm)); in TEST()
111 TF_ASSERT_OK(FindNodeByName(result.get(), "tr1")->input_tensor(1, &tr1_perm)); in TEST()
137 TF_ASSERT_OK(FindNodeByName(result.get(), "tr0")->input_tensor(1, &tr0_perm)); in TEST()
140 TF_ASSERT_OK(FindNodeByName(result.get(), "tr1")->input_tensor(1, &tr1_perm)); in TEST()
168 TF_ASSERT_OK(FindNodeByName(result.get(), "tr0")->input_tensor(1, &tr0_perm)); in TEST()
171 TF_ASSERT_OK(FindNodeByName(result.get(), "tr1")->input_tensor(1, &tr1_perm)); in TEST()
Dpartially_decluster_pass_test.cc105 Node* FindNodeByName(const Graph& graph, const string& name) { in FindNodeByName() function
116 const Node* node = FindNodeByName(graph, node_name); in GetInputsForNode()
319 const Node* n = FindNodeByName(*graph, "shape"); in TEST()
347 const Node* n = FindNodeByName(*graph, "shape"); in TEST()
372 const Node* n = FindNodeByName(*graph, "shape"); in TEST()
402 Node* n = FindNodeByName(*graph, "shape"); in TEST()
435 Node* n = FindNodeByName(*graph, "dynamic_slice"); in TEST()
466 EXPECT_EQ(FindNodeByName(*graph, kClusteredProducer0Name), nullptr); in TEST()
467 EXPECT_EQ(FindNodeByName(*graph, kClusteredProducer1Name), nullptr); in TEST()
Dincrease_dynamism_for_auto_jit_pass_test.cc140 Node* static_shaped_slice = testing::FindNodeByName( in TEST()
160 Node* static_shaped_slice = testing::FindNodeByName( in TEST()
182 Node* static_shaped_slice = testing::FindNodeByName( in TEST()
290 Node* static_shaped_slice = testing::FindNodeByName( in TEST()
346 Node* static_shaped_slice = testing::FindNodeByName( in TEST()
387 Node* static_shaped_slice = testing::FindNodeByName( in TEST()
427 Node* const_0 = testing::FindNodeByName(result.get(), in TEST()
449 Node* slice_node = testing::FindNodeByName(result.get(), "slice"); in TEST()
Dbuild_xla_ops_pass_test.cc48 using ::tensorflow::testing::FindNodeByName;
148 Node* write_op_new = FindNodeByName(graph.get(), write_op->name()); in TEST_F()
201 Node* write_op_new = FindNodeByName(graph.get(), write_op->name()); in TEST_F()
229 Node* write_op_new = FindNodeByName(graph.get(), write_op->name()); in TEST_F()
Dmark_for_compilation_pass_test.cc42 using ::tensorflow::testing::FindNodeByName;
1207 FindNodeByName(graph.get(), "test/x")->set_assigned_device_name(kGPU0); in TEST()
1208 FindNodeByName(graph.get(), "test/y")->set_assigned_device_name(kCPU0); in TEST()
1209 FindNodeByName(graph.get(), "test/z")->set_assigned_device_name(kGPU0); in TEST()
1232 FindNodeByName(graph.get(), "test/x")->set_assigned_device_name(kGPU0); in TEST()
1233 FindNodeByName(graph.get(), "test/y")->set_assigned_device_name(kGPU1); in TEST()
1254 FindNodeByName(graph.get(), "test/x")->set_assigned_device_name(kCPU0); in TEST()
1255 FindNodeByName(graph.get(), "test/y")->set_assigned_device_name(kXLA_GPU0); in TEST()
1278 FindNodeByName(graph.get(), "test/b")->set_assigned_device_name(kCPU0); in TEST()
1279 FindNodeByName(graph.get(), resource_read_name) in TEST()
[all …]
Dnode_matchers.h244 Node* FindNodeByName(Graph* g, absl::string_view name);
Dnode_matchers.cc528 Node* FindNodeByName(Graph* g, absl::string_view name) { in FindNodeByName() function
Dencapsulate_subgraphs_pass_test.cc763 const Node* FindNodeByName(const Graph& graph, const string& name) { in FindNodeByName() function
/external/tensorflow/tensorflow/core/kernels/
Dremote_fused_graph_execute_utils.cc35 const Node* FindNodeByName(const string& name, const Graph& graph) { in FindNodeByName() function
741 queue.emplace_back(FindNodeByName(*remaining_nodes.begin(), graph)); in ClusterizeNodes()