Searched refs:CudnnRNNV3Attr (Results 1 – 1 of 1) sorted by relevance
5633 type CudnnRNNV3Attr func(optionalAttr) type5637 func CudnnRNNV3RnnMode(value string) CudnnRNNV3Attr {5645 func CudnnRNNV3InputMode(value string) CudnnRNNV3Attr {5653 func CudnnRNNV3Direction(value string) CudnnRNNV3Attr {5661 func CudnnRNNV3Dropout(value float32) CudnnRNNV3Attr {5669 func CudnnRNNV3Seed(value int64) CudnnRNNV3Attr {5677 func CudnnRNNV3Seed2(value int64) CudnnRNNV3Attr {5685 func CudnnRNNV3NumProj(value int64) CudnnRNNV3Attr {5693 func CudnnRNNV3IsTraining(value bool) CudnnRNNV3Attr {5701 func CudnnRNNV3TimeMajor(value bool) CudnnRNNV3Attr {[all …]