Home
last modified time | relevance | path

Searched refs:grad_scale (Results 1 – 10 of 10) sorted by relevance

/external/tensorflow/tensorflow/compiler/tests/
Dfused_batchnorm_test.py68 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/
Dnn_grad.py962 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
980 grad_scale = math_ops.reduce_sum(
983 return math_ops.cast(grad_x, x_dtype), grad_scale, grad_offset
1016 grad_x, grad_scale, grad_offset = _BatchNormGrad(
1019 grad_grad_y, grad_x, grad_scale = tape.gradient(
1020 [grad_x, grad_scale, grad_offset], [grad_y, x, scale], grad_initial)
1021 return grad_grad_y, grad_x, grad_scale, None, None
Dnn_fused_batchnorm_test.py275 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/python/kernel_tests/distributions/
Dlaplace_test.py252 grad_loc, grad_scale = tape.gradient(samples, [loc, scale])
254 self.assertIsNotNone(grad_scale)
/external/tensorflow/tensorflow/core/kernels/
Dtraining_ops_gpu.cu.cc336 auto grad_scale = (logbase_bcast * sign_decay_bcast * sign_gm).exp(); in operator ()() local
337 var.device(d) -= lr_bcast * grad_scale * grad; in operator ()()
Dtraining_ops.cc453 auto grad_scale = (logbase() * sign_decay() * sign_gm).exp(); in operator ()() local
454 var.device(d) -= lr() * grad_scale * grad; in operator ()()
/external/tensorflow/tensorflow/compiler/xla/service/
Dbatchnorm_expander.cc539 auto grad_scale = add_binary(feature_shape, HloOpcode::kMultiply, in HandleBatchNormGrad() local
582 HloInstruction::CreateTuple({grad_activation, grad_scale, grad_beta}); in HandleBatchNormGrad()
/external/tensorflow/tensorflow/compiler/xla/tests/
Dbatch_normalization_test.cc748 auto grad_scale = ReferenceUtil::Reduce4DTo1D( in XLA_TEST_P() local
831 {expected_grad_activation, LiteralUtil::CreateR1<float>(grad_scale), in XLA_TEST_P()
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dtraining_ops.cc907 xla::XlaOp grad_scale = ComputeGradientScale(alpha, decay); in Compile() local
908 var = var - lr * grad_scale * grad; in Compile()
/external/tensorflow/tensorflow/compiler/xla/g3doc/
Doperation_semantics.md164 | `grad_scale` | `XlaOp` | gradient with respect to input |