|
DLPrimitives
|
This is the complete list of members for dlprim::SoftmaxWithLoss, including all inherited members.
| alias_generator() | dlprim::Operator | inlinevirtual |
| backward(std::vector< TensorAndGradient > &input, std::vector< TensorAndGradient > &output, std::vector< TensorAndGradient > &, Tensor &, ExecutionContext const &e) | dlprim::SoftmaxWithLoss | virtual |
| ctx_ | dlprim::Operator | protected |
| forward(std::vector< Tensor > &input, std::vector< Tensor > &output, std::vector< Tensor > ¶meters, Tensor &workspace, ExecutionContext const &ctx) | dlprim::SoftmaxWithLoss | virtual |
| initialize_params(std::vector< Tensor > &, ExecutionContext const &) | dlprim::Operator | inlinevirtual |
| items_per_wi_ (defined in dlprim::SoftmaxBase) | dlprim::SoftmaxBase | protected |
| mode(CalculationsMode mode) | dlprim::Operator | inlinevirtual |
| mode() | dlprim::Operator | inlinevirtual |
| mode_ | dlprim::Operator | protected |
| nd_range_ (defined in dlprim::SoftmaxBase) | dlprim::SoftmaxBase | 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::SoftmaxWithLoss | inlinevirtual |
| reshape(std::vector< Shape > const &in, std::vector< Shape > &out, size_t &ws) | dlprim::SoftmaxWithLoss | virtual |
| setup(std::vector< TensorSpecs > const &in, std::vector< TensorSpecs > &out, std::vector< TensorSpecs > ¶meters, size_t &workspace) | dlprim::SoftmaxWithLoss | virtual |
| setup_kernel_params(int sm_range) (defined in dlprim::SoftmaxBase) | dlprim::SoftmaxBase | 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 |
| sm_range_ (defined in dlprim::SoftmaxBase) | dlprim::SoftmaxBase | protected |
| SoftmaxWithLoss(Context &ctx, SoftmaxConfig const &cfg=SoftmaxConfig()) (defined in dlprim::SoftmaxWithLoss) | dlprim::SoftmaxWithLoss | |
| wg_size_ (defined in dlprim::SoftmaxBase) | dlprim::SoftmaxBase | protected |
| ~Operator() (defined in dlprim::Operator) | dlprim::Operator | inlinevirtual |
| ~SoftmaxWithLoss() (defined in dlprim::SoftmaxWithLoss) | dlprim::SoftmaxWithLoss | virtual |
1.8.11