org.tensorflow.op.core

Class CudnnRNN.Options

java.lang.Object

org.tensorflow.op.core.CudnnRNN.Options

Enclosing class:

CudnnRNN<T extends Number>

public static class CudnnRNN.Options
extends Object

Optional attributes for CudnnRNN

Method Summary

Modifier and Type	Method and Description
CudnnRNN.Options	direction(String direction)
CudnnRNN.Options	dropout(Float dropout)
CudnnRNN.Options	inputMode(String inputMode)
CudnnRNN.Options	isTraining(Boolean isTraining)
CudnnRNN.Options	rnnMode(String rnnMode)
CudnnRNN.Options	seed(Long seed)
CudnnRNN.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 CudnnRNN.Options rnnMode(String rnnMode)

Parameters:

rnnMode -

inputMode

public CudnnRNN.Options inputMode(String inputMode)

Parameters:

inputMode -

direction

public CudnnRNN.Options direction(String direction)

Parameters:

direction -

dropout

public CudnnRNN.Options dropout(Float dropout)

Parameters:

dropout -

seed

public CudnnRNN.Options seed(Long seed)

Parameters:

seed -

seed2

public CudnnRNN.Options seed2(Long seed2)

Parameters:

seed2 -

isTraining

public CudnnRNN.Options isTraining(Boolean isTraining)

Parameters:

isTraining -