OpenGrok
Home
Sort by relevance
Sort by last modified time
Full Search
Definition
Symbol
File Path
History
|
|
Help
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