Commit e06c3bbf authored by peastman's avatar peastman
Browse files

Made test more tolerant

parent abaa902d
Loading
Loading
Loading
Loading
+1 −1
Original line number Diff line number Diff line
@@ -1282,7 +1282,7 @@ class L2Loss(Layer):
    l2 = tf.square(guess - label)
    if len(inputs) > 2:
      l2 *= inputs[2]
    out_tensor = tf.reduce_mean(l2, axis=list(range(1, len(label._shape))))
    out_tensor = tf.reduce_mean(l2, axis=list(range(1, len(label.shape))))
    if set_tensors:
      self.out_tensor = out_tensor
    return out_tensor
+1 −1
Original line number Diff line number Diff line
@@ -360,7 +360,7 @@ class TestLayersEager(test_util.TensorFlowTestCase):
    with context.eager_mode():
      with tfe.IsolateTest():
        result = layers.Divide()([1, 2], [2, 5])
        assert np.array_equal(result, [0.5, 0.4])
        assert np.allclose(result, [0.5, 0.4])

  def test_log(self):
    """Test invoking Log in eager mode."""