| /aosp_15_r20/external/pytorch/aten/src/ATen/native/vulkan/ops/ |
| H A D | Select.cpp | 11 Tensor select_batch_4d(const Tensor& input_arg, uint32_t index) { in select_batch_4d()
63 Tensor select_depth_3d(const Tensor& input_arg, uint32_t index) { in select_depth_3d()
110 Tensor select_depth_4d(const Tensor& input_arg, uint32_t index) { in select_depth_4d()
163 Tensor select_height_3d(const Tensor& input_arg, uint32_t index) { in select_height_3d()
222 Tensor select_height_4d(const Tensor& input_arg, uint32_t index) { in select_height_4d()
276 Tensor select_width_3d(const Tensor& input_arg, uint32_t index) { in select_width_3d()
336 Tensor select_width_4d(const Tensor& input_arg, uint32_t index) { in select_width_4d()
|
| H A D | Mm.cpp | 20 vTensor pack_inputs_using_width_packing(const Tensor& input_arg) { in pack_inputs_using_width_packing()
415 static Tensor reshape_to_2d(const Tensor& input_arg) { in reshape_to_2d()
430 const Tensor& input_arg, in run_quantized_addmm_context()
643 const Tensor& input_arg, in run_addmm_context()
763 const Tensor& input_arg, in run_baddbmm_context()
989 const Tensor& input_arg, in run_qlinear_context()
|
| H A D | Softmax.cpp | 89 const at::Tensor& input_arg, in softmax_internal()
186 const at::Tensor& input_arg, in softmax()
193 const at::Tensor& input_arg, in log_softmax()
|
| H A D | QuantizedTensor.cpp | 32 const at::Tensor& input_arg, in quantize_per_tensor()
94 const at::Tensor& input_arg, in quantize_per_tensor_tensor_qparams()
107 const at::Tensor& input_arg, in dequantize_helper()
|
| H A D | Slice.cpp | 14 const Tensor& input_arg, in slice_4d()
83 const Tensor& input_arg, in slice_width()
159 const Tensor& input_arg, in slice_height()
|
| H A D | Layernorm.cpp | 59 const Tensor& input_arg, in run_layernorm_context()
82 const at::Tensor& input_arg, in layer_norm()
|
| H A D | Random.cpp | 63 const at::Tensor& input_arg, in rand_like()
124 const at::Tensor& input_arg, in randn_like()
|
| H A D | Upsample.cpp | 13 const Tensor& input_arg, in upsample_nearest2d()
98 const Tensor& input_arg, in upsample_bilinear2d()
|
| H A D | Convolution.cpp | 821 const Tensor& input_arg, in run_conv1d_context_impl()
1109 const Tensor& input_arg, in run_conv2d_context_impl()
1230 const Tensor& input_arg, in run_conv2d_context()
1236 const Tensor& input_arg, in run_tconv2d_context()
1242 const Tensor& input_arg, in run_qconv2d_context()
|
| H A D | Batchnorm.cpp | 75 const at::Tensor& input_arg, in batch_norm()
197 const Tensor& input_arg, in run_batchnorm_context()
|
| H A D | Glu.cpp | 12 Tensor glu(const at::Tensor& input_arg, const int64_t dim = -1) { in glu()
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/distribute/failure_handling/ |
| H A D | gce_failure_handler_test.py | 107 input_arg, argument
247 def test_basic_run(self, input_arg, mwms_mode): argument
322 input_arg, mwms_mode): argument
397 def test_grace_period_continue_training(self, input_arg, mwms_mode): argument
|
| H A D | failure_handler_test.py | 119 input_arg='checkpoint', argument
230 def test_preemption_checkpointing(self, input_arg, mwms_mode): argument
353 def test_grace_period_continue_training(self, input_arg, mwms_mode): argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/framework/ |
| H A D | node_def_builder.cc | 115 void NodeDefBuilder::SingleInput(const OpDef::ArgDef* input_arg, in SingleInput()
136 void NodeDefBuilder::ListInput(const OpDef::ArgDef* input_arg, in ListInput()
186 void NodeDefBuilder::VerifyInputType(const OpDef::ArgDef* input_arg, in VerifyInputType()
195 void NodeDefBuilder::VerifyInputRef(const OpDef::ArgDef* input_arg, in VerifyInputRef()
|
| H A D | node_def_builder.h | 175 DataType MaybeAddRef(const OpDef::ArgDef* input_arg, DataType dt) { in MaybeAddRef()
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/ |
| H A D | ScanKernels.cpp | 31 TensorArg input_arg{ self, "input", 3 }; in cummax_helper_cuda() local
48 TensorArg input_arg{ self, "input", 3 }; in cummin_helper_cuda() local
75 TensorArg input_arg{ self, "input", 2 }; in _logcumsumexp_out_cuda() local
|
| H A D | NaiveConvolutionTranspose2d.cu | 158 TensorArg input_arg{input, "input", 1}, output_arg{output, "output", 2}, in slow_conv_transpose2d_out_cuda_template() local
338 TensorArg input_arg{input_, "input", 1}, in slow_conv_transpose2d_backward_out_cuda_template() local
517 TensorArg input_arg{input_, "input", 1}, in slow_conv_transpose2d_acc_grad_parameters_cuda_template() local
|
| H A D | NaiveConvolutionTranspose3d.cu | 227 TensorArg input_arg{input_, "input", 1}, output_arg{output, "output", 2}, in slow_conv_transpose3d_out_cuda_template() local
448 TensorArg input_arg{input_, "input", 1}, in slow_conv_transpose3d_backward_out_cuda_template() local
656 TensorArg input_arg{input_, "input", 1}, in slow_conv_transpose3d_acc_grad_parameters_cuda() local
|
| H A D | AdaptiveMaxPooling2d.cu | 211 TensorArg input_arg{input, "input", 3}; in TORCH_IMPL_FUNC() local
333 TensorArg input_arg{input, "input", 3}; in TORCH_IMPL_FUNC() local
|
| H A D | AdaptiveMaxPooling3d.cu | 312 TensorArg input_arg{input, "input", 3}; in TORCH_IMPL_FUNC() local
392 TensorArg input_arg{input, "input", 3}; in TORCH_IMPL_FUNC() local
|
| H A D | AveragePool2d.cu | 256 TensorArg input_arg{ input_, "input_", 2 }; in TORCH_IMPL_FUNC() local
370 TensorArg input_arg{ input_, "input_", 3 }; in TORCH_IMPL_FUNC() local
|
| H A D | TensorTopK.cpp | 52 …TensorArg topK_arg{values, "topK", 1}, indices_arg{indices, "indices", 2}, input_arg{self, "self",… in TORCH_IMPL_FUNC() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/grappler/utils/ |
| H A D | functions.cc | 58 for (const InputArgInstantiation& input_arg : input_args_) { in GrapplerFunctionItem() local
327 const InputArgInstantiation& input_arg = item->input(input_index); in ReplaceInputWithConst() local
449 for (const InputArgInstantiation& input_arg : item.inputs()) { in Initialize() local
544 for (const InputArgInstantiation& input_arg : item.inputs()) { in MakeFunctionDef() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/framework/ |
| H A D | op_def_library_pybind.cc | 435 const py::object& input_tensor, const ArgDef& input_arg, in GetInputType()
487 for (const ArgDef& input_arg : op_def.input_arg()) { in ExtractInputsAndAttrs() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/grappler/optimizers/data/ |
| H A D | fusion_utils.cc | 161 for (const auto& input_arg : second_signature.input_arg()) { in GetUniqueSignature() local
387 for (const auto& input_arg : first_function.signature().input_arg()) { in LazyConjunctionNodes() local
|