Searched defs:saved_mean (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | fused_batch_norm_op.cc | 87 typename TTypes<U>::Vec saved_mean(saved_mean_output->vec<U>()); in operator ()() local
580 Tensor* saved_mean = nullptr; in Compute() local
|
| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| D | rocm_dnn.cc | 1015 void* saved_mean, in SetBatchNormForwardArgs()
1043 const void* saved_mean, in SetBatchNormBackwardArgs()
1384 void* saved_mean, void* saved_var, in SetBatchNormForwardArgs()
1484 const void* saved_mean, in SetBatchNormBackwardArgs()
3004 DeviceMemory<float>* batch_var, DeviceMemory<float>* saved_mean, in DoBatchNormalizationForward()
3023 DeviceMemory<float>* batch_var, DeviceMemory<float>* saved_mean, in DoBatchNormalizationForward()
3044 DeviceMemory<U>* saved_mean, DeviceMemory<U>* saved_inv_var, in DoBatchNormalizationForwardImpl()
|
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cuda_dnn.cc | 3186 DeviceMemory<float>* batch_var, DeviceMemory<float>* saved_mean, in DoBatchNormalizationForward()
3207 DeviceMemory<float>* batch_var, DeviceMemory<float>* saved_mean, in DoBatchNormalizationForward()
3230 DeviceMemory<U>* saved_mean, DeviceMemory<U>* saved_inv_var, in DoBatchNormalizationForwardImpl()
|
| /external/tensorflow/tensorflow/stream_executor/ |
| D | stream.cc | 347 DeviceMemory<float> *batch_var, DeviceMemory<float> *saved_mean, in ThenBatchNormalizationForward()
397 DeviceMemory<float> *batch_var, DeviceMemory<float> *saved_mean, in ThenBatchNormalizationForward()
|