Model: "sequential"
_________________________________________________________________
 Layer (type)                Output Shape              Param #
=================================================================
 dense (Dense)               (None, 9, 9, 81)          243

 dense_1 (Dense)             (None, 9, 9, 81)          6642

 flatten (Flatten)           (None, 6561)              0

 dense_2 (Dense)             (None, 82)                538084

=================================================================
Total params: 544,969
Trainable params: 544,969
Non-trainable params: 0
_________________________________________________________________