Home
last modified time | relevance | path

Searched refs:input_node_name (Results 1 – 18 of 18) 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
34 {string(input_node_name), string(batch_size_node_name), in MakeBatchV2Node()
41 NodeDef MakeParallelBatchNode(StringPiece name, StringPiece input_node_name, in MakeParallelBatchNode() argument
48 {string(input_node_name), string(batch_size_node_name), in MakeParallelBatchNode()
55 NodeDef MakeCacheV2Node(StringPiece name, StringPiece input_node_name, in MakeCacheV2Node() argument
61 string(input_node_name), in MakeCacheV2Node()
71 NodeDef MakeFilterNode(StringPiece name, StringPiece input_node_name, in MakeFilterNode() argument
74 name, "FilterDataset", {string(input_node_name)}, in MakeFilterNode()
81 NodeDef MakeMapAndBatchNode(StringPiece name, StringPiece input_node_name, in MakeMapAndBatchNode() argument
88 {string(input_node_name), string(batch_size_node_name), in MakeMapAndBatchNode()
[all …]
Dgraph_test_utils.h27 NodeDef MakeBatchV2Node(StringPiece name, StringPiece input_node_name,
33 NodeDef MakeParallelBatchNode(StringPiece name, StringPiece input_node_name,
40 NodeDef MakeCacheV2Node(StringPiece name, StringPiece input_node_name,
45 NodeDef MakeFilterNode(StringPiece name, StringPiece input_node_name,
49 NodeDef MakeMapNode(StringPiece name, StringPiece input_node_name,
53 NodeDef MakeMapAndBatchNode(StringPiece name, StringPiece input_node_name,
61 StringPiece input_node_name,
69 StringPiece input_node_name,
77 NodeDef MakeParallelMapNode(StringPiece name, StringPiece input_node_name,
82 NodeDef MakeParallelMapV2Node(StringPiece name, StringPiece input_node_name,
[all …]
Dauto_shard_test.cc47 void FinishItem(GrapplerItem* item, const string& input_node_name) { in FinishItem() argument
49 NDef("map_before_rebatch", "MapDataset", {input_node_name}, in FinishItem()
/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/data/
Drewrite_utils_test.cc44 NodeDef GetMapNode(absl::string_view name, absl::string_view input_node_name, in GetMapNode() argument
47 name, /*op=*/"MapDataset", {std::string(input_node_name)}, in GetMapNode()
/external/tensorflow/tensorflow/core/data/service/
Dtest_util.cc58 NodeDef GetMapNode(absl::string_view name, absl::string_view input_node_name, in GetMapNode() argument
61 name, /*op=*/"MapDataset", {std::string(input_node_name)}, in GetMapNode()
/external/tensorflow/tensorflow/core/grappler/costs/
Dutils.cc99 const string input_node_name(input_tensor_id.first); in ExtractExtraProperties() local
101 auto iter = name_to_node.find(input_node_name); in ExtractExtraProperties()
172 const string input_node_name(input_tensor_id.first); in FindInputFeatures() local
180 auto it = name_to_cost.find(input_node_name); in FindInputFeatures()
Dvirtual_scheduler.cc496 for (const string& input_node_name : inputs) { in Init() local
500 const NodeDef* input_node = name_to_node[NodeName(input_node_name)]; in Init()
504 const auto input_node_port_num = NodePosition(input_node_name); in Init()
511 if (curr_node_device == in_device || IsControlInput(input_node_name)) { in Init()
533 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.cc169 for (const string& input_node_name : node.input()) { in NumEdgesIfBypassed() local
170 if (IsControlInput(input_node_name)) { in NumEdgesIfBypassed()
Dconstant_folding.cc298 const string input_node_name = in ForwardInputs() local
300 if (input_node_name == node->name() && output_idx == input_idx) { in ForwardInputs()
/external/tensorflow/tensorflow/python/tools/
Doptimize_for_inference_lib.py142 input_node_name = node_name_from_input(input_name)
143 if input_node_name not in node_map:
/external/tensorflow/tensorflow/core/grappler/
Dmutable_graph_view.cc372 absl::string_view input_node_name = tensor_id.node(); in AddAndDedupFanouts() local
375 CanDedupControlWithRegularInput(*this, input_node_name); in AddAndDedupFanouts()
378 controlling_fanins.contains(input_node_name)); in AddAndDedupFanouts()
379 if (!gtl::InsertIfNotPresent(&fanins, input_node_name) && in AddAndDedupFanouts()
384 OutputPort output(nodes()[input_node_name], tensor_id.index()); in AddAndDedupFanouts()
397 controlling_fanins.insert(input_node_name); in AddAndDedupFanouts()