Example #1
0
 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)
Example #2
0
 def __init__(self):
     super(ErfcNet, self).__init__()
     self.erfc = P.Erfc()
Example #3
0
 def __init__(self):
     super(NetErfc, self).__init__()
     self.erfc = P.Erfc()