diff options
Diffstat (limited to 'compiler/tflchef')
-rw-r--r-- | compiler/tflchef/core/src/CustomOp/MaxPoolWithArgMax.cpp | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/compiler/tflchef/core/src/CustomOp/MaxPoolWithArgMax.cpp b/compiler/tflchef/core/src/CustomOp/MaxPoolWithArgMax.cpp index b1c92ecbd..13bf2e5e9 100644 --- a/compiler/tflchef/core/src/CustomOp/MaxPoolWithArgMax.cpp +++ b/compiler/tflchef/core/src/CustomOp/MaxPoolWithArgMax.cpp @@ -65,13 +65,13 @@ MaxPoolWithArgMaxChef::custom_value(flatbuffers::FlatBufferBuilder &fbb) const flex_buffers->Add(1); flex_buffers->EndVector(start, /*typed=*/true, /*fixed=*/false); auto output_type = operation.max_pool_with_argmax_options().output_type(); - assert(output_type == tflite::TensorType_INT64 || output_type == tflite::TensorType_INT32); + assert(output_type == tflchef::INT64 || output_type == tflchef::INT32); flex_buffers->Int("Targmax", output_type); std::string padding = operation.max_pool_with_argmax_options().padding() ? "VALID" : "SAME"; flex_buffers->String("padding", padding); flex_buffers->Bool("include_batch_in_index", operation.max_pool_with_argmax_options().include_batch_in_index()); - flex_buffers->Int("T", tflite::TensorType_FLOAT32); + flex_buffers->Int("T", tflchef::FLOAT32); flex_buffers->EndMap(map_start); flex_buffers->Finish(); |