Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_dnn.cc4143 ScopedTensorDescriptor scale_offset_mean_variance_nd{ in DoFusedBatchNormActivationInferenceImpl() local
4150 miopen.handle(), x_nd.handle(), scale_offset_mean_variance_nd.handle(), in DoFusedBatchNormActivationInferenceImpl()
4254 ScopedTensorDescriptor scale_offset_mean_variance_nd{ in DoFusedBatchNormActivationForwardImpl() local
4261 miopen.handle(), x_nd.handle(), scale_offset_mean_variance_nd.handle(), in DoFusedBatchNormActivationForwardImpl()
4370 ScopedTensorDescriptor scale_offset_mean_variance_nd{ in DoFusedBatchNormActivationBackwardImpl() local
4378 scale_offset_mean_variance_nd.handle(), activation_desc}; in DoFusedBatchNormActivationBackwardImpl()