Searched refs:saved_inv_var (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cuda_dnn.h | 226 DeviceMemory<float>* saved_inv_var, bool is_training,
239 DeviceMemory<float>* saved_inv_var, bool is_training,
567 DeviceMemory<U>* saved_inv_var, bool is_training,
|
| D | cuda_dnn.cc | 3187 DeviceMemory<float>* saved_inv_var, bool is_training, in DoBatchNormalizationForward() argument
3194 epsilon, y, batch_mean, batch_var, saved_mean, saved_inv_var, in DoBatchNormalizationForward()
3208 DeviceMemory<float>* saved_inv_var, bool is_training, in DoBatchNormalizationForward() argument
3215 epsilon, y, batch_mean, batch_var, saved_mean, saved_inv_var, in DoBatchNormalizationForward()
3230 DeviceMemory<U>* saved_mean, DeviceMemory<U>* saved_inv_var, in DoBatchNormalizationForwardImpl() argument
3267 saved_inv_var->opaque())); in DoBatchNormalizationForwardImpl()
|
| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| D | rocm_dnn.h | 217 DeviceMemory<float>* saved_inv_var, bool is_training,
230 DeviceMemory<float>* saved_inv_var, bool is_training,
639 DeviceMemory<U>* saved_inv_var, bool is_training,
|
| D | rocm_dnn.cc | 3005 DeviceMemory<float>* saved_inv_var, bool is_training, in DoBatchNormalizationForward() argument
3011 batch_mean, batch_var, saved_mean, saved_inv_var, is_training, in DoBatchNormalizationForward()
3024 DeviceMemory<float>* saved_inv_var, bool is_training, in DoBatchNormalizationForward() argument
3030 batch_mean, batch_var, saved_mean, saved_inv_var, is_training, in DoBatchNormalizationForward()
3044 DeviceMemory<U>* saved_mean, DeviceMemory<U>* saved_inv_var, in DoBatchNormalizationForwardImpl() argument
3066 saved_mean->opaque(), saved_inv_var->opaque()); in DoBatchNormalizationForwardImpl()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | fused_batch_norm_op.cc | 247 Tensor* saved_inv_var, TensorFormat tensor_format, in operator ()()
329 StreamExecutorUtil::AsDeviceMemory<U>(*saved_inv_var); in operator ()()
|
| /external/tensorflow/tensorflow/stream_executor/ |
| D | stream.h | 241 DeviceMemory<float> *saved_inv_var, bool is_training,
262 DeviceMemory<float> *saved_inv_var, bool is_training,
|
| D | stream.cc | 348 DeviceMemory<float> *saved_inv_var, bool is_training, in ThenBatchNormalizationForward() argument
358 saved_inv_var, is_training, std::move(var_to_inv_var), in ThenBatchNormalizationForward()
398 DeviceMemory<float> *saved_inv_var, bool is_training, in ThenBatchNormalizationForward() argument
408 saved_inv_var, is_training, std::move(var_to_inv_var), in ThenBatchNormalizationForward()
|