| |
| // Copyright (c) 2022-2024, ARM Limited. |
| // |
| // Licensed under the Apache License, Version 2.0 (the "License"); |
| // you may not use this file except in compliance with the License. |
| // You may obtain a copy of the License at |
| // |
| // http://www.apache.org/licenses/LICENSE-2.0 |
| // |
| // Unless required by applicable law or agreed to in writing, software |
| // distributed under the License is distributed on an "AS IS" BASIS, |
| // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
| // See the License for the specific language governing permissions and |
| // limitations under the License. |
| |
| #ifndef TYPES_H_ |
| #define TYPES_H_ |
| |
| #include <stddef.h> |
| #include <stdint.h> |
| |
| #ifdef __cplusplus |
| extern "C" |
| { |
| #endif /* __cplusplus */ |
| |
| enum tosa_datatype_t |
| { |
| tosa_datatype_bf16_t = 0, |
| tosa_datatype_bool_t = 1, |
| tosa_datatype_fp16_t = 2, |
| tosa_datatype_fp32_t = 3, |
| tosa_datatype_int16_t = 4, |
| tosa_datatype_int32_t = 5, |
| tosa_datatype_int48_t = 6, |
| tosa_datatype_int4_t = 7, |
| tosa_datatype_int8_t = 8, |
| tosa_datatype_uint16_t = 9, |
| tosa_datatype_uint8_t = 10, |
| tosa_datatype_shape_t = 11, |
| tosa_datatype_fp8e4m3_t = 12, |
| tosa_datatype_fp8e5m2_t = 13, |
| tosa_datatype_fp64_t = 99 |
| }; |
| |
| struct tosa_tensor_t |
| { |
| const char* name; |
| int32_t* shape; |
| int32_t num_dims; |
| tosa_datatype_t data_type; |
| uint8_t* data; |
| size_t size; |
| }; |
| |
| struct tosa_tensor_list_t |
| { |
| tosa_tensor_t* tensors; |
| size_t size; |
| }; |
| |
| #ifdef __cplusplus |
| } |
| #endif /* __cplusplus */ |
| |
| #endif // TYPES_H_ |