def benchmark_cudnn_train_large(self): if not tf.test.is_gpu_available(): return self._benchmark_train("graph_cudnn_train_large", rnn_ptb.large_model(True))
def benchmark_apply_large(self): self._benchmark_apply("graph_apply_large", rnn_ptb.large_model(False))
def benchmark_train_large(self): self._benchmark_train("graph_train_large", rnn_ptb.large_model(False))
def benchmark_cudnn_train_large(self): if not tfe.num_gpus(): return self._benchmark_train("eager_cudnn_train_large", rnn_ptb.large_model(True))