Home
last modified time | relevance | path

Searched defs:TransferFromOutfeed (Results 1 – 11 of 11) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dcompile_only_service.h74 Status TransferFromOutfeed(const TransferFromOutfeedRequest* arg, in TransferFromOutfeed() function
Dservice.cc1083 Status Service::TransferFromOutfeed(const TransferFromOutfeedRequest* arg, in TransferFromOutfeed() function in xla::Service
/external/tensorflow/tensorflow/compiler/xla/rpc/
Dxla_service.proto103 rpc TransferFromOutfeed(TransferFromOutfeedRequest) rpc
Dgrpc_service.cc112 ::grpc::Status GRPCService::TransferFromOutfeed( in TransferFromOutfeed() function in xla::GRPCService
Dgrpc_stub.cc51 Status GRPCStub::TransferFromOutfeed(const TransferFromOutfeedRequest* request, in TransferFromOutfeed() function in xla::GRPCStub
/external/tensorflow/tensorflow/compiler/xla/pjrt/
Dpjrt_c_api_client.h61 Status TransferFromOutfeed(MutableBorrowingLiteral literal) override { in TransferFromOutfeed() function
Dtfrt_cpu_pjrt_client.cc126 Status TfrtCpuDevice::TransferFromOutfeed(MutableBorrowingLiteral literal) { in TransferFromOutfeed() function in xla::TfrtCpuDevice
Dpjrt_stream_executor_client.cc1095 Status PjRtStreamExecutorDevice::TransferFromOutfeed( in TransferFromOutfeed() function in xla::PjRtStreamExecutorDevice
/external/tensorflow/tensorflow/compiler/xla/python/tpu_driver/client/
Dtpu_client.h83 Status TransferFromOutfeed(MutableBorrowingLiteral literal) override { in TransferFromOutfeed() function
Dtpu_client.cc139 StatusOr<Literal> PyTpuClient::TransferFromOutfeed(const Shape& shape, in TransferFromOutfeed() function in xla::PyTpuClient
/external/tensorflow/tensorflow/compiler/xla/client/
Dclient.cc117 StatusOr<Literal> Client::TransferFromOutfeed( in TransferFromOutfeed() function in xla::Client