Home
last modified time | relevance | path

Searched defs:input_node (Results 26 – 50 of 83) sorted by relevance

1234

/aosp_15_r20/external/tensorflow/tensorflow/tools/graph_transforms/
H A Dflatten_atrous.cc57 const NodeDef& input_node = match.inputs[0].inputs[0].inputs[0].node; in FlattenAtrousConv() local
H A Dfuse_convolutions.cc163 const NodeDef& input_node = match.inputs[0].inputs[0].node; in FusePadAndConv() local
H A Dfold_old_batch_norms.cc112 const NodeDef& input_node = conv_node_match.inputs[0].node; in FuseScaleOffsetToConvWeights() local
/aosp_15_r20/external/tensorflow/tensorflow/core/tfrt/utils/
H A Dgraph_partition_test.cc148 NodeDef* input_node = nullptr; in TEST_F() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/
H A Dhlo_phi_graph.cc110 Node* input_node = CreateOrReuseNode(*input); in RegisterPhi() local
/aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/
H A Dmkl_tfconversion_pass.cc385 Node* input_node = nullptr; in RunPass() local
H A Dlower_while_op.cc328 const Node* input_node; in CreateSwitchNodes() local
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/data/experimental/
H A Dtake_while_dataset_op.cc100 Node* input_node; in AsGraphDefInternal() local
H A Ddense_to_sparse_batch_dataset_op.cc137 Node* input_node; in AsGraphDefInternal() local
H A Dscan_dataset_op.cc140 Node* input_node; in AsGraphDefInternal() local
H A Dchoose_fastest_dataset_op.cc183 Node* input_node; in AsGraphDefInternal() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/jit/
H A Dshape_inference.cc188 const Node* input_node; in PropagateShapes() local
H A Drearrange_function_argument_pass_test.cc139 const Node *input_node; in TEST() local
/aosp_15_r20/external/tensorflow/tensorflow/core/grappler/optimizers/
H A Dauto_parallel.cc181 for (const auto& input_node : input_nodes) { in Initialize() local
H A Dmodel_pruner.cc161 const NodeDef* input_node = graph_view.GetNode(NodeName(input)); in ForwardInputsInternal() local
/aosp_15_r20/external/executorch/backends/qualcomm/builders/
H A Dnode_visitor.py92 def get_tensor(self, input_node, op_node, idx=None): argument
/aosp_15_r20/external/tensorflow/tensorflow/core/grappler/costs/
H A Dvirtual_scheduler.cc500 const NodeDef* input_node = name_to_node[NodeName(input_node_name)]; in Init() local
661 const NodeDef* from, const NodeDef* to, const NodeDef* input_node, in CreateSendRecv()
H A Dgraph_memory.cc227 string input_node = ParseNodeName(input, &position); in InferFromTrace() local
/aosp_15_r20/external/executorch/backends/arm/
H A Dtosa_quant_utils.py280 input_node, argument
/aosp_15_r20/external/mesa3d/src/gallium/drivers/r300/compiler/
H A Dradeon_pair_regalloc.c256 unsigned int i, input_node, node_count, node_index; in do_advanced_regalloc() local
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/data/
H A Dzip_dataset_op.cc152 Node* input_node; in AsGraphDefInternal() local
H A Dinterleave_dataset_op.cc104 Node* input_node; in AsGraphDefInternal() local
/aosp_15_r20/external/executorch/backends/xnnpack/partition/config/
H A Dgemm_configs.py381 def find_partition_args(input_node): argument
/aosp_15_r20/external/tensorflow/tensorflow/core/distributed_runtime/
H A Dcluster_function_library_runtime.cc84 Node* input_node; in ConstructFunctionGraph() local
/aosp_15_r20/external/pytorch/torch/ao/quantization/fx/
H A Dconvert.py142 def add_dequantize_op_kwargs(dequantize_op, input_node): argument

1234