Searched refs:KernelPack (Results 1 – 6 of 6) sorted by relevance
44 bool KernelPack::ReadFromJsonFileHelper(std::ifstream &kernel_bin) { in ReadFromJsonFileHelper()52 void *ptr = static_cast<void *>(new (std::nothrow) uint8_t[sizeof(KernelPack) + bin_size]); in ReadFromJsonFileHelper()61 if (memset_s(kernel_, sizeof(KernelPack) + bin_size, 0, sizeof(KernelPack) + bin_size) != EOK) { in ReadFromJsonFileHelper()74 bool KernelPack::ReadFromJsonFile(const std::string &json_f, const std::string &processor) { in ReadFromJsonFile()89 void *ptr = static_cast<void *>(new (std::nothrow) uint8_t[sizeof(KernelPack) + bin_size]); in ReadFromJsonFile()172 void KernelPack::ParseKernelJson(const nlohmann::json &js) { in ParseKernelJson()200 bool KernelPack::LoadKernelMeta(const std::string &json_f) { in LoadKernelMeta()252 KernelJsonInfo KernelPack::kernel_json_info() const { return kernel_json_info_; } in kernel_json_info()
130 class KernelPack {132 KernelPack() : json_(nullptr), kernel_(nullptr) {} in KernelPack() function133 KernelPack(const KernelPack &) = default;139 ~KernelPack() { in ~KernelPack()157 using KernelPackPtr = std::shared_ptr<KernelPack>;
221 KernelPackPtr kernel_pack = std::make_shared<KernelPack>(); in SearchCache()245 KernelPackPtr kernel_pack = std::make_shared<KernelPack>(); in InsertCache()
79 …static uintptr_t GenFuncStub(const KernelPack &kernel_pack, bool force_reload, uint32_t *block_dim,
277 uintptr_t KernelManager::GenFuncStub(const mindspore::kernel::KernelPack &kernel_pack, bool force_r… in GenFuncStub()511 ret = std::make_shared<KernelPack>(); in GetKernelPack()
88 auto tbe_kernel_pack_first = std::make_shared<kernel::KernelPack>(); in CreateKernelGraph()118 auto tbe_kernel_pack_floor = std::make_shared<kernel::KernelPack>(); in CreateKernelGraph()