Home
last modified time | relevance | path

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

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