Beispiel #1
0
 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))
Beispiel #2
0
 def benchmark_apply_large(self):
     self._benchmark_apply("graph_apply_large", rnn_ptb.large_model(False))
Beispiel #3
0
 def benchmark_train_large(self):
     self._benchmark_train("graph_train_large", rnn_ptb.large_model(False))
 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_train_large(self):
   self._benchmark_train("graph_train_large", rnn_ptb.large_model(False))
 def benchmark_apply_large(self):
   self._benchmark_apply("graph_apply_large", rnn_ptb.large_model(False))
Beispiel #7
0
 def benchmark_cudnn_train_large(self):
     if not tfe.num_gpus():
         return
     self._benchmark_train("eager_cudnn_train_large",
                           rnn_ptb.large_model(True))
 def benchmark_cudnn_train_large(self):
   if not tfe.num_gpus():
     return
   self._benchmark_train("eager_cudnn_train_large", rnn_ptb.large_model(True))