diff options
author | Suharsh Sivakumar <suharshs@google.com> | 2018-09-13 15:01:08 -0700 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2018-09-13 15:09:10 -0700 |
commit | fb50c8e9a3cb2ccfac9cf4a847d5841cba80b524 (patch) | |
tree | 969d44684c674fad7ef775323f6e150e616890d9 /tensorflow/contrib/lite/toco/model.h | |
parent | e8af4e1bb9496c111530e88263fb1b8dac8bdde9 (diff) |
Dilated Depthwise Conv reference implementations.
PiperOrigin-RevId: 212884951
Diffstat (limited to 'tensorflow/contrib/lite/toco/model.h')
-rw-r--r-- | tensorflow/contrib/lite/toco/model.h | 5 |
1 files changed, 5 insertions, 0 deletions
diff --git a/tensorflow/contrib/lite/toco/model.h b/tensorflow/contrib/lite/toco/model.h index 2e100e37f6..164b70f2df 100644 --- a/tensorflow/contrib/lite/toco/model.h +++ b/tensorflow/contrib/lite/toco/model.h @@ -477,6 +477,11 @@ struct DepthwiseConvOperator : Operator { int stride_height = 0; int stride_width = 0; int depth_multiplier = 0; + // A dilation_rate of 0 is invalid and this field is an optional attribute. + // Thus initializing it to 1 to allow default conv behavior when the + // attribute is not present. + int dilation_width_factor = 1; + int dilation_height_factor = 1; }; // Depth-to-space transform operator. |