rust: Add missing cases for ORT tensor handling.

robohouse-delft · Nov 23, 2023 · 5fb71ef · 5fb71ef
1 parent 9a050a1
commit 5fb71ef
Showing 1 changed file with 12 additions and 0 deletions.
diff --git a/rust/onnxruntime/src/tensor/ort_output_tensor.rs b/rust/onnxruntime/src/tensor/ort_output_tensor.rs
@@ -341,6 +341,18 @@ impl<'a> TryFrom<OrtOutputTensor> for OrtOutput<'a> {
                 sys::ONNXTensorElementDataType::ONNX_TENSOR_ELEMENT_DATA_TYPE_BFLOAT16 => {
                     unimplemented!()
                 }
+                sys::ONNXTensorElementDataType::ONNX_TENSOR_ELEMENT_DATA_TYPE_FLOAT8E4M3FN => {
+                    unimplemented!()
+                }
+                sys::ONNXTensorElementDataType::ONNX_TENSOR_ELEMENT_DATA_TYPE_FLOAT8E4M3FNUZ => {
+                    unimplemented!()
+                }
+                sys::ONNXTensorElementDataType::ONNX_TENSOR_ELEMENT_DATA_TYPE_FLOAT8E5M2 => {
+                    unimplemented!()
+                }
+                sys::ONNXTensorElementDataType::ONNX_TENSOR_ELEMENT_DATA_TYPE_FLOAT8E5M2FNUZ => {
+                    unimplemented!()
+                }
             }
         }
     }