Searched defs:x_descriptor (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/stream_executor/ |
| D | dnn.h | 2462 Stream* stream, const dnn::BatchDescriptor& x_descriptor, in DoFusedBatchNormActivationInference()
2475 Stream* stream, const dnn::BatchDescriptor& x_descriptor, in DoFusedBatchNormActivationInference()
2516 Stream* stream, const dnn::BatchDescriptor& x_descriptor, in DoFusedBatchNormActivationForward()
2529 Stream* stream, const dnn::BatchDescriptor& x_descriptor, in DoFusedBatchNormActivationForward()
|
| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| D | rocm_dnn.cc | 3693 ScopedTensorDescriptor x_descriptor{x_desc, in DoBatchNormalizationForwardImpl() local
3772 ScopedTensorDescriptor x_descriptor{ in DoBatchNormalizationBackwardImpl() local
4667 const dnn::BatchDescriptor& x_descriptor, const DeviceMemory<T>& x_data, in DoFusedBatchNormActivationInferenceImpl()
4741 Stream* stream, const dnn::BatchDescriptor& x_descriptor, in DoFusedBatchNormActivationInference()
4757 Stream* stream, const dnn::BatchDescriptor& x_descriptor, in DoFusedBatchNormActivationInference()
4776 const dnn::BatchDescriptor& x_descriptor, const DeviceMemory<T>& x_data, in DoFusedBatchNormActivationForwardImpl()
4853 Stream* stream, const dnn::BatchDescriptor& x_descriptor, in DoFusedBatchNormActivationForward()
4870 Stream* stream, const dnn::BatchDescriptor& x_descriptor, in DoFusedBatchNormActivationForward()
|
| /external/tensorflow/tensorflow/compiler/xla/stream_executor/cuda/ |
| D | cuda_dnn.cc | 1878 const CudnnTensorDescriptor& x_descriptor, in CreateBatchNormForwardWorkspace()
1902 const CudnnTensorDescriptor& x_descriptor, in CreateBatchNormBackwardWorkspace()
5470 CudnnTensorDescriptor x_descriptor(x_desc, ToCudnnDataType(input_data_type)); in DoBatchNormalizationForwardImpl() local
5659 CudnnTensorDescriptor x_descriptor( in DoBatchNormalizationBackwardImpl() local
|