Searched refs:forward_pass_name_scope (Results 1 – 4 of 4) sorted by relevance
128 out_grads, skip_input_indices, forward_pass_name_scope): argument154 if forward_pass_name_scope:155 gradient_name_scope += forward_pass_name_scope + "/"
285 PyObject* forward_pass_name_scope);
3226 PyObject* forward_pass_name_scope = nullptr) { argument3307 if (!forward_pass_name_scope) forward_pass_name_scope = Py_None;3312 forward_pass_name_scope]() {3318 Py_INCREF(forward_pass_name_scope);3321 forward_pass_name_scope](3343 forward_pass_name_scope));3355 forward_pass_name_scope](PyBackwardFunction* backward_function) {3361 Py_DECREF(forward_pass_name_scope);4016 PyObject* forward_pass_name_scope) { argument4022 forward_pass_name_scope);
1031 const py::handle& forward_pass_name_scope) { in PYBIND11_MODULE() argument1034 forward_pass_name_scope.ptr())); in PYBIND11_MODULE()