Searched defs:x_descriptor (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| D | rocm_dnn.cc | 3577 ScopedTensorDescriptor x_descriptor{x_desc, in DoBatchNormalizationForwardImpl() local
3652 ScopedTensorDescriptor x_descriptor{ in DoBatchNormalizationBackwardImpl() local
4728 const dnn::BatchDescriptor& x_descriptor, const DeviceMemory<T>& x_data, in DoFusedBatchNormActivationInferenceImpl()
4802 Stream* stream, const dnn::BatchDescriptor& x_descriptor, in DoFusedBatchNormActivationInference()
4818 Stream* stream, const dnn::BatchDescriptor& x_descriptor, in DoFusedBatchNormActivationInference()
4837 const dnn::BatchDescriptor& x_descriptor, const DeviceMemory<T>& x_data, in DoFusedBatchNormActivationForwardImpl()
4914 Stream* stream, const dnn::BatchDescriptor& x_descriptor, in DoFusedBatchNormActivationForward()
4931 Stream* stream, const dnn::BatchDescriptor& x_descriptor, in DoFusedBatchNormActivationForward()
|
| /external/tensorflow/tensorflow/stream_executor/ |
| D | dnn.h | 2528 Stream* stream, const dnn::BatchDescriptor& x_descriptor, in DoFusedBatchNormActivationInference()
2541 Stream* stream, const dnn::BatchDescriptor& x_descriptor, in DoFusedBatchNormActivationInference()
2582 Stream* stream, const dnn::BatchDescriptor& x_descriptor, in DoFusedBatchNormActivationForward()
2595 Stream* stream, const dnn::BatchDescriptor& x_descriptor, in DoFusedBatchNormActivationForward()
|
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cuda_dnn.cc | 1697 const CudnnTensorDescriptor& x_descriptor, in CreateBatchNormForwardWorkspace()
1720 const CudnnTensorDescriptor& x_descriptor, in CreateBatchNormBackwardWorkspace()
3508 CudnnTensorDescriptor x_descriptor(x_desc, ToCudnnDataType(input_data_type)); in DoBatchNormalizationForwardImpl() local
3688 CudnnTensorDescriptor x_descriptor( in DoBatchNormalizationBackwardImpl() local
|