Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/python/
Dpytree.h46 pybind11::function to_iterable; member
53 static void Register(pybind11::object type, pybind11::function to_iterable,
Dpytree.cc47 py::function to_iterable, in Register() argument
52 registration->to_iterable = std::move(to_iterable); in Register()
148 py::tuple out = py::cast<py::tuple>(node.custom->to_iterable(handle)); in FlattenInto()
407 py::tuple out = py::cast<py::tuple>(node.custom->to_iterable(object)); in FlattenUpTo()
651 pytree.def("register_node", [](py::object type, py::function to_iterable, in BuildPytreeSubmodule()
653 return CustomNodeRegistry::Register(type, to_iterable, from_iterable); in BuildPytreeSubmodule()