Home
last modified time | relevance | path

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

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