Searched defs:x_desc (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | fused_batch_norm_op.cc | 303 se::dnn::BatchDescriptor x_desc; in operator ()() local
457 se::dnn::BatchDescriptor x_desc; in operator ()() local
|
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cuda_dnn.cc | 3183 const dnn::BatchDescriptor& x_desc, in DoBatchNormalizationForward()
3204 const dnn::BatchDescriptor& x_desc, in DoBatchNormalizationForward()
3227 const dnn::BatchDescriptor& x_desc, in DoBatchNormalizationForwardImpl()
3283 const dnn::BatchDescriptor& x_desc, in DoBatchNormalizationBackward()
3298 const dnn::BatchDescriptor& x_desc, in DoBatchNormalizationBackward()
3314 const DeviceMemory<U>& inv_var, const dnn::BatchDescriptor& x_desc, in DoBatchNormalizationBackwardImpl()
|
| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| D | rocm_dnn.cc | 3001 const dnn::BatchDescriptor& x_desc, in DoBatchNormalizationForward()
3020 const dnn::BatchDescriptor& x_desc, in DoBatchNormalizationForward()
3041 const dnn::BatchDescriptor& x_desc, in DoBatchNormalizationForwardImpl()
3088 const dnn::BatchDescriptor& x_desc, in DoBatchNormalizationBackward()
3102 const dnn::BatchDescriptor& x_desc, in DoBatchNormalizationBackward()
3117 const DeviceMemory<U>& variance, const dnn::BatchDescriptor& x_desc, in DoBatchNormalizationBackwardImpl()
|
| /external/tensorflow/tensorflow/stream_executor/ |
| D | stream.cc | 344 const dnn::BatchDescriptor &x_desc, in ThenBatchNormalizationForward()
370 const DeviceMemory<float> &inv_var, const dnn::BatchDescriptor &x_desc, in ThenBatchNormalizationBackward()
394 const dnn::BatchDescriptor &x_desc, in ThenBatchNormalizationForward()
421 const dnn::BatchDescriptor &x_desc, in ThenBatchNormalizationBackward()
|