(step)
| 957 | |
| 958 | |
| 959 | def _choose_step(step): |
| 960 | if step is None: |
| 961 | return training_util.get_or_create_global_step() |
| 962 | if not isinstance(step, ops.Tensor): |
| 963 | return ops.convert_to_tensor(step, dtypes.int64) |
| 964 | return step |
| 965 | |
| 966 | |
| 967 | def _check_create_file_writer_args(inside_function, **kwargs): |