| commit | c887ebe555a725d3113890c60ad9f6c0f466d35b | [log] [tgz] |
|---|---|---|
| author | Patrice Vignola <vignola.patrice@gmail.com> | Fri Mar 25 20:00:46 2022 -0700 |
| committer | Patrice Vignola <vignola.patrice@gmail.com> | Fri Mar 25 20:00:46 2022 -0700 |
| tree | 3a81883288534698fe922143a9d5a47bf6f16e9e | |
| parent | 6184af5f864f5a88e92424b9d083212359e99663 [diff] |
Set OK status for TF_IsHostMemoryOutput
diff --git a/tensorflow/c/kernels.cc b/tensorflow/c/kernels.cc index 6f06f1e..dfdd999 100644 --- a/tensorflow/c/kernels.cc +++ b/tensorflow/c/kernels.cc
@@ -471,6 +471,7 @@ TF_SetStatus(status, TF_OUT_OF_RANGE, "input index out of range"); return false; } + TF_SetStatus(status, TF_OK, ""); return cc_ctx->output_memory_type(i) == tensorflow::HOST_MEMORY; }