| Package | Description |
|---|---|
| org.tensorflow.op | |
| org.tensorflow.op.core |
| Modifier and Type | Method and Description |
|---|---|
<U extends Number,T extends Number> |
Ops.cudnnRNNParamsSize(Operand<Integer> numLayers,
Operand<Integer> numUnits,
Operand<Integer> inputSize,
Class<T> T,
Class<U> S,
CudnnRNNParamsSize.Options... options)
Adds an
CudnnRNNParamsSize operation to the graph |
| Modifier and Type | Method and Description |
|---|---|
static CudnnRNNParamsSize.Options |
CudnnRNNParamsSize.direction(String direction) |
CudnnRNNParamsSize.Options |
CudnnRNNParamsSize.Options.direction(String direction) |
static CudnnRNNParamsSize.Options |
CudnnRNNParamsSize.dropout(Float dropout) |
CudnnRNNParamsSize.Options |
CudnnRNNParamsSize.Options.dropout(Float dropout) |
static CudnnRNNParamsSize.Options |
CudnnRNNParamsSize.inputMode(String inputMode) |
CudnnRNNParamsSize.Options |
CudnnRNNParamsSize.Options.inputMode(String inputMode) |
static CudnnRNNParamsSize.Options |
CudnnRNNParamsSize.rnnMode(String rnnMode) |
CudnnRNNParamsSize.Options |
CudnnRNNParamsSize.Options.rnnMode(String rnnMode) |
static CudnnRNNParamsSize.Options |
CudnnRNNParamsSize.seed(Long seed) |
CudnnRNNParamsSize.Options |
CudnnRNNParamsSize.Options.seed(Long seed) |
static CudnnRNNParamsSize.Options |
CudnnRNNParamsSize.seed2(Long seed2) |
CudnnRNNParamsSize.Options |
CudnnRNNParamsSize.Options.seed2(Long seed2) |
| Modifier and Type | Method and Description |
|---|---|
static <U extends Number,T extends Number> |
CudnnRNNParamsSize.create(Scope scope,
Operand<Integer> numLayers,
Operand<Integer> numUnits,
Operand<Integer> inputSize,
Class<T> T,
Class<U> S,
CudnnRNNParamsSize.Options... options)
Factory method to create a class to wrap a new CudnnRNNParamsSize operation to the graph.
|
Copyright © 2015–2019. All rights reserved.