Пример #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))