| /external/tensorflow/tensorflow/python/framework/ |
| D | convert_to_constants.py | 180 for input_node in graph_def.node:
183 if input_node.name in resource_placeholders:
184 dtype = resource_placeholders[input_node.name]["dtype"]
185 data = resource_placeholders[input_node.name]["data"]
188 output_node.name = input_node.name
195 elif input_node.name in resource_identities:
196 output_node.CopyFrom(input_node)
197 output_node.attr["T"].CopyFrom(resource_identities[input_node.name])
199 elif input_node.op == "ReadVariableOp":
201 output_node.name = input_node.name
[all …]
|
| D | graph_util_impl.py | 274 for input_node in inference_graph.node:
276 if input_node.name in found_variables:
278 output_node.name = input_node.name
279 dtype = input_node.attr["dtype"]
280 data = found_variables[input_node.name]
287 elif input_node.op == "ReadVariableOp" and (
288 input_node.input[0] in found_variables):
293 output_node.name = input_node.name
294 output_node.input.extend([input_node.input[0]])
295 output_node.attr["T"].CopyFrom(input_node.attr["dtype"])
[all …]
|
| /external/tensorflow/tensorflow/contrib/receptive_field/python/util/ |
| D | receptive_field_test.py | 289 input_node = 'input_image'
294 graph_def, input_node, output_node))
304 input_node = 'input_image'
309 graph_def, input_node, output_node))
319 input_node = 'input_image'
323 graph_def, input_node, output_node)
327 input_node = 'input_image'
332 graph_def, input_node, output_node))
342 input_node = 'input_image'
347 graph_def, input_node, output_node, input_resolution=[9, 9]))
[all …]
|
| D | receptive_field.py | 154 input_node, argument
203 if isinstance(input_node, framework_ops.Tensor):
204 input_node = input_node.op.name
213 input_node_name=input_node,
253 if node.name == input_node:
386 if input_node not in rf_sizes_x:
389 (rf_sizes_x[input_node], rf_sizes_y[input_node]),
390 (effective_strides_x[input_node], effective_strides_y[input_node]),
391 (effective_paddings_x[input_node], effective_paddings_y[input_node]))
|
| /external/tensorflow/tensorflow/tools/graph_transforms/ |
| D | obfuscate_names.cc | 48 for (const NodeDef& input_node : input_graph_def.node()) { in ObfuscateNames() local
49 const string& old_name = input_node.name(); in ObfuscateNames()
73 for (const NodeDef& input_node : input_graph_def.node()) { in ObfuscateNames() local
75 *node = input_node; in ObfuscateNames()
76 const string& old_name = input_node.name(); in ObfuscateNames()
79 for (const string& input_name : input_node.input()) { in ObfuscateNames()
|
| D | remove_nodes.cc | 82 const NodeDef& input_node = match.inputs[0].node; in RemoveNodes() local
83 string target_name = input_node.name(); in RemoveNodes()
95 "^" + input_node.name(); in RemoveNodes()
96 new_nodes->push_back(input_node); in RemoveNodes()
|
| D | fold_batch_norms.cc | 55 const NodeDef& input_node = match.inputs[0].inputs[0].node; in FoldBatchNorms() local
64 {mul_node, conv_node, input_node, weights_node, in FoldBatchNorms()
109 new_nodes->push_back(input_node); in FoldBatchNorms()
|
| D | flatten_atrous.cc | 57 const NodeDef& input_node = match.inputs[0].inputs[0].inputs[0].node; in FlattenAtrousConv() local
112 AddNodeInput(input_node.name(), &flattened_conv_node); in FlattenAtrousConv()
126 new_nodes->push_back(input_node); in FlattenAtrousConv()
|
| D | transform_utils.cc | 349 const NodeDef& input_node = *(node_map_[input_node_name]); in DoesOpTypeMatch() local
353 if (!DoesOpTypeMatch(input_node, input_pattern, previously_matched_nodes, in DoesOpTypeMatch()
386 for (const NodeDef& input_node : input_graph_def.node()) { in ReplaceMatchingOpTypes() local
387 if (matches_by_head_name.count(input_node.name())) { in ReplaceMatchingOpTypes()
390 const NodeMatch* match = matches_by_head_name[input_node.name()]; in ReplaceMatchingOpTypes()
462 } else if (!matched_nodes.count(input_node.name())) { in ReplaceMatchingOpTypes()
465 *added_node = input_node; in ReplaceMatchingOpTypes()
562 string input_node = NodeNameFromInput(input); in FindInvalidInputs() local
563 if (!node_map.count(input_node)) { in FindInvalidInputs()
564 invalid_inputs->push_back({node.name(), input_node}); in FindInvalidInputs()
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| D | convert_graph.cc | 251 auto input_node = edge->src(); in GetEngineInfo() local
252 if (input_node->IsSource() || segment_nodes.count(input_node)) { in GetEngineInfo()
257 info->connections.emplace_back(input_node->name(), input_node->id(), in GetEngineInfo()
260 } else if (input_node->type_string() == "Const") { in GetEngineInfo()
269 if (!added_const_nodes.insert(input_node).second) { in GetEngineInfo()
273 VLOG(1) << "Adding const node " << input_node->name(); in GetEngineInfo()
277 info->connections.emplace_back(input_node->name(), input_node->id(), in GetEngineInfo()
284 const string s = StrCat(input_node->name(), ":", edge->src_output()); in GetEngineInfo()
293 input_node->name(), input_node->id(), edge->src_output(), node_name, in GetEngineInfo()
425 Node* input_node = graph->FindNodeId(conn.outside_id); in CreateTRTNode() local
[all …]
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/data/ |
| D | map_vectorization_test.cc | 338 auto input_node = AddRangeNode(&graph); in TEST_P() local
355 auto map_and_batch_0 = make_map_and_batch(input_node, fuse_0); in TEST_P()
432 auto input_node = AddArbitraryInputNode(&graph, nullptr, &input_types); in TEST() local
435 AddMapNode(&graph, input_node->name(), map_fn->signature().name()); in TEST()
441 input_node->name()); in TEST()
452 auto input_node = AddArbitraryInputNode(&graph, &input_shapes, &input_types); in TEST() local
455 AddMapNode(&graph, input_node->name(), map_fn->signature().name()); in TEST()
461 input_node->name()); in TEST()
472 auto input_node = AddArbitraryInputNode(&graph, &input_shapes, &input_types); in TEST() local
475 AddMapNode(&graph, input_node->name(), map_fn->signature().name()); in TEST()
[all …]
|
| D | map_vectorization.cc | 191 Status AddNewBatchNode(const NodeDef& old_batch_node, const NodeDef& input_node, in AddNewBatchNode() argument
200 batch_node.add_input(input_node.name()); in AddNewBatchNode()
227 input_node.attr().at("output_shapes").list().shape(); in AddNewBatchNode()
457 const NodeDef*& input_node = *input_node_output; in FindMapAndBatchPattern() local
486 input_node = graph_utils::GetInputNode(*map_node, graph); in FindMapAndBatchPattern()
487 DCHECK_NE(input_node, nullptr); in FindMapAndBatchPattern()
489 if (!IsOutputShapesFullyDefined(*input_node)) { in FindMapAndBatchPattern()
525 const NodeDef* input_node; in OptimizeAndCollectStats() local
528 &optional_prefetch_node, &map_node, &input_node, in OptimizeAndCollectStats()
540 *batch_node, *input_node, *vectorized_func, &graph, &new_batch_node)); in OptimizeAndCollectStats()
[all …]
|
| D | noop_elimination_test.cc | 38 NodeDef *MakeUnaryNode(StringPiece node_type, int count, string input_node, in MakeUnaryNode() argument
42 {std::move(input_node), node_count->name()}, in MakeUnaryNode()
46 NodeDef *MakeUnaryNonConstNode(StringPiece node_type, string input_node, in MakeUnaryNonConstNode() argument
50 {std::move(input_node), node_count->name()}, in MakeUnaryNonConstNode()
54 NodeDef *MakeCacheNode(string input_node, MutableGraphView *graph) { in MakeCacheNode() argument
58 {std::move(input_node), node_filename->name()}, in MakeCacheNode()
|
| D | vectorization_utils.cc | 444 Node* input_node; in StackTensor() local
445 TF_RETURN_IF_ERROR(map_defun_node_->input_node(0, &input_node)); in StackTensor()
449 node_builder("Shape").Input(input_node).Finalize(g, &shape_node)); in StackTensor()
511 Node* input_node; in AddArgTensorMappings() local
512 TF_RETURN_IF_ERROR(map_defun_node_->input_node( in AddArgTensorMappings()
513 arg_node->attrs().Find("index")->i(), &input_node)); in AddArgTensorMappings()
515 conversion_map_.insert({{arg_node, 0}, {input_node, 0, stacked}}); in AddArgTensorMappings()
519 {input_node, Graph::kControlSlot, stacked}}); in AddArgTensorMappings()
|
| D | rebatch.cc | 194 NodeDef* input_node = graph_utils::GetInputNode(node, *graph, i); in RecursivelyHandleOp() local
196 RecursivelyHandleOp(*input_node, num_workers, flib, graph)); in RecursivelyHandleOp()
201 NodeDef* input_node = graph_utils::GetInputNode(node, *graph, 0); in RecursivelyHandleOp() local
203 RecursivelyHandleOp(*input_node, num_workers, flib, graph)); in RecursivelyHandleOp()
|
| D | auto_shard.cc | 187 const NodeDef* input_node = graph_utils::GetInputNode(node, *graph, i); in RecursivelyHandleOp() local
188 TF_RETURN_IF_ERROR(RecursivelyHandleOp(*input_node, num_workers, index, in RecursivelyHandleOp()
222 const NodeDef* input_node = graph_utils::GetInputNode(node, *graph, 0); in RecursivelyHandleOp() local
223 return RecursivelyHandleOp(*input_node, num_workers, index, flib, graph, in RecursivelyHandleOp()
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | dependency_optimizer.cc | 192 for (NodeDef* input_node : input_nodes) { in BypassingNodeIsBeneficial()
193 num_cross_in += static_cast<int>(input_node->device() != node_dev); in BypassingNodeIsBeneficial()
203 for (NodeDef* input_node : input_nodes) { in BypassingNodeIsBeneficial()
206 static_cast<int>(input_node->device() != output_node->device()); in BypassingNodeIsBeneficial()
359 NodeDef* input_node = node_map_->GetNode(node->input(i)); in OptimizeNode() local
360 if (input_node == nullptr) { in OptimizeNode()
364 input_nodes.push_back(input_node); in OptimizeNode()
490 const NodeDef* input_node = node_map_->GetNode(input); in TransitiveReduction() local
491 if (ModifiesFrameInfo(*input_node) || IsMerge(*input_node)) { in TransitiveReduction()
496 const int input_node_idx = node_to_idx_[input_node]; in TransitiveReduction()
|
| D | layout_optimizer.cc | 613 auto input_node = node_map_->GetNode(node_->input(input_index)); in UpdateAttrValueOfInput() local
619 *added_node = *input_node; in UpdateAttrValueOfInput()
646 auto input_node = node_map_->GetNode(node_->input(pos)); in AddLayoutTransposeToInputs() local
647 TF_RETURN_IF_ERROR(HasAttribute(*input_node, "_output_shapes")); in AddLayoutTransposeToInputs()
653 input_node->attr().at("_output_shapes").list().shape(output_pos), in AddLayoutTransposeToInputs()
913 auto input_node = node_map_->GetNode(input); in GetOrAddNodePermNHWCToNCHW() local
914 depended_node = NodeName(input_node->input(0)); in GetOrAddNodePermNHWCToNCHW()
1244 auto input_node = node_map_->GetNode(node.input(pos)); in IsNodeAfterNCHWToNHWC() local
1245 queue.push_back(input_node); in IsNodeAfterNCHWToNHWC()
1246 visited.insert(input_node->name()); in IsNodeAfterNCHWToNHWC()
[all …]
|
| D | memory_optimizer.cc | 95 const NodeDef* input_node = node_map.GetNode(input_name); in FindCandidateRecomputeNodes() local
96 if (is_target(*input_node)) { in FindCandidateRecomputeNodes()
128 const NodeDef* input_node = node_map.GetNode(input_name_raw); in connected_subgraph() local
129 if (expanded_nodes->count(input_node) == 0 && in connected_subgraph()
130 is_candidate(*input_node)) { in connected_subgraph()
131 to_visit.push(input_node); in connected_subgraph()
790 const NodeDef* input_node = it1->second; in FindSwapInTrigger() local
792 auto it2 = execution_times.find(input_node); in FindSwapInTrigger()
821 const NodeDef* input_node = it1->second; in FindSwapInTrigger() local
825 if (ModifiesFrameInfo(*input_node) || IsSwitch(*input_node) || in FindSwapInTrigger()
[all …]
|
| /external/tensorflow/tensorflow/core/kernels/data/experimental/ |
| D | stats_dataset_ops.cc | 84 Node* input_node; in AsGraphDefInternal() local
85 TF_RETURN_IF_ERROR(b->AddInputDataset(ctx, input_, &input_node)); in AsGraphDefInternal()
88 TF_RETURN_IF_ERROR(b->AddDataset(this, {input_node, tag_node}, output)); in AsGraphDefInternal()
194 Node* input_node; in AsGraphDefInternal() local
195 TF_RETURN_IF_ERROR(b->AddInputDataset(ctx, input_, &input_node)); in AsGraphDefInternal()
198 TF_RETURN_IF_ERROR(b->AddDataset(this, {input_node, tag_node}, output)); in AsGraphDefInternal()
|
| /external/tensorflow/tensorflow/core/kernels/hexagon/ |
| D | graph_transferer.cc | 129 for (const Node* const input_node : node->in_nodes()) { in LoadGraphFromProto() local
130 const string& name = input_node->name(); in LoadGraphFromProto()
313 for (const Node* const input_node : node.in_nodes()) { in AreAllInputsCached() local
314 if (node_name_to_id_cache_map_.count(input_node->name()) <= 0) { in AreAllInputsCached()
315 VLOG(1) << "input_node " << input_node->name() << " of " << node.name() in AreAllInputsCached()
351 Node* input_node; in TransformGraphToAddAggregatedInputNode() local
352 TF_RETURN_IF_ERROR(builder.Finalize(graph, &input_node)); in TransformGraphToAddAggregatedInputNode()
353 CHECK_NOTNULL(input_node); in TransformGraphToAddAggregatedInputNode()
357 shape_refiner->UpdateNode(input_node, false /* relax */, &refined)); in TransformGraphToAddAggregatedInputNode()
360 shape_refiner->GetContext(input_node); in TransformGraphToAddAggregatedInputNode()
[all …]
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | tf2xla_util.cc | 229 const Node* input_node; in PropagateConstIntoIfNode() local
230 TF_RETURN_IF_ERROR(if_node->input_node(i, &input_node)); in PropagateConstIntoIfNode()
231 if (input_node->type_string() == "Const") { in PropagateConstIntoIfNode()
232 const_input_index_to_node[i - 1] = input_node; in PropagateConstIntoIfNode()
267 const Node* input_node; in PropagateConstIntoWhileNode() local
268 TF_RETURN_IF_ERROR(while_node->input_node(i, &input_node)); in PropagateConstIntoWhileNode()
269 if (input_node->type_string() != "Const") { in PropagateConstIntoWhileNode()
293 const_input_index_to_node[i] = input_node; in PropagateConstIntoWhileNode()
635 Node* input_node; in RewriteAssociatedFunction() local
636 TF_RETURN_IF_ERROR(node->input_node(i, &input_node)); in RewriteAssociatedFunction()
[all …]
|
| /external/tensorflow/tensorflow/core/graph/ |
| D | benchmark_testlib.h | 128 const uint32 input_node = rnd.Uniform(i); in CreateRandomGraph() local
129 node.add_input(absl::StrCat(prefix, input_node)); in CreateRandomGraph()
151 NodeDef input_node = create_node(/*name=*/input_node_name); in CreateFaninFanoutNodeGraph() local
152 *graph.add_node() = std::move(input_node); in CreateFaninFanoutNodeGraph()
|
| /external/tensorflow/tensorflow/core/grappler/costs/ |
| D | virtual_scheduler.cc | 379 const NodeDef* input_node = name_to_node[NodeName(input_node_name)]; in Init() local
381 CHECK(input_node); in Init()
382 const string in_device = DeviceName(input_node); in Init()
388 std::make_pair(input_node, input_node_port_num)); in Init()
389 auto& input_node_state = GetNodeStateOrCreateIt(input_node); in Init()
392 RecvNodeDescriptor recv_node(input_node, input_node_port_num, in Init()
406 auto send_and_recv = CreateSendRecv(input_node, curr_node, input_node, in Init()
414 auto& input_node_state = GetNodeStateOrCreateIt(input_node); in Init()
545 const NodeDef* from, const NodeDef* to, const NodeDef* input_node, in CreateSendRecv() argument
581 if (input_node->attr().count(kAttrTensorName)) { in CreateSendRecv()
[all …]
|
| /external/tensorflow/tensorflow/contrib/framework/python/framework/ |
| D | graph_util.py | 98 input_node for input_node in name_to_input_name[cur_node]
99 if input_node not in visited
|