Searched refs:mutable_backend (Results 1 – 11 of 11) sorted by relevance
352 Backend* LocalClient::mutable_backend() { in mutable_backend() function in xla::LocalClient353 return local_service_->mutable_backend(); in mutable_backend()392 std::move(executable), local_service_->mutable_backend(), in Compile()409 mutable_backend()->BorrowStream(device_ordinal)); in LiteralToShapedBuffer()417 TF_ASSIGN_OR_RETURN(auto stream, mutable_backend()->BorrowStream( in ShapedBufferToLiteral()456 mutable_backend()->BorrowStream(device_ordinal)); in TransferToLocalServer()
210 Backend* mutable_backend();
51 xla::Backend* backend() { return client_->mutable_backend(); } in backend()
229 client_->mutable_backend()->BorrowStream( in CopyDeviceTensorToCPU()279 auto status_or_new_stream = client_->mutable_backend()->BorrowStream( in CopyDeviceTensorToCPU()
265 xla::Backend* backend = GetOrCreateClient().ValueOrDie()->mutable_backend(); in GetAllocatorLocked()295 xla::Backend* backend = client->mutable_backend(); in GetDeviceContextLocked()
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()972 client->mutable_backend()->BorrowStream(device_ordinal).ValueOrDie(); in BM_LocalClientOverhead()
772 client->mutable_backend()->BorrowStream(device_ordinal).ValueOrDie(); in BM_DynamicSlice()