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)