|
ONE - On-device Neural Engine
|
#include <Config.h>

Data Fields | |
| ModelConfig | model |
| int | cache_size = 32 |
| int | ubatch = 32 |
| KVCacheDataType | kv_cache_type = KVCacheDataType::FLOAT32 |
| int ggma::GGMAConfig::cache_size = 32 |
Definition at line 68 of file Config.h.
Referenced by ggma::Context::Context(), and ggma::Context::generate().
| KVCacheDataType ggma::GGMAConfig::kv_cache_type = KVCacheDataType::FLOAT32 |
Definition at line 70 of file Config.h.
Referenced by ggma::KVCache::init().
| ModelConfig ggma::GGMAConfig::model |
Definition at line 67 of file Config.h.
Referenced by ggma::Context::generate(), ggma::KVCache::init(), ggma::Context::prefill(), and ggma::Context::sample().
| int ggma::GGMAConfig::ubatch = 32 |
Definition at line 69 of file Config.h.
Referenced by ggma::Context::unemb().