#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_cat : public at::impl::MetaBase { template struct TORCH_API precompute_out { precompute_out set_dim(int64_t value) { static_assert(DIM == false, "dim already set"); precompute_out ret; ret.dim = value; ret.valid = this->valid; ret.all_contiguous = this->all_contiguous; ret.all_same_dtype = this->all_same_dtype; ret.all_same_sizes_and_stride = this->all_same_sizes_and_stride; ret.memory_format = this->memory_format; return ret; } precompute_out set_valid(int64_t value) { static_assert(VALID == false, "valid already set"); precompute_out ret; ret.dim = this->dim; ret.valid = value; ret.all_contiguous = this->all_contiguous; ret.all_same_dtype = this->all_same_dtype; ret.all_same_sizes_and_stride = this->all_same_sizes_and_stride; ret.memory_format = this->memory_format; return ret; } precompute_out set_all_contiguous(bool value) { static_assert(ALL_CONTIGUOUS == false, "all_contiguous already set"); precompute_out ret; ret.dim = this->dim; ret.valid = this->valid; ret.all_contiguous = value; ret.all_same_dtype = this->all_same_dtype; ret.all_same_sizes_and_stride = this->all_same_sizes_and_stride; ret.memory_format = this->memory_format; return ret; } precompute_out set_all_same_dtype(bool value) { static_assert(ALL_SAME_DTYPE == false, "all_same_dtype already set"); precompute_out ret; ret.dim = this->dim; ret.valid = this->valid; ret.all_contiguous = this->all_contiguous; ret.all_same_dtype = value; ret.all_same_sizes_and_stride = this->all_same_sizes_and_stride; ret.memory_format = this->memory_format; return ret; } precompute_out set_all_same_sizes_and_stride(bool value) { static_assert(ALL_SAME_SIZES_AND_STRIDE == false, "all_same_sizes_and_stride already set"); precompute_out ret; ret.dim = this->dim; ret.valid = this->valid; ret.all_contiguous = this->all_contiguous; ret.all_same_dtype = this->all_same_dtype; ret.all_same_sizes_and_stride = value; ret.memory_format = this->memory_format; return ret; } precompute_out set_memory_format(at::MemoryFormat value) { static_assert(MEMORY_FORMAT == false, "memory_format already set"); precompute_out ret; ret.dim = this->dim; ret.valid = this->valid; ret.all_contiguous = this->all_contiguous; ret.all_same_dtype = this->all_same_dtype; ret.all_same_sizes_and_stride = this->all_same_sizes_and_stride; ret.memory_format = value; return ret; } int64_t dim; int64_t valid; bool all_contiguous; bool all_same_dtype; bool all_same_sizes_and_stride; at::MemoryFormat memory_format; }; using meta_return_ty = precompute_out ; meta_return_ty meta(const at::ITensorListRef & tensors, int64_t dim); }; } // namespace native } // namespace at