39 lines
933 B
C
39 lines
933 B
C
//
|
|
// Created by xiangzhou on 11/9/23.
|
|
//
|
|
|
|
#pragma once
|
|
#include "C_Tensor.h"
|
|
#ifdef __cplusplus
|
|
extern "C" {
|
|
#endif
|
|
|
|
//----------------------------------------------------------------------------
|
|
// Multinomial
|
|
//------------------------------------------------------------------------
|
|
struct MultinomialAttribute {
|
|
// user
|
|
const char* name_;
|
|
int sample_size_;
|
|
float seed_;
|
|
};
|
|
typedef struct MultinomialAttribute MultinomialAttribute;
|
|
|
|
//----------------------------------------------------------------------------
|
|
// RandomUniformLike
|
|
//------------------------------------------------------------------------
|
|
struct RandomUniformLikeAttribute {
|
|
// user
|
|
const char* name_;
|
|
float high_;
|
|
float low_;
|
|
float seed_;
|
|
};
|
|
typedef struct RandomUniformLikeAttribute RandomUniformLikeAttribute;
|
|
|
|
CREATE_DECL_RUN_FUN(Multinomial)
|
|
CREATE_DECL_RUN_FUN(RandomUniformLike)
|
|
|
|
#ifdef __cplusplus
|
|
}
|
|
#endif |