Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/torch/csrc/autograd/
H A Dpython_function.cpp1221 THPObjectPtr ctx_input_tuple(PyTuple_New(num_args + 1)); in make_ctx_input_tuple() local
1222 if (!ctx_input_tuple) in make_ctx_input_tuple()
1225 PyTuple_SET_ITEM(ctx_input_tuple.get(), 0, (PyObject*)ctx); in make_ctx_input_tuple()
1229 PyTuple_SET_ITEM(ctx_input_tuple.get(), i + 1, arg); in make_ctx_input_tuple()
1231 return ctx_input_tuple; in make_ctx_input_tuple()
1370 auto ctx_input_tuple = in THPFunction_apply() local
1372 if (!ctx_input_tuple) { in THPFunction_apply()
1375 output = PyObject_CallObject(forward_fn, ctx_input_tuple); in THPFunction_apply()