Home
last modified time | relevance | path

Searched refs:inv_var (Results 1 – 9 of 9) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dfused_batch_norm_op.cu.cc50 T inv_var = variance[index]; in InvVarianceToVarianceKernel() local
51 T var = __fdividef(1, inv_var * inv_var) - T(epsilon); in InvVarianceToVarianceKernel()
Dfused_batch_norm_op.cc333 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/
Dcuda_dnn.h246 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,
Dcuda_dnn.cc3282 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/
Ddnn.h1042 const DeviceMemory<float>& mean, const DeviceMemory<float>& inv_var,
1056 const DeviceMemory<float>& mean, const DeviceMemory<float>& inv_var,
Dstream.h248 const DeviceMemory<float> &inv_var, const dnn::BatchDescriptor &x_desc,
269 const DeviceMemory<float> &mean, const DeviceMemory<float> &inv_var,
Dstream.cc370 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/
Drocm_dnn.h246 const DeviceMemory<float>& mean, const DeviceMemory<float>& inv_var,
Drocm_dnn.cc3087 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()