HomeSort by relevance Sort by last modified time
    Searched defs:estimated_variance (Results 1 - 2 of 2) sorted by null

  /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...]

Completed in 772 milliseconds