| /aosp_15_r20/external/pytorch/aten/src/ATen/native/vulkan/ops/ |
| H A D | Select.cpp | 15 const vTensor& v_input = convert(input); in select_batch_4d() local
67 const vTensor& v_input = convert(input); in select_depth_3d() local
114 const vTensor& v_input = convert(input); in select_depth_4d() local
167 const vTensor& v_input = convert(input); in select_height_3d() local
226 const vTensor& v_input = convert(input); in select_height_4d() local
280 const vTensor& v_input = convert(input); in select_width_3d() local
340 const vTensor& v_input = convert(input); in select_width_4d() local
|
| H A D | QuantizedTensor.cpp | 41 const vTensor& v_input = convert(input); in quantize_per_tensor() local
116 const vTensor& v_input = convert(input); in dequantize_helper() local
167 const vTensor& v_input = convert(self); in q_scale() local
173 const vTensor& v_input = convert(self); in q_zero_point() local
|
| H A D | Mm.cpp | 37 vTensor v_input = convert(input); in pack_inputs_using_width_packing() local
404 const auto v_input = convert(input); in usable() local
442 const vTensor& v_input = convert(input); in run_quantized_addmm_context() local
666 const vTensor& v_input = pack_inputs_using_width_packing(input); in run_addmm_context() local
|
| H A D | Upsample.cpp | 24 const vTensor& v_input = convert(input); in upsample_nearest2d() local
110 const vTensor& v_input = convert(input); in upsample_bilinear2d() local
|
| H A D | Batchnorm.cpp | 21 const vTensor& v_input, in record_op()
201 const vTensor& v_input = convert(input_arg); in run_batchnorm_context() local
|
| H A D | Convolution.cpp | 366 const vTensor& v_input, in record_op()
439 const vTensor& v_input, in record_quantized_op()
858 const vTensor& v_input = convert(input); in run_conv1d_context_impl() local
1116 const vTensor& v_input = convert(input_arg); in run_conv2d_context_impl() local
|
| H A D | Glu.cpp | 24 const vTensor& v_input = convert(input); in glu() local
|
| H A D | Flip.cpp | 23 const vTensor& v_input = convert(input); in flip() local
|
| H A D | Unsqueeze.cpp | 35 const vTensor& v_input = convert(input); in unsqueeze() local
|
| H A D | Mean.cpp | 35 const vTensor& v_input = convert(input); in mean_dim() local
|
| H A D | Sum.cpp | 27 const vTensor& v_input = convert(input); in sum_dim() local
|
| H A D | cumsum.cpp | 108 const vTensor& v_input = convert(input); in cumsum() local
|
| H A D | Softmax.cpp | 105 const vTensor& v_input = convert(input); in softmax_internal() local
|
| H A D | Utils.cpp | 324 const vTensor& v_input = convert(input); in extract_texel() local
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/vulkan/impl/ |
| H A D | Packing.cpp | 300 const vTensor& v_input, in channel_image_repacking()
350 vTensor convert_image_channels_packed_to_height_packed(const vTensor& v_input) { in convert_image_channels_packed_to_height_packed()
357 vTensor convert_image_channels_packed_to_width_packed(const vTensor& v_input) { in convert_image_channels_packed_to_width_packed()
|
| /aosp_15_r20/external/libaom/av1/encoder/arm/ |
| H A D | ml_neon.c | 152 const float32x4_t v_input = vld1q_f32(&inputs[in]); in nn_propagate_4to4() local
191 const float32x4_t v_input = vld1q_f32(&inputs[in]); in nn_propagate_4to8() local
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/test/ |
| H A D | vulkan_quantized_api_test.cpp | 3612 auto v_input = at::native::vulkan::ops::convert(data.vulkan()); in TEST_F() local
3664 auto v_input = at::native::vulkan::ops::convert(data.vulkan()); in TEST_F() local
|
| H A D | vulkan_api_test.cpp | 7281 auto run(const at::Tensor& input, const at::Tensor& v_input) { in run()
|