Searched refs:python_fallback (Results 1 – 1 of 1) sorted by relevance
112 const py::function& python_fallback) { in ShardArg() argument148 py::cast<py::list>(python_fallback(arg, py_devices, input_spec.indices)); in ShardArg()