diff options
author | 2018-02-04 11:24:51 -0800 | |
---|---|---|
committer | 2018-02-04 11:28:47 -0800 | |
commit | fc09f65a5d283baa9af182536e3e3652c7a41dd7 (patch) | |
tree | ded2909f5abbd5aff90c5d22e02d5a03ae059afb /arm_compiler.BUILD | |
parent | de6037d7505cd24b24fd937d1c011c8d24ca0e81 (diff) |
Avoid retaining two copies of each constant in `ConstantOp`.
Presently, the kernel keeps two copies of the constant tensor value, which can be large:
1. In the `ConstantOp::tensor_` field.
2. In the `OpKernel::def_` field (as an attr of the `NodeDef`).
Since we can be sure that `ConstantOp` will never need to access the
tensor value from `OpKernel::def_`, this change introduces a mechanism
for `OpKernel` implementations to store a stripped `NodeDef` in the
base class, and uses it in `ConstantOp` to avoid storing the tensor
value attr.
PiperOrigin-RevId: 184455793
Diffstat (limited to 'arm_compiler.BUILD')
0 files changed, 0 insertions, 0 deletions