HomeSort by relevance Sort by last modified time
    Searched refs:data_format_dst (Results 1 - 2 of 2) sorted by null

  /external/tensorflow/tensorflow/compiler/tests/
test_utils.py 25 def ConvertBetweenDataFormats(x, data_format_src, data_format_dst):
32 if data_format_dst not in valid_data_formats:
33 raise ValueError("data_format_dst must be of %s, got %s." %
34 (valid_data_formats, data_format_dst))
38 if data_format_src == data_format_dst:
42 transpose_dims = [dim_map[d] for d in data_format_dst]
46 def PermuteDimsBetweenDataFormats(dims, data_format_src, data_format_dst):
53 if data_format_dst not in valid_data_formats:
54 raise ValueError("data_format_dst must be of %s, got %s." %
55 (valid_data_formats, data_format_dst))
    [all...]
conv2d_test.py 51 data_format_dst="NHWC",
64 data_format_dst: Data format verification will run and input is converted
80 data_format_dst)
82 data_format_dst)
84 input_sizes, data_format_src, data_format_dst)
86 data_format_dst)
88 dilations, data_format_src, data_format_dst)
99 data_format=data_format_dst,
117 data_format_dst=data_format,
130 data_format_dst=data_format
    [all...]

Completed in 415 milliseconds