Home
last modified time | relevance | path

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

1234

/aosp_15_r20/external/tensorflow/tensorflow/core/grappler/optimizers/data/
H A Dauto_shard.cc486 const NodeDef* input_node = graph_utils::GetInputNode(*node, *graph, 0); in FindFuncAndTensorSliceDataset() local
503 const NodeDef* input_node = graph_utils::GetInputNode(*node, *graph, 0); in FindFuncAndTensorSliceDataset() local
551 const NodeDef* input_node = graph_utils::GetInputNode(node, *graph, 0); in RecursivelyHandleOp() local
563 const NodeDef* input_node = graph_utils::GetInputNode(node, *graph, i); in RecursivelyHandleOp() local
589 const NodeDef* input_node = graph_utils::GetInputNode(node, *graph, 0); in RecursivelyHandleOp() local
635 const NodeDef* input_node = graph_utils::GetInputNode(node, *graph, 0); in RecursivelyHandleOp() local
662 NodeDef* input_node = graph_utils::GetInputNode(sink_node, *graph); in RewriteRebatchV2ToV1() local
711 NodeDef* input_node = graph_utils::GetInputNode(sink_node, *graph); in ShardByData() local
829 NodeDef* input_node = graph_utils::GetInputNode(sink_node, graph); in IsEligibleRewriteBatchSize() local
H A Dnoop_elimination_test.cc39 string input_node, MutableGraphView *graph) { in MakeNode()
54 std::vector<DataType> param_dtypes, string input_node, in MakeNonConstNode()
69 NodeDef *MakeCacheNode(string input_node, MutableGraphView *graph) { in MakeCacheNode()
H A 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
/aosp_15_r20/external/tensorflow/tensorflow/core/graph/
H A 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
H A Dgraph.cc319 Status Node::input_node(int idx, Node** n) const { in input_node() function in tensorflow::Node
330 Status Node::input_node(int idx, const Node** const_n) const { in input_node() function in tensorflow::Node
/aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
H A Dconvert_graph.cc127 bool ShallKeepControlEdgeFrom(const Node* input_node) { in ShallKeepControlEdgeFrom()
175 auto input_node = edge->src(); in GetEngineInfo() local
374 Node* input_node = graph->FindNodeId(conn.outside_id); in CreateTRTNode() local
431 Node* input_node = graph->FindNodeId(conn.outside_id); in CreateTRTNode() local
/aosp_15_r20/external/tensorflow/tensorflow/tools/graph_transforms/
H A 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
H A 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
H A 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
H A Drename_node.cc43 for (const NodeDef& input_node : input_graph_def.node()) { in RenameNode() local
H A Dfold_batch_norms.cc54 const NodeDef& input_node = match.inputs[0].inputs[0].node; in FoldBatchNorms() local
H A Dremove_nodes.cc81 const NodeDef& input_node = match.inputs[0].node; in RemoveNodes() local
/aosp_15_r20/external/tensorflow/tensorflow/core/tfrt/utils/
H A Dgraph_partition.cc55 Node* input_node = nullptr; member
229 for (const auto& input_node : input_nodes) { in BuildPartitionedCallOp() local
249 for (const auto& input_node : input_nodes) { in BuildPartitionedCallOp() local
/aosp_15_r20/external/tensorflow/tensorflow/core/grappler/optimizers/
H A 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
1356 const NodeDef* input_node = node_map_->GetNode(input); in EvaluateOneFoldable() local
1434 NodeDef* input_node = node_map_->GetNode(input); in FoldMergeNode() local
1547 NodeDef* input_node = node_map_->GetNode(input); in FoldNode() local
3630 const NodeDef* input_node = node_map_->GetNode(NodeName(input)); in PartialConstPropThroughIdentityN() local
3661 const NodeDef* input_node = node_map_->GetNode(NodeName(input)); in PartialAssocOpConstFolding() local
3857 const NodeDef* input_node = node_map_->GetNode(node->input(i)); in MergeConcat() local
3882 const NodeDef* input_node = node_map_->GetNode(parent->input(i)); in MergeConcat() local
H A 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
H A Ddependency_optimizer.cc387 NodeDef* input_node = node_map_->GetNode(node->input(i)); in OptimizeNode() local
555 const NodeDef* input_node = node_map_->GetNode(input); in TransitiveReduction() local
/aosp_15_r20/external/pytorch/torch/csrc/jit/passes/
H A Dpeephole.cpp108 Node* input_node = node->input()->node(); in optimizeBlock() local
137 Node* input_node = node->input()->node(); in optimizeBlock() local
H A Dremove_redundant_profiles.cpp27 Node* input_node = n->input()->node(); in RemoveRedundantProfiles() local
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/data/experimental/
H A Dstats_dataset_ops.cc97 Node* input_node; in AsGraphDefInternal() local
216 Node* input_node; in AsGraphDefInternal() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/
H A Dtf2xla_util.cc288 const Node* input_node; in PropagateConstIntoIfNode() local
805 Node* input_node; in RewriteAssociatedFunction() local
1046 Node* input_node; in RewriteTensorListWithConstElement() local
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/data/
H A Dcache_dataset_ops.cc709 Node* input_node = nullptr; in AsGraphDefInternal() local
1048 Node* input_node = nullptr; in AsGraphDefInternal() local
1093 Node* input_node = nullptr; in AsGraphDefInternal() local
/aosp_15_r20/external/executorch/backends/example/example_operators/
H A Dutils.py22 def _annotate_nodes(node_tuples, quant_spec, input_node=False): argument
/aosp_15_r20/external/pytorch/torch/csrc/jit/passes/onnx/
H A Dunpack_quantized_weights.cpp30 double getScaleFromInput(Node* input_node) { in getScaleFromInput()
727 auto input_node = match_vmap.at(vmap.at("r"))->node()->inputs()[0]->node(); in insertPermutesHelper() local
/aosp_15_r20/external/executorch/backends/transforms/
H A Dduplicate_dynamic_quant_chain.py39 def _replace_input_node_with_new_node(node, input_node, new_node): argument
/aosp_15_r20/external/executorch/backends/arm/_passes/
H A Dannotate_channels_last_dim_order_pass.py86 def insert_input_transpose(self, node, input_node, graph_module): argument

1234