diff options
Diffstat (limited to 'compiler/tflchef/tflite/src/FillerHelper.cpp')
-rw-r--r-- | compiler/tflchef/tflite/src/FillerHelper.cpp | 15 |
1 files changed, 15 insertions, 0 deletions
diff --git a/compiler/tflchef/tflite/src/FillerHelper.cpp b/compiler/tflchef/tflite/src/FillerHelper.cpp index cf96d2e8c..1ac99ad40 100644 --- a/compiler/tflchef/tflite/src/FillerHelper.cpp +++ b/compiler/tflchef/tflite/src/FillerHelper.cpp @@ -48,3 +48,18 @@ void fill_tensor_to_import(int32_t idx, TFliteImport *import) } } // namespace tflchef + +// helpers of common codes for filling inputs +namespace tflchef +{ + +void fill_two_inputs(const tflite::Operator *op, TFliteImport *import) +{ + const std::vector<int32_t> &inputs = as_index_vector(op->inputs()); + assert(inputs.size() == 2); + + fill_tensor_to_import(inputs[0], import); + fill_tensor_to_import(inputs[1], import); +} + +} // namespace tflchef |