Skip to content

Add the BatchNorm train/test flag support

Benjamin Halimi requested to merge bn_flag into dev

Description

This MR aims to bring the support of the train/test mode flag, for the BatchNorm operator.

Two MRs will follow this one : one for the CPU backend and one for the CUDA backend.

Merge request reports

Copyright © Eclipse Foundation, Inc. All Rights Reserved.     Privacy Policy | Terms of Use | Copyright Agent