DLPrimitives
|
This is the complete list of members for dlprim::TransposedConvolution2D, including all inherited members.
activation_ (defined in dlprim::TransposedConvolution2D) | dlprim::TransposedConvolution2D | protected |
alias_generator() | dlprim::Operator | inlinevirtual |
backward(std::vector< TensorAndGradient > &input, std::vector< TensorAndGradient > &output, std::vector< TensorAndGradient > ¶meters, Tensor &workspace, ExecutionContext const &ctx) | dlprim::TransposedConvolution2D | virtual |
backward_data_cpu(Tensor &dy, Tensor &K, Tensor &dx, Tensor &ws, float factor) (defined in dlprim::TransposedConvolution2D) | dlprim::TransposedConvolution2D | protected |
backward_filter_cpu(Tensor &dy, Tensor &x, Tensor &dK, Tensor &ws, float factor) (defined in dlprim::TransposedConvolution2D) | dlprim::TransposedConvolution2D | protected |
bs_ (defined in dlprim::TransposedConvolution2D) | dlprim::TransposedConvolution2D | protected |
bwd_bias_ (defined in dlprim::TransposedConvolution2D) | dlprim::TransposedConvolution2D | protected |
calc_workspace(Shape const &in) (defined in dlprim::TransposedConvolution2D) | dlprim::TransposedConvolution2D | protected |
config_ (defined in dlprim::TransposedConvolution2D) | dlprim::TransposedConvolution2D | protected |
conv_bwd_data_ (defined in dlprim::TransposedConvolution2D) | dlprim::TransposedConvolution2D | protected |
conv_bwd_filter_ (defined in dlprim::TransposedConvolution2D) | dlprim::TransposedConvolution2D | protected |
conv_config_ (defined in dlprim::TransposedConvolution2D) | dlprim::TransposedConvolution2D | protected |
conv_fwd_ (defined in dlprim::TransposedConvolution2D) | dlprim::TransposedConvolution2D | protected |
ctx_ | dlprim::Operator | protected |
dtype_ (defined in dlprim::TransposedConvolution2D) | dlprim::TransposedConvolution2D | protected |
forward(std::vector< Tensor > &input, std::vector< Tensor > &output, std::vector< Tensor > ¶meters, Tensor &workspace, ExecutionContext const &ctx) | dlprim::TransposedConvolution2D | virtual |
forward_cpu(Tensor &in, Tensor &out, Tensor &M, Tensor *bias, void *ws) (defined in dlprim::TransposedConvolution2D) | dlprim::TransposedConvolution2D | protected |
forward_gpu(Tensor &in, Tensor &out, Tensor &M, Tensor *bias, ExecutionContext const &ctx) (defined in dlprim::TransposedConvolution2D) | dlprim::TransposedConvolution2D | protected |
fwd_bwd_cpu(GemmOpMode mode, Tensor &in, Tensor &out, Tensor &W, Tensor *bias_tensor, void *ws, Convolution2DConfig const &config, float fwd_beta=0.0f) (defined in dlprim::Convolution2DBase) | dlprim::Convolution2DBase | protectedstatic |
get_im2col_width() (defined in dlprim::TransposedConvolution2D) | dlprim::TransposedConvolution2D | protected |
get_output_shape(Shape const &in) (defined in dlprim::TransposedConvolution2D) | dlprim::TransposedConvolution2D | |
im2col(Shape const &in, Shape const &outs, DType *img_in, DType *mat_in, Convolution2DConfig const &config) (defined in dlprim::Convolution2DBase) | dlprim::Convolution2DBase | protectedstatic |
in_h_ (defined in dlprim::TransposedConvolution2D) | dlprim::TransposedConvolution2D | protected |
in_w_ (defined in dlprim::TransposedConvolution2D) | dlprim::TransposedConvolution2D | protected |
initialize_params(std::vector< Tensor > ¶meters, ExecutionContext const &e) | dlprim::TransposedConvolution2D | virtual |
mode(CalculationsMode mode) | dlprim::Operator | inlinevirtual |
mode() | dlprim::Operator | inlinevirtual |
mode_ | dlprim::Operator | protected |
Operator(Context const &ctx) | dlprim::Operator | inline |
Operator(Operator const &)=delete | dlprim::Operator | |
Operator(Operator &&)=delete (defined in dlprim::Operator) | dlprim::Operator | |
operator=(Operator const &)=delete (defined in dlprim::Operator) | dlprim::Operator | |
operator=(Operator &&)=delete (defined in dlprim::Operator) | dlprim::Operator | |
operator_type() const | dlprim::TransposedConvolution2D | inlinevirtual |
out_h_ (defined in dlprim::TransposedConvolution2D) | dlprim::TransposedConvolution2D | protected |
out_w_ (defined in dlprim::TransposedConvolution2D) | dlprim::TransposedConvolution2D | protected |
reshape(std::vector< Shape > const &in, std::vector< Shape > &out, size_t &ws) | dlprim::TransposedConvolution2D | virtual |
scale_cpu(Tensor &t, float v) (defined in dlprim::Convolution2DBase) | dlprim::Convolution2DBase | protectedstatic |
setup(std::vector< TensorSpecs > const &in, std::vector< TensorSpecs > &out, std::vector< TensorSpecs > ¶meters, size_t &workspace) | dlprim::TransposedConvolution2D | virtual |
setup_algo(Shape const &in) (defined in dlprim::TransposedConvolution2D) | dlprim::TransposedConvolution2D | protected |
shared_resource() | dlprim::Operator | inline |
shared_resource(std::shared_ptr< SharedResource > r) | dlprim::Operator | inline |
shared_resource_ (defined in dlprim::Operator) | dlprim::Operator | protected |
TransposedConvolution2D(Context &ctx, TransposedConvolution2DConfig const &cfg) (defined in dlprim::TransposedConvolution2D) | dlprim::TransposedConvolution2D | |
ws_size_ (defined in dlprim::TransposedConvolution2D) | dlprim::TransposedConvolution2D | protected |
~Operator() (defined in dlprim::Operator) | dlprim::Operator | inlinevirtual |
~TransposedConvolution2D() (defined in dlprim::TransposedConvolution2D) | dlprim::TransposedConvolution2D | virtual |