| /external/tensorflow/tensorflow/core/kernels/ |
| fused_batch_norm_op.cc | 65 typename TTypes<U>::ConstVec estimated_variance(
117 variance.device(d) = estimated_variance;
224 const Tensor& estimated_variance, U epsilon, Tensor* y,
302 StreamExecutorUtil::AsDeviceMemory<U>(estimated_variance);
315 estimated_variance.shape(), &inv_var));
318 [d, epsilon, estimated_variance,
321 StreamExecutorUtil::AsDeviceMemory<U>(estimated_variance);
521 const Tensor& estimated_variance = context->input(4); variable
536 context, estimated_variance.dims() == 1,
537 errors::InvalidArgument("estimated_variance must be 1-dimensional"
[all...] |
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| cuda_dnn.h | 204 const DeviceMemory<float>& estimated_variance,
217 const DeviceMemory<float>& estimated_variance,
[all...] |
| cuda_dnn.cc | [all...] |
| /external/tensorflow/tensorflow/stream_executor/ |
| dnn.h | 899 // estimated_variance: population variance estimated during training,
923 const DeviceMemory<float>& estimated_variance,
[all...] |
| stream.h | 225 const DeviceMemory<float> &estimated_variance,
246 const DeviceMemory<float> &estimated_variance,
[all...] |
| stream.cc | 318 const DeviceMemory<float> &estimated_variance,
331 this, x, scale, offset, estimated_mean, estimated_variance, x_desc,
368 const DeviceMemory<float> &estimated_variance,
381 this, x, scale, offset, estimated_mean, estimated_variance, x_desc,
[all...] |
