Mohsen Saffari 8d4f766f9c Add scale and bias to batch normalization

2025-12-04 16:22:32 +00:00

..

batchnorm_fwd_pipeline.hpp

Add scale and bias to batch normalization

2025-12-04 16:22:32 +00:00

batchnorm_fwd_policy.hpp

Change from NCHW to MHWC based on old-ck and manage verifying for c > 1

2025-12-02 15:47:46 +00:00

batchnorm_fwd_traits.hpp

Add trait file for update_moving_average and save_mean_inv_std

2025-11-28 10:33:58 +00:00

batchnorm_problem.hpp

add kSaveMeanInvStd, kUpdateMovingAverage in Traits

2025-11-27 15:06:13 +00:00

batchnorm_shape.hpp

add tiling, pipeline, follow layernorm2d

2025-11-28 18:29:10 +00:00