Searched refs:tuple_slice (Results 1 – 6 of 6) sorted by relevance
| /aosp_15_r20/external/pytorch/torch/csrc/jit/python/ |
| H A D | pybind_utils.h | 848 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() function
853 tuple_slice(py::tuple tup_, int64_t b_, int64_t e_) in tuple_slice() argument
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 D | python_custom_class.cpp | 51 callee, tuple_slice(std::move(args), 1), kwargs); in initPythonCustomClassBindings()
|
| H A D | script_init.cpp | 1441 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 D | test_slice.py | 131 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 D | Metaprogramming_test.cpp | 162 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 D | Metaprogramming.h | 164 constexpr auto tuple_slice(Tuple t) {
|