Home
last modified time | relevance | path

Searched defs:IntArrayRef (Results 1 – 15 of 15) sorted by relevance

/aosp_15_r20/external/pytorch/aten/src/ATen/native/xnnpack/
H A DShim.cpp35 const IntArrayRef, in use_convolution2d()
47 const IntArrayRef, in convolution2d()
70 const IntArrayRef, in use_max_pool2d()
82 const IntArrayRef, in max_pool2d()
/aosp_15_r20/external/executorch/runtime/core/exec_aten/
H A Dexec_aten.h87 using IntArrayRef = at::IntArrayRef; variable
132 using IntArrayRef = torch::executor::IntArrayRef; variable
/aosp_15_r20/external/pytorch/aten/src/ATen/native/quantized/
H A DTensorFactories.cpp107 IntArrayRef, in empty_affine_quantized_other_backends_stub() argument
119 IntArrayRef, in empty_per_channel_affine_quantized_other_backends_stub() argument
/aosp_15_r20/external/pytorch/aten/src/ATen/core/
H A DATen_fwd.h34 using IntArrayRef = c10::ArrayRef<int64_t>; variable
/aosp_15_r20/external/executorch/kernels/portable/cpu/
H A Dop_permute_copy.cpp18 using IntArrayRef = exec_aten::ArrayRef<int64_t>; typedef
H A Dop_max_pool2d_with_indices.cpp21 using IntArrayRef = exec_aten::ArrayRef<int64_t>; typedef
H A Dop_avg_pool2d.cpp21 using IntArrayRef = exec_aten::ArrayRef<int64_t>; typedef
H A Dop_convolution_backward.cpp21 using IntArrayRef = exec_aten::ArrayRef<int64_t>; typedef
H A Dop_convolution.cpp22 using IntArrayRef = exec_aten::ArrayRef<int64_t>; typedef
/aosp_15_r20/external/pytorch/aten/src/ATen/native/vulkan/ops/
H A DFactory.cpp50 const IntArrayRef sizes, in empty_strided()
H A DConvolution.cpp660 const IntArrayRef stride, in available()
/aosp_15_r20/external/executorch/kernels/test/
H A Dop_convolution_backward_test.cpp22 using IntArrayRef = exec_aten::ArrayRef<int64_t>; typedef
/aosp_15_r20/external/executorch/runtime/core/
H A Darray_ref.h237 using IntArrayRef = ArrayRef<int64_t>; variable
/aosp_15_r20/external/pytorch/c10/util/
H A DArrayRef.h374 using IntArrayRef = ArrayRef<int64_t>; variable
/aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/
H A DFractionalMaxPool3d.cu311 IntArrayRef /*pool_size*/, in fractional_max_pool3d_backward_out_cuda() argument