Home
last modified time | relevance | path

Searched defs:MaskedTensor (Results 1 – 5 of 5) sorted by relevance

/aosp_15_r20/external/pytorch/torch/masked/
H A D_ops.py865 def _input_mask(input: Union[Tensor, MaskedTensor], *args, **kwargs) -> Tensor: argument
997 op, input: Union[MaskedTensor, Tensor], mask, *args
1040 input: Union[Tensor, MaskedTensor], argument
1093 input: Union[Tensor, MaskedTensor], argument
1210 input: Union[Tensor, MaskedTensor], argument
1260 input: Union[Tensor, MaskedTensor], argument
1310 input: Union[Tensor, MaskedTensor], argument
1336 input: Union[Tensor, MaskedTensor], argument
1362 input: Union[Tensor, MaskedTensor], argument
1418 input: Union[Tensor, MaskedTensor], argument
[all …]
/aosp_15_r20/external/pytorch/docs/source/
H A Dmasked.rst59 …MaskedTensors and why they're useful: https://pytorch.org/tutorials/prototype/maskedtensor_overview
60 …se COO and CSR data and mask Tensors: https://pytorch.org/tutorials/prototype/maskedtensor_sparsity
61 … sparse semantics and implementations: https://pytorch.org/tutorials/prototype/maskedtensor_adagrad
62 …for binary/reduction operations), differences with NumPy's MaskedArray, and reduction semantics: h…
/aosp_15_r20/external/tensorflow/tensorflow/python/util/
H A Ddispatch_test.py334 class MaskedTensor(extension_type.ExtensionType): class
521 def masked_add(x: MaskedTensor, y: MaskedTensor, name=None):
817 def masked_add(x: MaskedTensor, y: MaskedTensor):
/aosp_15_r20/external/pytorch/torch/masked/maskedtensor/
H A Dcore.py144 class MaskedTensor(torch.Tensor): class
/aosp_15_r20/external/tensorflow/tensorflow/python/eager/
H A Ddef_function_test.py328 class MaskedTensor(extension_type.ExtensionType): class