Home
last modified time | relevance | path

Searched defs:grad_scale (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dtraining_ops_gpu.cu.cc336 auto grad_scale = (logbase_bcast * sign_decay_bcast * sign_gm).exp(); in operator ()() local
Dtraining_ops.cc453 auto grad_scale = (logbase() * sign_decay() * sign_gm).exp(); in operator ()() local
/external/tensorflow/tensorflow/compiler/xla/service/
Dbatchnorm_expander.cc539 auto grad_scale = add_binary(feature_shape, HloOpcode::kMultiply, in HandleBatchNormGrad() local
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dtraining_ops.cc907 xla::XlaOp grad_scale = ComputeGradientScale(alpha, decay); in Compile() local
/external/tensorflow/tensorflow/compiler/xla/tests/
Dbatch_normalization_test.cc748 auto grad_scale = ReferenceUtil::Reduce4DTo1D( in XLA_TEST_P() local