Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_dnn.h546 Stream* stream, const dnn::BatchDescriptor& x_descriptor,
557 Stream* stream, const dnn::BatchDescriptor& x_descriptor,
568 Stream* stream, const dnn::BatchDescriptor& x_descriptor,
579 Stream* stream, const dnn::BatchDescriptor& x_descriptor,
722 const dnn::BatchDescriptor& x_descriptor, const DeviceMemory<T>& x_data,
733 const dnn::BatchDescriptor& x_descriptor, const DeviceMemory<T>& x_data,
Drocm_dnn.cc3049 ScopedTensorDescriptor x_descriptor{x_desc, in DoBatchNormalizationForwardImpl() local
3062 miopen.handle(), mode, &one, &zero, x_descriptor.handle(), x.opaque(), in DoBatchNormalizationForwardImpl()
3063 x_descriptor.handle(), y->opaque(), scale_offset_descriptor.handle(), in DoBatchNormalizationForwardImpl()
3070 miopen.handle(), mode, &one, &zero, x_descriptor.handle(), x.opaque(), in DoBatchNormalizationForwardImpl()
3071 x_descriptor.handle(), y->opaque(), scale_offset_descriptor.handle(), in DoBatchNormalizationForwardImpl()
3122 ScopedTensorDescriptor x_descriptor{ in DoBatchNormalizationBackwardImpl() local
3131 miopen.handle(), mode, &one, &zero, &one, &zero, x_descriptor.handle(), in DoBatchNormalizationBackwardImpl()
3132 x.opaque(), x_descriptor.handle(), y_backprop.opaque(), in DoBatchNormalizationBackwardImpl()
3133 x_descriptor.handle(), x_backprop->opaque(), in DoBatchNormalizationBackwardImpl()
4132 const dnn::BatchDescriptor& x_descriptor, const DeviceMemory<T>& x_data, in DoFusedBatchNormActivationInferenceImpl() argument
[all …]
/external/tensorflow/tensorflow/stream_executor/
Ddnn.h2393 Stream* stream, const dnn::BatchDescriptor& x_descriptor,
2406 Stream* stream, const dnn::BatchDescriptor& x_descriptor,
2447 Stream* stream, const dnn::BatchDescriptor& x_descriptor,
2460 Stream* stream, const dnn::BatchDescriptor& x_descriptor,
/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_dnn.cc3233 CudnnTensorDescriptor x_descriptor(x_desc, ToCudnnDataType(input_data_type)); in DoBatchNormalizationForwardImpl() local
3263 cudnn.handle(), mode, &one, &zero, x_descriptor.handle(), x.opaque(), in DoBatchNormalizationForwardImpl()
3264 x_descriptor.handle(), y->opaque(), scale_offset_descriptor.handle(), in DoBatchNormalizationForwardImpl()
3271 cudnn.handle(), mode, &one, &zero, x_descriptor.handle(), x.opaque(), in DoBatchNormalizationForwardImpl()
3272 x_descriptor.handle(), y->opaque(), scale_offset_descriptor.handle(), in DoBatchNormalizationForwardImpl()
3318 CudnnTensorDescriptor x_descriptor( in DoBatchNormalizationBackwardImpl() local
3334 cudnn.handle(), mode, &one, &zero, &one, &zero, x_descriptor.handle(), in DoBatchNormalizationBackwardImpl()
3335 x.opaque(), x_descriptor.handle(), y_backprop.opaque(), in DoBatchNormalizationBackwardImpl()
3336 x_descriptor.handle(), x_backprop->opaque(), in DoBatchNormalizationBackwardImpl()