#pragma once // @generated by torchgen/gen.py from NativeMetaFunction.h #include #include #include #include #include #include #include #include #include #include #include namespace at { namespace meta { struct TORCH_API structured_index_Tensor : public TensorIteratorBase { template struct TORCH_API precompute_out { precompute_out set_sizes(at::DimVector value) { static_assert(SIZES == false, "sizes already set"); precompute_out ret; ret.sizes = value; ret.strides = this->strides; return ret; } precompute_out set_strides(at::DimVector value) { static_assert(STRIDES == false, "strides already set"); precompute_out ret; ret.sizes = this->sizes; ret.strides = value; return ret; } at::DimVector sizes; at::DimVector strides; }; using meta_return_ty = precompute_out ; meta_return_ty meta(const at::Tensor & self, at::IOptTensorListRef indices); }; } // namespace native } // namespace at