def stage2(x, c, label): with variable_scope.variable_scope("stage2", use_resource=True): with ops.control_dependencies([internal_ops.print_tensor(x)]): return x * 20, c, label
def stage2(x, c): internal_ops.print_tensor(c, "stage2_c") return math_ops.reduce_sum(x) + c
def stage3(x): internal_ops.print_tensor(x, "stage3_x") return x
def stage2(x, label): internal_ops.print_tensor(x) return x, label