bagua.torch_api.model_parallel.moe.utils

Module Contents

bagua.torch_api.model_parallel.moe.utils.is_moe_param(param)
Parameters

param (torch.Tensor) –

Return type

bool