Lines Matching full:indexing
7 namespace indexing { namespace
54 value = at::indexing::scalarToTensor(v, device(kCPU).dtype(kFloat), at::Device(kCPU)); in set_item()
56 value = at::indexing::scalarToTensor(v, self.options(), at::Device(kCPU)); in set_item()
58 value = at::indexing::scalarToTensor(v, self.options(), self_device); in set_item()
65 } // namespace indexing
67 Tensor Tensor::index(ArrayRef<at::indexing::TensorIndex> indices) const { in index()
70 return at::indexing::get_item(*this, indices); in index()
72 Tensor Tensor::index(std::initializer_list<at::indexing::TensorIndex> indices) const { in index()
73 return index(ArrayRef<at::indexing::TensorIndex>(indices)); in index()
76 Tensor & Tensor::index_put_(ArrayRef<at::indexing::TensorIndex> indices, Tensor const & rhs) { in index_put_()
79 at::indexing::set_item(*this, indices, rhs); in index_put_()
82 Tensor & Tensor::index_put_(ArrayRef<at::indexing::TensorIndex> indices, const Scalar& v) { in index_put_()
85 at::indexing::set_item(*this, indices, v); in index_put_()
88 Tensor & Tensor::index_put_(std::initializer_list<at::indexing::TensorIndex> indices, Tensor const … in index_put_()
89 return index_put_(ArrayRef<at::indexing::TensorIndex>(indices), rhs); in index_put_()
91 Tensor & Tensor::index_put_(std::initializer_list<at::indexing::TensorIndex> indices, const Scalar&… in index_put_()
92 return index_put_(ArrayRef<at::indexing::TensorIndex>(indices), v); in index_put_()