Lines Matching defs:Tensor

400 def _reduction_identity(op_name: str, input: Tensor, *args):
485 def _sparse_coo_flatten_indices(indices: Tensor, shape: tuple):
494 def _any(input: Tensor, dim: tuple, keepdim: bool):
503 def _sparse_coo_where(mask: Tensor, input: Tensor, fill_value: Tensor) -> Tensor:
617     mask_input: Tensor,
737     mask_input: Tensor,
821 def _sparse_csr_where(mask: Tensor, input: Tensor, fill_value: Tensor) -> Tensor:
829 def _where(mask: Tensor, input: Tensor, fill_value: Tensor) -> Tensor:
865 def _input_mask(input: Union[Tensor, MaskedTensor], *args, **kwargs) -> Tensor:
950 def _output_mask(op, input: Tensor, *args, **kwargs) -> Tensor:
997     op, input: Union[MaskedTensor, Tensor], mask, *args  argument
1040     input: Union[Tensor, MaskedTensor],
1045     mask: Optional[Tensor] = None,
1093     input: Union[Tensor, MaskedTensor],
1098     mask: Optional[Tensor] = None,
1170     input: Tensor,
1174     mask: Optional[Tensor] = None,
1190     input: Tensor,
1194     mask: Optional[Tensor] = None,
1210     input: Union[Tensor, MaskedTensor],
1215     mask: Optional[Tensor] = None,
1260     input: Union[Tensor, MaskedTensor],
1265     mask: Optional[Tensor] = None,
1310     input: Union[Tensor, MaskedTensor],
1315     mask: Optional[Tensor] = None,
1336     input: Union[Tensor, MaskedTensor],
1341     mask: Optional[Tensor] = None,
1362     input: Union[Tensor, MaskedTensor],
1367     mask: Optional[Tensor] = None,
1418     input: Union[Tensor, MaskedTensor],
1423     mask: Optional[Tensor] = None,
1461     input: Tensor,
1466     mask: Optional[Tensor] = None,
1482     input: Union[Tensor, MaskedTensor],
1483     other: Union[Tensor, MaskedTensor],
1486     input_mask: Optional[Tensor] = None,
1487     other_mask: Optional[Tensor] = None,
1544     input: Union[Tensor, MaskedTensor],
1550     mask: Optional[Tensor] = None,
1579     input: Union[Tensor, MaskedTensor],
1586     mask: Optional[Tensor],
1652     input: Union[Tensor, MaskedTensor],
1659     mask: Optional[Tensor] = None,
1683     input: Union[Tensor, MaskedTensor],
1690     mask: Optional[Tensor] = None,
1714     input: Union[Tensor, MaskedTensor],
1718     mask: Optional[Tensor] = None,
1734     input: Union[Tensor, MaskedTensor],
1738     mask: Optional[Tensor] = None,
1754     input: Union[Tensor, MaskedTensor],
1758     mask: Optional[Tensor] = None,
1774     input: Union[Tensor, MaskedTensor],
1780     mask: Optional[Tensor] = None,