用于获取dropout Tiling参数。
inline uint32_t GetDropOutMaxTmpSize(const ge::Shape srcShape, const uint32_t typeSize, const bool isReuseSource);
inline uint32_t GetDropOutMinTmpSize(const ge::Shape srcShape, const uint32_t typeSize, const bool isReuseSource);
inline void GetDropOutMaxMinTmpSize(const ge::Shape srcShape, const uint32_t typeSize, const bool isReuseSource, uint32_t& maxValue, uint32_t& minValue);
接口 |
输入/输出 |
功能 |
|---|---|---|
srcShape |
输入 |
输入的shape信息。 |
typeSize |
输入 |
计算的数据类型，比如half=2。 |
isReuseSource |
输入 |
是否复用src输入的空间，与kernel侧接口一致。 |
maxValue |
输出 |
输出dropout接口所需的tiling信息（最大临时空间大小）。 |
minValue |
输出 |
输出dropout接口所需的tiling信息（最小临时空间大小）。 |
GetDropOutMaxTmpSize返回dropout接口能完成计算所需最大临时空间大小
GetDropOutMinTmpSize返回dropout接口能完成计算所需最小临时空间大小
GetDropOutMaxMinTmpSize返回dropout接口能完成计算所需最大和小临时空间大小