summaryrefslogtreecommitdiff
path: root/compiler/tflchef/tflite/src/Convert.h
diff options
context:
space:
mode:
Diffstat (limited to 'compiler/tflchef/tflite/src/Convert.h')
-rw-r--r--compiler/tflchef/tflite/src/Convert.h2
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);