DLPrimitives
|
This is the complete list of members for dlprim::Threshold, including all inherited members.
alias_generator() | dlprim::Operator | inlinevirtual |
backward(std::vector< TensorAndGradient > &input, std::vector< TensorAndGradient > &output, std::vector< TensorAndGradient > ¶meters, Tensor &workspace, ExecutionContext const &ctx) | dlprim::PointwiseBase | virtual |
backward_cpu(Tensor &x, Tensor &dx, Tensor &y, Tensor &dy, float beta) (defined in dlprim::PointwiseBase) | dlprim::PointwiseBase | protectedvirtual |
backward_cpu_float(size_t n, float const *x, float *dx, float const *y, float const *dy, float beta) (defined in dlprim::Threshold) | dlprim::Threshold | virtual |
backward_gpu(Tensor &x, Tensor &dx, Tensor &y, Tensor &dy, float beta, ExecutionContext const &q) (defined in dlprim::Threshold) | dlprim::Threshold | virtual |
ctx_ | dlprim::Operator | protected |
forward(std::vector< Tensor > &input, std::vector< Tensor > &output, std::vector< Tensor > ¶meters, Tensor &ws, ExecutionContext const &q) | dlprim::PointwiseBase | virtual |
forward_cpu(Tensor &x, Tensor &y) (defined in dlprim::PointwiseBase) | dlprim::PointwiseBase | protectedvirtual |
forward_cpu_float(size_t n, float const *x, float *y) (defined in dlprim::Threshold) | dlprim::Threshold | virtual |
forward_gpu(Tensor &x, Tensor &y, ExecutionContext const &q) (defined in dlprim::Threshold) | dlprim::Threshold | virtual |
initialize_params(std::vector< Tensor > &, ExecutionContext const &) | dlprim::Operator | inlinevirtual |
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::Threshold | inlinevirtual |
PointwiseBase(Context &ctx) (defined in dlprim::PointwiseBase) | dlprim::PointwiseBase | |
reshape(std::vector< Shape > const &in, std::vector< Shape > &out, size_t &ws) | dlprim::PointwiseBase | virtual |
setup(std::vector< TensorSpecs > const &in, std::vector< TensorSpecs > &out, std::vector< TensorSpecs > ¶meters, size_t &workspace) | dlprim::PointwiseBase | virtual |
shared_resource() | dlprim::Operator | inline |
shared_resource(std::shared_ptr< SharedResource > r) | dlprim::Operator | inline |
shared_resource_ (defined in dlprim::Operator) | dlprim::Operator | protected |
Threshold(Context &ctx, ThresholdConfig const &cfg=ThresholdConfig()) (defined in dlprim::Threshold) | dlprim::Threshold | inline |
~Operator() (defined in dlprim::Operator) | dlprim::Operator | inlinevirtual |
~PointwiseBase() (defined in dlprim::PointwiseBase) | dlprim::PointwiseBase | virtual |