xmos-ai-tools 1.3.2.dev90__py3-none-macosx_10_15_universal2.whl → 1.3.2.dev114__py3-none-macosx_10_15_universal2.whl

xmos_ai_tools/runtime/include/lib_nn/api/nn_layers.h

@@ -300,4 +300,8 @@ void mean_int8(const int8_t *input, int8_t *output, const int start_dim_size,
                const float in_zero_point, const float out_zero_point,
                const float scale_mul);
 
+void mean_int16(const int16_t *input, int16_t *output, const int start_dim_size,
+                const int mean_dim_size, const int end_dim_size,
+                const float scale_mul);
+
 #endif // LAYERS_H_

xmos_ai_tools/runtime/include/lib_tflite_micro/src/tflite-xcore-kernels/xcore_ops.h

@@ -32,6 +32,7 @@ constexpr const char *XC_concat_OpCode = "XC_concat";
 constexpr const char *XC_pad_3_to_4_OpCode = "XC_pad_3_to_4";
 constexpr const char *XC_mul_OpCode = "XC_mul";
 constexpr const char *XC_mean_OpCode = "XC_mean";
+constexpr const char *XC_meani16_OpCode = "XC_meani16";
 constexpr const char *XC_expand_8_to_16_OpCode = "XC_expand_8_to_16";
 // Binarized ops
 constexpr const char *XC_bsign_8_OpCode = "XC_bsign_8";
@@ -59,6 +60,7 @@ TFLMRegistration *Register_XC_concat();
 TFLMRegistration *Register_XC_pad_3_to_4();
 TFLMRegistration *Register_XC_mul();
 TFLMRegistration *Register_XC_mean();
+TFLMRegistration *Register_XC_meani16();
 TFLMRegistration *Register_XC_expand_8_to_16();
 // Binarized ops
 TFLMRegistration *Register_XC_bsign_8();

xmos_ai_tools/xformer/__init__.py

@@ -5,7 +5,8 @@ from typing import Union, List, Optional
 from .flash import generate_flash
 import re
 
-__compilation_output = ""
+__compilation_stdout = ""
+__compilation_stderr = ""
 __arena_size = 0
 
 
@@ -29,20 +30,22 @@ def convert(
 
     args.append(str(filename))
 
-    process_call: subprocess.CompletedProcess = subprocess.run(
-        [arg for arg in args],
-        stdout=subprocess.PIPE,
-        stderr=subprocess.STDOUT,
-        check=True,
-    )
-
-    global __compilation_output, __arena_size
-    __compilation_output = process_call.stdout.decode("utf-8")
-    size_str = re.sub("((.|\n|\r)*)Tensor arena size :", "", __compilation_output)
-    size_str = re.sub("(\n|\r)((.|\n|\r)*)", "", size_str)
-    __arena_size = int(size_str.strip())
-
-    return process_call.returncode
+    try:
+        process_call: subprocess.CompletedProcess = subprocess.run(
+            [arg for arg in args],
+            check=True, text=True, capture_output=True,
+        )
+        global __compilation_stdout, __compilation_stderr, __arena_size
+        __compilation_stdout = process_call.stdout
+        __compilation_stderr = process_call.stderr
+        size_str = re.sub("((.|\n|\r)*)Tensor arena size :", "", __compilation_stdout)
+        size_str = re.sub("(\n|\r)((.|\n|\r)*)", "", size_str)
+        __arena_size = int(size_str.strip())
+        return process_call.returncode
+    except subprocess.CalledProcessError as e:
+        print(e)
+        print("Return code:", e.returncode)
+        print("Error output:", e.stderr)
 
 
 def tensor_arena_size() -> int:
@@ -50,7 +53,8 @@ def tensor_arena_size() -> int:
 
 
 def print_optimization_report():
-    print(__compilation_output)
+    print(__compilation_stderr)
+    print(__compilation_stdout)
 
 
 def print_help(show_hidden: Optional[bool] = False) -> int:

xmos_ai_tools/xinterpreters/host_interpreter.py

@@ -242,7 +242,6 @@ class TFLMHostInterpreter:
         if self.obj:
             lib.delete_interpreter(self.obj)
             self.obj = None
-            print(self.obj)
 
     def tensor_arena_size(self) -> int:
         """! Read the size of the tensor arena required.

{xmos_ai_tools-1.3.2.dev90.dist-info → xmos_ai_tools-1.3.2.dev114.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: xmos_ai_tools
-Version: 1.3.2.dev90
+Version: 1.3.2.dev114
 Summary: XMOS AI Tools
 Home-page: https://github.com/xmos/ai_tools
 Author: XMOS
@@ -30,4 +30,4 @@ Requires-Dist: tflite>=2.4.0
 Documentation
 -------------
 
-Click [here](https://github.com/xmos/ai_tools/blob/ba134f5c7797a561414d85cbaa98648f7dee8b9f/README.md) for documentation on using xmos-ai-tools to deploy AI models on xcore.
+Click [here](https://github.com/xmos/ai_tools/blob/f60ef11c52910b2184ad0b01460bb6d01f4f22a9/README.md) for documentation on using xmos-ai-tools to deploy AI models on xcore.

{xmos_ai_tools-1.3.2.dev90.dist-info → xmos_ai_tools-1.3.2.dev114.dist-info}/RECORD

@@ -49,7 +49,7 @@ xmos_ai_tools/runtime/include/lib_nn/api/nn_bin_types.h,sha256=DTFX2ecfzwl-_tHlq
 xmos_ai_tools/runtime/include/lib_nn/api/nn_config.h,sha256=KFoA5zyOTTHi6x-qDuIWkF8wAz4Gc4BdOBgUzOArOD0,11558
 xmos_ai_tools/runtime/include/lib_nn/api/nn_conv2d_structs.h,sha256=Kfyq3bD1aLcSnWTB_cijU_D5ThGol5utJ1k-WsjqeMY,2268
 xmos_ai_tools/runtime/include/lib_nn/api/nn_image.h,sha256=gz7NyxmZACZiDED0wT7px2qjEEMcUwfa5TjeOUmYzL4,514
-xmos_ai_tools/runtime/include/lib_nn/api/nn_layers.h,sha256=FqXpBLsch_IDH1ssRsDO4zt3N8L8v-PbMddbiwY6DS4,10509
+xmos_ai_tools/runtime/include/lib_nn/api/nn_layers.h,sha256=6vyEP1UTnuVqxCELCaCZx3L1kgMoVX4r20d4nucNZ0E,10696
 xmos_ai_tools/runtime/include/lib_nn/api/nn_op_helper.h,sha256=VMnBVEcgH0wtiiV91JjtFRG4xkCI0m7WrikEmU6ulss,3716
 xmos_ai_tools/runtime/include/lib_nn/api/nn_op_utils.h,sha256=zdCGBeyiHx4-T1hZ4Cjf-QTtXwIyWp4xzuI1dfOjHPA,4948
 xmos_ai_tools/runtime/include/lib_nn/api/nn_operator.h,sha256=Nj4jiXVHlnFjVBtV7mCGulTzSyNwCIpitz8Ao9kSIsU,337
@@ -83,7 +83,7 @@ xmos_ai_tools/runtime/include/lib_tflite_micro/src/tflite-xcore-kernels/conv2d_f
 xmos_ai_tools/runtime/include/lib_tflite_micro/src/tflite-xcore-kernels/xcore_custom_options.h,sha256=lC4Tw1Pxxg3zOXRdqNNtokuU-_cX9TTkYmGLe47-9dQ,630
 xmos_ai_tools/runtime/include/lib_tflite_micro/src/tflite-xcore-kernels/xcore_error_reporter.h,sha256=_NIzvBYMqlwJexYESP5t5JXpxYTt-ZKq-1AdqAB9-Sc,812
 xmos_ai_tools/runtime/include/lib_tflite_micro/src/tflite-xcore-kernels/xcore_interpreter.h,sha256=-0BNn65tzxWgNnHLolCYyUYhboL9nN3ksKni3BjH6QU,1801
-xmos_ai_tools/runtime/include/lib_tflite_micro/src/tflite-xcore-kernels/xcore_ops.h,sha256=YdprKTg4PO0KdlW1RLfb9GYxzRoXmLVxsZb0wRafF6s,2781
+xmos_ai_tools/runtime/include/lib_tflite_micro/src/tflite-xcore-kernels/xcore_ops.h,sha256=Sp1bmgrOwQmF6DzorbD5_Aqv1mwoAzP9CfSTp53a-cI,2878
 xmos_ai_tools/runtime/include/lib_tflite_micro/src/tflite-xcore-kernels/xcore_profiler.h,sha256=Ytqbj4TsbhZrtl42I2dgLyeloLi-1vZwjysIoOkgX9s,1239
 xmos_ai_tools/runtime/include/lib_tflite_micro/src/tflite-xcore-kernels/xcore_utils.h,sha256=CkxEhyN7i2rmlk_ua18XH1XDV_E4mS2u3Ph48mIhN7M,4747
 xmos_ai_tools/runtime/include/lib_xud/lib_xud/api/xud.h,sha256=aQNbN6EvpVQm-OkgE_JGn2SeqYE_sKtwWZebu50IwnE,20701
@@ -378,17 +378,17 @@ xmos_ai_tools/runtime/include/tensorflow/lite/micro/tools/benchmarking/metrics.h
 xmos_ai_tools/runtime/include/tensorflow/lite/micro/tools/benchmarking/op_resolver.h,sha256=g0dl9tzUqngiINvjBlqDclFqvkC85MC4kbU13vE4OkY,6071
 xmos_ai_tools/runtime/include/tensorflow/lite/schema/schema_generated.h,sha256=OzGCvswrW_FArm-HxD70BExn0PVtBbHAFPDO6ZZOFtc,1093177
 xmos_ai_tools/runtime/include/tensorflow/lite/schema/schema_utils.h,sha256=tkHMDPARjIqppYCVInIowwdHxjNP3pfSS9O7vx-ODeo,1333
-xmos_ai_tools/runtime/lib/libhost_xtflitemicro.a,sha256=02QNMG8z6Gc6O1hi6WhPoyxNaU7uX278J95a1fSLqUU,2431256
-xmos_ai_tools/runtime/lib/libxtflitemicro.a,sha256=mU05QFt_TIbBrNiqmLw3_dpMXQvQwFfBAnxeoB70Vn8,70440318
-xmos_ai_tools/xformer/__init__.py,sha256=jA0xba6ZitK9HwWvDJewM8AMU4IZnx_iB1GSBrRUGvU,1627
+xmos_ai_tools/runtime/lib/libhost_xtflitemicro.a,sha256=TENbm9fl5yFjqXLI8kLbzP12GpXt2nRYohMfbKzP4Og,2446328
+xmos_ai_tools/runtime/lib/libxtflitemicro.a,sha256=rN4SPvWgI9MgNuKuqhNosF1159ZNCfFGz033QmfANVU,70898304
+xmos_ai_tools/xformer/__init__.py,sha256=AxzDorvqJ0FiTc4CLecHPwbMkSuWzXGZT2-u72Aau5s,1904
 xmos_ai_tools/xformer/flash.py,sha256=MG4coi_Lvvg-oQmw1pomJD8eeOH4gAMjixjBFvO2BCk,6376
 xmos_ai_tools/xinterpreters/__init__.py,sha256=PFRB9VxOLKaA--j2ZvWGcmesv2C6uNYqJ_kBam68aUI,50
 xmos_ai_tools/xinterpreters/exceptions.py,sha256=HOjADxHYMPI9mN0YIbWxtw9hSeL2B6XWWwqtGtyJdVs,577
-xmos_ai_tools/xinterpreters/host_interpreter.py,sha256=No-g29_twbIaJ4S4fzU5TGpKd4WPVEE0737HtL877sw,25692
-xmos_ai_tools/xinterpreters/libs/macos/xtflm_python.1.0.1.dylib,sha256=uwA1dNlFvRyw-lyg7msQ2P_vi6syL4OGWWZj8pa3fWE,2321592
-xmos_ai_tools/xinterpreters/libs/macos/xtflm_python.dylib,sha256=uwA1dNlFvRyw-lyg7msQ2P_vi6syL4OGWWZj8pa3fWE,2321592
-xmos_ai_tools-1.3.2.dev90.data/data/bin/xcore-opt,sha256=s02tefnJ7Sm7TdbJPz0qB8SK6052cMXq2uL4uwB0BJE,283234872
-xmos_ai_tools-1.3.2.dev90.dist-info/METADATA,sha256=84FC4gHUEBTSb51pvaDCXRvn0z5zY717Q_nj9WrSQ8Q,1341
-xmos_ai_tools-1.3.2.dev90.dist-info/WHEEL,sha256=cblCMhPsi-5aS-1Fj5LRCkSbCUQwa2nrUvkiKN0vRpQ,113
-xmos_ai_tools-1.3.2.dev90.dist-info/top_level.txt,sha256=YWegea73ll3tMlRWRdHJemUy2VOuEYDdOIaffxu_eF0,14
-xmos_ai_tools-1.3.2.dev90.dist-info/RECORD,,
+xmos_ai_tools/xinterpreters/host_interpreter.py,sha256=rqoTcSWK7O9eboMsg0o2UYWapWTE5xiki7XTJZ_JbwQ,25664
+xmos_ai_tools/xinterpreters/libs/macos/xtflm_python.1.0.1.dylib,sha256=D3w9gI6jwmZw6pUwVN-wWNCEmFexquuF2txYuMyghYg,2338744
+xmos_ai_tools/xinterpreters/libs/macos/xtflm_python.dylib,sha256=D3w9gI6jwmZw6pUwVN-wWNCEmFexquuF2txYuMyghYg,2338744
+xmos_ai_tools-1.3.2.dev114.data/data/bin/xcore-opt,sha256=spwatc-p6m2J_AwvrhSGcYnafm9qYJf6LmZkP_T0c9c,283514744
+xmos_ai_tools-1.3.2.dev114.dist-info/METADATA,sha256=vYd-7FP2kIKiTF_R0hwdbiltMGH1gwKI6onXtYMZc04,1342
+xmos_ai_tools-1.3.2.dev114.dist-info/WHEEL,sha256=cblCMhPsi-5aS-1Fj5LRCkSbCUQwa2nrUvkiKN0vRpQ,113
+xmos_ai_tools-1.3.2.dev114.dist-info/top_level.txt,sha256=YWegea73ll3tMlRWRdHJemUy2VOuEYDdOIaffxu_eF0,14
+xmos_ai_tools-1.3.2.dev114.dist-info/RECORD,,