Internal implementation for the v1/v2 ones_like API calls.
(tensor, dtype, name, optimize=True)
| 2518 | |
| 2519 | |
| 2520 | def ones_like_impl(tensor, dtype, name, optimize=True): |
| 2521 | """Internal implementation for the v1/v2 ones_like API calls.""" |
| 2522 | with ops.name_scope(name, "ones_like", [tensor]) as name: |
| 2523 | tensor = ops.convert_to_tensor(tensor, name="tensor") |
| 2524 | ones_shape = shape_internal(tensor, optimize=optimize) |
| 2525 | if dtype is None: |
| 2526 | dtype = tensor.dtype |
| 2527 | ret = ones(ones_shape, dtype=dtype, name=name) |
| 2528 | if not context.executing_eagerly(): |
| 2529 | ret.set_shape(tensor.get_shape()) |
| 2530 | return ret |
| 2531 | |
| 2532 | |
| 2533 | @tf_export("ones") |
no test coverage detected