def testUnknownModelReturnsDefaultBatchSize(self):
batch_size = tensorflow_benchmark._GetBatchSizes(
'unknown_model', tensorflow_benchmark.CPU)
self.assertEqual([64], batch_size)
def testFlagOverridesDefaultBatchSize(self):
batch_size = tensorflow_benchmark._GetBatchSizes(
'alexnet', tensorflow_benchmark.CPU)
self.assertEqual([99], batch_size)
def testUnknownGpuTypeReturnsDefaultBatchSize(self):
batch_size = tensorflow_benchmark._GetBatchSizes(
'alexnet', 'unknown_gpu_type')
self.assertEqual([64], batch_size)
def testP100AlexnetDefault(self):
batch_size = tensorflow_benchmark._GetBatchSizes(
'alexnet', tensorflow_benchmark.NVIDIA_TESLA_P100)
self.assertEqual([512], batch_size)
def testFlagOverridesDefaultBatchSize(self):
batch_size = tensorflow_benchmark._GetBatchSizes(
'alexnet', tensorflow_benchmark.CPU)
self.assertEqual([99], batch_size)
def testCpuResnet152Default(self):
batch_size = tensorflow_benchmark._GetBatchSizes(
'resnet152', tensorflow_benchmark.CPU)
self.assertEqual([32], batch_size)
def testK80Resnet152Default(self):
batch_size = tensorflow_benchmark._GetBatchSizes(
'resnet152', tensorflow_benchmark.NVIDIA_TESLA_K80)
self.assertEqual([32], batch_size)
def testCpuVgg16Default(self):
batch_size = tensorflow_benchmark._GetBatchSizes(
'vgg16', tensorflow_benchmark.CPU)
self.assertEqual([32], batch_size)
def testK80Inception3Default(self):
batch_size = tensorflow_benchmark._GetBatchSizes(
'inception3', tensorflow_benchmark.NVIDIA_TESLA_K80)
self.assertEqual([64], batch_size)
def testCpuInception3Default(self):
batch_size = tensorflow_benchmark._GetBatchSizes(
'inception3', tensorflow_benchmark.CPU)
self.assertEqual([64], batch_size)
def testCpuResnet152Default(self):
batch_size = tensorflow_benchmark._GetBatchSizes(
'resnet152', tensorflow_benchmark.CPU)
self.assertEqual([32], batch_size)
def testCpuAlexnetDefault(self):
batch_size = tensorflow_benchmark._GetBatchSizes(
'alexnet', tensorflow_benchmark.CPU)
self.assertEqual([512], batch_size)
def testUnknownModelReturnsDefaultBatchSize(self):
batch_size = tensorflow_benchmark._GetBatchSizes(
'unknown_model', tensorflow_benchmark.CPU)
self.assertEqual([64], batch_size)
def testUnknownGpuTypeReturnsDefaultBatchSize(self):
batch_size = tensorflow_benchmark._GetBatchSizes(
'alexnet', 'unknown_gpu_type')
self.assertEqual([64], batch_size)
def testCpuAlexnetDefault(self):
batch_size = tensorflow_benchmark._GetBatchSizes(
'alexnet', tensorflow_benchmark.CPU)
self.assertEqual([512], batch_size)
def testK80Resnet152Default(self):
batch_size = tensorflow_benchmark._GetBatchSizes(
'resnet152', tensorflow_benchmark.NVIDIA_TESLA_K80)
self.assertEqual([32], batch_size)
def testCpuInception3Default(self):
batch_size = tensorflow_benchmark._GetBatchSizes(
'inception3', tensorflow_benchmark.CPU)
self.assertEqual([64], batch_size)
def testK80Vgg16Default(self):
batch_size = tensorflow_benchmark._GetBatchSizes(
'vgg16', tensorflow_benchmark.NVIDIA_TESLA_K80)
self.assertEqual([32], batch_size)
def testCpuVgg16Default(self):
batch_size = tensorflow_benchmark._GetBatchSizes(
'vgg16', tensorflow_benchmark.CPU)
self.assertEqual([32], batch_size)
def testP100AlexnetDefault(self):
batch_size = tensorflow_benchmark._GetBatchSizes(
'alexnet', tensorflow_benchmark.NVIDIA_TESLA_P100)
self.assertEqual([512], batch_size)
def testK80Vgg16Default(self):
batch_size = tensorflow_benchmark._GetBatchSizes(
'vgg16', tensorflow_benchmark.NVIDIA_TESLA_K80)
self.assertEqual([32], batch_size)
def testV100Resnet50Default(self):
batch_size = tensorflow_benchmark._GetBatchSizes(
'resnet50', tensorflow_benchmark.NVIDIA_TESLA_V100)
self.assertEqual([256], batch_size)
def testP100Inception3Default(self):
batch_size = tensorflow_benchmark._GetBatchSizes(
'inception3', tensorflow_benchmark.NVIDIA_TESLA_P100)
self.assertEqual([256], batch_size)
def testV100Resnet50Default(self):
batch_size = tensorflow_benchmark._GetBatchSizes(
'resnet50', tensorflow_benchmark.NVIDIA_TESLA_V100)
self.assertEqual([256], batch_size)
