| Package | Description |
|---|---|
| org.tensorflow.op | |
| org.tensorflow.op.train |
| Modifier and Type | Method and Description |
|---|---|
<T extends TType> |
TrainOps.conditionalAccumulator(Class<T> dtype,
org.tensorflow.ndarray.Shape shape,
ConditionalAccumulator.Options... options)
A conditional accumulator for aggregating gradients.
|
| Modifier and Type | Method and Description |
|---|---|
static ConditionalAccumulator.Options |
ConditionalAccumulator.container(String container)
Sets the container option.
|
ConditionalAccumulator.Options |
ConditionalAccumulator.Options.container(String container)
Sets the container option.
|
static ConditionalAccumulator.Options |
ConditionalAccumulator.reductionType(String reductionType)
Sets the reductionType option.
|
ConditionalAccumulator.Options |
ConditionalAccumulator.Options.reductionType(String reductionType)
Sets the reductionType option.
|
static ConditionalAccumulator.Options |
ConditionalAccumulator.sharedName(String sharedName)
Sets the sharedName option.
|
ConditionalAccumulator.Options |
ConditionalAccumulator.Options.sharedName(String sharedName)
Sets the sharedName option.
|
| Modifier and Type | Method and Description |
|---|---|
static <T extends TType> |
ConditionalAccumulator.create(Scope scope,
Class<T> dtype,
org.tensorflow.ndarray.Shape shape,
ConditionalAccumulator.Options... options)
Factory method to create a class wrapping a new ConditionalAccumulator operation.
|
Copyright © 2015–2022. All rights reserved.