| /external/tensorflow/tensorflow/core/kernels/ |
| remote_fused_graph_execute_op.cc | 87 const string& input_node_name = execute_info_.graph_input_node_name(i); variable
89 remote_fused_graph_executor_->FillInputNode(input_node_name,
|
| remote_fused_graph_execute_op_test.cc | 181 const string& input_node_name = info_->graph_input_node_name(0); variable
182 const Tensor& input_tensor = input_tensor_cache_[input_node_name];
|
| /external/tensorflow/tensorflow/core/graph/ |
| benchmark_testlib.h | 150 const string input_node_name = absl::StrFormat("in%05d", i); local
151 NodeDef input_node = create_node(/*name=*/input_node_name);
153 node.add_input(input_node_name);
|
| /external/tensorflow/tensorflow/tools/graph_transforms/ |
| obfuscate_names.cc | 81 string input_node_name; local
83 NodeNamePartsFromInput(input_name, &prefix, &input_node_name, &suffix);
84 if (new_names.count(input_node_name) == 0) {
85 return errors::InvalidArgument("No node named ", input_node_name,
88 string new_input_name = prefix + new_names[input_node_name] + suffix;
|
| strip_unused_nodes.cc | 151 string input_node_name = NodeNameFromInput(input_name); local
152 if (!required_nodes.count(input_node_name)) {
153 next_inputs.insert(input_node_name);
|
| transform_utils.cc | 76 string input_node_name = NodeNameFromInput(input); local
77 (*result)[input_node_name].push_back(&node);
217 const string& input_node_name = NodeNameFromInput(input_name); local
218 if (!name_index.count(input_node_name)) {
223 outputs[name_index[input_node_name]].push_back(n);
348 const string& input_node_name = NodeNameFromInput(non_control_inputs[i]); local
349 const NodeDef& input_node = *(node_map_[input_node_name]);
408 string input_node_name = NodeNameFromInput(input_name); local
409 if (!matched_nodes_lookup.count(input_node_name)) {
496 string input_node_name = NodeNameFromInput(new_input_name) local
[all...] |
| /external/tensorflow/tensorflow/core/grappler/ |
| mutable_graph_view.cc | 349 absl::string_view input_node_name = tensor_id.node(); local
352 CanDedupControlWithRegularInput(*this, input_node_name);
356 controlling_fanins.contains(input_node_name)));
357 if (!gtl::InsertIfNotPresent(&fanins, input_node_name) &&
362 OutputPort output(nodes()[input_node_name], tensor_id.index());
375 controlling_fanins.insert(input_node_name);
[all...] |
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| memory_optimizer.cc | 785 const string input_node_name = NodeName(node->input(i)); local
801 const string input_node_name = NodeName(node->input(i)); local
814 const string input_node_name = *possible_inputs.begin(); local
[all...] |
| scoped_allocator_optimizer.cc | 263 string input_node_name = ParseNodeName(input_name, &position); local
265 if (op_set.find(input_node_name) != op_set.end()) {
267 VLOG(1) << "Remove control output from " << input_node_name
269 TF_CHECK_OK(RemoveEdge(input_name, input_node_name, n, node_map));
[all...] |
| constant_folding.cc | 3042 const string input_node_name = local
[all...] |
