Home
last modified time | relevance | path

Searched full:__derive_index (Results 1 – 8 of 8) sorted by relevance

/aosp_15_r20/external/pytorch/torch/csrc/jit/runtime/
H A Dserialized_shape_function_registry.cpp44 i = torch.__derive_index(_2, 1, 1)
52 i0 = torch.__derive_index(_4, 0, 1)
1508 i = torch.__derive_index(_6, 2, 1)
1523 d = torch.__derive_index(_13, 2, 1)
1611 i = torch.__derive_index(_6, 2, 1)
1626 d = torch.__derive_index(_13, 2, 1)
1730 i = torch.__derive_index(_6, 2, 1)
1745 d = torch.__derive_index(_13, 2, 1)
1799 d = torch.__derive_index(_4, 2, 1)
1854 d = torch.__derive_index(_4, 2, 1)
[all …]
H A Dregister_prim_ops.cpp272 "aten::__derive_index(int index, int start, int step) -> int"),
/aosp_15_r20/external/pytorch/test/mobile/model_test/
H A Dcoverage.yaml26 - aten::__derive_index
720 aten::__derive_index: 24
H A Dmodel_ops.yaml21 aten::__derive_index: 106
/aosp_15_r20/external/pytorch/aten/src/ATen/core/
H A Dinterned_strings.h156 _(aten, __derive_index) \
/aosp_15_r20/external/pytorch/
H A Dpt_ops.bzl174 "aten::__derive_index",
/aosp_15_r20/external/pytorch/torch/csrc/jit/frontend/
H A Dsugared_value.cpp605 g.insert(aten::__derive_index, {idx, start_, step_}, {}, loc)); in getitem()
/aosp_15_r20/external/pytorch/torch/onnx/
H A Dsymbolic_opset9.py5944 @_onnx_symbolic("aten::__derive_index")
5945 def __derive_index(g: jit_utils.GraphContext, index, start, step): function