diff --git a/ethosu/regor/compiler/operation.cpp b/ethosu/regor/compiler/operation.cpp index 1aa57eee586187ccf8e07827e3122ac6bc692514..7a6e25780359cc7aca5eaa4f0b0ca79a612a57be 100644 --- a/ethosu/regor/compiler/operation.cpp +++ b/ethosu/regor/compiler/operation.cpp @@ -74,7 +74,9 @@ void Operation::CopyInput(TensorUsage usage, const TensorConnection &tensorConne ConnectInput(usage, tensorConnection.tensor) .Set(tensorConnection.shape) .Set(tensorConnection.slice) - .Set(tensorConnection.quantization); + .Set(tensorConnection.quantization) + .Set(tensorConnection.reverse) + .Set(tensorConnection.rounding); } TensorConnection &Operation::ConnectInput(TensorUsage usage, const std::shared_ptr &tensor) @@ -111,7 +113,9 @@ void Operation::CopyOutput(TensorUsage usage, const TensorConnection &tensorConn ConnectOutput(usage, tensorConnection.tensor) .Set(tensorConnection.shape) .Set(tensorConnection.slice) - .Set(tensorConnection.quantization); + .Set(tensorConnection.quantization) + .Set(tensorConnection.reverse) + .Set(tensorConnection.rounding); } TensorConnection &Operation::ConnectOutput(TensorUsage usage, const std::shared_ptr &tensor)