Home
last modified time | relevance | path

Searched refs:node_names (Results 1 – 25 of 26) sorted by relevance

12

/external/tensorflow/tensorflow/python/debug/lib/
Ddebug_utils_test.py87 node_names = []
89 node_names.append(watch.node_name)
95 return node_names
201 node_names = self._verify_watches(debug_watch_opts, 0,
206 self.assertTrue("a1_init" in node_names)
207 self.assertTrue("a1" in node_names)
208 self.assertTrue("a1/Assign" in node_names)
209 self.assertTrue("a1/read" in node_names)
211 self.assertTrue("b_init" in node_names)
212 self.assertTrue("b" in node_names)
[all …]
Dsession_debug_testlib.py766 node_names = dump.nodes(
769 node_names = dump.nodes()
770 self.assertTrue(u_name in node_names)
771 self.assertTrue(u_read_name in node_names)
/external/tensorflow/tensorflow/python/debug/cli/
Dstepper_cli_test.py57 node_names = []
64 node_names.append(line.split(" ")[-1])
75 return node_names, status_labels, node_pointer
161 def _assert_nodes_topologically_sorted_with_target_e(self, node_names): argument
164 self.assertGreaterEqual(len(node_names), 7)
165 self.assertLess(node_names.index("a"), node_names.index("a/read"))
166 self.assertLess(node_names.index("b"), node_names.index("b/read"))
167 self.assertLess(node_names.index("a/read"), node_names.index("c"))
168 self.assertLess(node_names.index("b/read"), node_names.index("c"))
169 self.assertLess(node_names.index("a/read"), node_names.index("d"))
[all …]
/external/tensorflow/tensorflow/compiler/tf2tensorrt/segment/
Dsegment_test.cc40 const std::set<string>& node_names) { in MakeCandidateFn() argument
41 return [node_names](const Node* node) -> Status { in MakeCandidateFn()
42 if (node_names.find(node->name()) != node_names.end()) { in MakeCandidateFn()
50 const std::set<string>& node_names) { in MakeInputEdgeCandidateFn() argument
51 return [node_names](const Edge* in_edge) -> bool { in MakeInputEdgeCandidateFn()
52 return node_names.find(in_edge->dst()->name()) != node_names.end(); in MakeInputEdgeCandidateFn()
57 const std::set<string>& node_names) { in MakeOutputEdgeCandidateFn() argument
58 return [node_names](const Edge* out_edge) -> bool { in MakeOutputEdgeCandidateFn()
59 return node_names.find(out_edge->src()->name()) != node_names.end(); in MakeOutputEdgeCandidateFn()
/external/tensorflow/tensorflow/core/distributed_runtime/rpc/
Dgrpc_session_test.cc44 void CreateGraphDef(GraphDef* graph_def, string node_names[3]) { in CreateGraphDef()
50 node_names[0] = a->name(); in CreateGraphDef()
55 node_names[1] = b->name(); in CreateGraphDef()
58 node_names[2] = c->name(); in CreateGraphDef()
89 string node_names[3]; in TEST() local
91 CreateGraphDef(&graph, node_names); in TEST()
105 std::vector<string> targets = {node_names[2]}; in TEST()
111 std::vector<string> names = {node_names[2] + ":0"}; in TEST()
112 std::vector<string> targets = {node_names[1]}; in TEST()
125 string node_names[3]; in TEST() local
[all …]
/external/tensorflow/tensorflow/core/debug/
Dgrpc_session_debug_test.cc52 void CreateGraphDef(GraphDef* graph_def, string node_names[3]) { in CreateGraphDef()
58 node_names[0] = a->name(); in CreateGraphDef()
63 node_names[1] = b->name(); in CreateGraphDef()
67 node_names[2] = c->name(); in CreateGraphDef()
153 string node_names[3]; in TEST_F() local
154 CreateGraphDef(&graph, node_names); in TEST_F()
171 watch->set_node_name(node_names[0]); in TEST_F()
178 watch->set_node_name(node_names[1]); in TEST_F()
188 session->Run(options, {}, {node_names[2]}, {}, &outputs, &metadata)); in TEST_F()
Ddebug_grpc_io_utils_test.cc119 ASSERT_EQ(1, server_data.server->node_names.size()); in TEST_F()
123 EXPECT_EQ(kDebugNodeKey.node_name, server_data.server->node_names[0]); in TEST_F()
139 ASSERT_EQ(1, server_data_.server->node_names.size()); in TEST_F()
143 EXPECT_EQ(kDebugNodeKey.node_name, server_data_.server->node_names[0]); in TEST_F()
246 ASSERT_EQ(kSends, server_data_.server->node_names.size()); in TEST_F()
247 for (size_t i = 0; i < server_data_.server->node_names.size(); ++i) { in TEST_F()
249 str_util::Split(server_data_.server->node_names[i], '_'); in TEST_F()
293 ASSERT_EQ(1, server_data_.server->node_names.size()); in TEST_F()
298 EXPECT_EQ(kDebugNodeKey.node_name, server_data_.server->node_names[0]); in TEST_F()
303 ASSERT_EQ(0, server_data_.server->node_names.size()); in TEST_F()
[all …]
Ddebug_grpc_testlib.cc74 node_names.push_back(node_name); in SendEvents()
125 node_names.clear(); in ClearReceivedDebugData()
Ddebug_grpc_testlib.h54 std::vector<string> node_names; variable
/external/tensorflow/tensorflow/core/grappler/utils/
Dgrappler_test.cc91 const GraphDef& graph, const std::vector<string>& node_names) const { in EvaluateNodes()
92 return EvaluateNodes(graph, node_names, {}); in EvaluateNodes()
96 const GraphDef& graph, const std::vector<string>& node_names, in EvaluateNodes() argument
102 TF_CHECK_OK(session->Run(run_options, inputs, node_names, node_names, in EvaluateNodes()
Dgrappler_test.h39 const GraphDef& graph, const std::vector<string>& node_names) const;
42 const GraphDef& graph, const std::vector<string>& node_names,
/external/tensorflow/tensorflow/core/framework/
Dgraph_to_functiondef.cc122 NodeNameMapping node_names; in GraphToFunctionDef() local
136 const string normalized = node_names.Normalize(node->name()); in GraphToFunctionDef()
153 const string normalized = node_names.Normalize(node->name()); in GraphToFunctionDef()
167 node_def->set_name(node_names.Uniquify(node->name())); in GraphToFunctionDef()
237 node_names.Renormalize(node_def->input(i).substr(1)); in GraphToFunctionDef()
/external/tensorflow/tensorflow/c/
Dc_api_function.cc161 const string& fn_name, const NodeNameMapping& node_names, in FillFunctionBody() argument
179 node_def->set_name(node_names.Lookup(node->name())); in FillFunctionBody()
235 const string normalized = node_names.Lookup(edge->src()->name()); in FillFunctionBody()
317 NodeNameMapping node_names; in GraphToFunctionDef() local
340 TF_RETURN_IF_ERROR(node_names.UseOutputName(output_names[i])); in GraphToFunctionDef()
343 argdef->set_name(node_names.GetOutputName(node->name())); in GraphToFunctionDef()
353 const string& input_name = node_names.GetInputName(node->name()); in GraphToFunctionDef()
368 const string& node_name = node_names.Uniquify(node->name()); in GraphToFunctionDef()
393 FillFunctionBody(fn_name, node_names, body_nodes, tensor_renaming, fdef)); in GraphToFunctionDef()
/external/tensorflow/tensorflow/core/kernels/
Dremote_fused_graph_execute_utils_test.cc482 const std::unordered_set<string>& node_names = std::get<0>(cluster); in TEST() local
485 EXPECT_EQ(1, node_names.size()) << IterToString(node_names); in TEST()
490 EXPECT_EQ(3, node_names.size()) << IterToString(node_names); in TEST()
495 EXPECT_EQ(5, node_names.size()) << IterToString(node_names); in TEST()
500 EXPECT_EQ(6, node_names.size()) << IterToString(node_names); in TEST()
505 EXPECT_EQ(2, node_names.size()) << IterToString(node_names); in TEST()
Dremote_fused_graph_execute_utils.cc728 const std::unordered_set<string>& node_names, const GraphDef& graph_def, in ClusterizeNodes() argument
733 std::unordered_set<string> remaining_nodes = node_names; in ClusterizeNodes()
747 if (node_names.count(node_name) > 0) { in ClusterizeNodes()
769 for (const string& node_name : node_names) { in ClusterizeNodes()
776 node_names.count(src_node->name()) <= 0 && !src_node->IsSource(); in ClusterizeNodes()
798 const bool dst_is_outside = node_names.count(dst_node->name()) <= 0; in ClusterizeNodes()
831 const std::unordered_set<string>& node_names = std::get<0>(cluster); in BuildClusterSubgraphDef() local
840 if (node != nullptr && node_names.count(node->name()) <= 0 && in BuildClusterSubgraphDef()
Dremote_fused_graph_execute_utils.h197 static Status ClusterizeNodes(const std::unordered_set<string>& node_names,
/external/tensorflow/tensorflow/python/data/experimental/kernel_tests/optimization/
Dmap_vectorization_test.py349 def _make_dataset(node_names): argument
350 dataset = base_dataset.apply(optimization.assert_next(node_names))
419 def _make_dataset(node_names): argument
420 dataset = base_dataset.apply(optimization.assert_next(node_names))
469 def make_dataset(node_names): argument
470 dataset = base_dataset.apply(optimization.assert_next(node_names))
/external/tensorflow/tensorflow/core/grappler/optimizers/
Dgraph_optimizer_stage.cc115 const std::vector<string> node_names, in MakeOptimizedNodeName() argument
119 for (const string& node_name : node_names) { in MakeOptimizedNodeName()
Dgraph_optimizer_stage.h102 const std::vector<string> node_names,
/external/tensorflow/tensorflow/python/debug/wrappers/
Dlocal_cli_wrapper_test.py426 node_names = [datum.node_name for datum in debug_dump.dumped_tensor_data]
429 node_names)
453 node_names = [datum.node_name for datum in debug_dump.dumped_tensor_data]
454 self.assertItemsEqual(["callable_a", "callable_b"], node_names)
480 node_names = [datum.node_name for datum in debug_dump.dumped_tensor_data]
481 self.assertItemsEqual(["callable_a", "callable_b"], node_names)
508 node_names = [datum.node_name for datum in debug_dump.dumped_tensor_data]
511 node_names)
/external/swiftshader/third_party/LLVM/lib/Target/CellSPU/
DSPUISelLowering.cpp41 std::map<unsigned, const char *> node_names; variable
468 if (node_names.empty()) { in getTargetNodeName()
469 node_names[(unsigned) SPUISD::RET_FLAG] = "SPUISD::RET_FLAG"; in getTargetNodeName()
470 node_names[(unsigned) SPUISD::Hi] = "SPUISD::Hi"; in getTargetNodeName()
471 node_names[(unsigned) SPUISD::Lo] = "SPUISD::Lo"; in getTargetNodeName()
472 node_names[(unsigned) SPUISD::PCRelAddr] = "SPUISD::PCRelAddr"; in getTargetNodeName()
473 node_names[(unsigned) SPUISD::AFormAddr] = "SPUISD::AFormAddr"; in getTargetNodeName()
474 node_names[(unsigned) SPUISD::IndirectAddr] = "SPUISD::IndirectAddr"; in getTargetNodeName()
475 node_names[(unsigned) SPUISD::LDRESULT] = "SPUISD::LDRESULT"; in getTargetNodeName()
476 node_names[(unsigned) SPUISD::CALL] = "SPUISD::CALL"; in getTargetNodeName()
[all …]
/external/curl/tests/python_dependencies/impacket/
Dnmb.py212 self.node_names = [ ]
224 self.node_names.append(NBNodeEntry(name, type ,flags))
245 return self.node_names
246 def add_node_name(self,node_names): argument
247 self.node_names.append(node_names)
252 res += self.node_names[i].rawData()
/external/tensorflow/tensorflow/python/compiler/tensorrt/test/
Dtf_trt_integration_test_base.py371 for engine_name, node_names in expected_engines.items():
372 for node_name in node_names:
/external/tensorflow/tensorflow/core/graph/
Dgraph_test.cc656 std::vector<string> node_names{"_SOURCE", "_SINK", "A", "B", "C"}; in TEST_F() local
657 for (const string& node_name : node_names) { in TEST_F()
/external/tensorflow/tensorflow/core/grappler/costs/
Dvirtual_scheduler_test.cc1657 std::vector<string> node_names; in ValidateNodeDefs() local
1659 std::back_inserter(node_names), in ValidateNodeDefs()
1661 ExpectVectorEq(expected, node_names); in ValidateNodeDefs()

12