Searched refs:device_oridinal (Results 1 – 2 of 2) sorted by relevance
314 const ::xla::BorrowingLiteral& literal, int device_oridinal) { in TransferToLocalServer() argument320 shape, backend().memory_allocator(), device_oridinal)); in TransferToLocalServer()322 mutable_backend()->BorrowStream(device_oridinal)); in TransferToLocalServer()
134 const ::xla::BorrowingLiteral& literal, int device_oridinal);