Lines Matching refs:directions
657 directions = [CUDNN_RNN_UNIDIRECTION, CUDNN_RNN_BIDIRECTION]
659 for direction, dtype in itertools.product(directions, dtype_list):
759 directions = [CUDNN_RNN_UNIDIRECTION, CUDNN_RNN_BIDIRECTION]
760 for cfg, direction in zip(configs, directions):
923 directions = [CUDNN_RNN_UNIDIRECTION, CUDNN_RNN_BIDIRECTION]
927 rnns, test_configs, dtype_list, directions):