|
DLPrimitives
|
This is the complete list of members for dlprim::Convolution2D, including all inherited members.
| activation_ (defined in dlprim::Convolution2D) | dlprim::Convolution2D | 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::Convolution2D | virtual |
| backward_data_cpu(Tensor &dy, Tensor &K, Tensor &dx, Tensor &ws, float factor) (defined in dlprim::Convolution2D) | dlprim::Convolution2D | protected |
| backward_filter_cpu(Tensor &dy, Tensor &x, Tensor &dK, Tensor &ws, float factor) (defined in dlprim::Convolution2D) | dlprim::Convolution2D | protected |
| bs_ (defined in dlprim::Convolution2D) | dlprim::Convolution2D | protected |
| bwd_bias_ (defined in dlprim::Convolution2D) | dlprim::Convolution2D | protected |
| calc_workspace(Shape const &in) (defined in dlprim::Convolution2D) | dlprim::Convolution2D | protected |
| config_ (defined in dlprim::Convolution2D) | dlprim::Convolution2D | protected |
| conv_ (defined in dlprim::Convolution2D) | dlprim::Convolution2D | protected |
| conv_bwd_data_ (defined in dlprim::Convolution2D) | dlprim::Convolution2D | protected |
| conv_bwd_filter_ (defined in dlprim::Convolution2D) | dlprim::Convolution2D | protected |
| Convolution2D(Context &ctx, Convolution2DConfig const &cfg) (defined in dlprim::Convolution2D) | dlprim::Convolution2D | |
| ctx_ | dlprim::Operator | protected |
| dtype_ (defined in dlprim::Convolution2D) | dlprim::Convolution2D | protected |
| forward(std::vector< Tensor > &input, std::vector< Tensor > &output, std::vector< Tensor > ¶meters, Tensor &workspace, ExecutionContext const &ctx) | dlprim::Convolution2D | virtual |
| forward_cpu(Tensor &in, Tensor &out, Tensor &M, Tensor *bias, void *ws) (defined in dlprim::Convolution2D) | dlprim::Convolution2D | protected |
| forward_gpu(Tensor &in, Tensor &out, Tensor &M, Tensor *bias, ExecutionContext const &ctx) (defined in dlprim::Convolution2D) | dlprim::Convolution2D | 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::Convolution2D) | dlprim::Convolution2D | protected |
| get_output_shape(Shape const &in) (defined in dlprim::Convolution2D) | dlprim::Convolution2D | |
| 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::Convolution2D) | dlprim::Convolution2D | protected |
| in_w_ (defined in dlprim::Convolution2D) | dlprim::Convolution2D | protected |
| initialize_params(std::vector< Tensor > ¶meters, ExecutionContext const &e) | dlprim::Convolution2D | 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::Convolution2D | inlinevirtual |
| out_h_ (defined in dlprim::Convolution2D) | dlprim::Convolution2D | protected |
| out_w_ (defined in dlprim::Convolution2D) | dlprim::Convolution2D | protected |
| reshape(std::vector< Shape > const &in, std::vector< Shape > &out, size_t &ws) | dlprim::Convolution2D | 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::Convolution2D | virtual |
| setup_algo(Shape const &in) (defined in dlprim::Convolution2D) | dlprim::Convolution2D | 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 |
| ws_size_ (defined in dlprim::Convolution2D) | dlprim::Convolution2D | protected |
| ~Convolution2D() (defined in dlprim::Convolution2D) | dlprim::Convolution2D | virtual |
| ~Operator() (defined in dlprim::Operator) | dlprim::Operator | inlinevirtual |
1.8.11