DLPrimitives
dlprim::core::BatchNormFwdBwd Member List

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

create(Context &ctx, Shape const &s, DataType dt=float_data) (defined in dlprim::core::BatchNormFwdBwd)dlprim::core::BatchNormFwdBwdstatic
enqueue_backward_affine(bool training_mode, Tensor &x, Tensor &dy, Tensor &mean, Tensor &var, Tensor &gamma, Tensor *dx, float fx_factor, Tensor *dgamma, float dgamma_factor, Tensor *dbeta, float dbeta_factor, float eps, Tensor &ws, ExecutionContext const &e)=0dlprim::core::BatchNormFwdBwdpure virtual
enqueue_backward_direct(bool training_mode, Tensor &x, Tensor &dy, Tensor &mean, Tensor &var, Tensor &dx, float dx_factor, float eps, Tensor &ws, ExecutionContext const &e)=0dlprim::core::BatchNormFwdBwdpure virtual
enqueue_calculate_batch_stats(Tensor &x, Tensor &mean, Tensor &var, Tensor &ws, ExecutionContext const &e)=0dlprim::core::BatchNormFwdBwdpure virtual
enqueue_forward_affine(Tensor &x, Tensor &y, Tensor &gamma, Tensor &beta, Tensor &mean, Tensor &var, float eps, Tensor &ws, ExecutionContext const &e)=0dlprim::core::BatchNormFwdBwdpure virtual
enqueue_forward_direct(Tensor &x, Tensor &y, Tensor &mean, Tensor &var, float eps, Tensor &ws, ExecutionContext const &e)=0dlprim::core::BatchNormFwdBwdpure virtual
enqueue_update_running_stats(float batch_mean_factor, float running_mean_factor, Tensor &batch_mean, Tensor &running_mean, float batch_var_factor, float running_var_factor, Tensor &batch_var, Tensor &running_var, Tensor &ws, ExecutionContext const &e)=0dlprim::core::BatchNormFwdBwdpure virtual
workspace()=0dlprim::core::BatchNormFwdBwdpure virtual
~BatchNormFwdBwd() (defined in dlprim::core::BatchNormFwdBwd)dlprim::core::BatchNormFwdBwdinlinevirtual