Home
last modified time | relevance | path

Searched refs:TransferFromOutfeed (Results 1 – 16 of 16) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/tests/
Doutfeed_in_nested_computation_test.cc90 local_client_->TransferFromOutfeed(&int_shape)); in XLA_TEST_F()
103 local_client_->TransferFromOutfeed(&xfeed_shape)); in XLA_TEST_F()
111 local_client_->TransferFromOutfeed(&int_shape)); in XLA_TEST_F()
160 local_client_->TransferFromOutfeed(&result_shape)); in XLA_TEST_F()
/external/tensorflow/tensorflow/compiler/xla/rpc/
Dgrpc_service.cc112 ::grpc::Status GRPCService::TransferFromOutfeed( in TransferFromOutfeed() function in xla::GRPCService
116 return service_->TransferFromOutfeed(arg, result); in TransferFromOutfeed()
Dgrpc_stub.cc51 Status GRPCStub::TransferFromOutfeed(const TransferFromOutfeedRequest* request, in TransferFromOutfeed() function in xla::GRPCStub
54 return grpc_stub_->TransferFromOutfeed(context, *request, response); in TransferFromOutfeed()
Dgrpc_stub.h40 Status TransferFromOutfeed(const TransferFromOutfeedRequest* arg,
Dgrpc_service.h73 ::grpc::Status TransferFromOutfeed(
Dxla_service.proto103 rpc TransferFromOutfeed(TransferFromOutfeedRequest) rpc
/external/tensorflow/tensorflow/compiler/xla/service/
Dcompile_only_service.h80 Status TransferFromOutfeed(const TransferFromOutfeedRequest* arg, in TransferFromOutfeed() function
Dservice.h150 Status TransferFromOutfeed(const TransferFromOutfeedRequest* arg,
Dservice.cc1038 Status Service::TransferFromOutfeed(const TransferFromOutfeedRequest* arg, in TransferFromOutfeed() function in xla::Service
/external/tensorflow/tensorflow/compiler/xla/
Dservice_interface.h44 virtual Status TransferFromOutfeed(const TransferFromOutfeedRequest* arg,
/external/tensorflow/tensorflow/compiler/xla/client/
Dclient.h160 StatusOr<Literal> TransferFromOutfeed(
Dclient.cc118 StatusOr<Literal> Client::TransferFromOutfeed( in TransferFromOutfeed() function in xla::Client
133 Status s = stub_->TransferFromOutfeed(&request, &response); in TransferFromOutfeed()
/external/tensorflow/tensorflow/compiler/xla/python/
Dlocal_computation_builder.cc90 StatusOr<Literal> LocalClient::TransferFromOutfeed(const Shape& shape, in TransferFromOutfeed() function in xla::swig::LocalClient
94 return client_->TransferFromOutfeed(&shape, device_ordinal); in TransferFromOutfeed()
Dlocal_computation_builder.h60 StatusOr<Literal> TransferFromOutfeed(const Shape& shape, int device_ordinal);
Dlocal_computation_builder.i263 %unignore xla::swig::LocalClient::TransferFromOutfeed;
Dxla_client.py703 return backend.client.TransferFromOutfeed(shape, device_ordinal)