| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v1/ |
| D | ResourceApplyPowerSign.pbtxt | 16 name: "logbase"
79 name: "logbase"
143 name: "logbase"
|
| D | ApplyPowerSign.pbtxt | 18 name: "logbase"
87 name: "logbase"
157 name: "logbase"
|
| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v2/ |
| D | ResourceApplyPowerSign.pbtxt | 16 name: "logbase"
79 name: "logbase"
143 name: "logbase"
|
| D | ApplyPowerSign.pbtxt | 18 name: "logbase"
87 name: "logbase"
157 name: "logbase"
|
| /external/tensorflow/tensorflow/core/api_def/base_api/ |
| D | api_def_ResourceApplyPowerSign.pbtxt | 22 name: "logbase"
56 update <- exp(logbase * sign_decay * sign(g) * sign(m_t)) * g
|
| D | api_def_ApplyPowerSign.pbtxt | 22 name: "logbase"
62 update <- exp(logbase * sign_decay * sign(g) * sign(m_t)) * g
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | training_ops_test.cc | 396 auto logbase = Scalar(g, 2); in PowerSign() local
401 {var, m, lr, logbase, sign_decay, beta, grad}); in PowerSign()
|
| D | training_ops.h | 300 typename TTypes<T>::ConstScalar logbase,
|
| D | training_ops.cc | 919 typename TTypes<T>::ConstScalar logbase, in operator ()()
925 auto grad_scale = (logbase() * sign_decay() * sign_gm).exp(); in operator ()()
4619 const Tensor& logbase = ctx->input(3); in Compute() local
4620 OP_REQUIRES(ctx, TensorShapeUtils::IsScalar(logbase.shape()), in Compute()
4622 logbase.shape().DebugString())); in Compute()
4624 OP_REQUIRES(ctx, TensorShapeUtils::IsScalar(logbase.shape()), in Compute()
4644 device, var.flat<T>(), m.flat<T>(), lr.scalar<T>(), logbase.scalar<T>(), in Compute()
4678 typename TTypes<T>::ConstScalar logbase, \
|
| D | training_ops_gpu.cu.cc | 1056 typename TTypes<T>::ConstScalar logbase, in operator ()()
1077 auto logbase_bcast = logbase.reshape(single).broadcast(bcast); in operator ()()
|
| /external/OpenCL-CTS/test_conformance/compiler/ |
| D | test_compile.cpp | 607 double logbase(double a, double base) in logbase() function
620 unsigned int level = (unsigned int)(logbase(numLines, 2.0) + 1.000001)/2; in test_large_multiple_libraries()
785 unsigned int level = (unsigned int)(logbase(numLines, 2.0) + 1.000001)/2; in test_large_multiple_files_multiple_libraries()
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.raw_ops.pbtxt | 169 …argspec: "args=[\'var\', \'m\', \'lr\', \'logbase\', \'sign_decay\', \'beta\', \'grad\', \'use_loc…
3597 …argspec: "args=[\'var\', \'m\', \'lr\', \'logbase\', \'sign_decay\', \'beta\', \'grad\', \'use_loc…
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.raw_ops.pbtxt | 169 …argspec: "args=[\'var\', \'m\', \'lr\', \'logbase\', \'sign_decay\', \'beta\', \'grad\', \'use_loc…
3597 …argspec: "args=[\'var\', \'m\', \'lr\', \'logbase\', \'sign_decay\', \'beta\', \'grad\', \'use_loc…
|
| /external/tensorflow/tensorflow/core/ops/compat/ |
| D | ops_history.v2.pbtxt | 4745 name: "logbase"
4814 name: "logbase"
4884 name: "logbase"
57146 name: "logbase"
57209 name: "logbase"
57273 name: "logbase"
|
| /external/tensorflow/tensorflow/go/op/ |
| D | wrappers.go | 41664 func ResourceApplyPowerSign(scope *Scope, var_ tf.Output, m tf.Output, lr tf.Output, logbase tf.Out…
41675 var_, m, lr, logbase, sign_decay, beta, grad,
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | ops.pbtxt | 1822 name: "logbase"
38563 name: "logbase"
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/ |
| D | tf_generated_ops.td | 11217 update <- exp(logbase * sign_decay * sign(g) * sign(m_t)) * g
11225 … TF_Qint8, TF_Quint8, TF_Uint16, TF_Uint32, TF_Uint64, TF_Uint8]>, [{Must be a scalar.}]>:$logbase,
|