Searched defs:saved_mean (Results 1 – 6 of 6) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/stream_executor/rocm/ |
| H A D | rocm_dnn.cc | 1185 void* running_variance, void* saved_mean, void* saved_inv_variance, in SetBatchNormForwardArgs()
1210 const void* saved_mean, in SetBatchNormBackwardArgs()
1551 void* saved_mean, void* saved_var, in SetBatchNormForwardArgs()
1651 const void* saved_mean, in SetBatchNormBackwardArgs()
3647 DeviceMemory<float>* saved_mean, DeviceMemory<float>* saved_inv_var, in DoBatchNormalizationForward()
3667 DeviceMemory<float>* saved_mean, DeviceMemory<float>* saved_inv_var, in DoBatchNormalizationForward()
3689 DeviceMemory<U>* saved_mean, DeviceMemory<U>* saved_inv_var, in DoBatchNormalizationForwardImpl()
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/mkldnn/ |
| H A D | Normalization.cpp | 163 ideep::tensor saved_mean; in mkldnn_batch_norm() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/ |
| H A D | fused_batch_norm_op.cc | 1364 Tensor* saved_mean = nullptr; in ComputeWithReservedSpace() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/ |
| H A D | stream.cc | 338 DeviceMemory<float> *saved_mean, DeviceMemory<float> *saved_inv_var, in ThenBatchNormalizationForward()
392 DeviceMemory<float> *saved_mean, DeviceMemory<float> *saved_inv_var, in ThenBatchNormalizationForward()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/cuda/ |
| H A D | cuda_dnn.cc | 5418 DeviceMemory<float>* saved_mean, DeviceMemory<float>* saved_inv_var, in DoBatchNormalizationForward()
5442 DeviceMemory<float>* saved_mean, DeviceMemory<float>* saved_inv_var, in DoBatchNormalizationForward()
5467 DeviceMemory<U>* saved_mean, DeviceMemory<U>* saved_inv_var, in DoBatchNormalizationForwardImpl()
|
| /aosp_15_r20/external/pytorch/torch/csrc/autograd/ |
| H A D | FunctionsManual.cpp | 6198 const Tensor& saved_mean, in batch_norm_jvp()
6251 const Tensor& saved_mean, in layer_norm_jvp()
6290 const Tensor& saved_mean, in group_norm_jvp()
|