Searched defs:py_devices (Results 1 – 1 of 1) sorted by relevance
111 const InputSpec& input_spec, py::handle py_devices, in ShardArg()171 py::object py_devices; // To pass back to Python. member