Home
last modified time | relevance | path

Searched refs:tuple_slice (Results 1 – 6 of 6) sorted by relevance

/aosp_15_r20/external/pytorch/torch/csrc/jit/python/
H A Dpybind_utils.h848 struct VISIBILITY_HIDDEN tuple_slice { struct
849 /*implicit*/ tuple_slice(py::tuple tup_) in tuple_slice() function
851 tuple_slice(py::tuple tup_, int64_t b_) in tuple_slice() argument
853 tuple_slice(py::tuple tup_, int64_t b_, int64_t e_) in tuple_slice() function
901 const tuple_slice& args, in matchSchemaAllowFakeScriptObject()
986 const tuple_slice& args, in createStackForSchema()
1101 const tuple_slice& args, in runAndInsertCall()
1161 const tuple_slice& args_no_self, in maybeTorchFunctionDispatch()
1211 const tuple_slice& args, in invokeScriptFunctionFromPython()
1228 const tuple_slice& args, in invokeScriptMethodFromPython()
H A Dpython_custom_class.cpp51 callee, tuple_slice(std::move(args), 1), kwargs); in initPythonCustomClassBindings()
H A Dscript_init.cpp1441 callee, tuple_slice(std::move(args), 1), kwargs); in initJitScriptBindings()
1547 method, tuple_slice(std::move(args), 1), kwargs); in initJitScriptBindings()
/aosp_15_r20/external/pytorch/test/jit/
H A Dtest_slice.py131 def tuple_slice(a): function
140 self.checkScript(tuple_slice, (torch.tensor([1]),), optimize=True)
141 scripted_fn = torch.jit.script(tuple_slice)
/aosp_15_r20/external/pytorch/c10/test/util/
H A DMetaprogramming_test.cpp162 auto y = tuple_slice<decltype(x), 1, 2>(x); in TEST()
169 auto y = tuple_slice<decltype(x), 0, 3>(x); in TEST()
/aosp_15_r20/external/pytorch/c10/util/
H A DMetaprogramming.h164 constexpr auto tuple_slice(Tuple t) {