| /external/tensorflow/tensorflow/python/debug/lib/ |
| D | debug_utils_test.py | 89 node_names = []
91 node_names.append(watch.node_name)
102 return node_names
208 node_names = self._verify_watches(debug_watch_opts, 0,
213 self.assertIn("a1_init", node_names)
214 self.assertIn("a1", node_names)
215 self.assertIn("a1/Assign", node_names)
216 self.assertIn("a1/read", node_names)
218 self.assertIn("b_init", node_names)
219 self.assertIn("b", node_names)
[all …]
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/translate/ |
| D | mlir_roundtrip_flags.cc | 54 std::vector<string> node_names = absl::StrSplit(array_names, ','); in ParseInputArrayInfo() local
70 return ParseInputArrayInfo(node_names, node_dtypes, node_shapes, inputs); in ParseInputArrayInfo()
73 Status ParseInputArrayInfo(const std::vector<string>& node_names, in ParseInputArrayInfo() argument
82 used_node_dtypes.resize(node_names.size(), DataType_Name(DT_INVALID)); in ParseInputArrayInfo()
83 } else if (node_names.size() == node_dtypes.size()) { in ParseInputArrayInfo()
97 node_names.size(), ", #data_types ", node_dtypes.size(), ")")); in ParseInputArrayInfo()
100 if (node_names.size() != node_shapes.size()) { in ParseInputArrayInfo()
103 node_names.size(), ", #input_shapes ", node_shapes.size(), ")")); in ParseInputArrayInfo()
107 for (int i = 0; i < node_names.size(); i++) { in ParseInputArrayInfo()
108 auto& name = node_names[i]; in ParseInputArrayInfo()
|
| D | mlir_roundtrip_flags.h | 89 Status ParseInputArrayInfo(const std::vector<string>& node_names,
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | placer_inspection_required_ops_utils.cc | 154 std::unordered_set<string>* node_names) { in Uniquify() argument
155 if (node_names->find(candidate_name) == node_names->end()) { in Uniquify()
156 node_names->insert(candidate_name); in Uniquify()
162 if (node_names->find(candidate) == node_names->end()) { in Uniquify()
163 node_names->insert(candidate); in Uniquify()
170 std::unordered_set<string>* node_names) { in AddInputIdentity() argument
175 absl::StrCat(edge->src()->name(), "_", node->name()), node_names); in AddInputIdentity()
213 std::unordered_set<string>* node_names) { in AddOutputIdentities() argument
252 Uniquify(absl::StrCat(node->name(), "_", dst->name()), node_names); in AddOutputIdentities()
270 string identity_name = Uniquify(node->name(), node_names); in AddOutputIdentities()
[all …]
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/quantization/ |
| D | quantization_config.cc | 42 bool ParseInputNodeQuantSpecs(absl::string_view node_names, in ParseInputNodeQuantSpecs() argument
47 std::vector<std::string> input_nodes = absl::StrSplit(node_names, ','); in ParseInputNodeQuantSpecs()
82 bool GetInputNodeQuantSpecs(const std::vector<std::string>& node_names, in GetInputNodeQuantSpecs() argument
95 if (node_names.size() != node_mins.size() || in GetInputNodeQuantSpecs()
96 node_names.size() != node_maxs.size()) { in GetInputNodeQuantSpecs()
99 for (int i = 0, e = node_names.size(); i != e; ++i) { in GetInputNodeQuantSpecs()
|
| D | quantization_config.h | 118 bool ParseInputNodeQuantSpecs(absl::string_view node_names,
127 bool GetInputNodeQuantSpecs(const std::vector<std::string>& node_names,
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/segment/ |
| D | segment_test.cc | 40 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/debug/ |
| D | grpc_session_debug_test.cc | 52 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()
|
| D | debug_grpc_io_utils_test.cc | 119 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 …]
|
| D | debug_grpc_testlib.cc | 74 node_names.push_back(node_name); in SendEvents()
125 node_names.clear(); in ClearReceivedDebugData()
|
| D | debug_grpc_testlib.h | 54 std::vector<string> node_names; variable
|
| /external/tensorflow/tensorflow/core/distributed_runtime/rpc/ |
| D | grpc_session_test.cc | 44 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/framework/ |
| D | graph_to_functiondef.cc | 155 const string& fn_name, const NodeNameMapping& node_names, in FillFunctionBody() argument
174 node_def->set_name(node_names.Lookup(node->name())); in FillFunctionBody()
236 const string normalized = node_names.Lookup(edge->src()->name()); in FillFunctionBody()
393 NodeNameMapping node_names; in GraphToFunctionDef() local
420 TF_RETURN_IF_ERROR(node_names.UseOutputName(output_names[i])); in GraphToFunctionDef()
423 argdef->set_name(node_names.GetOutputName(node->name())); in GraphToFunctionDef()
433 const string& input_name = node_names.GetInputName(node->name()); in GraphToFunctionDef()
474 const string& node_name = node_names.Uniquify(node->name()); in GraphToFunctionDef()
498 TF_RETURN_IF_ERROR(FillFunctionBody(fn_name, node_names, body_nodes, in GraphToFunctionDef()
568 node_names.Lookup(control_outputs[i]->name()); in GraphToFunctionDef()
|
| /external/tensorflow/tensorflow/core/grappler/utils/ |
| D | grappler_test.cc | 95 const GraphDef& graph, const std::vector<string>& node_names) const { in EvaluateNodes()
96 return EvaluateNodes(graph, node_names, {}); in EvaluateNodes()
100 const GraphDef& graph, const std::vector<string>& node_names, in EvaluateNodes() argument
106 TF_CHECK_OK(session->Run(run_options, inputs, node_names, node_names, in EvaluateNodes()
|
| D | grappler_test.h | 41 const GraphDef& graph, const std::vector<string>& node_names) const;
44 const GraphDef& graph, const std::vector<string>& node_names,
|
| D | graph_view.cc | 603 absl::flat_hash_map<absl::string_view, int>* node_names, in GetNodeNamesAndPartitionUpdatedNodes() argument
613 node_names->emplace(node_name, index); in GetNodeNamesAndPartitionUpdatedNodes()
619 node_names->emplace(node_name, node_index); in GetNodeNamesAndPartitionUpdatedNodes()
642 auto it = node_names->insert({node_name, internal::kNodeNamePresent}); in GetNodeNamesAndPartitionUpdatedNodes()
670 auto it = node_names->insert({node_name, internal::kNodeNamePresent}); in GetNodeNamesAndPartitionUpdatedNodes()
689 const absl::flat_hash_map<absl::string_view, int>& node_names, in RemovedOrMissingNodeFanoutsWellFormed() argument
708 for (const auto& node_name_state : node_names) { in RemovedOrMissingNodeFanoutsWellFormed()
767 const absl::flat_hash_map<absl::string_view, int>& node_names, in CheckNodeNamesAndFanins() argument
772 RemovedOrMissingNodeFanoutsWellFormed(node_names, renamed_nodes)); in CheckNodeNamesAndFanins()
777 if (!internal::IsWellFormed(&diff, node_names)) { in CheckNodeNamesAndFanins()
[all …]
|
| D | graph_view.h | 444 absl::flat_hash_map<absl::string_view, int>* node_names,
450 const absl::flat_hash_map<absl::string_view, int>& node_names,
454 const absl::flat_hash_map<absl::string_view, int>& node_names,
524 absl::flat_hash_map<absl::string_view, int>* node_names,
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/python/ |
| D | graphdef_to_tfl_flatbuffer.cc | 176 std::vector<string> node_names; in ConvertGraphDefToTFLiteFlatBuffer() local
192 node_names.push_back(flag.name()); in ConvertGraphDefToTFLiteFlatBuffer()
215 node_names, node_dtypes, node_shapes, &specs.inputs)); in ConvertGraphDefToTFLiteFlatBuffer()
216 if (mlir::TFL::GetInputNodeQuantSpecs(node_names, node_mins, node_maxs, in ConvertGraphDefToTFLiteFlatBuffer()
|
| /external/tensorflow/tensorflow/python/training/tracking/ |
| D | util.py | 88 def node_names(self): member in _ObjectGraphProtoPrettyPrinter
104 node_names = {}
106 node_names[node_id] = ".".join(path_to_root)
110 node_names[slot_reference.slot_variable_node_id] = (
112 node_names[node_id], slot_reference.slot_name,
113 node_names[slot_reference.original_variable_node_id]))
114 self._node_name_cache = node_names
115 return node_names
144 .format(pretty_printer.node_names[node_id]))
148 .format(pretty_printer.node_names[node_id], attribute_name)))
[all …]
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | remote_fused_graph_execute_utils_test.cc | 481 const std::unordered_set<string>& node_names = std::get<0>(cluster); in TEST() local
484 EXPECT_EQ(1, node_names.size()) << IterToString(node_names); in TEST()
489 EXPECT_EQ(3, node_names.size()) << IterToString(node_names); in TEST()
494 EXPECT_EQ(5, node_names.size()) << IterToString(node_names); in TEST()
499 EXPECT_EQ(6, node_names.size()) << IterToString(node_names); in TEST()
504 EXPECT_EQ(2, node_names.size()) << IterToString(node_names); in TEST()
|
| /external/tensorflow/tensorflow/python/debug/wrappers/ |
| D | local_cli_wrapper_test.py | 434 node_names = [datum.node_name for datum in debug_dump.dumped_tensor_data]
437 node_names)
461 node_names = [datum.node_name for datum in debug_dump.dumped_tensor_data]
462 self.assertIn("callable_a", node_names)
463 self.assertIn("callable_b", node_names)
489 node_names = [datum.node_name for datum in debug_dump.dumped_tensor_data]
490 self.assertIn("callable_a", node_names)
491 self.assertIn("callable_b", node_names)
518 node_names = [datum.node_name for datum in debug_dump.dumped_tensor_data]
521 node_names)
[all …]
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | graph_optimizer_stage.cc | 115 const std::vector<string> node_names, in MakeOptimizedNodeName() argument
119 for (const string& node_name : node_names) { in MakeOptimizedNodeName()
|
| /external/tensorflow/tensorflow/python/data/experimental/kernel_tests/optimization/ |
| D | map_vectorization_test.py | 222 def _make_dataset(node_names): argument
223 dataset = base_dataset.apply(testing.assert_next(node_names))
497 def _make_dataset(node_names): argument
498 dataset = base_dataset.apply(testing.assert_next(node_names))
544 def make_dataset(node_names): argument
545 dataset = base_dataset.apply(testing.assert_next(node_names))
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | functionalize_cond_test.cc | 152 absl::flat_hash_set<absl::string_view> node_names; in TEST() local
154 EXPECT_TRUE(node_names.insert(node.name()).second) in TEST()
|
| /external/curl/tests/python_dependencies/impacket/ |
| D | nmb.py | 214 self.node_names = [ ]
226 self.node_names.append(NBNodeEntry(name, type ,flags))
247 return self.node_names
248 def add_node_name(self,node_names): argument
249 self.node_names.append(node_names)
254 res += self.node_names[i].rawData()
|