diff options
| author |  Alan Chiao <alanchiao@google.com> | 2018-08-31 08:01:48 -0700 |
|---|---|---|
| committer |  TensorFlower Gardener <gardener@tensorflow.org> | 2018-08-31 08:06:27 -0700 |
| commit | 4902db818664b62f60c0f9572cc939f596d1c8bc (patch) | |
| tree | a722b04cecd1f7f92aea0f0afeb3fcc119f9f65d /tensorflow/contrib/lite/kernels/internal | |
| parent | a55ed5bd3c429dfff20654ccbacec2521c1b20d2 (diff) | |
Fixes to hybrid conv. Add additional tests for pointwise conv.
PiperOrigin-RevId: 211085787
Diffstat (limited to 'tensorflow/contrib/lite/kernels/internal')
| -rw-r--r-- | tensorflow/contrib/lite/kernels/internal/optimized/optimized_ops.h | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/tensorflow/contrib/lite/kernels/internal/optimized/optimized_ops.h b/tensorflow/contrib/lite/kernels/internal/optimized/optimized_ops.h index 85e631b852..70adffda3b 100644 --- a/tensorflow/contrib/lite/kernels/internal/optimized/optimized_ops.h +++ b/tensorflow/contrib/lite/kernels/internal/optimized/optimized_ops.h @@ -1948,7 +1948,7 @@ inline void HybridConv(const int8_t* input_data, const Dims<4>& input_dims, const int filter_width = ArraySize(filter_dims, 1); const int filter_height = ArraySize(filter_dims, 2); - const int8* gemm_input_data = nullptr; + const int8_t* gemm_input_data = nullptr; int num_input; const bool need_im2col = stride_width != 1 || stride_height != 1 || filter_width != 1 || filter_height != 1; |