6 lines
158 B
Python
6 lines
158 B
Python
import torch
|
|
import ixformer.functions as ixf_F
|
|
|
|
def topk_softmax(topk_weights,topk_ids,token_expert_indicies,gating_output):
|
|
raise NotImplementedError()
|