Home
last modified time | relevance | path

Searched refs:XLA_TransferManager (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/stream_executor/tpu/
Dtpu_executor_c_api.h176 XLA_TransferManager* TpuTransferManager_New();
177 void TpuTransferManager_Free(XLA_TransferManager* manager);
178 SE_PlatformId TpuTransferManager_PlatformId(XLA_TransferManager* manager);
179 void TpuTransferManager_HostShapeToDeviceShape(XLA_TransferManager* manager,
183 XLA_TransferManager* manager, SE_Stream* stream, XLA_Literal* literal,
186 XLA_TransferManager* manager, SE_Stream* stream,
189 int64_t TpuTransferManager_GetByteSizeRequirement(XLA_TransferManager* manager,
192 XLA_TransferManager* manager, XLA_Shape* host_shape, XLA_Shape* output,
195 XLA_TransferManager* manager, SE_StreamExecutor* executor,
198 XLA_TransferManager* manager, SE_StreamExecutor* executor,
[all …]
Dc_api_decl.h306 typedef struct XLA_TransferManager XLA_TransferManager; typedef
Dc_api_defn.h67 struct XLA_TransferManager {}; struct
Dtpu_transfer_manager.h92 XLA_TransferManager* manager_;