/external/tensorflow/tensorflow/core/kernels/ |
D | fused_batch_norm_op.cu.cc | 50 T inv_var = variance[index]; in InvVarianceToVarianceKernel() local 51 T var = __fdividef(1, inv_var * inv_var) - T(epsilon); in InvVarianceToVarianceKernel()
|
D | fused_batch_norm_op.cc | 333 Tensor inv_var; in operator ()() local 336 estimated_variance.shape(), &inv_var)); in operator ()() 337 auto inv_var_ptr = StreamExecutorUtil::AsDeviceMemory<U>(inv_var); in operator ()()
|
/external/tensorflow/tensorflow/stream_executor/cuda/ |
D | cuda_dnn.h | 246 const DeviceMemory<float>& mean, const DeviceMemory<float>& inv_var, 255 const DeviceMemory<float>& mean, const DeviceMemory<float>& inv_var, 576 const DeviceMemory<U>& inv_var, const dnn::BatchDescriptor& x_desc,
|
D | cuda_dnn.cc | 3282 const DeviceMemory<float>& mean, const DeviceMemory<float>& inv_var, in DoBatchNormalizationBackward() argument 3289 x, scale, mean, inv_var, x_desc, scale_offset_desc, in DoBatchNormalizationBackward() 3297 const DeviceMemory<float>& mean, const DeviceMemory<float>& inv_var, in DoBatchNormalizationBackward() argument 3304 x, scale, mean, inv_var, x_desc, scale_offset_desc, in DoBatchNormalizationBackward() 3314 const DeviceMemory<U>& inv_var, const dnn::BatchDescriptor& x_desc, in DoBatchNormalizationBackwardImpl() argument 3339 mean.opaque(), inv_var.opaque())); in DoBatchNormalizationBackwardImpl()
|
/external/tensorflow/tensorflow/stream_executor/ |
D | dnn.h | 1042 const DeviceMemory<float>& mean, const DeviceMemory<float>& inv_var, 1056 const DeviceMemory<float>& mean, const DeviceMemory<float>& inv_var,
|
D | stream.h | 248 const DeviceMemory<float> &inv_var, const dnn::BatchDescriptor &x_desc, 269 const DeviceMemory<float> &mean, const DeviceMemory<float> &inv_var,
|
D | stream.cc | 370 const DeviceMemory<float> &inv_var, const dnn::BatchDescriptor &x_desc, in ThenBatchNormalizationBackward() argument 380 this, y_backprop, x, scale, mean, inv_var, x_desc, scale_offset_desc, in ThenBatchNormalizationBackward() 420 const DeviceMemory<float> &mean, const DeviceMemory<float> &inv_var, in ThenBatchNormalizationBackward() argument 431 this, y_backprop, x, scale, mean, inv_var, x_desc, scale_offset_desc, in ThenBatchNormalizationBackward()
|
/external/tensorflow/tensorflow/stream_executor/rocm/ |
D | rocm_dnn.h | 246 const DeviceMemory<float>& mean, const DeviceMemory<float>& inv_var,
|
D | rocm_dnn.cc | 3087 const DeviceMemory<float>& mean, const DeviceMemory<float>& inv_var, in DoBatchNormalizationBackward() argument 3093 stream, miopenHalf, miopenFloat, y_backprop, x, scale, mean, inv_var, in DoBatchNormalizationBackward()
|