Searched refs:_wrap_functional_tensor (Results 1 – 4 of 4) sorted by relevance
28 def _wrap_functional_tensor(tensor: Tensor, level: int) -> Tensor: ...
42 Tensor _wrap_functional_tensor(const Tensor& self, int64_t level) { in _wrap_functional_tensor() function454 &_wrap_functional_tensor, in initFuncTorchBindings()
27 _wrap_functional_tensor,1462 wrapped = _wrap_functional_tensor(maybe_tensor, level)
29 _wrap_functional_tensor,1271 r = _wrap_functional_tensor(ft, t.current_level)