Eager-only version of Reshape op; requires tensor is an eager Tensor.
(tensor, shape, ctx)
| 35 | |
| 36 | |
| 37 | def _eager_reshape(tensor, shape, ctx): |
| 38 | """Eager-only version of Reshape op; requires tensor is an eager Tensor.""" |
| 39 | attr_t = tensor._datatype_enum() # pylint: disable=protected-access |
| 40 | attr_tshape, (shape,) = execute.args_to_matching_eager( |
| 41 | [shape], ctx, dtypes.int32) |
| 42 | inputs_flat = [tensor, shape] |
| 43 | attrs = ("T", attr_t, "Tshape", attr_tshape) |
| 44 | result, = execute.execute( |
| 45 | b"Reshape", 1, inputs=inputs_flat, attrs=attrs, ctx=ctx) |
| 46 | return result |
| 47 | |
| 48 | |
| 49 | def _eager_fill(dims, value, ctx): |
no test coverage detected