org.tensorflow.op.core

Class CudnnRNNV3.Options

java.lang.Object

org.tensorflow.op.core.CudnnRNNV3.Options

Enclosing class:

CudnnRNNV3<T extends Number>

public static class CudnnRNNV3.Options
extends Object

Optional attributes for CudnnRNNV3

Method Summary

Modifier and Type	Method and Description
CudnnRNNV3.Options	direction(String direction)
CudnnRNNV3.Options	dropout(Float dropout)
CudnnRNNV3.Options	inputMode(String inputMode)
CudnnRNNV3.Options	isTraining(Boolean isTraining)
CudnnRNNV3.Options	numProj(Long numProj)
CudnnRNNV3.Options	rnnMode(String rnnMode)
CudnnRNNV3.Options	seed(Long seed)
CudnnRNNV3.Options	seed2(Long seed2)
CudnnRNNV3.Options	timeMajor(Boolean timeMajor)

Methods inherited from class java.lang.Object

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

Method Detail

rnnMode

public CudnnRNNV3.Options rnnMode(String rnnMode)

Parameters:

rnnMode -

inputMode

public CudnnRNNV3.Options inputMode(String inputMode)

Parameters:

inputMode -

direction

public CudnnRNNV3.Options direction(String direction)

Parameters:

direction -

dropout

public CudnnRNNV3.Options dropout(Float dropout)

Parameters:

dropout -

seed

public CudnnRNNV3.Options seed(Long seed)

Parameters:

seed -

seed2

public CudnnRNNV3.Options seed2(Long seed2)

Parameters:

seed2 -

numProj

public CudnnRNNV3.Options numProj(Long numProj)

Parameters:

numProj -

isTraining

public CudnnRNNV3.Options isTraining(Boolean isTraining)

Parameters:

isTraining -

timeMajor

public CudnnRNNV3.Options timeMajor(Boolean timeMajor)

Parameters:

timeMajor -