Function GetTestConfigs

tensorflow/python/kernel_tests/conv_ops_test.py:151–161

Get all the valid tests configs to run. Returns: all the valid test configs as tuples of data_format and use_gpu.

()

def GetTestConfigs():
  """Get all the valid tests configs to run.

  Returns:
    all the valid test configs as tuples of data_format and use_gpu.
  """
  test_configs = [("NHWC", False), ("NHWC", True)]
  if test.is_gpu_available(cuda_only=True):
    # "NCHW" format is only supported on CUDA.
    test_configs += [("NCHW", True)]
  return test_configs


class Conv2DTest(test.TestCase):