Searched refs:optimizer_kernel (Results 1 – 1 of 1) sorted by relevance
117 std::shared_ptr<kernel::OptimizerKernel> optimizer_kernel = optimizer_with_params.first; in LaunchOptimizers() local118 MS_ERROR_IF_NULL_W_RET_VAL(optimizer_kernel, false); in LaunchOptimizers()119 bool ret = optimizer_kernel->Launch(params.inputs, params.workspace, params.outputs); in LaunchOptimizers()121 …MS_LOG(ERROR) << "Launching optimizer kernel " << typeid(optimizer_kernel.get()).name() << " faile… in LaunchOptimizers()234 auto optimizer_kernel = kernel::OptimizerKernelFactory::GetInstance().Create(name, cnode); in InitOptimizerKernels() local235 if (optimizer_kernel == nullptr) { in InitOptimizerKernels()240 optimizer_kernel->InitKernel(cnode); in InitOptimizerKernels()242 …const ReuseKernelNodeInfo &reuse_kernel_node_inputs_info = optimizer_kernel->reuse_kernel_node_inp… in InitOptimizerKernels()243 if (!AssignMemory(optimizer_kernel, cnode, reuse_kernel_node_inputs_info, memory_register_)) { in InitOptimizerKernels()248 if (!GenerateOptimizerKernelParams(optimizer_kernel, memory_register_)) { in InitOptimizerKernels()[all …]