| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | outfeed_in_nested_computation_test.cc | 90 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/ |
| D | grpc_service.cc | 112 ::grpc::Status GRPCService::TransferFromOutfeed( in TransferFromOutfeed() function in xla::GRPCService
116 return service_->TransferFromOutfeed(arg, result); in TransferFromOutfeed()
|
| D | grpc_stub.cc | 51 Status GRPCStub::TransferFromOutfeed(const TransferFromOutfeedRequest* request, in TransferFromOutfeed() function in xla::GRPCStub
54 return grpc_stub_->TransferFromOutfeed(context, *request, response); in TransferFromOutfeed()
|
| D | grpc_stub.h | 40 Status TransferFromOutfeed(const TransferFromOutfeedRequest* arg,
|
| D | grpc_service.h | 73 ::grpc::Status TransferFromOutfeed(
|
| D | xla_service.proto | 103 rpc TransferFromOutfeed(TransferFromOutfeedRequest) rpc
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | compile_only_service.h | 80 Status TransferFromOutfeed(const TransferFromOutfeedRequest* arg, in TransferFromOutfeed() function
|
| D | service.h | 150 Status TransferFromOutfeed(const TransferFromOutfeedRequest* arg,
|
| D | service.cc | 1015 Status Service::TransferFromOutfeed(const TransferFromOutfeedRequest* arg, in TransferFromOutfeed() function in xla::Service
|
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | service_interface.h | 44 virtual Status TransferFromOutfeed(const TransferFromOutfeedRequest* arg,
|
| /external/tensorflow/tensorflow/compiler/xla/python/tpu_driver/client/ |
| D | tpu_client.h | 72 Status TransferFromOutfeed(MutableBorrowingLiteral literal) override { in TransferFromOutfeed() function
104 StatusOr<Literal> TransferFromOutfeed(const Shape& shape, int device_id);
|
| D | tpu_client_extension.cc | 87 TF_ASSIGN_OR_RETURN(Literal literal, client->TransferFromOutfeed( in PYBIND11_MODULE()
|
| D | tpu_client.cc | 128 StatusOr<Literal> PyTpuClient::TransferFromOutfeed(const Shape& shape, in TransferFromOutfeed() function in xla::PyTpuClient
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | client.h | 160 StatusOr<Literal> TransferFromOutfeed(
|
| D | client.cc | 118 StatusOr<Literal> Client::TransferFromOutfeed( in TransferFromOutfeed() function in xla::Client
133 Status s = stub_->TransferFromOutfeed(&request, &response); in TransferFromOutfeed()
|
| /external/tensorflow/tensorflow/compiler/xla/pjrt/ |
| D | pjrt_client.h | 97 virtual Status TransferFromOutfeed(MutableBorrowingLiteral literal) = 0;
|
| D | pjrt_stream_executor_client.h | 110 Status TransferFromOutfeed(MutableBorrowingLiteral literal) override;
|
| D | pjrt_stream_executor_client.cc | 979 Status PjRtStreamExecutorDevice::TransferFromOutfeed( in TransferFromOutfeed() function in xla::PjRtStreamExecutorDevice
|
| /external/tensorflow/tensorflow/compiler/xla/python/ |
| D | outfeed_receiver.cc | 345 TF_RETURN_IF_ERROR(device->TransferFromOutfeed(literal.get())); in ReceiveRawFromOutfeed()
|
| D | xla.cc | 149 TF_RETURN_IF_ERROR(device.TransferFromOutfeed(literal.get())); in PYBIND11_MODULE()
|