Lines Matching refs:Tensor
77 std::initializer_list<std::reference_wrapper<const Tensor>> tensors, in check_defined()
99 inline c10::MaybeOwned<Tensor> expand_inplace( in expand_inplace()
100 const Tensor& tensor, in expand_inplace()
101 const Tensor& to_expand) { in expand_inplace()
103 return c10::MaybeOwned<Tensor>::borrowed(to_expand); in expand_inplace()
105 return c10::MaybeOwned<Tensor>::owned( in expand_inplace()
109 inline c10::MaybeOwned<Tensor> expand_inplace(
110 const Tensor& tensor,
111 Tensor&& to_expand) = delete;
113 inline c10::MaybeOwned<Tensor> expand_inplace( in expand_inplace()
114 const Tensor& tensor, in expand_inplace()
115 const Tensor& to_expand, in expand_inplace()
121 inline c10::MaybeOwned<Tensor> expand_inplace(
122 const Tensor& tensor,
123 Tensor&& to_expand,
126 inline std::tuple<c10::MaybeOwned<Tensor>, c10::MaybeOwned<Tensor>>
128 const Tensor& tensor, in expand_inplace()
129 const Tensor& to_expand1, in expand_inplace()
130 const Tensor& to_expand2) { in expand_inplace()
134 c10::MaybeOwned<Tensor>::borrowed(to_expand1), in expand_inplace()
135 c10::MaybeOwned<Tensor>::borrowed(to_expand2)); in expand_inplace()
139 c10::MaybeOwned<Tensor>::owned(to_expand1.expand(tensor.sizes())), in expand_inplace()
140 c10::MaybeOwned<Tensor>::owned(to_expand2.expand(tensor.sizes()))); in expand_inplace()
143 inline std::tuple<c10::MaybeOwned<Tensor>, c10::MaybeOwned<Tensor>>
145 const Tensor& tensor,
146 Tensor&& to_expand1,
147 const Tensor& to_expand2) = delete;
148 inline std::tuple<c10::MaybeOwned<Tensor>, c10::MaybeOwned<Tensor>>
150 const Tensor& tensor,
151 const Tensor& to_expand1,
152 Tensor&& to_expand2) = delete;
153 inline std::tuple<c10::MaybeOwned<Tensor>, c10::MaybeOwned<Tensor>>
154 expand_inplace(const Tensor& tensor, Tensor&& to_expand1, Tensor&& to_expand2) =
157 inline std::tuple<c10::MaybeOwned<Tensor>, c10::MaybeOwned<Tensor>>
159 const Tensor& tensor, in expand_inplace()
160 const Tensor& to_expand1, in expand_inplace()
161 const Tensor& to_expand2, in expand_inplace()
167 inline std::tuple<c10::MaybeOwned<Tensor>, c10::MaybeOwned<Tensor>>
169 const Tensor& tensor,
170 Tensor&& to_expand1,
171 const Tensor& to_expand2,
173 inline std::tuple<c10::MaybeOwned<Tensor>, c10::MaybeOwned<Tensor>>
175 const Tensor& tensor,
176 const Tensor& to_expand1,
177 Tensor&& to_expand2,
179 inline std::tuple<c10::MaybeOwned<Tensor>, c10::MaybeOwned<Tensor>>
181 const Tensor& tensor,
182 Tensor&& to_expand1,
183 Tensor&& to_expand2,
187 inline std::tuple<c10::MaybeOwned<Tensor>, c10::MaybeOwned<Tensor>>
188 expand_outplace(const Tensor& to_expand1, const Tensor& to_expand2) { in expand_outplace()
193 c10::MaybeOwned<Tensor>::borrowed(to_expand1), in expand_outplace()
194 c10::MaybeOwned<Tensor>::borrowed(to_expand2)); in expand_outplace()
199 c10::MaybeOwned<Tensor>::owned(to_expand1.expand_symint(expanded_size)), in expand_outplace()
200 c10::MaybeOwned<Tensor>::owned(to_expand2.expand_symint(expanded_size))); in expand_outplace()
203 inline std::tuple<c10::MaybeOwned<Tensor>, c10::MaybeOwned<Tensor>>
204 expand_outplace(Tensor&& to_expand1, const Tensor& to_expand2) = delete;
205 inline std::tuple<c10::MaybeOwned<Tensor>, c10::MaybeOwned<Tensor>>
206 expand_outplace(const Tensor& to_expand1, Tensor&& to_expand2) = delete;
207 inline std::tuple<c10::MaybeOwned<Tensor>, c10::MaybeOwned<Tensor>>
208 expand_outplace(Tensor&& to_expand1, Tensor&& to_expand2) = delete;
210 inline std::tuple<c10::MaybeOwned<Tensor>, c10::MaybeOwned<Tensor>>
212 const Tensor& to_expand1, in expand_outplace()
213 const Tensor& to_expand2, in expand_outplace()
219 inline std::tuple<c10::MaybeOwned<Tensor>, c10::MaybeOwned<Tensor>>
221 Tensor&& to_expand1,
222 const Tensor& to_expand2,
224 inline std::tuple<c10::MaybeOwned<Tensor>, c10::MaybeOwned<Tensor>>
226 const Tensor& to_expand1,
227 Tensor&& to_expand2,
229 inline std::tuple<c10::MaybeOwned<Tensor>, c10::MaybeOwned<Tensor>>
231 Tensor&& to_expand1,
232 Tensor&& to_expand2,
236 c10::MaybeOwned<Tensor>,
237 c10::MaybeOwned<Tensor>,
238 c10::MaybeOwned<Tensor>>
240 const Tensor& to_expand1, in expand_outplace()
241 const Tensor& to_expand2, in expand_outplace()
242 const Tensor& to_expand3) { in expand_outplace()
246 c10::MaybeOwned<Tensor>::borrowed(to_expand1), in expand_outplace()
247 c10::MaybeOwned<Tensor>::borrowed(to_expand2), in expand_outplace()
248 c10::MaybeOwned<Tensor>::borrowed(to_expand3)); in expand_outplace()
256 c10::MaybeOwned<Tensor>::owned(to_expand1.expand(expanded_size)), in expand_outplace()
257 c10::MaybeOwned<Tensor>::owned(to_expand2.expand(expanded_size)), in expand_outplace()
258 c10::MaybeOwned<Tensor>::owned(to_expand3.expand(expanded_size))); in expand_outplace()
262 c10::MaybeOwned<Tensor>,
263 c10::MaybeOwned<Tensor>,
264 c10::MaybeOwned<Tensor>>
266 Tensor&& to_expand1,
267 const Tensor& to_expand2,
268 const Tensor& to_expand3) = delete;
270 c10::MaybeOwned<Tensor>,
271 c10::MaybeOwned<Tensor>,
272 c10::MaybeOwned<Tensor>>
274 const Tensor& to_expand1,
275 Tensor&& to_expand2,
276 const Tensor& to_expand3) = delete;
278 c10::MaybeOwned<Tensor>,
279 c10::MaybeOwned<Tensor>,
280 c10::MaybeOwned<Tensor>>
282 Tensor&& to_expand1,
283 Tensor&& to_expand2,
284 const Tensor& to_expand3) = delete;
286 c10::MaybeOwned<Tensor>,
287 c10::MaybeOwned<Tensor>,
288 c10::MaybeOwned<Tensor>>
290 const Tensor& to_expand1,
291 const Tensor& to_expand2,
292 Tensor&& to_expand3) = delete;
294 c10::MaybeOwned<Tensor>,
295 c10::MaybeOwned<Tensor>,
296 c10::MaybeOwned<Tensor>>
298 Tensor&& to_expand1,
299 const Tensor& to_expand2,
300 Tensor&& to_expand3) = delete;
302 c10::MaybeOwned<Tensor>,
303 c10::MaybeOwned<Tensor>,
304 c10::MaybeOwned<Tensor>>
306 const Tensor& to_expand1,
307 Tensor&& to_expand2,
308 Tensor&& to_expand3) = delete;
310 c10::MaybeOwned<Tensor>,
311 c10::MaybeOwned<Tensor>,
312 c10::MaybeOwned<Tensor>>
313 expand_outplace(Tensor&& to_expand1, Tensor&& to_expand2, Tensor&& to_expand3) =
317 c10::MaybeOwned<Tensor>,
318 c10::MaybeOwned<Tensor>,
319 c10::MaybeOwned<Tensor>>
321 const Tensor& to_expand1, in expand_outplace()
322 const Tensor& to_expand2, in expand_outplace()
323 const Tensor& to_expand3, in expand_outplace()
330 c10::MaybeOwned<Tensor>,
331 c10::MaybeOwned<Tensor>,
332 c10::MaybeOwned<Tensor>>
334 Tensor&& to_expand1,
335 const Tensor& to_expand2,
336 const Tensor& to_expand3,
339 c10::MaybeOwned<Tensor>,
340 c10::MaybeOwned<Tensor>,
341 c10::MaybeOwned<Tensor>>
343 const Tensor& to_expand1,
344 Tensor&& to_expand2,
345 const Tensor& to_expand3,
348 c10::MaybeOwned<Tensor>,
349 c10::MaybeOwned<Tensor>,
350 c10::MaybeOwned<Tensor>>
352 Tensor&& to_expand1,
353 Tensor&& to_expand2,
354 const Tensor& to_expand3,
357 c10::MaybeOwned<Tensor>,
358 c10::MaybeOwned<Tensor>,
359 c10::MaybeOwned<Tensor>>
361 const Tensor& to_expand1,
362 const Tensor& to_expand2,
363 Tensor&& to_expand3,
366 c10::MaybeOwned<Tensor>,
367 c10::MaybeOwned<Tensor>,
368 c10::MaybeOwned<Tensor>>
370 Tensor&& to_expand1,
371 const Tensor& to_expand2,
372 Tensor&& to_expand3,
375 c10::MaybeOwned<Tensor>,
376 c10::MaybeOwned<Tensor>,
377 c10::MaybeOwned<Tensor>>
379 const Tensor& to_expand1,
380 Tensor&& to_expand2,
381 Tensor&& to_expand3,
384 c10::MaybeOwned<Tensor>,
385 c10::MaybeOwned<Tensor>,
386 c10::MaybeOwned<Tensor>>
388 Tensor&& to_expand1,
389 Tensor&& to_expand2,
390 Tensor&& to_expand3,
393 inline c10::MaybeOwned<Tensor> expand_size( in expand_size()
394 const Tensor& to_expand, in expand_size()
397 return c10::MaybeOwned<Tensor>::borrowed(to_expand); in expand_size()
400 return c10::MaybeOwned<Tensor>::owned(to_expand.expand(sizes)); in expand_size()
403 inline c10::MaybeOwned<Tensor> expand_size(
404 Tensor&& to_expand,
407 inline c10::MaybeOwned<Tensor> expand_size( in expand_size()
408 const Tensor& to_expand, in expand_size()
415 inline c10::MaybeOwned<Tensor> expand_size(
416 Tensor&& to_expand,
420 inline std::vector<Tensor> expand_outplace(TensorList to_expand) { in expand_outplace()
435 std::vector<Tensor> result(to_expand.size()); in expand_outplace()
449 inline Tensor _sum_to(
450 Tensor tensor,
485 inline Tensor sum_to(
486 Tensor tensor,
494 inline Tensor sum_to(
495 Tensor tensor,