| /external/tensorflow/tensorflow/core/graph/ |
| D | benchmark_testlib.h | 128 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/ |
| D | slack.cc | 91 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
|
| D | auto_shard.cc | 486 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
|
| D | noop_elimination_test.cc | 39 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()
|
| /external/tensorflow/tensorflow/tools/graph_transforms/ |
| D | obfuscate_names.cc | 47 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
|
| D | rename_node.cc | 43 for (const NodeDef& input_node : input_graph_def.node()) { in RenameNode() local
|
| D | fold_batch_norms.cc | 54 const NodeDef& input_node = match.inputs[0].inputs[0].node; in FoldBatchNorms() local
|
| D | remove_nodes.cc | 81 const NodeDef& input_node = match.inputs[0].node; in RemoveNodes() local
|
| D | sparsify_gather_test.cc | 81 NodeDef* input_node = CreateNode("ids", "Const", {}, &graph_def); in TestSinglePartition() local
295 NodeDef* input_node = CreateNode("ids", "Const", {}, &graph_def); in TestMultiPartition() local
|
| D | flatten_atrous.cc | 57 const NodeDef& input_node = match.inputs[0].inputs[0].inputs[0].node; in FlattenAtrousConv() local
|
| D | transform_utils.cc | 351 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
|
| /external/executorch/backends/example/example_operators/ |
| D | utils.py | 22 def _annotate_nodes(node_tuples, quant_spec, input_node=False): argument
|
| /external/pytorch/torch/csrc/jit/passes/ |
| D | remove_redundant_profiles.cpp | 27 Node* input_node = n->input()->node(); in RemoveRedundantProfiles() local
|
| D | peephole.cpp | 108 Node* input_node = node->input()->node(); in optimizeBlock() local
137 Node* input_node = node->input()->node(); in optimizeBlock() local
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| D | convert_graph.cc | 127 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
|
| /external/tensorflow/tensorflow/core/tfrt/utils/ |
| D | graph_partition.cc | 55 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
|
| D | graph_partition_test.cc | 148 NodeDef* input_node = nullptr; in TEST_F() local
|
| /external/executorch/backends/arm/_passes/ |
| D | annotate_channels_last_dim_order_pass.py | 86 def insert_input_transpose(self, node, input_node, graph_module): argument
|
| /external/executorch/backends/transforms/ |
| D | duplicate_dynamic_quant_chain.py | 39 def _replace_input_node_with_new_node(node, input_node, new_node): argument
|
| /external/tensorflow/tensorflow/core/kernels/data/experimental/ |
| D | stats_dataset_ops.cc | 97 Node* input_node; in AsGraphDefInternal() local
216 Node* input_node; in AsGraphDefInternal() local
|
| D | take_while_dataset_op.cc | 100 Node* input_node; in AsGraphDefInternal() local
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_phi_graph.cc | 110 Node* input_node = CreateOrReuseNode(*input); in RegisterPhi() local
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | mkl_tfconversion_pass.cc | 385 Node* input_node = nullptr; in RunPass() local
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | dependency_optimizer.cc | 387 NodeDef* input_node = node_map_->GetNode(node->input(i)); in OptimizeNode() local
555 const NodeDef* input_node = node_map_->GetNode(input); in TransitiveReduction() local
|
| D | constant_folding.cc | 278 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
|