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() 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 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) {
|