Searched defs:py_args (Results 1 – 4 of 4) sorted by relevance
57 auto py_args = py::tuple(args.size()); in ConvertDatatoPyTuple() local82 auto py_args = ConvertDatatoPyTuple(args); in RunComputeFunctionWithoutPyObj() local95 auto py_args = ConvertDatatoPyTuple(args); in RunComputeFunction() local
92 py::tuple check_bprop_out(const py::object &grads_obj, const py::tuple &py_args) { in check_bprop_out()270 py::tuple py_args = ConvertDatatoPyTuple(args); in RunHookFunction() local325 auto py_args = ConvertDatatoPyTuple(args); in RunComputeFunction() local
517 py::tuple py_args(size_args); in PreparePyInputs() local584 auto py_args = PreparePyInputs(prim_py, args); in RunPyInferValue() local618 auto py_args = PreparePyInputs(prim_py, args); in EvalPyCheckPrim() local688 auto py_args = PreparePyInputs(prim_py_, args); in EvalPrim() local
220 void GetDstType(const py::tuple &py_args, in GetDstType()