Home
last modified time | relevance | path

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

123

/external/tensorflow/tensorflow/core/kernels/
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 …]
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 …]
Dragged_range_op_test.cc50 AddInputFromArray<int>(TensorShape({4}), {0, 5, 8, 5}); // starts in TEST_F()
51 AddInputFromArray<int>(TensorShape({4}), {8, 7, 8, 1}); // limits in TEST_F()
52 AddInputFromArray<int>(TensorShape({4}), {2, 1, 1, -1}); // deltas in TEST_F()
65 AddInputFromArray<float>(TensorShape({4}), {0, 5, 8, 5}); // starts in TEST_F()
66 AddInputFromArray<float>(TensorShape({4}), {8, 7, 8, 1}); // limits in TEST_F()
67 AddInputFromArray<float>(TensorShape({4}), {2, 1, 1, -1}); // deltas in TEST_F()
80 AddInputFromArray<int>(TensorShape({3}), {0, 5, 8}); // starts in TEST_F()
81 AddInputFromArray<int>(TensorShape({3}), {8, 7, 8}); // limits in TEST_F()
82 AddInputFromArray<int>(TensorShape({}), {1}); // deltas in TEST_F()
95 AddInputFromArray<int>(TensorShape({}), {0}); // starts in TEST_F()
[all …]
Droll_op_test.cc56 AddInputFromArray<float>(TensorShape({5}), {0, 1, 2, 3, 4}); in TEST_F()
57 AddInputFromArray<int32>(TensorShape({}), {3}); in TEST_F()
58 AddInputFromArray<int32>(TensorShape({}), {0}); in TEST_F()
71 AddInputFromArray<string>(TensorShape({5}), {"a", "b", "c", "d", "e"}); in TEST_F()
72 AddInputFromArray<int32>(TensorShape({}), {3}); in TEST_F()
73 AddInputFromArray<int32>(TensorShape({}), {0}); in TEST_F()
86 AddInputFromArray<std::complex<float>>( in TEST_F()
90 AddInputFromArray<int32>(TensorShape({}), {3}); in TEST_F()
91 AddInputFromArray<int32>(TensorShape({}), {0}); in TEST_F()
107 AddInputFromArray<float>(TensorShape({3, 5}), in TEST_F()
[all …]
Dmkl_quantized_conv_ops_test.cc61 AddInputFromArray<T>(input.shape(), input.flat<T>()); in ConvertMklToTF()
62 AddInputFromArray<uint8>(input_metadata_tensor.shape(), in ConvertMklToTF()
143 AddInputFromArray<quint8>(image_quantized.shape(), in TEST_F()
145 AddInputFromArray<qint8>(filter_quantized.shape(), in TEST_F()
147 AddInputFromArray<float>(TensorShape({1}), {image_min}); in TEST_F()
148 AddInputFromArray<float>(TensorShape({1}), {image_max}); in TEST_F()
149 AddInputFromArray<float>(TensorShape({1}), {filter_min}); in TEST_F()
150 AddInputFromArray<float>(TensorShape({1}), {filter_max}); in TEST_F()
152 AddInputFromArray<uint8>(dummy_shape, dummy_tensor); in TEST_F()
153 AddInputFromArray<uint8>(dummy_shape, dummy_tensor); in TEST_F()
[all …]
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 …]
Dquantize_and_dequantize_op_test.cc52 AddInputFromArray<float>(TensorShape({1}), {-3.5}); in TEST_F()
53 AddInputFromArray<float>(TensorShape({}), {0.0}); // Min in TEST_F()
54 AddInputFromArray<float>(TensorShape({}), {0.0}); // Max in TEST_F()
77 AddInputFromArray<float>(TensorShape({1}), {-3.5}); in TEST_F()
78 AddInputFromArray<float>(TensorShape({}), {0.0}); // Min in TEST_F()
79 AddInputFromArray<float>(TensorShape({}), {0.0}); // Max in TEST_F()
80 AddInputFromArray<int32>(TensorShape({}), {8}); // num_bits in TEST_F()
104 AddInputFromArray<float>(TensorShape({7}), in TEST_F()
106 AddInputFromArray<float>(TensorShape({}), {0.0}); // Min in TEST_F()
107 AddInputFromArray<float>(TensorShape({}), {0.0}); // Max in TEST_F()
[all …]
Dresize_nearest_neighbor_op_test.cc77 AddInputFromArray<float>(TensorShape({1, 2, 2, 1}), {1, 2, 3, 4}); in TEST_F()
78 AddInputFromArray<int32>(TensorShape({2}), {1, 1}); in TEST_F()
95 AddInputFromArray<float>(TensorShape({1, 2, 2, 1}), {1, 2, 3, 4}); in TEST_F()
96 AddInputFromArray<int32>(TensorShape({2}), {1, 1}); in TEST_F()
112 AddInputFromArray<float>(TensorShape({1, 2, 2, 1}), {1, 2, 3, 4}); in TEST_F()
113 AddInputFromArray<int32>(TensorShape({2}), {3, 3}); in TEST_F()
133 AddInputFromArray<float>(TensorShape({1, 2, 2, 1}), {1, 2, 3, 4}); in TEST_F()
134 AddInputFromArray<int32>(TensorShape({2}), {3, 3}); in TEST_F()
154 AddInputFromArray<float>(TensorShape({1, 3, 3, 1}), in TEST_F()
156 AddInputFromArray<int32>(TensorShape({2}), {2, 2}); in TEST_F()
[all …]
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({1}), {0}); in TEST_F()
66 AddInputFromArray<float>(TensorShape({1}), {255.0f}); in TEST_F()
67 AddInputFromArray<float>(TensorShape({1}), {0}); in TEST_F()
68 AddInputFromArray<float>(TensorShape({1}), {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({1}), {-12.0f}); in TEST_F()
122 AddInputFromArray<float>(TensorShape({1}), {243.0f}); in TEST_F()
[all …]
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>(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 …]
Dscatter_op_test.cc53 AddInputFromArray<string>(TensorShape({1}), {"Brain"}); in TEST_F()
54 AddInputFromArray<int32>(TensorShape({1}), {0}); in TEST_F()
55 AddInputFromArray<string>(TensorShape({1}), {"TensorFlow"}); in TEST_F()
66 AddInputFromArray<bool>(TensorShape({1}), {false}); in TEST_F()
67 AddInputFromArray<int32>(TensorShape({1}), {0}); in TEST_F()
68 AddInputFromArray<bool>(TensorShape({1}), {true}); in TEST_F()
81 AddInputFromArray<float>(TensorShape({5, 3}), in TEST_F()
83 AddInputFromArray<int32>(TensorShape({3}), {0, 4, 2}); in TEST_F()
84 AddInputFromArray<float>(TensorShape({3, 3}), in TEST_F()
100 AddInputFromArray<float>(TensorShape({5, 3}), in TEST_F()
[all …]
Dsequence_ops_test.cc59 AddInputFromArray<int32>(TensorShape({}), {0}); in TEST_F()
60 AddInputFromArray<int32>(TensorShape({}), {10}); in TEST_F()
61 AddInputFromArray<int32>(TensorShape({}), {2}); in TEST_F()
74 AddInputFromArray<float>(TensorShape({}), {0.5}); in TEST_F()
75 AddInputFromArray<float>(TensorShape({}), {2}); in TEST_F()
76 AddInputFromArray<float>(TensorShape({}), {0.3}); in TEST_F()
89 AddInputFromArray<double>(TensorShape({}), {0.0}); in TEST_F()
90 AddInputFromArray<double>(TensorShape({}), {10000}); in TEST_F()
91 AddInputFromArray<double>(TensorShape({}), {0.5}); in TEST_F()
106 AddInputFromArray<float>(TensorShape({}), {3.0}); in TEST_F()
[all …]
Drequantize_op_test.cc54 AddInputFromArray<qint32>(TensorShape({value_count}), in TEST_F()
56 AddInputFromArray<float>(TensorShape({1}), {-256.0f}); in TEST_F()
57 AddInputFromArray<float>(TensorShape({1}), {256.0f}); in TEST_F()
58 AddInputFromArray<float>(TensorShape({1}), {-1.0f}); in TEST_F()
59 AddInputFromArray<float>(TensorShape({1}), {1.0f}); in TEST_F()
72 AddInputFromArray<qint32>(TensorShape({value_count}), in TEST_F()
74 AddInputFromArray<float>(TensorShape({1}), {-256.0f}); in TEST_F()
75 AddInputFromArray<float>(TensorShape({1}), {256.0f}); in TEST_F()
76 AddInputFromArray<float>(TensorShape({1}), {0.01f}); in TEST_F()
77 AddInputFromArray<float>(TensorShape({1}), {1.0f}); in TEST_F()
[all …]
Dsparse_to_dense_op_test.cc56 AddInputFromArray<int32>(TensorShape({3}), {1, 3, 4}); in TEST_F()
58 AddInputFromArray<int32>(TensorShape({1}), {5}); in TEST_F()
60 AddInputFromArray<float>(TensorShape({}), {2}); in TEST_F()
62 AddInputFromArray<float>(TensorShape({}), {-2}); in TEST_F()
75 AddInputFromArray<int64>(TensorShape({3}), {1, 3, 4}); in TEST_F()
77 AddInputFromArray<int64>(TensorShape({1}), {5}); in TEST_F()
79 AddInputFromArray<double>(TensorShape({}), {2}); in TEST_F()
81 AddInputFromArray<double>(TensorShape({}), {-2}); in TEST_F()
94 AddInputFromArray<int32>({3}, {1, 3, 4}); in TEST_F()
96 AddInputFromArray<int32>({1}, {5}); in TEST_F()
[all …]
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({1}), {image_min}); in TEST_F()
95 AddInputFromArray<float>(TensorShape({1}), {image_max}); in TEST_F()
96 AddInputFromArray<float>(TensorShape({1}), {filter_min}); in TEST_F()
97 AddInputFromArray<float>(TensorShape({1}), {filter_max}); in TEST_F()
153 AddInputFromArray<quint8>( in TEST_F()
158 AddInputFromArray<quint8>( in TEST_F()
161 AddInputFromArray<float>(TensorShape({1}), {0}); in TEST_F()
162 AddInputFromArray<float>(TensorShape({1}), {255.0f}); in TEST_F()
[all …]
Dquantized_batch_norm_op_test.cc98 AddInputFromArray<quint8>(input_quantized.shape(), in TEST_F()
100 AddInputFromArray<float>(TensorShape({1}), {input_min}); in TEST_F()
101 AddInputFromArray<float>(TensorShape({1}), {input_max}); in TEST_F()
102 AddInputFromArray<quint8>(mean_quantized.shape(), in TEST_F()
104 AddInputFromArray<float>(TensorShape({1}), {mean_min}); in TEST_F()
105 AddInputFromArray<float>(TensorShape({1}), {mean_max}); in TEST_F()
106 AddInputFromArray<quint8>(variance_quantized.shape(), in TEST_F()
108 AddInputFromArray<float>(TensorShape({1}), {variance_min}); in TEST_F()
109 AddInputFromArray<float>(TensorShape({1}), {variance_max}); in TEST_F()
110 AddInputFromArray<quint8>(beta_quantized.shape(), in TEST_F()
[all …]
Dgather_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<std::complex<float>>( in TEST_F()
76 AddInputFromArray<int32>(TensorShape({}), {3}); in TEST_F()
77 AddInputFromArray<int32>(TensorShape({}), {0}); in TEST_F()
91 AddInputFromArray<float>(TensorShape({5, 3}), in TEST_F()
93 AddInputFromArray<int32>(TensorShape({4}), {0, 4, 0, 2}); in TEST_F()
94 AddInputFromArray<int32>(TensorShape({}), {0}); in TEST_F()
107 AddInputFromArray<float>(TensorShape({5, 3}), in TEST_F()
[all …]
Dquantize_op_test.cc40 AddInputFromArray<float>(TensorShape({7}), in TEST_F()
43 AddInputFromArray<float>(TensorShape({1}), {0}); in TEST_F()
45 AddInputFromArray<float>(TensorShape({1}), {255.0f}); in TEST_F()
63 AddInputFromArray<float>(TensorShape({8}), in TEST_F()
65 AddInputFromArray<float>(TensorShape({1}), {-255.0f}); in TEST_F()
66 AddInputFromArray<float>(TensorShape({1}), {127.0f}); in TEST_F()
95 AddInputFromArray<float>(TensorShape({3}), {-1.0, 0.0, 2.0}); in TEST_F()
96 AddInputFromArray<float>(TensorShape({1}), {-1.0f}); in TEST_F()
97 AddInputFromArray<float>(TensorShape({1}), {2.0f}); in TEST_F()
125 AddInputFromArray<float>(TensorShape({7}), in TEST_F()
[all …]
Dmkl_quantized_concat_op_test.cc65 AddInputFromArray<T>(first.shape(), first.flat<T>()); in ConvertMKL2TF()
66 AddInputFromArray<uint8>(second.shape(), second.flat<uint8>()); in ConvertMKL2TF()
135 AddInputFromArray<quint8>(first_quantized.shape(), in TestSmall8Bit()
137 AddInputFromArray<quint8>(second_quantized.shape(), in TestSmall8Bit()
139 AddInputFromArray<int32>(TensorShape({}), {0}); in TestSmall8Bit()
140 AddInputFromArray<float>(TensorShape({}), {first_min}); in TestSmall8Bit()
141 AddInputFromArray<float>(TensorShape({}), {second_min}); in TestSmall8Bit()
142 AddInputFromArray<float>(TensorShape({}), {first_max}); in TestSmall8Bit()
143 AddInputFromArray<float>(TensorShape({}), {second_max}); in TestSmall8Bit()
144 AddInputFromArray<uint8>(dummy_shape, dummy_tensor); in TestSmall8Bit()
[all …]
Dresize_bilinear_op_test.cc126 AddInputFromArray<int32>(TensorShape({2}), {output_width, output_height}); in TestResize()
193 AddInputFromArray<float>(TensorShape({1, 2, 2, 1}), {1, 2, 3, 4}); in TEST_F()
194 AddInputFromArray<int32>(TensorShape({2}), {1, 1}); in TEST_F()
206 AddInputFromArray<int32>(TensorShape({2}), {1, 1}); in TEST_F()
225 AddInputFromArray<float>(TensorShape({1, 2, 2, 1}), {1, 2, 3, 4}); in TEST_F()
226 AddInputFromArray<int32>(TensorShape({2}), {1, 1}); in TEST_F()
240 AddInputFromArray<float>(TensorShape({1, 2, 2, 1}), {1, 2, 3, 4}); in TEST_F()
241 AddInputFromArray<int32>(TensorShape({2}), {3, 3}); in TEST_F()
260 AddInputFromArray<float>(TensorShape({1, 2, 2, 1}), {1, 2, 3, 4}); in TEST_F()
261 AddInputFromArray<int32>(TensorShape({2}), {3, 3}); in TEST_F()
[all …]
Dfused_batch_norm_op_test.cc43 AddInputFromArray<float>(TensorShape({1, 1, 6, 2}), in TEST_F()
45 AddInputFromArray<float>(TensorShape({2}), {4.0, 4.0}); in TEST_F()
46 AddInputFromArray<float>(TensorShape({2}), {2.0, 2.0}); in TEST_F()
47 AddInputFromArray<float>(TensorShape({0}), {}); in TEST_F()
48 AddInputFromArray<float>(TensorShape({0}), {}); in TEST_F()
77 AddInputFromArray<float>(TensorShape({1, 1, 6, 2}), in TEST_F()
79 AddInputFromArray<float>(TensorShape({2}), {4.0, 4.0}); in TEST_F()
80 AddInputFromArray<float>(TensorShape({2}), {2.0, 2.0}); in TEST_F()
81 AddInputFromArray<float>(TensorShape({2}), {10, 10}); in TEST_F()
82 AddInputFromArray<float>(TensorShape({2}), {11.67f, 11.67f}); in TEST_F()
[all …]
Dsparse_add_op_test.cc70 AddInputFromArray<int64>(indices_shape, indices); \ in TEST_F()
71 AddInputFromArray<float>(TensorShape({4}), {1, 2, 3, 4}); \ in TEST_F()
72 AddInputFromArray<int64>(TensorShape({2}), shape); in TEST_F()
76 AddInputFromArray<float>(TensorShape({}), {0.0}); in TEST_F()
109 AddInputFromArray<int64>(indices_shape, indices); \
110 AddInputFromArray<VALTYPE>(TensorShape({4}), {1, 2, 3, 4}); \
111 AddInputFromArray<int64>(TensorShape({2}), shape); \
113 AddInputFromArray<int64>(TensorShape({2, 2}), {0, 0, 1, 1}); \
114 AddInputFromArray<VALTYPE>(TensorShape({2}), {5, 6}); \
115 AddInputFromArray<int64>(TensorShape({2}), shape); \
[all …]
Dlogging_ops_test.cc45 AddInputFromArray<string>(TensorShape({}), {"bar"}); in TEST_F()
70 AddInputFromArray<int32>(TensorShape({6}), {1, 2, 3, 4, 5, 6}); in TEST_F()
71 AddInputFromArray<int32>(TensorShape({6}), {1, 2, 3, 4, 5, 6}); in TEST_F()
72 AddInputFromArray<int32>(TensorShape({6}), {1, 2, 3, 4, 5, 6}); in TEST_F()
81 AddInputFromArray<int32>(TensorShape({6}), {1, 2, 3, 4, 5, 6}); in TEST_F()
82 AddInputFromArray<int32>(TensorShape({6}), {1, 2, 3, 4, 5, 6}); in TEST_F()
83 AddInputFromArray<int32>(TensorShape({6}), {1, 2, 3, 4, 5, 6}); in TEST_F()
92 AddInputFromArray<int32>(TensorShape({6}), {1, 2, 3, 4, 5, 6}); in TEST_F()
93 AddInputFromArray<string>(TensorShape({}), {"foo"}); in TEST_F()
94 AddInputFromArray<string>(TensorShape({}), {"bar"}); in TEST_F()
[all …]
Dfake_quant_ops_test.cc66 AddInputFromArray<float>(shape, data); in RunTestFakeQuantWithMinMaxArgs()
91 AddInputFromArray<float>(shape, data); in RunTestFakeQuantWithMinMaxVars()
93 AddInputFromArray<float>(TensorShape({}), {min}); in RunTestFakeQuantWithMinMaxVars()
95 AddInputFromArray<float>(TensorShape({}), {max}); in RunTestFakeQuantWithMinMaxVars()
121 AddInputFromArray<float>(shape, data); in RunTestFakeQuantWithMinMaxVarsPerChannel()
123 AddInputFromArray<float>(minmax_shape, min); in RunTestFakeQuantWithMinMaxVarsPerChannel()
125 AddInputFromArray<float>(minmax_shape, max); in RunTestFakeQuantWithMinMaxVarsPerChannel()
372 AddInputFromArray<float>(TensorShape({2, 3}), in TEST_F()
402 AddInputFromArray<float>(TensorShape({2, 3}), in TEST_F()
433 AddInputFromArray<float>(TensorShape({2, 3}), in TEST_F()
[all …]
Dsparse_dense_binary_op_shared_test.cc75 AddInputFromArray<int64>(TensorShape({1, 1}), {0}); // indices in TEST_F()
76 AddInputFromArray<float>(TensorShape({1}), {1618}); // values in TEST_F()
77 AddInputFromArray<int64>(TensorShape({1}), {1}); // shape in TEST_F()
78 AddInputFromArray<float>(TensorShape({2, 1}), {17, 19}); // dense in TEST_F()
86 AddInputFromArray<int64>(TensorShape({1, 2}), {0, 0}); in TEST_F()
87 AddInputFromArray<float>(TensorShape({1}), {1618}); in TEST_F()
88 AddInputFromArray<int64>(TensorShape({2}), {1, 1}); in TEST_F()
89 AddInputFromArray<float>(TensorShape({2, 1}), {17, 19}); in TEST_F()
110 AddInputFromArray<int64>(indices_shape, indices); in TEST_F()
111 AddInputFromArray<float>(TensorShape({4}), {1, 2, 3, 4}); in TEST_F()
[all …]

123