Home
last modified time | relevance | path

Searched refs:activation_mode_ (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dconv_ops_fused_impl.h311 : ConvParameters(base), activation_mode_(activation_mode) {}
314 return absl::StrCat(ConvParameters::ToString(), ", ", activation_mode_);
326 activation_mode_);
329 se::dnn::ActivationMode activation_mode_;
Dfused_batch_norm_op.cc1202 activation_mode_ = FbnActivationMode::kIdentity; in FusedBatchNormOpBase()
1204 OP_REQUIRES_OK(context, ParseActivationMode(context, &activation_mode_)); 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()
1320 side_input, epsilon_, exponential_avg_factor_, activation_mode_, y, in ComputeWithReservedSpace()
1326 side_input, epsilon_, exponential_avg_factor_, activation_mode_, y, in ComputeWithReservedSpace()
1342 FbnActivationMode activation_mode_; member in tensorflow::FusedBatchNormOpBase
/external/tensorflow/tensorflow/core/kernels/mkl/
Dmkl_fused_batch_norm_op.cc660 activation_mode_ = FusedBNActivationMode::kIdentity; in MklFusedBatchNormOp()
670 OP_REQUIRES_OK(context, ParseActivationMode(context, &activation_mode_)); in MklFusedBatchNormOp()
671 OP_REQUIRES(context, activation_mode_ == FusedBNActivationMode::kRelu, in MklFusedBatchNormOp()
777 src_md, activation_mode_); in Compute()
925 FusedBNActivationMode activation_mode_; member in tensorflow::MklFusedBatchNormOp