HomeSort by relevance Sort by last modified time
    Searched refs:GenericTransferManager (Results 1 - 8 of 8) sorted by null

  /external/tensorflow/tensorflow/compiler/xla/service/interpreter/
interpreter_transfer_manager.h 24 // An implementation of the XLA GenericTransferManager for interpreter backend.
25 class InterpreterTransferManager : public GenericTransferManager {
interpreter_transfer_manager.cc 29 : GenericTransferManager(sei::kInterpreterPlatformId,
  /external/tensorflow/tensorflow/compiler/xla/service/
generic_transfer_manager.cc 39 GenericTransferManager::GenericTransferManager(se::Platform::Id platform_id,
50 se::Platform::Id GenericTransferManager::PlatformId() const {
54 Status GenericTransferManager::WriteSingleTupleIndexTable(
69 GenericTransferManager::TransferLiteralFromDevice(
100 Status GenericTransferManager::TransferLiteralToDevice(
148 Status GenericTransferManager::TransferLiteralToInfeed(
153 Status GenericTransferManager::TransferBufferToInfeed(
159 Status GenericTransferManager::TransferLiteralFromOutfeed(
166 Status GenericTransferManager::ResetDevices
    [all...]
generic_transfer_manager.h 35 // inherits from GenericTransferManager and handles CPU-specific
37 class GenericTransferManager : public TransferManager {
39 GenericTransferManager(perftools::gputools::Platform::Id platform_id,
41 ~GenericTransferManager() override {}
83 TF_DISALLOW_COPY_AND_ASSIGN(GenericTransferManager);
  /external/tensorflow/tensorflow/compiler/xla/service/gpu/
gpu_transfer_manager.h 32 // An implementation of the XLA GenericTransferManager that
34 class GpuTransferManager : public GenericTransferManager {
gpu_transfer_manager.cc 44 : GenericTransferManager(
  /external/tensorflow/tensorflow/compiler/xla/service/cpu/
cpu_transfer_manager.h 33 // An implementation of the XLA GenericTransferManager that
35 class CpuTransferManager : public GenericTransferManager {
cpu_transfer_manager.cc 90 : GenericTransferManager(se::host::kHostPlatformId,

Completed in 1054 milliseconds