Searched refs:device_array (Results 1 – 1 of 1) sorted by relevance
388 const auto& device_array = xla_module.attr("_DeviceArray"); in ArgSignatureOfValue() local456 (*p)[device_array.ptr()] = device_array_handler; in ArgSignatureOfValue()752 const auto& device_array = xla_module.attr("_DeviceArray"); in DevicePut() local764 (*p)[device_array.ptr()] = HandleDeviceArray; in DevicePut()954 const auto& device_array = xla_module->attr("_DeviceArray"); in ConvertArgsToBuffers() local970 arg.get_type().is(device_array)) { in ConvertArgsToBuffers()1197 static const auto* device_array = in Call() local1200 (*device_array)(out_avals[i], sticky_device, out_lazy_exprs[i], in Call()