Searched refs:mutable_backend (Results 1 – 14 of 14) sorted by relevance
355 Backend* LocalClient::mutable_backend() { in mutable_backend() function in xla::LocalClient356 return local_service_->mutable_backend(); in mutable_backend()402 std::move(executable), local_service_->mutable_backend(), in Compile()442 local_service_->mutable_backend(), in Load()456 mutable_backend()->BorrowStream(device_ordinal)); in LiteralToShapedBuffer()464 TF_ASSIGN_OR_RETURN(auto stream, mutable_backend()->BorrowStream( in ShapedBufferToLiteral()503 mutable_backend()->BorrowStream(device_ordinal)); in TransferToLocalServer()
225 Backend* mutable_backend();
231 client_->mutable_backend()->BorrowStream( in CopyDeviceTensorToCPU()281 auto status_or_new_stream = client_->mutable_backend()->BorrowStream( in CopyDeviceTensorToCPU()
274 xla::Backend* backend = GetOrCreateClient().ValueOrDie()->mutable_backend(); in GetAllocatorLocked()303 xla::Backend* backend = client->mutable_backend(); in GetDeviceContextLocked()
54 xla::Backend* backend() { return client_->mutable_backend(); } in backend()
108 allocator_ = client_->mutable_backend()->memory_allocator(); in Acquire()
203 stream = local_client_->mutable_backend() in ExecuteLocally()
760 ASSERT_IS_OK(local_client_->mutable_backend() in XLA_TEST_F()957 client->mutable_backend()->BorrowStream(device_ordinal).ValueOrDie(); in BM_LocalClientOverhead()
47 backend_ = client_->mutable_backend(); in BufferDonationTest()
136 Backend* backend = service->mutable_backend(); in ExecuteAndFetchProfile()
47 stream_ptr_ = local_client_->mutable_backend()
775 client->mutable_backend()->BorrowStream(device_ordinal).ValueOrDie(); in BM_DynamicSlice()
183 Backend* mutable_backend() { return execute_backend_.get(); } in mutable_backend() function
195 std::move(tpu_executable), client_->mutable_backend(), in DeserializeExecutable()