Home
last modified time | relevance | path

Searched refs:ActivationMode (Results 1 – 18 of 18) sorted by relevance

/external/tensorflow/tensorflow/contrib/fused_conv/kernels/
Dfused_conv_ops_gpu.h36 ActivationMode activation_mode) in FusedConvParameters()
60 std::tuple<ConvParameters::ParameterDataType, bool, ActivationMode>;
67 ActivationMode activation_mode_;
Dfused_conv2d_bias_activation_op.cc146 activation_mode_ == ActivationMode::RELU || in FusedConv2DBiasActivationOp()
147 activation_mode_ == ActivationMode::NONE, in FusedConv2DBiasActivationOp()
250 ActivationMode activation_mode_;
446 ActivationMode activation_mode, TensorFormat data_format, in launch()
680 dnn::ActivationMode dnn_activation_mode; in launch()
682 case ActivationMode::NONE: in launch()
683 dnn_activation_mode = dnn::ActivationMode::kNone; in launch()
685 case ActivationMode::RELU: in launch()
686 dnn_activation_mode = dnn::ActivationMode::kRelu; in launch()
700 if (activation_mode == ActivationMode::NONE) { in launch()
Dfused_conv2d_bias_activation_op.h44 ActivationMode activation_mode, TensorFormat data_format,
58 ActivationMode activation_mode, TensorFormat data_format,
/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_dnn.h272 const DeviceMemory<double>& biases, dnn::ActivationMode activation_mode,
286 const DeviceMemory<float>& biases, dnn::ActivationMode activation_mode,
303 dnn::ActivationMode activation_mode,
318 const DeviceMemory<float>& biases, dnn::ActivationMode activation_mode,
412 bool DoActivate(Stream* stream, dnn::ActivationMode activation_mode,
540 const DeviceMemory<float>& bias_data, dnn::ActivationMode activation_mode,
553 dnn::ActivationMode activation_mode, DeviceMemory<float>* y_data,
564 dnn::ActivationMode activation_mode, DeviceMemory<Eigen::half>* y_data,
573 dnn::ActivationMode activation_mode, DeviceMemory<float>* y_data,
584 dnn::ActivationMode activation_mode, DeviceMemory<Eigen::half>* y_data,
[all …]
Drocm_dnn.cc821 ScopedActivationDescriptor(dnn::ActivationMode activation_mode) in ScopedActivationDescriptor()
833 case dnn::ActivationMode::kNone: in ScopedActivationDescriptor()
837 case dnn::ActivationMode::kSigmoid: in ScopedActivationDescriptor()
841 case dnn::ActivationMode::kRelu: in ScopedActivationDescriptor()
845 case dnn::ActivationMode::kRelu6: in ScopedActivationDescriptor()
850 case dnn::ActivationMode::kTanh: in ScopedActivationDescriptor()
3153 const DeviceMemory<double>& biases, dnn::ActivationMode activation_mode, in DoFusedConvolve()
3170 const DeviceMemory<float>& biases, dnn::ActivationMode activation_mode, in DoFusedConvolve()
3188 dnn::ActivationMode activation_mode, in DoFusedConvolve()
3205 const DeviceMemory<float>& biases, dnn::ActivationMode activation_mode, in DoFusedConvolve()
[all …]
/external/tensorflow/tensorflow/core/util/
Dactivation_mode.h30 enum ActivationMode { enum
42 ActivationMode* value);
Dactivation_mode.cc24 ActivationMode* value) { in GetActivationModeFromString()
/external/tensorflow/tensorflow/stream_executor/
Ddnn.cc66 string ActivationModeString(ActivationMode mode) { in ActivationModeString()
68 case ActivationMode::kSigmoid: in ActivationModeString()
70 case ActivationMode::kRelu: in ActivationModeString()
72 case ActivationMode::kRelu6: in ActivationModeString()
74 case ActivationMode::kReluX: in ActivationModeString()
76 case ActivationMode::kTanh: in ActivationModeString()
78 case ActivationMode::kBandPass: in ActivationModeString()
Ddnn.h905 string ActivationModeString(ActivationMode mode);
1123 const DeviceMemory<double>& biases, dnn::ActivationMode activation_mode,
1140 const DeviceMemory<float>& biases, dnn::ActivationMode activation_mode,
1159 dnn::ActivationMode activation_mode,
1178 const DeviceMemory<float>& biases, dnn::ActivationMode activation_mode,
1698 virtual bool DoActivate(Stream* stream, ActivationMode activation_mode,
2359 const DeviceMemory<float>& bias_data, dnn::ActivationMode activation_mode,
2400 dnn::ActivationMode activation_mode, DeviceMemory<float>* y_data,
2413 dnn::ActivationMode activation_mode, DeviceMemory<Eigen::half>* y_data,
2452 dnn::ActivationMode activation_mode, DeviceMemory<float>* y_data,
[all …]
Ddnn.proto46 enum ActivationMode { enum
Dstream.h345 const DeviceMemory<double> &biases, dnn::ActivationMode activation_mode,
359 const DeviceMemory<float> &biases, dnn::ActivationMode activation_mode,
374 dnn::ActivationMode activation_mode,
388 const DeviceMemory<float> &biases, dnn::ActivationMode activation_mode,
586 Stream &ThenActivate(dnn::ActivationMode activation_mode,
593 Stream &ThenActivateWithOptions(dnn::ActivationMode activation_mode,
Dstream.cc58 string ToVlogString(dnn::ActivationMode mode) { in ToVlogString()
448 const DeviceMemory<double> &biases, dnn::ActivationMode activation_mode, in ThenFusedConvolveWithAlgorithm()
486 const DeviceMemory<float> &biases, dnn::ActivationMode activation_mode, in ThenFusedConvolveWithAlgorithm()
525 dnn::ActivationMode activation_mode, in ThenFusedConvolveWithAlgorithm()
563 const DeviceMemory<float> &biases, dnn::ActivationMode activation_mode, in ThenFusedConvolveWithAlgorithm()
1438 Stream &Stream::ThenActivate(dnn::ActivationMode activation_mode, in ThenActivate()
1446 Stream &Stream::ThenActivateWithOptions(dnn::ActivationMode activation_mode, in ThenActivateWithOptions()
/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_dnn.h281 const DeviceMemory<double>& biases, dnn::ActivationMode activation_mode,
295 const DeviceMemory<float>& biases, dnn::ActivationMode activation_mode,
312 dnn::ActivationMode activation_mode,
327 const DeviceMemory<float>& biases, dnn::ActivationMode activation_mode,
421 bool DoActivate(Stream* stream, dnn::ActivationMode activation_mode,
591 const DeviceMemory<BiasType>& biases, dnn::ActivationMode activation_mode,
Dcuda_dnn.cc798 CudnnActivationDescriptor(dnn::ActivationMode activation_mode, in CudnnActivationDescriptor()
806 case dnn::ActivationMode::kNone: in CudnnActivationDescriptor()
810 case dnn::ActivationMode::kRelu6: in CudnnActivationDescriptor()
814 case dnn::ActivationMode::kReluX: in CudnnActivationDescriptor()
818 case dnn::ActivationMode::kRelu: in CudnnActivationDescriptor()
821 case dnn::ActivationMode::kSigmoid: in CudnnActivationDescriptor()
824 case dnn::ActivationMode::kTanh: in CudnnActivationDescriptor()
2930 const DeviceMemory<BiasType>& biases, dnn::ActivationMode activation_mode, in DoFusedConvolveImpl()
2936 if (activation_mode != dnn::ActivationMode::kRelu && in DoFusedConvolveImpl()
2937 activation_mode != dnn::ActivationMode::kNone) { in DoFusedConvolveImpl()
[all …]
/external/tensorflow/tensorflow/core/kernels/
Dconv_ops_fused_impl.h449 const se::dnn::ActivationMode activation_mode)
461 std::tuple<ConvParameters::ParameterDataType, se::dnn::ActivationMode>;
468 se::dnn::ActivationMode activation_mode_;
669 se::dnn::ActivationMode dnn_activation_mode;
672 dnn_activation_mode = se::dnn::ActivationMode::kRelu;
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dbackend_configs.proto35 // stream_executor::dnn::ActivationMode.
Dcudnn_conv_runner.cc56 se::dnn::ActivationMode mode;
377 fusion.mode = static_cast<se::dnn::ActivationMode>( in GetCudnnConvParams()
Dcudnn_fused_conv_rewriter.cc233 static_cast<int64>(se::dnn::ActivationMode::kRelu)); in TryRewriteToCudnnForwardRelu()