diff options
| author |  A. Unique TensorFlower <gardener@tensorflow.org> | 2018-09-21 10:21:48 -0700 |
|---|---|---|
| committer | TensorFlower Gardener <gardener@tensorflow.org> | 2018-09-21 10:26:21 -0700 |
| commit | 010e8ed731d0e10c82fccbf6c119180ca1a36efd (patch) | |
| tree | 607103b255979fc1e459eeb624c7c2c283d8c00f /tensorflow/contrib/lite/kernels | |
| parent | 4e252b2f997904769711b242bb37027706b08b7f (diff) | |
Kernel signature reworking, misc fixes.
PiperOrigin-RevId: 214004752
Diffstat (limited to 'tensorflow/contrib/lite/kernels')
| -rw-r--r-- | tensorflow/contrib/lite/kernels/internal/optimized/optimized_ops.h | 5 |
1 files changed, 1 insertions, 4 deletions
diff --git a/tensorflow/contrib/lite/kernels/internal/optimized/optimized_ops.h b/tensorflow/contrib/lite/kernels/internal/optimized/optimized_ops.h index 1a2d45166a..0999738396 100644 --- a/tensorflow/contrib/lite/kernels/internal/optimized/optimized_ops.h +++ b/tensorflow/contrib/lite/kernels/internal/optimized/optimized_ops.h @@ -2256,10 +2256,7 @@ inline void HybridConv(const ConvParams& params, float* scaling_factors_ptr, const int output_rows = FlatSizeSkipDim(output_shape, 3); TFLITE_DCHECK_EQ(output_cols, filter_rows); TFLITE_DCHECK_EQ(output_rows, gemm_input_rows); - TFLITE_DCHECK_EQ(bias_shape.Dims(3), output_cols); - TFLITE_DCHECK_EQ(bias_shape.Dims(2), 1); - TFLITE_DCHECK_EQ(bias_shape.Dims(1), 1); - TFLITE_DCHECK_EQ(bias_shape.Dims(0), 1); + TFLITE_DCHECK_EQ(bias_shape.FlatSize(), output_cols); // MatrixBatchVectorMultiplyAccumulate assumes that each row of the second // input matrix has its own scale factor. This code duplicates the scale |