OpenGrok
Home
Sort by relevance
Sort by last modified time
Full Search
Definition
Symbol
File Path
History
|
|
Help
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