aboutsummaryrefslogtreecommitdiffhomepage
path: root/tensorflow/contrib/lite/toco/model.h
diff options
context:
space:
mode:
authorGravatar A. Unique TensorFlower <gardener@tensorflow.org>2018-06-08 14:58:43 -0700
committerGravatar TensorFlower Gardener <gardener@tensorflow.org>2018-06-08 15:01:43 -0700
commitbc65583b2b4e3f48b6a724832ef96ab176666d33 (patch)
tree67899478d7fae7244cc1751e2cf528473cdde625 /tensorflow/contrib/lite/toco/model.h
parent77f0772c0ead3e1402615022649aad2a721265fd (diff)
Allow large allocations in toco.
PiperOrigin-RevId: 199855838
Diffstat (limited to 'tensorflow/contrib/lite/toco/model.h')
-rw-r--r--tensorflow/contrib/lite/toco/model.h4
1 files changed, 2 insertions, 2 deletions
diff --git a/tensorflow/contrib/lite/toco/model.h b/tensorflow/contrib/lite/toco/model.h
index 2ec36d27ef..2f43adb07b 100644
--- a/tensorflow/contrib/lite/toco/model.h
+++ b/tensorflow/contrib/lite/toco/model.h
@@ -1644,8 +1644,8 @@ struct SparseToDenseOperator : Operator {
// be used for the transient array at hand. The 'start' and 'end' values are
// offsets from the start of the workspace buffer, expressed in bytes.
struct Alloc {
- int start = 0;
- int end = 0;
+ int64 start = 0;
+ int64 end = 0;
};
inline bool operator<(const Alloc& a, const Alloc& b) {