| Package | Description |
|---|---|
| org.tensorflow.op | |
| org.tensorflow.op.train |
| Modifier and Type | Method and Description |
|---|---|
<T,U extends Number> |
TrainOps.sparseApplyRmsProp(Operand<T> var,
Operand<T> ms,
Operand<T> mom,
Operand<T> lr,
Operand<T> rho,
Operand<T> momentum,
Operand<T> epsilon,
Operand<T> grad,
Operand<U> indices,
SparseApplyRmsProp.Options... options)
Builds an
SparseApplyRmsProp operation |
| Modifier and Type | Method and Description |
|---|---|
static SparseApplyRmsProp.Options |
SparseApplyRmsProp.useLocking(Boolean useLocking) |
SparseApplyRmsProp.Options |
SparseApplyRmsProp.Options.useLocking(Boolean useLocking) |
| Modifier and Type | Method and Description |
|---|---|
static <T,U extends Number> |
SparseApplyRmsProp.create(Scope scope,
Operand<T> var,
Operand<T> ms,
Operand<T> mom,
Operand<T> lr,
Operand<T> rho,
Operand<T> momentum,
Operand<T> epsilon,
Operand<T> grad,
Operand<U> indices,
SparseApplyRmsProp.Options... options)
Factory method to create a class wrapping a new SparseApplyRmsProp operation.
|
Copyright © 2015–2019. All rights reserved.