TensorRT  7.2.1.6
NVIDIA TensorRT
Looking for a C++ dev who knows TensorRT?
I'm looking for work. Hire me!
All Classes Namespaces Functions Variables Typedefs Enumerations Enumerator Friends Pages
util.test_misc Namespace Reference

Classes

class  FindInDictCase
 

Functions

def test_volume (case)
 
def test_find_in_dict (case)
 
def test_is_valid_shape_override (case)
 
def test_shape_matching (case)
 
def test_unpack_args (case)
 
def test_unique_list (case)
 

Variables

list VOLUME_CASES
 
list FIND_IN_DICT_CASES
 
list SHAPE_OVERRIDE_CASES
 
list SHAPE_MATCHING_CASES
 
list UNPACK_ARGS_CASES
 
list UNIQUE_LIST_CASES
 

Function Documentation

◆ test_volume()

def util.test_misc.test_volume (   case)

◆ test_find_in_dict()

def util.test_misc.test_find_in_dict (   case)

◆ test_is_valid_shape_override()

def util.test_misc.test_is_valid_shape_override (   case)

◆ test_shape_matching()

def util.test_misc.test_shape_matching (   case)

◆ test_unpack_args()

def util.test_misc.test_unpack_args (   case)

◆ test_unique_list()

def util.test_misc.test_unique_list (   case)

Variable Documentation

◆ VOLUME_CASES

list util.test_misc.VOLUME_CASES
Initial value:
1 = [
2  ((1, 1, 1), 1),
3  ((2, 3, 4), 24),
4  (tuple(), 1),
5 ]

◆ FIND_IN_DICT_CASES

list util.test_misc.FIND_IN_DICT_CASES
Initial value:
1 = [
2  FindInDictCase("resnet50_v1.5/output/Softmax:0", map={"resnet50_v1.5/output/Softmax:0": "x"}, index=None, expected="resnet50_v1.5/output/Softmax:0"),
3  FindInDictCase("resnet50_v1.5/output/Softmax:0", map={"resnet50_v1.5/output/softmax:0": "x"}, index=None, expected="resnet50_v1.5/output/softmax:0"),
4 ]

◆ SHAPE_OVERRIDE_CASES

list util.test_misc.SHAPE_OVERRIDE_CASES
Initial value:
1 = [
2  ((1, 3, 224, 224), (None, 3, 224, 224), True),
3 ]

◆ SHAPE_MATCHING_CASES

list util.test_misc.SHAPE_MATCHING_CASES
Initial value:
1 = [
2  (np.zeros((1, 1, 3, 3)), (3, 3), (3, 3)), # Squeeze array shape
3  (np.zeros((1, 3, 3, 1)), (1, 1, 3, 3), (1, 1, 3, 3)), # Permute
4  (np.zeros((3, 3)), (1, 1, 3, 3), (3, 3)), # Squeeze specified shape
5  (np.zeros((3, 3)), (-1, 3), (3, 3)), # Infer dynamic
6  (np.zeros((3 * 224 * 224)), (None, 3, 224, 224), (1, 3, 224, 224)), # Reshape and Permute
7  (np.zeros((1, 3, 224, 224)), (None, 224, 224, 3), (1, 224, 224, 3)), # Permute
8 ]

◆ UNPACK_ARGS_CASES

list util.test_misc.UNPACK_ARGS_CASES
Initial value:
1 = [
2  ((0, 1, 2), 3, (0, 1, 2)), # no extras
3  ((0, 1, 2), 4, (0, 1, 2, None)), # 1 extra
4  ((0, 1, 2), 2, (0, 1)), # 1 fewer
5 ]

◆ UNIQUE_LIST_CASES

list util.test_misc.UNIQUE_LIST_CASES
Initial value:
1 = [
2  ([], []),
3  ([3, 1, 2], [3, 1, 2]),
4  ([1, 2, 3, 2, 1], [1, 2, 3]),
5  ([0, 0, 0, 0, 1, 0, 0], [0, 1]),
6  ([5, 5, 5, 5, 5], [5]),
7 ]