Searched refs:TbeKernelMod (Results 1 – 5 of 5) sorted by relevance
29 class TbeKernelMod : public AscendKernelMod {31 explicit TbeKernelMod(KernelPackPtr kernel_pack) : kernel_pack_(std::move(kernel_pack)) {} in TbeKernelMod() function32 ~TbeKernelMod() override = default;55 using TbeKernelModPtr = std::shared_ptr<TbeKernelMod>;
29 bool TbeKernelMod::Launch(const std::vector<mindspore::kernel::AddressPtr> &inputs, in Launch()73 std::vector<TaskInfoPtr> TbeKernelMod::GenTask(const std::vector<AddressPtr> &inputs, in GenTask()113 device::DynamicKernelPtr TbeKernelMod::GenDynamicKernel(const CNodePtr &cnode_ptr, void *stream_ptr… in GenDynamicKernel()167 vector<size_t> TbeKernelMod::GenParameters() { in GenParameters()
291 auto kernel_mod_ptr = std::make_shared<TbeKernelMod>(kernel_pack); in GenKernelMod()
26 kernel_mod_ret[fusion_scope_iter.scope_id] = std::make_shared<TbeKernelMod>(nullptr); in KernelFusion()45 kernel_mod_ret[fusion_scope_iter.scope_id] = std::make_shared<TbeKernelMod>(nullptr); in AscendFusionOpCompile()
89 auto kernel_mod_first = std::make_shared<kernel::TbeKernelMod>(tbe_kernel_pack_first); in CreateKernelGraph()119 auto kernel_mod_floor = std::make_shared<kernel::TbeKernelMod>(tbe_kernel_pack_floor); in CreateKernelGraph()