#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_fractional_max_pool3d : public at::impl::MetaBase { template struct TORCH_API precompute_out { precompute_out set_poolSizeT(int64_t value) { static_assert(POOLSIZET == false, "poolSizeT already set"); precompute_out ret; ret.poolSizeT = value; ret.poolSizeH = this->poolSizeH; ret.poolSizeW = this->poolSizeW; ret.outputT = this->outputT; ret.outputH = this->outputH; ret.outputW = this->outputW; ret.numBatch = this->numBatch; ret.numPlanes = this->numPlanes; ret.inputT = this->inputT; ret.inputH = this->inputH; ret.inputW = this->inputW; return ret; } precompute_out set_poolSizeH(int64_t value) { static_assert(POOLSIZEH == false, "poolSizeH already set"); precompute_out ret; ret.poolSizeT = this->poolSizeT; ret.poolSizeH = value; ret.poolSizeW = this->poolSizeW; ret.outputT = this->outputT; ret.outputH = this->outputH; ret.outputW = this->outputW; ret.numBatch = this->numBatch; ret.numPlanes = this->numPlanes; ret.inputT = this->inputT; ret.inputH = this->inputH; ret.inputW = this->inputW; return ret; } precompute_out set_poolSizeW(int64_t value) { static_assert(POOLSIZEW == false, "poolSizeW already set"); precompute_out ret; ret.poolSizeT = this->poolSizeT; ret.poolSizeH = this->poolSizeH; ret.poolSizeW = value; ret.outputT = this->outputT; ret.outputH = this->outputH; ret.outputW = this->outputW; ret.numBatch = this->numBatch; ret.numPlanes = this->numPlanes; ret.inputT = this->inputT; ret.inputH = this->inputH; ret.inputW = this->inputW; return ret; } precompute_out set_outputT(int64_t value) { static_assert(OUTPUTT == false, "outputT already set"); precompute_out ret; ret.poolSizeT = this->poolSizeT; ret.poolSizeH = this->poolSizeH; ret.poolSizeW = this->poolSizeW; ret.outputT = value; ret.outputH = this->outputH; ret.outputW = this->outputW; ret.numBatch = this->numBatch; ret.numPlanes = this->numPlanes; ret.inputT = this->inputT; ret.inputH = this->inputH; ret.inputW = this->inputW; return ret; } precompute_out set_outputH(int64_t value) { static_assert(OUTPUTH == false, "outputH already set"); precompute_out ret; ret.poolSizeT = this->poolSizeT; ret.poolSizeH = this->poolSizeH; ret.poolSizeW = this->poolSizeW; ret.outputT = this->outputT; ret.outputH = value; ret.outputW = this->outputW; ret.numBatch = this->numBatch; ret.numPlanes = this->numPlanes; ret.inputT = this->inputT; ret.inputH = this->inputH; ret.inputW = this->inputW; return ret; } precompute_out set_outputW(int64_t value) { static_assert(OUTPUTW == false, "outputW already set"); precompute_out ret; ret.poolSizeT = this->poolSizeT; ret.poolSizeH = this->poolSizeH; ret.poolSizeW = this->poolSizeW; ret.outputT = this->outputT; ret.outputH = this->outputH; ret.outputW = value; ret.numBatch = this->numBatch; ret.numPlanes = this->numPlanes; ret.inputT = this->inputT; ret.inputH = this->inputH; ret.inputW = this->inputW; return ret; } precompute_out set_numBatch(int64_t value) { static_assert(NUMBATCH == false, "numBatch already set"); precompute_out ret; ret.poolSizeT = this->poolSizeT; ret.poolSizeH = this->poolSizeH; ret.poolSizeW = this->poolSizeW; ret.outputT = this->outputT; ret.outputH = this->outputH; ret.outputW = this->outputW; ret.numBatch = value; ret.numPlanes = this->numPlanes; ret.inputT = this->inputT; ret.inputH = this->inputH; ret.inputW = this->inputW; return ret; } precompute_out set_numPlanes(int64_t value) { static_assert(NUMPLANES == false, "numPlanes already set"); precompute_out ret; ret.poolSizeT = this->poolSizeT; ret.poolSizeH = this->poolSizeH; ret.poolSizeW = this->poolSizeW; ret.outputT = this->outputT; ret.outputH = this->outputH; ret.outputW = this->outputW; ret.numBatch = this->numBatch; ret.numPlanes = value; ret.inputT = this->inputT; ret.inputH = this->inputH; ret.inputW = this->inputW; return ret; } precompute_out set_inputT(int64_t value) { static_assert(INPUTT == false, "inputT already set"); precompute_out ret; ret.poolSizeT = this->poolSizeT; ret.poolSizeH = this->poolSizeH; ret.poolSizeW = this->poolSizeW; ret.outputT = this->outputT; ret.outputH = this->outputH; ret.outputW = this->outputW; ret.numBatch = this->numBatch; ret.numPlanes = this->numPlanes; ret.inputT = value; ret.inputH = this->inputH; ret.inputW = this->inputW; return ret; } precompute_out set_inputH(int64_t value) { static_assert(INPUTH == false, "inputH already set"); precompute_out ret; ret.poolSizeT = this->poolSizeT; ret.poolSizeH = this->poolSizeH; ret.poolSizeW = this->poolSizeW; ret.outputT = this->outputT; ret.outputH = this->outputH; ret.outputW = this->outputW; ret.numBatch = this->numBatch; ret.numPlanes = this->numPlanes; ret.inputT = this->inputT; ret.inputH = value; ret.inputW = this->inputW; return ret; } precompute_out set_inputW(int64_t value) { static_assert(INPUTW == false, "inputW already set"); precompute_out ret; ret.poolSizeT = this->poolSizeT; ret.poolSizeH = this->poolSizeH; ret.poolSizeW = this->poolSizeW; ret.outputT = this->outputT; ret.outputH = this->outputH; ret.outputW = this->outputW; ret.numBatch = this->numBatch; ret.numPlanes = this->numPlanes; ret.inputT = this->inputT; ret.inputH = this->inputH; ret.inputW = value; return ret; } int64_t poolSizeT; int64_t poolSizeH; int64_t poolSizeW; int64_t outputT; int64_t outputH; int64_t outputW; int64_t numBatch; int64_t numPlanes; int64_t inputT; int64_t inputH; int64_t inputW; }; using meta_return_ty = precompute_out ; meta_return_ty meta(const at::Tensor & self, at::IntArrayRef kernel_size, at::IntArrayRef output_size, const at::Tensor & random_samples); }; } // namespace native } // namespace at