Home
last modified time | relevance | path

Searched defs:index_put_ (Results 1 – 3 of 3) sorted by relevance

/external/pytorch/aten/src/ATen/
DTensorIndexing.cpp76 Tensor & Tensor::index_put_(ArrayRef<at::indexing::TensorIndex> indices, Tensor const & rhs) { in index_put_() function in at::Tensor
82 Tensor & Tensor::index_put_(ArrayRef<at::indexing::TensorIndex> indices, const Scalar& v) { in index_put_() function in at::Tensor
88 Tensor & Tensor::index_put_(std::initializer_list<at::indexing::TensorIndex> indices, Tensor const … in index_put_() function in at::Tensor
91 Tensor & Tensor::index_put_(std::initializer_list<at::indexing::TensorIndex> indices, const Scalar&… in index_put_() function in at::Tensor
/external/pytorch/aten/src/ATen/native/
DTensorAdvancedIndexing.cpp867 Tensor & index_put_(Tensor & self, const torch::List<std::optional<Tensor>>& indices, const Tensor … in index_put_() function
/external/pytorch/torch/_inductor/
Dlowering.py3190 def index_put_(self, indices, values, accumulate=False): function