def METHOD_NAME(dtype):
    image = [
        [[1], [1], [1], [1], [1]],
        [[1], [1], [1], [1], [1]],
        [[0], [1], [0], [1], [0]],
        [[1], [0], [1], [0], [1]],
        [[0], [1], [0], [1], [0]],
    ]
    expected_output = np.array(
        [
            2,
            2.23606801,
            2,
            2.23606801,
            2,
            1,
            1.41421354,
            1,
            1.41421354,
            1,
            0,
            1,
            0,
            1,
            0,
            1,
            0,
            1,
            0,
            1,
            0,
            1,
            0,
            1,
            0,
        ]
    )
    image = tf.constant(image, dtype=tf.uint8)
    output = dist_ops.euclidean_dist_transform(image, dtype=dtype)
    output_flat = tf.reshape(output, [-1])
    assert output.dtype == dtype
    assert output.shape == [5, 5, 1]
    test_utils.assert_allclose_according_to_type(output_flat, expected_output)