Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_dnn.cc4049 ScopedActivationDescriptor activation_desc{activation_mode}; in DoFusedConvolutionBiasActivationImpl() local
4053 conv.handle(), bias_nd.handle(), activation_desc}; in DoFusedConvolutionBiasActivationImpl()
4078 status = fusion_plan.SetActivationForwardArgs(activation_desc); in DoFusedConvolutionBiasActivationImpl()
4147 ScopedActivationDescriptor activation_desc{activation_mode}; in DoFusedBatchNormActivationInferenceImpl() local
4151 activation_desc}; in DoFusedBatchNormActivationInferenceImpl()
4174 status = fusion_plan.SetActivationForwardArgs(activation_desc); in DoFusedBatchNormActivationInferenceImpl()
4258 ScopedActivationDescriptor activation_desc{activation_mode}; in DoFusedBatchNormActivationForwardImpl() local
4262 activation_desc}; in DoFusedBatchNormActivationForwardImpl()
4286 status = fusion_plan.SetActivationForwardArgs(activation_desc); in DoFusedBatchNormActivationForwardImpl()
4374 ScopedActivationDescriptor activation_desc{activation_mode}; in DoFusedBatchNormActivationBackwardImpl() local
[all …]
/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_dnn.cc2981 CudnnActivationDescriptor activation_desc( in DoFusedConvolveImpl() local
3000 << "\nactivation_desc.handle() = " << activation_desc.handle() in DoFusedConvolveImpl()
3022 /*activationDesc=*/activation_desc.handle(), in DoFusedConvolveImpl()
3709 CudnnActivationDescriptor activation_desc( in DoActivate() local
3721 cudnn.handle(), activation_desc.handle(), &alpha, input_nd.handle(), in DoActivate()