Searched defs:maybe_tensor (Results 1 – 4 of 4) sorted by relevance
59 bool isBatchedAtLevel(const std::optional<Tensor>& maybe_tensor, int64_t level) { in isBatchedAtLevel()77 const auto& maybe_tensor = maybe_tensors.get(idx); in isBatchedAtLevel() local86 for (const auto& maybe_tensor : maybe_tensors) { in areAnyBatchedAtLevel() local
48 const auto& maybe_tensor = indices[i]; in get_max_index_logical_dim() local
135 def _is_differentiable(maybe_tensor): argument146 def _wrap_tensor_for_grad(maybe_tensor, level): argument1458 maybe_tensor, level, *, _python_functionalize: bool = False argument1484 def _maybe_unwrap_functional_tensor(maybe_tensor, *, reapply_views: bool): argument
60 def move_tensor(maybe_tensor): argument