Python _GetBatchSizes Examples

Example #1

File: tensorflow_benchmark_test.py Project: moesy/PerfKitBenchmarker

 def testUnknownModelReturnsDefaultBatchSize(self):
     batch_size = tensorflow_benchmark._GetBatchSizes(
         'unknown_model', tensorflow_benchmark.CPU)
     self.assertEqual([64], batch_size)

Example #2

File: tensorflow_benchmark_test.py Project: moesy/PerfKitBenchmarker

 def testFlagOverridesDefaultBatchSize(self):
     batch_size = tensorflow_benchmark._GetBatchSizes(
         'alexnet', tensorflow_benchmark.CPU)
     self.assertEqual([99], batch_size)

Example #3

File: tensorflow_benchmark_test.py Project: moesy/PerfKitBenchmarker

 def testUnknownGpuTypeReturnsDefaultBatchSize(self):
     batch_size = tensorflow_benchmark._GetBatchSizes(
         'alexnet', 'unknown_gpu_type')
     self.assertEqual([64], batch_size)

Example #4

File: tensorflow_benchmark_test.py Project: moesy/PerfKitBenchmarker

 def testP100AlexnetDefault(self):
     batch_size = tensorflow_benchmark._GetBatchSizes(
         'alexnet', tensorflow_benchmark.NVIDIA_TESLA_P100)
     self.assertEqual([512], batch_size)

Example #5

File: tensorflow_benchmark_test.py Project: 9723/PerfKitBenchmarker

 def testFlagOverridesDefaultBatchSize(self):
   batch_size = tensorflow_benchmark._GetBatchSizes(
       'alexnet', tensorflow_benchmark.CPU)
   self.assertEqual([99], batch_size)

Example #6

File: tensorflow_benchmark_test.py Project: moesy/PerfKitBenchmarker

 def testCpuResnet152Default(self):
     batch_size = tensorflow_benchmark._GetBatchSizes(
         'resnet152', tensorflow_benchmark.CPU)
     self.assertEqual([32], batch_size)

Example #7

File: tensorflow_benchmark_test.py Project: moesy/PerfKitBenchmarker

 def testK80Resnet152Default(self):
     batch_size = tensorflow_benchmark._GetBatchSizes(
         'resnet152', tensorflow_benchmark.NVIDIA_TESLA_K80)
     self.assertEqual([32], batch_size)

Example #8

File: tensorflow_benchmark_test.py Project: 9723/PerfKitBenchmarker

 def testCpuVgg16Default(self):
   batch_size = tensorflow_benchmark._GetBatchSizes(
       'vgg16', tensorflow_benchmark.CPU)
   self.assertEqual([32], batch_size)

Example #9

File: tensorflow_benchmark_test.py Project: 9723/PerfKitBenchmarker

 def testK80Inception3Default(self):
   batch_size = tensorflow_benchmark._GetBatchSizes(
       'inception3', tensorflow_benchmark.NVIDIA_TESLA_K80)
   self.assertEqual([64], batch_size)

Example #10

File: tensorflow_benchmark_test.py Project: 9723/PerfKitBenchmarker

 def testCpuInception3Default(self):
   batch_size = tensorflow_benchmark._GetBatchSizes(
       'inception3', tensorflow_benchmark.CPU)
   self.assertEqual([64], batch_size)

Example #11

File: tensorflow_benchmark_test.py Project: 9723/PerfKitBenchmarker

 def testCpuResnet152Default(self):
   batch_size = tensorflow_benchmark._GetBatchSizes(
       'resnet152', tensorflow_benchmark.CPU)
   self.assertEqual([32], batch_size)

Example #12

File: tensorflow_benchmark_test.py Project: 9723/PerfKitBenchmarker

 def testCpuAlexnetDefault(self):
   batch_size = tensorflow_benchmark._GetBatchSizes(
       'alexnet', tensorflow_benchmark.CPU)
   self.assertEqual([512], batch_size)

Example #13

File: tensorflow_benchmark_test.py Project: 9723/PerfKitBenchmarker

 def testUnknownModelReturnsDefaultBatchSize(self):
   batch_size = tensorflow_benchmark._GetBatchSizes(
       'unknown_model', tensorflow_benchmark.CPU)
   self.assertEqual([64], batch_size)

Example #14

File: tensorflow_benchmark_test.py Project: 9723/PerfKitBenchmarker

 def testUnknownGpuTypeReturnsDefaultBatchSize(self):
   batch_size = tensorflow_benchmark._GetBatchSizes(
       'alexnet', 'unknown_gpu_type')
   self.assertEqual([64], batch_size)

Example #15

File: tensorflow_benchmark_test.py Project: moesy/PerfKitBenchmarker

 def testCpuAlexnetDefault(self):
     batch_size = tensorflow_benchmark._GetBatchSizes(
         'alexnet', tensorflow_benchmark.CPU)
     self.assertEqual([512], batch_size)

Example #16

File: tensorflow_benchmark_test.py Project: 9723/PerfKitBenchmarker

 def testK80Resnet152Default(self):
   batch_size = tensorflow_benchmark._GetBatchSizes(
       'resnet152', tensorflow_benchmark.NVIDIA_TESLA_K80)
   self.assertEqual([32], batch_size)

Example #17

File: tensorflow_benchmark_test.py Project: moesy/PerfKitBenchmarker

 def testCpuInception3Default(self):
     batch_size = tensorflow_benchmark._GetBatchSizes(
         'inception3', tensorflow_benchmark.CPU)
     self.assertEqual([64], batch_size)

Example #18

File: tensorflow_benchmark_test.py Project: 9723/PerfKitBenchmarker

 def testK80Vgg16Default(self):
   batch_size = tensorflow_benchmark._GetBatchSizes(
       'vgg16', tensorflow_benchmark.NVIDIA_TESLA_K80)
   self.assertEqual([32], batch_size)

Example #19

File: tensorflow_benchmark_test.py Project: moesy/PerfKitBenchmarker

 def testCpuVgg16Default(self):
     batch_size = tensorflow_benchmark._GetBatchSizes(
         'vgg16', tensorflow_benchmark.CPU)
     self.assertEqual([32], batch_size)

Example #20

File: tensorflow_benchmark_test.py Project: 9723/PerfKitBenchmarker

 def testP100AlexnetDefault(self):
   batch_size = tensorflow_benchmark._GetBatchSizes(
       'alexnet', tensorflow_benchmark.NVIDIA_TESLA_P100)
   self.assertEqual([512], batch_size)

Example #21

File: tensorflow_benchmark_test.py Project: moesy/PerfKitBenchmarker

 def testK80Vgg16Default(self):
     batch_size = tensorflow_benchmark._GetBatchSizes(
         'vgg16', tensorflow_benchmark.NVIDIA_TESLA_K80)
     self.assertEqual([32], batch_size)

Example #22

File: tensorflow_benchmark_test.py Project: moesy/PerfKitBenchmarker

 def testV100Resnet50Default(self):
     batch_size = tensorflow_benchmark._GetBatchSizes(
         'resnet50', tensorflow_benchmark.NVIDIA_TESLA_V100)
     self.assertEqual([256], batch_size)

Example #23

File: tensorflow_benchmark_test.py Project: moesy/PerfKitBenchmarker

 def testP100Inception3Default(self):
     batch_size = tensorflow_benchmark._GetBatchSizes(
         'inception3', tensorflow_benchmark.NVIDIA_TESLA_P100)
     self.assertEqual([256], batch_size)

Example #24

File: tensorflow_benchmark_test.py Project: 9723/PerfKitBenchmarker

 def testV100Resnet50Default(self):
   batch_size = tensorflow_benchmark._GetBatchSizes(
       'resnet50', tensorflow_benchmark.NVIDIA_TESLA_V100)
   self.assertEqual([256], batch_size)