aboutsummaryrefslogtreecommitdiffhomepage
path: root/tensorflow/core/kernels/scoped_allocator_ops.cc
Commit message (Collapse)AuthorAge
* Improving error messages for dtypes.Gravatar A. Unique TensorFlower2018-08-09
| | | | | | 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
* Improve error status message in scoped_allocator_ops.cc.Gravatar A. Unique TensorFlower2018-05-09
| | | | PiperOrigin-RevId: 196051520
* Enable reshape of _ScopedAllocatorConcat output.Gravatar Ayush Dubey2018-05-02
| | | | | | | | | | | | 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
* Add new Ops for ScopedAllocator and the associated Concat and Split. TheGravatar Ayush Dubey2018-03-22
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