Exemplo n.º 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))
Exemplo n.º 2
0
 def benchmark_apply_large(self):
     self._benchmark_apply("graph_apply_large", rnn_ptb.large_model(False))
Exemplo n.º 3
0
 def benchmark_train_large(self):
     self._benchmark_train("graph_train_large", rnn_ptb.large_model(False))
Exemplo n.º 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))
Exemplo n.º 5
0
 def benchmark_train_large(self):
   self._benchmark_train("graph_train_large", rnn_ptb.large_model(False))
Exemplo n.º 6
0
 def benchmark_apply_large(self):
   self._benchmark_apply("graph_apply_large", rnn_ptb.large_model(False))
Exemplo n.º 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))
Exemplo n.º 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))