| Package | Description |
|---|---|
| org.tensorflow.op | |
| org.tensorflow.op.core |
| Modifier and Type | Method and Description |
|---|---|
<T extends Number> |
Ops.cudnnRNN(Operand<T> input,
Operand<T> inputH,
Operand<T> inputC,
Operand<T> params,
CudnnRNN.Options... options)
Adds an
CudnnRNN operation to the graph |
| Modifier and Type | Method and Description |
|---|---|
static CudnnRNN.Options |
CudnnRNN.direction(String direction) |
CudnnRNN.Options |
CudnnRNN.Options.direction(String direction) |
static CudnnRNN.Options |
CudnnRNN.dropout(Float dropout) |
CudnnRNN.Options |
CudnnRNN.Options.dropout(Float dropout) |
static CudnnRNN.Options |
CudnnRNN.inputMode(String inputMode) |
CudnnRNN.Options |
CudnnRNN.Options.inputMode(String inputMode) |
static CudnnRNN.Options |
CudnnRNN.isTraining(Boolean isTraining) |
CudnnRNN.Options |
CudnnRNN.Options.isTraining(Boolean isTraining) |
static CudnnRNN.Options |
CudnnRNN.rnnMode(String rnnMode) |
CudnnRNN.Options |
CudnnRNN.Options.rnnMode(String rnnMode) |
static CudnnRNN.Options |
CudnnRNN.seed(Long seed) |
CudnnRNN.Options |
CudnnRNN.Options.seed(Long seed) |
static CudnnRNN.Options |
CudnnRNN.seed2(Long seed2) |
CudnnRNN.Options |
CudnnRNN.Options.seed2(Long seed2) |
| Modifier and Type | Method and Description |
|---|---|
static <T extends Number> |
CudnnRNN.create(Scope scope,
Operand<T> input,
Operand<T> inputH,
Operand<T> inputC,
Operand<T> params,
CudnnRNN.Options... options)
Factory method to create a class to wrap a new CudnnRNN operation to the graph.
|
Copyright © 2015–2019. All rights reserved.