Searched defs:device_array (Results 1 – 1 of 1) sorted by relevance
388 const auto& device_array = xla_module.attr("_DeviceArray"); in ArgSignatureOfValue() local752 const auto& device_array = xla_module.attr("_DeviceArray"); in DevicePut() local954 const auto& device_array = xla_module->attr("_DeviceArray"); in ConvertArgsToBuffers() local1197 static const auto* device_array = in Call() local