Home
last modified time | relevance | path

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

/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/rocm/
Drocm_dnn.h264 dnn::ActivationMode activation_mode, DeviceMemory<float>* y,
278 dnn::ActivationMode activation_mode, DeviceMemory<Eigen::half>* y,
326 const DeviceMemory<double>& biases, dnn::ActivationMode activation_mode,
340 const DeviceMemory<float>& biases, dnn::ActivationMode activation_mode,
355 dnn::ActivationMode activation_mode,
370 const DeviceMemory<float>& biases, dnn::ActivationMode activation_mode,
464 bool DoActivate(Stream* stream, dnn::ActivationMode activation_mode,
592 const DeviceMemory<float>& bias_data, dnn::ActivationMode activation_mode,
605 dnn::ActivationMode activation_mode, DeviceMemory<float>* y_data,
616 dnn::ActivationMode activation_mode, DeviceMemory<Eigen::half>* y_data,
[all …]
Drocm_dnn.cc938 ScopedActivationDescriptor(dnn::ActivationMode activation_mode) in ScopedActivationDescriptor()
950 case dnn::ActivationMode::kNone: in ScopedActivationDescriptor()
954 case dnn::ActivationMode::kSigmoid: in ScopedActivationDescriptor()
958 case dnn::ActivationMode::kRelu: in ScopedActivationDescriptor()
962 case dnn::ActivationMode::kRelu6: in ScopedActivationDescriptor()
967 case dnn::ActivationMode::kTanh: in ScopedActivationDescriptor()
3529 dnn::ActivationMode activation_mode, DeviceMemory<Eigen::half>* y, in DoBatchNormalizationForward()
3549 dnn::ActivationMode activation_mode, DeviceMemory<float>* y, in DoBatchNormalizationForward()
3571 dnn::ActivationMode activation_mode, DeviceMemory<T>* y, in DoBatchNormalizationForwardImpl()
3683 const DeviceMemory<double>& biases, dnn::ActivationMode activation_mode, in DoFusedConvolve()
[all …]
/external/tensorflow/tensorflow/stream_executor/
Ddnn.cc95 std::string ActivationModeString(ActivationMode mode) { in ActivationModeString()
97 case ActivationMode::kNone: in ActivationModeString()
99 case ActivationMode::kSigmoid: in ActivationModeString()
101 case ActivationMode::kRelu: in ActivationModeString()
103 case ActivationMode::kRelu6: in ActivationModeString()
105 case ActivationMode::kReluX: in ActivationModeString()
107 case ActivationMode::kTanh: in ActivationModeString()
109 case ActivationMode::kBandPass: in ActivationModeString()
Ddnn.h927 std::string ActivationModeString(ActivationMode mode);
1019 dnn::ActivationMode activation_mode, DeviceMemory<float>* y, in DoBatchNormalizationForward()
1038 dnn::ActivationMode activation_mode, DeviceMemory<Eigen::half>* y, in DoBatchNormalizationForward()
1151 const DeviceMemory<double>& biases, dnn::ActivationMode activation_mode, in DoFusedConvolve()
1169 const DeviceMemory<float>& biases, dnn::ActivationMode activation_mode, in DoFusedConvolve()
1189 dnn::ActivationMode activation_mode, in DoFusedConvolve()
1209 const DeviceMemory<float>& biases, dnn::ActivationMode activation_mode, in DoFusedConvolve()
1230 dnn::ActivationMode /*activation_mode*/, in DoFusedConvolve() argument
1764 virtual bool DoActivate(Stream* stream, ActivationMode activation_mode, in DoActivate()
2494 const DeviceMemory<float>& bias_data, dnn::ActivationMode activation_mode, in DoFusedConvolutionBiasActivation()
[all …]
Ddnn.proto50 enum ActivationMode { enum
Dstream.h254 dnn::ActivationMode activation_mode, DeviceMemory<float> *y,
279 dnn::ActivationMode activation_mode, DeviceMemory<Eigen::half> *y,
362 const DeviceMemory<double> &biases, dnn::ActivationMode activation_mode,
376 const DeviceMemory<float> &biases, dnn::ActivationMode activation_mode,
391 dnn::ActivationMode activation_mode,
405 const DeviceMemory<float> &biases, dnn::ActivationMode activation_mode,
419 const DeviceMemory<float> &biases, dnn::ActivationMode activation_mode,
609 Stream &ThenActivate(dnn::ActivationMode activation_mode,
616 Stream &ThenActivateWithOptions(dnn::ActivationMode activation_mode,
Dstream.cc57 std::string ToVlogString(dnn::ActivationMode mode) { in ToVlogString()
350 dnn::ActivationMode activation_mode, DeviceMemory<float> *y, in ThenBatchNormalizationForward()
401 dnn::ActivationMode activation_mode, DeviceMemory<Eigen::half> *y, in ThenBatchNormalizationForward()
454 const DeviceMemory<double> &biases, dnn::ActivationMode activation_mode, in FusedConvolveWithAlgorithm()
485 const DeviceMemory<float> &biases, dnn::ActivationMode activation_mode, in FusedConvolveWithAlgorithm()
517 dnn::ActivationMode activation_mode, in FusedConvolveWithAlgorithm()
548 const DeviceMemory<float> &biases, dnn::ActivationMode activation_mode, in FusedConvolveWithAlgorithm()
579 const DeviceMemory<float> &biases, dnn::ActivationMode activation_mode, in FusedConvolveWithAlgorithm()
1017 Stream &Stream::ThenActivate(dnn::ActivationMode activation_mode, in ThenActivate()
1025 Stream &Stream::ThenActivateWithOptions(dnn::ActivationMode activation_mode, in ThenActivateWithOptions()
/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_dnn.h226 dnn::ActivationMode activation_mode, DeviceMemory<float>* y,
240 dnn::ActivationMode activation_mode, DeviceMemory<Eigen::half>* y,
288 const DeviceMemory<double>& biases, dnn::ActivationMode activation_mode,
302 const DeviceMemory<float>& biases, dnn::ActivationMode activation_mode,
317 dnn::ActivationMode activation_mode,
332 const DeviceMemory<float>& biases, dnn::ActivationMode activation_mode,
346 const DeviceMemory<float>& biases, dnn::ActivationMode activation_mode,
440 bool DoActivate(Stream* stream, dnn::ActivationMode activation_mode,
597 dnn::ActivationMode activation_mode, DeviceMemory<T>* y,
625 const DeviceMemory<BiasType>& biases, dnn::ActivationMode activation_mode,
Dcuda_dnn.cc859 CudnnActivationDescriptor(dnn::ActivationMode activation_mode, in CudnnActivationDescriptor()
866 case dnn::ActivationMode::kNone: in CudnnActivationDescriptor()
869 case dnn::ActivationMode::kRelu6: in CudnnActivationDescriptor()
873 case dnn::ActivationMode::kReluX: in CudnnActivationDescriptor()
877 case dnn::ActivationMode::kRelu: in CudnnActivationDescriptor()
880 case dnn::ActivationMode::kSigmoid: in CudnnActivationDescriptor()
883 case dnn::ActivationMode::kTanh: in CudnnActivationDescriptor()
3185 const DeviceMemory<BiasType>& biases, dnn::ActivationMode activation_mode, in DoFusedConvolveImpl()
3191 if (activation_mode != dnn::ActivationMode::kRelu && in DoFusedConvolveImpl()
3192 activation_mode != dnn::ActivationMode::kNone) { in DoFusedConvolveImpl()
[all …]
/external/tensorflow/tensorflow/compiler/mlir/xla/
Dattribute_exporter.h32 StatusOr<stream_executor::dnn::ActivationMode> ConvertConvActivationMode(
Dattribute_exporter.cc60 StatusOr<stream_executor::dnn::ActivationMode> ConvertConvActivationMode( in ConvertConvActivationMode()
/external/tensorflow/tensorflow/core/protobuf/
Dconv_autotuning.proto26 stream_executor.dnn.ActivationMode activation = 8;
/external/tensorflow/tensorflow/core/kernels/
Dconv_ops_fused_impl.h310 const se::dnn::ActivationMode activation_mode)
322 std::tuple<ConvParameters::ParameterDataType, se::dnn::ActivationMode>;
329 se::dnn::ActivationMode activation_mode_;
563 se::dnn::ActivationMode dnn_activation_mode;
566 dnn_activation_mode = se::dnn::ActivationMode::kRelu;
Dgpu_utils.h238 double side_value_scale, se::dnn::ActivationMode activation_mode,
Dgpu_utils.cc178 double side_value_scale, se::dnn::ActivationMode activation_mode, in LogFusedConvForwardAutotuneResults()
Dfused_batch_norm_op.cc622 se::dnn::ActivationMode AsDnnActivationMode( in AsDnnActivationMode()
626 return se::dnn::ActivationMode::kNone; in AsDnnActivationMode()
628 return se::dnn::ActivationMode::kRelu; in AsDnnActivationMode()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dgpu_conv_runner.h60 se::dnn::ActivationMode mode;
Dcudnn_batchnorm_runner.cc149 se::dnn::ActivationMode::kNone, // in RunCudnnBatchNormForwardInferenceImpl()
178 se::dnn::ActivationMode::kNone, // in RunCudnnBatchNormForwardTrainingImpl()
Dbackend_configs.proto37 // stream_executor::dnn::ActivationMode.
Dcudnn_fused_conv_rewriter.cc248 static_cast<int64>(se::dnn::ActivationMode::kRelu)); in TryRewriteToCudnnForwardRelu()
Dgpu_conv_runner.cc309 static_cast<se::dnn::ActivationMode>(backend_config.activation_mode()); in GetGpuConvConfig()
Dir_emitter_unnested.cc1187 TF_ASSIGN_OR_RETURN(stream_executor::dnn::ActivationMode activation_mode, in EmitConvolutionThunkFromMlir()
/external/tensorflow/tensorflow/compiler/mlir/xla/transforms/
Dmhlo_to_lhlo_with_xla.cc703 stream_executor::dnn::ActivationMode activation) { in GetLHLOActivation()
793 auto se_activation = static_cast<stream_executor::dnn::ActivationMode>( in EmitDnnConvolution()