IQ.Pilot Release Commit @ 7550fa9
This commit is contained in:
@@ -1 +1 @@
|
||||
95bfc521351d3acba7338605686155a9ee8b0009
|
||||
b80bc5202db4ecae448c210ec60702bc63a134af
|
||||
|
||||
@@ -46,6 +46,7 @@ class Domain(enum.Enum):
|
||||
MICROSOFT_NCHWC = "com.microsoft.nchwc"
|
||||
MICROSOFT_EXPERIMENTAL = "com.microsoft.experimental"
|
||||
PYTORCH_ATEN = "org.pytorch.aten"
|
||||
TINYGRAD = "org.tinygrad"
|
||||
@classmethod
|
||||
def from_onnx(cls, domain: str | None) -> "Domain": return cls.ONNX if domain is None or domain == "" else cls(domain)
|
||||
|
||||
@@ -555,6 +556,9 @@ def get_onnx_ops() -> dict[str, types.FunctionType|dict[OpSetId, types.FunctionT
|
||||
cond = _resolve_const(_to_python_const(condition))
|
||||
return tuple(t if cond else e for t,e in zip(then_out.values(), else_out.values()))
|
||||
|
||||
def contiguous_1(x:Tensor): return x.contiguous()
|
||||
Contiguous = {OpSetId(Domain.TINYGRAD, 1):contiguous_1}
|
||||
|
||||
def Identity(x:Tensor): return x
|
||||
def Constant(sparse_value:Tensor|None=None, value:Tensor|None=None, value_float:float|None=None, value_floats:list[float]|None=None,
|
||||
value_int:int|None=None, value_ints:list[int]|None=None, value_string:str|None=None, value_strings:list[str]|None=None):
|
||||
|
||||
Reference in New Issue
Block a user