| /external/tensorflow/tensorflow/compiler/xla/service/cpu/tests/ |
| D | cpu_infeed_test.cc | 48 ASSERT_IS_OK(client_->TransferToInfeed(literal)); in TestInfeedRoundTrip()
160 client_->TransferToInfeed(LiteralUtil::CreateR1<float>({1, 2, 3}))); in TEST_F()
162 client_->TransferToInfeed(LiteralUtil::CreateR1<float>({4, 5, 6}))); in TEST_F()
164 client_->TransferToInfeed(LiteralUtil::CreateR1<float>({7, 8, 9}))); in TEST_F()
166 client_->TransferToInfeed(LiteralUtil::CreateR1<float>({10, 11, 12}))); in TEST_F()
168 client_->TransferToInfeed(LiteralUtil::CreateR1<float>({13, 14, 15}))); in TEST_F()
252 ASSERT_IS_OK(client_->TransferToInfeed( in TEST_F()
255 ASSERT_IS_OK(client_->TransferToInfeed( in TEST_F()
258 ASSERT_IS_OK(client_->TransferToInfeed( in TEST_F()
261 ASSERT_IS_OK(client_->TransferToInfeed( in TEST_F()
[all …]
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | outfeed_in_nested_computation_test.cc | 84 local_client_->TransferToInfeed(LiteralUtil::CreateR0<int32_t>(1))); in XLA_TEST_F()
96 TF_ASSERT_OK(local_client_->TransferToInfeed( in XLA_TEST_F()
157 local_client_->TransferToInfeed(LiteralUtil::CreateR0<bool>(true))); in XLA_TEST_F()
|
| D | while_test.cc | 1255 TF_ASSERT_OK(client_->TransferToInfeed(LiteralUtil::CreateR0<bool>(true))); in XLA_TEST_F()
1256 TF_ASSERT_OK(client_->TransferToInfeed(LiteralUtil::CreateR0<bool>(true))); in XLA_TEST_F()
1257 TF_ASSERT_OK(client_->TransferToInfeed(LiteralUtil::CreateR0<bool>(false))); in XLA_TEST_F()
|
| /external/tensorflow/tensorflow/compiler/xla/rpc/ |
| D | grpc_service.cc | 104 ::grpc::Status GRPCService::TransferToInfeed(::grpc::ServerContext* context, in TransferToInfeed() function in xla::GRPCService
108 return service_->TransferToInfeed(arg, result); in TransferToInfeed()
|
| D | grpc_stub.cc | 44 Status GRPCStub::TransferToInfeed(const TransferToInfeedRequest* request, in TransferToInfeed() function in xla::GRPCStub
47 return grpc_stub_->TransferToInfeed(context, *request, response); in TransferToInfeed()
|
| D | grpc_stub.h | 37 Status TransferToInfeed(const TransferToInfeedRequest* arg,
|
| D | grpc_service.h | 69 ::grpc::Status TransferToInfeed(::grpc::ServerContext* context,
|
| D | xla_service.proto | 98 rpc TransferToInfeed(TransferToInfeedRequest) rpc
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | compile_only_service.h | 75 Status TransferToInfeed(const TransferToInfeedRequest* arg, in TransferToInfeed() function
|
| D | service.h | 145 Status TransferToInfeed(const TransferToInfeedRequest* arg,
|
| D | service.cc | 986 Status Service::TransferToInfeed(const TransferToInfeedRequest* arg, in TransferToInfeed() function in xla::Service
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/tests/ |
| D | infeed_test.cc | 42 ASSERT_IS_OK(client_->TransferToInfeed(literal)); in TestInfeedRoundTrip()
|
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | service_interface.h | 41 virtual Status TransferToInfeed(const TransferToInfeedRequest* arg,
|
| /external/tensorflow/tensorflow/compiler/xla/python/tpu_driver/client/ |
| D | tpu_client.h | 68 Status TransferToInfeed(const LiteralSlice& literal) override { in TransferToInfeed() function
103 Status TransferToInfeed(const LiteralSlice& literal, int device_id);
|
| D | tpu_client_extension.cc | 78 return client->TransferToInfeed(literal, device_ordinal); in PYBIND11_MODULE()
|
| D | tpu_client.cc | 123 Status PyTpuClient::TransferToInfeed(const LiteralSlice& literal, in TransferToInfeed() function in xla::PyTpuClient
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | client.h | 152 Status TransferToInfeed(const LiteralSlice& literal, int64 replica_id = 0,
|
| D | client.cc | 96 Status Client::TransferToInfeed(const LiteralSlice& literal, int64 replica_id, in TransferToInfeed() function in xla::Client
108 Status s = stub_->TransferToInfeed(&request, &response); in TransferToInfeed()
|
| /external/tensorflow/tensorflow/compiler/xla/pjrt/ |
| D | pjrt_client.h | 94 virtual Status TransferToInfeed(const LiteralSlice& literal) = 0;
|
| D | pjrt_stream_executor_client.h | 108 Status TransferToInfeed(const LiteralSlice& literal) override;
|
| D | pjrt_stream_executor_client.cc | 972 Status PjRtStreamExecutorDevice::TransferToInfeed(const LiteralSlice& literal) { in TransferToInfeed() function in xla::PjRtStreamExecutorDevice
|
| /external/tensorflow/tensorflow/compiler/xla/python/ |
| D | xla.cc | 133 return device.TransferToInfeed(literal); in PYBIND11_MODULE()
|
| /external/tensorflow/tensorflow/compiler/xla/tools/ |
| D | replay_computation.cc | 271 TF_CHECK_OK(client->TransferToInfeed(*infeed_data)); in ReplayComputation()
|