Home
last modified time | relevance | path

Searched refs:transfer_metadata (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dtransfer_manager.cc49 const TransferMetadata* transfer_metadata) { in TransferLiteralFromDevice() argument
66 transfer_metadata); in TransferLiteralFromDevice()
77 const TransferMetadata* transfer_metadata) { in TransferLiteralFromDevice() argument
90 transfer_metadata); in TransferLiteralFromDevice()
98 const TransferMetadata* transfer_metadata) { in TransferLiteralToDevice() argument
107 substream, literal, device_buffer, transfer_metadata)); in TransferLiteralToDevice()
113 const TransferMetadata* transfer_metadata) { in TransferArrayFromDevice() argument
131 transfer_metadata); in TransferArrayFromDevice()
142 const TransferMetadata* transfer_metadata) { in TransferArrayToDevice() argument
150 TransferArrayToDeviceAsync(substream, literal, dest, transfer_metadata)); in TransferArrayToDevice()
[all …]
Dtransfer_manager.h73 const TransferMetadata* transfer_metadata);
81 const TransferMetadata* transfer_metadata);
103 const TransferMetadata* transfer_metadata) = 0;
126 const TransferMetadata* transfer_metadata);
152 const TransferMetadata* transfer_metadata) = 0;
169 const TransferMetadata* transfer_metadata = nullptr);
174 const TransferMetadata* transfer_metadata = nullptr);
179 const TransferMetadata* transfer_metadata = nullptr);
183 const TransferMetadata* transfer_metadata = nullptr);
Dgeneric_transfer_manager.h46 const TransferMetadata* transfer_metadata) override;
51 const TransferMetadata* transfer_metadata) override;