OpenGrok
Home
Sort by relevance
Sort by last modified time
Full Search
Definition
Symbol
File Path
History
|
|
Help
Searched
defs:scale_backprop
(Results
1 - 3
of
3
) sorted by null
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
batch_norm_op.cc
147
xla::ComputationDataHandle
scale_backprop
;
variable
155
scale_backprop
= b->GetTupleElement(output, 1);
165
//
scale_backprop
= y_backprop * ((x - pop_mean) * rsqrt(pop_var +
185
scale_backprop
= b->Mul(scratch1, scratch2);
189
ctx->SetOutput(1,
scale_backprop
);
/external/tensorflow/tensorflow/core/kernels/
fused_batch_norm_op.cc
150
typename TTypes<U>::Vec
scale_backprop
(scale_backprop_output->vec<U>());
158
//
scale_backprop
= sum(y_backprop *
193
scale_backprop
.device(d) =
367
Tensor*
scale_backprop
, Tensor* offset_backprop,
457
StreamExecutorUtil::AsDeviceMemory<U>(*
scale_backprop
);
629
Tensor*
scale_backprop
= nullptr;
variable
631
&
scale_backprop
));
651
f(context->eigen_device<Device>(),
scale_backprop
->flat<U>());
659
saved_maybe_inv_var_or_pop_var, epsilon_, x_backprop,
scale_backprop
,
678
x_backprop,
scale_backprop
, offset_backprop, scratch1.vec<U>()
[
all
...]
mkl_fused_batch_norm_op.cc
450
Tensor*
scale_backprop
= nullptr;
variable
452
AllocateOutputSetMklShape(context, 1, &
scale_backprop
, tf_shape_scale_shift,
465
static_cast<const float*>(
scale_backprop
->flat<T>().data()));
[
all
...]
Completed in 157 milliseconds