Home
last modified time | relevance | path

Searched defs:input_node (Results 1 – 25 of 64) sorted by relevance

123

/external/tensorflow/tensorflow/core/graph/
Dbenchmark_testlib.h128 const uint32 input_node = rnd.Uniform(i); in CreateRandomGraph() local
155 NodeDef input_node = create_node(/*name=*/input_node_name); in CreateFaninFanoutNodeGraph() local
162 NodeDef input_node = create_node(/*name=*/input_node_name); in CreateFaninFanoutNodeGraph() local
/external/tensorflow/tensorflow/core/grappler/optimizers/data/
Dauto_shard.cc439 const NodeDef* input_node = graph_utils::GetInputNode(*node, *graph, 0); in FindFuncAndTensorSliceDataset() local
456 const NodeDef* input_node = graph_utils::GetInputNode(*node, *graph, 0); in FindFuncAndTensorSliceDataset() local
470 const NodeDef* input_node = graph_utils::GetInputNode(node, *graph, 0); in RecursivelyHandleOp() local
482 const NodeDef* input_node = graph_utils::GetInputNode(node, *graph, i); in RecursivelyHandleOp() local
508 const NodeDef* input_node = graph_utils::GetInputNode(node, *graph, 0); in RecursivelyHandleOp() local
554 const NodeDef* input_node = graph_utils::GetInputNode(node, *graph, 0); in RecursivelyHandleOp() local
581 NodeDef* input_node = graph_utils::GetInputNode(sink_node, *graph); in RewriteRebatchV2ToV1() local
630 NodeDef* input_node = graph_utils::GetInputNode(sink_node, *graph); in ShardByData() local
Dslack.cc91 NodeDef* input_node = graph_utils::GetInputNode(*dataset_node, graph, 0); in RecursivelyHandleOp() local
97 NodeDef* input_node = graph_utils::GetInputNode(*dataset_node, graph, i); in RecursivelyHandleOp() local
Dnoop_elimination_test.cc39 string input_node, MutableGraphView *graph) { in MakeNode()
53 std::vector<DataType> param_dtypes, string input_node, in MakeNonConstNode()
68 NodeDef *MakeCacheNode(string input_node, MutableGraphView *graph) { in MakeCacheNode()
/external/tensorflow/tensorflow/tools/graph_transforms/
Dobfuscate_names.cc47 for (const NodeDef& input_node : input_graph_def.node()) { in ObfuscateNames() local
72 for (const NodeDef& input_node : input_graph_def.node()) { in ObfuscateNames() local
Drename_node.cc43 for (const NodeDef& input_node : input_graph_def.node()) { in RenameNode() local
Dfold_batch_norms.cc54 const NodeDef& input_node = match.inputs[0].inputs[0].node; in FoldBatchNorms() local
Dremove_nodes.cc81 const NodeDef& input_node = match.inputs[0].node; in RemoveNodes() local
Dsparsify_gather_test.cc81 NodeDef* input_node = CreateNode("ids", "Const", {}, &graph_def); in TestSinglePartition() local
295 NodeDef* input_node = CreateNode("ids", "Const", {}, &graph_def); in TestMultiPartition() local
Dflatten_atrous.cc57 const NodeDef& input_node = match.inputs[0].inputs[0].inputs[0].node; in FlattenAtrousConv() local
Dtransform_utils.cc351 const NodeDef& input_node = *(node_map_[input_node_name]); in DoesOpTypeMatch() local
388 for (const NodeDef& input_node : input_graph_def.node()) { in ReplaceMatchingOpTypes() local
564 string input_node = NodeNameFromInput(input); in FindInvalidInputs() local
Dfuse_convolutions.cc163 const NodeDef& input_node = match.inputs[0].inputs[0].node; in FusePadAndConv() local
/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
Dconvert_graph.cc123 bool ShallKeepControlEdgeFrom(const Node* input_node) { in ShallKeepControlEdgeFrom()
172 auto input_node = edge->src(); in GetEngineInfo() local
353 Node* input_node = graph->FindNodeId(conn.outside_id); in CreateTRTNode() local
396 Node* input_node = graph->FindNodeId(conn.outside_id); in CreateTRTNode() local
/external/tensorflow/tensorflow/core/kernels/data/experimental/
Dstats_dataset_ops.cc95 Node* input_node; in AsGraphDefInternal() local
212 Node* input_node; in AsGraphDefInternal() local
Dtake_while_dataset_op.cc100 Node* input_node; in AsGraphDefInternal() local
/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_phi_graph.cc110 Node* input_node = CreateOrReuseNode(*input); in RegisterPhi() local
/external/tensorflow/tensorflow/core/common_runtime/
Dmkl_tfconversion_pass.cc385 Node* input_node = nullptr; in RunPass() local
/external/tensorflow/tensorflow/core/grappler/optimizers/
Ddependency_optimizer.cc385 NodeDef* input_node = node_map_->GetNode(node->input(i)); in OptimizeNode() local
553 const NodeDef* input_node = node_map_->GetNode(input); in TransitiveReduction() local
Dconstant_folding.cc278 const NodeDef* input_node = node_map_->GetNode(NodeName(input)); in ForwardInputs() local
849 NodeDef* input_node = nullptr; in MaterializeConstantValuedNode() local
979 const NodeDef* input_node = node_map_->GetNode(input); in IsFoldableUncached() local
1351 const NodeDef* input_node = node_map_->GetNode(input); in EvaluateOneFoldable() local
1424 NodeDef* input_node = node_map_->GetNode(input); in FoldMergeNode() local
1537 NodeDef* input_node = node_map_->GetNode(input); in FoldNode() local
3593 const NodeDef* input_node = node_map_->GetNode(NodeName(input)); in PartialConstPropThroughIdentityN() local
3624 const NodeDef* input_node = node_map_->GetNode(NodeName(input)); in PartialAssocOpConstFolding() local
3820 const NodeDef* input_node = node_map_->GetNode(node->input(i)); in MergeConcat() local
3845 const NodeDef* input_node = node_map_->GetNode(parent->input(i)); in MergeConcat() local
Dlayout_optimizer.cc626 auto input_node = node_map_->GetNode(node_->input(input_index)); in UpdateAttrValueOfInput() local
659 auto input_node = node_map_->GetNode(node_->input(pos)); in AddLayoutTransposeToInputs() local
953 auto input_node = node_map_->GetNode(input); in GetOrAddNodePermNHWCToNCHW() local
1304 auto input_node = node_map_->GetNode(node.input(pos)); in IsNodeAfterNCHWToNHWC() local
1324 auto input_node = node_map_->GetNode(current_node->input(pos)); in IsNodeAfterNCHWToNHWC() local
1460 auto input_node = node_map_->GetNode(node_->input(vector_index)); in CustomizedProcessing() local
1624 auto input_node = node_map_->GetNode(input); in IsEveryInputAfterNCHWToNHWC() local
Dmemory_optimizer.cc114 const NodeDef* input_node = node_map.GetNode(input_name); in FindCandidateRecomputeNodes() local
147 const NodeDef* input_node = node_map.GetNode(input_name_raw); in connected_subgraph() local
821 const NodeDef* input_node = it1->second; in FindSwapInTrigger() local
852 const NodeDef* input_node = it1->second; in FindSwapInTrigger() local
Dauto_parallel.cc181 for (const auto& input_node : input_nodes) { in Initialize() local
/external/tensorflow/tensorflow/compiler/jit/
Dshape_inference.cc191 const Node* input_node; in PropagateShapes() local
/external/tensorflow/tensorflow/compiler/tf2xla/
Dtf2xla_util.cc290 const Node* input_node; in PropagateConstIntoIfNode() local
812 Node* input_node; in RewriteAssociatedFunction() local
1056 Node* input_node; in RewriteTensorListWithConstElement() local
/external/tensorflow/tensorflow/core/kernels/data/
Dcache_dataset_ops.cc646 Node* input_node = nullptr; in AsGraphDefInternal() local
958 Node* input_node = nullptr; in AsGraphDefInternal() local
1003 Node* input_node = nullptr; in AsGraphDefInternal() local

123