diff options
Diffstat (limited to 'compiler/tflchef/tflite/src/Op/PadV2.cpp')
-rw-r--r-- | compiler/tflchef/tflite/src/Op/PadV2.cpp | 7 |
1 files changed, 5 insertions, 2 deletions
diff --git a/compiler/tflchef/tflite/src/Op/PadV2.cpp b/compiler/tflchef/tflite/src/Op/PadV2.cpp index 0b1c9f3b2..a6b657f59 100644 --- a/compiler/tflchef/tflite/src/Op/PadV2.cpp +++ b/compiler/tflchef/tflite/src/Op/PadV2.cpp @@ -16,6 +16,7 @@ #include "PadV2.h" +#include "Convert.h" #include "FillerHelper.h" namespace tflchef @@ -24,9 +25,11 @@ namespace tflchef void TFliteOpPadV2::filler(const tflite::Operator *op, TFliteImport *import, tflchef::ModelRecipe *model_recipe) const { + const std::vector<int32_t> &inputs = as_index_vector(op->inputs()); + // Filler for paddings and constant_values - fill_tensor_to_import(1, import); - fill_tensor_to_import(2, import); + fill_tensor_to_import(inputs[1], import); + fill_tensor_to_import(inputs[2], import); } tflchef::Operation *TFliteOpPadV2::build(const tflite::Operator *op, TFliteImport *import, |