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