org.tensorflow.op.core

Class CudnnRNNV2.Options

java.lang.Object

org.tensorflow.op.core.CudnnRNNV2.Options

Enclosing class:

CudnnRNNV2<T extends Number>

public static class CudnnRNNV2.Options
extends Object

Optional attributes for CudnnRNNV2

Method Summary

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

Parameters:

rnnMode -

inputMode

public CudnnRNNV2.Options inputMode(String inputMode)

Parameters:

inputMode -

direction

public CudnnRNNV2.Options direction(String direction)

Parameters:

direction -

dropout

public CudnnRNNV2.Options dropout(Float dropout)

Parameters:

dropout -

seed

public CudnnRNNV2.Options seed(Long seed)

Parameters:

seed -

seed2

public CudnnRNNV2.Options seed2(Long seed2)

Parameters:

seed2 -

isTraining

public CudnnRNNV2.Options isTraining(Boolean isTraining)

Parameters:

isTraining -