OpenGrok
Home
Sort by relevance
Sort by last modified time
Full Search
Definition
Symbol
File Path
History
|
|
Help
Searched
defs:offset_backprop
(Results
1 - 2
of
2
) sorted by null
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
batch_norm_op.cc
148
xla::ComputationDataHandle
offset_backprop
;
variable
156
offset_backprop
= b->GetTupleElement(output, 2);
164
//
offset_backprop
= sum(y_backprop)
168
offset_backprop
=
190
ctx->SetOutput(2,
offset_backprop
);
/external/tensorflow/tensorflow/core/kernels/
fused_batch_norm_op.cc
151
typename TTypes<U>::Vec
offset_backprop
(offset_backprop_output->vec<U>());
160
//
offset_backprop
= sum(y_backprop)
196
offset_backprop
.device(d) = y_backprop_sum;
367
Tensor* scale_backprop, Tensor*
offset_backprop
,
459
StreamExecutorUtil::AsDeviceMemory<U>(*
offset_backprop
);
632
Tensor*
offset_backprop
= nullptr;
variable
634
&
offset_backprop
));
652
f(context->eigen_device<Device>(),
offset_backprop
->flat<U>());
660
offset_backprop
, tensor_format_);
678
x_backprop, scale_backprop,
offset_backprop
, scratch1.vec<U>()
[
all
...]
Completed in 1007 milliseconds