Searched refs:bn_flags (Results 1 – 2 of 2) sorted by relevance
434 __u8 bn_flags; member
538 auto bn_flags = bwdParams.training in Setup() local543 prop_kind::forward_training, src_md, bwdParams.eps, bn_flags); in Setup()552 prop_kind::backward, diff_dst_md, src_md, bwdParams.eps, bn_flags); in Setup()