Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_dnn.cc823 miopen_activation_mode_(miopenActivationPASTHRU), in ScopedActivationDescriptor()
834 miopen_activation_mode_ = miopenActivationPASTHRU; in ScopedActivationDescriptor()
838 miopen_activation_mode_ = miopenActivationLOGISTIC; in ScopedActivationDescriptor()
842 miopen_activation_mode_ = miopenActivationRELU; in ScopedActivationDescriptor()
846 miopen_activation_mode_ = miopenActivationRELU; in ScopedActivationDescriptor()
851 miopen_activation_mode_ = miopenActivationTANH; in ScopedActivationDescriptor()
862 handle_, miopen_activation_mode_, alpha_, beta_, gamma_); in ScopedActivationDescriptor()
881 uint64 hash_value = tensorflow::hash<int>()(miopen_activation_mode_); in GetHashValue()
903 miopenActivationMode_t miopen_activation_mode_; member in stream_executor::gpu::ScopedActivationDescriptor
1151 activation_descriptor.miopen_activation_mode_); in ScopedFusionPlanConvolutionBiasActivation()
[all …]