Searched refs:mutable_backend (Results 1 – 11 of 11) sorted by relevance
336 Backend* LocalClient::mutable_backend() { in mutable_backend() function in xla::LocalClient337 return local_service_->mutable_backend(); in mutable_backend()359 std::move(executable), local_service_->mutable_backend(), in Compile()376 mutable_backend()->BorrowStream(device_ordinal)); in LiteralToShapedBuffer()384 TF_ASSIGN_OR_RETURN(auto stream, mutable_backend()->BorrowStream( in ShapedBufferToLiteral()425 mutable_backend()->BorrowStream(device_ordinal)); in TransferToLocalServer()
187 Backend* mutable_backend();
47 xla::Backend* backend() { return client_->mutable_backend(); } in backend()
264 xla::Backend* backend = GetOrCreateClient().ValueOrDie()->mutable_backend(); in GetAllocatorLocked()294 xla::Backend* backend = client->mutable_backend(); in GetDeviceContextLocked()
227 client_->mutable_backend()->BorrowStream( in CopyDeviceTensorToCPU()
184 Backend* mutable_backend() { return execute_backend_.get(); } in mutable_backend() function
206 stream = local_client_->mutable_backend() in ExecuteLocally()
136 Backend* backend = service->mutable_backend(); in ExecuteAndFetchProfile()
47 stream_ptr_ = local_client_->mutable_backend()
775 ASSERT_IS_OK(local_client_->mutable_backend() in XLA_TEST_F()951 client->mutable_backend()->BorrowStream(device_ordinal).ValueOrDie(); in BM_LocalClientOverhead()
755 client->mutable_backend()->BorrowStream(device_ordinal).ValueOrDie(); in BM_DynamicSlice()