| /external/tensorflow/tensorflow/core/kernels/ |
| fused_batch_norm_op.cc | 82 typename TTypes<U>::ConstVec estimated_variance(
134 variance.device(d) = estimated_variance;
245 const Tensor& estimated_variance, U epsilon, Tensor* y,
323 StreamExecutorUtil::AsDeviceMemory<U>(estimated_variance);
336 estimated_variance.shape(), &inv_var));
339 [d, epsilon, estimated_variance,
342 StreamExecutorUtil::AsDeviceMemory<U>(estimated_variance);
542 const Tensor& estimated_variance = context->input(4); variable
557 context, estimated_variance.dims() == 1,
558 errors::InvalidArgument("estimated_variance must be 1-dimensional"
[all...] |
| conv_ops_fused_impl.h | 381 // scaling_factor = (estimated_variance + epsilon).rsqrt() * scale
410 const Tensor& estimated_variance = context->input(5); local
422 estimated_variance.dims() == 1,
423 errors::InvalidArgument("estimated_variance must be 1-dimensional",
424 estimated_variance.shape().DebugString()));
429 InitDataPtr(estimated_variance, &args->estimated_variance_data);
433 (estimated_variance.flat<T>() + static_cast<T>(epsilon)).rsqrt() *
[all...] |
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| cuda_dnn.h | 221 const DeviceMemory<float>& estimated_variance,
234 const DeviceMemory<float>& estimated_variance,
562 const DeviceMemory<U>& estimated_variance,
[all...] |
| cuda_dnn.cc | [all...] |
| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| rocm_dnn.h | 212 const DeviceMemory<float>& estimated_variance,
225 const DeviceMemory<float>& estimated_variance,
[all...] |
| rocm_dnn.cc | [all...] |
| /external/tensorflow/tensorflow/stream_executor/ |
| dnn.h | [all...] |
| stream.h | 236 const DeviceMemory<float> &estimated_variance,
257 const DeviceMemory<float> &estimated_variance,
[all...] |
| stream.cc | 343 const DeviceMemory<float> &estimated_variance,
356 this, x, scale, offset, estimated_mean, estimated_variance, x_desc,
393 const DeviceMemory<float> &estimated_variance,
406 this, x, scale, offset, estimated_mean, estimated_variance, x_desc,
[all...] |
