Searched refs:FusedBNActivationMode (Results 1 – 1 of 1) sorted by relevance
41 using FusedBNActivationMode = functor::FusedBatchNormActivationMode; typedef48 FusedBNActivationMode activation_mode;53 FusedBNActivationMode activation_mode) in MklBatchNormFwdParams()193 if (fwdParams.activation_mode == FusedBNActivationMode::kRelu) { in Setup()359 key_creator.AddAsKey<FusedBNActivationMode>(fwdParams.activation_mode); in CreateKey()660 activation_mode_ = FusedBNActivationMode::kIdentity; in MklFusedBatchNormOp()671 OP_REQUIRES(context, activation_mode_ == FusedBNActivationMode::kRelu, in MklFusedBatchNormOp()785 if (fwdParams.activation_mode == FusedBNActivationMode::kRelu) { in Compute()925 FusedBNActivationMode activation_mode_;