Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/
H A Dcompile_only_service.h69 Status TransferToInfeed(const TransferToInfeedRequest* arg, in TransferToInfeed() function
H A Dservice.cc1054 Status Service::TransferToInfeed(const TransferToInfeedRequest* arg, in TransferToInfeed() function in xla::Service
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/rpc/
H A Dxla_service.proto98 rpc TransferToInfeed(TransferToInfeedRequest) rpc
H A Dgrpc_service.cc104 ::grpc::Status GRPCService::TransferToInfeed(::grpc::ServerContext* context, in TransferToInfeed() function in xla::GRPCService
H A Dgrpc_stub.cc44 Status GRPCStub::TransferToInfeed(const TransferToInfeedRequest* request, in TransferToInfeed() function in xla::GRPCStub
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/pjrt/
H A Dpjrt_c_api_client.h54 Status TransferToInfeed(const LiteralSlice& literal) override { in TransferToInfeed() function
H A Dtfrt_cpu_pjrt_client.cc122 Status TfrtCpuDevice::TransferToInfeed(const LiteralSlice& literal) { in TransferToInfeed() function in xla::TfrtCpuDevice
H A Dpjrt_stream_executor_client.cc1088 Status PjRtStreamExecutorDevice::TransferToInfeed(const LiteralSlice& literal) { in TransferToInfeed() function in xla::PjRtStreamExecutorDevice
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/python/tpu_driver/client/
H A Dtpu_client.h79 Status TransferToInfeed(const LiteralSlice& literal) override { in TransferToInfeed() function
H A Dtpu_client.cc134 Status PyTpuClient::TransferToInfeed(const LiteralSlice& literal, in TransferToInfeed() function in xla::PyTpuClient
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/client/
H A Dclient.cc95 Status Client::TransferToInfeed(const LiteralSlice& literal, int64_t replica_id, in TransferToInfeed() function in xla::Client