Searched refs:mutable_backend (Results 1 – 11 of 11) sorted by relevance
240 Backend* LocalClient::mutable_backend() { in mutable_backend() function in xla::LocalClient241 return local_service_->mutable_backend(); in mutable_backend()258 local_service_->mutable_backend(), in Compile()272 mutable_backend()->BorrowStream(device_ordinal)); in LiteralToShapedBuffer()280 TF_ASSIGN_OR_RETURN(auto stream, mutable_backend()->BorrowStream( in ShapedBufferToLiteral()322 mutable_backend()->BorrowStream(device_oridinal)); in TransferToLocalServer()
186 Backend* mutable_backend();
47 xla::Backend* backend() { return client_->mutable_backend(); } in backend()
251 xla::Backend* backend = client()->mutable_backend(); in GetAllocatorLocked()279 xla::Backend* backend = client()->mutable_backend(); in GetDeviceContextLocked()
221 client_->mutable_backend()->BorrowStream( in CopyDeviceTensorToCPU()
199 stream = local_client_->mutable_backend() in ExecuteLocally()
47 stream_ptr_ = local_client_->mutable_backend()
136 Backend* backend = service->mutable_backend(); in ExecuteAndFetchProfile()
772 ASSERT_IS_OK(local_client_->mutable_backend() in XLA_TEST_F()923 client->mutable_backend()->BorrowStream(device_ordinal).ValueOrDie(); in BM_LocalClientOverhead()
755 client->mutable_backend()->BorrowStream(device_ordinal).ValueOrDie(); in BM_DynamicSlice()
184 Backend* mutable_backend() { return execute_backend_.get(); } in mutable_backend() function