diff options
author | A. Unique TensorFlower <gardener@tensorflow.org> | 2018-06-20 11:49:22 -0700 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2018-06-20 11:54:27 -0700 |
commit | c40ed1d7cec07a0a8ffdfd263689e5db4fe38cc8 (patch) | |
tree | 840e293801aa0b30e3c0761b3a433e1b1db9abf5 /tensorflow/contrib/lite/toco/graph_transformations/propagate_fixed_sizes.cc | |
parent | 4efefb90391b12c95339ed3b46a02b62ea5e195d (diff) |
Fix a bug: the conversion of pure Conv to DepthwiseConv did not
properly check the necessary precondition that the input depth is 1.
PiperOrigin-RevId: 201389819
Diffstat (limited to 'tensorflow/contrib/lite/toco/graph_transformations/propagate_fixed_sizes.cc')
-rw-r--r-- | tensorflow/contrib/lite/toco/graph_transformations/propagate_fixed_sizes.cc | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/tensorflow/contrib/lite/toco/graph_transformations/propagate_fixed_sizes.cc b/tensorflow/contrib/lite/toco/graph_transformations/propagate_fixed_sizes.cc index e7da9051d8..beda187f13 100644 --- a/tensorflow/contrib/lite/toco/graph_transformations/propagate_fixed_sizes.cc +++ b/tensorflow/contrib/lite/toco/graph_transformations/propagate_fixed_sizes.cc @@ -325,7 +325,7 @@ void ProcessDepthwiseConvOperator(Model* model, DepthwiseConvOperator* op) { if (!op->depth_multiplier) { op->depth_multiplier = output_depth / input_depth; } - QCHECK_EQ(output_depth, input_depth * op->depth_multiplier) + CHECK_EQ(output_depth, input_depth * op->depth_multiplier) << "input/output depths and depth_multiplier don't match"; const int kheight = weights_shape.dims(1); |