/external/tensorflow/tensorflow/tools/graph_transforms/ |
obfuscate_names.cc | 82 string input_node_name; local 84 NodeNamePartsFromInput(input_name, &prefix, &input_node_name, &suffix); 85 if (new_names.count(input_node_name) == 0) { 86 return errors::InvalidArgument("No node named ", input_node_name, 89 string new_input_name = prefix + new_names[input_node_name] + suffix;
|
strip_unused_nodes.cc | 152 string input_node_name = NodeNameFromInput(input_name); local 153 if (!required_nodes.count(input_node_name)) { 154 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); 218 const string& input_node_name = NodeNameFromInput(input_name); local 219 if (!name_index.count(input_node_name)) { 224 outputs[name_index[input_node_name]].push_back(n); 342 const string& input_node_name = NodeNameFromInput(non_control_inputs[i]); local 343 const NodeDef& input_node = *(node_map_[input_node_name]); 402 string input_node_name = NodeNameFromInput(input_name); local 403 if (!matched_nodes_lookup.count(input_node_name)) { 490 string input_node_name = NodeNameFromInput(new_input_name) local [all...] |
/external/tensorflow/tensorflow/contrib/receptive_field/python/util/ |
graph_compute_order.py | 77 input_node_name='', 82 resolutions at each node. In this case, input_node_name and input_node_size 92 input_node_name: Name of node with fixed input resolution (optional). 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): 164 resolutions at each node. In this case, input_node_name and input_node_size 170 input_node_name: Name of node with fixed input resolution (optional). This 190 _get_computed_nodes(name_to_node, each.name, node_info, input_node_name,
|
graph_compute_order_test.py | 119 graph_def, input_node_name='input_image') 124 graph_def, input_node_name='input_image', input_node_size=[224, 224])
|
receptive_field.py | 218 input_node_name=input_node,
|
/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];
|
remote_fused_graph_execute_utils.h | 182 const string& input_node_name,
|
remote_fused_graph_execute_utils.cc | 712 const string& node_name, const string& input_node_name, 715 Node* node = FindMutableNodeByName(input_node_name, graph); [all...] |
/external/tensorflow/tensorflow/core/grappler/optimizers/ |
graph_rewriter.cc | 134 string input_node_name = ParseNodeName(input, &position); local 135 auto itr = nodes_.find(input_node_name); 194 const string input_node_name = NodeName(input); local 195 auto itr = nodes_.find(input_node_name);
|
memory_optimizer.cc | 780 const string input_node_name = NodeName(node->input(i)); local 796 const string input_node_name = NodeName(node->input(i)); local 809 const string input_node_name = *possible_inputs.begin(); local [all...] |
arithmetic_optimizer.cc | 629 string input_node_name = ParseNodeName(node->input(0), &output_pos); local 630 const NodeDef* input = node_map_->GetNode(input_node_name); [all...] |
/external/tensorflow/tensorflow/tools/quantization/ |
quantize_graph.py | 95 for input_node_name in current_node.input: 96 if input_node_name in already_visited: 98 input_node = nodes_map[input_node_name] 461 for input_node_name in current_node.input: 462 input_node_name = node_name_from_input(input_node_name) 463 input_node = self.nodes_map[input_node_name] 493 for input_node_name in current_node.input: 494 input_node_name = node_name_from_input(input_node_name) [all...] |
/external/tensorflow/tensorflow/core/grappler/costs/ |
utils.cc | 100 const string input_node_name = input_tensor_id.first.ToString(); local 102 auto iter = name_to_node.find(input_node_name); 173 const string input_node_name = input_tensor_id.first.ToString(); local 181 auto it = name_to_cost.find(input_node_name);
|
virtual_scheduler.cc | 376 for (const string& input_node_name : inputs) { 377 // Note that input_node_name may be in <prefix><node_name>:<port_num> 380 const NodeDef* input_node = name_to_node[NodeName(input_node_name)]; 384 const auto input_node_port_num = NodePosition(input_node_name); 408 CreateSendRecv(input_node, curr_node, input_node_name); 548 // input_node_name is the string from the "to" node to identify which output [all...] |
/external/tensorflow/tensorflow/python/tools/ |
optimize_for_inference_lib.py | 139 input_node_name = node_name_from_input(input_name) 140 if input_node_name not in node_map.keys():
|