Skip to content
Snippets Groups Projects
Commit de3fd61b authored by Maxence Naud's avatar Maxence Naud Committed by Inna Kucher
Browse files

Apply 1 suggestion(s) to 1 file(s)

parent 74656734
No related branches found
No related tags found
1 merge request!12Quantization ptq
Pipeline #32015 canceled
......@@ -71,8 +71,16 @@ public:
}
inline Tensor& input(__attribute__((unused)) const IOIndex_t inputIdx) const override final { return *(mInput.get()); }
inline Tensor& output(__attribute__((unused)) const IOIndex_t outputIdx) const override final { return *(mOutput.get()); }
inline Tensor& input(const IOIndex_t inputIdx) const override final {
assert((inputIdx == 0) && "Scaling Operator has only 1 input");
(void) inputIdx; // avoid unused warning
return *(mInput.get());
}
inline Tensor& output(const IOIndex_t outputIdx) const override final {
assert((outputIdx == 0) && "Scaling Operator has only 1 output");
(void) outputIdx; // avoid unused warning
return *(mOutput.get());
}
inline std::shared_ptr<Tensor> getInput(const IOIndex_t inputIdx) const override final {
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment