diff options
| author |  A. Unique TensorFlower <gardener@tensorflow.org> | 2018-06-08 13:14:59 -0700 |
|---|---|---|
| committer | TensorFlower Gardener <gardener@tensorflow.org> | 2018-06-08 13:18:20 -0700 |
| commit | 3b81d6e6055c529c00a165fd8e3431a6ba704e8e (patch) | |
| tree | c49f2661b14d3af7587634d21300004440712672 /tensorflow/contrib/lite/toco/graph_transformations/propagate_fixed_sizes.cc | |
| parent | 9affc2080bf9840f4c7da2990ba528114e25d3b1 (diff) | |
Optimizing transpose_conv.
PiperOrigin-RevId: 199839745
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 9e4262223e..170a499d4e 100644 --- a/tensorflow/contrib/lite/toco/graph_transformations/propagate_fixed_sizes.cc +++ b/tensorflow/contrib/lite/toco/graph_transformations/propagate_fixed_sizes.cc @@ -278,7 +278,7 @@ void ProcessTransposeConvOperator(Model* model, TransposeConvOperator* op) { << "TransposeConv input shape must have 4 dimensions. Input \"" << op->inputs[TransposeConvOperator::WEIGHTS] << "\" had shape " << toco::ShapeToString(weights_shape) << "."; - CHECK_EQ(input_shape.dims(3), weights_shape.dims(0)) + CHECK_EQ(input_shape.dims(3), weights_shape.dims(3)) << "Input shape depth and weight depth do not agree"; // Set the output shape according to the specified output shape. |