org.tensorflow.op.core

Class CudnnRNNBackprop.Options

java.lang.Object

org.tensorflow.op.core.CudnnRNNBackprop.Options

Enclosing class:

CudnnRNNBackprop<T extends Number>

public static class CudnnRNNBackprop.Options
extends Object

Optional attributes for CudnnRNNBackprop

Method Summary

Modifier and Type	Method and Description
CudnnRNNBackprop.Options	direction(String direction)
CudnnRNNBackprop.Options	dropout(Float dropout)
CudnnRNNBackprop.Options	inputMode(String inputMode)
CudnnRNNBackprop.Options	rnnMode(String rnnMode)
CudnnRNNBackprop.Options	seed(Long seed)
CudnnRNNBackprop.Options	seed2(Long seed2)

Methods inherited from class java.lang.Object

clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

Method Detail

rnnMode

public CudnnRNNBackprop.Options rnnMode(String rnnMode)

Parameters:

rnnMode -

inputMode

public CudnnRNNBackprop.Options inputMode(String inputMode)

Parameters:

inputMode -

direction

public CudnnRNNBackprop.Options direction(String direction)

Parameters:

direction -

dropout

public CudnnRNNBackprop.Options dropout(Float dropout)

Parameters:

dropout -

seed

public CudnnRNNBackprop.Options seed(Long seed)

Parameters:

seed -

seed2

public CudnnRNNBackprop.Options seed2(Long seed2)

Parameters:

seed2 -