OpenGrok
Home
Sort by relevance
Sort by last modified time
Full Search
Definition
Symbol
File Path
History
|
|
Help
Searched
defs:batch_var
(Results
1 - 2
of
2
) sorted by null
/external/tensorflow/tensorflow/core/kernels/
fused_batch_norm_op.cc
69
typename TTypes<U>::Vec
batch_var
(batch_var_output->vec<U>());
114
batch_var
.device(d) = variance * rest_size_adjust;
225
Tensor* batch_mean, Tensor*
batch_var
, Tensor* saved_mean,
247
f(context->eigen_device<GPUDevice>(),
batch_var
->flat<U>());
305
auto batch_var_ptr = StreamExecutorUtil::AsDeviceMemory<U>(*
batch_var
);
556
Tensor*
batch_var
= nullptr;
variable
558
context->allocate_output(2, scale.shape(), &
batch_var
));
568
y, batch_mean,
batch_var
, saved_mean, saved_maybe_inv_var,
[
all
...]
/external/tensorflow/tensorflow/core/grappler/costs/
virtual_scheduler_test.cc
248
auto
batch_var
= batch_norm.batch_variance;
local
251
auto z2 = ops::Add(s.WithOpName("z2"),
batch_var
,
batch_var
);
252
auto z3 = ops::Add(s.WithOpName("z3"),
batch_var
,
batch_var
);
259
auto z4 = ops::NoOp(s.WithControlDependencies(
batch_var
).WithOpName("z4"));
811
auto
batch_var
= batch_norm.batch_variance;
local
820
ops::Identity(s.WithOpName("batch_var1").WithDevice(kCPU1),
batch_var
);
[
all
...]
Completed in 85 milliseconds