def test_power_two_input_infer3(self):
graph = self.create_graph(single_input=False)
power_node = Node(graph, 'power')
input2 = Node(graph, 'input2')
input2.value = None
Power.infer(power_node)
self.assertIsNone(power_node.out_node().shape)
def test_power_two_input_infer1(self):
graph = self.create_graph(single_input=False)
graph.graph['layout'] = 'NCHW'
power_node = Node(graph, 'power')
Power.infer(power_node)
self.assertTrue(
np.array_equal(power_node.out_node().shape,
power_node.in_node(0).shape))
