diff options
| author |  A. Unique TensorFlower <nobody@tensorflow.org> | 2016-04-18 08:08:23 -0800 |
|---|---|---|
| committer |  TensorFlower Gardener <gardener@tensorflow.org> | 2016-04-18 09:12:35 -0700 |
| commit | 3c280f6fa0e0fcaa3d2cee5d2d8bb7ab3e25319f (patch) | |
| tree | fc67b33f56cc465486453b49789ea6a4d97b639d /tensorflow/core/kernels/immutable_constant_op.h | |
| parent | 517d3af445d85e2f6945fcdfc4fed4e46b1e0e35 (diff) | |
Added a format for saving an inference graph that can be memmapped and an utility to convert a freezed graph into this format.
Change: 120128412
Diffstat (limited to 'tensorflow/core/kernels/immutable_constant_op.h')
| -rw-r--r-- | tensorflow/core/kernels/immutable_constant_op.h | 5 |
1 files changed, 5 insertions, 0 deletions
diff --git a/tensorflow/core/kernels/immutable_constant_op.h b/tensorflow/core/kernels/immutable_constant_op.h index a731fa4b97..ecfc1a027a 100644 --- a/tensorflow/core/kernels/immutable_constant_op.h +++ b/tensorflow/core/kernels/immutable_constant_op.h @@ -32,6 +32,11 @@ class ImmutableConstantOp : public OpKernel { bool IsExpensive() override { return false; } ~ImmutableConstantOp() override; + // Names of attributes that are used by this op + static constexpr char kDTypeAttr[] = "dtype"; + static constexpr char kShapeAttr[] = "shape"; + static constexpr char kMemoryRegionNameAttr[] = "memory_region_name"; + private: class ReadOnlyMemoryRegionAllocator : public ::tensorflow::Allocator { public: |