diff options
| author |  A. Unique TensorFlower <gardener@tensorflow.org> | 2018-07-23 19:08:40 -0700 |
|---|---|---|
| committer | TensorFlower Gardener <gardener@tensorflow.org> | 2018-07-23 19:12:07 -0700 |
| commit | df7344f1933d932f03f472402068ff1883f0c011 (patch) | |
| tree | 56cd6abec5b5076195b86909eddd47696b4d3612 /tensorflow/contrib/lite/toco/import_tensorflow.cc | |
| parent | efe370fcb367efd069c8166120858492dffa9a33 (diff) | |
Implementation of stack.
PiperOrigin-RevId: 205763219
Diffstat (limited to 'tensorflow/contrib/lite/toco/import_tensorflow.cc')
| -rw-r--r-- | tensorflow/contrib/lite/toco/import_tensorflow.cc | 1 |
1 files changed, 1 insertions, 0 deletions
diff --git a/tensorflow/contrib/lite/toco/import_tensorflow.cc b/tensorflow/contrib/lite/toco/import_tensorflow.cc index 8bb797fe0f..032c863945 100644 --- a/tensorflow/contrib/lite/toco/import_tensorflow.cc +++ b/tensorflow/contrib/lite/toco/import_tensorflow.cc @@ -1529,6 +1529,7 @@ tensorflow::Status ConvertPackOperator( for (int i = 0; i < num_inputs; ++i) { op->inputs.push_back(node.input(i)); } + op->values_count = HasAttr(node, "N") ? GetIntAttr(node, "N") : num_inputs; op->axis = HasAttr(node, "axis") ? GetIntAttr(node, "axis") : 0; op->dtype = ConvertDataType(toco::GetDataTypeAttr(node, "T")); op->outputs.push_back(node.name()); |