Home
last modified time | relevance | path

Searched refs:AddInputFromArray (Results 1 – 25 of 92) sorted by relevance

1234

/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/image/
H A Dnon_max_suppression_op_gpu_test.cc64 AddInputFromArray<float>( in TEST_F()
68 AddInputFromArray<float>(TensorShape({6}), {.9f, .75f, .6f, .95f, .5f, .3f}); in TEST_F()
69 AddInputFromArray<int>(TensorShape({}), {3}); in TEST_F()
70 AddInputFromArray<float>(TensorShape({}), {.5f}); in TEST_F()
80 AddInputFromArray<float>( in TEST_F()
84 AddInputFromArray<float>(TensorShape({6}), {.9f, .75f, .6f, .95f, .5f, .3f}); in TEST_F()
85 AddInputFromArray<int>(TensorShape({}), {3}); in TEST_F()
86 AddInputFromArray<Eigen::half>(TensorShape({}), {Eigen::half(0.5)}); in TEST_F()
97 AddInputFromArray<float>(TensorShape({6, 4}), in TEST_F()
105 AddInputFromArray<float>(TensorShape({6}), {.9f, .75f, .6f, .95f, .5f, .3f}); in TEST_F()
[all …]
H A Dnon_max_suppression_op_test.cc47 AddInputFromArray<float>( in TEST_F()
51 AddInputFromArray<float>(TensorShape({6}), {.9f, .75f, .6f, .95f, .5f, .3f}); in TEST_F()
52 AddInputFromArray<int>(TensorShape({}), {3}); in TEST_F()
62 AddInputFromArray<float>(TensorShape({6, 4}), in TEST_F()
65 AddInputFromArray<float>(TensorShape({6}), {.9f, .75f, .6f, .95f, .5f, .3f}); in TEST_F()
66 AddInputFromArray<int>(TensorShape({}), {3}); in TEST_F()
76 AddInputFromArray<float>( in TEST_F()
80 AddInputFromArray<float>(TensorShape({6}), {.9f, .75f, .6f, .95f, .5f, .3f}); in TEST_F()
81 AddInputFromArray<int>(TensorShape({}), {2}); in TEST_F()
91 AddInputFromArray<float>( in TEST_F()
[all …]
H A Dcrop_and_resize_op_test.cc53 AddInputFromArray<T>(TensorShape({1, 2, 2, 1}), {1, 2, 3, 4}); \
54 AddInputFromArray<float>(TensorShape({1, 4}), {0, 0, 1, 1}); \
55 AddInputFromArray<int32>(TensorShape({1}), {0}); \
56 AddInputFromArray<int32>(TensorShape({2}), {1, 1}); \
66 AddInputFromArray<T>(TensorShape({1, 2, 2, 1}), {1, 2, 3, 4}); \
67 AddInputFromArray<float>(TensorShape({1, 4}), {0, 0, 1, 1}); \
68 AddInputFromArray<int32>(TensorShape({1}), {0}); \
69 AddInputFromArray<int32>(TensorShape({2}), {1, 1}); \
93 AddInputFromArray<uint8>(TensorShape({1, 2, 2, 1}), {1, 2, 3, 4}); in REGISTER_TEST()
94 AddInputFromArray<float>(TensorShape({1, 4}), {0, 0, 1, 1}); in REGISTER_TEST()
[all …]
H A Dresize_nearest_neighbor_op_test.cc103 AddInputFromArray<float>(TensorShape({1, 2, 2, 1}), {1, 2, 3, 4}); in TEST_P()
104 AddInputFromArray<int32>(TensorShape({2}), {1, 1}); in TEST_P()
121 AddInputFromArray<float>(TensorShape({1, 2, 2, 1}), {1, 2, 3, 4}); in TEST_P()
122 AddInputFromArray<int32>(TensorShape({2}), {1, 1}); in TEST_P()
138 AddInputFromArray<float>(TensorShape({1, 2, 2, 1}), {1, 2, 3, 4}); in TEST_P()
139 AddInputFromArray<int32>(TensorShape({2}), {3, 3}); in TEST_P()
159 AddInputFromArray<float>(TensorShape({1, 2, 2, 1}), {1, 2, 3, 4}); in TEST_P()
160 AddInputFromArray<int32>(TensorShape({2}), {3, 3}); in TEST_P()
180 AddInputFromArray<float>(TensorShape({1, 3, 3, 1}), in TEST_P()
182 AddInputFromArray<int32>(TensorShape({2}), {2, 2}); in TEST_P()
[all …]
H A Dresize_bilinear_op_test.cc140 AddInputFromArray<int32>(TensorShape({2}), {output_width, output_height}); in TestResize()
201 AddInputFromArray<float>(TensorShape({1, 2, 2, 1}), {1, 2, 3, 4}); in TEST_P()
202 AddInputFromArray<int32>(TensorShape({2}), {1, 1}); in TEST_P()
214 AddInputFromArray<int32>(TensorShape({2}), {1, 1}); in TEST_P()
232 AddInputFromArray<float>(TensorShape({1, 2, 2, 1}), {1, 2, 3, 4}); in TEST_P()
233 AddInputFromArray<int32>(TensorShape({2}), {1, 1}); in TEST_P()
247 AddInputFromArray<float>(TensorShape({1, 2, 2, 1}), {1, 2, 3, 4}); in TEST_P()
248 AddInputFromArray<int32>(TensorShape({2}), {3, 3}); in TEST_P()
267 AddInputFromArray<float>(TensorShape({1, 2, 2, 1}), {1, 2, 3, 4}); in TEST_P()
268 AddInputFromArray<int32>(TensorShape({2}), {3, 3}); in TEST_P()
[all …]
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/mkl/
H A Dmkl_qmatmul_op_test.cc59 AddInputFromArray<quint8>(TensorShape({2, 3}), {1, 2, 3, 4, 5, 6}); in TEST_F()
64 AddInputFromArray<qint8>(TensorShape({3, 4}), in TEST_F()
66 AddInputFromArray<qint32>(TensorShape({4}), {1, 2, 3, 4}); in TEST_F()
67 AddInputFromArray<float>(TensorShape({1}), {0}); in TEST_F()
68 AddInputFromArray<float>(TensorShape({1}), {255.0f}); in TEST_F()
69 AddInputFromArray<float>(TensorShape({1}), {-127.0f}); in TEST_F()
70 AddInputFromArray<float>(TensorShape({1}), {127.0f}); in TEST_F()
111 AddInputFromArray<quint8>(TensorShape({2, 3}), {1, 2, 3, 4, 5, 6}); in TEST_F()
116 AddInputFromArray<qint8>(TensorShape({3, 4}), in TEST_F()
118 AddInputFromArray<qint32>(TensorShape({4}), {100, -200, 300, -400}); in TEST_F()
[all …]
H A Dmkl_quantized_conv_ops_test.cc68 AddInputFromArray<quint8>( in RunQuantizedDepthwiseConv2DOp()
78 AddInputFromArray<qint8>( in RunQuantizedDepthwiseConv2DOp()
84 AddInputFromArray<float>(TensorShape({depth}), {1.0f, 1.0f}); in RunQuantizedDepthwiseConv2DOp()
88 AddInputFromArray<float>(TensorShape({1}), {0.0f}); in RunQuantizedDepthwiseConv2DOp()
89 AddInputFromArray<float>(TensorShape({1}), {255.0f}); in RunQuantizedDepthwiseConv2DOp()
92 AddInputFromArray<float>(TensorShape({1}), {-127.0f}); in RunQuantizedDepthwiseConv2DOp()
93 AddInputFromArray<float>(TensorShape({1}), {127.0f}); in RunQuantizedDepthwiseConv2DOp()
168 AddInputFromArray<quint8>(image_quantized.shape(), in TEST_F()
170 AddInputFromArray<qint8>(filter_quantized.shape(), in TEST_F()
172 AddInputFromArray<float>(TensorShape({1}), {image_min}); in TEST_F()
[all …]
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Droll_op_test.cc57 AddInputFromArray<float>(TensorShape({5}), {0, 1, 2, 3, 4}); in TEST_F()
58 AddInputFromArray<int32>(TensorShape({}), {3}); in TEST_F()
59 AddInputFromArray<int32>(TensorShape({}), {0}); in TEST_F()
72 AddInputFromArray<tstring>(TensorShape({5}), {"a", "b", "c", "d", "e"}); in TEST_F()
73 AddInputFromArray<int32>(TensorShape({}), {3}); in TEST_F()
74 AddInputFromArray<int32>(TensorShape({}), {0}); in TEST_F()
87 AddInputFromArray<std::complex<float>>( in TEST_F()
91 AddInputFromArray<int32>(TensorShape({}), {3}); in TEST_F()
92 AddInputFromArray<int32>(TensorShape({}), {0}); in TEST_F()
108 AddInputFromArray<float>(TensorShape({3, 5}), in TEST_F()
[all …]
H A Dragged_range_op_test.cc53 AddInputFromArray<int>(TensorShape({4}), {0, 5, 8, 5}); // starts in TEST_F()
54 AddInputFromArray<int>(TensorShape({4}), {8, 7, 8, 1}); // limits in TEST_F()
55 AddInputFromArray<int>(TensorShape({4}), {2, 1, 1, -1}); // deltas in TEST_F()
68 AddInputFromArray<float>(TensorShape({4}), {0, 5, 8, 5}); // starts in TEST_F()
69 AddInputFromArray<float>(TensorShape({4}), {8, 7, 8, 1}); // limits in TEST_F()
70 AddInputFromArray<float>(TensorShape({4}), {2, 1, 1, -1}); // deltas in TEST_F()
83 AddInputFromArray<float>(TensorShape({2}), {1.1, 0.1}); // starts in TEST_F()
84 AddInputFromArray<float>(TensorShape({2}), {10.0, 1e10}); // limits in TEST_F()
85 AddInputFromArray<float>(TensorShape({2}), {1, 1e-10}); // deltas in TEST_F()
94 AddInputFromArray<int>(TensorShape({3}), {0, 5, 8}); // starts in TEST_F()
[all …]
H A Dstring_ngrams_op_test.cc77 AddInputFromArray<tstring>(TensorShape({6}), {"a", "b", "c", "d", "e", "f"}); in TEST_F()
78 AddInputFromArray<int64_t>(TensorShape({3}), {0, 4, 6}); in TEST_F()
95 AddInputFromArray<tstring>(TensorShape({6}), {"a", "b", "c", "d", "e", "f"}); in TEST_F()
96 AddInputFromArray<int64_t>(TensorShape({3}), {0, 4, 6}); in TEST_F()
114 AddInputFromArray<tstring>(TensorShape({6}), {"a", "b", "c", "d", "e", "f"}); in TEST_F()
115 AddInputFromArray<int64_t>(TensorShape({3}), {0, 4, 6}); in TEST_F()
132 AddInputFromArray<tstring>(TensorShape({6}), {"a", "b", "c", "d", "e", "f"}); in TEST_F()
133 AddInputFromArray<int64_t>(TensorShape({3}), {0, 4, 6}); in TEST_F()
150 AddInputFromArray<tstring>(TensorShape({6}), {"a", "b", "c", "d", "e", "f"}); in TEST_F()
151 AddInputFromArray<int64_t>(TensorShape({3}), {0, 4, 6}); in TEST_F()
[all …]
H A Dquantize_and_dequantize_op_test.cc57 AddInputFromArray<float>(TensorShape({1}), {-3.5}); in TEST_F()
58 AddInputFromArray<float>(TensorShape({}), {0.0}); // Min in TEST_F()
59 AddInputFromArray<float>(TensorShape({}), {0.0}); // Max in TEST_F()
82 AddInputFromArray<float>(TensorShape({1}), {-3.5}); in TEST_F()
83 AddInputFromArray<float>(TensorShape({}), {0.0}); // Min in TEST_F()
84 AddInputFromArray<float>(TensorShape({}), {0.0}); // Max in TEST_F()
85 AddInputFromArray<int32>(TensorShape({}), {8}); // num_bits in TEST_F()
137 AddInputFromArray<float>( in TEST_P()
146 AddInputFromArray<float>(range_shape, init_value); // Min in TEST_P()
147 AddInputFromArray<float>(range_shape, init_value); // Max in TEST_P()
[all …]
H A Dquantized_matmul_op_test.cc58 AddInputFromArray<quint8>(TensorShape({2, 3}), {1, 2, 3, 4, 5, 6}); in TEST_F()
63 AddInputFromArray<quint8>(TensorShape({3, 4}), in TEST_F()
65 AddInputFromArray<float>(TensorShape({}), {0}); in TEST_F()
66 AddInputFromArray<float>(TensorShape({}), {255.0f}); in TEST_F()
67 AddInputFromArray<float>(TensorShape({}), {0}); in TEST_F()
68 AddInputFromArray<float>(TensorShape({}), {255.0f}); in TEST_F()
116 AddInputFromArray<quint8>(TensorShape({a_rows, a_cols}), {11}); in TEST_F()
120 AddInputFromArray<quint8>(TensorShape({b_rows, b_cols}), {0}); in TEST_F()
121 AddInputFromArray<float>(TensorShape({}), {-12.0f}); in TEST_F()
122 AddInputFromArray<float>(TensorShape({}), {243.0f}); in TEST_F()
[all …]
H A Ddynamic_stitch_op_test.cc52 AddInputFromArray<int32>(TensorShape({3}), {0, 4, 7}); in TEST_F()
53 AddInputFromArray<int32>(TensorShape({5}), {1, 6, 2, 3, 5}); in TEST_F()
54 AddInputFromArray<float>(TensorShape({3}), {0, 40, 70}); in TEST_F()
55 AddInputFromArray<float>(TensorShape({5}), {10, 60, 20, 30, 50}); in TEST_F()
68 AddInputFromArray<int32>(TensorShape({3}), {0, 4, 7}); in TEST_F()
69 AddInputFromArray<int32>(TensorShape({2}), {1, 6}); in TEST_F()
70 AddInputFromArray<int32>(TensorShape({3}), {2, 3, 5}); in TEST_F()
71 AddInputFromArray<float>(TensorShape({3, 2}), {0, 1, 40, 41, 70, 71}); in TEST_F()
72 AddInputFromArray<float>(TensorShape({2, 2}), {10, 11, 60, 61}); in TEST_F()
73 AddInputFromArray<float>(TensorShape({3, 2}), {20, 21, 30, 31, 50, 51}); in TEST_F()
[all …]
H A Dscatter_op_test.cc64 AddInputFromArray<tstring>(TensorShape({1}), {"Brain"}); in TEST_F()
65 AddInputFromArray<int32>(TensorShape({1}), {0}); in TEST_F()
66 AddInputFromArray<tstring>(TensorShape({1}), {"TensorFlow"}); in TEST_F()
77 AddInputFromArray<bool>(TensorShape({1}), {false}); in TEST_F()
78 AddInputFromArray<int32>(TensorShape({1}), {0}); in TEST_F()
79 AddInputFromArray<bool>(TensorShape({1}), {true}); in TEST_F()
92 AddInputFromArray<float>(TensorShape({5, 3}), in TEST_F()
94 AddInputFromArray<int32>(TensorShape({3}), {0, 4, 2}); in TEST_F()
95 AddInputFromArray<float>(TensorShape({3, 3}), in TEST_F()
111 AddInputFromArray<float>(TensorShape({5, 3}), in TEST_F()
[all …]
H A Dscatter_nd_op_test.cc82 AddInputFromArray<float>(TensorShape({5, 3}), in TEST_F()
84 AddInputFromArray<int32>(TensorShape({3, 1}), {0, 4, 2}); in TEST_F()
85 AddInputFromArray<float>(TensorShape({3, 3}), in TEST_F()
101 AddInputFromArray<float>(TensorShape({5, 3}), in TEST_F()
103 AddInputFromArray<int64_t>(TensorShape({3, 1}), {0, 4, 2}); in TEST_F()
104 AddInputFromArray<float>(TensorShape({3, 3}), in TEST_F()
133 AddInputFromArray<float>(TensorShape({5}), {0, 0, 0, 0, 0}); in TEST_F()
134 AddInputFromArray<int32>(TensorShape({1}), {3}); in TEST_F()
135 AddInputFromArray<float>(TensorShape({1}), {101}); in TEST_F()
149 AddInputFromArray<float>(TensorShape({5}), {0, 0, 0, 0, 0}); in TEST_F()
[all …]
H A Dquantize_op_test.cc46 AddInputFromArray<float>(TensorShape({7}), in TEST_F()
49 AddInputFromArray<float>(TensorShape({1}), {0}); in TEST_F()
51 AddInputFromArray<float>(TensorShape({1}), {255.0f}); in TEST_F()
100 AddInputFromArray<float>( in TEST_P()
109 AddInputFromArray<float>(TensorShape({num_slices}), min_ranges); in TEST_P()
110 AddInputFromArray<float>(TensorShape({num_slices}), max_ranges); in TEST_P()
144 AddInputFromArray<float>(TensorShape({3}), {-1.0, 0.0, 2.0}); in TEST_F()
145 AddInputFromArray<float>(TensorShape({1}), {-1.0f}); in TEST_F()
146 AddInputFromArray<float>(TensorShape({1}), {2.0f}); in TEST_F()
181 AddInputFromArray<float>( in TEST_P()
[all …]
H A Dquantized_batch_norm_op_test.cc97 AddInputFromArray<quint8>(input_quantized.shape(), in TEST_F()
99 AddInputFromArray<float>(TensorShape({1}), {input_min}); in TEST_F()
100 AddInputFromArray<float>(TensorShape({1}), {input_max}); in TEST_F()
101 AddInputFromArray<quint8>(mean_quantized.shape(), in TEST_F()
103 AddInputFromArray<float>(TensorShape({1}), {mean_min}); in TEST_F()
104 AddInputFromArray<float>(TensorShape({1}), {mean_max}); in TEST_F()
105 AddInputFromArray<quint8>(variance_quantized.shape(), in TEST_F()
107 AddInputFromArray<float>(TensorShape({1}), {variance_min}); in TEST_F()
108 AddInputFromArray<float>(TensorShape({1}), {variance_max}); in TEST_F()
109 AddInputFromArray<quint8>(beta_quantized.shape(), in TEST_F()
[all …]
H A Dfake_quant_ops_test.cc76 AddInputFromArray<float>(shape, data); in RunTestFakeQuantWithMinMaxArgs()
111 AddInputFromArray<float>(shape, data); in RunTestFakeQuantWithMinMaxVars()
113 AddInputFromArray<float>(TensorShape({}), {min}); in RunTestFakeQuantWithMinMaxVars()
115 AddInputFromArray<float>(TensorShape({}), {max}); in RunTestFakeQuantWithMinMaxVars()
148 AddInputFromArray<float>(shape, data); in RunTestFakeQuantWithMinMaxVarsPerChannel()
150 AddInputFromArray<float>(minmax_shape, min); in RunTestFakeQuantWithMinMaxVarsPerChannel()
152 AddInputFromArray<float>(minmax_shape, max); in RunTestFakeQuantWithMinMaxVarsPerChannel()
443 AddInputFromArray<float>(TensorShape({2, 3}), in TEST_F()
473 AddInputFromArray<float>(TensorShape({2, 3}), in TEST_F()
504 AddInputFromArray<float>(TensorShape({2, 3}), in TEST_F()
[all …]
H A Dsparse_to_dense_op_test.cc58 AddInputFromArray<int32>(TensorShape({3}), {1, 3, 4}); in TEST_F()
60 AddInputFromArray<int32>(TensorShape({1}), {5}); in TEST_F()
62 AddInputFromArray<float>(TensorShape({}), {2}); in TEST_F()
64 AddInputFromArray<float>(TensorShape({}), {-2}); in TEST_F()
77 AddInputFromArray<int64_t>(TensorShape({3}), {1, 3, 4}); in TEST_F()
79 AddInputFromArray<int64_t>(TensorShape({1}), {5}); in TEST_F()
81 AddInputFromArray<double>(TensorShape({}), {2}); in TEST_F()
83 AddInputFromArray<double>(TensorShape({}), {-2}); in TEST_F()
96 AddInputFromArray<int32>({3}, {1, 3, 4}); in TEST_F()
98 AddInputFromArray<int32>({1}, {5}); in TEST_F()
[all …]
H A Dgather_op_test.cc58 AddInputFromArray<float>(TensorShape({5}), {0, 1, 2, 3, 4}); in TEST_F()
59 AddInputFromArray<int32>(TensorShape({}), {3}); in TEST_F()
60 AddInputFromArray<int32>(TensorShape({}), {0}); in TEST_F()
73 AddInputFromArray<std::complex<float>>( in TEST_F()
77 AddInputFromArray<int32>(TensorShape({}), {3}); in TEST_F()
78 AddInputFromArray<int32>(TensorShape({}), {0}); in TEST_F()
92 AddInputFromArray<float>(TensorShape({5, 3}), in TEST_F()
94 AddInputFromArray<int32>(TensorShape({4}), {0, 4, 0, 2}); in TEST_F()
95 AddInputFromArray<int32>(TensorShape({}), {0}); in TEST_F()
108 AddInputFromArray<float>(TensorShape({5, 3}), in TEST_F()
[all …]
H A Dquantized_conv_ops_test.cc90 AddInputFromArray<quint8>(image_quantized.shape(), in TEST_F()
92 AddInputFromArray<quint8>(filter_quantized.shape(), in TEST_F()
94 AddInputFromArray<float>(TensorShape({}), {image_min}); in TEST_F()
95 AddInputFromArray<float>(TensorShape({}), {image_max}); in TEST_F()
96 AddInputFromArray<float>(TensorShape({}), {filter_min}); in TEST_F()
97 AddInputFromArray<float>(TensorShape({}), {filter_max}); in TEST_F()
153 AddInputFromArray<quint8>( in TEST_F()
158 AddInputFromArray<quint8>( in TEST_F()
161 AddInputFromArray<float>(TensorShape({}), {0}); in TEST_F()
162 AddInputFromArray<float>(TensorShape({}), {255.0f}); in TEST_F()
[all …]
H A Dsequence_ops_test.cc60 AddInputFromArray<int32>(TensorShape({}), {0}); in TEST_F()
61 AddInputFromArray<int32>(TensorShape({}), {10}); in TEST_F()
62 AddInputFromArray<int32>(TensorShape({}), {2}); in TEST_F()
75 AddInputFromArray<float>(TensorShape({}), {0.5}); in TEST_F()
76 AddInputFromArray<float>(TensorShape({}), {2}); in TEST_F()
77 AddInputFromArray<float>(TensorShape({}), {0.3}); in TEST_F()
90 AddInputFromArray<double>(TensorShape({}), {0.0}); in TEST_F()
91 AddInputFromArray<double>(TensorShape({}), {10000}); in TEST_F()
92 AddInputFromArray<double>(TensorShape({}), {0.5}); in TEST_F()
107 AddInputFromArray<float>(TensorShape({}), {3.0}); in TEST_F()
[all …]
H A Dsparse_dense_binary_op_shared_test.cc76 AddInputFromArray<int64_t>(TensorShape({1, 1}), {0}); // indices in TEST_F()
77 AddInputFromArray<float>(TensorShape({1}), {1618}); // values in TEST_F()
78 AddInputFromArray<int64_t>(TensorShape({1}), {1}); // shape in TEST_F()
79 AddInputFromArray<float>(TensorShape({2, 1}), {17, 19}); // dense in TEST_F()
87 AddInputFromArray<int64_t>(TensorShape({1, 2}), {0, 0}); in TEST_F()
88 AddInputFromArray<float>(TensorShape({1}), {1618}); in TEST_F()
89 AddInputFromArray<int64_t>(TensorShape({2}), {1, 1}); in TEST_F()
90 AddInputFromArray<float>(TensorShape({2, 1}), {17, 19}); in TEST_F()
111 AddInputFromArray<int64_t>(indices_shape, indices); in TEST_F()
112 AddInputFromArray<float>(TensorShape({4}), {1, 2, 3, 4}); in TEST_F()
[all …]
H A Dfused_batch_norm_op_test.cc48 AddInputFromArray<float>(TensorShape({1, 1, 6, 2}), in TEST_F()
50 AddInputFromArray<float>(TensorShape({2}), {4.0, 4.0}); in TEST_F()
51 AddInputFromArray<float>(TensorShape({2}), {2.0, 2.0}); in TEST_F()
52 AddInputFromArray<float>(TensorShape({0}), {}); in TEST_F()
53 AddInputFromArray<float>(TensorShape({0}), {}); in TEST_F()
83 AddInputFromArray<float>(TensorShape({1, 1, 6, 2}), in TEST_F()
85 AddInputFromArray<float>(TensorShape({2}), {4.0, 4.0}); in TEST_F()
86 AddInputFromArray<float>(TensorShape({2}), {2.0, 2.0}); in TEST_F()
87 AddInputFromArray<float>(TensorShape({2}), {6.0, 6.0}); in TEST_F()
88 AddInputFromArray<float>(TensorShape({2}), {16.0, 16.0}); in TEST_F()
[all …]
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/uniform_quant_ops/
H A Duniform_requantize_op_test.cc85 AddInputFromArray<qint32>(TensorShape({2, 3}), {0, 0, 0, 0, 0, 0}); in TEST_F()
86 AddInputFromArray<float>(TensorShape({}), {1.0}); in TEST_F()
87 AddInputFromArray<int32>(TensorShape({}), {0}); in TEST_F()
88 AddInputFromArray<float>(TensorShape({}), {1.0}); in TEST_F()
89 AddInputFromArray<int32>(TensorShape({}), {0}); in TEST_F()
112 AddInputFromArray<qint32>(TensorShape({2, 3}), {-28, -21, -1, 0, 4, 9}); in TEST_F()
113 AddInputFromArray<float>(TensorShape({}), {0.5}); in TEST_F()
114 AddInputFromArray<int32>(TensorShape({}), {-1}); in TEST_F()
115 AddInputFromArray<float>(TensorShape({}), {0.125}); in TEST_F()
116 AddInputFromArray<int32>(TensorShape({}), {-20}); in TEST_F()
[all …]

1234