Searched refs:PythonOpNode (Results 1 – 1 of 1) sorted by relevance
413 auto inlineAutograd = [&](Node* PythonOpNode) { in NodeToONNX() argument414 for (auto subblock : PythonOpNode->blocks()) { in NodeToONNX()415 for (const auto i : c10::irange(PythonOpNode->inputs().size())) { in NodeToONNX()416 auto py_value = env[py::cast(PythonOpNode->inputs()[i])]; in NodeToONNX()423 for (const auto i : c10::irange(PythonOpNode->outputs().size())) { in NodeToONNX()425 env[py::cast(PythonOpNode->outputs()[i])] = py_value; in NodeToONNX()