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