Searched defs:x_descriptor (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| D | rocm_dnn.cc | 3049 ScopedTensorDescriptor x_descriptor{x_desc, in DoBatchNormalizationForwardImpl() local
3122 ScopedTensorDescriptor x_descriptor{ in DoBatchNormalizationBackwardImpl() local
4132 const dnn::BatchDescriptor& x_descriptor, const DeviceMemory<T>& x_data, in DoFusedBatchNormActivationInferenceImpl()
4206 Stream* stream, const dnn::BatchDescriptor& x_descriptor, in DoFusedBatchNormActivationInference()
4222 Stream* stream, const dnn::BatchDescriptor& x_descriptor, in DoFusedBatchNormActivationInference()
4241 const dnn::BatchDescriptor& x_descriptor, const DeviceMemory<T>& x_data, in DoFusedBatchNormActivationForwardImpl()
4318 Stream* stream, const dnn::BatchDescriptor& x_descriptor, in DoFusedBatchNormActivationForward()
4335 Stream* stream, const dnn::BatchDescriptor& x_descriptor, in DoFusedBatchNormActivationForward()
|
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cuda_dnn.cc | 3233 CudnnTensorDescriptor x_descriptor(x_desc, ToCudnnDataType(input_data_type)); in DoBatchNormalizationForwardImpl() local
3318 CudnnTensorDescriptor x_descriptor( in DoBatchNormalizationBackwardImpl() local
|