| Commit message (Collapse) | Author | Age |
|
|
|
|
|
| |
Instead of printing the enum (integer value) for the dtype which is hard to understand by the users, we are now printing the dtype string value.
PiperOrigin-RevId: 208132121
|
|
|
|
| |
PiperOrigin-RevId: 196051520
|
|
|
|
|
|
|
|
|
|
|
|
| |
The _ScopedAllocatorConcat kernel outputs the backing tensor after performing
runtime bounds checks. However, the shape of the backing tensor may not match
the desired output shape of the concat operation.
This change adds a "reshape" boolean attribute to _ScopedAllocatorConcat kernel.
When this attribute is set to true, the kernel outputs a reshaped backing tensor
according to the "shape" attribute.
PiperOrigin-RevId: 195169105
|
|
ScopedAllocatorOp allocates a large backing tensor whose slices may be
concatenated or splitted with ScopedAllocatorConcatOp and
ScopedAllocatorSplitOp respectively. These ops should only be added via
Grappler optimizations on the dataflow graph provided by the user.
PiperOrigin-RevId: 190097586
|