diff options
| author |  Jared Duke <jdduke@google.com> | 2018-06-20 11:48:15 -0700 |
|---|---|---|
| committer |  TensorFlower Gardener <gardener@tensorflow.org> | 2018-06-20 11:51:26 -0700 |
| commit | 4efefb90391b12c95339ed3b46a02b62ea5e195d (patch) | |
| tree | bb3f9bb986b89287983ea8e7c35827993aad7206 /tensorflow/contrib/lite/toco/model.h | |
| parent | e51df5918020cdfada26022240091e5529f7da60 (diff) | |
Implement TFLite Shape operator
PiperOrigin-RevId: 201389618
Diffstat (limited to 'tensorflow/contrib/lite/toco/model.h')
| -rw-r--r-- | tensorflow/contrib/lite/toco/model.h | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/tensorflow/contrib/lite/toco/model.h b/tensorflow/contrib/lite/toco/model.h index 0faadedf3b..2585cff56e 100644 --- a/tensorflow/contrib/lite/toco/model.h +++ b/tensorflow/contrib/lite/toco/model.h @@ -1157,10 +1157,10 @@ struct StackOperator : Operator { // This operation outputs a 1-D integer tensor representing the shape of // the input. // -// TensorFlow equivalent: Shape. We currently assume that the output is int32 -// and not int64. The output type could be stored herein. +// TensorFlow equivalent: Shape. struct TensorFlowShapeOperator : Operator { TensorFlowShapeOperator() : Operator(OperatorType::kTensorFlowShape) {} + ArrayDataType output_data_type = ArrayDataType::kInt32; }; // Element-wise square-root (x^0.5) operator. |