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()
|