Searched refs:is_batch_norm_ex (Results 1 – 3 of 3) sorted by relevance
36 FusedBatchNormOp(OpKernelConstruction* ctx, bool is_batch_norm_ex) in FusedBatchNormOp() argument48 if (is_batch_norm_ex) { in FusedBatchNormOp()
638 bool is_batch_norm_ex = false, bool native_format = false>659 if (!is_batch_norm_ex) { in MklFusedBatchNormOp()
1185 bool is_batch_norm_ex = false) in FusedBatchNormOpBase() argument1200 if (!is_batch_norm_ex) { in FusedBatchNormOpBase()