| /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.h | 80 typename TTypes<U>::Vec offset_backprop(offset_backprop_output->vec<U>());
98 // offset_backprop = sum(y_backprop)
106 offset_backprop.device(d) = y_backprop_rest_by_depth.sum(reduction_axis);
|
| 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...] |
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| cuda_dnn.h | 233 DeviceMemory<float>* offset_backprop) override;
243 DeviceMemory<float>* offset_backprop) override;
[all...] |
| cuda_dnn.cc | [all...] |
| /external/tensorflow/tensorflow/stream_executor/ |
| dnn.h | [all...] |
| stream.cc | 348 DeviceMemory<float> *offset_backprop) {
351 PARAM(scale_backprop), PARAM(offset_backprop));
356 epsilon, x_backprop, scale_backprop, offset_backprop));
399 DeviceMemory<float> *offset_backprop) {
402 PARAM(scale_backprop), PARAM(offset_backprop));
407 epsilon, x_backprop, scale_backprop, offset_backprop));
[all...] |
| stream.h | 240 DeviceMemory<float> *offset_backprop);
263 DeviceMemory<float> *offset_backprop);
[all...] |
| /external/tensorflow/tensorflow/go/op/ |
| wrappers.go | [all...] |
