Home
last modified time | relevance | path

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

123

/external/tensorflow/tensorflow/tools/graph_transforms/
Dobfuscate_names.cc48 for (const NodeDef& input_node : input_graph_def.node()) { in ObfuscateNames() local
73 for (const NodeDef& input_node : input_graph_def.node()) { in ObfuscateNames() local
Dfold_batch_norms.cc55 const NodeDef& input_node = match.inputs[0].inputs[0].node; in FoldBatchNorms() local
Dremove_nodes.cc82 const NodeDef& input_node = match.inputs[0].node; in RemoveNodes() local
Dflatten_atrous.cc57 const NodeDef& input_node = match.inputs[0].inputs[0].inputs[0].node; in FlattenAtrousConv() 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
Dfuse_convolutions.cc164 const NodeDef& input_node = match.inputs[0].inputs[0].node; in FusePadAndConv() local
Dtransform_utils.cc349 const NodeDef& input_node = *(node_map_[input_node_name]); in DoesOpTypeMatch() local
386 for (const NodeDef& input_node : input_graph_def.node()) { in ReplaceMatchingOpTypes() local
562 string input_node = NodeNameFromInput(input); in FindInvalidInputs() local
/external/tensorflow/tensorflow/core/graph/
Dbenchmark_testlib.h128 const uint32 input_node = rnd.Uniform(i); in CreateRandomGraph() local
151 NodeDef input_node = create_node(/*name=*/input_node_name); in CreateFaninFanoutNodeGraph() local
Dmkl_tfconversion_pass.cc383 Node* input_node = nullptr; in RunPass() local
/external/tensorflow/tensorflow/core/grappler/optimizers/data/
Dnoop_elimination_test.cc38 NodeDef *MakeUnaryNode(StringPiece node_type, int count, string input_node, in MakeUnaryNode()
46 NodeDef *MakeUnaryNonConstNode(StringPiece node_type, string input_node, in MakeUnaryNonConstNode()
54 NodeDef *MakeCacheNode(string input_node, MutableGraphView *graph) { in MakeCacheNode()
Dauto_shard.cc187 const NodeDef* input_node = graph_utils::GetInputNode(node, *graph, i); in RecursivelyHandleOp() local
222 const NodeDef* input_node = graph_utils::GetInputNode(node, *graph, 0); in RecursivelyHandleOp() local
Dmap_vectorization_test.cc338 auto input_node = AddRangeNode(&graph); in TEST_P() local
432 auto input_node = AddArbitraryInputNode(&graph, nullptr, &input_types); in TEST() local
452 auto input_node = AddArbitraryInputNode(&graph, &input_shapes, &input_types); in TEST() local
472 auto input_node = AddArbitraryInputNode(&graph, &input_shapes, &input_types); in TEST() local
491 auto input_node = AddArbitraryInputNode(&graph, &input_shapes, nullptr); in TEST() local
Drebatch.cc194 NodeDef* input_node = graph_utils::GetInputNode(node, *graph, i); in RecursivelyHandleOp() local
201 NodeDef* input_node = graph_utils::GetInputNode(node, *graph, 0); in RecursivelyHandleOp() local
Dmap_vectorization.cc191 Status AddNewBatchNode(const NodeDef& old_batch_node, const NodeDef& input_node, in AddNewBatchNode()
457 const NodeDef*& input_node = *input_node_output; in FindMapAndBatchPattern() local
525 const NodeDef* input_node; in OptimizeAndCollectStats() local
Dvectorization_utils.cc444 Node* input_node; in StackTensor() local
511 Node* input_node; in AddArgTensorMappings() local
/external/tensorflow/tensorflow/core/kernels/data/
Ddataset_ops.cc35 Node* input_node = nullptr; in Compute() local
Dzip_dataset_op.cc101 Node* input_node; in AsGraphDefInternal() local
Dgraph_rewrite_dataset.cc35 Node* input_node = nullptr; in Optimize() local
/external/tensorflow/tensorflow/contrib/receptive_field/python/util/
Dreceptive_field.py154 input_node, argument
/external/tensorflow/tensorflow/core/kernels/data/experimental/
Dstats_dataset_ops.cc84 Node* input_node; in AsGraphDefInternal() local
194 Node* input_node; in AsGraphDefInternal() local
/external/tensorflow/tensorflow/core/kernels/hexagon/
Dgraph_transferer.cc129 for (const Node* const input_node : node->in_nodes()) { in LoadGraphFromProto() local
313 for (const Node* const input_node : node.in_nodes()) { in AreAllInputsCached() local
351 Node* input_node; in TransformGraphToAddAggregatedInputNode() local
733 const Node* input_node = edge->src(); in RegisterPadNode() local
874 const Node* input_node = edge->src(); in AddNodeInputByInputIndex() local
/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
Dconvert_graph.cc251 auto input_node = edge->src(); in GetEngineInfo() local
425 Node* input_node = graph->FindNodeId(conn.outside_id); in CreateTRTNode() local
475 Node* input_node = graph->FindNodeId(conn.outside_id); in CreateTRTNode() local
/external/tensorflow/tensorflow/compiler/tf2xla/
Dtf2xla_util.cc229 const Node* input_node; in PropagateConstIntoIfNode() local
267 const Node* input_node; in PropagateConstIntoWhileNode() local
635 Node* input_node; in RewriteAssociatedFunction() local
/external/tensorflow/tensorflow/core/grappler/optimizers/
Ddependency_optimizer.cc359 NodeDef* input_node = node_map_->GetNode(node->input(i)); in OptimizeNode() local
490 const NodeDef* input_node = node_map_->GetNode(input); in TransitiveReduction() local
/external/tensorflow/tensorflow/core/distributed_runtime/
Dcluster_function_library_runtime.cc47 NodeDef* input_node = g->add_node(); in ConstructFunctionGraph() local

123