Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/grappler/optimizers/data/
Dgraph_test_utils.cc28 NodeDef MakeFilterNode(StringPiece name, StringPiece input_node_name, in MakeFilterNode() argument
31 name, "FilterDataset", {string(input_node_name)}, in MakeFilterNode()
38 NodeDef MakeMapAndBatchNode(StringPiece name, StringPiece input_node_name, in MakeMapAndBatchNode() argument
45 {string(input_node_name), string(batch_size_node_name), in MakeMapAndBatchNode()
53 NodeDef MakeMapNode(StringPiece name, StringPiece input_node_name, in MakeMapNode() argument
56 name, "MapDataset", {string(input_node_name)}, in MakeMapNode()
64 StringPiece input_node_name, in MakeParallelInterleaveNode() argument
71 {string(input_node_name), string(cycle_length_node_name), in MakeParallelInterleaveNode()
82 NodeDef MakeParallelMapNode(StringPiece name, StringPiece input_node_name, in MakeParallelMapNode() argument
87 {string(input_node_name), string(num_parallel_calls_node_name)}, in MakeParallelMapNode()
[all …]
Dgraph_test_utils.h27 NodeDef MakeFilterNode(StringPiece name, StringPiece input_node_name,
31 NodeDef MakeMapNode(StringPiece name, StringPiece input_node_name,
35 NodeDef MakeMapAndBatchNode(StringPiece name, StringPiece input_node_name,
43 StringPiece input_node_name,
50 NodeDef MakeParallelMapNode(StringPiece name, StringPiece input_node_name,
55 NodeDef MakeParseExampleNode(StringPiece name, StringPiece input_node_name,
/external/tensorflow/tensorflow/tools/graph_transforms/
Dobfuscate_names.cc81 string input_node_name; in ObfuscateNames() local
83 NodeNamePartsFromInput(input_name, &prefix, &input_node_name, &suffix); in ObfuscateNames()
84 if (new_names.count(input_node_name) == 0) { in ObfuscateNames()
85 return errors::InvalidArgument("No node named ", input_node_name, in ObfuscateNames()
88 string new_input_name = prefix + new_names[input_node_name] + suffix; in ObfuscateNames()
Dtransform_utils.cc76 string input_node_name = NodeNameFromInput(input); in MapNodesToOutputs() local
77 (*result)[input_node_name].push_back(&node); in MapNodesToOutputs()
217 const string& input_node_name = NodeNameFromInput(input_name); in SortByExecutionOrder() local
218 if (!name_index.count(input_node_name)) { in SortByExecutionOrder()
223 outputs[name_index[input_node_name]].push_back(n); in SortByExecutionOrder()
348 const string& input_node_name = NodeNameFromInput(non_control_inputs[i]); in DoesOpTypeMatch() local
349 const NodeDef& input_node = *(node_map_[input_node_name]); in DoesOpTypeMatch()
408 string input_node_name = NodeNameFromInput(input_name); in ReplaceMatchingOpTypes() local
409 if (!matched_nodes_lookup.count(input_node_name)) { in ReplaceMatchingOpTypes()
496 string input_node_name = NodeNameFromInput(new_input_name); in RenameNodeInputs() local
[all …]
Dstrip_unused_nodes.cc151 string input_node_name = NodeNameFromInput(input_name); in StripUnusedNodes() local
152 if (!required_nodes.count(input_node_name)) { in StripUnusedNodes()
153 next_inputs.insert(input_node_name); in StripUnusedNodes()
/external/tensorflow/tensorflow/contrib/receptive_field/python/util/
Dgraph_compute_order.py77 input_node_name='', argument
105 if current == input_node_name:
126 name_to_node, each, node_info, input_node_name, input_node_size)
160 def get_compute_order(graph_def, input_node_name='', input_node_size=None): argument
190 _get_computed_nodes(name_to_node, each.name, node_info, input_node_name,
Dgraph_compute_order_test.py119 graph_def, input_node_name='input_image')
124 graph_def, input_node_name='input_image', input_node_size=[224, 224])
Dreceptive_field.py213 input_node_name=input_node,
/external/tensorflow/tensorflow/core/graph/
Dbenchmark_testlib.h150 const string input_node_name = absl::StrFormat("in%05d", i); in CreateFaninFanoutNodeGraph() local
151 NodeDef input_node = create_node(/*name=*/input_node_name); in CreateFaninFanoutNodeGraph()
153 node.add_input(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.cc713 const string& node_name, const string& input_node_name, in BuildIdentityOpNode() argument
716 Node* node = FindMutableNodeByName(input_node_name, graph); in BuildIdentityOpNode()
/external/tensorflow/tensorflow/core/grappler/costs/
Dutils.cc97 const string input_node_name(input_tensor_id.first); in ExtractExtraProperties() local
99 auto iter = name_to_node.find(input_node_name); in ExtractExtraProperties()
170 const string input_node_name(input_tensor_id.first); in FindInputFeatures() local
178 auto it = name_to_cost.find(input_node_name); in FindInputFeatures()
Dvirtual_scheduler.cc375 for (const string& input_node_name : inputs) { in Init() local
379 const NodeDef* input_node = name_to_node[NodeName(input_node_name)]; in Init()
383 const auto input_node_port_num = NodePosition(input_node_name); in Init()
407 input_node_name); in Init()
/external/tensorflow/tensorflow/core/grappler/optimizers/
Dmemory_optimizer.cc785 const string input_node_name = NodeName(node->input(i)); in FindSwapInTrigger() local
786 auto it1 = name_map.find(input_node_name); in FindSwapInTrigger()
797 possible_inputs.insert(input_node_name); in FindSwapInTrigger()
801 const string input_node_name = NodeName(node->input(i)); in FindSwapInTrigger() local
802 possible_inputs.erase(input_node_name); in FindSwapInTrigger()
814 const string input_node_name = *possible_inputs.begin(); in FindSwapInTrigger() local
816 already_processed.insert(input_node_name); in FindSwapInTrigger()
817 auto it1 = name_map.find(input_node_name); in FindSwapInTrigger()
Dscoped_allocator_optimizer.cc263 string input_node_name = ParseNodeName(input_name, &position); in ClearInternalControlInputs() local
265 if (op_set.find(input_node_name) != op_set.end()) { in ClearInternalControlInputs()
267 VLOG(1) << "Remove control output from " << input_node_name in ClearInternalControlInputs()
269 TF_CHECK_OK(RemoveEdge(input_name, input_node_name, n, node_map)); in ClearInternalControlInputs()
Ddependency_optimizer.cc139 for (string input_node_name : node.input()) { in NumEdgesIfBypassed() local
140 if (IsControlInput(input_node_name)) { in NumEdgesIfBypassed()
Dconstant_folding.cc3042 const string input_node_name = in PartialConstPropThroughIdentityN() local
3044 if (input_node_name == node->name() && output_idx == input_idx) { in PartialConstPropThroughIdentityN()
/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.cc349 absl::string_view input_node_name = tensor_id.node(); in AddAndDedupFanouts() local
352 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()