Home
last modified time | relevance | path

Searched defs:gather_out (Results 1 – 4 of 4) sorted by relevance

/aosp_15_r20/external/executorch/kernels/portable/cpu/
H A Dop_gather.cpp62 Tensor& gather_out( in gather_out() function
/aosp_15_r20/external/pytorch/torch/csrc/cuda/
H A Dcomm.cpp391 at::Tensor& gather_out( in _broadcast_out_impl() function
/aosp_15_r20/external/pytorch/aten/src/ATen/native/
H A DNamedTensor.cpp338 Tensor& gather_out(const Tensor& self, Dimname dim, const Tensor& index, bool sparse_grad, Tensor& … in gather_out() function
H A DTensorAdvancedIndexing.cpp1638 TORCH_IMPL_FUNC(gather_out) in TORCH_IMPL_FUNC() argument