Searched refs:host_input_arrays (Results 1 – 1 of 1) sorted by relevance
75 py::tuple host_input_arrays(arity); in XlaPythonGpuCallback() local79 host_input_arrays[i] = py::none(); in XlaPythonGpuCallback()84 host_input_arrays[i] = in XlaPythonGpuCallback()87 host_input_arrays[i].attr("flags").attr("writeable") = Py_False; in XlaPythonGpuCallback()90 callback->Call(host_input_arrays, status); in XlaPythonGpuCallback()