diff options
| author |  A. Unique TensorFlower <gardener@tensorflow.org> | 2018-06-29 12:52:33 -0700 |
|---|---|---|
| committer | TensorFlower Gardener <gardener@tensorflow.org> | 2018-06-29 12:55:32 -0700 |
| commit | 5be44f95a612b59be2182769c6650e4bb82c7355 (patch) | |
| tree | 8826c2141c4edf6c7f3e2e284417889183052cbe /tensorflow/contrib/lite/toco/model.h | |
| parent | 4b988cac015c1b464b2f83024a2853f628a7c938 (diff) | |
Allow transposition of the weights in fully connected ops.
PiperOrigin-RevId: 202693036
Diffstat (limited to 'tensorflow/contrib/lite/toco/model.h')
| -rw-r--r-- | tensorflow/contrib/lite/toco/model.h | 2 |
1 files changed, 2 insertions, 0 deletions
diff --git a/tensorflow/contrib/lite/toco/model.h b/tensorflow/contrib/lite/toco/model.h index abe0bf3c54..3a1d243f87 100644 --- a/tensorflow/contrib/lite/toco/model.h +++ b/tensorflow/contrib/lite/toco/model.h @@ -837,6 +837,8 @@ struct BatchMatMulOperator : Operator { // TensorFlow equivalent: MatMul struct TensorFlowMatMulOperator : Operator { TensorFlowMatMulOperator() : Operator(OperatorType::kMatMul) {} + bool transpose_a = false; + bool transpose_b = false; }; // Padding operator. Pads a tensor with zeros. |