projects / platform / upstream / tensorflow.git / commit
? search:
summary | shortlog | log | commit | commitdiff | tree
(parent: 7c4cdb8) | patch
Add new Ops for ScopedAllocator and the associated Concat and Split. The
authorAyush Dubey <ayushd@google.com>
Thu, 22 Mar 2018 18:25:58 +0000 (11:25 -0700)
committerTensorFlower Gardener <gardener@tensorflow.org>
Thu, 22 Mar 2018 18:32:16 +0000 (11:32 -0700)
commit282750fee5e2df502436ca9ef6a95283f8adab34
tree3c24d054b88ab0757c9b3e392faf0b1dcbddfaaatree | snapshot
parent7c4cdb8bae0e8760ebe4793d49ea5aee68768655commit | diff
Add new Ops for ScopedAllocator and the associated Concat and Split.  The
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
16 files changed:
tensorflow/core/BUILD diff | blob | history
tensorflow/core/common_runtime/gpu/gpu_device.cc diff | blob | history
tensorflow/core/common_runtime/gpu/gpu_device.h diff | blob | history
tensorflow/core/common_runtime/scoped_allocator.cc diff | blob | history
tensorflow/core/common_runtime/scoped_allocator_mgr.cc diff | blob | history
tensorflow/core/common_runtime/scoped_allocator_mgr.h diff | blob | history
tensorflow/core/common_runtime/scoped_allocator_mgr_test.cc diff | blob | history
tensorflow/core/common_runtime/threadpool_device.cc diff | blob | history
tensorflow/core/common_runtime/threadpool_device.h diff | blob | history
tensorflow/core/framework/allocator.h diff | blob | history
tensorflow/core/framework/device_base.h diff | blob | history
tensorflow/core/framework/op_kernel.cc diff | blob | history
tensorflow/core/kernels/BUILD diff | blob | history
tensorflow/core/kernels/scoped_allocator_ops.cc [new file with mode: 0644] blob
tensorflow/core/kernels/scoped_allocator_ops_test.cc [new file with mode: 0644] blob
tensorflow/core/ops/scoped_allocator_ops.cc [new file with mode: 0644] blob
Domain: Machine Learning / ML Framework;