Home | History | Annotate | Download | only in tests

Lines Matching refs:array

58                    [np.array([[1, 2, 3]], dtype=np.float32)],
59 expected=np.array([[1, 2, 3]], dtype=np.float32))
62 [np.array([1, 2], dtype=np.float32),
63 np.array([10, 20], dtype=np.float32)],
64 expected=np.array([11, 22], dtype=np.float32))
66 [np.array([-4], dtype=np.float32),
67 np.array([10], dtype=np.float32),
68 np.array([42], dtype=np.float32)],
69 expected=np.array([48], dtype=np.float32))
74 math_ops.add_n, [np.array([[1 + 2j, 2 - 3j, 3 + 4j]], dtype=dtype)],
75 expected=np.array([[1 + 2j, 2 - 3j, 3 + 4j]], dtype=dtype))
79 np.array([1 + 2j, 2 - 3j], dtype=dtype),
80 np.array([10j, 20], dtype=dtype)
82 expected=np.array([1 + 12j, 22 - 3j], dtype=dtype))
85 np.array([-4, 5j], dtype=dtype),
86 np.array([2 + 10j, -2], dtype=dtype),
87 np.array([42j, 3 + 3j], dtype=dtype)
89 expected=np.array([-2 + 52j, 1 + 8j], dtype=dtype))
94 [np.array([[1, 2, 3]], dtype=np.float32)],
95 expected=[np.array([[1, 2, 3]], dtype=np.float32)])
97 [np.array([[1, 2], [3, 4]], dtype=np.float32),
98 np.array([[3, 2, 1], [6, 5, 1]], dtype=np.float32)],
100 np.array([[1, 2], [3, 4]], dtype=np.float32),
101 np.array([[3, 2, 1], [6, 5, 1]], dtype=np.float32)])
103 [np.array([[1], [2], [3], [4]], dtype=np.int32),
104 np.array([[3, 2, 1], [6, 5, 1]], dtype=np.float32)],
106 np.array([[1], [2], [3], [4]], dtype=np.int32),
107 np.array([[3, 2, 1], [6, 5, 1]], dtype=np.float32)])
112 np.array(
113 [[1, 2, 3], [4, 5, 6]], dtype=np.float32), np.array(
116 expected=np.array(
121 np.array(
122 [[1, 2, 3], [4, 5, 6]], dtype=np.float32), np.array(
125 expected=np.array(
130 indices = array_ops.constant(np.array([[2, 3], [0, 1]], dtype=np.int32))
135 expected = np.array([[[3, 3, 7, 3], [3, 3, 3, 7]],
145 expected = np.array([[[1, 1], [1, 1], [2, 1], [1, 2]],
154 array_ops.split(np.array([[1, 2, 3, 4], [5, 6, 7, 8], [9, 0, 1, 2]],
157 expected = [np.array([[1, 2], [5, 6], [9, 0]], dtype=np.float32),
158 np.array([[3, 4], [7, 8], [1, 2]], dtype=np.float32)]
163 [np.array([[], [], []], dtype=np.float32),
164 np.array([1, 0], dtype=np.int32),
165 np.array([3, 0], dtype=np.int32),
166 np.array([1, 1], dtype=np.int32)],
167 expected=np.array([[], []], dtype=np.float32))
172 np.array([[], [], []], dtype=np.float32), np.array(
173 [1, 0], dtype=np.int64), np.array([3, 0], dtype=np.int64),
174 np.array([1, 1], dtype=np.int64)
176 expected=np.array([[], []], dtype=np.float32))
179 [np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9]],
181 np.array([1, 1], dtype=np.int32),
182 np.array([3, 3], dtype=np.int32),
183 np.array([1, 1], dtype=np.int32)],
184 expected=np.array([[5, 6], [8, 9]], dtype=np.float32))
187 [np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9]],
189 np.array([0, 2], dtype=np.int32),
190 np.array([2, 0], dtype=np.int32),
191 np.array([1, -1], dtype=np.int32)],
192 expected=np.array([[3, 2], [6, 5]], dtype=np.float32))
195 [np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9]],
197 expected=np.array([[[3, 2, 1]], [[6, 5, 4]]],
201 [np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9]],
203 expected=np.array([[4], [5], [6]], dtype=np.float32))
208 [np.array([], dtype=np.int32),
209 np.array([], dtype=np.int32),
210 np.array([], dtype=np.int32),
211 np.array([], dtype=np.int32),
213 expected=np.array(np.float32(0.5), dtype=np.float32))
217 [np.array([3], dtype=np.int32),
218 np.array([0], dtype=np.int32),
219 np.array([0], dtype=np.int32),
220 np.array([1], dtype=np.int32),
221 np.array([], dtype=np.float32)],
222 expected=np.array([0, 0, 0], dtype=np.float32))
225 [np.array([3, 0], dtype=np.int32),
226 np.array([1, 0], dtype=np.int32),
227 np.array([3, 0], dtype=np.int32),
228 np.array([1, 1], dtype=np.int32),
229 np.array([[], []], dtype=np.float32)],
230 expected=np.array([[], [], []], dtype=np.float32))
233 [np.array([3, 3], dtype=np.int32),
234 np.array([1, 1], dtype=np.int32),
235 np.array([3, 3], dtype=np.int32),
236 np.array([1, 1], dtype=np.int32),
237 np.array([[5, 6], [8, 9]], dtype=np.float32)],
238 expected=np.array([[0, 0, 0], [0, 5, 6], [0, 8, 9]],
246 [np.array([3, 1, 3], dtype=np.int32),
247 np.array([0, 0, 0, 2], dtype=np.int32),
248 np.array([0, 3, 1, -4], dtype=np.int32),
249 np.array([1, 2, 1, -3], dtype=np.int32),
250 np.array([[[1], [2]]], dtype=np.float32)],
251 expected=np.array([[[0, 0, 1]], [[0, 0, 0]], [[0, 0, 2]]],
256 [np.array([4, 4], dtype=np.int32),
257 np.array([0, 0, 0, 1, 0], dtype=np.int32),
258 np.array([0, 3, 0, 4, 0], dtype=np.int32),
259 np.array([1, 2, 1, 2, 1], dtype=np.int32),
260 np.array([[[[[1], [2]]], [[[3], [4]]]]], dtype=np.float32)],
261 expected=np.array([[0, 1, 0, 2], [0, 0, 0, 0], [0, 3, 0, 4],
265 [np.array([3, 3], dtype=np.int32),
266 np.array([0, 2], dtype=np.int32),
267 np.array([2, 0], dtype=np.int32),
268 np.array([1, -1], dtype=np.int32),
269 np.array([[1, 2], [3, 4]], dtype=np.float32)],
270 expected=np.array([[0, 2, 1], [0, 4, 3], [0, 0, 0]],
274 [np.array([3, 3], dtype=np.int32),
275 np.array([2, 2], dtype=np.int32),
276 np.array([0, 1], dtype=np.int32),
277 np.array([-1, -2], dtype=np.int32),
278 np.array([[1], [2]], dtype=np.float32)],
279 expected=np.array([[0, 0, 0], [0, 0, 2], [0, 0, 1]],