Home
last modified time | relevance | path

Searched refs:input_node_name (Results 1 – 19 of 19) sorted by relevance

/external/tensorflow/tensorflow/core/grappler/optimizers/data/
Dgraph_test_utils.cc28 NodeDef MakeBatchV2Node(StringPiece name, StringPiece input_node_name, in MakeBatchV2Node() argument
33 {string(input_node_name), string(batch_size_node_name), in MakeBatchV2Node()
40 NodeDef MakeCacheV2Node(StringPiece name, StringPiece input_node_name, in MakeCacheV2Node() argument
46 string(input_node_name), in MakeCacheV2Node()
56 NodeDef MakeFilterNode(StringPiece name, StringPiece input_node_name, in MakeFilterNode() argument
59 name, "FilterDataset", {string(input_node_name)}, in MakeFilterNode()
66 NodeDef MakeMapAndBatchNode(StringPiece name, StringPiece input_node_name, in MakeMapAndBatchNode() argument
73 {string(input_node_name), string(batch_size_node_name), in MakeMapAndBatchNode()
81 NodeDef MakeMapNode(StringPiece name, StringPiece input_node_name, in MakeMapNode() argument
84 name, "MapDataset", {string(input_node_name)}, in MakeMapNode()
[all …]
Dgraph_test_utils.h27 NodeDef MakeBatchV2Node(StringPiece name, StringPiece input_node_name,
32 NodeDef MakeCacheV2Node(StringPiece name, StringPiece input_node_name,
37 NodeDef MakeFilterNode(StringPiece name, StringPiece input_node_name,
41 NodeDef MakeMapNode(StringPiece name, StringPiece input_node_name,
45 NodeDef MakeMapAndBatchNode(StringPiece name, StringPiece input_node_name,
53 StringPiece input_node_name,
60 NodeDef MakeParallelMapNode(StringPiece name, StringPiece input_node_name,
65 NodeDef MakeParseExampleNode(StringPiece name, StringPiece input_node_name,
70 NodeDef MakeShuffleV2Node(StringPiece name, StringPiece input_node_name,
75 NodeDef MakeTakeNode(StringPiece name, StringPiece input_node_name,
[all …]
/external/tensorflow/tensorflow/tools/graph_transforms/
Dobfuscate_names.cc80 string input_node_name; in ObfuscateNames() local
82 NodeNamePartsFromInput(input_name, &prefix, &input_node_name, &suffix); in ObfuscateNames()
83 if (new_names.count(input_node_name) == 0) { in ObfuscateNames()
84 return errors::InvalidArgument("No node named ", input_node_name, in ObfuscateNames()
87 string new_input_name = prefix + new_names[input_node_name] + suffix; in ObfuscateNames()
Drename_node.cc55 std::string input_node_name; in RenameNode() local
57 NodeNamePartsFromInput(input_name, &prefix, &input_node_name, &suffix); in RenameNode()
58 if (input_node_name == old_node_name) { in RenameNode()
Dtransform_utils.cc78 string input_node_name = NodeNameFromInput(input); in MapNodesToOutputs() local
79 (*result)[input_node_name].push_back(&node); in MapNodesToOutputs()
219 const string& input_node_name = NodeNameFromInput(input_name); in SortByExecutionOrder() local
220 if (!name_index.count(input_node_name)) { in SortByExecutionOrder()
225 outputs[name_index[input_node_name]].push_back(n); in SortByExecutionOrder()
350 const string& input_node_name = NodeNameFromInput(non_control_inputs[i]); in DoesOpTypeMatch() local
351 const NodeDef& input_node = *(node_map_[input_node_name]); in DoesOpTypeMatch()
410 string input_node_name = NodeNameFromInput(input_name); in ReplaceMatchingOpTypes() local
411 if (!matched_nodes_lookup.count(input_node_name)) { in ReplaceMatchingOpTypes()
498 string input_node_name = NodeNameFromInput(new_input_name); in RenameNodeInputs() local
[all …]
Dstrip_unused_nodes.cc150 string input_node_name = NodeNameFromInput(input_name); in StripUnusedNodes() local
151 if (!required_nodes.count(input_node_name)) { in StripUnusedNodes()
152 next_inputs.insert(input_node_name); in StripUnusedNodes()
/external/tensorflow/tensorflow/core/graph/
Dbenchmark_testlib.h154 const string input_node_name = absl::StrFormat("in%05d", i); in CreateFaninFanoutNodeGraph() local
155 NodeDef input_node = create_node(/*name=*/input_node_name); in CreateFaninFanoutNodeGraph()
157 node.add_input(input_node_name); in CreateFaninFanoutNodeGraph()
161 const string input_node_name = absl::StrFormat("control_in%05d", i); in CreateFaninFanoutNodeGraph() local
162 NodeDef input_node = create_node(/*name=*/input_node_name); in CreateFaninFanoutNodeGraph()
164 node.add_input(absl::StrCat("^", input_node_name)); in CreateFaninFanoutNodeGraph()
/external/tensorflow/tensorflow/core/kernels/
Dremote_fused_graph_execute_op.cc87 const string& input_node_name = execute_info_.graph_input_node_name(i); in Compute() local
89 remote_fused_graph_executor_->FillInputNode(input_node_name, in Compute()
Dremote_fused_graph_execute_op_test.cc181 const string& input_node_name = info_->graph_input_node_name(0); in ExecuteGraph() local
182 const Tensor& input_tensor = input_tensor_cache_[input_node_name]; in ExecuteGraph()
Dremote_fused_graph_execute_utils.h191 const string& input_node_name,
Dremote_fused_graph_execute_utils.cc714 const string& node_name, const string& input_node_name, in BuildIdentityOpNode() argument
717 Node* node = FindMutableNodeByName(input_node_name, graph); in BuildIdentityOpNode()
/external/tensorflow/tensorflow/core/grappler/costs/
Dutils.cc98 const string input_node_name(input_tensor_id.first); in ExtractExtraProperties() local
100 auto iter = name_to_node.find(input_node_name); in ExtractExtraProperties()
171 const string input_node_name(input_tensor_id.first); in FindInputFeatures() local
179 auto it = name_to_cost.find(input_node_name); in FindInputFeatures()
Dvirtual_scheduler.cc491 for (const string& input_node_name : inputs) { in Init() local
495 const NodeDef* input_node = name_to_node[NodeName(input_node_name)]; in Init()
499 const auto input_node_port_num = NodePosition(input_node_name); in Init()
506 if (curr_node_device == in_device || IsControlInput(input_node_name)) { in Init()
528 CreateSendRecv(input_node, curr_node, input_node, input_node_name, in Init()
/external/tensorflow/tensorflow/core/grappler/optimizers/
Dmemory_optimizer.cc816 const string input_node_name = NodeName(node->input(i)); in FindSwapInTrigger() local
817 auto it1 = name_map.find(input_node_name); in FindSwapInTrigger()
828 possible_inputs.insert(input_node_name); in FindSwapInTrigger()
832 const string input_node_name = NodeName(node->input(i)); in FindSwapInTrigger() local
833 possible_inputs.erase(input_node_name); in FindSwapInTrigger()
845 const string input_node_name = *possible_inputs.begin(); in FindSwapInTrigger() local
847 already_processed.insert(input_node_name); in FindSwapInTrigger()
848 auto it1 = name_map.find(input_node_name); in FindSwapInTrigger()
Dscoped_allocator_optimizer.cc430 string input_node_name = ParseNodeName(input_name, &position); in ClearInternalControlInputs() local
432 if (op_set.find(input_node_name) != op_set.end()) { in ClearInternalControlInputs()
434 VLOG(1) << "Remove control output from " << input_node_name in ClearInternalControlInputs()
436 TF_CHECK_OK(RemoveEdge(input_name, input_node_name, n, node_map)); in ClearInternalControlInputs()
Ddependency_optimizer.cc161 for (const string& input_node_name : node.input()) { in NumEdgesIfBypassed() local
162 if (IsControlInput(input_node_name)) { in NumEdgesIfBypassed()
Dconstant_folding.cc294 const string input_node_name = in ForwardInputs() local
296 if (input_node_name == node->name() && output_idx == input_idx) { in ForwardInputs()
/external/tensorflow/tensorflow/python/tools/
Doptimize_for_inference_lib.py146 input_node_name = node_name_from_input(input_name)
147 if input_node_name not in node_map:
/external/tensorflow/tensorflow/core/grappler/
Dmutable_graph_view.cc350 absl::string_view input_node_name = tensor_id.node(); in AddAndDedupFanouts() local
353 CanDedupControlWithRegularInput(*this, input_node_name); in AddAndDedupFanouts()
356 controlling_fanins.contains(input_node_name)); in AddAndDedupFanouts()
357 if (!gtl::InsertIfNotPresent(&fanins, input_node_name) && in AddAndDedupFanouts()
362 OutputPort output(nodes()[input_node_name], tensor_id.index()); in AddAndDedupFanouts()
375 controlling_fanins.insert(input_node_name); in AddAndDedupFanouts()