Eager-only version of Fill op; requires value is an eager Tensor.
(dims, value, ctx)
| 47 | |
| 48 | |
| 49 | def _eager_fill(dims, value, ctx): |
| 50 | """Eager-only version of Fill op; requires value is an eager Tensor.""" |
| 51 | attr_t = value.dtype.as_datatype_enum |
| 52 | dims = convert_to_eager_tensor(dims, ctx, dtypes.int32) |
| 53 | inputs_flat = [dims, value] |
| 54 | attrs = ("T", attr_t, "index_type", types_pb2.DT_INT32) |
| 55 | result, = execute.execute( |
| 56 | b"Fill", 1, inputs=inputs_flat, attrs=attrs, ctx=ctx) |
| 57 | return result |
| 58 | |
| 59 | |
| 60 | def _eager_identity(tensor, ctx): |
no test coverage detected