leap-model-parser 0.1.96__py3-none-any.whl → 0.1.98__py3-none-any.whl

leap_model_parser/contract/nodedata.py

@@ -1,6 +1,6 @@
 from dataclasses import dataclass
-from typing import List
 from typing import Union
+from typing import List
 from enum import Enum
 
 
@@ -203,6 +203,7 @@ class NodeType(Enum):
     SGD = "SGD"
     OnnxAbs = "OnnxAbs"
     OnnxErf = "OnnxErf"
+    OnnxGridSampler = "OnnxGridSampler"
     OnnxHardSigmoid = "OnnxHardSigmoid"
     OnnxLSTM = "OnnxLSTM"
     OnnxReduceMean = "OnnxReduceMean"
@@ -1801,6 +1802,12 @@ class OnnxErf:
     type: NodeType
 
 
+@dataclass
+class OnnxGridSampler:
+    sample_grid: List[int]
+    type: NodeType
+
+
 @dataclass
 class OnnxHardSigmoid:
     alpha: float
@@ -1931,6 +1938,6 @@ class NodeDataTypes:
                  CategoricalCrossentropy, CategoricalHinge, CosineSimilarity, Hinge, Huber, KLDivergence, LogCosh, 
                  MeanAbsoluteError, MeanAbsolutePercentageError, MeanSquaredError, MeanSquaredLogarithmicError, Poisson, 
                  SquaredHinge, Adadelta, Adagrad, Adam, Adamax, Ftrl, Nadam, RMSprop, SGD, OnnxAbs, OnnxErf, 
-                 OnnxHardSigmoid, OnnxLSTM, OnnxReduceMean, OnnxSqrt, Dataset, Input, RepresentationBlock, GroundTruth, 
-                 CustomLayer, CustomLoss, Visualizer, Metric, Lambda, TFOpLambda, SlicingOpLambda, Repeat, Variable, 
-                 Gather, FixedDropout]
+                 OnnxGridSampler, OnnxHardSigmoid, OnnxLSTM, OnnxReduceMean, OnnxSqrt, Dataset, Input, 
+                 RepresentationBlock, GroundTruth, CustomLayer, CustomLoss, Visualizer, Metric, Lambda, TFOpLambda, 
+                 SlicingOpLambda, Repeat, Variable, Gather, FixedDropout]

leap_model_parser/contract/ui_components.json

@@ -14509,6 +14509,53 @@
         "properties": [],
         "options": null
     },
+    {
+        "type": "Layer",
+        "shape_calc": [
+            "IdentityFunc"
+        ],
+        "receptive_fields_func": "IdentityReceptiveFieldsFunc",
+        "menu_sections": [
+            "Layer",
+            "Onnx"
+        ],
+        "inputs_data": {
+            "add_input_by_drag": false,
+            "inputs": [
+                {
+                    "name": "input",
+                    "approval_connection": [
+                        "Input",
+                        "Dataset",
+                        "Layer",
+                        "CustomLayer"
+                    ]
+                }
+            ]
+        },
+        "outputs_data": {
+            "outputs": [
+                {
+                    "name": "feature_map"
+                }
+            ]
+        },
+        "name": "OnnxGridSampler",
+        "parents": [
+            "onnx2kerastl",
+            "customonnxlayer",
+            "onnxgridsampler"
+        ],
+        "properties": [
+            {
+                "name": "sample_grid",
+                "isdefault": false,
+                "default_val": null,
+                "type": "tuple"
+            }
+        ],
+        "options": null
+    },
     {
         "type": "Layer",
         "shape_calc": [

{leap_model_parser-0.1.96.dist-info → leap_model_parser-0.1.98.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: leap-model-parser
-Version: 0.1.96
+Version: 0.1.98
 Summary: 
 Home-page: https://github.com/tensorleap/leap-model-parser
 License: MIT
@@ -16,7 +16,7 @@ Requires-Dist: keras-data-format-converter (==0.1.14)
 Requires-Dist: leap-model-rebuilder (==0.1.5)
 Requires-Dist: numpy (>=1.22.3,<2.0.0)
 Requires-Dist: onnx (==1.10.1)
-Requires-Dist: onnx2kerastl (==0.0.88)
+Requires-Dist: onnx2kerastl (==0.0.93)
 Requires-Dist: tensorflow (==2.11.0) ; platform_machine == "x86_64"
 Requires-Dist: tensorflow-macos (==2.11.0) ; platform_machine == "arm64"
 Project-URL: Repository, https://github.com/tensorleap/leap-model-parser

{leap_model_parser-0.1.96.dist-info → leap_model_parser-0.1.98.dist-info}/RECORD

@@ -2,8 +2,8 @@ LICENSE,sha256=qIwWjdspQeSMTtnFZBC8MuT-95L02FPvzRUdWFxrwJY,1067
 leap_model_parser/__init__.py,sha256=OAU7rFHAVVWUM-cDtQ4Ohum567KN8M-YTkHZp5KiYbo,132
 leap_model_parser/contract/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 leap_model_parser/contract/importmodelresponse.py,sha256=YNtTTJX40GmkzP6gYDw7_JcnNgwqtrTHkaV9vTFSYs4,691
-leap_model_parser/contract/nodedata.py,sha256=1_ML0nzp3QUZ0_9mGSLhfO4_hqjYMwi0DWLwymUnWEs,43326
-leap_model_parser/contract/ui_components.json,sha256=lhKWOQjvGCxQakDcNHq7aBmORUf67ZodZcCtJ_TIcsI,416303
+leap_model_parser/contract/nodedata.py,sha256=zDeea3uIypxzU3MTo10OfEmog_3N1PeqmljVHYcP1wI,43465
+leap_model_parser/contract/ui_components.json,sha256=sOozS6vyG0GF3NS8msZPG2byMqMIhi4tw2Vydox_6wQ,417479
 leap_model_parser/keras_json_model_import.py,sha256=D1a7V7lV9yiFPqhhrjHKmwNIRYWO-QuFxlRwTdvjsOE,15485
 leap_model_parser/model_parser.py,sha256=pvDGFyez_RV1S4khlfIWCmxFi3hfMNzH_FkUzdMCDiw,6766
 leap_model_parser/utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -15,8 +15,8 @@ leap_model_parser/utils/uicomponents/__init__.py,sha256=ShNUN5NLzaxOQ5ByNUUWJG6v
 leap_model_parser/utils/uicomponents/generatenodedata.py,sha256=LRaPlO5jJ9pUtkvLG9QaGBRnu860wNH_OkBM6UKLGGQ,3035
 leap_model_parser/utils/uicomponents/tensorflowinscpection.py,sha256=ym613z9iQKPDBpr0RYD35bTABdm1L-Ez86G47BYT7qw,6775
 leap_model_parser/utils/uicomponents/ui_components_config.yaml,sha256=nzhcSR-00erM-D09CczOmfF4a6w9kZUgDjDksHtLsUo,19604
-leap_model_parser-0.1.96.dist-info/LICENSE,sha256=qIwWjdspQeSMTtnFZBC8MuT-95L02FPvzRUdWFxrwJY,1067
-leap_model_parser-0.1.96.dist-info/METADATA,sha256=qHpLTTqzW9c_scz6_ysdJ6rCXDRIyzndj6Fyt9G7z_0,1008
-leap_model_parser-0.1.96.dist-info/WHEEL,sha256=7Z8_27uaHI_UZAc4Uox4PpBhQ9Y5_modZXWMxtUi4NU,88
-leap_model_parser-0.1.96.dist-info/entry_points.txt,sha256=ZvV6EuQt1uAqwapNg5Lo2qjJM9ZG5g2wfzZoLh_Ztyk,77
-leap_model_parser-0.1.96.dist-info/RECORD,,
+leap_model_parser-0.1.98.dist-info/LICENSE,sha256=qIwWjdspQeSMTtnFZBC8MuT-95L02FPvzRUdWFxrwJY,1067
+leap_model_parser-0.1.98.dist-info/METADATA,sha256=NOf6UayoXFekinO9ZI5ysF12Cknr-CkczqEyE36MItw,1008
+leap_model_parser-0.1.98.dist-info/WHEEL,sha256=7Z8_27uaHI_UZAc4Uox4PpBhQ9Y5_modZXWMxtUi4NU,88
+leap_model_parser-0.1.98.dist-info/entry_points.txt,sha256=ZvV6EuQt1uAqwapNg5Lo2qjJM9ZG5g2wfzZoLh_Ztyk,77
+leap_model_parser-0.1.98.dist-info/RECORD,,