8000 Add default implementation for Finalize(). · IBMZ-Linux-OSS-Python/tensorflow@36f863b · GitHub
[go: up one dir, main page]

Skip to content

Commit 36f863b

Browse files
Add default implementation for Finalize().
PiperOrigin-RevId: 673107887
1 parent b3c0594 commit 36f863b

File tree

2 files changed

+4
-0
lines changed

2 files changed

+4
-0
lines changed

tensorflow/core/tfrt/tfrt_session/tfrt_session.cc

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -481,6 +481,8 @@ class TfrtSession : public tensorflow::Session {
481481
return absl::OkStatus();
482482
}
483483

484+
Status Finalize() override { return absl::OkStatus(); }
485+
484486
private:
485487
tfrt::HostContext* GetHostContext() {
486488
return runtime_->core_runtime()->GetHostContext();

tensorflow/core/tfrt/tfrt_session/tfrt_session_test.cc

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -399,6 +399,8 @@ TEST_F(TfrtSessionTest, Callable) {
399399
TF_ASSERT_OK(session_->ReleaseCallable(callable_handle));
400400
}
401401

402+
TEST_F(TfrtSessionTest, Finalize) { TF_ASSERT_OK(session_->Finalize()); }
403+
402404
TEST_F(TfrtSessionTest, WithTargetNodes) {
403405
std::vector<Tensor> outputs;
404406
TF_ASSERT_OK(session_->Run(inputs_, output_tensor_names_, target_node_names_,

0 commit comments

Comments
 (0)
0