Home
last modified time | relevance | path

Searched defs:target_sizes (Results 1 – 12 of 12) sorted by relevance

/aosp_15_r20/external/executorch/kernels/portable/cpu/
H A Dop_narrow_copy.cpp39 Tensor::SizesType target_sizes[kTensorDimensionLimit]; in narrow_copy_out() local
H A Dop_reflection_pad1d.cpp36 Tensor::SizesType target_sizes[kTensorDimensionLimit]; in reflection_pad1d_out() local
H A Dop_pdist_forward.cpp32 Tensor::SizesType target_sizes[kTensorDimensionLimit]; in _pdist_forward_out() local
H A Dop_reflection_pad3d.cpp36 Tensor::SizesType target_sizes[kTensorDimensionLimit]; in reflection_pad3d_out() local
H A Dop_replication_pad2d.cpp28 Tensor::SizesType target_sizes[kTensorDimensionLimit]; in replication_pad2d_out() local
H A Dop_reflection_pad2d.cpp36 Tensor::SizesType target_sizes[kTensorDimensionLimit]; in reflection_pad2d_out() local
H A Dop_replication_pad3d.cpp28 Tensor::SizesType target_sizes[kTensorDimensionLimit]; in replication_pad3d_out() local
H A Dop_replication_pad1d.cpp28 Tensor::SizesType target_sizes[kTensorDimensionLimit]; in replication_pad1d_out() local
H A Dop_slice_copy.cpp49 Tensor::SizesType target_sizes[kTensorDimensionLimit]; in slice_copy_Tensor_out() local
H A Dop_cdist_forward.cpp138 Tensor::SizesType target_sizes[kTensorDimensionLimit]; in _cdist_forward_out() local
/aosp_15_r20/external/executorch/kernels/portable/cpu/util/
H A Dselect_copy_util.cpp33 Tensor::SizesType target_sizes[kTensorDimensionLimit]; in select_copy_util() local
/aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/
H A DSpectralOps.cpp359 auto target_sizes = dim.size() == 1 ? out_sizes : onesided_sizes; in _fft_r2c_cufft() local