Searched refs:hash_swap_out_mod (Results 1 – 1 of 1) sorted by relevance
217 auto hash_swap_out_mod = std::make_shared<kernel::AicpuOpKernelMod>(); in HashSwapOut() local218 MS_ERROR_IF_NULL(hash_swap_out_mod); in HashSwapOut()219 hash_swap_out_mod->SetNodeName(kEmbeddingLookupOpName); in HashSwapOut()234 RETURN_IF_FALSE(SetNodedefProto(op_info, hash_swap_out_mod)); in HashSwapOut()244 auto ret = hash_swap_out_mod->Launch(kernel_inputs, kernel_workspaces, kernel_outputs, stream_); in HashSwapOut()