| /aosp_15_r20/external/pytorch/aten/src/ATen/functorch/ |
| H A D | BatchRulesModules.cpp | 116 auto new_input = reshape_dim_into(*input_bdim, 1, input); in grid_sample_batch_rule() local
127 auto new_input = reshape_dim_into(*input_bdim, 0, input); in grid_sample_batch_rule() local
189 auto new_input = std::get<1>(new_bw_input); in grid_sample_backward_batch_rule() local
209 auto new_input = std::get<1>(new_bw_input); in cudnn_grid_sample_backward_batch_rule() local
|
| /aosp_15_r20/external/pytorch/torch/csrc/jit/passes/ |
| H A D | autocast.cpp | 169 const auto new_input = graph->insert( in castTensorInputs() local
176 const auto new_input = graph->insert( in castTensorInputs() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| H A D | conv_canonicalization.cc | 77 HloInstruction* new_input = module->entry_computation()->AddInstruction( in Run() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/ops/ |
| H A D | while_v2_indexed_slices_rewriter.py | 231 def rewrite(old_output, new_input): argument
|
| /aosp_15_r20/external/executorch/kernels/quantized/test/ |
| H A D | op_choose_qparams_test.cpp | 147 Tensor new_input = tf_float.make( in TEST() local
|
| /aosp_15_r20/external/rust/android-crates-io/crates/toml_edit/src/parser/ |
| D | mod.rs | 86 pub(crate) fn new_input(s: &str) -> Input<'_> { in new_input() function
|
| /aosp_15_r20/art/compiler/optimizing/ |
| H A D | intrinsics.cc | 254 HInvokeStaticOrDirect* new_input = new (allocator) HInvokeStaticOrDirect( in InsertFpToIntegralIntrinsic() local
|
| /aosp_15_r20/external/pytorch/torch/csrc/jit/passes/onnx/ |
| H A D | helper.cpp | 212 auto new_input = in transformToONNXConcatNode() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| H A D | gpu_conv_padding_legalization.cc | 161 HloInstruction* new_input = MaybePaddedAndSlicedInput( in CanonicalizeForwardConvolution() local
|
| /aosp_15_r20/external/mesa3d/src/gallium/drivers/r300/compiler/ |
| H A D | radeon_compiler.c | 224 void rc_transform_fragment_wpos(struct radeon_compiler * c, unsigned wpos, unsigned new_input, in rc_transform_fragment_wpos()
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/internal/reference/ |
| H A D | lstm_cell.h | 120 const float new_input = std::tanh(activ_temp_data[Offset( in LstmCell() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/grappler/optimizers/ |
| H A D | dependency_optimizer.cc | 415 string new_input; in OptimizeNode() local
|
| H A D | scoped_allocator_optimizer.cc | 219 int output_index, NodeDef* op, NodeDef** new_input, in MaybeRewriteInput()
|
| H A D | arithmetic_optimizer.cc | 117 bool MaybeAddControlInput(const string& new_input, NodeDef* node, in MaybeAddControlInput()
256 Status UpdateConsumers(NodeDef* node, const string& new_input) { in UpdateConsumers()
|
| /aosp_15_r20/external/mesa3d/src/microsoft/clc/ |
| H A D | clc_compiler.c | 222 nir_variable *new_input = NULL; in clc_lower_input_image_deref() local
|
| /aosp_15_r20/external/pytorch/torch/csrc/jit/runtime/ |
| H A D | autodiff.cpp | 802 auto new_input = reverse_block->addInput()->copyMetadata(captured); in lambdaLiftReverse() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/graph/ |
| H A D | graph.cc | 645 const std::string new_input = strings::StrCat("^", source->name()); in AddControlEdge() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/ |
| H A D | graph_constructor.cc | 874 TensorId new_input = iter->second; in RemapNodeDefInputs() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/spmd/ |
| H A D | spmd_partitioner.cc | 3000 auto new_input = in HandleDynamicSlice() local
3207 auto new_input = in HandleDynamicUpdateSlice() local
|
| /aosp_15_r20/external/pytorch/torch/csrc/jit/frontend/ |
| H A D | ir_emitter.cpp | 858 Value* new_input = block->addInput()->setDebugName(name); in emitFormalArguments() local
877 Value* new_input = block->addInput(); in emitFormalArguments() local
|
| /aosp_15_r20/external/mesa3d/src/compiler/nir/ |
| H A D | nir_opt_varyings.c | 3087 nir_def *new_input, *new_tes_loads[3]; in try_move_postdominator() local
|