Searched refs:FbnActivationMode (Results 1 – 1 of 1) sorted by relevance
1181 using FbnActivationMode = functor::FusedBatchNormActivationMode; typedef in tensorflow::FusedBatchNormOpBase1202 activation_mode_ = FbnActivationMode::kIdentity; in FusedBatchNormOpBase()1215 context, activation_mode_ != FbnActivationMode::kIdentity, in FusedBatchNormOpBase()1221 if (activation_mode_ != FbnActivationMode::kIdentity && is_training_) { in FusedBatchNormOpBase()1290 if (activation_mode_ != FbnActivationMode::kIdentity) { in ComputeWithReservedSpace()1342 FbnActivationMode activation_mode_;