| /external/tensorflow/tensorflow/compiler/tests/ |
| fused_batchnorm_test.py | 54 # grad_scale =
68 grad_scale = np.sum(
71 return grad_x, grad_scale, grad_offset
229 grad_x, grad_scale, grad_offset, _, _ = gen_nn_ops.fused_batch_norm_grad(
239 [grad_x, grad_scale, grad_offset], {
287 grad_x, grad_scale, grad_offset, _, _ = out
293 [grad_x, grad_scale, grad_offset], {
|
| /external/tensorflow/tensorflow/python/ops/ |
| nn_grad.py | 850 grad_scale: gradient for scale, which is sum(grad_y * (x - mean(x)) *
931 A tuple (grad_x, grad_scale, grad_offset), where grad_x is the gradient
932 for x, grad_scale the gradient for scale, and grad_offset the gradient
962 grad_scale = math_ops.rsqrt(var_x + epsilon) * math_ops.reduce_sum(
965 grad_scale = array_ops.squeeze(grad_scale)
967 return math_ops.cast(grad_x, x_dtype), grad_scale, grad_offset
[all...] |
| nn_fused_batchnorm_test.py | 275 grad_x, grad_scale, grad_offset = gradients_impl.gradients(
281 grad_vals = self.evaluate([grad_x, grad_scale, grad_offset])
292 grad_y, x_shape, grad_scale, scale_shape)
300 x, x_shape, grad_scale, scale_shape)
321 grad_y, grad_y32, x_shape, grad_scale, grad_scale32, scale_shape)
329 x, x32, x_shape, grad_scale, grad_scale32, scale_shape)
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| batchnorm_expander.cc | 539 auto grad_scale = add_binary(feature_shape, HloOpcode::kMultiply, local
582 HloInstruction::CreateTuple({grad_activation, grad_scale, grad_beta});
|
| /external/tensorflow/tensorflow/python/kernel_tests/distributions/ |
| laplace_test.py | 252 grad_loc, grad_scale = tape.gradient(samples, [loc, scale])
254 self.assertIsNotNone(grad_scale)
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| batch_normalization_test.cc | 748 auto grad_scale = ReferenceUtil::Reduce4DTo1D( local
[all...] |
| /external/tensorflow/tensorflow/core/kernels/ |
| training_ops_gpu.cu.cc | 330 // auto grad_scale = (logbase() * sign_decay() * sign_gm).exp();
331 // var.device(d) -= lr() * grad_scale * grad;
336 auto grad_scale = (logbase_bcast * sign_decay_bcast * sign_gm).exp(); local
337 var.device(d) -= lr_bcast * grad_scale * grad;
|
| training_ops.cc | 453 auto grad_scale = (logbase() * sign_decay() * sign_gm).exp(); local
454 var.device(d) -= lr() * grad_scale * grad;
[all...] |
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| training_ops.cc | 907 xla::XlaOp grad_scale = ComputeGradientScale(alpha, decay); variable
[all...] |
