Searched defs:data_device (Results 1 – 1 of 1) sorted by relevance
697 xla::PjRtDevice* data_device = nullptr; in ComputeSignature() local741 xla::PjRtDevice* data_device = arguments.signature.device; in CopyBuffersToDevice() local