| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cuda_dnn.h | 222 const dnn::BatchDescriptor& x_desc,
235 const dnn::BatchDescriptor& x_desc,
247 const dnn::BatchDescriptor& x_desc,
256 const dnn::BatchDescriptor& x_desc,
563 const dnn::BatchDescriptor& x_desc,
576 const DeviceMemory<U>& inv_var, const dnn::BatchDescriptor& x_desc,
|
| D | cuda_dnn.cc | 3183 const dnn::BatchDescriptor& x_desc, in DoBatchNormalizationForward() argument
3193 offset, estimated_mean, estimated_variance, x_desc, scale_offset_desc, in DoBatchNormalizationForward()
3204 const dnn::BatchDescriptor& x_desc, in DoBatchNormalizationForward() argument
3214 estimated_mean, estimated_variance, x_desc, scale_offset_desc, in DoBatchNormalizationForward()
3227 const dnn::BatchDescriptor& x_desc, in DoBatchNormalizationForwardImpl() argument
3233 CudnnTensorDescriptor x_descriptor(x_desc, ToCudnnDataType(input_data_type)); in DoBatchNormalizationForwardImpl()
3283 const dnn::BatchDescriptor& x_desc, in DoBatchNormalizationBackward() argument
3289 x, scale, mean, inv_var, x_desc, scale_offset_desc, in DoBatchNormalizationBackward()
3298 const dnn::BatchDescriptor& x_desc, in DoBatchNormalizationBackward() argument
3304 x, scale, mean, inv_var, x_desc, scale_offset_desc, in DoBatchNormalizationBackward()
[all …]
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | fused_batch_norm_op.cc | 303 se::dnn::BatchDescriptor x_desc; in operator ()() local
304 x_desc.set_count(batch_size) in operator ()()
362 estimated_variance_ptr, x_desc, scale_offset_desc, in operator ()()
457 se::dnn::BatchDescriptor x_desc; in operator ()() local
458 x_desc.set_count(batch_size) in operator ()()
488 x_desc, scale_offset_desc, static_cast<double>(epsilon), in operator ()()
|
| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| D | rocm_dnn.h | 213 const dnn::BatchDescriptor& x_desc,
226 const dnn::BatchDescriptor& x_desc,
238 const dnn::BatchDescriptor& x_desc,
247 const dnn::BatchDescriptor& x_desc,
635 const dnn::BatchDescriptor& x_desc,
648 const DeviceMemory<U>& variance, const dnn::BatchDescriptor& x_desc,
|
| D | rocm_dnn.cc | 3001 const dnn::BatchDescriptor& x_desc, in DoBatchNormalizationForward() argument
3010 estimated_mean, estimated_variance, x_desc, scale_offset_desc, epsilon, y, in DoBatchNormalizationForward()
3020 const dnn::BatchDescriptor& x_desc, in DoBatchNormalizationForward() argument
3029 estimated_mean, estimated_variance, x_desc, scale_offset_desc, epsilon, y, in DoBatchNormalizationForward()
3041 const dnn::BatchDescriptor& x_desc, in DoBatchNormalizationForwardImpl() argument
3049 ScopedTensorDescriptor x_descriptor{x_desc, in DoBatchNormalizationForwardImpl()
3088 const dnn::BatchDescriptor& x_desc, in DoBatchNormalizationBackward() argument
3094 x_desc, scale_offset_desc, epsilon, x_backprop, scale_backprop, in DoBatchNormalizationBackward()
3102 const dnn::BatchDescriptor& x_desc, in DoBatchNormalizationBackward() argument
3108 x_desc, scale_offset_desc, epsilon, x_backprop, scale_backprop, in DoBatchNormalizationBackward()
[all …]
|
| /external/tensorflow/tensorflow/stream_executor/ |
| D | dnn.h | 995 const dnn::BatchDescriptor& x_desc,
1012 const dnn::BatchDescriptor& x_desc,
1043 const dnn::BatchDescriptor& x_desc,
1057 const dnn::BatchDescriptor& x_desc,
|
| D | stream.cc | 344 const dnn::BatchDescriptor &x_desc, in ThenBatchNormalizationForward() argument
351 VLOG_CALL(PARAM(x), PARAM(scale), PARAM(offset), PARAM(x_desc), in ThenBatchNormalizationForward()
356 this, x, scale, offset, estimated_mean, estimated_variance, x_desc, in ThenBatchNormalizationForward()
370 const DeviceMemory<float> &inv_var, const dnn::BatchDescriptor &x_desc, in ThenBatchNormalizationBackward() argument
374 VLOG_CALL(PARAM(y_backprop), PARAM(x), PARAM(scale), PARAM(x_desc), in ThenBatchNormalizationBackward()
380 this, y_backprop, x, scale, mean, inv_var, x_desc, scale_offset_desc, in ThenBatchNormalizationBackward()
394 const dnn::BatchDescriptor &x_desc, in ThenBatchNormalizationForward() argument
401 VLOG_CALL(PARAM(x), PARAM(scale), PARAM(offset), PARAM(x_desc), in ThenBatchNormalizationForward()
406 this, x, scale, offset, estimated_mean, estimated_variance, x_desc, in ThenBatchNormalizationForward()
421 const dnn::BatchDescriptor &x_desc, in ThenBatchNormalizationBackward() argument
[all …]
|
| D | stream.h | 237 const dnn::BatchDescriptor &x_desc,
248 const DeviceMemory<float> &inv_var, const dnn::BatchDescriptor &x_desc,
258 const dnn::BatchDescriptor &x_desc,
270 const dnn::BatchDescriptor &x_desc,
|