49 lines
1.2 KiB
C
49 lines
1.2 KiB
C
//
|
|
// Created by sjt on 11/9/23.
|
|
//
|
|
|
|
#pragma once
|
|
#include "C_Tensor.h"
|
|
#ifdef __cplusplus
|
|
extern "C" {
|
|
#endif
|
|
|
|
|
|
|
|
struct ReduceAttribute {
|
|
// user
|
|
const char* name_;
|
|
C_Shape axes_;
|
|
int keepdims_;
|
|
int noop_with_empty_axes_;
|
|
int opset_;
|
|
};
|
|
|
|
typedef struct ReduceAttribute ReduceLogSumAttribute;
|
|
typedef struct ReduceAttribute ReduceLogSumExpAttribute;
|
|
typedef struct ReduceAttribute ReduceMeanAttribute;
|
|
typedef struct ReduceAttribute ReduceSumAttribute;
|
|
typedef struct ReduceAttribute ReduceMaxAttribute;
|
|
typedef struct ReduceAttribute ReduceSumSquareAttribute;
|
|
|
|
//----------------------------------------------------------------------------
|
|
// MeanVarianceNormalization
|
|
//------------------------------------------------------------------------
|
|
struct MeanVarianceNormalizationAttribute{
|
|
// user
|
|
const char* name_;
|
|
C_Shape axes_;
|
|
};
|
|
typedef struct MeanVarianceNormalizationAttribute MeanVarianceNormalizationAttribute;
|
|
|
|
CREATE_DECL_RUN_FUN(ReduceLogSum)
|
|
CREATE_DECL_RUN_FUN(ReduceLogSumExp)
|
|
CREATE_DECL_RUN_FUN(ReduceMax)
|
|
CREATE_DECL_RUN_FUN(ReduceMean)
|
|
CREATE_DECL_RUN_FUN(ReduceSum)
|
|
CREATE_DECL_RUN_FUN(ReduceSumSquare)
|
|
CREATE_DECL_RUN_FUN(MeanVarianceNormalization)
|
|
|
|
#ifdef __cplusplus
|
|
}
|
|
#endif |