HomeSort by relevance Sort by last modified time
    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