diff options
Diffstat (limited to 'compiler/tflchef/tflite/src/Convert.h')
-rw-r--r-- | compiler/tflchef/tflite/src/Convert.h | 2 |
1 files changed, 2 insertions, 0 deletions
diff --git a/compiler/tflchef/tflite/src/Convert.h b/compiler/tflchef/tflite/src/Convert.h index 8623e7b78..770bffa4d 100644 --- a/compiler/tflchef/tflite/src/Convert.h +++ b/compiler/tflchef/tflite/src/Convert.h @@ -27,12 +27,14 @@ namespace tflchef tflchef::TensorType as_tflchef_type(const tflite::TensorType type); tflchef::Activation as_tflchef_activation(const tflite::ActivationFunctionType type); tflchef::Padding as_tflchef_padding(const tflite::Padding padding); +tflchef::MirrorPadMode as_tflchef_mirrorpadmode(const tflite::MirrorPadMode mode); /** * @brief extract buffer data to std::vector<DT> */ template <typename DT> std::vector<DT> extract_buffer(const tflite::Buffer *buffer) { + assert(buffer->data() != nullptr); auto buffer_length = buffer->data()->size(); auto num_elements = buffer_length / sizeof(DT); std::vector<DT> result(num_elements); |