Searched refs:CudnnRNNV2Attr (Results 1 – 1 of 1) sorted by relevance
4932 type CudnnRNNV2Attr func(optionalAttr) type4936 func CudnnRNNV2RnnMode(value string) CudnnRNNV2Attr {4944 func CudnnRNNV2InputMode(value string) CudnnRNNV2Attr {4952 func CudnnRNNV2Direction(value string) CudnnRNNV2Attr {4960 func CudnnRNNV2Dropout(value float32) CudnnRNNV2Attr {4968 func CudnnRNNV2Seed(value int64) CudnnRNNV2Attr {4976 func CudnnRNNV2Seed2(value int64) CudnnRNNV2Attr {4984 func CudnnRNNV2IsTraining(value bool) CudnnRNNV2Attr {5025 … input_h tf.Output, input_c tf.Output, params tf.Output, optional ...CudnnRNNV2Attr) (output tf.Ou… argument