Program Listing for File reduce_fusion.h
↰ Return to documentation for file (include/converter/include/ops/fusion/reduce_fusion.h
)
#ifndef MINDSPORE_CORE_OPS_REDUCE_FUSION_H_
#define MINDSPORE_CORE_OPS_REDUCE_FUSION_H_
#include <map>
#include <memory>
#include <string>
#include "mindapi/base/types.h"
#include "ops/reduce.h"
namespace mindspore {
namespace ops {
constexpr auto kNameReduceFusion = "ReduceFusion";
class MIND_API ReduceFusion : public Reduce {
public:
MIND_API_BASE_MEMBER(ReduceFusion);
ReduceFusion() : Reduce(kNameReduceFusion) {}
void Init(const bool keep_dims = false, const ReduceMode mode = ReduceMode::Reduce_Mean,
const bool reduce_to_end = false, const float coeff = 1.0);
void set_keep_dims(const bool keep_dims);
void set_mode(const ReduceMode mode);
void set_reduce_to_end(const bool reduce_to_end);
void set_coeff(const float coeff);
bool get_keep_dims() const;
ReduceMode get_mode() const;
bool get_reduce_to_end() const;
float get_coeff() const;
};
} // namespace ops
} // namespace mindspore
#endif // MINDSPORE_CORE_OPS_REDUCE_FUSION_H_