diff --git a/lib/transpose_conv_bias.cc b/lib/transpose_conv_bias.cc index 169b3ca..4df1e76 100644 --- a/lib/transpose_conv_bias.cc +++ b/lib/transpose_conv_bias.cc @@ -221,11 +221,13 @@ TfLiteStatus Eval(TfLiteContext* context, TfLiteNode* node) { op_params.stride_height = stride_height; TransposeConvBias( - op_params, ::tflite::GetTensorShape(input), + op_params, + ::tflite::GetTensorShape(input), ::tflite::GetTensorData(input), ::tflite::GetTensorShape(weights), ::tflite::GetTensorData(weights), - ::tflite::GetTensorShape(bias), ::tflite::GetTensorData(bias), + ::tflite::GetTensorShape(bias), + ::tflite::GetTensorData(bias), ::tflite::GetTensorShape(output), ::tflite::GetTensorData(output), // Last two args specify im2col which reference_ops ignores.