leap-model-parser 0.1.96__tar.gz → 0.1.98__tar.gz

{leap_model_parser-0.1.96 → leap_model_parser-0.1.98}/PKG-INFO

@@ -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 → leap_model_parser-0.1.98}/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-0.1.96 → leap_model_parser-0.1.98}/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 → leap_model_parser-0.1.98}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "leap-model-parser"
-version = "0.1.96"
+version = "0.1.98"
 description = ""
 authors = ["idan <idan.yogev@tensorleap.ai>"]
 license = "MIT"
@@ -18,7 +18,7 @@ tensorflow = {version = "2.11.0", markers = "platform_machine  == 'x86_64'"}
 tensorflow-macos = {version = "2.11.0", markers = "platform_machine  == 'arm64'"}
 numpy = "^1.22.3"
 onnx = "1.10.1"
-onnx2kerastl = "0.0.88"
+onnx2kerastl = "0.0.93"
 keras-data-format-converter = "0.1.14"
 leap-model-rebuilder = "0.1.5"