diff options
author | A. Unique TensorFlower <gardener@tensorflow.org> | 2018-07-17 12:04:04 -0700 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2018-07-17 12:07:13 -0700 |
commit | 9b12cb84292d23522c1c3f75700d97d9f9af8abd (patch) | |
tree | 2806fabac9dc13e3d29f496daf15e01f1a6ab22e /tensorflow/contrib/lite/toco/model.h | |
parent | 9d126b26ff219d9258a78832ead0bb272b898bf0 (diff) |
toco thinks Stack is the same as Pack.
Stack is not Pack. tf.stack() yields Pack ops. Pack ops stack tensors. Stack ops manipulate the TF runtime stack.
This cl unaliases "Stack" and "Pack" ops in toco, and renames most things that refer to "Stack" ops to "Pack" to be consistent across the codebase.
In summary:
Stack is whack. 'Stack' should be 'Pack'. Hack 'Stack's into 'Pack's like a maniac. This keeps 'Stack's from wracking runtime graphs.
(We apologize for the fault in the change description. Those responsible have been...er...sacked).
PiperOrigin-RevId: 204951155
Diffstat (limited to 'tensorflow/contrib/lite/toco/model.h')
-rw-r--r-- | tensorflow/contrib/lite/toco/model.h | 9 |
1 files changed, 5 insertions, 4 deletions
diff --git a/tensorflow/contrib/lite/toco/model.h b/tensorflow/contrib/lite/toco/model.h index 37f4188cf7..8fff68cf47 100644 --- a/tensorflow/contrib/lite/toco/model.h +++ b/tensorflow/contrib/lite/toco/model.h @@ -81,7 +81,7 @@ enum class OperatorType : uint8 { kResizeBilinear, kSin, kSpaceToBatchND, - kStack, + kPack, kBatchToSpaceND, kPad, kPadV2, @@ -1157,10 +1157,11 @@ struct TensorFlowRsqrtOperator : Operator { // Inputs: this operator accepts any number >= 1 of inputs. // inputs[i]: the i-th array to merge. // -// TensorFlow equivalent: Stack or Pack -struct StackOperator : Operator { - StackOperator() : Operator(OperatorType::kStack) {} +// TensorFlow equivalent: Pack +struct PackOperator : Operator { + PackOperator() : Operator(OperatorType::kPack) {} int axis = 0; + ArrayDataType dtype = ArrayDataType::kNone; }; // Shape operator. Extracts the shape of the tensor. |