diff options
author | A. Unique TensorFlower <gardener@tensorflow.org> | 2018-06-21 12:41:24 -0700 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2018-06-21 12:44:26 -0700 |
commit | 846520326327d5eb8e1be13cad7d5526adf67db2 (patch) | |
tree | 77f3f0a266306c8d8591a5ecf617d13a1eb94440 /tensorflow/contrib/lite/toco/model.h | |
parent | fc4484c359cab66bd5bfdfaab936b1a5128850be (diff) |
Graduate the 'experimental' fully connected weights shuffling feature
(which enables faster runtime kernels) as no longer experimental,
no longer behind a flag, and replace the
bool experimental_shuffled_weights
field on FullyConnectedOperator with an enum.
PiperOrigin-RevId: 201569224
Diffstat (limited to 'tensorflow/contrib/lite/toco/model.h')
-rw-r--r-- | tensorflow/contrib/lite/toco/model.h | 3 |
1 files changed, 2 insertions, 1 deletions
diff --git a/tensorflow/contrib/lite/toco/model.h b/tensorflow/contrib/lite/toco/model.h index ef170b3884..89cb061499 100644 --- a/tensorflow/contrib/lite/toco/model.h +++ b/tensorflow/contrib/lite/toco/model.h @@ -433,7 +433,8 @@ struct SpaceToDepthOperator : Operator { // input activations as a matrix, followed by a MatMul node. struct FullyConnectedOperator : Operator { FullyConnectedOperator() : Operator(OperatorType::kFullyConnected) {} - bool experimental_shuffled_weights = false; + FullyConnectedWeightsFormat weights_format = + FullyConnectedWeightsFormat::kDefault; }; // Dequantization operator, converting a quantized array of integers with |