experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Add Pack to operator_properties.

Browse Source 
Fixes #33572

PiperOrigin-RevId: 281981340
Change-Id: I79e971c35f328131df684f80a6cb9aa1d77abb4c
This commit is contained in:
Suharsh Sivakumar 2019-11-22 09:15:47 -08:00 committed by TensorFlower Gardener
parent e515908529
commit d8668d9e03
1 changed files with 6 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
tensorflow/lite/tools/optimize/operator_property.cc
View File

@ -286,6 +286,12 @@ OperatorProperty GetOperatorProperty(const ModelT* model, int subgraph_index,
property.outputs = {{0, {}}};
property.version = 2;
break;
case BuiltinOperator_PACK:
property.arbitrary_inputs = true;
property.outputs = {{0, {}}};
property.restrict_same_input_output_scale = true;
property.version = 2;
break;
case BuiltinOperator_PAD:
case BuiltinOperator_PADV2:
property.inputs = {{0, {}}};