コード例 #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))
コード例 #2
0
 def benchmark_apply_large(self):
     self._benchmark_apply("graph_apply_large", rnn_ptb.large_model(False))
コード例 #3
0
 def benchmark_train_large(self):
     self._benchmark_train("graph_train_large", rnn_ptb.large_model(False))
コード例 #4
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))
コード例 #5
0
 def benchmark_train_large(self):
   self._benchmark_train("graph_train_large", rnn_ptb.large_model(False))
コード例 #6
0
 def benchmark_apply_large(self):
   self._benchmark_apply("graph_apply_large", rnn_ptb.large_model(False))
コード例 #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))
コード例 #8
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))