DLPrimitives
dlprim::Hardtanh Member List

This is the complete list of members for dlprim::Hardtanh, including all inherited members.

alias_generator()dlprim::Operatorinlinevirtual
backward(std::vector< TensorAndGradient > &input, std::vector< TensorAndGradient > &output, std::vector< TensorAndGradient > &parameters, Tensor &workspace, ExecutionContext const &ctx)dlprim::PointwiseBasevirtual
backward_cpu(Tensor &x, Tensor &dx, Tensor &y, Tensor &dy, float beta) (defined in dlprim::PointwiseBase)dlprim::PointwiseBaseprotectedvirtual
backward_cpu_float(size_t n, float const *, float *, float const *, float const *, float beta) (defined in dlprim::Hardtanh)dlprim::Hardtanhvirtual
backward_gpu(Tensor &x, Tensor &dx, Tensor &y, Tensor &dy, float beta, ExecutionContext const &q) (defined in dlprim::Hardtanh)dlprim::Hardtanhvirtual
ctx_dlprim::Operatorprotected
forward(std::vector< Tensor > &input, std::vector< Tensor > &output, std::vector< Tensor > &parameters, Tensor &ws, ExecutionContext const &q)dlprim::PointwiseBasevirtual
forward_cpu(Tensor &x, Tensor &y) (defined in dlprim::PointwiseBase)dlprim::PointwiseBaseprotectedvirtual
forward_cpu_float(size_t n, float const *x, float *y) (defined in dlprim::Hardtanh)dlprim::Hardtanhvirtual
forward_gpu(Tensor &x, Tensor &y, ExecutionContext const &q) (defined in dlprim::Hardtanh)dlprim::Hardtanhvirtual
Hardtanh(Context &ctx, HardtanhConfig const &cfg=HardtanhConfig()) (defined in dlprim::Hardtanh)dlprim::Hardtanhinline
initialize_params(std::vector< Tensor > &, ExecutionContext const &)dlprim::Operatorinlinevirtual
mode(CalculationsMode mode)dlprim::Operatorinlinevirtual
mode()dlprim::Operatorinlinevirtual
mode_dlprim::Operatorprotected
Operator(Context const &ctx)dlprim::Operatorinline
Operator(Operator const &)=deletedlprim::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::Hardtanhinlinevirtual
PointwiseBase(Context &ctx) (defined in dlprim::PointwiseBase)dlprim::PointwiseBase
reshape(std::vector< Shape > const &in, std::vector< Shape > &out, size_t &ws)dlprim::PointwiseBasevirtual
setup(std::vector< TensorSpecs > const &in, std::vector< TensorSpecs > &out, std::vector< TensorSpecs > &parameters, size_t &workspace)dlprim::PointwiseBasevirtual
shared_resource()dlprim::Operatorinline
shared_resource(std::shared_ptr< SharedResource > r)dlprim::Operatorinline
shared_resource_ (defined in dlprim::Operator)dlprim::Operatorprotected
~Operator() (defined in dlprim::Operator)dlprim::Operatorinlinevirtual
~PointwiseBase() (defined in dlprim::PointwiseBase)dlprim::PointwiseBasevirtual