def __init__(self, strategy1, strategy2): super().__init__() self.matmul = P.MatMul().set_strategy(strategy1) self.erfc = P.Erfc().set_strategy(strategy2) self.matmul2 = P.MatMul().set_strategy(strategy1)
def __init__(self): super(ErfcNet, self).__init__() self.erfc = P.Erfc()
def __init__(self): super(NetErfc, self).__init__() self.erfc = P.Erfc()