Searched refs:activation_desc (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/stream_executor/cuda/ |
| D | cuda_dnn.cc | 1877 const cudnnActivationDescriptor_t& activation_desc, in CreateBatchNormForwardWorkspace() argument
1889 /*activationDesc=*/activation_desc, in CreateBatchNormForwardWorkspace()
1901 const cudnnActivationDescriptor_t& activation_desc, in CreateBatchNormBackwardWorkspace() argument
1915 /*activationDesc=*/activation_desc, in CreateBatchNormBackwardWorkspace()
4886 CudnnActivationDescriptor activation_desc) { in Create() argument
4905 std::move(bias_nd), std::move(conv), std::move(activation_desc)}}; in Create()
5024 CudnnActivationDescriptor activation_desc) in CudnnLegacyFusedConvRunner() argument
5038 activation_desc_(std::move(activation_desc)) {} in CudnnLegacyFusedConvRunner()
5096 CudnnActivationDescriptor activation_desc(activation_mode, in FusedConvolveRunnerFromDesc() local
5106 std::move(conv), std::move(activation_desc))); in FusedConvolveRunnerFromDesc()
[all …]
|
| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| D | rocm_dnn.cc | 4584 ScopedActivationDescriptor activation_desc{activation_mode}; in DoFusedConvolutionBiasActivationImpl() local
4588 conv.handle(), bias_nd.handle(), activation_desc}; in DoFusedConvolutionBiasActivationImpl()
4613 status = fusion_plan.SetActivationForwardArgs(activation_desc); in DoFusedConvolutionBiasActivationImpl()
4682 ScopedActivationDescriptor activation_desc{activation_mode}; in DoFusedBatchNormActivationInferenceImpl() local
4686 activation_desc}; in DoFusedBatchNormActivationInferenceImpl()
4709 status = fusion_plan.SetActivationForwardArgs(activation_desc); in DoFusedBatchNormActivationInferenceImpl()
4793 ScopedActivationDescriptor activation_desc{activation_mode}; in DoFusedBatchNormActivationForwardImpl() local
4797 activation_desc}; in DoFusedBatchNormActivationForwardImpl()
4821 status = fusion_plan.SetActivationForwardArgs(activation_desc); in DoFusedBatchNormActivationForwardImpl()
4909 ScopedActivationDescriptor activation_desc{activation_mode}; in DoFusedBatchNormActivationBackwardImpl() local
[all …]
|