Home
last modified time | relevance | path

Searched defs:triu (Results 1 – 6 of 6) sorted by relevance

/aosp_15_r20/external/pytorch/torch/onnx/
H A Dsymbolic_opset14.py54 def triu(g: jit_utils.GraphContext, self, diagonal, out=None): function
/aosp_15_r20/external/pytorch/aten/src/ATen/native/
H A DTriangularOps.cpp29 TORCH_META_FUNC(triu)(const Tensor& self, int64_t k) { in TORCH_META_FUNC() argument
/aosp_15_r20/external/pytorch/test/functorch/
H A Dtest_dims.py66 def triu(A): function
/aosp_15_r20/external/tensorflow/tensorflow/python/ops/numpy_ops/
H A Dnp_array_ops.py1225 def triu(m, k=0): # pylint: disable=missing-docstring function
/aosp_15_r20/external/pytorch/torch/_numpy/
H A D_funcs_impl.py773 def triu(m: ArrayLike, k=0): function
/aosp_15_r20/external/pytorch/torch/_refs/
H A D__init__.py5736 def triu(a: TensorLikeType, diagonal: int = 0) -> TensorLikeType: function