10902930dcd12210c6946897e7f6203e085fbad0,test/test_node.py,TestNode,test_average_pool,#TestNode#,78

Before Change

  def test_average_pool(self):
    // TODO: uncomment this in the future
    return
    shape = [1, 1, 40, 40]
    node_def = helper.make_node("AveragePool", ["X"], ["Y"],
      kernel_shape=[1,2],
      pads=[0, 0], strides=[1,1])

After Change

  def test_average_pool(self):
    device = "CUDA"
    if not supports_device(device):
                raise unittest.SkipTest(
                    "Backend doesn"t support device {}".format(device))
    shape = [1, 1, 40, 40]
    node_def = helper.make_node("AveragePool", ["X"], ["Y"],
      kernel_shape=[1,2],
      pads=[0, 0], strides=[1,1])

Instances