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

  /external/tensorflow/tensorflow/compiler/tests/
test_utils.py 25 def ConvertBetweenDataFormats(x, data_format_src, data_format_dst):
29 if data_format_src not in valid_data_formats:
30 raise ValueError("data_format_src must be of %s, got %s." %
31 (valid_data_formats, data_format_src))
38 if data_format_src == data_format_dst:
41 dim_map = {d: i for i, d in enumerate(data_format_src)}
46 def PermuteDimsBetweenDataFormats(dims, data_format_src, data_format_dst):
50 if data_format_src not in valid_data_formats:
51 raise ValueError("data_format_src must be of %s, got %s." %
52 (valid_data_formats, data_format_src))
    [all...]
conv2d_test.py 50 data_format_src="NHWC",
63 data_format_src: Data format input is in.
79 expected = test_utils.ConvertBetweenDataFormats(expected, data_format_src,
81 x1 = test_utils.ConvertBetweenDataFormats(x1, data_format_src,
84 input_sizes, data_format_src, data_format_dst)
85 strides = test_utils.PermuteDimsBetweenDataFormats(strides, data_format_src,
88 dilations, data_format_src, data_format_dst)
116 data_format_src="NHWC",
129 data_format_src="NHWC",
142 data_format_src="NHWC"
    [all...]
fused_batchnorm_test.py 82 data_format_src = "NHWC"
84 x_val, scale_val, offset_val, epsilon, data_format_src)
89 x_val, data_format_src, data_format)
91 y_ref, data_format_src, data_format)
125 data_format_src = "NHWC"
127 x_val, scale_val, offset_val, epsilon, data_format_src)
132 x_val, data_format_src, data_format)
134 y_ref, data_format_src, data_format)
208 data_format_src = "NHWC"
210 x_val, grad_val, scale_val, mean_val, var_val, epsilon, data_format_src)
    [all...]

Completed in 421 milliseconds