Ejemplo n.º 1
0
 def function(tag, scope):
   bad_color_ = (constant_op.constant([255, 0, 0, 255], dtype=dtypes.uint8)
                 if bad_color is None else bad_color)
   # Note the identity to move the tensor to the CPU.
   return gen_summary_ops.write_image_summary(
       _summary_state.writer._resource,  # pylint: disable=protected-access
       _choose_step(step),
       tag,
       array_ops.identity(tensor),
       bad_color_,
       max_images,
       name=scope)
Ejemplo n.º 2
0
 def function(tag, scope):
   bad_color_ = (constant_op.constant([255, 0, 0, 255], dtype=dtypes.uint8)
                 if bad_color is None else bad_color)
   # Note the identity to move the tensor to the CPU.
   return gen_summary_ops.write_image_summary(
       context.context().summary_writer_resource,
       _choose_step(step),
       tag,
       array_ops.identity(tensor),
       bad_color_,
       max_images,
       name=scope)
Ejemplo n.º 3
0
 def function(tag, scope):
   bad_color_ = (constant_op.constant([255, 0, 0, 255], dtype=dtypes.uint8)
                 if bad_color is None else bad_color)
   # Note the identity to move the tensor to the CPU.
   return gen_summary_ops.write_image_summary(
       context.context().summary_writer_resource,
       _choose_step(step),
       tag,
       array_ops.identity(tensor),
       bad_color_,
       max_images,
       name=scope)