Searched refs:activation_mode_ (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/contrib/fused_conv/kernels/ |
| D | fused_conv_ops_gpu.h | 39 activation_mode_(activation_mode), in FusedConvParameters()
55 ", ", activation_mode_, ", "); in ToString()
64 activation_mode_); in get_data_as_tuple()
67 ActivationMode activation_mode_; variable
|
| D | fused_conv2d_bias_activation_op.cc | 144 &activation_mode_)); in FusedConv2DBiasActivationOp()
146 activation_mode_ == ActivationMode::RELU || in FusedConv2DBiasActivationOp()
147 activation_mode_ == ActivationMode::NONE, in FusedConv2DBiasActivationOp()
242 side_input, side_input_scale, bias, activation_mode_, in Compute()
250 ActivationMode activation_mode_; member in tensorflow::FusedConv2DBiasActivationOp
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | conv_ops_fused_impl.h | 450 : ConvParameters(base), activation_mode_(activation_mode) {}
453 return absl::StrCat(ConvParameters::ToString(), ", ", activation_mode_);
465 activation_mode_);
468 se::dnn::ActivationMode activation_mode_;
|