Searched defs:py_args (Results 1 – 7 of 7) sorted by relevance
26 py::tuple py_args(kIndex1); in GetPythonArg() local86 auto py_args = GetPythonArg(grad); in operator ()() local
65 auto py_args = py::tuple(args.size()); in ConvertDatatoPyTuple() local92 auto py_args = ConvertDatatoPyTuple(args); in RunComputeFunctionWithoutPyObj() local105 auto py_args = ConvertDatatoPyTuple(args); in RunComputeFunction() local
207 void check_bprop_input_grads(const py::tuple &py_args, const py::tuple &grads, const std::string &b… in check_bprop_input_grads()249 py::tuple check_bprop_out(const py::object &grads_obj, const py::tuple &py_args, const std::string … in check_bprop_out()479 BaseRef PrimitivePy::RunVariableHookFunction(const py::tuple &py_args, bool is_tensor_hook) const { in RunVariableHookFunction() argument521 py::tuple py_args = ConvertDatatoPyTuple(args); in RunHookFunction() local593 auto py_args = ConvertDatatoPyTuple(args); in RunComputeFunction() local
133 auto py_args = PreparePyInputs(input_args); in InferShapeAndType() local
342 def _call_new_python(self, context, *py_args, **kwargs):
1086 PyObject *py_args = args.ptr(); in CallGraph() local1093 … PyObject *py_kwvargs) { in CallGraph()
1031 py::tuple py_args(args_size); in PreparePyInputs() local1071 auto py_args = PreparePyInputs(args); in RunPyInferValue() local1137 auto py_args = PreparePyInputs(args); in EvalPyCheckPrim() local1395 auto py_args = PreparePyInputs(args); in EvalPrim() local3052 auto py_args = PreparePyInputs(args_abs_list); in EvalPrim() local