1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17
#pragma once #include "op.h" #include "mean.h" namespace ctranslate2 { namespace ops { class Sum : public Mean { public: Sum(const dim_t axis); void operator()(const StorageView& input, StorageView& output) const override; }; } }