Lines Matching refs:num_elem
24 num_elem = 3
26 x = ceed.Vector(num_elem + 1)
27 a = np.arange(10, 10 + num_elem + 1, dtype=ceed.scalar_type())
30 ind = np.zeros(2 * num_elem, dtype="int32")
31 for i in range(num_elem):
34 r = ceed.ElemRestriction(num_elem, 2, 1, 1, num_elem + 1, ind,
37 y = ceed.Vector(2 * num_elem)
43 for i in range(2 * num_elem):
54 num_elem = 3
56 x = ceed.Vector(2 * num_elem)
57 a = np.arange(10, 10 + 2 * num_elem, dtype=ceed.scalar_type())
61 r = ceed.StridedElemRestriction(num_elem, 2, 1, 2 * num_elem, strides)
63 y = ceed.Vector(2 * num_elem)
69 for i in range(2 * num_elem):
80 num_elem = 8
85 x = ceed.Vector(num_elem + 1)
86 a = np.arange(10, 10 + num_elem + 1, dtype=ceed.scalar_type())
89 ind = np.zeros(2 * num_elem, dtype="int32")
90 for i in range(num_elem):
93 r = ceed.BlockedElemRestriction(num_elem, elem_size, blk_size, 1, 1,
94 num_elem + 1, ind, cmode=libceed.COPY_VALUES)
105 for k in range(num_elem):
116 for i in range(num_elem + 1):
118 0 and i < num_elem else 1.0)
128 num_elem = 8
133 x = ceed.Vector(num_elem + 1)
134 a = np.arange(10, 10 + num_elem + 1, dtype=ceed.scalar_type())
137 ind = np.zeros(2 * num_elem, dtype="int32")
138 for i in range(num_elem):
141 r = ceed.BlockedElemRestriction(num_elem, elem_size, blk_size, 1, 1,
142 num_elem + 1, ind, cmode=libceed.COPY_VALUES)
153 for k in range(blk_size, num_elem):
164 for i in range(blk_size, num_elem + 1):
166 blk_size and i < num_elem else 1.0)
176 num_elem = 3
178 ind = np.zeros(4 * num_elem, dtype="int32")
179 for i in range(num_elem):
184 r = ceed.ElemRestriction(num_elem, 4, 1, 1, 3 * num_elem + 1, ind,
190 for i in range(3 * num_elem + 1):
191 val = 1 + (1 if (i > 0 and i < 3 * num_elem and i % 3 == 0) else 0)
202 num_elem = 3
204 ind = np.zeros(2 * num_elem, dtype="int32")
205 for i in range(num_elem):
208 r = ceed.ElemRestriction(num_elem, 2, 1, 1, num_elem + 1, ind,
225 num_elem = 3
228 r = ceed.StridedElemRestriction(num_elem, 2, 1, num_elem * 2, strides)
244 num_elem = 3
248 num_elem, 2, 2, 1, num_elem * 2, strides)
264 num_elem = 3
266 x = ceed.Vector(num_elem + 1)
267 a = np.arange(10, 10 + num_elem + 1, dtype=ceed.scalar_type())
270 ind = np.zeros(2 * num_elem, dtype="int32")
271 orients = np.zeros(2 * num_elem, dtype="bool")
272 for i in range(num_elem):
279 num_elem,
283 num_elem + 1,
288 y = ceed.Vector(2 * num_elem)
294 for i in range(num_elem):
307 num_elem = 3
309 x = ceed.Vector(num_elem + 1)
310 a = np.arange(10, 10 + num_elem + 1, dtype=ceed.scalar_type())
313 ind = np.zeros(2 * num_elem, dtype="int32")
314 curl_orients = np.zeros(3 * 2 * num_elem, dtype="int8")
315 for i in range(num_elem):
333 num_elem,
337 num_elem + 1,
342 y = ceed.Vector(2 * num_elem)
348 for i in range(num_elem):