Home
last modified time | relevance | path

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

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