def test_reduce_mean(test_case):
    arg_dict = OrderedDict()
    arg_dict['device_type'] = ['gpu']
    arg_dict['input_shape'] = [(64, 64, 64)]
    arg_dict['axis'] = [None, [1], [0, 2]]
    arg_dict['keepdims'] = [True, False]
    for arg in GenArgList(arg_dict):
        compare_with_tensorflow(*arg)
def test_reduce_mean(test_case):
    arg_dict = OrderedDict()
    arg_dict['device_type'] = ['gpu']
    arg_dict['input_shape'] = [(64, 64, 64)]
    arg_dict['axis'] = [None, [1], [0, 2]]
    arg_dict['keepdims'] = [True, False]
    for arg in GenArgList(arg_dict):
        compare_with_tensorflow(*arg)