Searched defs:maybe_wrap_dim (Results 1 – 5 of 5) sorted by relevance
17 inline int64_t maybe_wrap_dim(int64_t dim, TensorImpl* tensor) { in maybe_wrap_dim() function21 inline int64_t maybe_wrap_dim(int64_t dim, TensorList tensors) { in maybe_wrap_dim() function30 inline int64_t maybe_wrap_dim( in maybe_wrap_dim() function
124 def maybe_wrap_dim(self, dim: int, n: int) -> List[int]: member in SliceScatterVisitor
196 def maybe_wrap_dim(dim: int, n: int) -> List[int]: function
928 def maybe_wrap_dim(dim: int, dim_post_expr: int, wrap_scalar: bool = True): function
4848 def maybe_wrap_dim(dim: int, dim_post_expr: int, wrap_scalar: bool = True): function