diff options
Diffstat (limited to 'tensorflow/core/debug')
-rw-r--r-- | tensorflow/core/debug/debug_gateway.cc | 6 |
1 files changed, 4 insertions, 2 deletions
diff --git a/tensorflow/core/debug/debug_gateway.cc b/tensorflow/core/debug/debug_gateway.cc index 24b9dd799a..1031ea843e 100644 --- a/tensorflow/core/debug/debug_gateway.cc +++ b/tensorflow/core/debug/debug_gateway.cc @@ -15,6 +15,8 @@ limitations under the License. #include "tensorflow/core/debug/debug_gateway.h" +#include <utility> + #include "tensorflow/core/common_runtime/device_factory.h" #include "tensorflow/core/common_runtime/session_factory.h" #include "tensorflow/core/framework/tensor.h" @@ -56,11 +58,11 @@ DebugGateway::~DebugGateway() { } void DebugGateway::SetNodeCompletionCallback(NodeCompletionCallback callback) { - comp_cb_ = callback; + comp_cb_ = std::move(callback); } void DebugGateway::SetNodeValueCallback(NodeValueCallback callback) { - val_cb_ = callback; + val_cb_ = std::move(callback); } void DebugGateway::CopyTensor(const string& node_name, const int output_slot, |