def graph_fn():
    boxes = tf.zeros([0, 4], dtype=tf.float32)
    blackout = tf.zeros([0], dtype=tf.bool)
    blackout_pixel_weights_by_box_regions = tf.function(ta_utils.blackout_pixel_weights_by_box_regions)
    output = blackout_pixel_weights_by_box_regions(10, 20, boxes, blackout)
    return output
def graph_fn():
    boxes = tf.zeros([0, 4], dtype=tf.float32)
    blackout = tf.zeros([0], dtype=tf.bool)
    blackout_pixel_weights_by_box_regions = tf.function(ta_utils.blackout_pixel_weights_by_box_regions)
    output = blackout_pixel_weights_by_box_regions(10, 20, boxes, blackout)
    return output