Home
last modified time | relevance | path

Searched refs:is_empty_tensor (Results 1 – 3 of 3) sorted by relevance

/external/pytorch/aten/src/ATen/native/mps/operations/
DUnaryOps.mm72 static bool is_empty_tensor(const Tensor& self) {
130 is_noop_p is_noop = is_empty_tensor) {
/external/pytorch/torch/_inductor/fx_passes/
Dsplit_cat.py322 def is_empty_tensor(x): function
328 ndim == x.meta["example_value"].dim() or is_empty_tensor(x) for x in tensors
1596 def is_empty_tensor(x: torch.fx.Node) -> bool: function
1601 assert all(ndim == x.meta["val"].dim() or is_empty_tensor(x) for x in tensors)
/external/pytorch/aten/src/ATen/functorch/
DBatchRulesNorm.cpp14 static bool is_empty_tensor(const Tensor& tensor) { in is_empty_tensor() function
25 if (input_bdim.has_value() && !is_empty_tensor(stat)) { in compute_stat_bdim()