Searched refs:GradNetWrap (Results 1 – 2 of 2) sorted by relevance
69 class GradNetWrap(nn.Cell): class73 super(GradNetWrap, self).__init__()83 'block': GradNetWrap(DisOrderTest1()),86 'block': GradNetWrap(DisOrderTest2()),
116 class GradNetWrap(nn.Cell): class120 super(GradNetWrap, self).__init__()146 GradNetWrap(Mul())(input_x, input_y)