| /external/tensorflow/tensorflow/core/kernels/ |
| fused_batch_norm_op.cc | 87 typename TTypes<U>::Vec saved_mean(saved_mean_output->vec<U>());
121 saved_mean.device(d) = mean;
246 Tensor* batch_mean, Tensor* batch_var, Tensor* saved_mean,
327 auto saved_mean_ptr = StreamExecutorUtil::AsDeviceMemory<U>(*saved_mean);
580 Tensor* saved_mean = nullptr; variable
582 context->allocate_output(3, scale.shape(), &saved_mean));
589 y, batch_mean, batch_var, saved_mean, saved_maybe_inv_var,
[all...] |
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| cuda_dnn.h | 225 DeviceMemory<float>* batch_var, DeviceMemory<float>* saved_mean,
238 DeviceMemory<float>* batch_var, DeviceMemory<float>* saved_mean,
566 DeviceMemory<U>* batch_var, DeviceMemory<U>* saved_mean,
[all...] |
| cuda_dnn.cc | [all...] |
| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| rocm_dnn.cc | [all...] |
| rocm_dnn.h | 216 DeviceMemory<float>* batch_var, DeviceMemory<float>* saved_mean,
229 DeviceMemory<float>* batch_var, DeviceMemory<float>* saved_mean,
[all...] |
| /external/tensorflow/tensorflow/stream_executor/ |
| stream.h | 240 DeviceMemory<float> *batch_var, DeviceMemory<float> *saved_mean,
261 DeviceMemory<float> *batch_var, DeviceMemory<float> *saved_mean,
[all...] |
| stream.cc | 347 DeviceMemory<float> *batch_var, DeviceMemory<float> *saved_mean,
357 scale_offset_desc, epsilon, y, batch_mean, batch_var, saved_mean,
397 DeviceMemory<float> *batch_var, DeviceMemory<float> *saved_mean,
407 scale_offset_desc, epsilon, y, batch_mean, batch_var, saved_mean,
[all...] |
