Home
last modified time | relevance | path

Searched refs:ConvertBetweenDataFormats (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/tests/
H A Dfused_batchnorm_test.py102 x_val_converted = test_utils.ConvertBetweenDataFormats(
104 y_ref_converted = test_utils.ConvertBetweenDataFormats(
149 x_val_converted = test_utils.ConvertBetweenDataFormats(
151 y_ref_converted = test_utils.ConvertBetweenDataFormats(
253 grad_val_converted = test_utils.ConvertBetweenDataFormats(
255 x_val_converted = test_utils.ConvertBetweenDataFormats(
257 grad_x_ref_converted = test_utils.ConvertBetweenDataFormats(
306 grad_val_converted = test_utils.ConvertBetweenDataFormats(
308 x_val_converted = test_utils.ConvertBetweenDataFormats(
H A Dconv2d_test.py75 expected = test_utils.ConvertBetweenDataFormats(expected, data_format_src,
77 x1 = test_utils.ConvertBetweenDataFormats(x1, data_format_src,
273 expected = test_utils.ConvertBetweenDataFormats(expected, data_format_src,
275 x2 = test_utils.ConvertBetweenDataFormats(x2, data_format_src,
587 x1 = test_utils.ConvertBetweenDataFormats(x1, data_format_src,
589 x2 = test_utils.ConvertBetweenDataFormats(x2, data_format_src,
H A Dtest_utils.py21 def ConvertBetweenDataFormats(x, data_format_src, data_format_dst): function