Searched defs:TransferLiteralFromDevice (Results 1 – 5 of 5) sorted by relevance
80 StatusOr<Literal> TransferLiteralFromDevice( in TransferLiteralFromDevice() function88 Status TransferLiteralFromDevice(se::Stream* stream, in TransferLiteralFromDevice() function110 void TransferLiteralFromDevice(se::Stream* stream, in TransferLiteralFromDevice() function
61 void GenericTransferManager::TransferLiteralFromDevice( in TransferLiteralFromDevice() function in xla::GenericTransferManager
50 StatusOr<Literal> TransferManager::TransferLiteralFromDevice( in TransferLiteralFromDevice() function in xla::TransferManager77 Status TransferManager::TransferLiteralFromDevice( in TransferLiteralFromDevice() function in xla::TransferManager
83 StatusOr<Literal> HloRunner::TransferLiteralFromDevice( in TransferLiteralFromDevice() function in xla::HloRunner
190 void TpuTransferManager::TransferLiteralFromDevice( in TransferLiteralFromDevice() function in tensorflow::tpu::TpuTransferManager