xmos-ai-tools 1.1.2.dev216__py3-none-macosx_11_0_arm64.whl → 1.1.2.dev236__py3-none-macosx_11_0_arm64.whl

xmos_ai_tools/runtime/include/tensorflow/lite/core/c/builtin_op_data.h

@@ -20,6 +20,7 @@ limitations under the License.
 #ifndef TENSORFLOW_LITE_CORE_C_BUILTIN_OP_DATA_H_
 #define TENSORFLOW_LITE_CORE_C_BUILTIN_OP_DATA_H_
 
+#include <stdbool.h>
 #include <stdint.h>
 
 #include "tensorflow/lite/core/c/common.h"
@@ -31,6 +32,8 @@ extern "C" {
 // TfLiteReshapeParams can't have dynamic data so we fix the maximum possible
 // number of dimensions.
 #define TFLITE_RESHAPE_PARAMS_MAX_DIMENSION_COUNT 8
+#define TFLITE_STABLEHLO_SCATTER_PARAMS_MAX_DIMENSION_COUNT 8
+#define TFLITE_STABLEHLO_GATHER_PARAMS_MAX_DIMENSION_COUNT 8
 
 // TODO(aselle): Consider using "if this then that" for testing.
 
@@ -89,6 +92,10 @@ typedef struct {
   // Note: Version 2 supports dilation values not equal to 1.
   int dilation_width_factor;
   int dilation_height_factor;
+
+  // Parameters for CONV_2D version 7 or above.
+  // Used to determine the default value for the quantized bias.
+  TfLiteType quantized_bias_type;
 } TfLiteConvParams;
 
 typedef struct {
@@ -166,7 +173,7 @@ typedef struct {
   TfLiteFusedActivation activation;
   bool merge_outputs;
 
-  // Parameter for Bidirectional RNN verison 3.
+  // Parameter for Bidirectional RNN version 3.
   bool asymmetric_quantize_inputs;
 } TfLiteBidirectionalSequenceRNNParams;
 
@@ -192,6 +199,10 @@ typedef struct {
   // If set to true and the weights are quantized, then non constant inputs
   // are quantized at evaluation time with asymmetric quantization.
   bool asymmetric_quantize_inputs;
+
+  // Parameters for FullyConnected version 10 or above.
+  // Used to determine the default value for the quantized bias.
+  TfLiteType quantized_bias_type;
 } TfLiteFullyConnectedParams;
 
 typedef enum {
@@ -340,7 +351,7 @@ typedef struct {
   // These fields are only used in old models for backward compatibility.
   // In the current implementation, we use the 2nd input of the op as the shape,
   // and these fields are unused.
-  int shape[TFLITE_RESHAPE_PARAMS_MAX_DIMENSION_COUNT];
+  int32_t shape[TFLITE_RESHAPE_PARAMS_MAX_DIMENSION_COUNT];
   int num_dimensions;
 } TfLiteReshapeParams;
 
@@ -397,7 +408,7 @@ typedef struct {
 typedef struct {
   // TODO(ahentz): We can't have dynamic data in this struct, at least not yet.
   // For now we will fix the maximum possible number of dimensions.
-  int squeeze_dims[8];
+  int32_t squeeze_dims[8];
   int num_squeeze_dims;
 } TfLiteSqueezeParams;
 
@@ -407,6 +418,10 @@ typedef struct {
   int ellipsis_mask;
   int new_axis_mask;
   int shrink_axis_mask;
+
+  // Parameters supported by version 8:
+  // If true, then the end tensor is an offset of the begin tensor.
+  bool offset;
 } TfLiteStridedSliceParams;
 
 typedef struct {
@@ -425,6 +440,10 @@ typedef struct {
 
   // Parameters supported by version 4:
   TfLiteFusedActivation activation;
+
+  // Parameters for TransposeConv version 5 or above.
+  // Used to determine the default value for the quantized bias.
+  TfLiteType quantized_bias_type;
 } TfLiteTransposeConvParams;
 
 typedef struct {
@@ -530,6 +549,76 @@ typedef struct {
   bool approximate;
 } TfLiteGeluParams;
 
+typedef struct {
+  int64_t dimension;
+} TfLiteStablehloConcatenateParams;
+
+typedef struct {
+  // See the stablehlo spec for the explanation of the attributes:
+  // https://github.com/openxla/stablehlo/blob/main/docs/spec.md#scatter
+  bool indices_are_sorted;
+  int64_t
+      update_window_dims[TFLITE_STABLEHLO_SCATTER_PARAMS_MAX_DIMENSION_COUNT];
+  int num_update_window_dims;
+  int64_t
+      inserted_window_dims[TFLITE_STABLEHLO_SCATTER_PARAMS_MAX_DIMENSION_COUNT];
+  int num_inserted_window_dims;
+  int64_t scatter_dims_to_operand_dims
+      [TFLITE_STABLEHLO_SCATTER_PARAMS_MAX_DIMENSION_COUNT];
+  int num_scatter_dims_to_operand_dims;
+  int64_t index_vector_dim;
+  bool unique_indices;
+  int update_computation_subgraph_index;
+} TfLiteStablehloScatterParams;
+
+typedef enum {
+  kTfLiteRngAlgorithmUnknown = 0,
+  // An algorithm auto-selected by the system according to device type.
+  kTfLiteRngAlgorithmDefault,
+  // The Philox algorithm, as described in paper
+  // ['Parallel Random Numbers: As Easy as 1, 2, 3']
+  // (https://www.thesalmons.org/john/random123/papers/random123sc11.pdf)
+  kTfLiteRngAlgorithmPhilox,
+  // The ThreeFry algorithm, as described in paper
+  // ['Parallel Random Numbers: As Easy as 1, 2, 3']
+  // (https://www.thesalmons.org/john/random123/papers/random123sc11.pdf)
+  kTfLiteRngAlgorithmThreefry,
+} TfLiteRngAlgorithm;
+
+typedef struct {
+  TfLiteRngAlgorithm algorithm;
+} TfLiteStablehloRngBitGeneratorParams;
+
+typedef struct {
+  // See the stablehlo spec for the explanation of the attributes:
+  // https://github.com/openxla/stablehlo/blob/main/docs/spec.md#gather
+  int64_t offset_dims[TFLITE_STABLEHLO_GATHER_PARAMS_MAX_DIMENSION_COUNT];
+  int num_offset_dims;
+  int64_t
+      collapsed_slice_dims[TFLITE_STABLEHLO_GATHER_PARAMS_MAX_DIMENSION_COUNT];
+  int num_collapsed_slice_dims;
+  int64_t start_index_map[TFLITE_STABLEHLO_GATHER_PARAMS_MAX_DIMENSION_COUNT];
+  int num_start_index_map;
+  int64_t index_vector_dim;
+  int64_t slice_sizes[TFLITE_STABLEHLO_GATHER_PARAMS_MAX_DIMENSION_COUNT];
+  int num_slice_sizes;
+  bool indices_are_sorted;
+} TfLiteStablehloGatherParams;
+
+enum TfLiteReduceWindowFunction {
+  TfLiteReduceWindowFunctionUnsupported,
+  TfLiteReduceWindowFunctionAdd,
+  TfLiteReduceWindowFunctionMul,
+  TfLiteReduceWindowFunctionMin,
+  TfLiteReduceWindowFunctionMax,
+  TfLiteReduceWindowFunctionAll,
+  TfLiteReduceWindowFunctionAny
+};
+
+typedef struct {
+  enum TfLiteReduceWindowFunction reduce_function;
+} TfLiteReduceWindowParams;
+
 #ifdef __cplusplus
 }  // extern "C"
 #endif  // __cplusplus

xmos_ai_tools/runtime/include/tensorflow/lite/core/c/c_api_types.h

@@ -13,14 +13,17 @@ See the License for the specific language governing permissions and
 limitations under the License.
 ==============================================================================*/
 
-// This file declares types used by the pure C inference API defined in c_api.h,
-// some of which are also used in the C++ and C kernel and interpreter APIs.
+/// This file declares types used by the pure C inference API defined in
+/// c_api.h, some of which are also used in the C++ and C kernel and interpreter
+/// APIs.
 
-/// WARNING: Users of TensorFlow Lite should not include this file directly,
-/// but should instead include
-/// "third_party/tensorflow/lite/c/c_api_types.h".
-/// Only the TensorFlow Lite implementation itself should include this
-/// file directly.
+// WARNING: Users of TensorFlow Lite should not include this file directly,
+// but should instead include
+// "third_party/tensorflow/lite/c/c_api_types.h".
+// Only the TensorFlow Lite implementation itself should include this
+// file directly.
+
+// IWYU pragma: private, include "third_party/tensorflow/lite/c/c_api_types.h"
 
 #ifndef TENSORFLOW_LITE_CORE_C_C_API_TYPES_H_
 #define TENSORFLOW_LITE_CORE_C_C_API_TYPES_H_
@@ -31,6 +34,10 @@ limitations under the License.
 extern "C" {
 #endif
 
+/** \addtogroup c_api_types tensorflow/lite/c/c_api_types.h
+ *  @{
+ */
+
 // Define TFL_CAPI_EXPORT macro to export a function properly with a shared
 // library.
 #ifdef SWIG
@@ -49,50 +56,51 @@ extern "C" {
 #endif  // _WIN32
 #endif  // SWIG
 
-// Note that new error status values may be added in future in order to
-// indicate more fine-grained internal states, therefore, applications should
-// not rely on status values being members of the enum.
+/// Note that new error status values may be added in future in order to
+/// indicate more fine-grained internal states, therefore, applications should
+/// not rely on status values being members of the enum.
 typedef enum TfLiteStatus {
+  /// Success
   kTfLiteOk = 0,
 
-  // Generally referring to an error in the runtime (i.e. interpreter)
+  /// Generally referring to an error in the runtime (i.e. interpreter)
   kTfLiteError = 1,
 
-  // Generally referring to an error from a TfLiteDelegate itself.
+  /// Generally referring to an error from a TfLiteDelegate itself.
   kTfLiteDelegateError = 2,
 
-  // Generally referring to an error in applying a delegate due to
-  // incompatibility between runtime and delegate, e.g., this error is returned
-  // when trying to apply a TF Lite delegate onto a model graph that's already
-  // immutable.
+  /// Generally referring to an error in applying a delegate due to
+  /// incompatibility between runtime and delegate, e.g., this error is returned
+  /// when trying to apply a TF Lite delegate onto a model graph that's already
+  /// immutable.
   kTfLiteApplicationError = 3,
 
-  // Generally referring to serialized delegate data not being found.
-  // See tflite::delegates::Serialization.
+  /// Generally referring to serialized delegate data not being found.
+  /// See tflite::delegates::Serialization.
   kTfLiteDelegateDataNotFound = 4,
 
-  // Generally referring to data-writing issues in delegate serialization.
-  // See tflite::delegates::Serialization.
+  /// Generally referring to data-writing issues in delegate serialization.
+  /// See tflite::delegates::Serialization.
   kTfLiteDelegateDataWriteError = 5,
 
-  // Generally referring to data-reading issues in delegate serialization.
-  // See tflite::delegates::Serialization.
+  /// Generally referring to data-reading issues in delegate serialization.
+  /// See tflite::delegates::Serialization.
   kTfLiteDelegateDataReadError = 6,
 
-  // Generally referring to issues when the TF Lite model has ops that cannot be
-  // resolved at runtime. This could happen when the specific op is not
-  // registered or built with the TF Lite framework.
+  /// Generally referring to issues when the TF Lite model has ops that cannot
+  /// be resolved at runtime. This could happen when the specific op is not
+  /// registered or built with the TF Lite framework.
   kTfLiteUnresolvedOps = 7,
 
-  // Generally referring to invocation cancelled by the user.
-  // See `interpreter::Cancel`.
+  /// Generally referring to invocation cancelled by the user.
+  /// See `interpreter::Cancel`.
   // TODO(b/194915839): Implement `interpreter::Cancel`.
   // TODO(b/250636993): Cancellation triggered by `SetCancellationFunction`
   // should also return this status code.
   kTfLiteCancelled = 8,
 } TfLiteStatus;
 
-// Types supported by tensor
+/// Types supported by tensor
 typedef enum {
   kTfLiteNoType = 0,
   kTfLiteFloat32 = 1,
@@ -115,12 +123,12 @@ typedef enum {
   kTfLiteInt4 = 18,
 } TfLiteType;
 
-// Legacy. Will be deprecated in favor of TfLiteAffineQuantization.
-// If per-layer quantization is specified this field will still be populated in
-// addition to TfLiteAffineQuantization.
-// Parameters for asymmetric quantization. Quantized values can be converted
-// back to float using:
-//     real_value = scale * (quantized_value - zero_point)
+/// Legacy. Will be deprecated in favor of TfLiteAffineQuantization.
+/// If per-layer quantization is specified this field will still be populated in
+/// addition to TfLiteAffineQuantization.
+/// Parameters for asymmetric quantization. Quantized values can be converted
+/// back to float using:
+///     real_value = scale * (quantized_value - zero_point)
 typedef struct TfLiteQuantizationParams {
   float scale;
   int32_t zero_point;
@@ -129,39 +137,41 @@ typedef struct TfLiteQuantizationParams {
 // --------------------------------------------------------------------------
 // Opaque types used by c_api.h, c_api_opaque.h and common.h.
 
-// TfLiteOpaqueContext is an opaque version of TfLiteContext;
+/// TfLiteOpaqueContext is an opaque version of TfLiteContext;
 typedef struct TfLiteOpaqueContext TfLiteOpaqueContext;
 
-// TfLiteOpaqueNode is an opaque version of TfLiteNode;
+/// TfLiteOpaqueNode is an opaque version of TfLiteNode;
 typedef struct TfLiteOpaqueNode TfLiteOpaqueNode;
 
-// TfLiteOpaqueTensor is an opaque version of TfLiteTensor;
+/// TfLiteOpaqueTensor is an opaque version of TfLiteTensor;
 typedef struct TfLiteOpaqueTensor TfLiteOpaqueTensor;
 
-// TfLiteDelegate: allows delegation of nodes to alternative backends.
-// Forward declaration of concrete type declared in common.h.
+/// TfLiteDelegate: allows delegation of nodes to alternative backends.
+/// Forward declaration of concrete type declared in common.h.
 typedef struct TfLiteDelegate TfLiteDelegate;
 
-// TfLiteOpaqueDelegateStruct: unconditionally opaque version of
-// TfLiteDelegate; allows delegation of nodes to alternative backends.
-//
-// This is an abstract type that is intended to have the same
-// role as TfLiteDelegate, but without exposing the implementation
-// details of how delegates are implemented.
-// WARNING: This is an experimental type and subject to change.
+/// TfLiteOpaqueDelegateStruct: unconditionally opaque version of
+/// TfLiteDelegate; allows delegation of nodes to alternative backends.
+///
+/// This is an abstract type that is intended to have the same
+/// role as TfLiteDelegate, but without exposing the implementation
+/// details of how delegates are implemented.
+/// WARNING: This is an experimental type and subject to change.
 typedef struct TfLiteOpaqueDelegateStruct TfLiteOpaqueDelegateStruct;
 
-// TfLiteOpaqueDelegate: conditionally opaque version of
-// TfLiteDelegate; allows delegation of nodes to alternative backends.
-// For TF Lite in Play Services, this is an opaque type,
-// but for regular TF Lite, this is just a typedef for TfLiteDelegate.
-// WARNING: This is an experimental type and subject to change.
+/// TfLiteOpaqueDelegate: conditionally opaque version of
+/// TfLiteDelegate; allows delegation of nodes to alternative backends.
+/// For TF Lite in Play Services, this is an opaque type,
+/// but for regular TF Lite, this is just a typedef for TfLiteDelegate.
+/// WARNING: This is an experimental type and subject to change.
 #if TFLITE_WITH_STABLE_ABI || TFLITE_USE_OPAQUE_DELEGATE
 typedef TfLiteOpaqueDelegateStruct TfLiteOpaqueDelegate;
 #else
 typedef TfLiteDelegate TfLiteOpaqueDelegate;
 #endif
 
+/** @} */
+
 #ifdef __cplusplus
 }  // extern C
 #endif