diff --git a/contrib/codegen-tools/libnd4j-gen/op-ir.proto b/contrib/codegen-tools/libnd4j-gen/op-ir.proto
index b7ee920f0..9cbb9c962 100644
--- a/contrib/codegen-tools/libnd4j-gen/op-ir.proto
+++ b/contrib/codegen-tools/libnd4j-gen/op-ir.proto
@@ -12,23 +12,14 @@ opList {
}
opList {
name: "BinaryMinimalRelativeError"
- argDescriptor {
- name: "outputs"
- argType: OUTPUT_TENSOR
- }
- argDescriptor {
- name: "input"
- argType: INPUT_TENSOR
- }
- argDescriptor {
- name: "y"
- argType: INPUT_TENSOR
- argIndex: 1
- }
argDescriptor {
name: "dataType"
argType: DATA_TYPE
}
+ argDescriptor {
+ name: "outputs"
+ argType: OUTPUT_TENSOR
+ }
argDescriptor {
name: "thresholdRelative"
argType: DOUBLE
@@ -38,18 +29,6 @@ opList {
argType: DOUBLE
argIndex: 1
}
- opDeclarationType: LEGACY_XYZ
-}
-opList {
- name: "BinaryRelativeError"
- argDescriptor {
- name: "outputs"
- argType: OUTPUT_TENSOR
- }
- argDescriptor {
- name: "inPlace"
- argType: BOOL
- }
argDescriptor {
name: "input"
argType: INPUT_TENSOR
@@ -59,14 +38,35 @@ opList {
argType: INPUT_TENSOR
argIndex: 1
}
+ opDeclarationType: LEGACY_XYZ
+}
+opList {
+ name: "BinaryRelativeError"
argDescriptor {
name: "dataType"
argType: DATA_TYPE
}
+ argDescriptor {
+ name: "outputs"
+ argType: OUTPUT_TENSOR
+ }
+ argDescriptor {
+ name: "inPlace"
+ argType: BOOL
+ }
argDescriptor {
name: "threshold"
argType: DOUBLE
}
+ argDescriptor {
+ name: "input"
+ argType: INPUT_TENSOR
+ }
+ argDescriptor {
+ name: "y"
+ argType: INPUT_TENSOR
+ argIndex: 1
+ }
opDeclarationType: LEGACY_XYZ
}
opList {
@@ -79,10 +79,6 @@ opList {
name: "inPlace"
argType: BOOL
}
- argDescriptor {
- name: "input"
- argType: INPUT_TENSOR
- }
argDescriptor {
name: "clipValueMin"
argType: DOUBLE
@@ -92,6 +88,10 @@ opList {
argType: DOUBLE
argIndex: 1
}
+ argDescriptor {
+ name: "input"
+ argType: INPUT_TENSOR
+ }
}
opList {
name: "Conditional"
@@ -158,6 +158,10 @@ opList {
}
opList {
name: "Pow_bp"
+ argDescriptor {
+ name: "dtype"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "dLdx"
argType: OUTPUT_TENSOR
@@ -195,13 +199,13 @@ opList {
argType: INPUT_TENSOR
argIndex: 4
}
- argDescriptor {
- name: "dtype"
- argType: DATA_TYPE
- }
}
opList {
name: "Reciprocal"
+ argDescriptor {
+ name: "dataType"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -210,14 +214,14 @@ opList {
name: "input"
argType: INPUT_TENSOR
}
- argDescriptor {
- name: "dataType"
- argType: DATA_TYPE
- }
opDeclarationType: LEGACY_XYZ
}
opList {
name: "RelativeError"
+ argDescriptor {
+ name: "dataType"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -235,10 +239,6 @@ opList {
argType: INPUT_TENSOR
argIndex: 1
}
- argDescriptor {
- name: "dataType"
- argType: DATA_TYPE
- }
opDeclarationType: LEGACY_XYZ
}
opList {
@@ -276,6 +276,10 @@ opList {
}
opList {
name: "Where"
+ argDescriptor {
+ name: "dtype"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -294,10 +298,6 @@ opList {
argType: INPUT_TENSOR
argIndex: 2
}
- argDescriptor {
- name: "dtype"
- argType: DATA_TYPE
- }
}
opList {
name: "While"
@@ -305,6 +305,10 @@ opList {
name: "outputs"
argType: OUTPUT_TENSOR
}
+ argDescriptor {
+ name: "frameName"
+ argType: STRING
+ }
argDescriptor {
name: "isConstant"
argType: BOOL
@@ -313,14 +317,14 @@ opList {
name: "condition"
argType: INPUT_TENSOR
}
- argDescriptor {
- name: "frameName"
- argType: STRING
- }
opDeclarationType: LOGIC_OP_IMPL
}
opList {
name: "_geluderivative"
+ argDescriptor {
+ name: "dataType"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -333,14 +337,14 @@ opList {
name: "input"
argType: INPUT_TENSOR
}
- argDescriptor {
- name: "dataType"
- argType: DATA_TYPE
- }
opDeclarationType: LEGACY_XYZ
}
opList {
name: "_mishderivative"
+ argDescriptor {
+ name: "dataType"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -358,10 +362,6 @@ opList {
argType: INPUT_TENSOR
argIndex: 1
}
- argDescriptor {
- name: "dataType"
- argType: DATA_TYPE
- }
opDeclarationType: LEGACY_XYZ
}
opList {
@@ -374,18 +374,22 @@ opList {
name: "inPlace"
argType: BOOL
}
- argDescriptor {
- name: "input"
- argType: INPUT_TENSOR
- }
argDescriptor {
name: "pow"
argType: DOUBLE
}
+ argDescriptor {
+ name: "input"
+ argType: INPUT_TENSOR
+ }
opDeclarationType: LEGACY_XYZ
}
opList {
name: "_precise_geluderivative"
+ argDescriptor {
+ name: "dataType"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -403,14 +407,14 @@ opList {
name: "input"
argType: INPUT_TENSOR
}
- argDescriptor {
- name: "dataType"
- argType: DATA_TYPE
- }
opDeclarationType: LEGACY_XYZ
}
opList {
name: "_sigmoidderivative"
+ argDescriptor {
+ name: "dataType"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -428,14 +432,14 @@ opList {
argType: INPUT_TENSOR
argIndex: 1
}
- argDescriptor {
- name: "dataType"
- argType: DATA_TYPE
- }
opDeclarationType: LEGACY_XYZ
}
opList {
name: "_swishderivative"
+ argDescriptor {
+ name: "dataType"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -453,14 +457,14 @@ opList {
argType: INPUT_TENSOR
argIndex: 1
}
- argDescriptor {
- name: "dataType"
- argType: DATA_TYPE
- }
opDeclarationType: LEGACY_XYZ
}
opList {
name: "_tanhderivative"
+ argDescriptor {
+ name: "dataType"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -473,14 +477,14 @@ opList {
name: "input"
argType: INPUT_TENSOR
}
- argDescriptor {
- name: "dataType"
- argType: DATA_TYPE
- }
opDeclarationType: LEGACY_XYZ
}
opList {
name: "abs"
+ argDescriptor {
+ name: "dataType"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -493,14 +497,18 @@ opList {
name: "input"
argType: INPUT_TENSOR
}
- argDescriptor {
- name: "dataType"
- argType: DATA_TYPE
- }
opDeclarationType: LEGACY_XYZ
}
opList {
name: "absolute_difference_loss"
+ argDescriptor {
+ name: "reductionMode"
+ argType: INT64
+ }
+ argDescriptor {
+ name: "dtype"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -519,6 +527,9 @@ opList {
argType: INPUT_TENSOR
argIndex: 2
}
+}
+opList {
+ name: "absolute_difference_loss_grad"
argDescriptor {
name: "reductionMode"
argType: INT64
@@ -527,9 +538,6 @@ opList {
name: "dtype"
argType: DATA_TYPE
}
-}
-opList {
- name: "absolute_difference_loss_grad"
argDescriptor {
name: "dLdp"
argType: OUTPUT_TENSOR
@@ -558,17 +566,13 @@ opList {
argType: INPUT_TENSOR
argIndex: 2
}
- argDescriptor {
- name: "reductionMode"
- argType: INT64
- }
- argDescriptor {
- name: "dtype"
- argType: DATA_TYPE
- }
}
opList {
name: "acos"
+ argDescriptor {
+ name: "dataType"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -581,14 +585,14 @@ opList {
name: "input"
argType: INPUT_TENSOR
}
- argDescriptor {
- name: "dataType"
- argType: DATA_TYPE
- }
opDeclarationType: LEGACY_XYZ
}
opList {
name: "acosh"
+ argDescriptor {
+ name: "dataType"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -601,10 +605,6 @@ opList {
name: "input"
argType: INPUT_TENSOR
}
- argDescriptor {
- name: "dataType"
- argType: DATA_TYPE
- }
opDeclarationType: LEGACY_XYZ
}
opList {
@@ -623,6 +623,15 @@ opList {
argType: OUTPUT_TENSOR
argIndex: 2
}
+ argDescriptor {
+ name: "dRho"
+ argType: DOUBLE
+ }
+ argDescriptor {
+ name: "dEpsilon"
+ argType: DOUBLE
+ argIndex: 1
+ }
argDescriptor {
name: "gradient"
argType: INPUT_TENSOR
@@ -652,15 +661,6 @@ opList {
argType: INPUT_TENSOR
argIndex: 5
}
- argDescriptor {
- name: "dRho"
- argType: DOUBLE
- }
- argDescriptor {
- name: "dEpsilon"
- argType: DOUBLE
- argIndex: 1
- }
opDeclarationType: CONFIGURABLE_OP_IMPL
}
opList {
@@ -674,6 +674,15 @@ opList {
argType: OUTPUT_TENSOR
argIndex: 1
}
+ argDescriptor {
+ name: "dLr"
+ argType: DOUBLE
+ }
+ argDescriptor {
+ name: "dEpsilon"
+ argType: DOUBLE
+ argIndex: 1
+ }
argDescriptor {
name: "gradient"
argType: INPUT_TENSOR
@@ -693,19 +702,14 @@ opList {
argType: INPUT_TENSOR
argIndex: 3
}
- argDescriptor {
- name: "dLr"
- argType: DOUBLE
- }
- argDescriptor {
- name: "dEpsilon"
- argType: DOUBLE
- argIndex: 1
- }
opDeclarationType: CONFIGURABLE_OP_IMPL
}
opList {
name: "ada_max_updater"
+ argDescriptor {
+ name: "iteration"
+ argType: INT64
+ }
argDescriptor {
name: "update"
argType: OUTPUT_TENSOR
@@ -720,6 +724,25 @@ opList {
argType: OUTPUT_TENSOR
argIndex: 2
}
+ argDescriptor {
+ name: "dLr"
+ argType: DOUBLE
+ }
+ argDescriptor {
+ name: "dBeta1"
+ argType: DOUBLE
+ argIndex: 1
+ }
+ argDescriptor {
+ name: "dBeta2"
+ argType: DOUBLE
+ argIndex: 2
+ }
+ argDescriptor {
+ name: "dEpsilon"
+ argType: DOUBLE
+ argIndex: 3
+ }
argDescriptor {
name: "gradient"
argType: INPUT_TENSOR
@@ -754,10 +777,28 @@ opList {
argType: INPUT_TENSOR
argIndex: 6
}
+ opDeclarationType: CONFIGURABLE_OP_IMPL
+}
+opList {
+ name: "adabelief_updater"
argDescriptor {
name: "iteration"
argType: INT64
}
+ argDescriptor {
+ name: "update"
+ argType: OUTPUT_TENSOR
+ }
+ argDescriptor {
+ name: "stateU"
+ argType: OUTPUT_TENSOR
+ argIndex: 1
+ }
+ argDescriptor {
+ name: "stateM"
+ argType: OUTPUT_TENSOR
+ argIndex: 2
+ }
argDescriptor {
name: "dLr"
argType: DOUBLE
@@ -777,10 +818,48 @@ opList {
argType: DOUBLE
argIndex: 3
}
+ argDescriptor {
+ name: "gradient"
+ argType: INPUT_TENSOR
+ }
+ argDescriptor {
+ name: "initStateU"
+ argType: INPUT_TENSOR
+ argIndex: 1
+ }
+ argDescriptor {
+ name: "initStateM"
+ argType: INPUT_TENSOR
+ argIndex: 2
+ }
+ argDescriptor {
+ name: "lr"
+ argType: INPUT_TENSOR
+ argIndex: 3
+ }
+ argDescriptor {
+ name: "beta1"
+ argType: INPUT_TENSOR
+ argIndex: 4
+ }
+ argDescriptor {
+ name: "beta2"
+ argType: INPUT_TENSOR
+ argIndex: 5
+ }
+ argDescriptor {
+ name: "epsilon"
+ argType: INPUT_TENSOR
+ argIndex: 6
+ }
opDeclarationType: CONFIGURABLE_OP_IMPL
}
opList {
name: "adam_updater"
+ argDescriptor {
+ name: "iteration"
+ argType: INT64
+ }
argDescriptor {
name: "update"
argType: OUTPUT_TENSOR
@@ -795,6 +874,25 @@ opList {
argType: OUTPUT_TENSOR
argIndex: 2
}
+ argDescriptor {
+ name: "dLr"
+ argType: DOUBLE
+ }
+ argDescriptor {
+ name: "dBeta1"
+ argType: DOUBLE
+ argIndex: 1
+ }
+ argDescriptor {
+ name: "dBeta2"
+ argType: DOUBLE
+ argIndex: 2
+ }
+ argDescriptor {
+ name: "dEpsilon"
+ argType: DOUBLE
+ argIndex: 3
+ }
argDescriptor {
name: "gradient"
argType: INPUT_TENSOR
@@ -829,33 +927,14 @@ opList {
argType: INPUT_TENSOR
argIndex: 6
}
- argDescriptor {
- name: "iteration"
- argType: INT64
- }
- argDescriptor {
- name: "dLr"
- argType: DOUBLE
- }
- argDescriptor {
- name: "dBeta1"
- argType: DOUBLE
- argIndex: 1
- }
- argDescriptor {
- name: "dBeta2"
- argType: DOUBLE
- argIndex: 2
- }
- argDescriptor {
- name: "dEpsilon"
- argType: DOUBLE
- argIndex: 3
- }
opDeclarationType: CONFIGURABLE_OP_IMPL
}
opList {
name: "add"
+ argDescriptor {
+ name: "dataType"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -873,14 +952,14 @@ opList {
argType: INPUT_TENSOR
argIndex: 1
}
- argDescriptor {
- name: "dataType"
- argType: DATA_TYPE
- }
opDeclarationType: BROADCASTABLE_OP_IMPL
}
opList {
name: "add_bp"
+ argDescriptor {
+ name: "dtype"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "gradX"
argType: OUTPUT_TENSOR
@@ -908,10 +987,6 @@ opList {
argType: INPUT_TENSOR
argIndex: 2
}
- argDescriptor {
- name: "dtype"
- argType: DATA_TYPE
- }
}
opList {
name: "add_scalar"
@@ -952,6 +1027,10 @@ opList {
name: "outputs"
argType: OUTPUT_TENSOR
}
+ argDescriptor {
+ name: "factor"
+ argType: DOUBLE
+ }
argDescriptor {
name: "input"
argType: INPUT_TENSOR
@@ -961,14 +1040,14 @@ opList {
argType: INPUT_TENSOR
argIndex: 1
}
- argDescriptor {
- name: "factor"
- argType: DOUBLE
- }
opDeclarationType: CONFIGURABLE_OP_IMPL
}
opList {
name: "adjust_hue"
+ argDescriptor {
+ name: "dimC"
+ argType: INT64
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -982,22 +1061,22 @@ opList {
argType: INPUT_TENSOR
argIndex: 1
}
- argDescriptor {
- name: "dimC"
- argType: INT64
- }
- argDescriptor {
- name: "factor"
- argType: DOUBLE
- }
opDeclarationType: CONFIGURABLE_OP_IMPL
}
opList {
name: "adjust_saturation"
+ argDescriptor {
+ name: "dimC"
+ argType: INT64
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
}
+ argDescriptor {
+ name: "factor"
+ argType: DOUBLE
+ }
argDescriptor {
name: "input"
argType: INPUT_TENSOR
@@ -1007,18 +1086,14 @@ opList {
argType: INPUT_TENSOR
argIndex: 1
}
- argDescriptor {
- name: "dimC"
- argType: INT64
- }
- argDescriptor {
- name: "factor"
- argType: DOUBLE
- }
opDeclarationType: CONFIGURABLE_OP_IMPL
}
opList {
name: "all"
+ argDescriptor {
+ name: "dimensions"
+ argType: INT64
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -1036,10 +1111,6 @@ opList {
argType: INPUT_TENSOR
argIndex: 1
}
- argDescriptor {
- name: "dimensions"
- argType: INT64
- }
opDeclarationType: LEGACY_XYZ
}
opList {
@@ -1048,10 +1119,6 @@ opList {
name: "outputs"
argType: OUTPUT_TENSOR
}
- argDescriptor {
- name: "input"
- argType: INPUT_TENSOR
- }
argDescriptor {
name: "a"
argType: DOUBLE
@@ -1071,32 +1138,22 @@ opList {
argType: DOUBLE
argIndex: 3
}
+ argDescriptor {
+ name: "input"
+ argType: INPUT_TENSOR
+ }
opDeclarationType: LEGACY_XYZ
}
opList {
name: "alpha_dropout_bp"
- argDescriptor {
- name: "outputs"
- argType: OUTPUT_TENSOR
- }
- argDescriptor {
- name: "input"
- argType: INPUT_TENSOR
- }
- argDescriptor {
- name: "gradOut"
- argType: INPUT_TENSOR
- argIndex: 1
- }
- argDescriptor {
- name: "reduceShape"
- argType: INPUT_TENSOR
- argIndex: 2
- }
argDescriptor {
name: "seed"
argType: INT64
}
+ argDescriptor {
+ name: "outputs"
+ argType: OUTPUT_TENSOR
+ }
argDescriptor {
name: "probValue"
argType: DOUBLE
@@ -1116,10 +1173,28 @@ opList {
argType: DOUBLE
argIndex: 3
}
+ argDescriptor {
+ name: "input"
+ argType: INPUT_TENSOR
+ }
+ argDescriptor {
+ name: "gradOut"
+ argType: INPUT_TENSOR
+ argIndex: 1
+ }
+ argDescriptor {
+ name: "reduceShape"
+ argType: INPUT_TENSOR
+ argIndex: 2
+ }
opDeclarationType: CONFIGURABLE_OP_IMPL
}
opList {
name: "amax"
+ argDescriptor {
+ name: "dimensions"
+ argType: INT64
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -1137,14 +1212,14 @@ opList {
argType: INPUT_TENSOR
argIndex: 1
}
- argDescriptor {
- name: "dimensions"
- argType: INT64
- }
opDeclarationType: LEGACY_XYZ
}
opList {
name: "amax_pairwise"
+ argDescriptor {
+ name: "dataType"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -1158,14 +1233,14 @@ opList {
argType: INPUT_TENSOR
argIndex: 1
}
- argDescriptor {
- name: "dataType"
- argType: DATA_TYPE
- }
opDeclarationType: LEGACY_XYZ
}
opList {
name: "amean"
+ argDescriptor {
+ name: "dimensions"
+ argType: INT64
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -1183,14 +1258,14 @@ opList {
argType: INPUT_TENSOR
argIndex: 1
}
- argDescriptor {
- name: "dimensions"
- argType: INT64
- }
opDeclarationType: LEGACY_XYZ
}
opList {
name: "amin"
+ argDescriptor {
+ name: "dimensions"
+ argType: INT64
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -1208,14 +1283,14 @@ opList {
argType: INPUT_TENSOR
argIndex: 1
}
- argDescriptor {
- name: "dimensions"
- argType: INT64
- }
opDeclarationType: LEGACY_XYZ
}
opList {
name: "amin_pairwise"
+ argDescriptor {
+ name: "dataType"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -1229,14 +1304,14 @@ opList {
argType: INPUT_TENSOR
argIndex: 1
}
- argDescriptor {
- name: "dataType"
- argType: DATA_TYPE
- }
opDeclarationType: LEGACY_XYZ
}
opList {
name: "ams_grad_updater"
+ argDescriptor {
+ name: "iteration"
+ argType: INT64
+ }
argDescriptor {
name: "update"
argType: OUTPUT_TENSOR
@@ -1256,6 +1331,25 @@ opList {
argType: OUTPUT_TENSOR
argIndex: 3
}
+ argDescriptor {
+ name: "dLr"
+ argType: DOUBLE
+ }
+ argDescriptor {
+ name: "dBeta1"
+ argType: DOUBLE
+ argIndex: 1
+ }
+ argDescriptor {
+ name: "dBeta2"
+ argType: DOUBLE
+ argIndex: 2
+ }
+ argDescriptor {
+ name: "dEpsilon"
+ argType: DOUBLE
+ argIndex: 3
+ }
argDescriptor {
name: "gradient"
argType: INPUT_TENSOR
@@ -1295,52 +1389,30 @@ opList {
argType: INPUT_TENSOR
argIndex: 7
}
- argDescriptor {
- name: "iteration"
- argType: INT64
- }
- argDescriptor {
- name: "dLr"
- argType: DOUBLE
- }
- argDescriptor {
- name: "dBeta1"
- argType: DOUBLE
- argIndex: 1
- }
- argDescriptor {
- name: "dBeta2"
- argType: DOUBLE
- argIndex: 2
- }
- argDescriptor {
- name: "dEpsilon"
- argType: DOUBLE
- argIndex: 3
- }
opDeclarationType: CONFIGURABLE_OP_IMPL
}
opList {
name: "and"
+ argDescriptor {
+ name: "dataType"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
}
argDescriptor {
- name: "inPlace"
- argType: BOOL
+ name: "comparable"
+ argType: DOUBLE
}
argDescriptor {
name: "input"
argType: INPUT_TENSOR
}
argDescriptor {
- name: "dataType"
- argType: DATA_TYPE
- }
- argDescriptor {
- name: "comparable"
- argType: DOUBLE
+ name: "y"
+ argType: INPUT_TENSOR
+ argIndex: 1
}
opDeclarationType: LEGACY_XYZ
}
@@ -1362,6 +1434,10 @@ opList {
}
opList {
name: "any"
+ argDescriptor {
+ name: "dimensions"
+ argType: INT64
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -1379,10 +1455,6 @@ opList {
argType: INPUT_TENSOR
argIndex: 1
}
- argDescriptor {
- name: "dimensions"
- argType: INT64
- }
opDeclarationType: LEGACY_XYZ
}
opList {
@@ -1391,6 +1463,10 @@ opList {
name: "Z"
argType: OUTPUT_TENSOR
}
+ argDescriptor {
+ name: "lr"
+ argType: DOUBLE
+ }
argDescriptor {
name: "parameters"
argType: INPUT_TENSOR
@@ -1405,10 +1481,6 @@ opList {
argType: INPUT_TENSOR
argIndex: 2
}
- argDescriptor {
- name: "lr"
- argType: DOUBLE
- }
opDeclarationType: CONFIGURABLE_OP_IMPL
}
opList {
@@ -1420,6 +1492,14 @@ opList {
}
opList {
name: "argamax"
+ argDescriptor {
+ name: "dimensions"
+ argType: INT64
+ }
+ argDescriptor {
+ name: "dtype"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -1437,17 +1517,17 @@ opList {
argType: INPUT_TENSOR
argIndex: 1
}
- argDescriptor {
- name: "dimensions"
- argType: INT64
- }
- argDescriptor {
- name: "dtype"
- argType: DATA_TYPE
- }
}
opList {
name: "argamin"
+ argDescriptor {
+ name: "dimensions"
+ argType: INT64
+ }
+ argDescriptor {
+ name: "dtype"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -1465,17 +1545,17 @@ opList {
argType: INPUT_TENSOR
argIndex: 1
}
- argDescriptor {
- name: "dimensions"
- argType: INT64
- }
- argDescriptor {
- name: "dtype"
- argType: DATA_TYPE
- }
}
opList {
name: "argmax"
+ argDescriptor {
+ name: "dimensions"
+ argType: INT64
+ }
+ argDescriptor {
+ name: "dtype"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -1493,17 +1573,17 @@ opList {
argType: INPUT_TENSOR
argIndex: 1
}
- argDescriptor {
- name: "dimensions"
- argType: INT64
- }
- argDescriptor {
- name: "dtype"
- argType: DATA_TYPE
- }
}
opList {
name: "argmin"
+ argDescriptor {
+ name: "dimensions"
+ argType: INT64
+ }
+ argDescriptor {
+ name: "dtype"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -1521,17 +1601,13 @@ opList {
argType: INPUT_TENSOR
argIndex: 1
}
- argDescriptor {
- name: "dimensions"
- argType: INT64
- }
- argDescriptor {
- name: "dtype"
- argType: DATA_TYPE
- }
}
opList {
name: "asin"
+ argDescriptor {
+ name: "dataType"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -1544,14 +1620,14 @@ opList {
name: "input"
argType: INPUT_TENSOR
}
- argDescriptor {
- name: "dataType"
- argType: DATA_TYPE
- }
opDeclarationType: LEGACY_XYZ
}
opList {
name: "asinh"
+ argDescriptor {
+ name: "dataType"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -1564,10 +1640,6 @@ opList {
name: "input"
argType: INPUT_TENSOR
}
- argDescriptor {
- name: "dataType"
- argType: DATA_TYPE
- }
opDeclarationType: LEGACY_XYZ
}
opList {
@@ -1589,6 +1661,10 @@ opList {
}
opList {
name: "assign_bp"
+ argDescriptor {
+ name: "dtype"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "gradX"
argType: OUTPUT_TENSOR
@@ -1612,13 +1688,13 @@ opList {
argType: INPUT_TENSOR
argIndex: 2
}
- argDescriptor {
- name: "dtype"
- argType: DATA_TYPE
- }
}
opList {
name: "asum"
+ argDescriptor {
+ name: "dimensions"
+ argType: INT64
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -1636,14 +1712,14 @@ opList {
argType: INPUT_TENSOR
argIndex: 1
}
- argDescriptor {
- name: "dimensions"
- argType: INT64
- }
opDeclarationType: LEGACY_XYZ
}
opList {
name: "atan"
+ argDescriptor {
+ name: "dataType"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -1656,14 +1732,14 @@ opList {
name: "input"
argType: INPUT_TENSOR
}
- argDescriptor {
- name: "dataType"
- argType: DATA_TYPE
- }
opDeclarationType: LEGACY_XYZ
}
opList {
name: "atanh"
+ argDescriptor {
+ name: "dataType"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -1676,22 +1752,10 @@ opList {
name: "input"
argType: INPUT_TENSOR
}
- argDescriptor {
- name: "dataType"
- argType: DATA_TYPE
- }
opDeclarationType: LEGACY_XYZ
}
opList {
name: "avgpool2d"
- argDescriptor {
- name: "outputs"
- argType: OUTPUT_TENSOR
- }
- argDescriptor {
- name: "input"
- argType: INPUT_TENSOR
- }
argDescriptor {
name: "kH"
argType: INT64
@@ -1750,22 +1814,17 @@ opList {
name: "dtype"
argType: DATA_TYPE
}
+ argDescriptor {
+ name: "outputs"
+ argType: OUTPUT_TENSOR
+ }
+ argDescriptor {
+ name: "input"
+ argType: INPUT_TENSOR
+ }
}
opList {
name: "avgpool2d_bp"
- argDescriptor {
- name: "gradI"
- argType: OUTPUT_TENSOR
- }
- argDescriptor {
- name: "input"
- argType: INPUT_TENSOR
- }
- argDescriptor {
- name: "gradO"
- argType: INPUT_TENSOR
- argIndex: 1
- }
argDescriptor {
name: "kH"
argType: INT64
@@ -1824,98 +1883,6 @@ opList {
name: "dtype"
argType: DATA_TYPE
}
-}
-opList {
- name: "avgpool3dnew"
- argDescriptor {
- name: "outputs"
- argType: OUTPUT_TENSOR
- }
- argDescriptor {
- name: "input"
- argType: INPUT_TENSOR
- }
- argDescriptor {
- name: "kD"
- argType: INT64
- }
- argDescriptor {
- name: "kH"
- argType: INT64
- argIndex: 1
- }
- argDescriptor {
- name: "kW"
- argType: INT64
- argIndex: 2
- }
- argDescriptor {
- name: "sD"
- argType: INT64
- argIndex: 3
- }
- argDescriptor {
- name: "sH"
- argType: INT64
- argIndex: 4
- }
- argDescriptor {
- name: "sW"
- argType: INT64
- argIndex: 5
- }
- argDescriptor {
- name: "pD"
- argType: INT64
- argIndex: 6
- }
- argDescriptor {
- name: "pH"
- argType: INT64
- argIndex: 7
- }
- argDescriptor {
- name: "pW"
- argType: INT64
- argIndex: 8
- }
- argDescriptor {
- name: "dD"
- argType: INT64
- argIndex: 9
- }
- argDescriptor {
- name: "dH"
- argType: INT64
- argIndex: 10
- }
- argDescriptor {
- name: "dW"
- argType: INT64
- argIndex: 11
- }
- argDescriptor {
- name: "isSameMode"
- argType: INT64
- argIndex: 12
- }
- argDescriptor {
- name: "extraParam0"
- argType: INT64
- argIndex: 13
- }
- argDescriptor {
- name: "isNCDHW"
- argType: INT64
- argIndex: 14
- }
- argDescriptor {
- name: "dtype"
- argType: DATA_TYPE
- }
-}
-opList {
- name: "avgpool3dnew_bp"
argDescriptor {
name: "gradI"
argType: OUTPUT_TENSOR
@@ -1929,6 +1896,9 @@ opList {
argType: INPUT_TENSOR
argIndex: 1
}
+}
+opList {
+ name: "avgpool3dnew"
argDescriptor {
name: "kD"
argType: INT64
@@ -2007,13 +1977,127 @@ opList {
name: "dtype"
argType: DATA_TYPE
}
+ argDescriptor {
+ name: "outputs"
+ argType: OUTPUT_TENSOR
+ }
+ argDescriptor {
+ name: "input"
+ argType: INPUT_TENSOR
+ }
+}
+opList {
+ name: "avgpool3dnew_bp"
+ argDescriptor {
+ name: "kD"
+ argType: INT64
+ }
+ argDescriptor {
+ name: "kH"
+ argType: INT64
+ argIndex: 1
+ }
+ argDescriptor {
+ name: "kW"
+ argType: INT64
+ argIndex: 2
+ }
+ argDescriptor {
+ name: "sD"
+ argType: INT64
+ argIndex: 3
+ }
+ argDescriptor {
+ name: "sH"
+ argType: INT64
+ argIndex: 4
+ }
+ argDescriptor {
+ name: "sW"
+ argType: INT64
+ argIndex: 5
+ }
+ argDescriptor {
+ name: "pD"
+ argType: INT64
+ argIndex: 6
+ }
+ argDescriptor {
+ name: "pH"
+ argType: INT64
+ argIndex: 7
+ }
+ argDescriptor {
+ name: "pW"
+ argType: INT64
+ argIndex: 8
+ }
+ argDescriptor {
+ name: "dD"
+ argType: INT64
+ argIndex: 9
+ }
+ argDescriptor {
+ name: "dH"
+ argType: INT64
+ argIndex: 10
+ }
+ argDescriptor {
+ name: "dW"
+ argType: INT64
+ argIndex: 11
+ }
+ argDescriptor {
+ name: "isSameMode"
+ argType: INT64
+ argIndex: 12
+ }
+ argDescriptor {
+ name: "extraParam0"
+ argType: INT64
+ argIndex: 13
+ }
+ argDescriptor {
+ name: "isNCDHW"
+ argType: INT64
+ argIndex: 14
+ }
+ argDescriptor {
+ name: "dtype"
+ argType: DATA_TYPE
+ }
+ argDescriptor {
+ name: "gradI"
+ argType: OUTPUT_TENSOR
+ }
+ argDescriptor {
+ name: "input"
+ argType: INPUT_TENSOR
+ }
+ argDescriptor {
+ name: "gradO"
+ argType: INPUT_TENSOR
+ argIndex: 1
+ }
}
opList {
name: "axpy"
+ argDescriptor {
+ name: "n"
+ argType: INT64
+ }
+ argDescriptor {
+ name: "dataType"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
}
+ argDescriptor {
+ name: "a"
+ argType: DOUBLE
+ }
argDescriptor {
name: "input"
argType: INPUT_TENSOR
@@ -2028,22 +2112,18 @@ opList {
argType: INPUT_TENSOR
argIndex: 2
}
- argDescriptor {
- name: "n"
- argType: INT64
- }
- argDescriptor {
- name: "dataType"
- argType: DATA_TYPE
- }
- argDescriptor {
- name: "a"
- argType: DOUBLE
- }
opDeclarationType: CONFIGURABLE_OP_IMPL
}
opList {
name: "barnes_edge_forces"
+ argDescriptor {
+ name: "N"
+ argType: INT64
+ }
+ argDescriptor {
+ name: "dtype"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -2067,14 +2147,6 @@ opList {
argType: INPUT_TENSOR
argIndex: 3
}
- argDescriptor {
- name: "N"
- argType: INT64
- }
- argDescriptor {
- name: "dtype"
- argType: DATA_TYPE
- }
}
opList {
name: "barnes_gains"
@@ -2100,6 +2172,14 @@ opList {
}
opList {
name: "barnes_symmetrized"
+ argDescriptor {
+ name: "N"
+ argType: INT64
+ }
+ argDescriptor {
+ name: "dtype"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputRows"
argType: OUTPUT_TENSOR
@@ -2133,30 +2213,9 @@ opList {
argType: INPUT_TENSOR
argIndex: 3
}
- argDescriptor {
- name: "N"
- argType: INT64
- }
- argDescriptor {
- name: "dtype"
- argType: DATA_TYPE
- }
}
opList {
name: "batch_to_space"
- argDescriptor {
- name: "outputs"
- argType: OUTPUT_TENSOR
- }
- argDescriptor {
- name: "input"
- argType: INPUT_TENSOR
- }
- argDescriptor {
- name: "crop"
- argType: INPUT_TENSOR
- argIndex: 1
- }
argDescriptor {
name: "blockSize"
argType: INT64
@@ -2175,9 +2234,30 @@ opList {
name: "dtype"
argType: DATA_TYPE
}
+ argDescriptor {
+ name: "outputs"
+ argType: OUTPUT_TENSOR
+ }
+ argDescriptor {
+ name: "input"
+ argType: INPUT_TENSOR
+ }
+ argDescriptor {
+ name: "crop"
+ argType: INPUT_TENSOR
+ argIndex: 1
+ }
}
opList {
name: "batch_to_space_nd"
+ argDescriptor {
+ name: "blocks"
+ argType: INT64
+ }
+ argDescriptor {
+ name: "dtype"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -2200,49 +2280,9 @@ opList {
argType: INPUT_TENSOR
argIndex: 2
}
- argDescriptor {
- name: "blocks"
- argType: INT64
- }
- argDescriptor {
- name: "dtype"
- argType: DATA_TYPE
- }
}
opList {
name: "batched_gemm"
- argDescriptor {
- name: "vC"
- argType: OUTPUT_TENSOR
- }
- argDescriptor {
- name: "transposeA"
- argType: BOOL
- }
- argDescriptor {
- name: "transposeB"
- argType: BOOL
- argIndex: 1
- }
- argDescriptor {
- name: "alpha"
- argType: INPUT_TENSOR
- }
- argDescriptor {
- name: "beta"
- argType: INPUT_TENSOR
- argIndex: 1
- }
- argDescriptor {
- name: "vA"
- argType: INPUT_TENSOR
- argIndex: 2
- }
- argDescriptor {
- name: "vB"
- argType: INPUT_TENSOR
- argIndex: 3
- }
argDescriptor {
name: "transA"
argType: INT64
@@ -2291,13 +2331,62 @@ opList {
name: "dtype"
argType: DATA_TYPE
}
+ argDescriptor {
+ name: "vC"
+ argType: OUTPUT_TENSOR
+ }
+ argDescriptor {
+ name: "transposeA"
+ argType: BOOL
+ }
+ argDescriptor {
+ name: "transposeB"
+ argType: BOOL
+ argIndex: 1
+ }
+ argDescriptor {
+ name: "alpha"
+ argType: INPUT_TENSOR
+ }
+ argDescriptor {
+ name: "beta"
+ argType: INPUT_TENSOR
+ argIndex: 1
+ }
+ argDescriptor {
+ name: "vA"
+ argType: INPUT_TENSOR
+ argIndex: 2
+ }
+ argDescriptor {
+ name: "vB"
+ argType: INPUT_TENSOR
+ argIndex: 3
+ }
}
opList {
name: "batchnorm"
+ argDescriptor {
+ name: "applyScale"
+ argType: INT64
+ }
+ argDescriptor {
+ name: "applyOffset"
+ argType: INT64
+ argIndex: 1
+ }
+ argDescriptor {
+ name: "dtype"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
}
+ argDescriptor {
+ name: "epsilon"
+ argType: DOUBLE
+ }
argDescriptor {
name: "inPlace"
argType: BOOL
@@ -2331,6 +2420,9 @@ opList {
argType: INPUT_TENSOR
argIndex: 3
}
+}
+opList {
+ name: "batchnorm_bp"
argDescriptor {
name: "applyScale"
argType: INT64
@@ -2344,13 +2436,6 @@ opList {
name: "dtype"
argType: DATA_TYPE
}
- argDescriptor {
- name: "epsilon"
- argType: DOUBLE
- }
-}
-opList {
- name: "batchnorm_bp"
argDescriptor {
name: "dLdI"
argType: OUTPUT_TENSOR
@@ -2370,6 +2455,10 @@ opList {
argType: OUTPUT_TENSOR
argIndex: 3
}
+ argDescriptor {
+ name: "epsilon"
+ argType: DOUBLE
+ }
argDescriptor {
name: "inPlace"
argType: BOOL
@@ -2403,23 +2492,6 @@ opList {
argType: INPUT_TENSOR
argIndex: 3
}
- argDescriptor {
- name: "applyScale"
- argType: INT64
- }
- argDescriptor {
- name: "applyOffset"
- argType: INT64
- argIndex: 1
- }
- argDescriptor {
- name: "dtype"
- argType: DATA_TYPE
- }
- argDescriptor {
- name: "epsilon"
- argType: DOUBLE
- }
}
opList {
name: "betainc"
@@ -2445,6 +2517,10 @@ opList {
}
opList {
name: "biasadd"
+ argDescriptor {
+ name: "dtype"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -2462,13 +2538,13 @@ opList {
argType: INPUT_TENSOR
argIndex: 1
}
+}
+opList {
+ name: "biasadd_bp"
argDescriptor {
name: "dtype"
argType: DATA_TYPE
}
-}
-opList {
- name: "biasadd_bp"
argDescriptor {
name: "gradI"
argType: OUTPUT_TENSOR
@@ -2496,13 +2572,27 @@ opList {
argType: INPUT_TENSOR
argIndex: 2
}
- argDescriptor {
- name: "dtype"
- argType: DATA_TYPE
- }
}
opList {
name: "bincount"
+ argDescriptor {
+ name: "minLength"
+ argType: INT64
+ }
+ argDescriptor {
+ name: "maxLength"
+ argType: INT64
+ argIndex: 1
+ }
+ argDescriptor {
+ name: "outputType"
+ argType: INT64
+ argIndex: 2
+ }
+ argDescriptor {
+ name: "outputType"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -2526,35 +2616,9 @@ opList {
argType: INPUT_TENSOR
argIndex: 3
}
- argDescriptor {
- name: "minLength"
- argType: INT64
- }
- argDescriptor {
- name: "maxLength"
- argType: INT64
- argIndex: 1
- }
- argDescriptor {
- name: "outputType"
- argType: INT64
- argIndex: 2
- }
- argDescriptor {
- name: "outputType"
- argType: DATA_TYPE
- }
}
opList {
name: "bitcast"
- argDescriptor {
- name: "outputs"
- argType: OUTPUT_TENSOR
- }
- argDescriptor {
- name: "input"
- argType: INPUT_TENSOR
- }
argDescriptor {
name: "newType"
argType: INT64
@@ -2563,9 +2627,26 @@ opList {
name: "dtype"
argType: DATA_TYPE
}
+ argDescriptor {
+ name: "outputs"
+ argType: OUTPUT_TENSOR
+ }
+ argDescriptor {
+ name: "input"
+ argType: INPUT_TENSOR
+ }
+ argDescriptor {
+ name: "dataType"
+ argType: INPUT_TENSOR
+ argIndex: 1
+ }
}
opList {
name: "bits_hamming_distance"
+ argDescriptor {
+ name: "dtype"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -2579,13 +2660,13 @@ opList {
argType: INPUT_TENSOR
argIndex: 1
}
- argDescriptor {
- name: "dtype"
- argType: DATA_TYPE
- }
}
opList {
name: "bitwise_and"
+ argDescriptor {
+ name: "dataType"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -2603,14 +2684,14 @@ opList {
argType: INPUT_TENSOR
argIndex: 1
}
- argDescriptor {
- name: "dataType"
- argType: DATA_TYPE
- }
opDeclarationType: BROADCASTABLE_OP_IMPL
}
opList {
name: "bitwise_or"
+ argDescriptor {
+ name: "dataType"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -2628,14 +2709,14 @@ opList {
argType: INPUT_TENSOR
argIndex: 1
}
- argDescriptor {
- name: "dataType"
- argType: DATA_TYPE
- }
opDeclarationType: BROADCASTABLE_OP_IMPL
}
opList {
name: "bitwise_xor"
+ argDescriptor {
+ name: "dataType"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -2653,14 +2734,14 @@ opList {
argType: INPUT_TENSOR
argIndex: 1
}
- argDescriptor {
- name: "dataType"
- argType: DATA_TYPE
- }
opDeclarationType: BROADCASTABLE_OP_IMPL
}
opList {
name: "bool_not"
+ argDescriptor {
+ name: "dataType"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -2669,10 +2750,6 @@ opList {
name: "input"
argType: INPUT_TENSOR
}
- argDescriptor {
- name: "dataType"
- argType: DATA_TYPE
- }
opDeclarationType: LEGACY_XYZ
}
opList {
@@ -2745,6 +2822,15 @@ opList {
}
opList {
name: "broadcast_amax"
+ argDescriptor {
+ name: "shape"
+ argType: INT64
+ }
+ argDescriptor {
+ name: "dimension"
+ argType: INT64
+ argIndex: 1
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -2757,19 +2843,19 @@ opList {
name: "input"
argType: INPUT_TENSOR
}
- argDescriptor {
- name: "shape"
- argType: INT64
- }
- argDescriptor {
- name: "dimension"
- argType: INT64
- argIndex: 1
- }
opDeclarationType: LEGACY_XYZ
}
opList {
name: "broadcast_amin"
+ argDescriptor {
+ name: "shape"
+ argType: INT64
+ }
+ argDescriptor {
+ name: "dimension"
+ argType: INT64
+ argIndex: 1
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -2782,19 +2868,14 @@ opList {
name: "input"
argType: INPUT_TENSOR
}
- argDescriptor {
- name: "shape"
- argType: INT64
- }
- argDescriptor {
- name: "dimension"
- argType: INT64
- argIndex: 1
- }
opDeclarationType: LEGACY_XYZ
}
opList {
name: "broadcast_dynamic_shape"
+ argDescriptor {
+ name: "dtype"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -2808,13 +2889,13 @@ opList {
argType: INPUT_TENSOR
argIndex: 1
}
- argDescriptor {
- name: "dtype"
- argType: DATA_TYPE
- }
}
opList {
name: "broadcast_equalto"
+ argDescriptor {
+ name: "dimension"
+ argType: INT64
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -2832,14 +2913,19 @@ opList {
argType: INPUT_TENSOR
argIndex: 1
}
- argDescriptor {
- name: "dimension"
- argType: INT64
- }
opDeclarationType: LEGACY_XYZ
}
opList {
name: "broadcast_greaterthan"
+ argDescriptor {
+ name: "shape"
+ argType: INT64
+ }
+ argDescriptor {
+ name: "dimension"
+ argType: INT64
+ argIndex: 1
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -2852,19 +2938,19 @@ opList {
name: "input"
argType: INPUT_TENSOR
}
- argDescriptor {
- name: "shape"
- argType: INT64
- }
- argDescriptor {
- name: "dimension"
- argType: INT64
- argIndex: 1
- }
opDeclarationType: LEGACY_XYZ
}
opList {
name: "broadcast_greaterthanorequal"
+ argDescriptor {
+ name: "shape"
+ argType: INT64
+ }
+ argDescriptor {
+ name: "dimension"
+ argType: INT64
+ argIndex: 1
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -2877,19 +2963,19 @@ opList {
name: "input"
argType: INPUT_TENSOR
}
- argDescriptor {
- name: "shape"
- argType: INT64
- }
- argDescriptor {
- name: "dimension"
- argType: INT64
- argIndex: 1
- }
opDeclarationType: LEGACY_XYZ
}
opList {
name: "broadcast_lessthan"
+ argDescriptor {
+ name: "shape"
+ argType: INT64
+ }
+ argDescriptor {
+ name: "dimension"
+ argType: INT64
+ argIndex: 1
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -2902,19 +2988,19 @@ opList {
name: "input"
argType: INPUT_TENSOR
}
- argDescriptor {
- name: "shape"
- argType: INT64
- }
- argDescriptor {
- name: "dimension"
- argType: INT64
- argIndex: 1
- }
opDeclarationType: LEGACY_XYZ
}
opList {
name: "broadcast_lessthanorequal"
+ argDescriptor {
+ name: "shape"
+ argType: INT64
+ }
+ argDescriptor {
+ name: "dimension"
+ argType: INT64
+ argIndex: 1
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -2927,19 +3013,19 @@ opList {
name: "input"
argType: INPUT_TENSOR
}
- argDescriptor {
- name: "shape"
- argType: INT64
- }
- argDescriptor {
- name: "dimension"
- argType: INT64
- argIndex: 1
- }
opDeclarationType: LEGACY_XYZ
}
opList {
name: "broadcast_max"
+ argDescriptor {
+ name: "shape"
+ argType: INT64
+ }
+ argDescriptor {
+ name: "dimension"
+ argType: INT64
+ argIndex: 1
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -2952,19 +3038,19 @@ opList {
name: "input"
argType: INPUT_TENSOR
}
- argDescriptor {
- name: "shape"
- argType: INT64
- }
- argDescriptor {
- name: "dimension"
- argType: INT64
- argIndex: 1
- }
opDeclarationType: LEGACY_XYZ
}
opList {
name: "broadcast_min"
+ argDescriptor {
+ name: "shape"
+ argType: INT64
+ }
+ argDescriptor {
+ name: "dimension"
+ argType: INT64
+ argIndex: 1
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -2977,19 +3063,19 @@ opList {
name: "input"
argType: INPUT_TENSOR
}
- argDescriptor {
- name: "shape"
- argType: INT64
- }
- argDescriptor {
- name: "dimension"
- argType: INT64
- argIndex: 1
- }
opDeclarationType: LEGACY_XYZ
}
opList {
name: "broadcast_notequal"
+ argDescriptor {
+ name: "shape"
+ argType: INT64
+ }
+ argDescriptor {
+ name: "dimension"
+ argType: INT64
+ argIndex: 1
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -3002,19 +3088,14 @@ opList {
name: "input"
argType: INPUT_TENSOR
}
- argDescriptor {
- name: "shape"
- argType: INT64
- }
- argDescriptor {
- name: "dimension"
- argType: INT64
- argIndex: 1
- }
opDeclarationType: LEGACY_XYZ
}
opList {
name: "broadcast_to"
+ argDescriptor {
+ name: "dtype"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -3028,13 +3109,13 @@ opList {
argType: INPUT_TENSOR
argIndex: 1
}
- argDescriptor {
- name: "dtype"
- argType: DATA_TYPE
- }
}
opList {
name: "broadcastadd"
+ argDescriptor {
+ name: "dimension"
+ argType: INT64
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -3048,14 +3129,19 @@ opList {
argType: INPUT_TENSOR
argIndex: 1
}
- argDescriptor {
- name: "dimension"
- argType: INT64
- }
opDeclarationType: LEGACY_XYZ
}
opList {
name: "broadcastcopy"
+ argDescriptor {
+ name: "shape"
+ argType: INT64
+ }
+ argDescriptor {
+ name: "dimension"
+ argType: INT64
+ argIndex: 1
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -3068,19 +3154,19 @@ opList {
name: "input"
argType: INPUT_TENSOR
}
- argDescriptor {
- name: "shape"
- argType: INT64
- }
- argDescriptor {
- name: "dimension"
- argType: INT64
- argIndex: 1
- }
opDeclarationType: LEGACY_XYZ
}
opList {
name: "broadcastdiv"
+ argDescriptor {
+ name: "shape"
+ argType: INT64
+ }
+ argDescriptor {
+ name: "dimension"
+ argType: INT64
+ argIndex: 1
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -3093,26 +3179,17 @@ opList {
name: "input"
argType: INPUT_TENSOR
}
- argDescriptor {
- name: "shape"
- argType: INT64
- }
- argDescriptor {
- name: "dimension"
- argType: INT64
- argIndex: 1
- }
opDeclarationType: LEGACY_XYZ
}
opList {
name: "broadcastgradientargs"
argDescriptor {
- name: "outputs"
- argType: OUTPUT_TENSOR
+ name: "dimension"
+ argType: INT64
}
argDescriptor {
- name: "inPlace"
- argType: BOOL
+ name: "outputs"
+ argType: OUTPUT_TENSOR
}
argDescriptor {
name: "input"
@@ -3123,14 +3200,19 @@ opList {
argType: INPUT_TENSOR
argIndex: 1
}
- argDescriptor {
- name: "dimension"
- argType: INT64
- }
opDeclarationType: OP_IMPL
}
opList {
name: "broadcastmul"
+ argDescriptor {
+ name: "shape"
+ argType: INT64
+ }
+ argDescriptor {
+ name: "dimension"
+ argType: INT64
+ argIndex: 1
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -3143,19 +3225,19 @@ opList {
name: "input"
argType: INPUT_TENSOR
}
- argDescriptor {
- name: "shape"
- argType: INT64
- }
- argDescriptor {
- name: "dimension"
- argType: INT64
- argIndex: 1
- }
opDeclarationType: LEGACY_XYZ
}
opList {
name: "broadcastrdiv"
+ argDescriptor {
+ name: "shape"
+ argType: INT64
+ }
+ argDescriptor {
+ name: "dimension"
+ argType: INT64
+ argIndex: 1
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -3168,19 +3250,19 @@ opList {
name: "input"
argType: INPUT_TENSOR
}
- argDescriptor {
- name: "shape"
- argType: INT64
- }
- argDescriptor {
- name: "dimension"
- argType: INT64
- argIndex: 1
- }
opDeclarationType: LEGACY_XYZ
}
opList {
name: "broadcastrsub"
+ argDescriptor {
+ name: "shape"
+ argType: INT64
+ }
+ argDescriptor {
+ name: "dimension"
+ argType: INT64
+ argIndex: 1
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -3193,31 +3275,10 @@ opList {
name: "input"
argType: INPUT_TENSOR
}
- argDescriptor {
- name: "shape"
- argType: INT64
- }
- argDescriptor {
- name: "dimension"
- argType: INT64
- argIndex: 1
- }
opDeclarationType: LEGACY_XYZ
}
opList {
name: "broadcastsub"
- argDescriptor {
- name: "outputs"
- argType: OUTPUT_TENSOR
- }
- argDescriptor {
- name: "inPlace"
- argType: BOOL
- }
- argDescriptor {
- name: "input"
- argType: INPUT_TENSOR
- }
argDescriptor {
name: "shape"
argType: INT64
@@ -3227,83 +3288,6 @@ opList {
argType: INT64
argIndex: 1
}
- opDeclarationType: LEGACY_XYZ
-}
-opList {
- name: "car"
- argDescriptor {
- name: "outputs"
- argType: OUTPUT_TENSOR
- }
- argDescriptor {
- name: "to"
- argType: INPUT_TENSOR
- }
- argDescriptor {
- name: "from"
- argType: INPUT_TENSOR
- argIndex: 1
- }
- argDescriptor {
- name: "mode"
- argType: INT64
- }
- argDescriptor {
- name: "dataType"
- argType: DATA_TYPE
- }
- argDescriptor {
- name: "compare"
- argType: DOUBLE
- }
- argDescriptor {
- name: "set"
- argType: DOUBLE
- argIndex: 1
- }
- argDescriptor {
- name: "eps"
- argType: DOUBLE
- argIndex: 2
- }
- opDeclarationType: LEGACY_XYZ
-}
-opList {
- name: "cas"
- argDescriptor {
- name: "outputs"
- argType: OUTPUT_TENSOR
- }
- argDescriptor {
- name: "input"
- argType: INPUT_TENSOR
- }
- argDescriptor {
- name: "mode"
- argType: INT64
- }
- argDescriptor {
- name: "dataType"
- argType: DATA_TYPE
- }
- argDescriptor {
- name: "compare"
- argType: DOUBLE
- }
- argDescriptor {
- name: "set"
- argType: DOUBLE
- argIndex: 1
- }
- argDescriptor {
- name: "eps"
- argType: DOUBLE
- argIndex: 2
- }
- opDeclarationType: LEGACY_XYZ
-}
-opList {
- name: "cast"
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -3316,6 +3300,83 @@ opList {
name: "input"
argType: INPUT_TENSOR
}
+ opDeclarationType: LEGACY_XYZ
+}
+opList {
+ name: "car"
+ argDescriptor {
+ name: "mode"
+ argType: INT64
+ }
+ argDescriptor {
+ name: "dataType"
+ argType: DATA_TYPE
+ }
+ argDescriptor {
+ name: "outputs"
+ argType: OUTPUT_TENSOR
+ }
+ argDescriptor {
+ name: "compare"
+ argType: DOUBLE
+ }
+ argDescriptor {
+ name: "set"
+ argType: DOUBLE
+ argIndex: 1
+ }
+ argDescriptor {
+ name: "eps"
+ argType: DOUBLE
+ argIndex: 2
+ }
+ argDescriptor {
+ name: "input"
+ argType: INPUT_TENSOR
+ }
+ argDescriptor {
+ name: "y"
+ argType: INPUT_TENSOR
+ argIndex: 1
+ }
+ opDeclarationType: LEGACY_XYZ
+}
+opList {
+ name: "cas"
+ argDescriptor {
+ name: "mode"
+ argType: INT64
+ }
+ argDescriptor {
+ name: "dataType"
+ argType: DATA_TYPE
+ }
+ argDescriptor {
+ name: "outputs"
+ argType: OUTPUT_TENSOR
+ }
+ argDescriptor {
+ name: "compare"
+ argType: DOUBLE
+ }
+ argDescriptor {
+ name: "set"
+ argType: DOUBLE
+ argIndex: 1
+ }
+ argDescriptor {
+ name: "eps"
+ argType: DOUBLE
+ argIndex: 2
+ }
+ argDescriptor {
+ name: "input"
+ argType: INPUT_TENSOR
+ }
+ opDeclarationType: LEGACY_XYZ
+}
+opList {
+ name: "cast"
argDescriptor {
name: "dst"
argType: INT64
@@ -3324,9 +3385,30 @@ opList {
name: "dtype"
argType: DATA_TYPE
}
+ argDescriptor {
+ name: "outputs"
+ argType: OUTPUT_TENSOR
+ }
+ argDescriptor {
+ name: "inPlace"
+ argType: BOOL
+ }
+ argDescriptor {
+ name: "input"
+ argType: INPUT_TENSOR
+ }
}
opList {
name: "cbow"
+ argDescriptor {
+ name: "numWorkers"
+ argType: INT64
+ }
+ argDescriptor {
+ name: "nsRounds"
+ argType: INT64
+ argIndex: 1
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -3414,19 +3496,14 @@ opList {
argType: INPUT_TENSOR
argIndex: 14
}
- argDescriptor {
- name: "numWorkers"
- argType: INT64
- }
- argDescriptor {
- name: "nsRounds"
- argType: INT64
- argIndex: 1
- }
opDeclarationType: CONFIGURABLE_OP_IMPL
}
opList {
name: "ceil"
+ argDescriptor {
+ name: "dataType"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -3439,14 +3516,18 @@ opList {
name: "input"
argType: INPUT_TENSOR
}
- argDescriptor {
- name: "dataType"
- argType: DATA_TYPE
- }
opDeclarationType: LEGACY_XYZ
}
opList {
name: "cell_contains"
+ argDescriptor {
+ name: "dimension"
+ argType: INT64
+ }
+ argDescriptor {
+ name: "dtype"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -3469,17 +3550,13 @@ opList {
argType: INPUT_TENSOR
argIndex: 2
}
- argDescriptor {
- name: "dimension"
- argType: INT64
- }
+}
+opList {
+ name: "check_numerics"
argDescriptor {
name: "dtype"
argType: DATA_TYPE
}
-}
-opList {
- name: "check_numerics"
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -3493,10 +3570,6 @@ opList {
argType: INPUT_TENSOR
argIndex: 1
}
- argDescriptor {
- name: "dtype"
- argType: DATA_TYPE
- }
}
opList {
name: "choice"
@@ -3529,6 +3602,14 @@ opList {
}
opList {
name: "choose"
+ argDescriptor {
+ name: "mode"
+ argType: INT64
+ }
+ argDescriptor {
+ name: "dtype"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -3538,6 +3619,10 @@ opList {
argType: OUTPUT_TENSOR
argIndex: 1
}
+ argDescriptor {
+ name: "scalar"
+ argType: DOUBLE
+ }
argDescriptor {
name: "arg"
argType: INPUT_TENSOR
@@ -3547,64 +3632,60 @@ opList {
argType: INPUT_TENSOR
argIndex: 1
}
- argDescriptor {
- name: "mode"
- argType: INT64
- }
+}
+opList {
+ name: "clip_by_global_norm"
argDescriptor {
name: "dtype"
argType: DATA_TYPE
}
- argDescriptor {
- name: "scalar"
- argType: DOUBLE
- }
-}
-opList {
- name: "clip_by_global_norm"
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
}
- argDescriptor {
- name: "input"
- argType: INPUT_TENSOR
- }
- argDescriptor {
- name: "dtype"
- argType: DATA_TYPE
- }
argDescriptor {
name: "clipNorm"
argType: DOUBLE
}
+ argDescriptor {
+ name: "input"
+ argType: INPUT_TENSOR
+ }
}
opList {
name: "clipbyavgnorm"
- argDescriptor {
- name: "outputs"
- argType: OUTPUT_TENSOR
- }
- argDescriptor {
- name: "input"
- argType: INPUT_TENSOR
- }
argDescriptor {
name: "dimensions"
argType: INT64
}
+ argDescriptor {
+ name: "outputs"
+ argType: OUTPUT_TENSOR
+ }
argDescriptor {
name: "clipNorm"
argType: DOUBLE
}
+ argDescriptor {
+ name: "input"
+ argType: INPUT_TENSOR
+ }
opDeclarationType: CONFIGURABLE_OP_IMPL
}
opList {
name: "clipbyavgnorm_bp"
+ argDescriptor {
+ name: "dtype"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "gradI"
argType: OUTPUT_TENSOR
}
+ argDescriptor {
+ name: "clipNorm"
+ argType: DOUBLE
+ }
argDescriptor {
name: "input"
argType: INPUT_TENSOR
@@ -3614,41 +3695,45 @@ opList {
argType: INPUT_TENSOR
argIndex: 1
}
- argDescriptor {
- name: "dtype"
- argType: DATA_TYPE
- }
- argDescriptor {
- name: "clipNorm"
- argType: DOUBLE
- }
}
opList {
name: "clipbynorm"
- argDescriptor {
- name: "outputs"
- argType: OUTPUT_TENSOR
- }
- argDescriptor {
- name: "input"
- argType: INPUT_TENSOR
- }
argDescriptor {
name: "dimensions"
argType: INT64
}
+ argDescriptor {
+ name: "outputs"
+ argType: OUTPUT_TENSOR
+ }
argDescriptor {
name: "clipValue"
argType: DOUBLE
}
+ argDescriptor {
+ name: "input"
+ argType: INPUT_TENSOR
+ }
opDeclarationType: CONFIGURABLE_OP_IMPL
}
opList {
name: "clipbynorm_bp"
+ argDescriptor {
+ name: "dimensions"
+ argType: INT64
+ }
+ argDescriptor {
+ name: "dtype"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "gradI"
argType: OUTPUT_TENSOR
}
+ argDescriptor {
+ name: "clipValue"
+ argType: DOUBLE
+ }
argDescriptor {
name: "input"
argType: INPUT_TENSOR
@@ -3658,18 +3743,6 @@ opList {
argType: INPUT_TENSOR
argIndex: 1
}
- argDescriptor {
- name: "dimensions"
- argType: INT64
- }
- argDescriptor {
- name: "dtype"
- argType: DATA_TYPE
- }
- argDescriptor {
- name: "clipValue"
- argType: DOUBLE
- }
}
opList {
name: "clipbyvalue"
@@ -3677,10 +3750,6 @@ opList {
name: "outputs"
argType: OUTPUT_TENSOR
}
- argDescriptor {
- name: "input"
- argType: INPUT_TENSOR
- }
argDescriptor {
name: "left"
argType: DOUBLE
@@ -3690,6 +3759,10 @@ opList {
argType: DOUBLE
argIndex: 1
}
+ argDescriptor {
+ name: "input"
+ argType: INPUT_TENSOR
+ }
opDeclarationType: CONFIGURABLE_OP_IMPL
}
opList {
@@ -3706,19 +3779,6 @@ opList {
}
opList {
name: "col2im"
- argDescriptor {
- name: "outputs"
- argType: OUTPUT_TENSOR
- }
- argDescriptor {
- name: "input"
- argType: INPUT_TENSOR
- }
- argDescriptor {
- name: "inputArrays"
- argType: INPUT_TENSOR
- argIndex: 1
- }
argDescriptor {
name: "strideY"
argType: INT64
@@ -3762,13 +3822,34 @@ opList {
name: "dtype"
argType: DATA_TYPE
}
+ argDescriptor {
+ name: "outputs"
+ argType: OUTPUT_TENSOR
+ }
+ argDescriptor {
+ name: "input"
+ argType: INPUT_TENSOR
+ }
+ argDescriptor {
+ name: "inputArrays"
+ argType: INPUT_TENSOR
+ argIndex: 1
+ }
}
opList {
name: "compare_and_bitpack"
+ argDescriptor {
+ name: "dtype"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
}
+ argDescriptor {
+ name: "threshold"
+ argType: DOUBLE
+ }
argDescriptor {
name: "input"
argType: INPUT_TENSOR
@@ -3778,17 +3859,13 @@ opList {
argType: INPUT_TENSOR
argIndex: 1
}
+}
+opList {
+ name: "compat_sparse_to_dense"
argDescriptor {
name: "dtype"
argType: DATA_TYPE
}
- argDescriptor {
- name: "threshold"
- argType: DOUBLE
- }
-}
-opList {
- name: "compat_sparse_to_dense"
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -3812,13 +3889,13 @@ opList {
argType: INPUT_TENSOR
argIndex: 3
}
+}
+opList {
+ name: "compat_string_split"
argDescriptor {
name: "dtype"
argType: DATA_TYPE
}
-}
-opList {
- name: "compat_string_split"
argDescriptor {
name: "indices"
argType: OUTPUT_TENSOR
@@ -3847,13 +3924,17 @@ opList {
argType: INPUT_TENSOR
argIndex: 3
}
+}
+opList {
+ name: "concat"
+ argDescriptor {
+ name: "concatDimension"
+ argType: INT64
+ }
argDescriptor {
name: "dtype"
argType: DATA_TYPE
}
-}
-opList {
- name: "concat"
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -3871,6 +3952,9 @@ opList {
argType: INPUT_TENSOR
argIndex: 1
}
+}
+opList {
+ name: "concat_bp"
argDescriptor {
name: "concatDimension"
argType: INT64
@@ -3879,9 +3963,6 @@ opList {
name: "dtype"
argType: DATA_TYPE
}
-}
-opList {
- name: "concat_bp"
argDescriptor {
name: "epsilonChunk"
argType: OUTPUT_TENSOR
@@ -3899,17 +3980,22 @@ opList {
argType: INPUT_TENSOR
argIndex: 1
}
+}
+opList {
+ name: "confusion_matrix"
argDescriptor {
- name: "concatDimension"
+ name: "numClasses"
argType: INT64
}
+ argDescriptor {
+ name: "dataType"
+ argType: INT64
+ argIndex: 1
+ }
argDescriptor {
name: "dtype"
argType: DATA_TYPE
}
-}
-opList {
- name: "confusion_matrix"
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -3928,35 +4014,9 @@ opList {
argType: INPUT_TENSOR
argIndex: 2
}
- argDescriptor {
- name: "dataType"
- argType: INT64
- }
- argDescriptor {
- name: "dtype"
- argType: DATA_TYPE
- }
}
opList {
name: "conv1d"
- argDescriptor {
- name: "outputs"
- argType: OUTPUT_TENSOR
- }
- argDescriptor {
- name: "input"
- argType: INPUT_TENSOR
- }
- argDescriptor {
- name: "weights"
- argType: INPUT_TENSOR
- argIndex: 1
- }
- argDescriptor {
- name: "bias"
- argType: INPUT_TENSOR
- argIndex: 2
- }
argDescriptor {
name: "kW"
argType: INT64
@@ -3995,42 +4055,27 @@ opList {
name: "dtype"
argType: DATA_TYPE
}
+ argDescriptor {
+ name: "outputs"
+ argType: OUTPUT_TENSOR
+ }
+ argDescriptor {
+ name: "input"
+ argType: INPUT_TENSOR
+ }
+ argDescriptor {
+ name: "weights"
+ argType: INPUT_TENSOR
+ argIndex: 1
+ }
+ argDescriptor {
+ name: "bias"
+ argType: INPUT_TENSOR
+ argIndex: 2
+ }
}
opList {
name: "conv1d_bp"
- argDescriptor {
- name: "gradI"
- argType: OUTPUT_TENSOR
- }
- argDescriptor {
- name: "gradW"
- argType: OUTPUT_TENSOR
- argIndex: 1
- }
- argDescriptor {
- name: "gradB"
- argType: OUTPUT_TENSOR
- argIndex: 2
- }
- argDescriptor {
- name: "input"
- argType: INPUT_TENSOR
- }
- argDescriptor {
- name: "weights"
- argType: INPUT_TENSOR
- argIndex: 1
- }
- argDescriptor {
- name: "bias"
- argType: INPUT_TENSOR
- argIndex: 2
- }
- argDescriptor {
- name: "gradO"
- argType: INPUT_TENSOR
- argIndex: 3
- }
argDescriptor {
name: "kW"
argType: INT64
@@ -4069,88 +4114,6 @@ opList {
name: "dtype"
argType: DATA_TYPE
}
-}
-opList {
- name: "conv2d"
- argDescriptor {
- name: "outputs"
- argType: OUTPUT_TENSOR
- }
- argDescriptor {
- name: "input"
- argType: INPUT_TENSOR
- }
- argDescriptor {
- name: "weights"
- argType: INPUT_TENSOR
- argIndex: 1
- }
- argDescriptor {
- name: "bias"
- argType: INPUT_TENSOR
- argIndex: 2
- }
- argDescriptor {
- name: "kH"
- argType: INT64
- }
- argDescriptor {
- name: "kW"
- argType: INT64
- argIndex: 1
- }
- argDescriptor {
- name: "sH"
- argType: INT64
- argIndex: 2
- }
- argDescriptor {
- name: "sW"
- argType: INT64
- argIndex: 3
- }
- argDescriptor {
- name: "pH"
- argType: INT64
- argIndex: 4
- }
- argDescriptor {
- name: "pW"
- argType: INT64
- argIndex: 5
- }
- argDescriptor {
- name: "dH"
- argType: INT64
- argIndex: 6
- }
- argDescriptor {
- name: "dW"
- argType: INT64
- argIndex: 7
- }
- argDescriptor {
- name: "isSameMode"
- argType: INT64
- argIndex: 8
- }
- argDescriptor {
- name: "isNCHW"
- argType: INT64
- argIndex: 9
- }
- argDescriptor {
- name: "wFormat"
- argType: INT64
- argIndex: 10
- }
- argDescriptor {
- name: "dtype"
- argType: DATA_TYPE
- }
-}
-opList {
- name: "conv2d_bp"
argDescriptor {
name: "gradI"
argType: OUTPUT_TENSOR
@@ -4184,6 +4147,9 @@ opList {
argType: INPUT_TENSOR
argIndex: 3
}
+}
+opList {
+ name: "conv2d"
argDescriptor {
name: "kH"
argType: INT64
@@ -4242,9 +4208,179 @@ opList {
name: "dtype"
argType: DATA_TYPE
}
+ argDescriptor {
+ name: "outputs"
+ argType: OUTPUT_TENSOR
+ }
+ argDescriptor {
+ name: "input"
+ argType: INPUT_TENSOR
+ }
+ argDescriptor {
+ name: "weights"
+ argType: INPUT_TENSOR
+ argIndex: 1
+ }
+ argDescriptor {
+ name: "bias"
+ argType: INPUT_TENSOR
+ argIndex: 2
+ }
+}
+opList {
+ name: "conv2d_bp"
+ argDescriptor {
+ name: "kH"
+ argType: INT64
+ }
+ argDescriptor {
+ name: "kW"
+ argType: INT64
+ argIndex: 1
+ }
+ argDescriptor {
+ name: "sH"
+ argType: INT64
+ argIndex: 2
+ }
+ argDescriptor {
+ name: "sW"
+ argType: INT64
+ argIndex: 3
+ }
+ argDescriptor {
+ name: "pH"
+ argType: INT64
+ argIndex: 4
+ }
+ argDescriptor {
+ name: "pW"
+ argType: INT64
+ argIndex: 5
+ }
+ argDescriptor {
+ name: "dH"
+ argType: INT64
+ argIndex: 6
+ }
+ argDescriptor {
+ name: "dW"
+ argType: INT64
+ argIndex: 7
+ }
+ argDescriptor {
+ name: "isSameMode"
+ argType: INT64
+ argIndex: 8
+ }
+ argDescriptor {
+ name: "isNCHW"
+ argType: INT64
+ argIndex: 9
+ }
+ argDescriptor {
+ name: "wFormat"
+ argType: INT64
+ argIndex: 10
+ }
+ argDescriptor {
+ name: "dtype"
+ argType: DATA_TYPE
+ }
+ argDescriptor {
+ name: "gradI"
+ argType: OUTPUT_TENSOR
+ }
+ argDescriptor {
+ name: "gradW"
+ argType: OUTPUT_TENSOR
+ argIndex: 1
+ }
+ argDescriptor {
+ name: "gradB"
+ argType: OUTPUT_TENSOR
+ argIndex: 2
+ }
+ argDescriptor {
+ name: "input"
+ argType: INPUT_TENSOR
+ }
+ argDescriptor {
+ name: "weights"
+ argType: INPUT_TENSOR
+ argIndex: 1
+ }
+ argDescriptor {
+ name: "bias"
+ argType: INPUT_TENSOR
+ argIndex: 2
+ }
+ argDescriptor {
+ name: "gradO"
+ argType: INPUT_TENSOR
+ argIndex: 3
+ }
}
opList {
name: "conv2d_input_bp"
+ argDescriptor {
+ name: "kH"
+ argType: INT64
+ }
+ argDescriptor {
+ name: "kW"
+ argType: INT64
+ argIndex: 1
+ }
+ argDescriptor {
+ name: "sH"
+ argType: INT64
+ argIndex: 2
+ }
+ argDescriptor {
+ name: "sW"
+ argType: INT64
+ argIndex: 3
+ }
+ argDescriptor {
+ name: "pH"
+ argType: INT64
+ argIndex: 4
+ }
+ argDescriptor {
+ name: "pW"
+ argType: INT64
+ argIndex: 5
+ }
+ argDescriptor {
+ name: "dH"
+ argType: INT64
+ argIndex: 6
+ }
+ argDescriptor {
+ name: "dW"
+ argType: INT64
+ argIndex: 7
+ }
+ argDescriptor {
+ name: "isSameMode"
+ argType: INT64
+ argIndex: 8
+ }
+ argDescriptor {
+ name: "isNCHW"
+ argType: INT64
+ argIndex: 9
+ }
+ argDescriptor {
+ name: "wFormat"
+ argType: INT64
+ argIndex: 10
+ }
+ argDescriptor {
+ name: "dtype"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "gradI"
argType: OUTPUT_TENSOR
@@ -4263,85 +4399,9 @@ opList {
argType: INPUT_TENSOR
argIndex: 2
}
- argDescriptor {
- name: "kH"
- argType: INT64
- }
- argDescriptor {
- name: "kW"
- argType: INT64
- argIndex: 1
- }
- argDescriptor {
- name: "sH"
- argType: INT64
- argIndex: 2
- }
- argDescriptor {
- name: "sW"
- argType: INT64
- argIndex: 3
- }
- argDescriptor {
- name: "pH"
- argType: INT64
- argIndex: 4
- }
- argDescriptor {
- name: "pW"
- argType: INT64
- argIndex: 5
- }
- argDescriptor {
- name: "dH"
- argType: INT64
- argIndex: 6
- }
- argDescriptor {
- name: "dW"
- argType: INT64
- argIndex: 7
- }
- argDescriptor {
- name: "isSameMode"
- argType: INT64
- argIndex: 8
- }
- argDescriptor {
- name: "isNCHW"
- argType: INT64
- argIndex: 9
- }
- argDescriptor {
- name: "wFormat"
- argType: INT64
- argIndex: 10
- }
- argDescriptor {
- name: "dtype"
- argType: DATA_TYPE
- }
}
opList {
name: "conv3dnew"
- argDescriptor {
- name: "outputs"
- argType: OUTPUT_TENSOR
- }
- argDescriptor {
- name: "input"
- argType: INPUT_TENSOR
- }
- argDescriptor {
- name: "weights"
- argType: INPUT_TENSOR
- argIndex: 1
- }
- argDescriptor {
- name: "bias"
- argType: INPUT_TENSOR
- argIndex: 2
- }
argDescriptor {
name: "kD"
argType: INT64
@@ -4420,9 +4480,105 @@ opList {
name: "dtype"
argType: DATA_TYPE
}
+ argDescriptor {
+ name: "outputs"
+ argType: OUTPUT_TENSOR
+ }
+ argDescriptor {
+ name: "input"
+ argType: INPUT_TENSOR
+ }
+ argDescriptor {
+ name: "weights"
+ argType: INPUT_TENSOR
+ argIndex: 1
+ }
+ argDescriptor {
+ name: "bias"
+ argType: INPUT_TENSOR
+ argIndex: 2
+ }
}
opList {
name: "conv3dnew_bp"
+ argDescriptor {
+ name: "kD"
+ argType: INT64
+ }
+ argDescriptor {
+ name: "kH"
+ argType: INT64
+ argIndex: 1
+ }
+ argDescriptor {
+ name: "kW"
+ argType: INT64
+ argIndex: 2
+ }
+ argDescriptor {
+ name: "sD"
+ argType: INT64
+ argIndex: 3
+ }
+ argDescriptor {
+ name: "sH"
+ argType: INT64
+ argIndex: 4
+ }
+ argDescriptor {
+ name: "sW"
+ argType: INT64
+ argIndex: 5
+ }
+ argDescriptor {
+ name: "pD"
+ argType: INT64
+ argIndex: 6
+ }
+ argDescriptor {
+ name: "pH"
+ argType: INT64
+ argIndex: 7
+ }
+ argDescriptor {
+ name: "pW"
+ argType: INT64
+ argIndex: 8
+ }
+ argDescriptor {
+ name: "dD"
+ argType: INT64
+ argIndex: 9
+ }
+ argDescriptor {
+ name: "dH"
+ argType: INT64
+ argIndex: 10
+ }
+ argDescriptor {
+ name: "dW"
+ argType: INT64
+ argIndex: 11
+ }
+ argDescriptor {
+ name: "paddingMode"
+ argType: INT64
+ argIndex: 12
+ }
+ argDescriptor {
+ name: "isNCDHW"
+ argType: INT64
+ argIndex: 13
+ }
+ argDescriptor {
+ name: "wFormat"
+ argType: INT64
+ argIndex: 14
+ }
+ argDescriptor {
+ name: "dtype"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "gradI"
argType: OUTPUT_TENSOR
@@ -4456,87 +4612,13 @@ opList {
argType: INPUT_TENSOR
argIndex: 3
}
- argDescriptor {
- name: "kD"
- argType: INT64
- }
- argDescriptor {
- name: "kH"
- argType: INT64
- argIndex: 1
- }
- argDescriptor {
- name: "kW"
- argType: INT64
- argIndex: 2
- }
- argDescriptor {
- name: "sD"
- argType: INT64
- argIndex: 3
- }
- argDescriptor {
- name: "sH"
- argType: INT64
- argIndex: 4
- }
- argDescriptor {
- name: "sW"
- argType: INT64
- argIndex: 5
- }
- argDescriptor {
- name: "pD"
- argType: INT64
- argIndex: 6
- }
- argDescriptor {
- name: "pH"
- argType: INT64
- argIndex: 7
- }
- argDescriptor {
- name: "pW"
- argType: INT64
- argIndex: 8
- }
- argDescriptor {
- name: "dD"
- argType: INT64
- argIndex: 9
- }
- argDescriptor {
- name: "dH"
- argType: INT64
- argIndex: 10
- }
- argDescriptor {
- name: "dW"
- argType: INT64
- argIndex: 11
- }
- argDescriptor {
- name: "paddingMode"
- argType: INT64
- argIndex: 12
- }
- argDescriptor {
- name: "isNCDHW"
- argType: INT64
- argIndex: 13
- }
- argDescriptor {
- name: "wFormat"
- argType: INT64
- argIndex: 14
- }
- argDescriptor {
- name: "dtype"
- argType: DATA_TYPE
- }
}
opList {
name: "copy"
+ argDescriptor {
+ name: "dataType"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -4554,14 +4636,14 @@ opList {
argType: INPUT_TENSOR
argIndex: 1
}
- argDescriptor {
- name: "dataType"
- argType: DATA_TYPE
- }
opDeclarationType: LEGACY_XYZ
}
opList {
name: "cos"
+ argDescriptor {
+ name: "dataType"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -4574,14 +4656,14 @@ opList {
name: "input"
argType: INPUT_TENSOR
}
- argDescriptor {
- name: "dataType"
- argType: DATA_TYPE
- }
opDeclarationType: LEGACY_XYZ
}
opList {
name: "cosh"
+ argDescriptor {
+ name: "dataType"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -4594,14 +4676,23 @@ opList {
name: "input"
argType: INPUT_TENSOR
}
- argDescriptor {
- name: "dataType"
- argType: DATA_TYPE
- }
opDeclarationType: LEGACY_XYZ
}
opList {
name: "cosine_distance_loss"
+ argDescriptor {
+ name: "reductionMode"
+ argType: INT64
+ }
+ argDescriptor {
+ name: "dimensions"
+ argType: INT64
+ argIndex: 1
+ }
+ argDescriptor {
+ name: "dtype"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -4620,6 +4711,9 @@ opList {
argType: INPUT_TENSOR
argIndex: 2
}
+}
+opList {
+ name: "cosine_distance_loss_grad"
argDescriptor {
name: "reductionMode"
argType: INT64
@@ -4633,9 +4727,6 @@ opList {
name: "dtype"
argType: DATA_TYPE
}
-}
-opList {
- name: "cosine_distance_loss_grad"
argDescriptor {
name: "dLdp"
argType: OUTPUT_TENSOR
@@ -4664,22 +4755,13 @@ opList {
argType: INPUT_TENSOR
argIndex: 2
}
- argDescriptor {
- name: "reductionMode"
- argType: INT64
- }
- argDescriptor {
- name: "dimensions"
- argType: INT64
- argIndex: 1
- }
- argDescriptor {
- name: "dtype"
- argType: DATA_TYPE
- }
}
opList {
name: "cosinedistance"
+ argDescriptor {
+ name: "dimensions"
+ argType: INT64
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -4702,14 +4784,14 @@ opList {
argType: INPUT_TENSOR
argIndex: 1
}
- argDescriptor {
- name: "dimensions"
- argType: INT64
- }
opDeclarationType: LEGACY_XYZ
}
opList {
name: "cosinesimilarity"
+ argDescriptor {
+ name: "dimensions"
+ argType: INT64
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -4732,14 +4814,14 @@ opList {
argType: INPUT_TENSOR
argIndex: 1
}
- argDescriptor {
- name: "dimensions"
- argType: INT64
- }
opDeclarationType: LEGACY_XYZ
}
opList {
name: "countNonZero"
+ argDescriptor {
+ name: "dimensions"
+ argType: INT64
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -4757,14 +4839,14 @@ opList {
argType: INPUT_TENSOR
argIndex: 1
}
- argDescriptor {
- name: "dimensions"
- argType: INT64
- }
opDeclarationType: LEGACY_XYZ
}
opList {
name: "countZero"
+ argDescriptor {
+ name: "dimensions"
+ argType: INT64
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -4782,26 +4864,10 @@ opList {
argType: INPUT_TENSOR
argIndex: 1
}
- argDescriptor {
- name: "dimensions"
- argType: INT64
- }
opDeclarationType: LEGACY_XYZ
}
opList {
name: "create"
- argDescriptor {
- name: "outputs"
- argType: OUTPUT_TENSOR
- }
- argDescriptor {
- name: "init"
- argType: BOOL
- }
- argDescriptor {
- name: "input"
- argType: INPUT_TENSOR
- }
argDescriptor {
name: "order"
argType: INT64
@@ -4815,17 +4881,21 @@ opList {
name: "outputType"
argType: DATA_TYPE
}
-}
-opList {
- name: "create_list"
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
}
+ argDescriptor {
+ name: "init"
+ argType: BOOL
+ }
argDescriptor {
name: "input"
argType: INPUT_TENSOR
}
+}
+opList {
+ name: "create_list"
argDescriptor {
name: "height"
argType: INT64
@@ -4835,10 +4905,6 @@ opList {
argType: INT64
argIndex: 1
}
- opDeclarationType: LIST_OP_IMPL
-}
-opList {
- name: "crelu"
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -4847,13 +4913,29 @@ opList {
name: "input"
argType: INPUT_TENSOR
}
+ opDeclarationType: LIST_OP_IMPL
+}
+opList {
+ name: "crelu"
argDescriptor {
name: "dtype"
argType: DATA_TYPE
}
+ argDescriptor {
+ name: "outputs"
+ argType: OUTPUT_TENSOR
+ }
+ argDescriptor {
+ name: "input"
+ argType: INPUT_TENSOR
+ }
}
opList {
name: "crelu_bp"
+ argDescriptor {
+ name: "dtype"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "epsilon"
argType: OUTPUT_TENSOR
@@ -4867,17 +4949,25 @@ opList {
argType: INPUT_TENSOR
argIndex: 1
}
- argDescriptor {
- name: "dtype"
- argType: DATA_TYPE
- }
}
opList {
name: "crop_and_resize"
+ argDescriptor {
+ name: "method"
+ argType: INT64
+ }
+ argDescriptor {
+ name: "dtype"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
}
+ argDescriptor {
+ name: "extrapolationVal"
+ argType: DOUBLE
+ }
argDescriptor {
name: "image"
argType: INPUT_TENSOR
@@ -4897,18 +4987,6 @@ opList {
argType: INPUT_TENSOR
argIndex: 3
}
- argDescriptor {
- name: "method"
- argType: INT64
- }
- argDescriptor {
- name: "dtype"
- argType: DATA_TYPE
- }
- argDescriptor {
- name: "extrapolationVal"
- argType: DOUBLE
- }
}
opList {
name: "cross"
@@ -4927,8 +5005,80 @@ opList {
}
opDeclarationType: OP_IMPL
}
+opList {
+ name: "ctc_loss"
+ argDescriptor {
+ name: "blankIndex"
+ argType: INT64
+ }
+ argDescriptor {
+ name: "dtype"
+ argType: DATA_TYPE
+ }
+ argDescriptor {
+ name: "outputLosses"
+ argType: OUTPUT_TENSOR
+ }
+ argDescriptor {
+ name: "targetLabels"
+ argType: INPUT_TENSOR
+ }
+ argDescriptor {
+ name: "logitInput"
+ argType: INPUT_TENSOR
+ argIndex: 1
+ }
+ argDescriptor {
+ name: "targetLabelLengths"
+ argType: INPUT_TENSOR
+ argIndex: 2
+ }
+ argDescriptor {
+ name: "logitInputLengths"
+ argType: INPUT_TENSOR
+ argIndex: 3
+ }
+}
+opList {
+ name: "ctc_loss_grad"
+ argDescriptor {
+ name: "blankIndex"
+ argType: INT64
+ }
+ argDescriptor {
+ name: "dtype"
+ argType: DATA_TYPE
+ }
+ argDescriptor {
+ name: "outputGradients"
+ argType: OUTPUT_TENSOR
+ }
+ argDescriptor {
+ name: "targetLabels"
+ argType: INPUT_TENSOR
+ }
+ argDescriptor {
+ name: "logitInput"
+ argType: INPUT_TENSOR
+ argIndex: 1
+ }
+ argDescriptor {
+ name: "targetLabelLengths"
+ argType: INPUT_TENSOR
+ argIndex: 2
+ }
+ argDescriptor {
+ name: "logitInputLengths"
+ argType: INPUT_TENSOR
+ argIndex: 3
+ }
+}
opList {
name: "cube"
+ argDescriptor {
+ name: "dataType"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -4941,10 +5091,6 @@ opList {
name: "input"
argType: INPUT_TENSOR
}
- argDescriptor {
- name: "dataType"
- argType: DATA_TYPE
- }
opDeclarationType: CONFIGURABLE_OP_IMPL
}
opList {
@@ -4966,6 +5112,10 @@ opList {
}
opList {
name: "cubederivative"
+ argDescriptor {
+ name: "dataType"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -4978,14 +5128,24 @@ opList {
name: "input"
argType: INPUT_TENSOR
}
- argDescriptor {
- name: "dataType"
- argType: DATA_TYPE
- }
opDeclarationType: LEGACY_XYZ
}
opList {
name: "cumprod"
+ argDescriptor {
+ name: "exclusive"
+ argType: INT64
+ }
+ argDescriptor {
+ name: "reverse"
+ argType: INT64
+ argIndex: 1
+ }
+ argDescriptor {
+ name: "dimensions"
+ argType: INT64
+ argIndex: 2
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -5008,24 +5168,28 @@ opList {
argType: INPUT_TENSOR
argIndex: 1
}
- argDescriptor {
- name: "exclusive"
- argType: INT64
- }
- argDescriptor {
- name: "reverse"
- argType: INT64
- argIndex: 1
- }
- argDescriptor {
- name: "dimensions"
- argType: INT64
- argIndex: 2
- }
opDeclarationType: CONFIGURABLE_OP_IMPL
}
opList {
name: "cumprod_bp"
+ argDescriptor {
+ name: "exclusive"
+ argType: INT64
+ }
+ argDescriptor {
+ name: "reverse"
+ argType: INT64
+ argIndex: 1
+ }
+ argDescriptor {
+ name: "dimensions"
+ argType: INT64
+ argIndex: 2
+ }
+ argDescriptor {
+ name: "dtype"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -5053,6 +5217,9 @@ opList {
argType: INPUT_TENSOR
argIndex: 2
}
+}
+opList {
+ name: "cumsum"
argDescriptor {
name: "exclusive"
argType: INT64
@@ -5067,13 +5234,6 @@ opList {
argType: INT64
argIndex: 2
}
- argDescriptor {
- name: "dtype"
- argType: DATA_TYPE
- }
-}
-opList {
- name: "cumsum"
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -5096,24 +5256,28 @@ opList {
argType: INPUT_TENSOR
argIndex: 1
}
- argDescriptor {
- name: "exclusive"
- argType: INT64
- }
- argDescriptor {
- name: "reverse"
- argType: INT64
- argIndex: 1
- }
- argDescriptor {
- name: "dimensions"
- argType: INT64
- argIndex: 2
- }
opDeclarationType: CONFIGURABLE_OP_IMPL
}
opList {
name: "cumsum_bp"
+ argDescriptor {
+ name: "exclusive"
+ argType: INT64
+ }
+ argDescriptor {
+ name: "reverse"
+ argType: INT64
+ argIndex: 1
+ }
+ argDescriptor {
+ name: "dimensions"
+ argType: INT64
+ argIndex: 2
+ }
+ argDescriptor {
+ name: "dtype"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -5141,27 +5305,13 @@ opList {
argType: INPUT_TENSOR
argIndex: 2
}
- argDescriptor {
- name: "exclusive"
- argType: INT64
- }
- argDescriptor {
- name: "reverse"
- argType: INT64
- argIndex: 1
- }
- argDescriptor {
- name: "dimensions"
- argType: INT64
- argIndex: 2
- }
- argDescriptor {
- name: "dtype"
- argType: DATA_TYPE
- }
}
opList {
name: "cyclic_rshift_bits"
+ argDescriptor {
+ name: "dataType"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -5179,14 +5329,14 @@ opList {
argType: INPUT_TENSOR
argIndex: 1
}
- argDescriptor {
- name: "dataType"
- argType: DATA_TYPE
- }
opDeclarationType: BROADCASTABLE_OP_IMPL
}
opList {
name: "cyclic_shift_bits"
+ argDescriptor {
+ name: "dataType"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -5204,14 +5354,14 @@ opList {
argType: INPUT_TENSOR
argIndex: 1
}
- argDescriptor {
- name: "dataType"
- argType: DATA_TYPE
- }
opDeclarationType: BROADCASTABLE_OP_IMPL
}
opList {
name: "decode_bitmap"
+ argDescriptor {
+ name: "dtype"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "updates"
argType: OUTPUT_TENSOR
@@ -5225,13 +5375,13 @@ opList {
argType: INPUT_TENSOR
argIndex: 1
}
+}
+opList {
+ name: "decode_threshold"
argDescriptor {
name: "dtype"
argType: DATA_TYPE
}
-}
-opList {
- name: "decode_threshold"
argDescriptor {
name: "updates"
argType: OUTPUT_TENSOR
@@ -5245,31 +5395,9 @@ opList {
argType: INPUT_TENSOR
argIndex: 1
}
- argDescriptor {
- name: "dtype"
- argType: DATA_TYPE
- }
}
opList {
name: "deconv2d"
- argDescriptor {
- name: "outputs"
- argType: OUTPUT_TENSOR
- }
- argDescriptor {
- name: "input"
- argType: INPUT_TENSOR
- }
- argDescriptor {
- name: "weights"
- argType: INPUT_TENSOR
- argIndex: 1
- }
- argDescriptor {
- name: "bias"
- argType: INPUT_TENSOR
- argIndex: 2
- }
argDescriptor {
name: "kH"
argType: INT64
@@ -5328,9 +5456,85 @@ opList {
name: "dtype"
argType: DATA_TYPE
}
+ argDescriptor {
+ name: "outputs"
+ argType: OUTPUT_TENSOR
+ }
+ argDescriptor {
+ name: "input"
+ argType: INPUT_TENSOR
+ }
+ argDescriptor {
+ name: "weights"
+ argType: INPUT_TENSOR
+ argIndex: 1
+ }
+ argDescriptor {
+ name: "bias"
+ argType: INPUT_TENSOR
+ argIndex: 2
+ }
}
opList {
name: "deconv2d_bp"
+ argDescriptor {
+ name: "kH"
+ argType: INT64
+ }
+ argDescriptor {
+ name: "kW"
+ argType: INT64
+ argIndex: 1
+ }
+ argDescriptor {
+ name: "sH"
+ argType: INT64
+ argIndex: 2
+ }
+ argDescriptor {
+ name: "sW"
+ argType: INT64
+ argIndex: 3
+ }
+ argDescriptor {
+ name: "pH"
+ argType: INT64
+ argIndex: 4
+ }
+ argDescriptor {
+ name: "pW"
+ argType: INT64
+ argIndex: 5
+ }
+ argDescriptor {
+ name: "dH"
+ argType: INT64
+ argIndex: 6
+ }
+ argDescriptor {
+ name: "dW"
+ argType: INT64
+ argIndex: 7
+ }
+ argDescriptor {
+ name: "isSameMode"
+ argType: INT64
+ argIndex: 8
+ }
+ argDescriptor {
+ name: "isNCHW"
+ argType: INT64
+ argIndex: 9
+ }
+ argDescriptor {
+ name: "wFormat"
+ argType: INT64
+ argIndex: 10
+ }
+ argDescriptor {
+ name: "dtype"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "gradI"
argType: OUTPUT_TENSOR
@@ -5364,6 +5568,9 @@ opList {
argType: INPUT_TENSOR
argIndex: 3
}
+}
+opList {
+ name: "deconv2d_tf"
argDescriptor {
name: "kH"
argType: INT64
@@ -5422,9 +5629,6 @@ opList {
name: "dtype"
argType: DATA_TYPE
}
-}
-opList {
- name: "deconv2d_tf"
argDescriptor {
name: "gradI"
argType: OUTPUT_TENSOR
@@ -5443,85 +5647,9 @@ opList {
argType: INPUT_TENSOR
argIndex: 2
}
- argDescriptor {
- name: "kH"
- argType: INT64
- }
- argDescriptor {
- name: "kW"
- argType: INT64
- argIndex: 1
- }
- argDescriptor {
- name: "sH"
- argType: INT64
- argIndex: 2
- }
- argDescriptor {
- name: "sW"
- argType: INT64
- argIndex: 3
- }
- argDescriptor {
- name: "pH"
- argType: INT64
- argIndex: 4
- }
- argDescriptor {
- name: "pW"
- argType: INT64
- argIndex: 5
- }
- argDescriptor {
- name: "dH"
- argType: INT64
- argIndex: 6
- }
- argDescriptor {
- name: "dW"
- argType: INT64
- argIndex: 7
- }
- argDescriptor {
- name: "isSameMode"
- argType: INT64
- argIndex: 8
- }
- argDescriptor {
- name: "isNCHW"
- argType: INT64
- argIndex: 9
- }
- argDescriptor {
- name: "wFormat"
- argType: INT64
- argIndex: 10
- }
- argDescriptor {
- name: "dtype"
- argType: DATA_TYPE
- }
}
opList {
name: "deconv3d"
- argDescriptor {
- name: "outputs"
- argType: OUTPUT_TENSOR
- }
- argDescriptor {
- name: "input"
- argType: INPUT_TENSOR
- }
- argDescriptor {
- name: "weights"
- argType: INPUT_TENSOR
- argIndex: 1
- }
- argDescriptor {
- name: "bias"
- argType: INPUT_TENSOR
- argIndex: 2
- }
argDescriptor {
name: "kD"
argType: INT64
@@ -5600,9 +5728,105 @@ opList {
name: "dtype"
argType: DATA_TYPE
}
+ argDescriptor {
+ name: "outputs"
+ argType: OUTPUT_TENSOR
+ }
+ argDescriptor {
+ name: "input"
+ argType: INPUT_TENSOR
+ }
+ argDescriptor {
+ name: "weights"
+ argType: INPUT_TENSOR
+ argIndex: 1
+ }
+ argDescriptor {
+ name: "bias"
+ argType: INPUT_TENSOR
+ argIndex: 2
+ }
}
opList {
name: "deconv3d_bp"
+ argDescriptor {
+ name: "kD"
+ argType: INT64
+ }
+ argDescriptor {
+ name: "kH"
+ argType: INT64
+ argIndex: 1
+ }
+ argDescriptor {
+ name: "kW"
+ argType: INT64
+ argIndex: 2
+ }
+ argDescriptor {
+ name: "sD"
+ argType: INT64
+ argIndex: 3
+ }
+ argDescriptor {
+ name: "sH"
+ argType: INT64
+ argIndex: 4
+ }
+ argDescriptor {
+ name: "sW"
+ argType: INT64
+ argIndex: 5
+ }
+ argDescriptor {
+ name: "pD"
+ argType: INT64
+ argIndex: 6
+ }
+ argDescriptor {
+ name: "pH"
+ argType: INT64
+ argIndex: 7
+ }
+ argDescriptor {
+ name: "pW"
+ argType: INT64
+ argIndex: 8
+ }
+ argDescriptor {
+ name: "dD"
+ argType: INT64
+ argIndex: 9
+ }
+ argDescriptor {
+ name: "dH"
+ argType: INT64
+ argIndex: 10
+ }
+ argDescriptor {
+ name: "dW"
+ argType: INT64
+ argIndex: 11
+ }
+ argDescriptor {
+ name: "isSameMode"
+ argType: INT64
+ argIndex: 12
+ }
+ argDescriptor {
+ name: "isNCDHW"
+ argType: INT64
+ argIndex: 13
+ }
+ argDescriptor {
+ name: "wFormat"
+ argType: INT64
+ argIndex: 14
+ }
+ argDescriptor {
+ name: "dtype"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "gradI"
argType: OUTPUT_TENSOR
@@ -5636,84 +5860,6 @@ opList {
argType: INPUT_TENSOR
argIndex: 3
}
- argDescriptor {
- name: "kD"
- argType: INT64
- }
- argDescriptor {
- name: "kH"
- argType: INT64
- argIndex: 1
- }
- argDescriptor {
- name: "kW"
- argType: INT64
- argIndex: 2
- }
- argDescriptor {
- name: "sD"
- argType: INT64
- argIndex: 3
- }
- argDescriptor {
- name: "sH"
- argType: INT64
- argIndex: 4
- }
- argDescriptor {
- name: "sW"
- argType: INT64
- argIndex: 5
- }
- argDescriptor {
- name: "pD"
- argType: INT64
- argIndex: 6
- }
- argDescriptor {
- name: "pH"
- argType: INT64
- argIndex: 7
- }
- argDescriptor {
- name: "pW"
- argType: INT64
- argIndex: 8
- }
- argDescriptor {
- name: "dD"
- argType: INT64
- argIndex: 9
- }
- argDescriptor {
- name: "dH"
- argType: INT64
- argIndex: 10
- }
- argDescriptor {
- name: "dW"
- argType: INT64
- argIndex: 11
- }
- argDescriptor {
- name: "isSameMode"
- argType: INT64
- argIndex: 12
- }
- argDescriptor {
- name: "isNCDHW"
- argType: INT64
- argIndex: 13
- }
- argDescriptor {
- name: "wFormat"
- argType: INT64
- argIndex: 14
- }
- argDescriptor {
- name: "dtype"
- argType: DATA_TYPE
- }
}
opList {
name: "deconv3d_tf"
@@ -5738,14 +5884,6 @@ opList {
}
opList {
name: "depth_to_space"
- argDescriptor {
- name: "outputs"
- argType: OUTPUT_TENSOR
- }
- argDescriptor {
- name: "input"
- argType: INPUT_TENSOR
- }
argDescriptor {
name: "block_size"
argType: INT64
@@ -5759,9 +5897,6 @@ opList {
name: "dtype"
argType: DATA_TYPE
}
-}
-opList {
- name: "depthwise_conv2d"
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -5770,16 +5905,9 @@ opList {
name: "input"
argType: INPUT_TENSOR
}
- argDescriptor {
- name: "weights"
- argType: INPUT_TENSOR
- argIndex: 1
- }
- argDescriptor {
- name: "bias"
- argType: INPUT_TENSOR
- argIndex: 2
- }
+}
+opList {
+ name: "depthwise_conv2d"
argDescriptor {
name: "kH"
argType: INT64
@@ -5838,9 +5966,85 @@ opList {
name: "dtype"
argType: DATA_TYPE
}
+ argDescriptor {
+ name: "outputs"
+ argType: OUTPUT_TENSOR
+ }
+ argDescriptor {
+ name: "input"
+ argType: INPUT_TENSOR
+ }
+ argDescriptor {
+ name: "weights"
+ argType: INPUT_TENSOR
+ argIndex: 1
+ }
+ argDescriptor {
+ name: "bias"
+ argType: INPUT_TENSOR
+ argIndex: 2
+ }
}
opList {
name: "depthwise_conv2d_bp"
+ argDescriptor {
+ name: "kH"
+ argType: INT64
+ }
+ argDescriptor {
+ name: "kW"
+ argType: INT64
+ argIndex: 1
+ }
+ argDescriptor {
+ name: "sH"
+ argType: INT64
+ argIndex: 2
+ }
+ argDescriptor {
+ name: "sW"
+ argType: INT64
+ argIndex: 3
+ }
+ argDescriptor {
+ name: "pH"
+ argType: INT64
+ argIndex: 4
+ }
+ argDescriptor {
+ name: "pW"
+ argType: INT64
+ argIndex: 5
+ }
+ argDescriptor {
+ name: "dH"
+ argType: INT64
+ argIndex: 6
+ }
+ argDescriptor {
+ name: "dW"
+ argType: INT64
+ argIndex: 7
+ }
+ argDescriptor {
+ name: "isSameMode"
+ argType: INT64
+ argIndex: 8
+ }
+ argDescriptor {
+ name: "isNCHW"
+ argType: INT64
+ argIndex: 9
+ }
+ argDescriptor {
+ name: "wFormat"
+ argType: INT64
+ argIndex: 10
+ }
+ argDescriptor {
+ name: "dtype"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "gradI"
argType: OUTPUT_TENSOR
@@ -5874,67 +6078,13 @@ opList {
argType: INPUT_TENSOR
argIndex: 3
}
- argDescriptor {
- name: "kH"
- argType: INT64
- }
- argDescriptor {
- name: "kW"
- argType: INT64
- argIndex: 1
- }
- argDescriptor {
- name: "sH"
- argType: INT64
- argIndex: 2
- }
- argDescriptor {
- name: "sW"
- argType: INT64
- argIndex: 3
- }
- argDescriptor {
- name: "pH"
- argType: INT64
- argIndex: 4
- }
- argDescriptor {
- name: "pW"
- argType: INT64
- argIndex: 5
- }
- argDescriptor {
- name: "dH"
- argType: INT64
- argIndex: 6
- }
- argDescriptor {
- name: "dW"
- argType: INT64
- argIndex: 7
- }
- argDescriptor {
- name: "isSameMode"
- argType: INT64
- argIndex: 8
- }
- argDescriptor {
- name: "isNCHW"
- argType: INT64
- argIndex: 9
- }
- argDescriptor {
- name: "wFormat"
- argType: INT64
- argIndex: 10
- }
+}
+opList {
+ name: "diag"
argDescriptor {
name: "dtype"
argType: DATA_TYPE
}
-}
-opList {
- name: "diag"
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -5947,13 +6097,13 @@ opList {
name: "input"
argType: INPUT_TENSOR
}
- argDescriptor {
- name: "dtype"
- argType: DATA_TYPE
- }
}
opList {
name: "diag_part"
+ argDescriptor {
+ name: "dtype"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -5966,10 +6116,6 @@ opList {
name: "input"
argType: INPUT_TENSOR
}
- argDescriptor {
- name: "dtype"
- argType: DATA_TYPE
- }
}
opList {
name: "digamma"
@@ -5985,6 +6131,24 @@ opList {
}
opList {
name: "dilation2d"
+ argDescriptor {
+ name: "isSameMode"
+ argType: INT64
+ }
+ argDescriptor {
+ name: "rates"
+ argType: INT64
+ argIndex: 1
+ }
+ argDescriptor {
+ name: "strides"
+ argType: INT64
+ argIndex: 2
+ }
+ argDescriptor {
+ name: "dtype"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -6017,31 +6181,9 @@ opList {
argType: INPUT_TENSOR
argIndex: 3
}
- argDescriptor {
- name: "isSameMode"
- argType: INT64
- }
- argDescriptor {
- name: "rates"
- argType: INT64
- argIndex: 1
- }
- argDescriptor {
- name: "strides"
- argType: INT64
- argIndex: 2
- }
- argDescriptor {
- name: "dtype"
- argType: DATA_TYPE
- }
}
opList {
name: "distribution_bernoulli"
- argDescriptor {
- name: "outputs"
- argType: OUTPUT_TENSOR
- }
argDescriptor {
name: "dataType"
argType: INT64
@@ -6051,6 +6193,10 @@ opList {
argType: INT64
argIndex: 1
}
+ argDescriptor {
+ name: "outputs"
+ argType: OUTPUT_TENSOR
+ }
argDescriptor {
name: "prob"
argType: DOUBLE
@@ -6059,10 +6205,6 @@ opList {
}
opList {
name: "distribution_binomial"
- argDescriptor {
- name: "outputs"
- argType: OUTPUT_TENSOR
- }
argDescriptor {
name: "trials"
argType: INT64
@@ -6077,6 +6219,10 @@ opList {
argType: INT64
argIndex: 2
}
+ argDescriptor {
+ name: "outputs"
+ argType: OUTPUT_TENSOR
+ }
argDescriptor {
name: "probability"
argType: DOUBLE
@@ -6085,18 +6231,14 @@ opList {
}
opList {
name: "distribution_binomial_ex"
- argDescriptor {
- name: "outputs"
- argType: OUTPUT_TENSOR
- }
- argDescriptor {
- name: "probabilities"
- argType: INPUT_TENSOR
- }
argDescriptor {
name: "trials"
argType: INT64
}
+ argDescriptor {
+ name: "outputs"
+ argType: OUTPUT_TENSOR
+ }
argDescriptor {
name: "probability"
argType: DOUBLE
@@ -6105,10 +6247,6 @@ opList {
}
opList {
name: "distribution_gaussian"
- argDescriptor {
- name: "outputs"
- argType: OUTPUT_TENSOR
- }
argDescriptor {
name: "dataType"
argType: INT64
@@ -6118,6 +6256,10 @@ opList {
argType: INT64
argIndex: 1
}
+ argDescriptor {
+ name: "outputs"
+ argType: OUTPUT_TENSOR
+ }
argDescriptor {
name: "mean"
argType: DOUBLE
@@ -6131,10 +6273,6 @@ opList {
}
opList {
name: "distribution_lognormal"
- argDescriptor {
- name: "outputs"
- argType: OUTPUT_TENSOR
- }
argDescriptor {
name: "dataType"
argType: INT64
@@ -6144,6 +6282,10 @@ opList {
argType: INT64
argIndex: 1
}
+ argDescriptor {
+ name: "outputs"
+ argType: OUTPUT_TENSOR
+ }
argDescriptor {
name: "mean"
argType: DOUBLE
@@ -6157,10 +6299,6 @@ opList {
}
opList {
name: "distribution_truncated"
- argDescriptor {
- name: "outputs"
- argType: OUTPUT_TENSOR
- }
argDescriptor {
name: "dataType"
argType: INT64
@@ -6170,6 +6308,10 @@ opList {
argType: INT64
argIndex: 1
}
+ argDescriptor {
+ name: "outputs"
+ argType: OUTPUT_TENSOR
+ }
argDescriptor {
name: "mean"
argType: DOUBLE
@@ -6183,10 +6325,6 @@ opList {
}
opList {
name: "distribution_uniform"
- argDescriptor {
- name: "outputs"
- argType: OUTPUT_TENSOR
- }
argDescriptor {
name: "dataType"
argType: INT64
@@ -6196,6 +6334,10 @@ opList {
argType: INT64
argIndex: 1
}
+ argDescriptor {
+ name: "outputs"
+ argType: OUTPUT_TENSOR
+ }
argDescriptor {
name: "from"
argType: DOUBLE
@@ -6226,6 +6368,10 @@ opList {
}
opList {
name: "divide"
+ argDescriptor {
+ name: "dataType"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -6243,14 +6389,14 @@ opList {
argType: INPUT_TENSOR
argIndex: 1
}
- argDescriptor {
- name: "dataType"
- argType: DATA_TYPE
- }
opDeclarationType: BROADCASTABLE_OP_IMPL
}
opList {
name: "divide_bp"
+ argDescriptor {
+ name: "dtype"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "gradX"
argType: OUTPUT_TENSOR
@@ -6278,10 +6424,6 @@ opList {
argType: INPUT_TENSOR
argIndex: 2
}
- argDescriptor {
- name: "dtype"
- argType: DATA_TYPE
- }
}
opList {
name: "divide_no_nan"
@@ -6302,6 +6444,10 @@ opList {
}
opList {
name: "dot"
+ argDescriptor {
+ name: "dimensions"
+ argType: INT64
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -6324,14 +6470,23 @@ opList {
argType: INPUT_TENSOR
argIndex: 1
}
- argDescriptor {
- name: "dimensions"
- argType: INT64
- }
opDeclarationType: LEGACY_XYZ
}
opList {
name: "dot_product_attention"
+ argDescriptor {
+ name: "normalization"
+ argType: INT64
+ }
+ argDescriptor {
+ name: "outputWeights"
+ argType: INT64
+ argIndex: 1
+ }
+ argDescriptor {
+ name: "dtype"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -6369,22 +6524,17 @@ opList {
argType: INPUT_TENSOR
argIndex: 3
}
+}
+opList {
+ name: "dot_product_attention_bp"
argDescriptor {
name: "normalization"
argType: INT64
}
- argDescriptor {
- name: "outputWeights"
- argType: INT64
- argIndex: 1
- }
argDescriptor {
name: "dtype"
argType: DATA_TYPE
}
-}
-opList {
- name: "dot_product_attention_bp"
argDescriptor {
name: "dLdq"
argType: OUTPUT_TENSOR
@@ -6427,14 +6577,6 @@ opList {
argType: INPUT_TENSOR
argIndex: 4
}
- argDescriptor {
- name: "normalization"
- argType: INT64
- }
- argDescriptor {
- name: "dtype"
- argType: DATA_TYPE
- }
}
opList {
name: "draw_bounding_boxes"
@@ -6460,10 +6602,18 @@ opList {
}
opList {
name: "dropout"
+ argDescriptor {
+ name: "seed"
+ argType: INT64
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
}
+ argDescriptor {
+ name: "probValue"
+ argType: DOUBLE
+ }
argDescriptor {
name: "input"
argType: INPUT_TENSOR
@@ -6473,22 +6623,22 @@ opList {
argType: INPUT_TENSOR
argIndex: 1
}
- argDescriptor {
- name: "seed"
- argType: INT64
- }
- argDescriptor {
- name: "probValue"
- argType: DOUBLE
- }
opDeclarationType: CONFIGURABLE_OP_IMPL
}
opList {
name: "dropout_bp"
+ argDescriptor {
+ name: "seed"
+ argType: INT64
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
}
+ argDescriptor {
+ name: "probValue"
+ argType: DOUBLE
+ }
argDescriptor {
name: "input"
argType: INPUT_TENSOR
@@ -6503,14 +6653,6 @@ opList {
argType: INPUT_TENSOR
argIndex: 2
}
- argDescriptor {
- name: "seed"
- argType: INT64
- }
- argDescriptor {
- name: "probValue"
- argType: DOUBLE
- }
opDeclarationType: CONFIGURABLE_OP_IMPL
}
opList {
@@ -6519,18 +6661,26 @@ opList {
name: "outputs"
argType: OUTPUT_TENSOR
}
- argDescriptor {
- name: "input"
- argType: INPUT_TENSOR
- }
argDescriptor {
name: "p"
argType: DOUBLE
}
+ argDescriptor {
+ name: "input"
+ argType: INPUT_TENSOR
+ }
opDeclarationType: LEGACY_XYZ
}
opList {
name: "dynamic_bidirectional_rnn"
+ argDescriptor {
+ name: "timeMajor"
+ argType: INT64
+ }
+ argDescriptor {
+ name: "dtype"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "hFW"
argType: OUTPUT_TENSOR
@@ -6594,17 +6744,17 @@ opList {
argType: INPUT_TENSOR
argIndex: 8
}
+}
+opList {
+ name: "dynamic_partition"
argDescriptor {
- name: "timeMajor"
+ name: "numPartitions"
argType: INT64
}
argDescriptor {
name: "dtype"
argType: DATA_TYPE
}
-}
-opList {
- name: "dynamic_partition"
argDescriptor {
name: "outputList"
argType: OUTPUT_TENSOR
@@ -6618,17 +6768,17 @@ opList {
argType: INPUT_TENSOR
argIndex: 1
}
+}
+opList {
+ name: "dynamic_partition_bp"
argDescriptor {
- name: "numPartitions"
+ name: "numPartition"
argType: INT64
}
argDescriptor {
name: "dtype"
argType: DATA_TYPE
}
-}
-opList {
- name: "dynamic_partition_bp"
argDescriptor {
name: "outputList"
argType: OUTPUT_TENSOR
@@ -6647,17 +6797,17 @@ opList {
argType: INPUT_TENSOR
argIndex: 2
}
+}
+opList {
+ name: "dynamic_rnn"
argDescriptor {
- name: "numPartition"
+ name: "timeMajor"
argType: INT64
}
argDescriptor {
name: "dtype"
argType: DATA_TYPE
}
-}
-opList {
- name: "dynamic_rnn"
argDescriptor {
name: "h"
argType: OUTPUT_TENSOR
@@ -6691,17 +6841,17 @@ opList {
argType: INPUT_TENSOR
argIndex: 4
}
+}
+opList {
+ name: "dynamic_stitch"
argDescriptor {
- name: "timeMajor"
+ name: "numPartitions"
argType: INT64
}
argDescriptor {
name: "dtype"
argType: DATA_TYPE
}
-}
-opList {
- name: "dynamic_stitch"
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -6715,14 +6865,6 @@ opList {
argType: INPUT_TENSOR
argIndex: 1
}
- argDescriptor {
- name: "numPartitions"
- argType: INT64
- }
- argDescriptor {
- name: "dtype"
- argType: DATA_TYPE
- }
}
opList {
name: "elu"
@@ -6730,14 +6872,14 @@ opList {
name: "outputs"
argType: OUTPUT_TENSOR
}
- argDescriptor {
- name: "input"
- argType: INPUT_TENSOR
- }
argDescriptor {
name: "alpha"
argType: DOUBLE
}
+ argDescriptor {
+ name: "input"
+ argType: INPUT_TENSOR
+ }
opDeclarationType: CONFIGURABLE_OP_IMPL
}
opList {
@@ -6746,6 +6888,10 @@ opList {
name: "outputs"
argType: OUTPUT_TENSOR
}
+ argDescriptor {
+ name: "alpha"
+ argType: DOUBLE
+ }
argDescriptor {
name: "input"
argType: INPUT_TENSOR
@@ -6755,14 +6901,18 @@ opList {
argType: INPUT_TENSOR
argIndex: 1
}
- argDescriptor {
- name: "alpha"
- argType: DOUBLE
- }
opDeclarationType: CONFIGURABLE_OP_IMPL
}
opList {
name: "embedding_lookup"
+ argDescriptor {
+ name: "partition_mode"
+ argType: INT64
+ }
+ argDescriptor {
+ name: "dtype"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -6776,17 +6926,13 @@ opList {
argType: INPUT_TENSOR
argIndex: 1
}
- argDescriptor {
- name: "partition_mode"
- argType: INT64
- }
+}
+opList {
+ name: "encode_bitmap"
argDescriptor {
name: "dtype"
argType: DATA_TYPE
}
-}
-opList {
- name: "encode_bitmap"
argDescriptor {
name: "encoded"
argType: OUTPUT_TENSOR
@@ -6796,6 +6942,10 @@ opList {
argType: OUTPUT_TENSOR
argIndex: 1
}
+ argDescriptor {
+ name: "threshold"
+ argType: DOUBLE
+ }
argDescriptor {
name: "input"
argType: INPUT_TENSOR
@@ -6810,17 +6960,17 @@ opList {
argType: INPUT_TENSOR
argIndex: 2
}
+}
+opList {
+ name: "encode_threshold"
+ argDescriptor {
+ name: "boundary"
+ argType: INT64
+ }
argDescriptor {
name: "dtype"
argType: DATA_TYPE
}
- argDescriptor {
- name: "threshold"
- argType: DOUBLE
- }
-}
-opList {
- name: "encode_threshold"
argDescriptor {
name: "updated"
argType: OUTPUT_TENSOR
@@ -6830,6 +6980,10 @@ opList {
argType: OUTPUT_TENSOR
argIndex: 1
}
+ argDescriptor {
+ name: "threshold"
+ argType: DOUBLE
+ }
argDescriptor {
name: "input"
argType: INPUT_TENSOR
@@ -6839,18 +6993,6 @@ opList {
argType: INPUT_TENSOR
argIndex: 1
}
- argDescriptor {
- name: "boundary"
- argType: INT64
- }
- argDescriptor {
- name: "dtype"
- argType: DATA_TYPE
- }
- argDescriptor {
- name: "threshold"
- argType: DOUBLE
- }
}
opList {
name: "enter"
@@ -6858,6 +7000,10 @@ opList {
name: "outputs"
argType: OUTPUT_TENSOR
}
+ argDescriptor {
+ name: "frameName"
+ argType: STRING
+ }
argDescriptor {
name: "isConstant"
argType: BOOL
@@ -6866,13 +7012,13 @@ opList {
name: "input"
argType: INPUT_TENSOR
}
- argDescriptor {
- name: "frameName"
- argType: STRING
- }
}
opList {
name: "entropy"
+ argDescriptor {
+ name: "dimensions"
+ argType: INT64
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -6890,21 +7036,17 @@ opList {
argType: INPUT_TENSOR
argIndex: 1
}
- argDescriptor {
- name: "dimensions"
- argType: INT64
- }
opDeclarationType: LEGACY_XYZ
}
opList {
name: "eps"
argDescriptor {
- name: "outputs"
- argType: OUTPUT_TENSOR
+ name: "dataType"
+ argType: DATA_TYPE
}
argDescriptor {
- name: "inPlace"
- argType: BOOL
+ name: "outputs"
+ argType: OUTPUT_TENSOR
}
argDescriptor {
name: "input"
@@ -6915,10 +7057,6 @@ opList {
argType: INPUT_TENSOR
argIndex: 1
}
- argDescriptor {
- name: "dataType"
- argType: DATA_TYPE
- }
opDeclarationType: LEGACY_XYZ
}
opList {
@@ -6939,6 +7077,10 @@ opList {
}
opList {
name: "equals"
+ argDescriptor {
+ name: "dataType"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -6956,10 +7098,6 @@ opList {
argType: INPUT_TENSOR
argIndex: 1
}
- argDescriptor {
- name: "dataType"
- argType: DATA_TYPE
- }
opDeclarationType: BROADCASTABLE_BOOL_OP_IMPL
}
opList {
@@ -6980,6 +7118,10 @@ opList {
}
opList {
name: "equals_with_eps"
+ argDescriptor {
+ name: "dimensions"
+ argType: INT64
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -6988,6 +7130,10 @@ opList {
name: "keepDims"
argType: BOOL
}
+ argDescriptor {
+ name: "eps"
+ argType: DOUBLE
+ }
argDescriptor {
name: "input"
argType: INPUT_TENSOR
@@ -6997,18 +7143,14 @@ opList {
argType: INPUT_TENSOR
argIndex: 1
}
- argDescriptor {
- name: "dimensions"
- argType: INT64
- }
- argDescriptor {
- name: "eps"
- argType: DOUBLE
- }
opDeclarationType: LEGACY_XYZ
}
opList {
name: "erf"
+ argDescriptor {
+ name: "dataType"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -7021,14 +7163,14 @@ opList {
name: "input"
argType: INPUT_TENSOR
}
- argDescriptor {
- name: "dataType"
- argType: DATA_TYPE
- }
opDeclarationType: LEGACY_XYZ
}
opList {
name: "erfc"
+ argDescriptor {
+ name: "dataType"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -7041,14 +7183,14 @@ opList {
name: "input"
argType: INPUT_TENSOR
}
- argDescriptor {
- name: "dataType"
- argType: DATA_TYPE
- }
opDeclarationType: LEGACY_XYZ
}
opList {
name: "euclidean"
+ argDescriptor {
+ name: "dimensions"
+ argType: INT64
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -7071,14 +7213,14 @@ opList {
argType: INPUT_TENSOR
argIndex: 1
}
- argDescriptor {
- name: "dimensions"
- argType: INT64
- }
opDeclarationType: LEGACY_XYZ
}
opList {
name: "evaluate_reduction_shape"
+ argDescriptor {
+ name: "dtype"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -7101,10 +7243,6 @@ opList {
argType: INPUT_TENSOR
argIndex: 1
}
- argDescriptor {
- name: "dtype"
- argType: DATA_TYPE
- }
}
opList {
name: "exit"
@@ -7112,17 +7250,21 @@ opList {
name: "outputs"
argType: OUTPUT_TENSOR
}
- argDescriptor {
- name: "input"
- argType: INPUT_TENSOR
- }
argDescriptor {
name: "frameName"
argType: STRING
}
+ argDescriptor {
+ name: "input"
+ argType: INPUT_TENSOR
+ }
}
opList {
name: "exp"
+ argDescriptor {
+ name: "dataType"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -7135,22 +7277,10 @@ opList {
name: "input"
argType: INPUT_TENSOR
}
- argDescriptor {
- name: "dataType"
- argType: DATA_TYPE
- }
opDeclarationType: LEGACY_XYZ
}
opList {
name: "expand_dims"
- argDescriptor {
- name: "outputs"
- argType: OUTPUT_TENSOR
- }
- argDescriptor {
- name: "input"
- argType: INPUT_TENSOR
- }
argDescriptor {
name: "dimensions"
argType: INT64
@@ -7159,9 +7289,6 @@ opList {
name: "dtype"
argType: DATA_TYPE
}
-}
-opList {
- name: "expm1"
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -7174,41 +7301,44 @@ opList {
name: "input"
argType: INPUT_TENSOR
}
+}
+opList {
+ name: "expm1"
argDescriptor {
name: "dataType"
argType: DATA_TYPE
}
- opDeclarationType: LEGACY_XYZ
-}
-opList {
- name: "expose"
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
}
argDescriptor {
- name: "input"
- argType: INPUT_TENSOR
- }
- argDescriptor {
- name: "dtype"
- argType: DATA_TYPE
- }
-}
-opList {
- name: "extract_image_patches"
- argDescriptor {
- name: "outputs"
- argType: OUTPUT_TENSOR
- }
- argDescriptor {
- name: "sameMode"
+ name: "inPlace"
argType: BOOL
}
argDescriptor {
name: "input"
argType: INPUT_TENSOR
}
+ opDeclarationType: LEGACY_XYZ
+}
+opList {
+ name: "expose"
+ argDescriptor {
+ name: "dtype"
+ argType: DATA_TYPE
+ }
+ argDescriptor {
+ name: "outputs"
+ argType: OUTPUT_TENSOR
+ }
+ argDescriptor {
+ name: "input"
+ argType: INPUT_TENSOR
+ }
+}
+opList {
+ name: "extract_image_patches"
argDescriptor {
name: "ksizeRows"
argType: INT64
@@ -7247,13 +7377,21 @@ opList {
name: "dtype"
argType: DATA_TYPE
}
-}
-opList {
- name: "eye"
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
}
+ argDescriptor {
+ name: "sameMode"
+ argType: BOOL
+ }
+ argDescriptor {
+ name: "input"
+ argType: INPUT_TENSOR
+ }
+}
+opList {
+ name: "eye"
argDescriptor {
name: "numRows"
argType: INT64
@@ -7277,13 +7415,30 @@ opList {
name: "dtype"
argType: DATA_TYPE
}
+ argDescriptor {
+ name: "outputs"
+ argType: OUTPUT_TENSOR
+ }
argDescriptor {
name: "dataType"
argType: DOUBLE
}
+ argDescriptor {
+ name: "numRows"
+ argType: INPUT_TENSOR
+ }
+ argDescriptor {
+ name: "numCols"
+ argType: INPUT_TENSOR
+ argIndex: 1
+ }
}
opList {
name: "fake_quant_with_min_max_args"
+ argDescriptor {
+ name: "numBits"
+ argType: INT64
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -7292,14 +7447,6 @@ opList {
name: "narrowRange"
argType: BOOL
}
- argDescriptor {
- name: "input"
- argType: INPUT_TENSOR
- }
- argDescriptor {
- name: "numBits"
- argType: INT64
- }
argDescriptor {
name: "min"
argType: DOUBLE
@@ -7309,9 +7456,17 @@ opList {
argType: DOUBLE
argIndex: 1
}
+ argDescriptor {
+ name: "input"
+ argType: INPUT_TENSOR
+ }
}
opList {
name: "fake_quant_with_min_max_vars"
+ argDescriptor {
+ name: "numBits"
+ argType: INT64
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -7320,24 +7475,6 @@ opList {
name: "narrowed"
argType: BOOL
}
- argDescriptor {
- name: "input"
- argType: INPUT_TENSOR
- }
- argDescriptor {
- name: "min"
- argType: INPUT_TENSOR
- argIndex: 1
- }
- argDescriptor {
- name: "max"
- argType: INPUT_TENSOR
- argIndex: 2
- }
- argDescriptor {
- name: "numBits"
- argType: INT64
- }
argDescriptor {
name: "m"
argType: DOUBLE
@@ -7347,10 +7484,28 @@ opList {
argType: DOUBLE
argIndex: 1
}
+ argDescriptor {
+ name: "input"
+ argType: INPUT_TENSOR
+ }
+ argDescriptor {
+ name: "min"
+ argType: INPUT_TENSOR
+ argIndex: 1
+ }
+ argDescriptor {
+ name: "max"
+ argType: INPUT_TENSOR
+ argIndex: 2
+ }
opDeclarationType: CONFIGURABLE_OP_IMPL
}
opList {
name: "fake_quant_with_min_max_vars_per_channel"
+ argDescriptor {
+ name: "numBits"
+ argType: INT64
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -7373,27 +7528,10 @@ opList {
argType: INPUT_TENSOR
argIndex: 2
}
- argDescriptor {
- name: "numBits"
- argType: INT64
- }
opDeclarationType: CONFIGURABLE_OP_IMPL
}
opList {
name: "fill"
- argDescriptor {
- name: "outputs"
- argType: OUTPUT_TENSOR
- }
- argDescriptor {
- name: "shape"
- argType: INPUT_TENSOR
- }
- argDescriptor {
- name: "outputs"
- argType: INPUT_TENSOR
- argIndex: 1
- }
argDescriptor {
name: "dtype"
argType: INT64
@@ -7402,10 +7540,23 @@ opList {
name: "dtype"
argType: DATA_TYPE
}
+ argDescriptor {
+ name: "outputs"
+ argType: OUTPUT_TENSOR
+ }
argDescriptor {
name: "value"
argType: DOUBLE
}
+ argDescriptor {
+ name: "shape"
+ argType: INPUT_TENSOR
+ }
+ argDescriptor {
+ name: "outputs"
+ argType: INPUT_TENSOR
+ argIndex: 1
+ }
}
opList {
name: "fill_as"
@@ -7421,6 +7572,10 @@ opList {
}
opList {
name: "firas_sparse"
+ argDescriptor {
+ name: "dtype"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -7429,13 +7584,13 @@ opList {
name: "input"
argType: INPUT_TENSOR
}
- argDescriptor {
- name: "dtype"
- argType: DATA_TYPE
- }
}
opList {
name: "first_index"
+ argDescriptor {
+ name: "dimensions"
+ argType: INT64
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -7444,14 +7599,6 @@ opList {
name: "keepDims"
argType: BOOL
}
- argDescriptor {
- name: "input"
- argType: INPUT_TENSOR
- }
- argDescriptor {
- name: "dimensions"
- argType: INT64
- }
argDescriptor {
name: "compare"
argType: DOUBLE
@@ -7461,18 +7608,14 @@ opList {
argType: DOUBLE
argIndex: 1
}
- opDeclarationType: LEGACY_XYZ
-}
-opList {
- name: "flatten"
- argDescriptor {
- name: "outputs"
- argType: OUTPUT_TENSOR
- }
argDescriptor {
name: "input"
argType: INPUT_TENSOR
}
+ opDeclarationType: LEGACY_XYZ
+}
+opList {
+ name: "flatten"
argDescriptor {
name: "order"
argType: INT64
@@ -7481,9 +7624,6 @@ opList {
name: "dtype"
argType: DATA_TYPE
}
-}
-opList {
- name: "flatten_2d"
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -7492,17 +7632,32 @@ opList {
name: "input"
argType: INPUT_TENSOR
}
+}
+opList {
+ name: "flatten_2d"
argDescriptor {
- name: "dimensions"
+ name: "flattenDimension"
argType: INT64
}
argDescriptor {
name: "dtype"
argType: DATA_TYPE
}
+ argDescriptor {
+ name: "outputs"
+ argType: OUTPUT_TENSOR
+ }
+ argDescriptor {
+ name: "input"
+ argType: INPUT_TENSOR
+ }
}
opList {
name: "floor"
+ argDescriptor {
+ name: "dataType"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -7515,14 +7670,14 @@ opList {
name: "input"
argType: INPUT_TENSOR
}
- argDescriptor {
- name: "dataType"
- argType: DATA_TYPE
- }
opDeclarationType: LEGACY_XYZ
}
opList {
name: "floordiv"
+ argDescriptor {
+ name: "dataType"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -7540,14 +7695,14 @@ opList {
argType: INPUT_TENSOR
argIndex: 1
}
- argDescriptor {
- name: "dataType"
- argType: DATA_TYPE
- }
opDeclarationType: BROADCASTABLE_OP_IMPL
}
opList {
name: "floordiv_bp"
+ argDescriptor {
+ name: "dtype"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "gradX"
argType: OUTPUT_TENSOR
@@ -7575,13 +7730,13 @@ opList {
argType: INPUT_TENSOR
argIndex: 2
}
- argDescriptor {
- name: "dtype"
- argType: DATA_TYPE
- }
}
opList {
name: "floormod"
+ argDescriptor {
+ name: "dataType"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -7599,14 +7754,14 @@ opList {
argType: INPUT_TENSOR
argIndex: 1
}
- argDescriptor {
- name: "dataType"
- argType: DATA_TYPE
- }
opDeclarationType: BROADCASTABLE_OP_IMPL
}
opList {
name: "floormod_bp"
+ argDescriptor {
+ name: "dtype"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "gradX"
argType: OUTPUT_TENSOR
@@ -7634,13 +7789,13 @@ opList {
argType: INPUT_TENSOR
argIndex: 2
}
- argDescriptor {
- name: "dtype"
- argType: DATA_TYPE
- }
}
opList {
name: "fmod"
+ argDescriptor {
+ name: "dataType"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -7658,10 +7813,6 @@ opList {
argType: INPUT_TENSOR
argIndex: 1
}
- argDescriptor {
- name: "dataType"
- argType: DATA_TYPE
- }
opDeclarationType: LEGACY_XYZ
}
opList {
@@ -7683,6 +7834,19 @@ opList {
}
opList {
name: "fused_batch_norm"
+ argDescriptor {
+ name: "dataFormat"
+ argType: INT64
+ }
+ argDescriptor {
+ name: "isTraining"
+ argType: INT64
+ argIndex: 1
+ }
+ argDescriptor {
+ name: "dtype"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "y"
argType: OUTPUT_TENSOR
@@ -7697,6 +7861,10 @@ opList {
argType: OUTPUT_TENSOR
argIndex: 2
}
+ argDescriptor {
+ name: "epsilon"
+ argType: DOUBLE
+ }
argDescriptor {
name: "input"
argType: INPUT_TENSOR
@@ -7726,26 +7894,17 @@ opList {
argType: INPUT_TENSOR
argIndex: 5
}
+}
+opList {
+ name: "gather"
argDescriptor {
- name: "dataFormat"
+ name: "dimensions"
argType: INT64
}
- argDescriptor {
- name: "isTraining"
- argType: INT64
- argIndex: 1
- }
argDescriptor {
name: "dtype"
argType: DATA_TYPE
}
- argDescriptor {
- name: "epsilon"
- argType: DOUBLE
- }
-}
-opList {
- name: "gather"
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -7768,17 +7927,13 @@ opList {
argType: INPUT_TENSOR
argIndex: 2
}
- argDescriptor {
- name: "dimensions"
- argType: INT64
- }
+}
+opList {
+ name: "gather_list"
argDescriptor {
name: "dtype"
argType: DATA_TYPE
}
-}
-opList {
- name: "gather_list"
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -7792,14 +7947,14 @@ opList {
argType: INPUT_TENSOR
argIndex: 1
}
- argDescriptor {
- name: "dtype"
- argType: DATA_TYPE
- }
opDeclarationType: LIST_OP_IMPL
}
opList {
name: "gather_nd"
+ argDescriptor {
+ name: "dtype"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -7817,13 +7972,13 @@ opList {
argType: INPUT_TENSOR
argIndex: 1
}
- argDescriptor {
- name: "dtype"
- argType: DATA_TYPE
- }
}
opList {
name: "gelu"
+ argDescriptor {
+ name: "dataType"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -7841,22 +7996,18 @@ opList {
name: "input"
argType: INPUT_TENSOR
}
- argDescriptor {
- name: "dataType"
- argType: DATA_TYPE
- }
opDeclarationType: LEGACY_XYZ
}
opList {
name: "get_seed"
- argDescriptor {
- name: "outputs"
- argType: OUTPUT_TENSOR
- }
argDescriptor {
name: "dtype"
argType: DATA_TYPE
}
+ argDescriptor {
+ name: "outputs"
+ argType: OUTPUT_TENSOR
+ }
}
opList {
name: "gradientbackwards"
@@ -7876,6 +8027,10 @@ opList {
}
opList {
name: "greater"
+ argDescriptor {
+ name: "dataType"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -7893,14 +8048,14 @@ opList {
argType: INPUT_TENSOR
argIndex: 1
}
- argDescriptor {
- name: "dataType"
- argType: DATA_TYPE
- }
opDeclarationType: BROADCASTABLE_BOOL_OP_IMPL
}
opList {
name: "greater_equal"
+ argDescriptor {
+ name: "dataType"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -7918,10 +8073,6 @@ opList {
argType: INPUT_TENSOR
argIndex: 1
}
- argDescriptor {
- name: "dataType"
- argType: DATA_TYPE
- }
opDeclarationType: BROADCASTABLE_BOOL_OP_IMPL
}
opList {
@@ -7975,6 +8126,10 @@ opList {
}
opList {
name: "gru"
+ argDescriptor {
+ name: "dtype"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "h"
argType: OUTPUT_TENSOR
@@ -8003,13 +8158,13 @@ opList {
argType: INPUT_TENSOR
argIndex: 4
}
+}
+opList {
+ name: "gruCell"
argDescriptor {
name: "dtype"
argType: DATA_TYPE
}
-}
-opList {
- name: "gruCell"
argDescriptor {
name: "r"
argType: OUTPUT_TENSOR
@@ -8058,13 +8213,13 @@ opList {
argType: INPUT_TENSOR
argIndex: 5
}
+}
+opList {
+ name: "gruCell_bp"
argDescriptor {
name: "dtype"
argType: DATA_TYPE
}
-}
-opList {
- name: "gruCell_bp"
argDescriptor {
name: "dLdx"
argType: OUTPUT_TENSOR
@@ -8143,13 +8298,13 @@ opList {
argType: INPUT_TENSOR
argIndex: 9
}
+}
+opList {
+ name: "gru_bp"
argDescriptor {
name: "dtype"
argType: DATA_TYPE
}
-}
-opList {
- name: "gru_bp"
argDescriptor {
name: "dLdx"
argType: OUTPUT_TENSOR
@@ -8203,13 +8358,13 @@ opList {
argType: INPUT_TENSOR
argIndex: 5
}
- argDescriptor {
- name: "dtype"
- argType: DATA_TYPE
- }
}
opList {
name: "hammingdistance"
+ argDescriptor {
+ name: "dimensions"
+ argType: INT64
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -8232,14 +8387,14 @@ opList {
argType: INPUT_TENSOR
argIndex: 1
}
- argDescriptor {
- name: "dimensions"
- argType: INT64
- }
opDeclarationType: LEGACY_XYZ
}
opList {
name: "hard_sigmoid"
+ argDescriptor {
+ name: "dataType"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -8252,14 +8407,14 @@ opList {
name: "input"
argType: INPUT_TENSOR
}
- argDescriptor {
- name: "dataType"
- argType: DATA_TYPE
- }
opDeclarationType: LEGACY_XYZ
}
opList {
name: "hard_sigmoidderivative"
+ argDescriptor {
+ name: "dataType"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -8272,10 +8427,6 @@ opList {
name: "input"
argType: INPUT_TENSOR
}
- argDescriptor {
- name: "dataType"
- argType: DATA_TYPE
- }
opDeclarationType: LEGACY_XYZ
}
opList {
@@ -8309,6 +8460,10 @@ opList {
}
opList {
name: "hardtanh"
+ argDescriptor {
+ name: "dataType"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -8321,10 +8476,6 @@ opList {
name: "input"
argType: INPUT_TENSOR
}
- argDescriptor {
- name: "dataType"
- argType: DATA_TYPE
- }
opDeclarationType: CONFIGURABLE_OP_IMPL
}
opList {
@@ -8346,6 +8497,10 @@ opList {
}
opList {
name: "hardtanhderivative"
+ argDescriptor {
+ name: "dataType"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -8358,14 +8513,14 @@ opList {
name: "input"
argType: INPUT_TENSOR
}
- argDescriptor {
- name: "dataType"
- argType: DATA_TYPE
- }
opDeclarationType: LEGACY_XYZ
}
opList {
name: "hashcode"
+ argDescriptor {
+ name: "dtype"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -8374,13 +8529,13 @@ opList {
name: "input"
argType: INPUT_TENSOR
}
- argDescriptor {
- name: "dtype"
- argType: DATA_TYPE
- }
}
opList {
name: "hasinf"
+ argDescriptor {
+ name: "dimensions"
+ argType: INT64
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -8393,14 +8548,14 @@ opList {
name: "input"
argType: INPUT_TENSOR
}
- argDescriptor {
- name: "dimensions"
- argType: INT64
- }
opDeclarationType: LEGACY_XYZ
}
opList {
name: "hasnan"
+ argDescriptor {
+ name: "dimensions"
+ argType: INT64
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -8413,14 +8568,18 @@ opList {
name: "input"
argType: INPUT_TENSOR
}
- argDescriptor {
- name: "dimensions"
- argType: INT64
- }
opDeclarationType: LEGACY_XYZ
}
opList {
name: "hinge_loss"
+ argDescriptor {
+ name: "reductionMode"
+ argType: INT64
+ }
+ argDescriptor {
+ name: "dtype"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -8439,6 +8598,9 @@ opList {
argType: INPUT_TENSOR
argIndex: 2
}
+}
+opList {
+ name: "hinge_loss_grad"
argDescriptor {
name: "reductionMode"
argType: INT64
@@ -8447,9 +8609,6 @@ opList {
name: "dtype"
argType: DATA_TYPE
}
-}
-opList {
- name: "hinge_loss_grad"
argDescriptor {
name: "dLdp"
argType: OUTPUT_TENSOR
@@ -8478,25 +8637,9 @@ opList {
argType: INPUT_TENSOR
argIndex: 2
}
- argDescriptor {
- name: "reductionMode"
- argType: INT64
- }
- argDescriptor {
- name: "dtype"
- argType: DATA_TYPE
- }
}
opList {
name: "histogram"
- argDescriptor {
- name: "outputs"
- argType: OUTPUT_TENSOR
- }
- argDescriptor {
- name: "input"
- argType: INPUT_TENSOR
- }
argDescriptor {
name: "numBins"
argType: INT64
@@ -8505,9 +8648,25 @@ opList {
name: "dtype"
argType: DATA_TYPE
}
+ argDescriptor {
+ name: "outputs"
+ argType: OUTPUT_TENSOR
+ }
+ argDescriptor {
+ name: "input"
+ argType: INPUT_TENSOR
+ }
}
opList {
name: "histogram_fixed_width"
+ argDescriptor {
+ name: "nbins"
+ argType: INT64
+ }
+ argDescriptor {
+ name: "dtype"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -8526,17 +8685,13 @@ opList {
argType: INPUT_TENSOR
argIndex: 2
}
- argDescriptor {
- name: "nbins"
- argType: INT64
- }
- argDescriptor {
- name: "dtype"
- argType: DATA_TYPE
- }
}
opList {
name: "hsv_to_rgb"
+ argDescriptor {
+ name: "dimC"
+ argType: INT64
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -8545,18 +8700,26 @@ opList {
name: "input"
argType: INPUT_TENSOR
}
- argDescriptor {
- name: "dimC"
- argType: INT64
- }
opDeclarationType: CONFIGURABLE_OP_IMPL
}
opList {
name: "huber_loss"
+ argDescriptor {
+ name: "reductionMode"
+ argType: INT64
+ }
+ argDescriptor {
+ name: "dtype"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
}
+ argDescriptor {
+ name: "delta"
+ argType: DOUBLE
+ }
argDescriptor {
name: "predictions"
argType: INPUT_TENSOR
@@ -8571,6 +8734,9 @@ opList {
argType: INPUT_TENSOR
argIndex: 2
}
+}
+opList {
+ name: "huber_loss_grad"
argDescriptor {
name: "reductionMode"
argType: INT64
@@ -8579,13 +8745,6 @@ opList {
name: "dtype"
argType: DATA_TYPE
}
- argDescriptor {
- name: "delta"
- argType: DOUBLE
- }
-}
-opList {
- name: "huber_loss_grad"
argDescriptor {
name: "dLdp"
argType: OUTPUT_TENSOR
@@ -8600,6 +8759,10 @@ opList {
argType: OUTPUT_TENSOR
argIndex: 2
}
+ argDescriptor {
+ name: "delta"
+ argType: DOUBLE
+ }
argDescriptor {
name: "predictions"
argType: INPUT_TENSOR
@@ -8614,21 +8777,13 @@ opList {
argType: INPUT_TENSOR
argIndex: 2
}
- argDescriptor {
- name: "reductionMode"
- argType: INT64
- }
- argDescriptor {
- name: "dtype"
- argType: DATA_TYPE
- }
- argDescriptor {
- name: "delta"
- argType: DOUBLE
- }
}
opList {
name: "identity"
+ argDescriptor {
+ name: "dataType"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -8641,10 +8796,6 @@ opList {
name: "input"
argType: INPUT_TENSOR
}
- argDescriptor {
- name: "dataType"
- argType: DATA_TYPE
- }
opDeclarationType: OP_IMPL
}
opList {
@@ -8666,6 +8817,10 @@ opList {
}
opList {
name: "identity_n"
+ argDescriptor {
+ name: "dtype"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -8674,10 +8829,6 @@ opList {
name: "input"
argType: INPUT_TENSOR
}
- argDescriptor {
- name: "dtype"
- argType: DATA_TYPE
- }
}
opList {
name: "igamma"
@@ -8715,19 +8866,6 @@ opList {
}
opList {
name: "im2col"
- argDescriptor {
- name: "outputs"
- argType: OUTPUT_TENSOR
- }
- argDescriptor {
- name: "input"
- argType: INPUT_TENSOR
- }
- argDescriptor {
- name: "inputArrays"
- argType: INPUT_TENSOR
- argIndex: 1
- }
argDescriptor {
name: "kernelHeight"
argType: INT64
@@ -8771,26 +8909,26 @@ opList {
name: "dtype"
argType: DATA_TYPE
}
- argDescriptor {
- name: "zeroPadVal"
- argType: DOUBLE
- }
-}
-opList {
- name: "im2col_bp"
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
}
+ argDescriptor {
+ name: "zeroPadVal"
+ argType: DOUBLE
+ }
argDescriptor {
name: "input"
argType: INPUT_TENSOR
}
argDescriptor {
- name: "gradAtOutput"
+ name: "inputArrays"
argType: INPUT_TENSOR
argIndex: 1
}
+}
+opList {
+ name: "im2col_bp"
argDescriptor {
name: "kernelHeight"
argType: INT64
@@ -8834,13 +8972,34 @@ opList {
name: "dtype"
argType: DATA_TYPE
}
+ argDescriptor {
+ name: "outputs"
+ argType: OUTPUT_TENSOR
+ }
argDescriptor {
name: "zeroPadVal"
argType: DOUBLE
}
+ argDescriptor {
+ name: "input"
+ argType: INPUT_TENSOR
+ }
+ argDescriptor {
+ name: "gradAtOutput"
+ argType: INPUT_TENSOR
+ argIndex: 1
+ }
}
opList {
name: "image_resize"
+ argDescriptor {
+ name: "method"
+ argType: INT64
+ }
+ argDescriptor {
+ name: "dtype"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -8863,17 +9022,17 @@ opList {
argType: INPUT_TENSOR
argIndex: 1
}
+}
+opList {
+ name: "in_top_k"
argDescriptor {
- name: "method"
+ name: "k"
argType: INT64
}
argDescriptor {
name: "dtype"
argType: DATA_TYPE
}
-}
-opList {
- name: "in_top_k"
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -8891,17 +9050,13 @@ opList {
argType: INPUT_TENSOR
argIndex: 1
}
- argDescriptor {
- name: "k"
- argType: INT64
- }
- argDescriptor {
- name: "dtype"
- argType: DATA_TYPE
- }
}
opList {
name: "invert_permutation"
+ argDescriptor {
+ name: "dataType"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -8914,10 +9069,6 @@ opList {
name: "input"
argType: INPUT_TENSOR
}
- argDescriptor {
- name: "dataType"
- argType: DATA_TYPE
- }
opDeclarationType: CONFIGURABLE_OP_IMPL
}
opList {
@@ -8970,6 +9121,10 @@ opList {
}
opList {
name: "isfinite"
+ argDescriptor {
+ name: "dataType"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -8982,14 +9137,14 @@ opList {
name: "input"
argType: INPUT_TENSOR
}
- argDescriptor {
- name: "dataType"
- argType: DATA_TYPE
- }
opDeclarationType: LEGACY_XYZ
}
opList {
name: "isinf"
+ argDescriptor {
+ name: "dataType"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -9002,14 +9157,14 @@ opList {
name: "input"
argType: INPUT_TENSOR
}
- argDescriptor {
- name: "dataType"
- argType: DATA_TYPE
- }
opDeclarationType: LEGACY_XYZ
}
opList {
name: "ismax"
+ argDescriptor {
+ name: "dimensions"
+ argType: INT64
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -9018,14 +9173,14 @@ opList {
name: "input"
argType: INPUT_TENSOR
}
- argDescriptor {
- name: "dimensions"
- argType: INT64
- }
opDeclarationType: CONFIGURABLE_OP_IMPL
}
opList {
name: "isnan"
+ argDescriptor {
+ name: "dataType"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -9038,14 +9193,14 @@ opList {
name: "input"
argType: INPUT_TENSOR
}
- argDescriptor {
- name: "dataType"
- argType: DATA_TYPE
- }
opDeclarationType: LEGACY_XYZ
}
opList {
name: "jaccarddistance"
+ argDescriptor {
+ name: "dimensions"
+ argType: INT64
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -9068,14 +9223,14 @@ opList {
argType: INPUT_TENSOR
argIndex: 1
}
- argDescriptor {
- name: "dimensions"
- argType: INT64
- }
opDeclarationType: LEGACY_XYZ
}
opList {
name: "knn_mindistance"
+ argDescriptor {
+ name: "dtype"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -9099,13 +9254,13 @@ opList {
argType: INPUT_TENSOR
argIndex: 3
}
+}
+opList {
+ name: "l2_loss"
argDescriptor {
name: "dtype"
argType: DATA_TYPE
}
-}
-opList {
- name: "l2_loss"
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -9114,13 +9269,13 @@ opList {
name: "input"
argType: INPUT_TENSOR
}
- argDescriptor {
- name: "dtype"
- argType: DATA_TYPE
- }
}
opList {
name: "last_index"
+ argDescriptor {
+ name: "dimensions"
+ argType: INT64
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -9129,14 +9284,6 @@ opList {
name: "keepDims"
argType: BOOL
}
- argDescriptor {
- name: "input"
- argType: INPUT_TENSOR
- }
- argDescriptor {
- name: "dimensions"
- argType: INT64
- }
argDescriptor {
name: "compare"
argType: DOUBLE
@@ -9146,16 +9293,24 @@ opList {
argType: DOUBLE
argIndex: 1
}
+ argDescriptor {
+ name: "input"
+ argType: INPUT_TENSOR
+ }
opDeclarationType: LEGACY_XYZ
}
opList {
name: "layer_norm"
+ argDescriptor {
+ name: "dimensions"
+ argType: INT64
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
}
argDescriptor {
- name: "channelsFirst"
+ name: "noBias"
argType: BOOL
}
argDescriptor {
@@ -9172,14 +9327,18 @@ opList {
argType: INPUT_TENSOR
argIndex: 2
}
- argDescriptor {
- name: "dimensions"
- argType: INT64
- }
opDeclarationType: CONFIGURABLE_OP_IMPL
}
opList {
name: "layer_norm_bp"
+ argDescriptor {
+ name: "dimensions"
+ argType: INT64
+ }
+ argDescriptor {
+ name: "dtype"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "dLdx"
argType: OUTPUT_TENSOR
@@ -9195,7 +9354,7 @@ opList {
argIndex: 2
}
argDescriptor {
- name: "channelsFirst"
+ name: "noBias"
argType: BOOL
}
argDescriptor {
@@ -9232,14 +9391,6 @@ opList {
argType: INPUT_TENSOR
argIndex: 6
}
- argDescriptor {
- name: "dimensions"
- argType: INT64
- }
- argDescriptor {
- name: "dtype"
- argType: DATA_TYPE
- }
}
opList {
name: "leakyrelu"
@@ -9248,17 +9399,13 @@ opList {
argType: OUTPUT_TENSOR
}
argDescriptor {
- name: "inPlace"
- argType: BOOL
+ name: "alpha"
+ argType: DOUBLE
}
argDescriptor {
name: "input"
argType: INPUT_TENSOR
}
- argDescriptor {
- name: "alpha"
- argType: DOUBLE
- }
opDeclarationType: LEGACY_XYZ
}
opList {
@@ -9268,21 +9415,21 @@ opList {
argType: OUTPUT_TENSOR
}
argDescriptor {
- name: "inPlace"
- argType: BOOL
+ name: "alpha"
+ argType: DOUBLE
}
argDescriptor {
name: "input"
argType: INPUT_TENSOR
}
- argDescriptor {
- name: "alpha"
- argType: DOUBLE
- }
opDeclarationType: LEGACY_XYZ
}
opList {
name: "less"
+ argDescriptor {
+ name: "dataType"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -9300,14 +9447,14 @@ opList {
argType: INPUT_TENSOR
argIndex: 1
}
- argDescriptor {
- name: "dataType"
- argType: DATA_TYPE
- }
opDeclarationType: BROADCASTABLE_BOOL_OP_IMPL
}
opList {
name: "less_equal"
+ argDescriptor {
+ name: "dataType"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -9325,10 +9472,6 @@ opList {
argType: INPUT_TENSOR
argIndex: 1
}
- argDescriptor {
- name: "dataType"
- argType: DATA_TYPE
- }
opDeclarationType: BROADCASTABLE_BOOL_OP_IMPL
}
opList {
@@ -9377,10 +9520,27 @@ opList {
}
opList {
name: "lin_space"
+ argDescriptor {
+ name: "dataType"
+ argType: INT64
+ }
+ argDescriptor {
+ name: "dataType"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
}
+ argDescriptor {
+ name: "start"
+ argType: DOUBLE
+ }
+ argDescriptor {
+ name: "stop"
+ argType: DOUBLE
+ argIndex: 1
+ }
argDescriptor {
name: "start"
argType: INPUT_TENSOR
@@ -9395,45 +9555,23 @@ opList {
argType: INPUT_TENSOR
argIndex: 2
}
- argDescriptor {
- name: "dataType"
- argType: INT64
- }
- argDescriptor {
- name: "dataType"
- argType: DATA_TYPE
- }
- argDescriptor {
- name: "start"
- argType: DOUBLE
- }
- argDescriptor {
- name: "stop"
- argType: DOUBLE
- argIndex: 1
- }
}
opList {
name: "linspace_random"
- argDescriptor {
- name: "outputs"
- argType: OUTPUT_TENSOR
- }
argDescriptor {
name: "length"
argType: INT64
}
argDescriptor {
- name: "dataType"
- argType: INT64
- argIndex: 1
+ name: "outputs"
+ argType: OUTPUT_TENSOR
}
argDescriptor {
name: "from"
argType: DOUBLE
}
argDescriptor {
- name: "step"
+ name: "to"
argType: DOUBLE
argIndex: 1
}
@@ -9441,6 +9579,10 @@ opList {
}
opList {
name: "listdiff"
+ argDescriptor {
+ name: "dtype"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "output1"
argType: OUTPUT_TENSOR
@@ -9459,13 +9601,13 @@ opList {
argType: INPUT_TENSOR
argIndex: 1
}
- argDescriptor {
- name: "dtype"
- argType: DATA_TYPE
- }
}
opList {
name: "log"
+ argDescriptor {
+ name: "dataType"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -9478,14 +9620,14 @@ opList {
name: "input"
argType: INPUT_TENSOR
}
- argDescriptor {
- name: "dataType"
- argType: DATA_TYPE
- }
opDeclarationType: LEGACY_XYZ
}
opList {
name: "log1p"
+ argDescriptor {
+ name: "dataType"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -9498,18 +9640,26 @@ opList {
name: "input"
argType: INPUT_TENSOR
}
- argDescriptor {
- name: "dataType"
- argType: DATA_TYPE
- }
opDeclarationType: LEGACY_XYZ
}
opList {
name: "log_loss"
+ argDescriptor {
+ name: "reductionMode"
+ argType: INT64
+ }
+ argDescriptor {
+ name: "dtype"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
}
+ argDescriptor {
+ name: "epsilon"
+ argType: DOUBLE
+ }
argDescriptor {
name: "predictions"
argType: INPUT_TENSOR
@@ -9524,6 +9674,9 @@ opList {
argType: INPUT_TENSOR
argIndex: 2
}
+}
+opList {
+ name: "log_loss_grad"
argDescriptor {
name: "reductionMode"
argType: INT64
@@ -9532,13 +9685,6 @@ opList {
name: "dtype"
argType: DATA_TYPE
}
- argDescriptor {
- name: "epsilon"
- argType: DOUBLE
- }
-}
-opList {
- name: "log_loss_grad"
argDescriptor {
name: "dLdp"
argType: OUTPUT_TENSOR
@@ -9553,6 +9699,10 @@ opList {
argType: OUTPUT_TENSOR
argIndex: 2
}
+ argDescriptor {
+ name: "epsilon"
+ argType: DOUBLE
+ }
argDescriptor {
name: "predictions"
argType: INPUT_TENSOR
@@ -9567,21 +9717,13 @@ opList {
argType: INPUT_TENSOR
argIndex: 2
}
- argDescriptor {
- name: "reductionMode"
- argType: INT64
- }
+}
+opList {
+ name: "log_matrix_determinant"
argDescriptor {
name: "dtype"
argType: DATA_TYPE
}
- argDescriptor {
- name: "epsilon"
- argType: DOUBLE
- }
-}
-opList {
- name: "log_matrix_determinant"
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -9594,13 +9736,17 @@ opList {
name: "input"
argType: INPUT_TENSOR
}
+}
+opList {
+ name: "log_poisson_loss"
+ argDescriptor {
+ name: "reductionMode"
+ argType: INT64
+ }
argDescriptor {
name: "dtype"
argType: DATA_TYPE
}
-}
-opList {
- name: "log_poisson_loss"
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -9623,6 +9769,9 @@ opList {
argType: INPUT_TENSOR
argIndex: 2
}
+}
+opList {
+ name: "log_poisson_loss_grad"
argDescriptor {
name: "reductionMode"
argType: INT64
@@ -9631,9 +9780,6 @@ opList {
name: "dtype"
argType: DATA_TYPE
}
-}
-opList {
- name: "log_poisson_loss_grad"
argDescriptor {
name: "dLdp"
argType: OUTPUT_TENSOR
@@ -9666,17 +9812,13 @@ opList {
argType: INPUT_TENSOR
argIndex: 2
}
- argDescriptor {
- name: "reductionMode"
- argType: INT64
- }
- argDescriptor {
- name: "dtype"
- argType: DATA_TYPE
- }
}
opList {
name: "log_softmax"
+ argDescriptor {
+ name: "dimension"
+ argType: INT64
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -9685,14 +9827,14 @@ opList {
name: "input"
argType: INPUT_TENSOR
}
- argDescriptor {
- name: "dimension"
- argType: INT64
- }
opDeclarationType: CONFIGURABLE_OP_IMPL
}
opList {
name: "log_softmax_bp"
+ argDescriptor {
+ name: "dimension"
+ argType: INT64
+ }
argDescriptor {
name: "gradI"
argType: OUTPUT_TENSOR
@@ -9706,10 +9848,6 @@ opList {
argType: INPUT_TENSOR
argIndex: 1
}
- argDescriptor {
- name: "dimension"
- argType: INT64
- }
opDeclarationType: CONFIGURABLE_OP_IMPL
}
opList {
@@ -9718,18 +9856,22 @@ opList {
name: "outputs"
argType: OUTPUT_TENSOR
}
- argDescriptor {
- name: "input"
- argType: INPUT_TENSOR
- }
argDescriptor {
name: "base"
argType: DOUBLE
}
+ argDescriptor {
+ name: "input"
+ argType: INPUT_TENSOR
+ }
opDeclarationType: LEGACY_XYZ
}
opList {
name: "logdet"
+ argDescriptor {
+ name: "dtype"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -9738,13 +9880,13 @@ opList {
name: "input"
argType: INPUT_TENSOR
}
- argDescriptor {
- name: "dtype"
- argType: DATA_TYPE
- }
}
opList {
name: "logentropy"
+ argDescriptor {
+ name: "dimensions"
+ argType: INT64
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -9762,14 +9904,14 @@ opList {
argType: INPUT_TENSOR
argIndex: 1
}
- argDescriptor {
- name: "dimensions"
- argType: INT64
- }
opDeclarationType: LEGACY_XYZ
}
opList {
name: "logsigmoid"
+ argDescriptor {
+ name: "dataType"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -9782,10 +9924,6 @@ opList {
name: "input"
argType: INPUT_TENSOR
}
- argDescriptor {
- name: "dataType"
- argType: DATA_TYPE
- }
opDeclarationType: LEGACY_XYZ
}
opList {
@@ -9794,14 +9932,14 @@ opList {
name: "outputs"
argType: OUTPUT_TENSOR
}
- argDescriptor {
- name: "input"
- argType: INPUT_TENSOR
- }
argDescriptor {
name: "frameName"
argType: STRING
}
+ argDescriptor {
+ name: "input"
+ argType: INPUT_TENSOR
+ }
}
opList {
name: "lrelu"
@@ -9809,14 +9947,14 @@ opList {
name: "outputs"
argType: OUTPUT_TENSOR
}
- argDescriptor {
- name: "input"
- argType: INPUT_TENSOR
- }
argDescriptor {
name: "alpha"
argType: DOUBLE
}
+ argDescriptor {
+ name: "input"
+ argType: INPUT_TENSOR
+ }
opDeclarationType: CONFIGURABLE_OP_IMPL
}
opList {
@@ -9825,6 +9963,10 @@ opList {
name: "outputs"
argType: OUTPUT_TENSOR
}
+ argDescriptor {
+ name: "alpha"
+ argType: DOUBLE
+ }
argDescriptor {
name: "input"
argType: INPUT_TENSOR
@@ -9834,14 +9976,14 @@ opList {
argType: INPUT_TENSOR
argIndex: 1
}
- argDescriptor {
- name: "alpha"
- argType: DOUBLE
- }
opDeclarationType: CONFIGURABLE_OP_IMPL
}
opList {
name: "lrn"
+ argDescriptor {
+ name: "depth"
+ argType: INT64
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -9850,14 +9992,6 @@ opList {
name: "inPlace"
argType: BOOL
}
- argDescriptor {
- name: "input"
- argType: INPUT_TENSOR
- }
- argDescriptor {
- name: "depth"
- argType: INT64
- }
argDescriptor {
name: "bias"
argType: DOUBLE
@@ -9872,10 +10006,18 @@ opList {
argType: DOUBLE
argIndex: 2
}
+ argDescriptor {
+ name: "input"
+ argType: INPUT_TENSOR
+ }
opDeclarationType: CONFIGURABLE_OP_IMPL
}
opList {
name: "lrn_bp"
+ argDescriptor {
+ name: "depth"
+ argType: INT64
+ }
argDescriptor {
name: "gradI"
argType: OUTPUT_TENSOR
@@ -9884,6 +10026,20 @@ opList {
name: "inPlace"
argType: BOOL
}
+ argDescriptor {
+ name: "bias"
+ argType: DOUBLE
+ }
+ argDescriptor {
+ name: "alpha"
+ argType: DOUBLE
+ argIndex: 1
+ }
+ argDescriptor {
+ name: "beta"
+ argType: DOUBLE
+ argIndex: 2
+ }
argDescriptor {
name: "input"
argType: INPUT_TENSOR
@@ -9893,28 +10049,23 @@ opList {
argType: INPUT_TENSOR
argIndex: 1
}
- argDescriptor {
- name: "depth"
- argType: INT64
- }
- argDescriptor {
- name: "bias"
- argType: DOUBLE
- }
- argDescriptor {
- name: "alpha"
- argType: DOUBLE
- argIndex: 1
- }
- argDescriptor {
- name: "beta"
- argType: DOUBLE
- argIndex: 2
- }
opDeclarationType: CONFIGURABLE_OP_IMPL
}
opList {
name: "lstm"
+ argDescriptor {
+ name: "peephole"
+ argType: INT64
+ }
+ argDescriptor {
+ name: "projection"
+ argType: INT64
+ argIndex: 1
+ }
+ argDescriptor {
+ name: "dtype"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "h"
argType: OUTPUT_TENSOR
@@ -9924,6 +10075,20 @@ opList {
argType: OUTPUT_TENSOR
argIndex: 1
}
+ argDescriptor {
+ name: "clippingCellValue"
+ argType: DOUBLE
+ }
+ argDescriptor {
+ name: "clippingProjValue"
+ argType: DOUBLE
+ argIndex: 1
+ }
+ argDescriptor {
+ name: "forgetBias"
+ argType: DOUBLE
+ argIndex: 2
+ }
argDescriptor {
name: "input"
argType: INPUT_TENSOR
@@ -9963,12 +10128,15 @@ opList {
argType: INPUT_TENSOR
argIndex: 7
}
+}
+opList {
+ name: "lstmBlock"
argDescriptor {
name: "peephole"
argType: INT64
}
argDescriptor {
- name: "projection"
+ name: "dataFormat"
argType: INT64
argIndex: 1
}
@@ -9976,23 +10144,6 @@ opList {
name: "dtype"
argType: DATA_TYPE
}
- argDescriptor {
- name: "clippingCellValue"
- argType: DOUBLE
- }
- argDescriptor {
- name: "clippingProjValue"
- argType: DOUBLE
- argIndex: 1
- }
- argDescriptor {
- name: "forgetBias"
- argType: DOUBLE
- argIndex: 2
- }
-}
-opList {
- name: "lstmBlock"
argDescriptor {
name: "i"
argType: OUTPUT_TENSOR
@@ -10027,6 +10178,15 @@ opList {
argType: OUTPUT_TENSOR
argIndex: 6
}
+ argDescriptor {
+ name: "forgetBias"
+ argType: DOUBLE
+ }
+ argDescriptor {
+ name: "clippingCellValue"
+ argType: DOUBLE
+ argIndex: 1
+ }
argDescriptor {
name: "maxTSLength"
argType: INPUT_TENSOR
@@ -10071,31 +10231,17 @@ opList {
argType: INPUT_TENSOR
argIndex: 8
}
+}
+opList {
+ name: "lstmBlockCell"
argDescriptor {
name: "peephole"
argType: INT64
}
- argDescriptor {
- name: "dataFormat"
- argType: INT64
- argIndex: 1
- }
argDescriptor {
name: "dtype"
argType: DATA_TYPE
}
- argDescriptor {
- name: "forgetBias"
- argType: DOUBLE
- }
- argDescriptor {
- name: "clippingCellValue"
- argType: DOUBLE
- argIndex: 1
- }
-}
-opList {
- name: "lstmBlockCell"
argDescriptor {
name: "i"
argType: OUTPUT_TENSOR
@@ -10130,6 +10276,15 @@ opList {
argType: OUTPUT_TENSOR
argIndex: 6
}
+ argDescriptor {
+ name: "forgetBias"
+ argType: DOUBLE
+ }
+ argDescriptor {
+ name: "clippingCellValue"
+ argType: DOUBLE
+ argIndex: 1
+ }
argDescriptor {
name: "xt"
argType: INPUT_TENSOR
@@ -10169,26 +10324,22 @@ opList {
argType: INPUT_TENSOR
argIndex: 7
}
+}
+opList {
+ name: "lstmCell"
argDescriptor {
name: "peephole"
argType: INT64
}
+ argDescriptor {
+ name: "projection"
+ argType: INT64
+ argIndex: 1
+ }
argDescriptor {
name: "dtype"
argType: DATA_TYPE
}
- argDescriptor {
- name: "forgetBias"
- argType: DOUBLE
- }
- argDescriptor {
- name: "clippingCellValue"
- argType: DOUBLE
- argIndex: 1
- }
-}
-opList {
- name: "lstmCell"
argDescriptor {
name: "ht"
argType: OUTPUT_TENSOR
@@ -10198,6 +10349,20 @@ opList {
argType: OUTPUT_TENSOR
argIndex: 1
}
+ argDescriptor {
+ name: "clippingCellValue"
+ argType: DOUBLE
+ }
+ argDescriptor {
+ name: "clippingProjValue"
+ argType: DOUBLE
+ argIndex: 1
+ }
+ argDescriptor {
+ name: "forgetBias"
+ argType: DOUBLE
+ argIndex: 2
+ }
argDescriptor {
name: "xt"
argType: INPUT_TENSOR
@@ -10237,36 +10402,37 @@ opList {
argType: INPUT_TENSOR
argIndex: 7
}
+}
+opList {
+ name: "lstmLayer"
argDescriptor {
- name: "peephole"
+ name: "dataFormat"
argType: INT64
}
argDescriptor {
- name: "projection"
+ name: "directionMode"
argType: INT64
argIndex: 1
}
+ argDescriptor {
+ name: "gateAct"
+ argType: INT64
+ argIndex: 2
+ }
+ argDescriptor {
+ name: "cellAct"
+ argType: INT64
+ argIndex: 3
+ }
+ argDescriptor {
+ name: "outAct"
+ argType: INT64
+ argIndex: 4
+ }
argDescriptor {
name: "dtype"
argType: DATA_TYPE
}
- argDescriptor {
- name: "clippingCellValue"
- argType: DOUBLE
- }
- argDescriptor {
- name: "clippingProjValue"
- argType: DOUBLE
- argIndex: 1
- }
- argDescriptor {
- name: "forgetBias"
- argType: DOUBLE
- argIndex: 2
- }
-}
-opList {
- name: "lstmLayer"
argDescriptor {
name: "h"
argType: OUTPUT_TENSOR
@@ -10320,6 +10486,40 @@ opList {
argType: BOOL
argIndex: 7
}
+ argDescriptor {
+ name: "cellClip"
+ argType: DOUBLE
+ }
+ argDescriptor {
+ name: "gateAlpha"
+ argType: DOUBLE
+ argIndex: 1
+ }
+ argDescriptor {
+ name: "gateBeta"
+ argType: DOUBLE
+ argIndex: 2
+ }
+ argDescriptor {
+ name: "cellAlpha"
+ argType: DOUBLE
+ argIndex: 3
+ }
+ argDescriptor {
+ name: "cellBeta"
+ argType: DOUBLE
+ argIndex: 4
+ }
+ argDescriptor {
+ name: "outAlpha"
+ argType: DOUBLE
+ argIndex: 5
+ }
+ argDescriptor {
+ name: "outBeta"
+ argType: DOUBLE
+ argIndex: 6
+ }
argDescriptor {
name: "input"
argType: INPUT_TENSOR
@@ -10359,34 +10559,45 @@ opList {
argType: INPUT_TENSOR
argIndex: 7
}
- argDescriptor {
- name: "dataFormat"
- argType: INT64
- }
- argDescriptor {
- name: "directionMode"
- argType: INT64
- argIndex: 1
- }
+}
+opList {
+ name: "lstmLayerCell"
argDescriptor {
name: "gateAct"
argType: INT64
- argIndex: 2
}
argDescriptor {
name: "cellAct"
argType: INT64
- argIndex: 3
+ argIndex: 1
}
argDescriptor {
name: "outAct"
argType: INT64
- argIndex: 4
+ argIndex: 2
}
argDescriptor {
name: "dtype"
argType: DATA_TYPE
}
+ argDescriptor {
+ name: "h"
+ argType: OUTPUT_TENSOR
+ }
+ argDescriptor {
+ name: "c"
+ argType: OUTPUT_TENSOR
+ argIndex: 1
+ }
+ argDescriptor {
+ name: "hasBiases"
+ argType: BOOL
+ }
+ argDescriptor {
+ name: "hasPH"
+ argType: BOOL
+ argIndex: 1
+ }
argDescriptor {
name: "cellClip"
argType: DOUBLE
@@ -10421,27 +10632,6 @@ opList {
argType: DOUBLE
argIndex: 6
}
-}
-opList {
- name: "lstmLayerCell"
- argDescriptor {
- name: "h"
- argType: OUTPUT_TENSOR
- }
- argDescriptor {
- name: "c"
- argType: OUTPUT_TENSOR
- argIndex: 1
- }
- argDescriptor {
- name: "hasBiases"
- argType: BOOL
- }
- argDescriptor {
- name: "hasPH"
- argType: BOOL
- argIndex: 1
- }
argDescriptor {
name: "input"
argType: INPUT_TENSOR
@@ -10476,6 +10666,9 @@ opList {
argType: INPUT_TENSOR
argIndex: 6
}
+}
+opList {
+ name: "lstmLayerCellBp"
argDescriptor {
name: "gateAct"
argType: INT64
@@ -10494,43 +10687,6 @@ opList {
name: "dtype"
argType: DATA_TYPE
}
- argDescriptor {
- name: "cellClip"
- argType: DOUBLE
- }
- argDescriptor {
- name: "gateAlpha"
- argType: DOUBLE
- argIndex: 1
- }
- argDescriptor {
- name: "gateBeta"
- argType: DOUBLE
- argIndex: 2
- }
- argDescriptor {
- name: "cellAlpha"
- argType: DOUBLE
- argIndex: 3
- }
- argDescriptor {
- name: "cellBeta"
- argType: DOUBLE
- argIndex: 4
- }
- argDescriptor {
- name: "outAlpha"
- argType: DOUBLE
- argIndex: 5
- }
- argDescriptor {
- name: "outBeta"
- argType: DOUBLE
- argIndex: 6
- }
-}
-opList {
- name: "lstmLayerCellBp"
argDescriptor {
name: "dLdx"
argType: OUTPUT_TENSOR
@@ -10574,6 +10730,40 @@ opList {
argType: BOOL
argIndex: 1
}
+ argDescriptor {
+ name: "cellClip"
+ argType: DOUBLE
+ }
+ argDescriptor {
+ name: "gateAlpha"
+ argType: DOUBLE
+ argIndex: 1
+ }
+ argDescriptor {
+ name: "gateBeta"
+ argType: DOUBLE
+ argIndex: 2
+ }
+ argDescriptor {
+ name: "cellAlpha"
+ argType: DOUBLE
+ argIndex: 3
+ }
+ argDescriptor {
+ name: "cellBeta"
+ argType: DOUBLE
+ argIndex: 4
+ }
+ argDescriptor {
+ name: "outAlpha"
+ argType: DOUBLE
+ argIndex: 5
+ }
+ argDescriptor {
+ name: "outBeta"
+ argType: DOUBLE
+ argIndex: 6
+ }
argDescriptor {
name: "input"
argType: INPUT_TENSOR
@@ -10613,61 +10803,37 @@ opList {
argType: INPUT_TENSOR
argIndex: 7
}
+}
+opList {
+ name: "lstmLayer_bp"
argDescriptor {
- name: "gateAct"
+ name: "dataFormat"
argType: INT64
}
argDescriptor {
- name: "cellAct"
+ name: "directionMode"
argType: INT64
argIndex: 1
}
argDescriptor {
- name: "outAct"
+ name: "gateAct"
argType: INT64
argIndex: 2
}
+ argDescriptor {
+ name: "cellAct"
+ argType: INT64
+ argIndex: 3
+ }
+ argDescriptor {
+ name: "outAct"
+ argType: INT64
+ argIndex: 4
+ }
argDescriptor {
name: "dtype"
argType: DATA_TYPE
}
- argDescriptor {
- name: "cellClip"
- argType: DOUBLE
- }
- argDescriptor {
- name: "gateAlpha"
- argType: DOUBLE
- argIndex: 1
- }
- argDescriptor {
- name: "gateBeta"
- argType: DOUBLE
- argIndex: 2
- }
- argDescriptor {
- name: "cellAlpha"
- argType: DOUBLE
- argIndex: 3
- }
- argDescriptor {
- name: "cellBeta"
- argType: DOUBLE
- argIndex: 4
- }
- argDescriptor {
- name: "outAlpha"
- argType: DOUBLE
- argIndex: 5
- }
- argDescriptor {
- name: "outBeta"
- argType: DOUBLE
- argIndex: 6
- }
-}
-opList {
- name: "lstmLayer_bp"
argDescriptor {
name: "dLdx"
argType: OUTPUT_TENSOR
@@ -10741,6 +10907,40 @@ opList {
argType: BOOL
argIndex: 7
}
+ argDescriptor {
+ name: "cellClip"
+ argType: DOUBLE
+ }
+ argDescriptor {
+ name: "gateAlpha"
+ argType: DOUBLE
+ argIndex: 1
+ }
+ argDescriptor {
+ name: "gateBeta"
+ argType: DOUBLE
+ argIndex: 2
+ }
+ argDescriptor {
+ name: "cellAlpha"
+ argType: DOUBLE
+ argIndex: 3
+ }
+ argDescriptor {
+ name: "cellBeta"
+ argType: DOUBLE
+ argIndex: 4
+ }
+ argDescriptor {
+ name: "outAlpha"
+ argType: DOUBLE
+ argIndex: 5
+ }
+ argDescriptor {
+ name: "outBeta"
+ argType: DOUBLE
+ argIndex: 6
+ }
argDescriptor {
name: "input"
argType: INPUT_TENSOR
@@ -10800,71 +11000,13 @@ opList {
argType: INPUT_TENSOR
argIndex: 11
}
- argDescriptor {
- name: "dataFormat"
- argType: INT64
- }
- argDescriptor {
- name: "directionMode"
- argType: INT64
- argIndex: 1
- }
- argDescriptor {
- name: "gateAct"
- argType: INT64
- argIndex: 2
- }
- argDescriptor {
- name: "cellAct"
- argType: INT64
- argIndex: 3
- }
- argDescriptor {
- name: "outAct"
- argType: INT64
- argIndex: 4
- }
+}
+opList {
+ name: "lstsq"
argDescriptor {
name: "dtype"
argType: DATA_TYPE
}
- argDescriptor {
- name: "cellClip"
- argType: DOUBLE
- }
- argDescriptor {
- name: "gateAlpha"
- argType: DOUBLE
- argIndex: 1
- }
- argDescriptor {
- name: "gateBeta"
- argType: DOUBLE
- argIndex: 2
- }
- argDescriptor {
- name: "cellAlpha"
- argType: DOUBLE
- argIndex: 3
- }
- argDescriptor {
- name: "cellBeta"
- argType: DOUBLE
- argIndex: 4
- }
- argDescriptor {
- name: "outAlpha"
- argType: DOUBLE
- argIndex: 5
- }
- argDescriptor {
- name: "outBeta"
- argType: DOUBLE
- argIndex: 6
- }
-}
-opList {
- name: "lstsq"
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -10873,6 +11015,10 @@ opList {
name: "fastFlag"
argType: BOOL
}
+ argDescriptor {
+ name: "l2_factor"
+ argType: DOUBLE
+ }
argDescriptor {
name: "a"
argType: INPUT_TENSOR
@@ -10882,17 +11028,13 @@ opList {
argType: INPUT_TENSOR
argIndex: 1
}
+}
+opList {
+ name: "lu"
argDescriptor {
name: "dtype"
argType: DATA_TYPE
}
- argDescriptor {
- name: "l2_factor"
- argType: DOUBLE
- }
-}
-opList {
- name: "lu"
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -10906,13 +11048,13 @@ opList {
name: "input"
argType: INPUT_TENSOR
}
- argDescriptor {
- name: "dtype"
- argType: DATA_TYPE
- }
}
opList {
name: "manhattan"
+ argDescriptor {
+ name: "dimensions"
+ argType: INT64
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -10935,14 +11077,14 @@ opList {
argType: INPUT_TENSOR
argIndex: 1
}
- argDescriptor {
- name: "dimensions"
- argType: INT64
- }
opDeclarationType: LEGACY_XYZ
}
opList {
name: "match_condition"
+ argDescriptor {
+ name: "dimensions"
+ argType: INT64
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -10951,14 +11093,6 @@ opList {
name: "keepDims"
argType: BOOL
}
- argDescriptor {
- name: "input"
- argType: INPUT_TENSOR
- }
- argDescriptor {
- name: "dimensions"
- argType: INT64
- }
argDescriptor {
name: "compare"
argType: DOUBLE
@@ -10968,18 +11102,14 @@ opList {
argType: DOUBLE
argIndex: 1
}
- opDeclarationType: LEGACY_XYZ
-}
-opList {
- name: "match_condition_transform"
- argDescriptor {
- name: "outputs"
- argType: OUTPUT_TENSOR
- }
argDescriptor {
name: "input"
argType: INPUT_TENSOR
}
+ opDeclarationType: LEGACY_XYZ
+}
+opList {
+ name: "match_condition_transform"
argDescriptor {
name: "mode"
argType: INT64
@@ -10988,32 +11118,19 @@ opList {
name: "dataType"
argType: DATA_TYPE
}
- argDescriptor {
- name: "compare"
- argType: DOUBLE
- }
- opDeclarationType: LEGACY_XYZ
-}
-opList {
- name: "matmul"
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
}
argDescriptor {
- name: "transposeX"
- argType: BOOL
+ name: "compare"
+ argType: DOUBLE
}
argDescriptor {
- name: "transposeY"
- argType: BOOL
+ name: "eps"
+ argType: DOUBLE
argIndex: 1
}
- argDescriptor {
- name: "transposeZ"
- argType: BOOL
- argIndex: 2
- }
argDescriptor {
name: "input"
argType: INPUT_TENSOR
@@ -11023,6 +11140,10 @@ opList {
argType: INPUT_TENSOR
argIndex: 1
}
+ opDeclarationType: LEGACY_XYZ
+}
+opList {
+ name: "matmul"
argDescriptor {
name: "transX"
argType: INT64
@@ -11041,6 +11162,24 @@ opList {
name: "dtype"
argType: DATA_TYPE
}
+ argDescriptor {
+ name: "outputs"
+ argType: OUTPUT_TENSOR
+ }
+ argDescriptor {
+ name: "transposeX"
+ argType: BOOL
+ }
+ argDescriptor {
+ name: "transposeY"
+ argType: BOOL
+ argIndex: 1
+ }
+ argDescriptor {
+ name: "transposeZ"
+ argType: BOOL
+ argIndex: 2
+ }
argDescriptor {
name: "alpha"
argType: DOUBLE
@@ -11050,9 +11189,36 @@ opList {
argType: DOUBLE
argIndex: 1
}
+ argDescriptor {
+ name: "input"
+ argType: INPUT_TENSOR
+ }
+ argDescriptor {
+ name: "y"
+ argType: INPUT_TENSOR
+ argIndex: 1
+ }
}
opList {
name: "matmul_bp"
+ argDescriptor {
+ name: "transX"
+ argType: INT64
+ }
+ argDescriptor {
+ name: "transY"
+ argType: INT64
+ argIndex: 1
+ }
+ argDescriptor {
+ name: "transZ"
+ argType: INT64
+ argIndex: 2
+ }
+ argDescriptor {
+ name: "dtype"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "dldx"
argType: OUTPUT_TENSOR
@@ -11062,6 +11228,15 @@ opList {
argType: OUTPUT_TENSOR
argIndex: 1
}
+ argDescriptor {
+ name: "alpha"
+ argType: DOUBLE
+ }
+ argDescriptor {
+ name: "beta"
+ argType: DOUBLE
+ argIndex: 1
+ }
argDescriptor {
name: "input"
argType: INPUT_TENSOR
@@ -11086,36 +11261,18 @@ opList {
argType: INPUT_TENSOR
argIndex: 4
}
- argDescriptor {
- name: "transX"
- argType: INT64
- }
- argDescriptor {
- name: "transY"
- argType: INT64
- argIndex: 1
- }
- argDescriptor {
- name: "transZ"
- argType: INT64
- argIndex: 2
- }
- argDescriptor {
- name: "dtype"
- argType: DATA_TYPE
- }
- argDescriptor {
- name: "alpha"
- argType: DOUBLE
- }
- argDescriptor {
- name: "beta"
- argType: DOUBLE
- argIndex: 1
- }
}
opList {
name: "matrix_band_part"
+ argDescriptor {
+ name: "minLower"
+ argType: INT64
+ }
+ argDescriptor {
+ name: "maxUpper"
+ argType: INT64
+ argIndex: 1
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -11134,19 +11291,14 @@ opList {
argType: INPUT_TENSOR
argIndex: 2
}
- argDescriptor {
- name: "minLower"
- argType: INT64
- }
- argDescriptor {
- name: "maxUpper"
- argType: INT64
- argIndex: 1
- }
opDeclarationType: CONFIGURABLE_OP_IMPL
}
opList {
name: "matrix_determinant"
+ argDescriptor {
+ name: "dtype"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -11159,13 +11311,13 @@ opList {
name: "input"
argType: INPUT_TENSOR
}
+}
+opList {
+ name: "matrix_diag"
argDescriptor {
name: "dtype"
argType: DATA_TYPE
}
-}
-opList {
- name: "matrix_diag"
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -11178,13 +11330,13 @@ opList {
name: "diagonal"
argType: INPUT_TENSOR
}
+}
+opList {
+ name: "matrix_diag_part"
argDescriptor {
name: "dtype"
argType: DATA_TYPE
}
-}
-opList {
- name: "matrix_diag_part"
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -11197,10 +11349,6 @@ opList {
name: "input"
argType: INPUT_TENSOR
}
- argDescriptor {
- name: "dtype"
- argType: DATA_TYPE
- }
}
opList {
name: "matrix_inverse"
@@ -11241,6 +11389,10 @@ opList {
}
opList {
name: "max_pairwise"
+ argDescriptor {
+ name: "dataType"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -11254,32 +11406,10 @@ opList {
argType: INPUT_TENSOR
argIndex: 1
}
- argDescriptor {
- name: "dataType"
- argType: DATA_TYPE
- }
opDeclarationType: LEGACY_XYZ
}
opList {
name: "max_pool_with_argmax"
- argDescriptor {
- name: "outputs"
- argType: OUTPUT_TENSOR
- }
- argDescriptor {
- name: "indices"
- argType: OUTPUT_TENSOR
- argIndex: 1
- }
- argDescriptor {
- name: "input"
- argType: INPUT_TENSOR
- }
- argDescriptor {
- name: "outArgMax"
- argType: INPUT_TENSOR
- argIndex: 1
- }
argDescriptor {
name: "kH"
argType: INT64
@@ -11338,6 +11468,24 @@ opList {
name: "dtype"
argType: DATA_TYPE
}
+ argDescriptor {
+ name: "outputs"
+ argType: OUTPUT_TENSOR
+ }
+ argDescriptor {
+ name: "indices"
+ argType: OUTPUT_TENSOR
+ argIndex: 1
+ }
+ argDescriptor {
+ name: "input"
+ argType: INPUT_TENSOR
+ }
+ argDescriptor {
+ name: "outArgMax"
+ argType: INPUT_TENSOR
+ argIndex: 1
+ }
}
opList {
name: "max_scalar"
@@ -11358,6 +11506,10 @@ opList {
}
opList {
name: "maximum"
+ argDescriptor {
+ name: "dataType"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -11375,14 +11527,14 @@ opList {
argType: INPUT_TENSOR
argIndex: 1
}
- argDescriptor {
- name: "dataType"
- argType: DATA_TYPE
- }
opDeclarationType: BROADCASTABLE_OP_IMPL
}
opList {
name: "maximum_bp"
+ argDescriptor {
+ name: "dtype"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "gradX"
argType: OUTPUT_TENSOR
@@ -11406,41 +11558,25 @@ opList {
argType: INPUT_TENSOR
argIndex: 2
}
- argDescriptor {
- name: "dtype"
- argType: DATA_TYPE
- }
}
opList {
name: "maxout"
+ argDescriptor {
+ name: "dataType"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
}
- argDescriptor {
- name: "inPlace"
- argType: BOOL
- }
argDescriptor {
name: "input"
argType: INPUT_TENSOR
}
- argDescriptor {
- name: "dataType"
- argType: DATA_TYPE
- }
opDeclarationType: LEGACY_XYZ
}
opList {
name: "maxpool2d"
- argDescriptor {
- name: "outputs"
- argType: OUTPUT_TENSOR
- }
- argDescriptor {
- name: "input"
- argType: INPUT_TENSOR
- }
argDescriptor {
name: "kH"
argType: INT64
@@ -11499,22 +11635,17 @@ opList {
name: "dtype"
argType: DATA_TYPE
}
+ argDescriptor {
+ name: "outputs"
+ argType: OUTPUT_TENSOR
+ }
+ argDescriptor {
+ name: "input"
+ argType: INPUT_TENSOR
+ }
}
opList {
name: "maxpool2d_bp"
- argDescriptor {
- name: "gradI"
- argType: OUTPUT_TENSOR
- }
- argDescriptor {
- name: "input"
- argType: INPUT_TENSOR
- }
- argDescriptor {
- name: "gradO"
- argType: INPUT_TENSOR
- argIndex: 1
- }
argDescriptor {
name: "kH"
argType: INT64
@@ -11573,98 +11704,6 @@ opList {
name: "dtype"
argType: DATA_TYPE
}
-}
-opList {
- name: "maxpool3dnew"
- argDescriptor {
- name: "outputs"
- argType: OUTPUT_TENSOR
- }
- argDescriptor {
- name: "input"
- argType: INPUT_TENSOR
- }
- argDescriptor {
- name: "kD"
- argType: INT64
- }
- argDescriptor {
- name: "kH"
- argType: INT64
- argIndex: 1
- }
- argDescriptor {
- name: "kW"
- argType: INT64
- argIndex: 2
- }
- argDescriptor {
- name: "sD"
- argType: INT64
- argIndex: 3
- }
- argDescriptor {
- name: "sH"
- argType: INT64
- argIndex: 4
- }
- argDescriptor {
- name: "sW"
- argType: INT64
- argIndex: 5
- }
- argDescriptor {
- name: "pD"
- argType: INT64
- argIndex: 6
- }
- argDescriptor {
- name: "pH"
- argType: INT64
- argIndex: 7
- }
- argDescriptor {
- name: "pW"
- argType: INT64
- argIndex: 8
- }
- argDescriptor {
- name: "dD"
- argType: INT64
- argIndex: 9
- }
- argDescriptor {
- name: "dH"
- argType: INT64
- argIndex: 10
- }
- argDescriptor {
- name: "dW"
- argType: INT64
- argIndex: 11
- }
- argDescriptor {
- name: "isSameMode"
- argType: INT64
- argIndex: 12
- }
- argDescriptor {
- name: "extraParam0"
- argType: INT64
- argIndex: 13
- }
- argDescriptor {
- name: "isNCDHW"
- argType: INT64
- argIndex: 14
- }
- argDescriptor {
- name: "dtype"
- argType: DATA_TYPE
- }
-}
-opList {
- name: "maxpool3dnew_bp"
argDescriptor {
name: "gradI"
argType: OUTPUT_TENSOR
@@ -11678,6 +11717,9 @@ opList {
argType: INPUT_TENSOR
argIndex: 1
}
+}
+opList {
+ name: "maxpool3dnew"
argDescriptor {
name: "kD"
argType: INT64
@@ -11756,9 +11798,124 @@ opList {
name: "dtype"
argType: DATA_TYPE
}
+ argDescriptor {
+ name: "outputs"
+ argType: OUTPUT_TENSOR
+ }
+ argDescriptor {
+ name: "input"
+ argType: INPUT_TENSOR
+ }
+ argDescriptor {
+ name: "arrayOutput"
+ argType: INPUT_TENSOR
+ argIndex: 1
+ }
+}
+opList {
+ name: "maxpool3dnew_bp"
+ argDescriptor {
+ name: "kD"
+ argType: INT64
+ }
+ argDescriptor {
+ name: "kH"
+ argType: INT64
+ argIndex: 1
+ }
+ argDescriptor {
+ name: "kW"
+ argType: INT64
+ argIndex: 2
+ }
+ argDescriptor {
+ name: "sD"
+ argType: INT64
+ argIndex: 3
+ }
+ argDescriptor {
+ name: "sH"
+ argType: INT64
+ argIndex: 4
+ }
+ argDescriptor {
+ name: "sW"
+ argType: INT64
+ argIndex: 5
+ }
+ argDescriptor {
+ name: "pD"
+ argType: INT64
+ argIndex: 6
+ }
+ argDescriptor {
+ name: "pH"
+ argType: INT64
+ argIndex: 7
+ }
+ argDescriptor {
+ name: "pW"
+ argType: INT64
+ argIndex: 8
+ }
+ argDescriptor {
+ name: "dD"
+ argType: INT64
+ argIndex: 9
+ }
+ argDescriptor {
+ name: "dH"
+ argType: INT64
+ argIndex: 10
+ }
+ argDescriptor {
+ name: "dW"
+ argType: INT64
+ argIndex: 11
+ }
+ argDescriptor {
+ name: "isSameMode"
+ argType: INT64
+ argIndex: 12
+ }
+ argDescriptor {
+ name: "extraParam0"
+ argType: INT64
+ argIndex: 13
+ }
+ argDescriptor {
+ name: "isNCDHW"
+ argType: INT64
+ argIndex: 14
+ }
+ argDescriptor {
+ name: "dtype"
+ argType: DATA_TYPE
+ }
+ argDescriptor {
+ name: "gradI"
+ argType: OUTPUT_TENSOR
+ }
+ argDescriptor {
+ name: "input"
+ argType: INPUT_TENSOR
+ }
+ argDescriptor {
+ name: "gradO"
+ argType: INPUT_TENSOR
+ argIndex: 1
+ }
}
opList {
name: "mean_pairwssqerr_loss"
+ argDescriptor {
+ name: "reductionMode"
+ argType: INT64
+ }
+ argDescriptor {
+ name: "dtype"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -11777,17 +11934,17 @@ opList {
argType: INPUT_TENSOR
argIndex: 2
}
- argDescriptor {
- name: "reductionMode"
- argType: INT64
- }
- argDescriptor {
- name: "dtype"
- argType: DATA_TYPE
- }
}
opList {
name: "mean_pairwssqerr_loss_grad"
+ argDescriptor {
+ name: "reductionMode"
+ argType: INT64
+ }
+ argDescriptor {
+ name: "dtype"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "dLdp"
argType: OUTPUT_TENSOR
@@ -11816,6 +11973,9 @@ opList {
argType: INPUT_TENSOR
argIndex: 2
}
+}
+opList {
+ name: "mean_sqerr_loss"
argDescriptor {
name: "reductionMode"
argType: INT64
@@ -11824,9 +11984,6 @@ opList {
name: "dtype"
argType: DATA_TYPE
}
-}
-opList {
- name: "mean_sqerr_loss"
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -11845,6 +12002,9 @@ opList {
argType: INPUT_TENSOR
argIndex: 2
}
+}
+opList {
+ name: "mean_sqerr_loss_grad"
argDescriptor {
name: "reductionMode"
argType: INT64
@@ -11853,9 +12013,6 @@ opList {
name: "dtype"
argType: DATA_TYPE
}
-}
-opList {
- name: "mean_sqerr_loss_grad"
argDescriptor {
name: "dLdp"
argType: OUTPUT_TENSOR
@@ -11884,14 +12041,6 @@ opList {
argType: INPUT_TENSOR
argIndex: 2
}
- argDescriptor {
- name: "reductionMode"
- argType: INT64
- }
- argDescriptor {
- name: "dtype"
- argType: DATA_TYPE
- }
}
opList {
name: "merge"
@@ -11906,6 +12055,10 @@ opList {
}
opList {
name: "mergeadd"
+ argDescriptor {
+ name: "dataType"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -11918,14 +12071,14 @@ opList {
name: "inArrs"
argType: INPUT_TENSOR
}
- argDescriptor {
- name: "dataType"
- argType: DATA_TYPE
- }
opDeclarationType: OP_IMPL
}
opList {
name: "mergeadd_bp"
+ argDescriptor {
+ name: "dtype"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outArrs"
argType: OUTPUT_TENSOR
@@ -11939,10 +12092,6 @@ opList {
argType: INPUT_TENSOR
argIndex: 1
}
- argDescriptor {
- name: "dtype"
- argType: DATA_TYPE
- }
}
opList {
name: "mergeavg"
@@ -11958,6 +12107,10 @@ opList {
}
opList {
name: "mergeavg_bp"
+ argDescriptor {
+ name: "dtype"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outArrs"
argType: OUTPUT_TENSOR
@@ -11971,10 +12124,6 @@ opList {
argType: INPUT_TENSOR
argIndex: 1
}
- argDescriptor {
- name: "dtype"
- argType: DATA_TYPE
- }
}
opList {
name: "mergemax"
@@ -11990,6 +12139,10 @@ opList {
}
opList {
name: "mergemax_bp"
+ argDescriptor {
+ name: "dtype"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outArrs"
argType: OUTPUT_TENSOR
@@ -12003,21 +12156,9 @@ opList {
argType: INPUT_TENSOR
argIndex: 1
}
- argDescriptor {
- name: "dtype"
- argType: DATA_TYPE
- }
}
opList {
name: "mergemaxindex"
- argDescriptor {
- name: "outputs"
- argType: OUTPUT_TENSOR
- }
- argDescriptor {
- name: "inArrs"
- argType: INPUT_TENSOR
- }
argDescriptor {
name: "dataType"
argType: INT64
@@ -12026,6 +12167,14 @@ opList {
name: "dtype"
argType: DATA_TYPE
}
+ argDescriptor {
+ name: "outputs"
+ argType: OUTPUT_TENSOR
+ }
+ argDescriptor {
+ name: "inArrs"
+ argType: INPUT_TENSOR
+ }
}
opList {
name: "mergesum"
@@ -12040,6 +12189,14 @@ opList {
}
opList {
name: "meshgrid"
+ argDescriptor {
+ name: "swapFirst2Dims"
+ argType: INT64
+ }
+ argDescriptor {
+ name: "dtype"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outArrs"
argType: OUTPUT_TENSOR
@@ -12052,14 +12209,6 @@ opList {
name: "inArrs"
argType: INPUT_TENSOR
}
- argDescriptor {
- name: "swapFirst2Dims"
- argType: INT64
- }
- argDescriptor {
- name: "dtype"
- argType: DATA_TYPE
- }
}
opList {
name: "meta_postulate"
@@ -12067,6 +12216,15 @@ opList {
name: "outputs"
argType: OUTPUT_TENSOR
}
+ argDescriptor {
+ name: "input"
+ argType: INPUT_TENSOR
+ }
+ argDescriptor {
+ name: "y"
+ argType: INPUT_TENSOR
+ argIndex: 1
+ }
opDeclarationType: LEGACY_XYZ
}
opList {
@@ -12083,31 +12241,26 @@ opList {
name: "outputs"
argType: OUTPUT_TENSOR
}
- argDescriptor {
- name: "input"
- argType: INPUT_TENSOR
- }
- argDescriptor {
- name: "y"
- argType: INPUT_TENSOR
- argIndex: 1
- }
opDeclarationType: LEGACY_XYZ
}
opList {
name: "meta_reduce"
- argDescriptor {
- name: "outputs"
- argType: OUTPUT_TENSOR
- }
argDescriptor {
name: "dimensions"
argType: INT64
}
+ argDescriptor {
+ name: "outputs"
+ argType: OUTPUT_TENSOR
+ }
opDeclarationType: LEGACY_XYZ
}
opList {
name: "min_pairwise"
+ argDescriptor {
+ name: "dataType"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -12121,14 +12274,14 @@ opList {
argType: INPUT_TENSOR
argIndex: 1
}
- argDescriptor {
- name: "dataType"
- argType: DATA_TYPE
- }
opDeclarationType: LEGACY_XYZ
}
opList {
name: "minimum"
+ argDescriptor {
+ name: "dataType"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -12146,14 +12299,14 @@ opList {
argType: INPUT_TENSOR
argIndex: 1
}
- argDescriptor {
- name: "dataType"
- argType: DATA_TYPE
- }
opDeclarationType: BROADCASTABLE_OP_IMPL
}
opList {
name: "minimum_bp"
+ argDescriptor {
+ name: "dtype"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "gradX"
argType: OUTPUT_TENSOR
@@ -12177,13 +12330,17 @@ opList {
argType: INPUT_TENSOR
argIndex: 2
}
+}
+opList {
+ name: "mirror_pad"
+ argDescriptor {
+ name: "mode"
+ argType: INT64
+ }
argDescriptor {
name: "dtype"
argType: DATA_TYPE
}
-}
-opList {
- name: "mirror_pad"
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -12201,17 +12358,13 @@ opList {
argType: INPUT_TENSOR
argIndex: 1
}
- argDescriptor {
- name: "mode"
- argType: INT64
- }
- argDescriptor {
- name: "dtype"
- argType: DATA_TYPE
- }
}
opList {
name: "mish"
+ argDescriptor {
+ name: "dataType"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -12224,14 +12377,14 @@ opList {
name: "input"
argType: INPUT_TENSOR
}
- argDescriptor {
- name: "dataType"
- argType: DATA_TYPE
- }
opDeclarationType: LEGACY_XYZ
}
opList {
name: "mod"
+ argDescriptor {
+ name: "dataType"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -12249,14 +12402,14 @@ opList {
argType: INPUT_TENSOR
argIndex: 1
}
- argDescriptor {
- name: "dataType"
- argType: DATA_TYPE
- }
opDeclarationType: BROADCASTABLE_OP_IMPL
}
opList {
name: "mod_bp"
+ argDescriptor {
+ name: "dtype"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "gradX"
argType: OUTPUT_TENSOR
@@ -12284,13 +12437,17 @@ opList {
argType: INPUT_TENSOR
argIndex: 2
}
+}
+opList {
+ name: "moments"
+ argDescriptor {
+ name: "dimensions"
+ argType: INT64
+ }
argDescriptor {
name: "dtype"
argType: DATA_TYPE
}
-}
-opList {
- name: "moments"
argDescriptor {
name: "means"
argType: OUTPUT_TENSOR
@@ -12300,6 +12457,10 @@ opList {
argType: OUTPUT_TENSOR
argIndex: 1
}
+ argDescriptor {
+ name: "keepDims"
+ argType: DOUBLE
+ }
argDescriptor {
name: "input"
argType: INPUT_TENSOR
@@ -12314,18 +12475,6 @@ opList {
argType: INPUT_TENSOR
argIndex: 2
}
- argDescriptor {
- name: "dimensions"
- argType: INT64
- }
- argDescriptor {
- name: "dtype"
- argType: DATA_TYPE
- }
- argDescriptor {
- name: "keepDims"
- argType: DOUBLE
- }
}
opList {
name: "mul_scalar"
@@ -12334,18 +12483,30 @@ opList {
argType: OUTPUT_TENSOR
}
argDescriptor {
- name: "input"
- argType: INPUT_TENSOR
+ name: "inPlace"
+ argType: BOOL
}
argDescriptor {
- name: "y"
+ name: "input"
argType: INPUT_TENSOR
- argIndex: 1
}
opDeclarationType: LEGACY_XYZ
}
opList {
name: "multi_head_dot_product_attention"
+ argDescriptor {
+ name: "normalization"
+ argType: INT64
+ }
+ argDescriptor {
+ name: "weights"
+ argType: INT64
+ argIndex: 1
+ }
+ argDescriptor {
+ name: "dtype"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -12398,22 +12559,17 @@ opList {
argType: INPUT_TENSOR
argIndex: 7
}
+}
+opList {
+ name: "multi_head_dot_product_attention_bp"
argDescriptor {
name: "normalization"
argType: INT64
}
- argDescriptor {
- name: "weights"
- argType: INT64
- argIndex: 1
- }
argDescriptor {
name: "dtype"
argType: DATA_TYPE
}
-}
-opList {
- name: "multi_head_dot_product_attention_bp"
argDescriptor {
name: "dLdq"
argType: OUTPUT_TENSOR
@@ -12496,17 +12652,13 @@ opList {
argType: INPUT_TENSOR
argIndex: 8
}
- argDescriptor {
- name: "normalization"
- argType: INT64
- }
- argDescriptor {
- name: "dtype"
- argType: DATA_TYPE
- }
}
opList {
name: "multiply"
+ argDescriptor {
+ name: "dataType"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -12524,14 +12676,14 @@ opList {
argType: INPUT_TENSOR
argIndex: 1
}
- argDescriptor {
- name: "dataType"
- argType: DATA_TYPE
- }
opDeclarationType: BROADCASTABLE_OP_IMPL
}
opList {
name: "multiply_bp"
+ argDescriptor {
+ name: "dtype"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "gradX"
argType: OUTPUT_TENSOR
@@ -12559,13 +12711,13 @@ opList {
argType: INPUT_TENSOR
argIndex: 2
}
- argDescriptor {
- name: "dtype"
- argType: DATA_TYPE
- }
}
opList {
name: "nadam_updater"
+ argDescriptor {
+ name: "iteration"
+ argType: INT64
+ }
argDescriptor {
name: "update"
argType: OUTPUT_TENSOR
@@ -12580,6 +12732,25 @@ opList {
argType: OUTPUT_TENSOR
argIndex: 2
}
+ argDescriptor {
+ name: "dLr"
+ argType: DOUBLE
+ }
+ argDescriptor {
+ name: "dBeta1"
+ argType: DOUBLE
+ argIndex: 1
+ }
+ argDescriptor {
+ name: "dBeta2"
+ argType: DOUBLE
+ argIndex: 2
+ }
+ argDescriptor {
+ name: "dEpsilon"
+ argType: DOUBLE
+ argIndex: 3
+ }
argDescriptor {
name: "gradient"
argType: INPUT_TENSOR
@@ -12614,33 +12785,14 @@ opList {
argType: INPUT_TENSOR
argIndex: 6
}
- argDescriptor {
- name: "iteration"
- argType: INT64
- }
- argDescriptor {
- name: "dLr"
- argType: DOUBLE
- }
- argDescriptor {
- name: "dBeta1"
- argType: DOUBLE
- argIndex: 1
- }
- argDescriptor {
- name: "dBeta2"
- argType: DOUBLE
- argIndex: 2
- }
- argDescriptor {
- name: "dEpsilon"
- argType: DOUBLE
- argIndex: 3
- }
opDeclarationType: CONFIGURABLE_OP_IMPL
}
opList {
name: "neg"
+ argDescriptor {
+ name: "dataType"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -12653,10 +12805,6 @@ opList {
name: "input"
argType: INPUT_TENSOR
}
- argDescriptor {
- name: "dataType"
- argType: DATA_TYPE
- }
opDeclarationType: LEGACY_XYZ
}
opList {
@@ -12670,6 +12818,15 @@ opList {
argType: OUTPUT_TENSOR
argIndex: 1
}
+ argDescriptor {
+ name: "dLr"
+ argType: DOUBLE
+ }
+ argDescriptor {
+ name: "dMomentum"
+ argType: DOUBLE
+ argIndex: 1
+ }
argDescriptor {
name: "gradient"
argType: INPUT_TENSOR
@@ -12689,15 +12846,6 @@ opList {
argType: INPUT_TENSOR
argIndex: 3
}
- argDescriptor {
- name: "dLr"
- argType: DOUBLE
- }
- argDescriptor {
- name: "dMomentum"
- argType: DOUBLE
- argIndex: 1
- }
opDeclarationType: CONFIGURABLE_OP_IMPL
}
opList {
@@ -12706,17 +12854,25 @@ opList {
name: "outputs"
argType: OUTPUT_TENSOR
}
- argDescriptor {
- name: "input"
- argType: INPUT_TENSOR
- }
argDescriptor {
name: "frameName"
argType: STRING
}
+ argDescriptor {
+ name: "input"
+ argType: INPUT_TENSOR
+ }
}
opList {
name: "non_max_suppression"
+ argDescriptor {
+ name: "maxOutputSize"
+ argType: INT64
+ }
+ argDescriptor {
+ name: "dtype"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -12736,7 +12892,7 @@ opList {
argIndex: 2
}
argDescriptor {
- name: "overlayThreshold"
+ name: "iouThreshold"
argType: INPUT_TENSOR
argIndex: 3
}
@@ -12745,6 +12901,9 @@ opList {
argType: INPUT_TENSOR
argIndex: 4
}
+}
+opList {
+ name: "non_max_suppression_overlaps"
argDescriptor {
name: "maxOutputSize"
argType: INT64
@@ -12754,7 +12913,11 @@ opList {
argType: DATA_TYPE
}
argDescriptor {
- name: "iouThreshold"
+ name: "outputs"
+ argType: OUTPUT_TENSOR
+ }
+ argDescriptor {
+ name: "overlapThreshold"
argType: DOUBLE
}
argDescriptor {
@@ -12762,13 +12925,6 @@ opList {
argType: DOUBLE
argIndex: 1
}
-}
-opList {
- name: "non_max_suppression_overlaps"
- argDescriptor {
- name: "outputs"
- argType: OUTPUT_TENSOR
- }
argDescriptor {
name: "boxes"
argType: INPUT_TENSOR
@@ -12778,6 +12934,24 @@ opList {
argType: INPUT_TENSOR
argIndex: 1
}
+ argDescriptor {
+ name: "maxOutSize"
+ argType: INPUT_TENSOR
+ argIndex: 2
+ }
+ argDescriptor {
+ name: "iouThreshold"
+ argType: INPUT_TENSOR
+ argIndex: 3
+ }
+ argDescriptor {
+ name: "scoreThreshold"
+ argType: INPUT_TENSOR
+ argIndex: 4
+ }
+}
+opList {
+ name: "non_max_suppression_v3"
argDescriptor {
name: "maxOutputSize"
argType: INT64
@@ -12786,18 +12960,6 @@ opList {
name: "dtype"
argType: DATA_TYPE
}
- argDescriptor {
- name: "overlapThreshold"
- argType: DOUBLE
- }
- argDescriptor {
- name: "scoreThreshold"
- argType: DOUBLE
- argIndex: 1
- }
-}
-opList {
- name: "non_max_suppression_v3"
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -12826,14 +12988,6 @@ opList {
argType: INPUT_TENSOR
argIndex: 4
}
- argDescriptor {
- name: "maxOutputSize"
- argType: INT64
- }
- argDescriptor {
- name: "dtype"
- argType: DATA_TYPE
- }
}
opList {
name: "noop"
@@ -12850,9 +13004,13 @@ opList {
opList {
name: "norm"
argDescriptor {
- name: "outputs"
+ name: "*output"
argType: OUTPUT_TENSOR
}
+ argDescriptor {
+ name: "mode"
+ argType: DOUBLE
+ }
argDescriptor {
name: "input"
argType: INPUT_TENSOR
@@ -12862,14 +13020,14 @@ opList {
argType: INPUT_TENSOR
argIndex: 1
}
- argDescriptor {
- name: "mode"
- argType: DOUBLE
- }
opDeclarationType: REDUCTION_OP_IMPL
}
opList {
name: "normalize_moments"
+ argDescriptor {
+ name: "dtype"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "resMeans"
argType: OUTPUT_TENSOR
@@ -12879,6 +13037,10 @@ opList {
argType: OUTPUT_TENSOR
argIndex: 1
}
+ argDescriptor {
+ name: "shift"
+ argType: DOUBLE
+ }
argDescriptor {
name: "counts"
argType: INPUT_TENSOR
@@ -12903,21 +13065,21 @@ opList {
argType: INPUT_TENSOR
argIndex: 4
}
- argDescriptor {
- name: "dtype"
- argType: DATA_TYPE
- }
- argDescriptor {
- name: "shift"
- argType: DOUBLE
- }
}
opList {
name: "not"
+ argDescriptor {
+ name: "dataType"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
}
+ argDescriptor {
+ name: "comparable"
+ argType: DOUBLE
+ }
argDescriptor {
name: "input"
argType: INPUT_TENSOR
@@ -12927,18 +13089,14 @@ opList {
argType: INPUT_TENSOR
argIndex: 1
}
- argDescriptor {
- name: "dataType"
- argType: DATA_TYPE
- }
- argDescriptor {
- name: "comparable"
- argType: DOUBLE
- }
opDeclarationType: LEGACY_XYZ
}
opList {
name: "not_equals"
+ argDescriptor {
+ name: "dataType"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -12956,10 +13114,6 @@ opList {
argType: INPUT_TENSOR
argIndex: 1
}
- argDescriptor {
- name: "dataType"
- argType: DATA_TYPE
- }
opDeclarationType: BROADCASTABLE_BOOL_OP_IMPL
}
opList {
@@ -12996,6 +13150,14 @@ opList {
}
opList {
name: "nth_element"
+ argDescriptor {
+ name: "reverse"
+ argType: INT64
+ }
+ argDescriptor {
+ name: "dtype"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -13013,17 +13175,13 @@ opList {
argType: INPUT_TENSOR
argIndex: 1
}
- argDescriptor {
- name: "reverse"
- argType: INT64
- }
- argDescriptor {
- name: "dtype"
- argType: DATA_TYPE
- }
}
opList {
name: "old_assign"
+ argDescriptor {
+ name: "dataType"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -13036,18 +13194,41 @@ opList {
name: "input"
argType: INPUT_TENSOR
}
- argDescriptor {
- name: "dataType"
- argType: DATA_TYPE
- }
opDeclarationType: LEGACY_XYZ
}
opList {
name: "onehot"
+ argDescriptor {
+ name: "dimensions"
+ argType: INT64
+ }
+ argDescriptor {
+ name: "depth"
+ argType: INT64
+ argIndex: 1
+ }
+ argDescriptor {
+ name: "dataType"
+ argType: INT64
+ argIndex: 2
+ }
+ argDescriptor {
+ name: "dtype"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
}
+ argDescriptor {
+ name: "on"
+ argType: DOUBLE
+ }
+ argDescriptor {
+ name: "off"
+ argType: DOUBLE
+ argIndex: 1
+ }
argDescriptor {
name: "input"
argType: INPUT_TENSOR
@@ -13067,36 +13248,13 @@ opList {
argType: INPUT_TENSOR
argIndex: 3
}
- argDescriptor {
- name: "dimensions"
- argType: INT64
- }
- argDescriptor {
- name: "depth"
- argType: INT64
- argIndex: 1
- }
- argDescriptor {
- name: "dataType"
- argType: INT64
- argIndex: 2
- }
- argDescriptor {
- name: "dtype"
- argType: DATA_TYPE
- }
- argDescriptor {
- name: "on"
- argType: DOUBLE
- }
- argDescriptor {
- name: "off"
- argType: DOUBLE
- argIndex: 1
- }
}
opList {
name: "oneminus"
+ argDescriptor {
+ name: "dataType"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -13109,22 +13267,10 @@ opList {
name: "input"
argType: INPUT_TENSOR
}
- argDescriptor {
- name: "dataType"
- argType: DATA_TYPE
- }
opDeclarationType: LEGACY_XYZ
}
opList {
name: "ones_as"
- argDescriptor {
- name: "outputs"
- argType: OUTPUT_TENSOR
- }
- argDescriptor {
- name: "input"
- argType: INPUT_TENSOR
- }
argDescriptor {
name: "dataType"
argType: INT64
@@ -13133,9 +13279,21 @@ opList {
name: "dtype"
argType: DATA_TYPE
}
+ argDescriptor {
+ name: "outputs"
+ argType: OUTPUT_TENSOR
+ }
+ argDescriptor {
+ name: "input"
+ argType: INPUT_TENSOR
+ }
}
opList {
name: "or"
+ argDescriptor {
+ name: "dataType"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -13144,23 +13302,14 @@ opList {
name: "inPlace"
argType: BOOL
}
- argDescriptor {
- name: "input"
- argType: INPUT_TENSOR
- }
- argDescriptor {
- name: "y"
- argType: INPUT_TENSOR
- argIndex: 1
- }
- argDescriptor {
- name: "dataType"
- argType: DATA_TYPE
- }
argDescriptor {
name: "comparable"
argType: DOUBLE
}
+ argDescriptor {
+ name: "input"
+ argType: INPUT_TENSOR
+ }
opDeclarationType: LEGACY_XYZ
}
opList {
@@ -13181,6 +13330,10 @@ opList {
}
opList {
name: "order"
+ argDescriptor {
+ name: "dtype"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -13189,17 +13342,25 @@ opList {
name: "input"
argType: INPUT_TENSOR
}
- argDescriptor {
- name: "dtype"
- argType: DATA_TYPE
- }
}
opList {
name: "pad"
+ argDescriptor {
+ name: "mode"
+ argType: INT64
+ }
+ argDescriptor {
+ name: "dtype"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
}
+ argDescriptor {
+ name: "padValue"
+ argType: DOUBLE
+ }
argDescriptor {
name: "input"
argType: INPUT_TENSOR
@@ -13209,21 +13370,13 @@ opList {
argType: INPUT_TENSOR
argIndex: 1
}
- argDescriptor {
- name: "mode"
- argType: INT64
- }
+}
+opList {
+ name: "parallel_stack"
argDescriptor {
name: "dtype"
argType: DATA_TYPE
}
- argDescriptor {
- name: "padValue"
- argType: DOUBLE
- }
-}
-opList {
- name: "parallel_stack"
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -13237,21 +13390,9 @@ opList {
argType: INPUT_TENSOR
argIndex: 1
}
- argDescriptor {
- name: "dtype"
- argType: DATA_TYPE
- }
}
opList {
name: "percentile"
- argDescriptor {
- name: "outputs"
- argType: OUTPUT_TENSOR
- }
- argDescriptor {
- name: "input"
- argType: INPUT_TENSOR
- }
argDescriptor {
name: "dimensions"
argType: INT64
@@ -13260,6 +13401,10 @@ opList {
name: "dtype"
argType: DATA_TYPE
}
+ argDescriptor {
+ name: "outputs"
+ argType: OUTPUT_TENSOR
+ }
argDescriptor {
name: "q"
argType: DOUBLE
@@ -13274,9 +13419,21 @@ opList {
argType: DOUBLE
argIndex: 2
}
+ argDescriptor {
+ name: "input"
+ argType: INPUT_TENSOR
+ }
}
opList {
name: "permute"
+ argDescriptor {
+ name: "reverseDims"
+ argType: INT64
+ }
+ argDescriptor {
+ name: "dtype"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -13286,18 +13443,10 @@ opList {
argType: INPUT_TENSOR
}
argDescriptor {
- name: "permuteDims"
+ name: "permutationVector"
argType: INPUT_TENSOR
argIndex: 1
}
- argDescriptor {
- name: "reverseDims"
- argType: INT64
- }
- argDescriptor {
- name: "dtype"
- argType: DATA_TYPE
- }
}
opList {
name: "pick_list"
@@ -13318,14 +13467,6 @@ opList {
}
opList {
name: "pnormpool2d"
- argDescriptor {
- name: "outputs"
- argType: OUTPUT_TENSOR
- }
- argDescriptor {
- name: "input"
- argType: INPUT_TENSOR
- }
argDescriptor {
name: "kY"
argType: INT64
@@ -13384,22 +13525,17 @@ opList {
name: "dtype"
argType: DATA_TYPE
}
-}
-opList {
- name: "pnormpool2d_bp"
argDescriptor {
- name: "gradI"
+ name: "outputs"
argType: OUTPUT_TENSOR
}
argDescriptor {
name: "input"
argType: INPUT_TENSOR
}
- argDescriptor {
- name: "gradO"
- argType: INPUT_TENSOR
- argIndex: 1
- }
+}
+opList {
+ name: "pnormpool2d_bp"
argDescriptor {
name: "kH"
argType: INT64
@@ -13458,13 +13594,39 @@ opList {
name: "dtype"
argType: DATA_TYPE
}
+ argDescriptor {
+ name: "gradI"
+ argType: OUTPUT_TENSOR
+ }
argDescriptor {
name: "eps"
argType: DOUBLE
}
+ argDescriptor {
+ name: "input"
+ argType: INPUT_TENSOR
+ }
+ argDescriptor {
+ name: "gradO"
+ argType: INPUT_TENSOR
+ argIndex: 1
+ }
}
opList {
name: "pointwise_conv2d"
+ argDescriptor {
+ name: "isNCHW"
+ argType: INT64
+ }
+ argDescriptor {
+ name: "wFormat"
+ argType: INT64
+ argIndex: 1
+ }
+ argDescriptor {
+ name: "dtype"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -13483,19 +13645,6 @@ opList {
argType: INPUT_TENSOR
argIndex: 2
}
- argDescriptor {
- name: "isNCHW"
- argType: INT64
- }
- argDescriptor {
- name: "wFormat"
- argType: INT64
- argIndex: 1
- }
- argDescriptor {
- name: "dtype"
- argType: DATA_TYPE
- }
}
opList {
name: "polygamma"
@@ -13544,6 +13693,10 @@ opList {
name: "inPlace"
argType: BOOL
}
+ argDescriptor {
+ name: "pow"
+ argType: DOUBLE
+ }
argDescriptor {
name: "input"
argType: INPUT_TENSOR
@@ -13553,14 +13706,14 @@ opList {
argType: INPUT_TENSOR
argIndex: 1
}
- argDescriptor {
- name: "pow"
- argType: DOUBLE
- }
opDeclarationType: LEGACY_XYZ
}
opList {
name: "pow_pairwise"
+ argDescriptor {
+ name: "dataType"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -13578,14 +13731,14 @@ opList {
argType: INPUT_TENSOR
argIndex: 1
}
- argDescriptor {
- name: "dataType"
- argType: DATA_TYPE
- }
opDeclarationType: LEGACY_XYZ
}
opList {
name: "precise_gelu"
+ argDescriptor {
+ name: "dataType"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -13603,14 +13756,14 @@ opList {
name: "input"
argType: INPUT_TENSOR
}
- argDescriptor {
- name: "dataType"
- argType: DATA_TYPE
- }
opDeclarationType: LEGACY_XYZ
}
opList {
name: "prelu"
+ argDescriptor {
+ name: "sharedAxes"
+ argType: INT64
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -13624,14 +13777,14 @@ opList {
argType: INPUT_TENSOR
argIndex: 1
}
- argDescriptor {
- name: "sharedAxes"
- argType: INT64
- }
opDeclarationType: CONFIGURABLE_OP_IMPL
}
opList {
name: "prelu_bp"
+ argDescriptor {
+ name: "sharedAxes"
+ argType: INT64
+ }
argDescriptor {
name: "dLdI"
argType: OUTPUT_TENSOR
@@ -13665,14 +13818,14 @@ opList {
argType: INPUT_TENSOR
argIndex: 4
}
- argDescriptor {
- name: "sharedAxes"
- argType: INT64
- }
opDeclarationType: CONFIGURABLE_OP_IMPL
}
opList {
name: "print_affinity"
+ argDescriptor {
+ name: "dtype"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -13681,13 +13834,13 @@ opList {
name: "input"
argType: INPUT_TENSOR
}
+}
+opList {
+ name: "print_variable"
argDescriptor {
name: "dtype"
argType: DATA_TYPE
}
-}
-opList {
- name: "print_variable"
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -13705,10 +13858,6 @@ opList {
argType: INPUT_TENSOR
argIndex: 1
}
- argDescriptor {
- name: "dtype"
- argType: DATA_TYPE
- }
}
opList {
name: "probablistic_merge"
@@ -13716,6 +13865,10 @@ opList {
name: "outputs"
argType: OUTPUT_TENSOR
}
+ argDescriptor {
+ name: "probability"
+ argType: DOUBLE
+ }
argDescriptor {
name: "input"
argType: INPUT_TENSOR
@@ -13725,14 +13878,14 @@ opList {
argType: INPUT_TENSOR
argIndex: 1
}
- argDescriptor {
- name: "probability"
- argType: DOUBLE
- }
opDeclarationType: LEGACY_XYZ
}
opList {
name: "qr"
+ argDescriptor {
+ name: "dtype"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputQ"
argType: OUTPUT_TENSOR
@@ -13750,32 +13903,36 @@ opList {
name: "input"
argType: INPUT_TENSOR
}
+}
+opList {
+ name: "random_bernoulli"
argDescriptor {
name: "dtype"
argType: DATA_TYPE
}
-}
-opList {
- name: "random_bernoulli"
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
}
+ argDescriptor {
+ name: "f"
+ argType: DOUBLE
+ }
argDescriptor {
name: "input"
argType: INPUT_TENSOR
}
+}
+opList {
+ name: "random_crop"
+ argDescriptor {
+ name: "seed"
+ argType: INT64
+ }
argDescriptor {
name: "dtype"
argType: DATA_TYPE
}
- argDescriptor {
- name: "f"
- argType: DOUBLE
- }
-}
-opList {
- name: "random_crop"
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -13789,21 +13946,9 @@ opList {
argType: INPUT_TENSOR
argIndex: 1
}
- argDescriptor {
- name: "seed"
- argType: INT64
- }
- argDescriptor {
- name: "dtype"
- argType: DATA_TYPE
- }
}
opList {
name: "random_exponential"
- argDescriptor {
- name: "outputs"
- argType: OUTPUT_TENSOR
- }
argDescriptor {
name: "shape"
argType: INT64
@@ -13812,6 +13957,10 @@ opList {
name: "dtype"
argType: DATA_TYPE
}
+ argDescriptor {
+ name: "outputs"
+ argType: OUTPUT_TENSOR
+ }
argDescriptor {
name: "lambda"
argType: DOUBLE
@@ -13819,6 +13968,14 @@ opList {
}
opList {
name: "random_gamma"
+ argDescriptor {
+ name: "seed"
+ argType: INT64
+ }
+ argDescriptor {
+ name: "dtype"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -13837,17 +13994,17 @@ opList {
argType: INPUT_TENSOR
argIndex: 2
}
+}
+opList {
+ name: "random_multinomial"
argDescriptor {
- name: "seed"
+ name: "dimC"
argType: INT64
}
argDescriptor {
name: "dtype"
argType: DATA_TYPE
}
-}
-opList {
- name: "random_multinomial"
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -13861,17 +14018,13 @@ opList {
argType: INPUT_TENSOR
argIndex: 1
}
- argDescriptor {
- name: "dimC"
- argType: INT64
- }
+}
+opList {
+ name: "random_normal"
argDescriptor {
name: "dtype"
argType: DATA_TYPE
}
-}
-opList {
- name: "random_normal"
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -13880,13 +14033,17 @@ opList {
name: "input"
argType: INPUT_TENSOR
}
+}
+opList {
+ name: "random_poisson"
+ argDescriptor {
+ name: "seed"
+ argType: INT64
+ }
argDescriptor {
name: "dtype"
argType: DATA_TYPE
}
-}
-opList {
- name: "random_poisson"
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -13900,17 +14057,13 @@ opList {
argType: INPUT_TENSOR
argIndex: 1
}
- argDescriptor {
- name: "seed"
- argType: INT64
- }
- argDescriptor {
- name: "dtype"
- argType: DATA_TYPE
- }
}
opList {
name: "random_shuffle"
+ argDescriptor {
+ name: "seeds"
+ argType: INT64
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -13919,10 +14072,6 @@ opList {
name: "input"
argType: INPUT_TENSOR
}
- argDescriptor {
- name: "seeds"
- argType: INT64
- }
opDeclarationType: OP_IMPL
}
opList {
@@ -13931,10 +14080,6 @@ opList {
name: "outputs"
argType: OUTPUT_TENSOR
}
- argDescriptor {
- name: "shape"
- argType: INPUT_TENSOR
- }
argDescriptor {
name: "mean"
argType: DOUBLE
@@ -13944,27 +14089,13 @@ opList {
argType: DOUBLE
argIndex: 1
}
-}
-opList {
- name: "randomuniform"
- argDescriptor {
- name: "outputs"
- argType: OUTPUT_TENSOR
- }
argDescriptor {
name: "shape"
argType: INPUT_TENSOR
}
- argDescriptor {
- name: "min"
- argType: INPUT_TENSOR
- argIndex: 1
- }
- argDescriptor {
- name: "max"
- argType: INPUT_TENSOR
- argIndex: 2
- }
+}
+opList {
+ name: "randomuniform"
argDescriptor {
name: "dtype"
argType: INT64
@@ -13978,6 +14109,10 @@ opList {
name: "dataType"
argType: DATA_TYPE
}
+ argDescriptor {
+ name: "outputs"
+ argType: OUTPUT_TENSOR
+ }
argDescriptor {
name: "min"
argType: DOUBLE
@@ -13987,27 +14122,23 @@ opList {
argType: DOUBLE
argIndex: 1
}
-}
-opList {
- name: "range"
argDescriptor {
- name: "outputs"
- argType: OUTPUT_TENSOR
- }
- argDescriptor {
- name: "from"
+ name: "shape"
argType: INPUT_TENSOR
}
argDescriptor {
- name: "to"
+ name: "min"
argType: INPUT_TENSOR
argIndex: 1
}
argDescriptor {
- name: "step"
+ name: "max"
argType: INPUT_TENSOR
argIndex: 2
}
+}
+opList {
+ name: "range"
argDescriptor {
name: "from"
argType: INT64
@@ -14026,6 +14157,10 @@ opList {
name: "dtype"
argType: DATA_TYPE
}
+ argDescriptor {
+ name: "outputs"
+ argType: OUTPUT_TENSOR
+ }
argDescriptor {
name: "from"
argType: DOUBLE
@@ -14040,9 +14175,27 @@ opList {
argType: DOUBLE
argIndex: 2
}
+ argDescriptor {
+ name: "from"
+ argType: INPUT_TENSOR
+ }
+ argDescriptor {
+ name: "to"
+ argType: INPUT_TENSOR
+ argIndex: 1
+ }
+ argDescriptor {
+ name: "step"
+ argType: INPUT_TENSOR
+ argIndex: 2
+ }
}
opList {
name: "rank"
+ argDescriptor {
+ name: "dtype"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -14055,13 +14208,13 @@ opList {
name: "input"
argType: INPUT_TENSOR
}
- argDescriptor {
- name: "dtype"
- argType: DATA_TYPE
- }
}
opList {
name: "rational_tanh"
+ argDescriptor {
+ name: "dataType"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -14074,14 +14227,14 @@ opList {
name: "input"
argType: INPUT_TENSOR
}
- argDescriptor {
- name: "dataType"
- argType: DATA_TYPE
- }
opDeclarationType: LEGACY_XYZ
}
opList {
name: "rational_tanh_derivative"
+ argDescriptor {
+ name: "dataType"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -14094,10 +14247,6 @@ opList {
name: "input"
argType: INPUT_TENSOR
}
- argDescriptor {
- name: "dataType"
- argType: DATA_TYPE
- }
opDeclarationType: LEGACY_XYZ
}
opList {
@@ -14135,18 +14284,27 @@ opList {
name: "outputs"
argType: OUTPUT_TENSOR
}
- argDescriptor {
- name: "inPlace"
- argType: BOOL
- }
argDescriptor {
name: "input"
argType: INPUT_TENSOR
}
+ argDescriptor {
+ name: "y"
+ argType: INPUT_TENSOR
+ argIndex: 1
+ }
opDeclarationType: LEGACY_XYZ
}
opList {
name: "read_list"
+ argDescriptor {
+ name: "index"
+ argType: INT64
+ }
+ argDescriptor {
+ name: "importDataType"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -14160,18 +14318,14 @@ opList {
argType: INPUT_TENSOR
argIndex: 1
}
- argDescriptor {
- name: "index"
- argType: INT64
- }
- argDescriptor {
- name: "importDataType"
- argType: DATA_TYPE
- }
opDeclarationType: LIST_OP_IMPL
}
opList {
name: "realdiv"
+ argDescriptor {
+ name: "dataType"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -14189,14 +14343,14 @@ opList {
argType: INPUT_TENSOR
argIndex: 1
}
- argDescriptor {
- name: "dataType"
- argType: DATA_TYPE
- }
opDeclarationType: BROADCASTABLE_OP_IMPL
}
opList {
name: "realdiv_bp"
+ argDescriptor {
+ name: "dtype"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "gradX"
argType: OUTPUT_TENSOR
@@ -14220,13 +14374,13 @@ opList {
argType: INPUT_TENSOR
argIndex: 2
}
- argDescriptor {
- name: "dtype"
- argType: DATA_TYPE
- }
}
opList {
name: "rectified_tanh"
+ argDescriptor {
+ name: "dataType"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -14239,14 +14393,14 @@ opList {
name: "input"
argType: INPUT_TENSOR
}
- argDescriptor {
- name: "dataType"
- argType: DATA_TYPE
- }
opDeclarationType: LEGACY_XYZ
}
opList {
name: "rectified_tanh_derivative"
+ argDescriptor {
+ name: "dataType"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -14259,10 +14413,6 @@ opList {
name: "input"
argType: INPUT_TENSOR
}
- argDescriptor {
- name: "dataType"
- argType: DATA_TYPE
- }
opDeclarationType: LEGACY_XYZ
}
opList {
@@ -14296,6 +14446,14 @@ opList {
}
opList {
name: "reduce_dot_bp"
+ argDescriptor {
+ name: "dimensions"
+ argType: INT64
+ }
+ argDescriptor {
+ name: "dtype"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "gradX"
argType: OUTPUT_TENSOR
@@ -14333,34 +14491,9 @@ opList {
argType: INPUT_TENSOR
argIndex: 4
}
- argDescriptor {
- name: "dimensions"
- argType: INT64
- }
- argDescriptor {
- name: "dtype"
- argType: DATA_TYPE
- }
}
opList {
name: "reduce_logsumexp"
- argDescriptor {
- name: "outputs"
- argType: OUTPUT_TENSOR
- }
- argDescriptor {
- name: "keepDim"
- argType: BOOL
- }
- argDescriptor {
- name: "input"
- argType: INPUT_TENSOR
- }
- argDescriptor {
- name: "dimensions"
- argType: INPUT_TENSOR
- argIndex: 1
- }
argDescriptor {
name: "dimensions"
argType: INT64
@@ -14369,13 +14502,38 @@ opList {
name: "dtype"
argType: DATA_TYPE
}
+ argDescriptor {
+ name: "outputs"
+ argType: OUTPUT_TENSOR
+ }
argDescriptor {
name: "keepDims"
argType: DOUBLE
}
+ argDescriptor {
+ name: "keepDims"
+ argType: BOOL
+ }
+ argDescriptor {
+ name: "input"
+ argType: INPUT_TENSOR
+ }
+ argDescriptor {
+ name: "dimensions"
+ argType: INPUT_TENSOR
+ argIndex: 1
+ }
}
opList {
name: "reduce_max"
+ argDescriptor {
+ name: "dimensions"
+ argType: INT64
+ }
+ argDescriptor {
+ name: "dtype"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -14393,17 +14551,17 @@ opList {
argType: INPUT_TENSOR
argIndex: 1
}
- argDescriptor {
- name: "dimensions"
- argType: INT64
- }
- argDescriptor {
- name: "dtype"
- argType: DATA_TYPE
- }
}
opList {
name: "reduce_max_bp"
+ argDescriptor {
+ name: "dimensions"
+ argType: INT64
+ }
+ argDescriptor {
+ name: "dtype"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "gradI"
argType: OUTPUT_TENSOR
@@ -14426,17 +14584,17 @@ opList {
argType: INPUT_TENSOR
argIndex: 2
}
- argDescriptor {
- name: "dimensions"
- argType: INT64
- }
- argDescriptor {
- name: "dtype"
- argType: DATA_TYPE
- }
}
opList {
name: "reduce_mean"
+ argDescriptor {
+ name: "dimensions"
+ argType: INT64
+ }
+ argDescriptor {
+ name: "dtype"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -14454,17 +14612,17 @@ opList {
argType: INPUT_TENSOR
argIndex: 1
}
- argDescriptor {
- name: "dimensions"
- argType: INT64
- }
- argDescriptor {
- name: "dtype"
- argType: DATA_TYPE
- }
}
opList {
name: "reduce_mean_bp"
+ argDescriptor {
+ name: "dimensions"
+ argType: INT64
+ }
+ argDescriptor {
+ name: "dtype"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "gradI"
argType: OUTPUT_TENSOR
@@ -14487,17 +14645,17 @@ opList {
argType: INPUT_TENSOR
argIndex: 2
}
- argDescriptor {
- name: "dimensions"
- argType: INT64
- }
- argDescriptor {
- name: "dtype"
- argType: DATA_TYPE
- }
}
opList {
name: "reduce_min"
+ argDescriptor {
+ name: "dimensions"
+ argType: INT64
+ }
+ argDescriptor {
+ name: "dtype"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -14515,17 +14673,17 @@ opList {
argType: INPUT_TENSOR
argIndex: 1
}
- argDescriptor {
- name: "dimensions"
- argType: INT64
- }
- argDescriptor {
- name: "dtype"
- argType: DATA_TYPE
- }
}
opList {
name: "reduce_min_bp"
+ argDescriptor {
+ name: "dimensions"
+ argType: INT64
+ }
+ argDescriptor {
+ name: "dtype"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "gradI"
argType: OUTPUT_TENSOR
@@ -14548,17 +14706,17 @@ opList {
argType: INPUT_TENSOR
argIndex: 2
}
- argDescriptor {
- name: "dimensions"
- argType: INT64
- }
- argDescriptor {
- name: "dtype"
- argType: DATA_TYPE
- }
}
opList {
name: "reduce_norm1"
+ argDescriptor {
+ name: "dimensions"
+ argType: INT64
+ }
+ argDescriptor {
+ name: "dtype"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -14576,17 +14734,17 @@ opList {
argType: INPUT_TENSOR
argIndex: 1
}
- argDescriptor {
- name: "dimensions"
- argType: INT64
- }
- argDescriptor {
- name: "dtype"
- argType: DATA_TYPE
- }
}
opList {
name: "reduce_norm1_bp"
+ argDescriptor {
+ name: "dimensions"
+ argType: INT64
+ }
+ argDescriptor {
+ name: "dtype"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "gradI"
argType: OUTPUT_TENSOR
@@ -14609,17 +14767,17 @@ opList {
argType: INPUT_TENSOR
argIndex: 2
}
- argDescriptor {
- name: "dimensions"
- argType: INT64
- }
- argDescriptor {
- name: "dtype"
- argType: DATA_TYPE
- }
}
opList {
name: "reduce_norm2"
+ argDescriptor {
+ name: "dimensions"
+ argType: INT64
+ }
+ argDescriptor {
+ name: "dtype"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -14637,17 +14795,17 @@ opList {
argType: INPUT_TENSOR
argIndex: 1
}
- argDescriptor {
- name: "dimensions"
- argType: INT64
- }
- argDescriptor {
- name: "dtype"
- argType: DATA_TYPE
- }
}
opList {
name: "reduce_norm2_bp"
+ argDescriptor {
+ name: "dimensions"
+ argType: INT64
+ }
+ argDescriptor {
+ name: "dtype"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "gradI"
argType: OUTPUT_TENSOR
@@ -14670,17 +14828,13 @@ opList {
argType: INPUT_TENSOR
argIndex: 2
}
- argDescriptor {
- name: "dimensions"
- argType: INT64
- }
- argDescriptor {
- name: "dtype"
- argType: DATA_TYPE
- }
}
opList {
name: "reduce_norm_max"
+ argDescriptor {
+ name: "dtype"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -14698,13 +14852,17 @@ opList {
argType: INPUT_TENSOR
argIndex: 1
}
+}
+opList {
+ name: "reduce_norm_max_bp"
+ argDescriptor {
+ name: "dimensions"
+ argType: INT64
+ }
argDescriptor {
name: "dtype"
argType: DATA_TYPE
}
-}
-opList {
- name: "reduce_norm_max_bp"
argDescriptor {
name: "gradI"
argType: OUTPUT_TENSOR
@@ -14727,17 +14885,13 @@ opList {
argType: INPUT_TENSOR
argIndex: 2
}
+}
+opList {
+ name: "reduce_normmax"
argDescriptor {
name: "dimensions"
argType: INT64
}
- argDescriptor {
- name: "dtype"
- argType: DATA_TYPE
- }
-}
-opList {
- name: "reduce_normmax"
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -14755,14 +14909,18 @@ opList {
argType: INPUT_TENSOR
argIndex: 1
}
- argDescriptor {
- name: "dimensions"
- argType: INT64
- }
opDeclarationType: LEGACY_XYZ
}
opList {
name: "reduce_prod"
+ argDescriptor {
+ name: "dimensions"
+ argType: INT64
+ }
+ argDescriptor {
+ name: "dtype"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -14780,17 +14938,17 @@ opList {
argType: INPUT_TENSOR
argIndex: 1
}
- argDescriptor {
- name: "dimensions"
- argType: INT64
- }
- argDescriptor {
- name: "dtype"
- argType: DATA_TYPE
- }
}
opList {
name: "reduce_prod_bp"
+ argDescriptor {
+ name: "dimensions"
+ argType: INT64
+ }
+ argDescriptor {
+ name: "dtype"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "gradI"
argType: OUTPUT_TENSOR
@@ -14813,17 +14971,17 @@ opList {
argType: INPUT_TENSOR
argIndex: 2
}
- argDescriptor {
- name: "dimensions"
- argType: INT64
- }
- argDescriptor {
- name: "dtype"
- argType: DATA_TYPE
- }
}
opList {
name: "reduce_sqnorm"
+ argDescriptor {
+ name: "dimensions"
+ argType: INT64
+ }
+ argDescriptor {
+ name: "dtype"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "gradI"
argType: OUTPUT_TENSOR
@@ -14841,6 +14999,9 @@ opList {
argType: INPUT_TENSOR
argIndex: 1
}
+}
+opList {
+ name: "reduce_sqnorm_bp"
argDescriptor {
name: "dimensions"
argType: INT64
@@ -14849,9 +15010,6 @@ opList {
name: "dtype"
argType: DATA_TYPE
}
-}
-opList {
- name: "reduce_sqnorm_bp"
argDescriptor {
name: "gradI"
argType: OUTPUT_TENSOR
@@ -14874,34 +15032,17 @@ opList {
argType: INPUT_TENSOR
argIndex: 2
}
- argDescriptor {
- name: "dimensions"
- argType: INT64
- }
- argDescriptor {
- name: "dtype"
- argType: DATA_TYPE
- }
}
opList {
name: "reduce_stdev"
- argDescriptor {
- name: "outputs"
- argType: OUTPUT_TENSOR
- }
- argDescriptor {
- name: "input"
- argType: INPUT_TENSOR
- }
- argDescriptor {
- name: "dimensions"
- argType: INPUT_TENSOR
- argIndex: 1
- }
argDescriptor {
name: "dtype"
argType: DATA_TYPE
}
+ argDescriptor {
+ name: "outputs"
+ argType: OUTPUT_TENSOR
+ }
argDescriptor {
name: "keepDims"
argType: DOUBLE
@@ -14911,9 +15052,26 @@ opList {
argType: DOUBLE
argIndex: 1
}
+ argDescriptor {
+ name: "input"
+ argType: INPUT_TENSOR
+ }
+ argDescriptor {
+ name: "dimensions"
+ argType: INPUT_TENSOR
+ argIndex: 1
+ }
}
opList {
name: "reduce_stdev_bp"
+ argDescriptor {
+ name: "dimensions"
+ argType: INT64
+ }
+ argDescriptor {
+ name: "dtype"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "gradI"
argType: OUTPUT_TENSOR
@@ -14927,6 +15085,15 @@ opList {
argType: BOOL
argIndex: 1
}
+ argDescriptor {
+ name: "keepDims"
+ argType: DOUBLE
+ }
+ argDescriptor {
+ name: "biasCorrected"
+ argType: DOUBLE
+ argIndex: 1
+ }
argDescriptor {
name: "input"
argType: INPUT_TENSOR
@@ -14941,26 +15108,17 @@ opList {
argType: INPUT_TENSOR
argIndex: 2
}
- argDescriptor {
- name: "dimensions"
- argType: INT64
- }
- argDescriptor {
- name: "dtype"
- argType: DATA_TYPE
- }
- argDescriptor {
- name: "keepDims"
- argType: DOUBLE
- }
- argDescriptor {
- name: "biasCorrected"
- argType: DOUBLE
- argIndex: 1
- }
}
opList {
name: "reduce_sum"
+ argDescriptor {
+ name: "dimensions"
+ argType: INT64
+ }
+ argDescriptor {
+ name: "dtype"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -14978,6 +15136,9 @@ opList {
argType: INPUT_TENSOR
argIndex: 1
}
+}
+opList {
+ name: "reduce_sum_bp"
argDescriptor {
name: "dimensions"
argType: INT64
@@ -14986,9 +15147,6 @@ opList {
name: "dtype"
argType: DATA_TYPE
}
-}
-opList {
- name: "reduce_sum_bp"
argDescriptor {
name: "gradI"
argType: OUTPUT_TENSOR
@@ -15011,34 +15169,17 @@ opList {
argType: INPUT_TENSOR
argIndex: 2
}
- argDescriptor {
- name: "dimensions"
- argType: INT64
- }
- argDescriptor {
- name: "dtype"
- argType: DATA_TYPE
- }
}
opList {
name: "reduce_variance"
- argDescriptor {
- name: "outputs"
- argType: OUTPUT_TENSOR
- }
- argDescriptor {
- name: "input"
- argType: INPUT_TENSOR
- }
- argDescriptor {
- name: "dimensions"
- argType: INPUT_TENSOR
- argIndex: 1
- }
argDescriptor {
name: "dtype"
argType: DATA_TYPE
}
+ argDescriptor {
+ name: "outputs"
+ argType: OUTPUT_TENSOR
+ }
argDescriptor {
name: "keepDims"
argType: DOUBLE
@@ -15048,9 +15189,26 @@ opList {
argType: DOUBLE
argIndex: 1
}
+ argDescriptor {
+ name: "input"
+ argType: INPUT_TENSOR
+ }
+ argDescriptor {
+ name: "dimensions"
+ argType: INPUT_TENSOR
+ argIndex: 1
+ }
}
opList {
name: "reduce_variance_bp"
+ argDescriptor {
+ name: "dimensions"
+ argType: INT64
+ }
+ argDescriptor {
+ name: "dtype"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "gradI"
argType: OUTPUT_TENSOR
@@ -15064,6 +15222,15 @@ opList {
argType: BOOL
argIndex: 1
}
+ argDescriptor {
+ name: "keepDims"
+ argType: DOUBLE
+ }
+ argDescriptor {
+ name: "biasCorrected"
+ argType: DOUBLE
+ argIndex: 1
+ }
argDescriptor {
name: "input"
argType: INPUT_TENSOR
@@ -15078,23 +15245,6 @@ opList {
argType: INPUT_TENSOR
argIndex: 2
}
- argDescriptor {
- name: "dimensions"
- argType: INT64
- }
- argDescriptor {
- name: "dtype"
- argType: DATA_TYPE
- }
- argDescriptor {
- name: "keepDims"
- argType: DOUBLE
- }
- argDescriptor {
- name: "biasCorrected"
- argType: DOUBLE
- argIndex: 1
- }
}
opList {
name: "relu"
@@ -15106,18 +15256,22 @@ opList {
name: "inPlace"
argType: BOOL
}
- argDescriptor {
- name: "input"
- argType: INPUT_TENSOR
- }
argDescriptor {
name: "cutoff"
argType: DOUBLE
}
+ argDescriptor {
+ name: "input"
+ argType: INPUT_TENSOR
+ }
opDeclarationType: CONFIGURABLE_OP_IMPL
}
opList {
name: "relu6"
+ argDescriptor {
+ name: "dtype"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -15126,18 +15280,14 @@ opList {
name: "inPlace"
argType: BOOL
}
- argDescriptor {
- name: "input"
- argType: INPUT_TENSOR
- }
- argDescriptor {
- name: "dtype"
- argType: DATA_TYPE
- }
argDescriptor {
name: "cutoff"
argType: DOUBLE
}
+ argDescriptor {
+ name: "input"
+ argType: INPUT_TENSOR
+ }
opDeclarationType: CONFIGURABLE_OP_IMPL
}
opList {
@@ -15146,6 +15296,10 @@ opList {
name: "gradI"
argType: OUTPUT_TENSOR
}
+ argDescriptor {
+ name: "cutoff"
+ argType: DOUBLE
+ }
argDescriptor {
name: "input"
argType: INPUT_TENSOR
@@ -15155,10 +15309,6 @@ opList {
argType: INPUT_TENSOR
argIndex: 1
}
- argDescriptor {
- name: "cutoff"
- argType: DOUBLE
- }
opDeclarationType: CONFIGURABLE_OP_IMPL
}
opList {
@@ -15167,6 +15317,10 @@ opList {
name: "outputs"
argType: OUTPUT_TENSOR
}
+ argDescriptor {
+ name: "scalar"
+ argType: DOUBLE
+ }
argDescriptor {
name: "input"
argType: INPUT_TENSOR
@@ -15176,14 +15330,14 @@ opList {
argType: INPUT_TENSOR
argIndex: 1
}
- argDescriptor {
- name: "scalar"
- argType: DOUBLE
- }
opDeclarationType: CONFIGURABLE_OP_IMPL
}
opList {
name: "relu_layer"
+ argDescriptor {
+ name: "dtype"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -15202,13 +15356,13 @@ opList {
argType: INPUT_TENSOR
argIndex: 2
}
- argDescriptor {
- name: "dtype"
- argType: DATA_TYPE
- }
}
opList {
name: "remainder"
+ argDescriptor {
+ name: "dataType"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -15226,10 +15380,6 @@ opList {
argType: INPUT_TENSOR
argIndex: 1
}
- argDescriptor {
- name: "dataType"
- argType: DATA_TYPE
- }
opDeclarationType: LEGACY_XYZ
}
opList {
@@ -15250,14 +15400,6 @@ opList {
}
opList {
name: "repeat"
- argDescriptor {
- name: "outputs"
- argType: OUTPUT_TENSOR
- }
- argDescriptor {
- name: "input"
- argType: INPUT_TENSOR
- }
argDescriptor {
name: "dimensions"
argType: INT64
@@ -15266,6 +15408,14 @@ opList {
name: "dtype"
argType: DATA_TYPE
}
+ argDescriptor {
+ name: "outputs"
+ argType: OUTPUT_TENSOR
+ }
+ argDescriptor {
+ name: "input"
+ argType: INPUT_TENSOR
+ }
}
opList {
name: "replace_nans"
@@ -15274,21 +15424,25 @@ opList {
argType: OUTPUT_TENSOR
}
argDescriptor {
- name: "inPlace"
- argType: BOOL
+ name: "set"
+ argType: DOUBLE
}
argDescriptor {
name: "input"
argType: INPUT_TENSOR
}
- argDescriptor {
- name: "set"
- argType: DOUBLE
- }
opDeclarationType: LEGACY_XYZ
}
opList {
name: "reshape"
+ argDescriptor {
+ name: "shapeArr"
+ argType: INT64
+ }
+ argDescriptor {
+ name: "dtype"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -15302,17 +15456,13 @@ opList {
argType: INPUT_TENSOR
argIndex: 1
}
- argDescriptor {
- name: "shapeArr"
- argType: INT64
- }
+}
+opList {
+ name: "reshapeas"
argDescriptor {
name: "dtype"
argType: DATA_TYPE
}
-}
-opList {
- name: "reshapeas"
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -15326,13 +15476,22 @@ opList {
argType: INPUT_TENSOR
argIndex: 1
}
+}
+opList {
+ name: "resize_area"
+ argDescriptor {
+ name: "height"
+ argType: INT64
+ }
+ argDescriptor {
+ name: "width"
+ argType: INT64
+ argIndex: 1
+ }
argDescriptor {
name: "dtype"
argType: DATA_TYPE
}
-}
-opList {
- name: "resize_area"
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -15350,22 +15509,13 @@ opList {
argType: INPUT_TENSOR
argIndex: 1
}
- argDescriptor {
- name: "height"
- argType: INT64
- }
- argDescriptor {
- name: "width"
- argType: INT64
- argIndex: 1
- }
+}
+opList {
+ name: "resize_bicubic"
argDescriptor {
name: "dtype"
argType: DATA_TYPE
}
-}
-opList {
- name: "resize_bicubic"
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -15388,35 +15538,9 @@ opList {
argType: INPUT_TENSOR
argIndex: 1
}
- argDescriptor {
- name: "dtype"
- argType: DATA_TYPE
- }
}
opList {
name: "resize_bilinear"
- argDescriptor {
- name: "outputs"
- argType: OUTPUT_TENSOR
- }
- argDescriptor {
- name: "alignCorners"
- argType: BOOL
- }
- argDescriptor {
- name: "halfPixelCenters"
- argType: BOOL
- argIndex: 1
- }
- argDescriptor {
- name: "image"
- argType: INPUT_TENSOR
- }
- argDescriptor {
- name: "newImageSize"
- argType: INPUT_TENSOR
- argIndex: 1
- }
argDescriptor {
name: "height"
argType: INT64
@@ -15430,9 +15554,35 @@ opList {
name: "dtype"
argType: DATA_TYPE
}
+ argDescriptor {
+ name: "outputs"
+ argType: OUTPUT_TENSOR
+ }
+ argDescriptor {
+ name: "alignCorners"
+ argType: BOOL
+ }
+ argDescriptor {
+ name: "halfPixelCenter"
+ argType: BOOL
+ argIndex: 1
+ }
+ argDescriptor {
+ name: "image"
+ argType: INPUT_TENSOR
+ }
+ argDescriptor {
+ name: "newImageSize"
+ argType: INPUT_TENSOR
+ argIndex: 1
+ }
}
opList {
name: "resize_images"
+ argDescriptor {
+ name: "dtype"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -15460,13 +15610,22 @@ opList {
argType: INPUT_TENSOR
argIndex: 2
}
+}
+opList {
+ name: "resize_nearest_neighbor"
+ argDescriptor {
+ name: "height"
+ argType: INT64
+ }
+ argDescriptor {
+ name: "width"
+ argType: INT64
+ argIndex: 1
+ }
argDescriptor {
name: "dtype"
argType: DATA_TYPE
}
-}
-opList {
- name: "resize_nearest_neighbor"
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -15489,19 +15648,6 @@ opList {
argType: INPUT_TENSOR
argIndex: 1
}
- argDescriptor {
- name: "height"
- argType: INT64
- }
- argDescriptor {
- name: "width"
- argType: INT64
- argIndex: 1
- }
- argDescriptor {
- name: "dtype"
- argType: DATA_TYPE
- }
}
opList {
name: "restorev2"
@@ -15512,6 +15658,10 @@ opList {
}
opList {
name: "reverse"
+ argDescriptor {
+ name: "dimensions"
+ argType: INT64
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -15525,14 +15675,18 @@ opList {
argType: INPUT_TENSOR
argIndex: 1
}
- argDescriptor {
- name: "dimensions"
- argType: INT64
- }
opDeclarationType: CONFIGURABLE_OP_IMPL
}
opList {
name: "reverse_bp"
+ argDescriptor {
+ name: "dimensions"
+ argType: INT64
+ }
+ argDescriptor {
+ name: "dtype"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -15551,30 +15705,9 @@ opList {
argType: INPUT_TENSOR
argIndex: 2
}
- argDescriptor {
- name: "dimensions"
- argType: INT64
- }
- argDescriptor {
- name: "dtype"
- argType: DATA_TYPE
- }
}
opList {
name: "reverse_sequence"
- argDescriptor {
- name: "outputs"
- argType: OUTPUT_TENSOR
- }
- argDescriptor {
- name: "input"
- argType: INPUT_TENSOR
- }
- argDescriptor {
- name: "seqLengths"
- argType: INPUT_TENSOR
- argIndex: 1
- }
argDescriptor {
name: "seqDim"
argType: INT64
@@ -15588,6 +15721,19 @@ opList {
name: "dtype"
argType: DATA_TYPE
}
+ argDescriptor {
+ name: "outputs"
+ argType: OUTPUT_TENSOR
+ }
+ argDescriptor {
+ name: "input"
+ argType: INPUT_TENSOR
+ }
+ argDescriptor {
+ name: "seqLengths"
+ argType: INPUT_TENSOR
+ argIndex: 1
+ }
}
opList {
name: "reverse_v2"
@@ -15602,6 +15748,10 @@ opList {
}
opList {
name: "reversedivide"
+ argDescriptor {
+ name: "dataType"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -15619,14 +15769,14 @@ opList {
argType: INPUT_TENSOR
argIndex: 1
}
- argDescriptor {
- name: "dataType"
- argType: DATA_TYPE
- }
opDeclarationType: BROADCASTABLE_OP_IMPL
}
opList {
name: "reversedivide_bp"
+ argDescriptor {
+ name: "dtype"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "gradX"
argType: OUTPUT_TENSOR
@@ -15654,10 +15804,6 @@ opList {
argType: INPUT_TENSOR
argIndex: 2
}
- argDescriptor {
- name: "dtype"
- argType: DATA_TYPE
- }
}
opList {
name: "reversemod"
@@ -15678,6 +15824,10 @@ opList {
}
opList {
name: "reversemod_bp"
+ argDescriptor {
+ name: "dtype"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "gradX"
argType: OUTPUT_TENSOR
@@ -15701,13 +15851,13 @@ opList {
argType: INPUT_TENSOR
argIndex: 2
}
- argDescriptor {
- name: "dtype"
- argType: DATA_TYPE
- }
}
opList {
name: "reversesubtract"
+ argDescriptor {
+ name: "dataType"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -15725,14 +15875,14 @@ opList {
argType: INPUT_TENSOR
argIndex: 1
}
- argDescriptor {
- name: "dataType"
- argType: DATA_TYPE
- }
opDeclarationType: BROADCASTABLE_OP_IMPL
}
opList {
name: "reversesubtract_bp"
+ argDescriptor {
+ name: "dtype"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "gradX"
argType: OUTPUT_TENSOR
@@ -15760,21 +15910,9 @@ opList {
argType: INPUT_TENSOR
argIndex: 2
}
- argDescriptor {
- name: "dtype"
- argType: DATA_TYPE
- }
}
opList {
name: "rgb_to_grs"
- argDescriptor {
- name: "outputs"
- argType: OUTPUT_TENSOR
- }
- argDescriptor {
- name: "input"
- argType: INPUT_TENSOR
- }
argDescriptor {
name: "dimC"
argType: INT64
@@ -15783,9 +15921,21 @@ opList {
name: "dtype"
argType: DATA_TYPE
}
+ argDescriptor {
+ name: "outputs"
+ argType: OUTPUT_TENSOR
+ }
+ argDescriptor {
+ name: "input"
+ argType: INPUT_TENSOR
+ }
}
opList {
name: "rgb_to_hsv"
+ argDescriptor {
+ name: "dimC"
+ argType: INT64
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -15794,14 +15944,14 @@ opList {
name: "input"
argType: INPUT_TENSOR
}
- argDescriptor {
- name: "dimC"
- argType: INT64
- }
opDeclarationType: CONFIGURABLE_OP_IMPL
}
opList {
name: "rgb_to_yiq"
+ argDescriptor {
+ name: "dimC"
+ argType: INT64
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -15810,14 +15960,14 @@ opList {
name: "input"
argType: INPUT_TENSOR
}
- argDescriptor {
- name: "dimC"
- argType: INT64
- }
opDeclarationType: CONFIGURABLE_OP_IMPL
}
opList {
name: "rgb_to_yuv"
+ argDescriptor {
+ name: "dimC"
+ argType: INT64
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -15826,14 +15976,14 @@ opList {
name: "input"
argType: INPUT_TENSOR
}
- argDescriptor {
- name: "dimC"
- argType: INT64
- }
opDeclarationType: CONFIGURABLE_OP_IMPL
}
opList {
name: "rint"
+ argDescriptor {
+ name: "dataType"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -15846,10 +15996,6 @@ opList {
name: "input"
argType: INPUT_TENSOR
}
- argDescriptor {
- name: "dataType"
- argType: DATA_TYPE
- }
opDeclarationType: OP_IMPL
}
opList {
@@ -15863,6 +16009,20 @@ opList {
argType: OUTPUT_TENSOR
argIndex: 1
}
+ argDescriptor {
+ name: "dLr"
+ argType: DOUBLE
+ }
+ argDescriptor {
+ name: "dRmsDecay"
+ argType: DOUBLE
+ argIndex: 1
+ }
+ argDescriptor {
+ name: "dEpsilon"
+ argType: DOUBLE
+ argIndex: 2
+ }
argDescriptor {
name: "gradient"
argType: INPUT_TENSOR
@@ -15887,24 +16047,14 @@ opList {
argType: INPUT_TENSOR
argIndex: 4
}
- argDescriptor {
- name: "dLr"
- argType: DOUBLE
- }
- argDescriptor {
- name: "dRmsDecay"
- argType: DOUBLE
- argIndex: 1
- }
- argDescriptor {
- name: "dEpsilon"
- argType: DOUBLE
- argIndex: 2
- }
opDeclarationType: CONFIGURABLE_OP_IMPL
}
opList {
name: "roll"
+ argDescriptor {
+ name: "shift"
+ argType: INT64
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -15923,14 +16073,14 @@ opList {
argType: INPUT_TENSOR
argIndex: 2
}
- argDescriptor {
- name: "shift"
- argType: INT64
- }
opDeclarationType: CONFIGURABLE_OP_IMPL
}
opList {
name: "round"
+ argDescriptor {
+ name: "dataType"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -15943,14 +16093,14 @@ opList {
name: "input"
argType: INPUT_TENSOR
}
- argDescriptor {
- name: "dataType"
- argType: DATA_TYPE
- }
opDeclarationType: LEGACY_XYZ
}
opList {
name: "rshift_bits"
+ argDescriptor {
+ name: "dataType"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -15968,14 +16118,14 @@ opList {
argType: INPUT_TENSOR
argIndex: 1
}
- argDescriptor {
- name: "dataType"
- argType: DATA_TYPE
- }
opDeclarationType: BROADCASTABLE_OP_IMPL
}
opList {
name: "rsqrt"
+ argDescriptor {
+ name: "dataType"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -15988,10 +16138,6 @@ opList {
name: "input"
argType: INPUT_TENSOR
}
- argDescriptor {
- name: "dataType"
- argType: DATA_TYPE
- }
opDeclarationType: LEGACY_XYZ
}
opList {
@@ -16098,6 +16244,10 @@ opList {
}
opList {
name: "scatter_list"
+ argDescriptor {
+ name: "dtype"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -16116,10 +16266,6 @@ opList {
argType: INPUT_TENSOR
argIndex: 2
}
- argDescriptor {
- name: "dtype"
- argType: DATA_TYPE
- }
opDeclarationType: LIST_OP_IMPL
}
opList {
@@ -16217,6 +16363,10 @@ opList {
}
opList {
name: "scatter_nd"
+ argDescriptor {
+ name: "dtype"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -16244,10 +16394,6 @@ opList {
argType: INPUT_TENSOR
argIndex: 2
}
- argDescriptor {
- name: "dtype"
- argType: DATA_TYPE
- }
}
opList {
name: "scatter_nd_add"
@@ -16421,35 +16567,13 @@ opList {
}
argDescriptor {
name: "indices"
- argType: INT64
- }
- argDescriptor {
- name: "dimension"
- argType: INT64
- argIndex: 1
+ argType: INPUT_TENSOR
+ argIndex: 2
}
opDeclarationType: CONFIGURABLE_OP_IMPL
}
opList {
name: "sconv2d"
- argDescriptor {
- name: "*output"
- argType: OUTPUT_TENSOR
- }
- argDescriptor {
- name: "*input"
- argType: INPUT_TENSOR
- }
- argDescriptor {
- name: "*weightsDepth"
- argType: INPUT_TENSOR
- argIndex: 1
- }
- argDescriptor {
- name: "bias"
- argType: INPUT_TENSOR
- argIndex: 2
- }
argDescriptor {
name: "kH"
argType: INT64
@@ -16508,9 +16632,85 @@ opList {
name: "dtype"
argType: DATA_TYPE
}
+ argDescriptor {
+ name: "*output"
+ argType: OUTPUT_TENSOR
+ }
+ argDescriptor {
+ name: "*input"
+ argType: INPUT_TENSOR
+ }
+ argDescriptor {
+ name: "*weightsDepth"
+ argType: INPUT_TENSOR
+ argIndex: 1
+ }
+ argDescriptor {
+ name: "bias"
+ argType: INPUT_TENSOR
+ argIndex: 2
+ }
}
opList {
name: "sconv2d_bp"
+ argDescriptor {
+ name: "kH"
+ argType: INT64
+ }
+ argDescriptor {
+ name: "kW"
+ argType: INT64
+ argIndex: 1
+ }
+ argDescriptor {
+ name: "sH"
+ argType: INT64
+ argIndex: 2
+ }
+ argDescriptor {
+ name: "sW"
+ argType: INT64
+ argIndex: 3
+ }
+ argDescriptor {
+ name: "pH"
+ argType: INT64
+ argIndex: 4
+ }
+ argDescriptor {
+ name: "pW"
+ argType: INT64
+ argIndex: 5
+ }
+ argDescriptor {
+ name: "dH"
+ argType: INT64
+ argIndex: 6
+ }
+ argDescriptor {
+ name: "dW"
+ argType: INT64
+ argIndex: 7
+ }
+ argDescriptor {
+ name: "isSameMode"
+ argType: INT64
+ argIndex: 8
+ }
+ argDescriptor {
+ name: "isNCHW"
+ argType: INT64
+ argIndex: 9
+ }
+ argDescriptor {
+ name: "wFormat"
+ argType: INT64
+ argIndex: 10
+ }
+ argDescriptor {
+ name: "dtype"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "*gradI"
argType: OUTPUT_TENSOR
@@ -16545,71 +16745,17 @@ opList {
argIndex: 2
}
argDescriptor {
- name: "weightsPoint"
+ name: "bias"
argType: INPUT_TENSOR
argIndex: 3
}
- argDescriptor {
- name: "kH"
- argType: INT64
- }
- argDescriptor {
- name: "kW"
- argType: INT64
- argIndex: 1
- }
- argDescriptor {
- name: "sH"
- argType: INT64
- argIndex: 2
- }
- argDescriptor {
- name: "sW"
- argType: INT64
- argIndex: 3
- }
- argDescriptor {
- name: "pH"
- argType: INT64
- argIndex: 4
- }
- argDescriptor {
- name: "pW"
- argType: INT64
- argIndex: 5
- }
- argDescriptor {
- name: "dH"
- argType: INT64
- argIndex: 6
- }
- argDescriptor {
- name: "dW"
- argType: INT64
- argIndex: 7
- }
- argDescriptor {
- name: "isSameMode"
- argType: INT64
- argIndex: 8
- }
- argDescriptor {
- name: "isNCHW"
- argType: INT64
- argIndex: 9
- }
- argDescriptor {
- name: "wFormat"
- argType: INT64
- argIndex: 10
- }
- argDescriptor {
- name: "dtype"
- argType: DATA_TYPE
- }
}
opList {
name: "segment_max"
+ argDescriptor {
+ name: "dtype"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "segmentedOutput"
argType: OUTPUT_TENSOR
@@ -16623,13 +16769,13 @@ opList {
argType: INPUT_TENSOR
argIndex: 1
}
- argDescriptor {
- name: "dtype"
- argType: DATA_TYPE
- }
}
opList {
name: "segment_max_bp"
+ argDescriptor {
+ name: "dtype"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -16653,13 +16799,13 @@ opList {
argType: INPUT_TENSOR
argIndex: 2
}
- argDescriptor {
- name: "dtype"
- argType: DATA_TYPE
- }
}
opList {
name: "segment_mean"
+ argDescriptor {
+ name: "dtype"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "segmentedOutput"
argType: OUTPUT_TENSOR
@@ -16673,13 +16819,13 @@ opList {
argType: INPUT_TENSOR
argIndex: 1
}
- argDescriptor {
- name: "dtype"
- argType: DATA_TYPE
- }
}
opList {
name: "segment_mean_bp"
+ argDescriptor {
+ name: "dtype"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -16703,13 +16849,13 @@ opList {
argType: INPUT_TENSOR
argIndex: 2
}
- argDescriptor {
- name: "dtype"
- argType: DATA_TYPE
- }
}
opList {
name: "segment_min"
+ argDescriptor {
+ name: "dtype"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "segmentedOutput"
argType: OUTPUT_TENSOR
@@ -16723,13 +16869,13 @@ opList {
argType: INPUT_TENSOR
argIndex: 1
}
- argDescriptor {
- name: "dtype"
- argType: DATA_TYPE
- }
}
opList {
name: "segment_min_bp"
+ argDescriptor {
+ name: "dtype"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -16753,13 +16899,13 @@ opList {
argType: INPUT_TENSOR
argIndex: 2
}
- argDescriptor {
- name: "dtype"
- argType: DATA_TYPE
- }
}
opList {
name: "segment_prod"
+ argDescriptor {
+ name: "dtype"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "segmentedOutput"
argType: OUTPUT_TENSOR
@@ -16773,13 +16919,13 @@ opList {
argType: INPUT_TENSOR
argIndex: 1
}
+}
+opList {
+ name: "segment_prod_bp"
argDescriptor {
name: "dtype"
argType: DATA_TYPE
}
-}
-opList {
- name: "segment_prod_bp"
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -16803,13 +16949,13 @@ opList {
argType: INPUT_TENSOR
argIndex: 2
}
+}
+opList {
+ name: "segment_sum"
argDescriptor {
name: "dtype"
argType: DATA_TYPE
}
-}
-opList {
- name: "segment_sum"
argDescriptor {
name: "segmentedOutput"
argType: OUTPUT_TENSOR
@@ -16823,13 +16969,13 @@ opList {
argType: INPUT_TENSOR
argIndex: 1
}
+}
+opList {
+ name: "segment_sum_bp"
argDescriptor {
name: "dtype"
argType: DATA_TYPE
}
-}
-opList {
- name: "segment_sum_bp"
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -16848,13 +16994,13 @@ opList {
argType: INPUT_TENSOR
argIndex: 2
}
+}
+opList {
+ name: "select"
argDescriptor {
name: "dtype"
argType: DATA_TYPE
}
-}
-opList {
- name: "select"
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -16873,13 +17019,13 @@ opList {
argType: INPUT_TENSOR
argIndex: 2
}
- argDescriptor {
- name: "dtype"
- argType: DATA_TYPE
- }
}
opList {
name: "selu"
+ argDescriptor {
+ name: "dataType"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -16892,10 +17038,6 @@ opList {
name: "input"
argType: INPUT_TENSOR
}
- argDescriptor {
- name: "dataType"
- argType: DATA_TYPE
- }
opDeclarationType: CONFIGURABLE_OP_IMPL
}
opList {
@@ -16917,6 +17059,10 @@ opList {
}
opList {
name: "seluderivative"
+ argDescriptor {
+ name: "dataType"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -16929,14 +17075,18 @@ opList {
name: "input"
argType: INPUT_TENSOR
}
- argDescriptor {
- name: "dataType"
- argType: DATA_TYPE
- }
opDeclarationType: LEGACY_XYZ
}
opList {
name: "sequence_mask"
+ argDescriptor {
+ name: "maxInd"
+ argType: INT64
+ }
+ argDescriptor {
+ name: "dtype"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -16954,22 +17104,13 @@ opList {
argType: INPUT_TENSOR
argIndex: 1
}
- argDescriptor {
- name: "maxInd"
- argType: INT64
- }
- argDescriptor {
- name: "dataType"
- argType: INT64
- argIndex: 1
- }
- argDescriptor {
- name: "dtype"
- argType: DATA_TYPE
- }
}
opList {
name: "set"
+ argDescriptor {
+ name: "dataType"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -16983,10 +17124,6 @@ opList {
argType: INPUT_TENSOR
argIndex: 1
}
- argDescriptor {
- name: "dataType"
- argType: DATA_TYPE
- }
opDeclarationType: LEGACY_XYZ
}
opList {
@@ -17008,14 +17145,6 @@ opList {
}
opList {
name: "set_seed"
- argDescriptor {
- name: "outputs"
- argType: OUTPUT_TENSOR
- }
- argDescriptor {
- name: "input"
- argType: INPUT_TENSOR
- }
argDescriptor {
name: "seed"
argType: INT64
@@ -17024,9 +17153,21 @@ opList {
name: "dtype"
argType: DATA_TYPE
}
+ argDescriptor {
+ name: "outputs"
+ argType: OUTPUT_TENSOR
+ }
+ argDescriptor {
+ name: "input"
+ argType: INPUT_TENSOR
+ }
}
opList {
name: "setrange"
+ argDescriptor {
+ name: "dataType"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -17035,14 +17176,6 @@ opList {
name: "inPlace"
argType: BOOL
}
- argDescriptor {
- name: "input"
- argType: INPUT_TENSOR
- }
- argDescriptor {
- name: "dataType"
- argType: DATA_TYPE
- }
argDescriptor {
name: "min"
argType: DOUBLE
@@ -17052,6 +17185,10 @@ opList {
argType: DOUBLE
argIndex: 1
}
+ argDescriptor {
+ name: "input"
+ argType: INPUT_TENSOR
+ }
opDeclarationType: LEGACY_XYZ
}
opList {
@@ -17076,6 +17213,10 @@ opList {
name: "outputs"
argType: OUTPUT_TENSOR
}
+ argDescriptor {
+ name: "lr"
+ argType: DOUBLE
+ }
argDescriptor {
name: "input"
argType: INPUT_TENSOR
@@ -17085,14 +17226,14 @@ opList {
argType: INPUT_TENSOR
argIndex: 1
}
- argDescriptor {
- name: "lr"
- argType: DOUBLE
- }
opDeclarationType: CONFIGURABLE_OP_IMPL
}
opList {
name: "shannonentropy"
+ argDescriptor {
+ name: "dimensions"
+ argType: INT64
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -17110,14 +17251,14 @@ opList {
argType: INPUT_TENSOR
argIndex: 1
}
- argDescriptor {
- name: "dimensions"
- argType: INT64
- }
opDeclarationType: LEGACY_XYZ
}
opList {
name: "shape_of"
+ argDescriptor {
+ name: "dtype"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -17130,13 +17271,13 @@ opList {
name: "input"
argType: INPUT_TENSOR
}
- argDescriptor {
- name: "dtype"
- argType: DATA_TYPE
- }
}
opList {
name: "shapes_of"
+ argDescriptor {
+ name: "dtype"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -17149,13 +17290,13 @@ opList {
name: "input"
argType: INPUT_TENSOR
}
- argDescriptor {
- name: "dtype"
- argType: DATA_TYPE
- }
}
opList {
name: "shift_bits"
+ argDescriptor {
+ name: "dataType"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -17173,18 +17314,26 @@ opList {
argType: INPUT_TENSOR
argIndex: 1
}
- argDescriptor {
- name: "dataType"
- argType: DATA_TYPE
- }
opDeclarationType: BROADCASTABLE_OP_IMPL
}
opList {
name: "sigm_cross_entropy_loss"
+ argDescriptor {
+ name: "reductionMode"
+ argType: INT64
+ }
+ argDescriptor {
+ name: "dtype"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
}
+ argDescriptor {
+ name: "labelsSmoothing"
+ argType: DOUBLE
+ }
argDescriptor {
name: "logits"
argType: INPUT_TENSOR
@@ -17199,6 +17348,9 @@ opList {
argType: INPUT_TENSOR
argIndex: 2
}
+}
+opList {
+ name: "sigm_cross_entropy_loss_grad"
argDescriptor {
name: "reductionMode"
argType: INT64
@@ -17207,13 +17359,6 @@ opList {
name: "dtype"
argType: DATA_TYPE
}
- argDescriptor {
- name: "labelsSmoothing"
- argType: DOUBLE
- }
-}
-opList {
- name: "sigm_cross_entropy_loss_grad"
argDescriptor {
name: "dLdp"
argType: OUTPUT_TENSOR
@@ -17228,6 +17373,10 @@ opList {
argType: OUTPUT_TENSOR
argIndex: 2
}
+ argDescriptor {
+ name: "labelSmoothing"
+ argType: DOUBLE
+ }
argDescriptor {
name: "logits"
argType: INPUT_TENSOR
@@ -17242,21 +17391,13 @@ opList {
argType: INPUT_TENSOR
argIndex: 2
}
- argDescriptor {
- name: "reductionMode"
- argType: INT64
- }
- argDescriptor {
- name: "dtype"
- argType: DATA_TYPE
- }
- argDescriptor {
- name: "labelSmoothing"
- argType: DOUBLE
- }
}
opList {
name: "sigmoid"
+ argDescriptor {
+ name: "dataType"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -17269,10 +17410,6 @@ opList {
name: "input"
argType: INPUT_TENSOR
}
- argDescriptor {
- name: "dataType"
- argType: DATA_TYPE
- }
opDeclarationType: CONFIGURABLE_OP_IMPL
}
opList {
@@ -17294,6 +17431,10 @@ opList {
}
opList {
name: "sign"
+ argDescriptor {
+ name: "dataType"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -17306,14 +17447,14 @@ opList {
name: "input"
argType: INPUT_TENSOR
}
- argDescriptor {
- name: "dataType"
- argType: DATA_TYPE
- }
opDeclarationType: LEGACY_XYZ
}
opList {
name: "sin"
+ argDescriptor {
+ name: "dataType"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -17326,14 +17467,14 @@ opList {
name: "input"
argType: INPUT_TENSOR
}
- argDescriptor {
- name: "dataType"
- argType: DATA_TYPE
- }
opDeclarationType: LEGACY_XYZ
}
opList {
name: "sinh"
+ argDescriptor {
+ name: "dataType"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -17346,14 +17487,14 @@ opList {
name: "input"
argType: INPUT_TENSOR
}
- argDescriptor {
- name: "dataType"
- argType: DATA_TYPE
- }
opDeclarationType: LEGACY_XYZ
}
opList {
name: "size"
+ argDescriptor {
+ name: "dtype"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -17362,21 +17503,9 @@ opList {
name: "input"
argType: INPUT_TENSOR
}
- argDescriptor {
- name: "dtype"
- argType: DATA_TYPE
- }
}
opList {
name: "size_at"
- argDescriptor {
- name: "outputs"
- argType: OUTPUT_TENSOR
- }
- argDescriptor {
- name: "input"
- argType: INPUT_TENSOR
- }
argDescriptor {
name: "dimensions"
argType: INT64
@@ -17385,6 +17514,14 @@ opList {
name: "dtype"
argType: DATA_TYPE
}
+ argDescriptor {
+ name: "outputs"
+ argType: OUTPUT_TENSOR
+ }
+ argDescriptor {
+ name: "input"
+ argType: INPUT_TENSOR
+ }
}
opList {
name: "size_list"
@@ -17400,6 +17537,15 @@ opList {
}
opList {
name: "skipgram"
+ argDescriptor {
+ name: "numWorkers"
+ argType: INT64
+ }
+ argDescriptor {
+ name: "nsRounds"
+ argType: INT64
+ argIndex: 1
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -17472,19 +17618,18 @@ opList {
argType: INPUT_TENSOR
argIndex: 11
}
- argDescriptor {
- name: "numWorkers"
- argType: INT64
- }
- argDescriptor {
- name: "nsRounds"
- argType: INT64
- argIndex: 1
- }
opDeclarationType: CONFIGURABLE_OP_IMPL
}
opList {
name: "slice"
+ argDescriptor {
+ name: "size"
+ argType: INT64
+ }
+ argDescriptor {
+ name: "dtype"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -17503,6 +17648,9 @@ opList {
argType: INPUT_TENSOR
argIndex: 2
}
+}
+opList {
+ name: "slice_bp"
argDescriptor {
name: "size"
argType: INT64
@@ -17511,9 +17659,6 @@ opList {
name: "dtype"
argType: DATA_TYPE
}
-}
-opList {
- name: "slice_bp"
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -17537,17 +17682,17 @@ opList {
argType: INPUT_TENSOR
argIndex: 3
}
- argDescriptor {
- name: "size"
- argType: INT64
- }
- argDescriptor {
- name: "dtype"
- argType: DATA_TYPE
- }
}
opList {
name: "softmax"
+ argDescriptor {
+ name: "dimension"
+ argType: INT64
+ }
+ argDescriptor {
+ name: "dataType"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -17560,18 +17705,14 @@ opList {
name: "input"
argType: INPUT_TENSOR
}
- argDescriptor {
- name: "dimension"
- argType: INT64
- }
- argDescriptor {
- name: "dataType"
- argType: DATA_TYPE
- }
opDeclarationType: CONFIGURABLE_OP_IMPL
}
opList {
name: "softmax_bp"
+ argDescriptor {
+ name: "dimension"
+ argType: INT64
+ }
argDescriptor {
name: "gradI"
argType: OUTPUT_TENSOR
@@ -17585,18 +17726,26 @@ opList {
argType: INPUT_TENSOR
argIndex: 1
}
- argDescriptor {
- name: "dimension"
- argType: INT64
- }
opDeclarationType: CONFIGURABLE_OP_IMPL
}
opList {
name: "softmax_cross_entropy_loss"
+ argDescriptor {
+ name: "reductionMode"
+ argType: INT64
+ }
+ argDescriptor {
+ name: "dtype"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
}
+ argDescriptor {
+ name: "labelsSmoothing"
+ argType: DOUBLE
+ }
argDescriptor {
name: "logits"
argType: INPUT_TENSOR
@@ -17611,6 +17760,9 @@ opList {
argType: INPUT_TENSOR
argIndex: 2
}
+}
+opList {
+ name: "softmax_cross_entropy_loss_grad"
argDescriptor {
name: "reductionMode"
argType: INT64
@@ -17619,13 +17771,6 @@ opList {
name: "dtype"
argType: DATA_TYPE
}
- argDescriptor {
- name: "labelsSmoothing"
- argType: DOUBLE
- }
-}
-opList {
- name: "softmax_cross_entropy_loss_grad"
argDescriptor {
name: "dLdp"
argType: OUTPUT_TENSOR
@@ -17640,6 +17785,10 @@ opList {
argType: OUTPUT_TENSOR
argIndex: 2
}
+ argDescriptor {
+ name: "labelsSmoothing"
+ argType: DOUBLE
+ }
argDescriptor {
name: "logits"
argType: INPUT_TENSOR
@@ -17654,21 +17803,17 @@ opList {
argType: INPUT_TENSOR
argIndex: 2
}
+}
+opList {
+ name: "softmax_cross_entropy_loss_with_logits"
argDescriptor {
- name: "reductionMode"
+ name: "classesDim"
argType: INT64
}
argDescriptor {
name: "dtype"
argType: DATA_TYPE
}
- argDescriptor {
- name: "labelsSmoothing"
- argType: DOUBLE
- }
-}
-opList {
- name: "softmax_cross_entropy_loss_with_logits"
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -17682,6 +17827,9 @@ opList {
argType: INPUT_TENSOR
argIndex: 1
}
+}
+opList {
+ name: "softmax_cross_entropy_loss_with_logits_grad"
argDescriptor {
name: "classesDim"
argType: INT64
@@ -17690,9 +17838,6 @@ opList {
name: "dtype"
argType: DATA_TYPE
}
-}
-opList {
- name: "softmax_cross_entropy_loss_with_logits_grad"
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -17711,17 +17856,13 @@ opList {
argType: INPUT_TENSOR
argIndex: 1
}
- argDescriptor {
- name: "classesDim"
- argType: INT64
- }
- argDescriptor {
- name: "dtype"
- argType: DATA_TYPE
- }
}
opList {
name: "softplus"
+ argDescriptor {
+ name: "dataType"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -17734,10 +17875,6 @@ opList {
name: "input"
argType: INPUT_TENSOR
}
- argDescriptor {
- name: "dataType"
- argType: DATA_TYPE
- }
opDeclarationType: CONFIGURABLE_OP_IMPL
}
opList {
@@ -17759,6 +17896,10 @@ opList {
}
opList {
name: "softsign"
+ argDescriptor {
+ name: "dataType"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -17771,10 +17912,6 @@ opList {
name: "input"
argType: INPUT_TENSOR
}
- argDescriptor {
- name: "dataType"
- argType: DATA_TYPE
- }
opDeclarationType: CONFIGURABLE_OP_IMPL
}
opList {
@@ -17796,6 +17933,10 @@ opList {
}
opList {
name: "softsignderivative"
+ argDescriptor {
+ name: "dataType"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -17808,14 +17949,14 @@ opList {
name: "input"
argType: INPUT_TENSOR
}
- argDescriptor {
- name: "dataType"
- argType: DATA_TYPE
- }
opDeclarationType: LEGACY_XYZ
}
opList {
name: "solve"
+ argDescriptor {
+ name: "dtype"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -17838,13 +17979,13 @@ opList {
argType: INPUT_TENSOR
argIndex: 2
}
+}
+opList {
+ name: "solve_ls"
argDescriptor {
name: "dtype"
argType: DATA_TYPE
}
-}
-opList {
- name: "solve_ls"
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -17853,6 +17994,10 @@ opList {
name: "fastFlag"
argType: BOOL
}
+ argDescriptor {
+ name: "l2_factor"
+ argType: DOUBLE
+ }
argDescriptor {
name: "a"
argType: INPUT_TENSOR
@@ -17862,14 +18007,6 @@ opList {
argType: INPUT_TENSOR
argIndex: 1
}
- argDescriptor {
- name: "dtype"
- argType: DATA_TYPE
- }
- argDescriptor {
- name: "l2_factor"
- argType: DOUBLE
- }
}
opList {
name: "somepoolingpool2d"
@@ -17901,12 +18038,26 @@ opList {
opList {
name: "space_to_batch"
argDescriptor {
- name: "outputs"
- argType: OUTPUT_TENSOR
+ name: "blockSize"
+ argType: INT64
}
argDescriptor {
- name: "inPlace"
- argType: BOOL
+ name: "paddingTop"
+ argType: INT64
+ argIndex: 1
+ }
+ argDescriptor {
+ name: "paddingBottom"
+ argType: INT64
+ argIndex: 2
+ }
+ argDescriptor {
+ name: "dtype"
+ argType: DATA_TYPE
+ }
+ argDescriptor {
+ name: "outputs"
+ argType: OUTPUT_TENSOR
}
argDescriptor {
name: "input"
@@ -17917,17 +18068,17 @@ opList {
argType: INPUT_TENSOR
argIndex: 1
}
+}
+opList {
+ name: "space_to_batch_nd"
argDescriptor {
- name: "blockSize"
+ name: "blocks"
argType: INT64
}
argDescriptor {
name: "dtype"
argType: DATA_TYPE
}
-}
-opList {
- name: "space_to_batch_nd"
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -17950,25 +18101,9 @@ opList {
argType: INPUT_TENSOR
argIndex: 2
}
- argDescriptor {
- name: "blocks"
- argType: INT64
- }
- argDescriptor {
- name: "dtype"
- argType: DATA_TYPE
- }
}
opList {
name: "space_to_depth"
- argDescriptor {
- name: "outputs"
- argType: OUTPUT_TENSOR
- }
- argDescriptor {
- name: "input"
- argType: INPUT_TENSOR
- }
argDescriptor {
name: "block_size"
argType: INT64
@@ -17982,9 +18117,21 @@ opList {
name: "dtype"
argType: DATA_TYPE
}
+ argDescriptor {
+ name: "outputs"
+ argType: OUTPUT_TENSOR
+ }
+ argDescriptor {
+ name: "input"
+ argType: INPUT_TENSOR
+ }
}
opList {
name: "sparse_softmax_cross_entropy_loss_with_logits"
+ argDescriptor {
+ name: "dtype"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -17998,13 +18145,13 @@ opList {
argType: INPUT_TENSOR
argIndex: 1
}
+}
+opList {
+ name: "sparse_softmax_cross_entropy_loss_with_logits_grad"
argDescriptor {
name: "dtype"
argType: DATA_TYPE
}
-}
-opList {
- name: "sparse_softmax_cross_entropy_loss_with_logits_grad"
argDescriptor {
name: "dLdp"
argType: OUTPUT_TENSOR
@@ -18018,26 +18165,9 @@ opList {
argType: INPUT_TENSOR
argIndex: 1
}
- argDescriptor {
- name: "dtype"
- argType: DATA_TYPE
- }
}
opList {
name: "split"
- argDescriptor {
- name: "outArrs"
- argType: OUTPUT_TENSOR
- }
- argDescriptor {
- name: "a"
- argType: INPUT_TENSOR
- }
- argDescriptor {
- name: "b"
- argType: INPUT_TENSOR
- argIndex: 1
- }
argDescriptor {
name: "numSplit"
argType: INT64
@@ -18051,9 +18181,26 @@ opList {
name: "dtype"
argType: DATA_TYPE
}
+ argDescriptor {
+ name: "outArrs"
+ argType: OUTPUT_TENSOR
+ }
+ argDescriptor {
+ name: "a"
+ argType: INPUT_TENSOR
+ }
+ argDescriptor {
+ name: "b"
+ argType: INPUT_TENSOR
+ argIndex: 1
+ }
}
opList {
name: "split_list"
+ argDescriptor {
+ name: "dtype"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -18072,14 +18219,14 @@ opList {
argType: INPUT_TENSOR
argIndex: 2
}
- argDescriptor {
- name: "dtype"
- argType: DATA_TYPE
- }
opDeclarationType: LIST_OP_IMPL
}
opList {
name: "split_string"
+ argDescriptor {
+ name: "dtype"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -18093,13 +18240,22 @@ opList {
argType: INPUT_TENSOR
argIndex: 1
}
+}
+opList {
+ name: "split_v"
+ argDescriptor {
+ name: "dimensions"
+ argType: INT64
+ }
+ argDescriptor {
+ name: "numSplit"
+ argType: INT64
+ argIndex: 1
+ }
argDescriptor {
name: "dtype"
argType: DATA_TYPE
}
-}
-opList {
- name: "split_v"
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -18118,22 +18274,13 @@ opList {
argType: INPUT_TENSOR
argIndex: 2
}
- argDescriptor {
- name: "dimensions"
- argType: INT64
- }
- argDescriptor {
- name: "numSplit"
- argType: INT64
- argIndex: 1
- }
- argDescriptor {
- name: "dtype"
- argType: DATA_TYPE
- }
}
opList {
name: "sqrt"
+ argDescriptor {
+ name: "dataType"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -18146,10 +18293,6 @@ opList {
name: "input"
argType: INPUT_TENSOR
}
- argDescriptor {
- name: "dataType"
- argType: DATA_TYPE
- }
opDeclarationType: LEGACY_XYZ
}
opList {
@@ -18166,6 +18309,10 @@ opList {
}
opList {
name: "square"
+ argDescriptor {
+ name: "dataType"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -18178,14 +18325,14 @@ opList {
name: "input"
argType: INPUT_TENSOR
}
- argDescriptor {
- name: "dataType"
- argType: DATA_TYPE
- }
opDeclarationType: OP_IMPL
}
opList {
name: "squaredsubtract"
+ argDescriptor {
+ name: "dataType"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -18203,14 +18350,14 @@ opList {
argType: INPUT_TENSOR
argIndex: 1
}
- argDescriptor {
- name: "dataType"
- argType: DATA_TYPE
- }
opDeclarationType: BROADCASTABLE_OP_IMPL
}
opList {
name: "squaredsubtract_bp"
+ argDescriptor {
+ name: "dtype"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "gradX"
argType: OUTPUT_TENSOR
@@ -18234,13 +18381,17 @@ opList {
argType: INPUT_TENSOR
argIndex: 2
}
+}
+opList {
+ name: "squeeze"
+ argDescriptor {
+ name: "_a"
+ argType: INT64
+ }
argDescriptor {
name: "dtype"
argType: DATA_TYPE
}
-}
-opList {
- name: "squeeze"
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -18254,17 +18405,13 @@ opList {
argType: INPUT_TENSOR
argIndex: 1
}
- argDescriptor {
- name: "_a"
- argType: INT64
- }
+}
+opList {
+ name: "sru"
argDescriptor {
name: "dtype"
argType: DATA_TYPE
}
-}
-opList {
- name: "sru"
argDescriptor {
name: "h"
argType: OUTPUT_TENSOR
@@ -18298,13 +18445,13 @@ opList {
argType: INPUT_TENSOR
argIndex: 4
}
+}
+opList {
+ name: "sruCell"
argDescriptor {
name: "dtype"
argType: DATA_TYPE
}
-}
-opList {
- name: "sruCell"
argDescriptor {
name: "ht"
argType: OUTPUT_TENSOR
@@ -18333,13 +18480,13 @@ opList {
argType: INPUT_TENSOR
argIndex: 3
}
+}
+opList {
+ name: "sru_bi"
argDescriptor {
name: "dtype"
argType: DATA_TYPE
}
-}
-opList {
- name: "sru_bi"
argDescriptor {
name: "ht"
argType: OUTPUT_TENSOR
@@ -18373,13 +18520,13 @@ opList {
argType: INPUT_TENSOR
argIndex: 4
}
+}
+opList {
+ name: "sru_bi_bp"
argDescriptor {
name: "dtype"
argType: DATA_TYPE
}
-}
-opList {
- name: "sru_bi_bp"
argDescriptor {
name: "gradI"
argType: OUTPUT_TENSOR
@@ -18438,13 +18585,13 @@ opList {
argType: INPUT_TENSOR
argIndex: 7
}
+}
+opList {
+ name: "sru_bp"
argDescriptor {
name: "dtype"
argType: DATA_TYPE
}
-}
-opList {
- name: "sru_bp"
argDescriptor {
name: "gradX"
argType: OUTPUT_TENSOR
@@ -18503,13 +18650,13 @@ opList {
argType: INPUT_TENSOR
argIndex: 7
}
- argDescriptor {
- name: "dtype"
- argType: DATA_TYPE
- }
}
opList {
name: "stabilize"
+ argDescriptor {
+ name: "dataType"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -18518,14 +18665,6 @@ opList {
name: "inPlace"
argType: BOOL
}
- argDescriptor {
- name: "input"
- argType: INPUT_TENSOR
- }
- argDescriptor {
- name: "dataType"
- argType: DATA_TYPE
- }
argDescriptor {
name: "realMin"
argType: DOUBLE
@@ -18540,10 +18679,22 @@ opList {
argType: DOUBLE
argIndex: 2
}
+ argDescriptor {
+ name: "input"
+ argType: INPUT_TENSOR
+ }
opDeclarationType: LEGACY_XYZ
}
opList {
name: "stack"
+ argDescriptor {
+ name: "dimensions"
+ argType: INT64
+ }
+ argDescriptor {
+ name: "dtype"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -18557,17 +18708,13 @@ opList {
argType: INPUT_TENSOR
argIndex: 1
}
- argDescriptor {
- name: "dimensions"
- argType: INT64
- }
+}
+opList {
+ name: "stack_list"
argDescriptor {
name: "dtype"
argType: DATA_TYPE
}
-}
-opList {
- name: "stack_list"
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -18576,14 +18723,14 @@ opList {
name: "list"
argType: INPUT_TENSOR
}
- argDescriptor {
- name: "dtype"
- argType: DATA_TYPE
- }
opDeclarationType: LIST_OP_IMPL
}
opList {
name: "standardize"
+ argDescriptor {
+ name: "dimensions"
+ argType: INT64
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -18597,14 +18744,18 @@ opList {
argType: INPUT_TENSOR
argIndex: 1
}
- argDescriptor {
- name: "dimensions"
- argType: INT64
- }
opDeclarationType: CONFIGURABLE_OP_IMPL
}
opList {
name: "standardize_bp"
+ argDescriptor {
+ name: "dimensions"
+ argType: INT64
+ }
+ argDescriptor {
+ name: "dtype"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -18623,17 +18774,13 @@ opList {
argType: INPUT_TENSOR
argIndex: 2
}
- argDescriptor {
- name: "dimensions"
- argType: INT64
- }
+}
+opList {
+ name: "static_bidirectional_rnn"
argDescriptor {
name: "dtype"
argType: DATA_TYPE
}
-}
-opList {
- name: "static_bidirectional_rnn"
argDescriptor {
name: "h"
argType: OUTPUT_TENSOR
@@ -18692,13 +18839,13 @@ opList {
argType: INPUT_TENSOR
argIndex: 8
}
+}
+opList {
+ name: "static_rnn"
argDescriptor {
name: "dtype"
argType: DATA_TYPE
}
-}
-opList {
- name: "static_rnn"
argDescriptor {
name: "h"
argType: OUTPUT_TENSOR
@@ -18732,19 +18879,19 @@ opList {
argType: INPUT_TENSOR
argIndex: 4
}
- argDescriptor {
- name: "dtype"
- argType: DATA_TYPE
- }
}
opList {
name: "std"
+ argDescriptor {
+ name: "dimensions"
+ argType: INT64
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
}
argDescriptor {
- name: "newFormat"
+ name: "biasCorrected"
argType: BOOL
}
argDescriptor {
@@ -18756,10 +18903,6 @@ opList {
name: "input"
argType: INPUT_TENSOR
}
- argDescriptor {
- name: "dimensions"
- argType: INT64
- }
opDeclarationType: LEGACY_XYZ
}
opList {
@@ -18772,18 +18915,22 @@ opList {
name: "inPlace"
argType: BOOL
}
- argDescriptor {
- name: "input"
- argType: INPUT_TENSOR
- }
argDescriptor {
name: "cutoff"
argType: DOUBLE
}
+ argDescriptor {
+ name: "input"
+ argType: INPUT_TENSOR
+ }
opDeclarationType: LEGACY_XYZ
}
opList {
name: "stop_gradient"
+ argDescriptor {
+ name: "dataType"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -18796,37 +18943,10 @@ opList {
name: "input"
argType: INPUT_TENSOR
}
- argDescriptor {
- name: "dataType"
- argType: DATA_TYPE
- }
opDeclarationType: OP_IMPL
}
opList {
name: "strided_slice"
- argDescriptor {
- name: "outputs"
- argType: OUTPUT_TENSOR
- }
- argDescriptor {
- name: "input"
- argType: INPUT_TENSOR
- }
- argDescriptor {
- name: "v_begin"
- argType: INPUT_TENSOR
- argIndex: 1
- }
- argDescriptor {
- name: "v_end"
- argType: INPUT_TENSOR
- argIndex: 2
- }
- argDescriptor {
- name: "v_stride"
- argType: INPUT_TENSOR
- argIndex: 3
- }
argDescriptor {
name: "begin_mask"
argType: INT64
@@ -18855,9 +18975,60 @@ opList {
name: "dtype"
argType: DATA_TYPE
}
+ argDescriptor {
+ name: "outputs"
+ argType: OUTPUT_TENSOR
+ }
+ argDescriptor {
+ name: "input"
+ argType: INPUT_TENSOR
+ }
+ argDescriptor {
+ name: "v_begin"
+ argType: INPUT_TENSOR
+ argIndex: 1
+ }
+ argDescriptor {
+ name: "v_end"
+ argType: INPUT_TENSOR
+ argIndex: 2
+ }
+ argDescriptor {
+ name: "v_stride"
+ argType: INPUT_TENSOR
+ argIndex: 3
+ }
}
opList {
name: "strided_slice_bp"
+ argDescriptor {
+ name: "begin_mask"
+ argType: INT64
+ }
+ argDescriptor {
+ name: "ellipsis_mask"
+ argType: INT64
+ argIndex: 1
+ }
+ argDescriptor {
+ name: "end_mask"
+ argType: INT64
+ argIndex: 2
+ }
+ argDescriptor {
+ name: "new_axis_mask"
+ argType: INT64
+ argIndex: 3
+ }
+ argDescriptor {
+ name: "shrink_axis_mask"
+ argType: INT64
+ argIndex: 4
+ }
+ argDescriptor {
+ name: "dtype"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -18886,34 +19057,6 @@ opList {
argType: INPUT_TENSOR
argIndex: 4
}
- argDescriptor {
- name: "begin_mask"
- argType: INT64
- }
- argDescriptor {
- name: "ellipsis_mask"
- argType: INT64
- argIndex: 1
- }
- argDescriptor {
- name: "end_mask"
- argType: INT64
- argIndex: 2
- }
- argDescriptor {
- name: "new_axis_mask"
- argType: INT64
- argIndex: 3
- }
- argDescriptor {
- name: "shrink_axis_mask"
- argType: INT64
- argIndex: 4
- }
- argDescriptor {
- name: "dtype"
- argType: DATA_TYPE
- }
}
opList {
name: "sub_scalar"
@@ -18921,18 +19064,23 @@ opList {
name: "outputs"
argType: OUTPUT_TENSOR
}
- argDescriptor {
- name: "inPlace"
- argType: BOOL
- }
argDescriptor {
name: "input"
argType: INPUT_TENSOR
}
+ argDescriptor {
+ name: "y"
+ argType: INPUT_TENSOR
+ argIndex: 1
+ }
opDeclarationType: LEGACY_XYZ
}
opList {
name: "subtract"
+ argDescriptor {
+ name: "dataType"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -18950,14 +19098,14 @@ opList {
argType: INPUT_TENSOR
argIndex: 1
}
- argDescriptor {
- name: "dataType"
- argType: DATA_TYPE
- }
opDeclarationType: BROADCASTABLE_OP_IMPL
}
opList {
name: "subtract_bp"
+ argDescriptor {
+ name: "dtype"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "gradX"
argType: OUTPUT_TENSOR
@@ -18985,13 +19133,13 @@ opList {
argType: INPUT_TENSOR
argIndex: 2
}
+}
+opList {
+ name: "sufficient_statistics"
argDescriptor {
name: "dtype"
argType: DATA_TYPE
}
-}
-opList {
- name: "sufficient_statistics"
argDescriptor {
name: "dataCount"
argType: OUTPUT_TENSOR
@@ -19025,19 +19173,33 @@ opList {
argType: INPUT_TENSOR
argIndex: 2
}
+}
+opList {
+ name: "svd"
+ argDescriptor {
+ name: "fullUV"
+ argType: INT64
+ }
+ argDescriptor {
+ name: "calcUV"
+ argType: INT64
+ argIndex: 1
+ }
+ argDescriptor {
+ name: "switchNum"
+ argType: INT64
+ argIndex: 2
+ }
argDescriptor {
name: "dtype"
argType: DATA_TYPE
}
-}
-opList {
- name: "svd"
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
}
argDescriptor {
- name: "fullUV"
+ name: "full_matrices"
argType: BOOL
}
argDescriptor {
@@ -19064,27 +19226,13 @@ opList {
argType: INPUT_TENSOR
argIndex: 3
}
- argDescriptor {
- name: "fullUV"
- argType: INT64
- }
- argDescriptor {
- name: "calcUV"
- argType: INT64
- argIndex: 1
- }
- argDescriptor {
- name: "switchNum"
- argType: INT64
- argIndex: 2
- }
- argDescriptor {
- name: "dtype"
- argType: DATA_TYPE
- }
}
opList {
name: "swish"
+ argDescriptor {
+ name: "dataType"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -19097,10 +19245,6 @@ opList {
name: "input"
argType: INPUT_TENSOR
}
- argDescriptor {
- name: "dataType"
- argType: DATA_TYPE
- }
opDeclarationType: LEGACY_XYZ
}
opList {
@@ -19109,6 +19253,10 @@ opList {
name: "outputs"
argType: OUTPUT_TENSOR
}
+ argDescriptor {
+ name: "frameName"
+ argType: STRING
+ }
argDescriptor {
name: "input"
argType: INPUT_TENSOR
@@ -19118,13 +19266,13 @@ opList {
argType: INPUT_TENSOR
argIndex: 1
}
- argDescriptor {
- name: "frameName"
- argType: STRING
- }
}
opList {
name: "tan"
+ argDescriptor {
+ name: "dataType"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -19137,14 +19285,14 @@ opList {
name: "input"
argType: INPUT_TENSOR
}
- argDescriptor {
- name: "dataType"
- argType: DATA_TYPE
- }
opDeclarationType: LEGACY_XYZ
}
opList {
name: "tanderivative"
+ argDescriptor {
+ name: "dataType"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -19153,14 +19301,14 @@ opList {
name: "input"
argType: INPUT_TENSOR
}
- argDescriptor {
- name: "dataType"
- argType: DATA_TYPE
- }
opDeclarationType: LEGACY_XYZ
}
opList {
name: "tanh"
+ argDescriptor {
+ name: "dataType"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -19173,10 +19321,6 @@ opList {
name: "input"
argType: INPUT_TENSOR
}
- argDescriptor {
- name: "dataType"
- argType: DATA_TYPE
- }
opDeclarationType: CONFIGURABLE_OP_IMPL
}
opList {
@@ -19198,6 +19342,10 @@ opList {
}
opList {
name: "tear"
+ argDescriptor {
+ name: "dtype"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outE"
argType: OUTPUT_TENSOR
@@ -19206,21 +19354,17 @@ opList {
name: "input"
argType: INPUT_TENSOR
}
- argDescriptor {
- name: "dtype"
- argType: DATA_TYPE
- }
}
opList {
name: "tensorarrayv3"
- argDescriptor {
- name: "outputs"
- argType: OUTPUT_TENSOR
- }
argDescriptor {
name: "dataType"
argType: INT64
}
+ argDescriptor {
+ name: "outputs"
+ argType: OUTPUT_TENSOR
+ }
}
opList {
name: "tensorarraywritev3"
@@ -19235,6 +19379,10 @@ opList {
}
opList {
name: "tensordot"
+ argDescriptor {
+ name: "dimensionsY"
+ argType: INT64
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -19262,26 +19410,9 @@ opList {
argType: INPUT_TENSOR
argIndex: 1
}
- argDescriptor {
- name: "dimensionsY"
- argType: INT64
- }
}
opList {
name: "tensormmul"
- argDescriptor {
- name: "c"
- argType: OUTPUT_TENSOR
- }
- argDescriptor {
- name: "a"
- argType: INPUT_TENSOR
- }
- argDescriptor {
- name: "b"
- argType: INPUT_TENSOR
- argIndex: 1
- }
argDescriptor {
name: "axe0_size"
argType: INT64
@@ -19295,9 +19426,30 @@ opList {
name: "dtype"
argType: DATA_TYPE
}
+ argDescriptor {
+ name: "c"
+ argType: OUTPUT_TENSOR
+ }
+ argDescriptor {
+ name: "a"
+ argType: INPUT_TENSOR
+ }
+ argDescriptor {
+ name: "b"
+ argType: INPUT_TENSOR
+ argIndex: 1
+ }
}
opList {
name: "tensormmul_bp"
+ argDescriptor {
+ name: "axe0Size"
+ argType: INT64
+ }
+ argDescriptor {
+ name: "dtype"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "dLdA"
argType: OUTPUT_TENSOR
@@ -19321,14 +19473,6 @@ opList {
argType: INPUT_TENSOR
argIndex: 2
}
- argDescriptor {
- name: "axe0Size"
- argType: INT64
- }
- argDescriptor {
- name: "dtype"
- argType: DATA_TYPE
- }
}
opList {
name: "test_output_reshape"
@@ -19344,6 +19488,10 @@ opList {
}
opList {
name: "test_scalar"
+ argDescriptor {
+ name: "dtype"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -19352,21 +19500,17 @@ opList {
name: "input"
argType: INPUT_TENSOR
}
- argDescriptor {
- name: "dtype"
- argType: DATA_TYPE
- }
}
opList {
name: "testcustom"
- argDescriptor {
- name: "outputs"
- argType: OUTPUT_TENSOR
- }
argDescriptor {
name: "dtype"
argType: DATA_TYPE
}
+ argDescriptor {
+ name: "outputs"
+ argType: OUTPUT_TENSOR
+ }
}
opList {
name: "testop2i2o"
@@ -19400,6 +19544,10 @@ opList {
}
opList {
name: "tf_atan2"
+ argDescriptor {
+ name: "dataType"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -19417,10 +19565,6 @@ opList {
argType: INPUT_TENSOR
argIndex: 1
}
- argDescriptor {
- name: "dataType"
- argType: DATA_TYPE
- }
opDeclarationType: BROADCASTABLE_OP_IMPL
}
opList {
@@ -19433,14 +19577,14 @@ opList {
name: "inPlace"
argType: BOOL
}
- argDescriptor {
- name: "input"
- argType: INPUT_TENSOR
- }
argDescriptor {
name: "cutoff"
argType: DOUBLE
}
+ argDescriptor {
+ name: "input"
+ argType: INPUT_TENSOR
+ }
opDeclarationType: CONFIGURABLE_OP_IMPL
}
opList {
@@ -19449,6 +19593,10 @@ opList {
name: "dLdI"
argType: OUTPUT_TENSOR
}
+ argDescriptor {
+ name: "cutoff"
+ argType: DOUBLE
+ }
argDescriptor {
name: "input"
argType: INPUT_TENSOR
@@ -19458,14 +19606,18 @@ opList {
argType: INPUT_TENSOR
argIndex: 1
}
- argDescriptor {
- name: "cutoff"
- argType: DOUBLE
- }
opDeclarationType: CONFIGURABLE_OP_IMPL
}
opList {
name: "tile"
+ argDescriptor {
+ name: "dimensions"
+ argType: INT64
+ }
+ argDescriptor {
+ name: "dtype"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -19483,17 +19635,17 @@ opList {
argType: INPUT_TENSOR
argIndex: 1
}
+}
+opList {
+ name: "tile_bp"
argDescriptor {
- name: "dimensions"
+ name: "repeat"
argType: INT64
}
argDescriptor {
name: "dtype"
argType: DATA_TYPE
}
-}
-opList {
- name: "tile_bp"
argDescriptor {
name: "gradI"
argType: OUTPUT_TENSOR
@@ -19507,17 +19659,13 @@ opList {
argType: INPUT_TENSOR
argIndex: 1
}
- argDescriptor {
- name: "repeat"
- argType: INT64
- }
+}
+opList {
+ name: "tile_to_shape"
argDescriptor {
name: "dtype"
argType: DATA_TYPE
}
-}
-opList {
- name: "tile_to_shape"
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -19526,13 +19674,13 @@ opList {
name: "input"
argType: INPUT_TENSOR
}
+}
+opList {
+ name: "tile_to_shape_bp"
argDescriptor {
name: "dtype"
argType: DATA_TYPE
}
-}
-opList {
- name: "tile_to_shape_bp"
argDescriptor {
name: "gradX"
argType: OUTPUT_TENSOR
@@ -19546,13 +19694,13 @@ opList {
argType: INPUT_TENSOR
argIndex: 1
}
- argDescriptor {
- name: "dtype"
- argType: DATA_TYPE
- }
}
opList {
name: "timesoneminus"
+ argDescriptor {
+ name: "dataType"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -19565,14 +19713,14 @@ opList {
name: "input"
argType: INPUT_TENSOR
}
- argDescriptor {
- name: "dataType"
- argType: DATA_TYPE
- }
opDeclarationType: LEGACY_XYZ
}
opList {
name: "to_double"
+ argDescriptor {
+ name: "dtype"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -19581,13 +19729,13 @@ opList {
name: "input"
argType: INPUT_TENSOR
}
- argDescriptor {
- name: "dtype"
- argType: DATA_TYPE
- }
}
opList {
name: "to_float16"
+ argDescriptor {
+ name: "dtype"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -19596,13 +19744,13 @@ opList {
name: "input"
argType: INPUT_TENSOR
}
- argDescriptor {
- name: "dtype"
- argType: DATA_TYPE
- }
}
opList {
name: "to_float32"
+ argDescriptor {
+ name: "dtype"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -19611,13 +19759,13 @@ opList {
name: "input"
argType: INPUT_TENSOR
}
- argDescriptor {
- name: "dtype"
- argType: DATA_TYPE
- }
}
opList {
name: "to_int32"
+ argDescriptor {
+ name: "dtype"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -19626,13 +19774,13 @@ opList {
name: "input"
argType: INPUT_TENSOR
}
- argDescriptor {
- name: "dtype"
- argType: DATA_TYPE
- }
}
opList {
name: "to_int64"
+ argDescriptor {
+ name: "dtype"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -19641,13 +19789,13 @@ opList {
name: "input"
argType: INPUT_TENSOR
}
- argDescriptor {
- name: "dtype"
- argType: DATA_TYPE
- }
}
opList {
name: "to_uint32"
+ argDescriptor {
+ name: "dtype"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -19656,13 +19804,13 @@ opList {
name: "input"
argType: INPUT_TENSOR
}
- argDescriptor {
- name: "dtype"
- argType: DATA_TYPE
- }
}
opList {
name: "to_uint64"
+ argDescriptor {
+ name: "dtype"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -19671,10 +19819,6 @@ opList {
name: "input"
argType: INPUT_TENSOR
}
- argDescriptor {
- name: "dtype"
- argType: DATA_TYPE
- }
}
opList {
name: "toggle_bits"
@@ -19690,6 +19834,14 @@ opList {
}
opList {
name: "top_k"
+ argDescriptor {
+ name: "k"
+ argType: INT64
+ }
+ argDescriptor {
+ name: "dtype"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "values"
argType: OUTPUT_TENSOR
@@ -19707,17 +19859,13 @@ opList {
name: "input"
argType: INPUT_TENSOR
}
- argDescriptor {
- name: "k"
- argType: INT64
- }
+}
+opList {
+ name: "trace"
argDescriptor {
name: "dtype"
argType: DATA_TYPE
}
-}
-opList {
- name: "trace"
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -19726,13 +19874,13 @@ opList {
name: "input"
argType: INPUT_TENSOR
}
+}
+opList {
+ name: "transpose"
argDescriptor {
name: "dtype"
argType: DATA_TYPE
}
-}
-opList {
- name: "transpose"
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -19746,17 +19894,9 @@ opList {
argType: INPUT_TENSOR
argIndex: 1
}
- argDescriptor {
- name: "dtype"
- argType: DATA_TYPE
- }
}
opList {
name: "tri"
- argDescriptor {
- name: "outputs"
- argType: OUTPUT_TENSOR
- }
argDescriptor {
name: "row"
argType: INT64
@@ -19775,9 +19915,17 @@ opList {
name: "dtype"
argType: DATA_TYPE
}
+ argDescriptor {
+ name: "outputs"
+ argType: OUTPUT_TENSOR
+ }
}
opList {
name: "triangular_solve"
+ argDescriptor {
+ name: "dtype"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -19810,21 +19958,9 @@ opList {
argType: INPUT_TENSOR
argIndex: 3
}
- argDescriptor {
- name: "dtype"
- argType: DATA_TYPE
- }
}
opList {
name: "triu"
- argDescriptor {
- name: "outputs"
- argType: OUTPUT_TENSOR
- }
- argDescriptor {
- name: "input"
- argType: INPUT_TENSOR
- }
argDescriptor {
name: "diag"
argType: INT64
@@ -19833,9 +19969,25 @@ opList {
name: "dtype"
argType: DATA_TYPE
}
+ argDescriptor {
+ name: "outputs"
+ argType: OUTPUT_TENSOR
+ }
+ argDescriptor {
+ name: "input"
+ argType: INPUT_TENSOR
+ }
}
opList {
name: "triu_bp"
+ argDescriptor {
+ name: "diag"
+ argType: INT64
+ }
+ argDescriptor {
+ name: "dtype"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "gradI"
argType: OUTPUT_TENSOR
@@ -19849,17 +20001,13 @@ opList {
argType: INPUT_TENSOR
argIndex: 1
}
- argDescriptor {
- name: "diag"
- argType: INT64
- }
- argDescriptor {
- name: "dtype"
- argType: DATA_TYPE
- }
}
opList {
name: "truncatediv"
+ argDescriptor {
+ name: "dataType"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -19877,14 +20025,14 @@ opList {
argType: INPUT_TENSOR
argIndex: 1
}
- argDescriptor {
- name: "dataType"
- argType: DATA_TYPE
- }
opDeclarationType: BROADCASTABLE_OP_IMPL
}
opList {
name: "unique"
+ argDescriptor {
+ name: "dtype"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "values"
argType: OUTPUT_TENSOR
@@ -19898,13 +20046,13 @@ opList {
name: "input"
argType: INPUT_TENSOR
}
+}
+opList {
+ name: "unique_with_counts"
argDescriptor {
name: "dtype"
argType: DATA_TYPE
}
-}
-opList {
- name: "unique_with_counts"
argDescriptor {
name: "values"
argType: OUTPUT_TENSOR
@@ -19923,13 +20071,17 @@ opList {
name: "input"
argType: INPUT_TENSOR
}
+}
+opList {
+ name: "unsorted_segment_max"
+ argDescriptor {
+ name: "numSegments"
+ argType: INT64
+ }
argDescriptor {
name: "dtype"
argType: DATA_TYPE
}
-}
-opList {
- name: "unsorted_segment_max"
argDescriptor {
name: "segmentedOutput"
argType: OUTPUT_TENSOR
@@ -19948,17 +20100,17 @@ opList {
argType: INPUT_TENSOR
argIndex: 2
}
- argDescriptor {
- name: "numSegments"
- argType: INT64
- }
- argDescriptor {
- name: "dtype"
- argType: DATA_TYPE
- }
}
opList {
name: "unsorted_segment_max_bp"
+ argDescriptor {
+ name: "numSegments"
+ argType: INT64
+ }
+ argDescriptor {
+ name: "dtype"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -19977,17 +20129,17 @@ opList {
argType: INPUT_TENSOR
argIndex: 2
}
- argDescriptor {
- name: "numSegments"
- argType: INT64
- }
- argDescriptor {
- name: "dtype"
- argType: DATA_TYPE
- }
}
opList {
name: "unsorted_segment_mean"
+ argDescriptor {
+ name: "numSegments"
+ argType: INT64
+ }
+ argDescriptor {
+ name: "dtype"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "segmentedOutput"
argType: OUTPUT_TENSOR
@@ -20006,17 +20158,17 @@ opList {
argType: INPUT_TENSOR
argIndex: 2
}
- argDescriptor {
- name: "numSegments"
- argType: INT64
- }
- argDescriptor {
- name: "dtype"
- argType: DATA_TYPE
- }
}
opList {
name: "unsorted_segment_mean_bp"
+ argDescriptor {
+ name: "numSegments"
+ argType: INT64
+ }
+ argDescriptor {
+ name: "dtype"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -20035,17 +20187,17 @@ opList {
argType: INPUT_TENSOR
argIndex: 2
}
- argDescriptor {
- name: "numSegments"
- argType: INT64
- }
- argDescriptor {
- name: "dtype"
- argType: DATA_TYPE
- }
}
opList {
name: "unsorted_segment_min"
+ argDescriptor {
+ name: "numSegments"
+ argType: INT64
+ }
+ argDescriptor {
+ name: "dtype"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "segmentedOutput"
argType: OUTPUT_TENSOR
@@ -20064,17 +20216,17 @@ opList {
argType: INPUT_TENSOR
argIndex: 2
}
- argDescriptor {
- name: "numSegments"
- argType: INT64
- }
- argDescriptor {
- name: "dtype"
- argType: DATA_TYPE
- }
}
opList {
name: "unsorted_segment_min_bp"
+ argDescriptor {
+ name: "numSegments"
+ argType: INT64
+ }
+ argDescriptor {
+ name: "dtype"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -20093,17 +20245,17 @@ opList {
argType: INPUT_TENSOR
argIndex: 2
}
- argDescriptor {
- name: "numSegments"
- argType: INT64
- }
- argDescriptor {
- name: "dtype"
- argType: DATA_TYPE
- }
}
opList {
name: "unsorted_segment_prod"
+ argDescriptor {
+ name: "numSegments"
+ argType: INT64
+ }
+ argDescriptor {
+ name: "dtype"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "segmentedOutput"
argType: OUTPUT_TENSOR
@@ -20122,17 +20274,17 @@ opList {
argType: INPUT_TENSOR
argIndex: 2
}
- argDescriptor {
- name: "numSegments"
- argType: INT64
- }
- argDescriptor {
- name: "dtype"
- argType: DATA_TYPE
- }
}
opList {
name: "unsorted_segment_prod_bp"
+ argDescriptor {
+ name: "numSegments"
+ argType: INT64
+ }
+ argDescriptor {
+ name: "dtype"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -20151,17 +20303,17 @@ opList {
argType: INPUT_TENSOR
argIndex: 2
}
- argDescriptor {
- name: "numSegments"
- argType: INT64
- }
- argDescriptor {
- name: "dtype"
- argType: DATA_TYPE
- }
}
opList {
name: "unsorted_segment_sqrt_n"
+ argDescriptor {
+ name: "numSegments"
+ argType: INT64
+ }
+ argDescriptor {
+ name: "dtype"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "segmentedOutput"
argType: OUTPUT_TENSOR
@@ -20180,17 +20332,17 @@ opList {
argType: INPUT_TENSOR
argIndex: 2
}
- argDescriptor {
- name: "numSegments"
- argType: INT64
- }
- argDescriptor {
- name: "dtype"
- argType: DATA_TYPE
- }
}
opList {
name: "unsorted_segment_sqrt_n_bp"
+ argDescriptor {
+ name: "numSegments"
+ argType: INT64
+ }
+ argDescriptor {
+ name: "dtype"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -20209,6 +20361,9 @@ opList {
argType: INPUT_TENSOR
argIndex: 2
}
+}
+opList {
+ name: "unsorted_segment_sum"
argDescriptor {
name: "numSegments"
argType: INT64
@@ -20217,9 +20372,6 @@ opList {
name: "dtype"
argType: DATA_TYPE
}
-}
-opList {
- name: "unsorted_segment_sum"
argDescriptor {
name: "segmentedOutput"
argType: OUTPUT_TENSOR
@@ -20238,6 +20390,9 @@ opList {
argType: INPUT_TENSOR
argIndex: 2
}
+}
+opList {
+ name: "unsorted_segment_sum_bp"
argDescriptor {
name: "numSegments"
argType: INT64
@@ -20246,9 +20401,6 @@ opList {
name: "dtype"
argType: DATA_TYPE
}
-}
-opList {
- name: "unsorted_segment_sum_bp"
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -20267,25 +20419,9 @@ opList {
argType: INPUT_TENSOR
argIndex: 2
}
- argDescriptor {
- name: "numSegments"
- argType: INT64
- }
- argDescriptor {
- name: "dtype"
- argType: DATA_TYPE
- }
}
opList {
name: "unstack"
- argDescriptor {
- name: "outArrs"
- argType: OUTPUT_TENSOR
- }
- argDescriptor {
- name: "input"
- argType: INPUT_TENSOR
- }
argDescriptor {
name: "dimensions"
argType: INT64
@@ -20299,6 +20435,14 @@ opList {
name: "dtype"
argType: DATA_TYPE
}
+ argDescriptor {
+ name: "outArrs"
+ argType: OUTPUT_TENSOR
+ }
+ argDescriptor {
+ name: "input"
+ argType: INPUT_TENSOR
+ }
}
opList {
name: "unstack_list"
@@ -20314,18 +20458,6 @@ opList {
}
opList {
name: "upsampling2d"
- argDescriptor {
- name: "outputs"
- argType: OUTPUT_TENSOR
- }
- argDescriptor {
- name: "nchw"
- argType: BOOL
- }
- argDescriptor {
- name: "input"
- argType: INPUT_TENSOR
- }
argDescriptor {
name: "factorH"
argType: INT64
@@ -20344,9 +20476,29 @@ opList {
name: "dtype"
argType: DATA_TYPE
}
+ argDescriptor {
+ name: "outputs"
+ argType: OUTPUT_TENSOR
+ }
+ argDescriptor {
+ name: "nchw"
+ argType: BOOL
+ }
+ argDescriptor {
+ name: "input"
+ argType: INPUT_TENSOR
+ }
}
opList {
name: "upsampling2d_bp"
+ argDescriptor {
+ name: "scaleW"
+ argType: INT64
+ }
+ argDescriptor {
+ name: "dtype"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "gradI"
argType: OUTPUT_TENSOR
@@ -20364,29 +20516,9 @@ opList {
argType: INPUT_TENSOR
argIndex: 1
}
- argDescriptor {
- name: "scaleW"
- argType: INT64
- }
- argDescriptor {
- name: "dtype"
- argType: DATA_TYPE
- }
}
opList {
name: "upsampling3d"
- argDescriptor {
- name: "outputs"
- argType: OUTPUT_TENSOR
- }
- argDescriptor {
- name: "ncdhw"
- argType: BOOL
- }
- argDescriptor {
- name: "input"
- argType: INPUT_TENSOR
- }
argDescriptor {
name: "factorD"
argType: INT64
@@ -20410,9 +20542,29 @@ opList {
name: "dtype"
argType: DATA_TYPE
}
+ argDescriptor {
+ name: "outputs"
+ argType: OUTPUT_TENSOR
+ }
+ argDescriptor {
+ name: "ncdhw"
+ argType: BOOL
+ }
+ argDescriptor {
+ name: "input"
+ argType: INPUT_TENSOR
+ }
}
opList {
name: "upsampling3d_bp"
+ argDescriptor {
+ name: "isNCDHW"
+ argType: INT64
+ }
+ argDescriptor {
+ name: "dtype"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "gradI"
argType: OUTPUT_TENSOR
@@ -20430,17 +20582,13 @@ opList {
argType: INPUT_TENSOR
argIndex: 1
}
- argDescriptor {
- name: "isNCDHW"
- argType: INT64
- }
- argDescriptor {
- name: "dtype"
- argType: DATA_TYPE
- }
}
opList {
name: "var"
+ argDescriptor {
+ name: "dimensions"
+ argType: INT64
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -20458,10 +20606,6 @@ opList {
name: "input"
argType: INPUT_TENSOR
}
- argDescriptor {
- name: "dimensions"
- argType: INT64
- }
opDeclarationType: LEGACY_XYZ
}
opList {
@@ -20488,6 +20632,10 @@ opList {
}
opList {
name: "where_np"
+ argDescriptor {
+ name: "dtype"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -20506,13 +20654,13 @@ opList {
argType: INPUT_TENSOR
argIndex: 2
}
- argDescriptor {
- name: "dtype"
- argType: DATA_TYPE
- }
}
opList {
name: "write_list"
+ argDescriptor {
+ name: "idx"
+ argType: INT64
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -20526,18 +20674,22 @@ opList {
argType: INPUT_TENSOR
argIndex: 1
}
- argDescriptor {
- name: "idx"
- argType: INT64
- }
opDeclarationType: LIST_OP_IMPL
}
opList {
name: "xor"
+ argDescriptor {
+ name: "dataType"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
}
+ argDescriptor {
+ name: "comparable"
+ argType: DOUBLE
+ }
argDescriptor {
name: "input"
argType: INPUT_TENSOR
@@ -20547,14 +20699,6 @@ opList {
argType: INPUT_TENSOR
argIndex: 1
}
- argDescriptor {
- name: "dataType"
- argType: DATA_TYPE
- }
- argDescriptor {
- name: "comparable"
- argType: DOUBLE
- }
opDeclarationType: LEGACY_XYZ
}
opList {
@@ -20575,6 +20719,14 @@ opList {
}
opList {
name: "xw_plus_b"
+ argDescriptor {
+ name: "bTranspose"
+ argType: INT64
+ }
+ argDescriptor {
+ name: "dtype"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -20593,6 +20745,9 @@ opList {
argType: INPUT_TENSOR
argIndex: 2
}
+}
+opList {
+ name: "xw_plus_b_bp"
argDescriptor {
name: "bTranspose"
argType: INT64
@@ -20601,9 +20756,6 @@ opList {
name: "dtype"
argType: DATA_TYPE
}
-}
-opList {
- name: "xw_plus_b_bp"
argDescriptor {
name: "dLdx"
argType: OUTPUT_TENSOR
@@ -20637,17 +20789,13 @@ opList {
argType: INPUT_TENSOR
argIndex: 3
}
- argDescriptor {
- name: "bTranspose"
- argType: INT64
- }
- argDescriptor {
- name: "dtype"
- argType: DATA_TYPE
- }
}
opList {
name: "yiq_to_rgb"
+ argDescriptor {
+ name: "dimC"
+ argType: INT64
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -20656,14 +20804,14 @@ opList {
name: "input"
argType: INPUT_TENSOR
}
- argDescriptor {
- name: "dimC"
- argType: INT64
- }
opDeclarationType: CONFIGURABLE_OP_IMPL
}
opList {
name: "yuv_to_rgb"
+ argDescriptor {
+ name: "dimC"
+ argType: INT64
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -20672,14 +20820,14 @@ opList {
name: "input"
argType: INPUT_TENSOR
}
- argDescriptor {
- name: "dimC"
- argType: INT64
- }
opDeclarationType: CONFIGURABLE_OP_IMPL
}
opList {
name: "zero_fraction"
+ argDescriptor {
+ name: "dtype"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -20688,21 +20836,17 @@ opList {
name: "input"
argType: INPUT_TENSOR
}
- argDescriptor {
- name: "dtype"
- argType: DATA_TYPE
- }
}
opList {
name: "zeros_as"
- argDescriptor {
- name: "outputs"
- argType: OUTPUT_TENSOR
- }
argDescriptor {
name: "dtype"
argType: DATA_TYPE
}
+ argDescriptor {
+ name: "outputs"
+ argType: OUTPUT_TENSOR
+ }
}
opList {
name: "zeros_like"
@@ -20717,6 +20861,10 @@ opList {
}
opList {
name: "zeroslike"
+ argDescriptor {
+ name: "dataType"
+ argType: INT64
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -20729,13 +20877,13 @@ opList {
name: "input"
argType: INPUT_TENSOR
}
- argDescriptor {
- name: "dataType"
- argType: INT64
- }
}
opList {
name: "zeta"
+ argDescriptor {
+ name: "dataType"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -20753,10 +20901,6 @@ opList {
argType: INPUT_TENSOR
argIndex: 1
}
- argDescriptor {
- name: "dataType"
- argType: DATA_TYPE
- }
opDeclarationType: CONFIGURABLE_OP_IMPL
}
opList {
diff --git a/contrib/codegen-tools/libnd4j-gen/src/main/java/org/nd4j/descriptor/proposal/impl/JavaSourceArgDescriptorSource.java b/contrib/codegen-tools/libnd4j-gen/src/main/java/org/nd4j/descriptor/proposal/impl/JavaSourceArgDescriptorSource.java
index d4216e925..c9cc9b799 100644
--- a/contrib/codegen-tools/libnd4j-gen/src/main/java/org/nd4j/descriptor/proposal/impl/JavaSourceArgDescriptorSource.java
+++ b/contrib/codegen-tools/libnd4j-gen/src/main/java/org/nd4j/descriptor/proposal/impl/JavaSourceArgDescriptorSource.java
@@ -551,6 +551,18 @@ public class JavaSourceArgDescriptorSource implements ArgDescriptorSource {
}
}
+ if(name.contains("scatter_update")) {
+ argDescriptorProposals.add(ArgDescriptorProposal.builder()
+ .sourceOfProposal("java")
+ .proposalWeight(Double.MAX_VALUE)
+ .descriptor(OpNamespace.ArgDescriptor.newBuilder()
+ .setArgType(OpNamespace.ArgDescriptor.ArgType.INPUT_TENSOR)
+ .setName("indices")
+ .setIsArray(false)
+ .setArgIndex(2)
+ .build()).build());
+
+ }
if(name.contains("fill")) {
diff --git a/contrib/codegen-tools/libnd4j-gen/src/main/java/org/nd4j/descriptor/proposal/impl/Libnd4jArgDescriptorSource.java b/contrib/codegen-tools/libnd4j-gen/src/main/java/org/nd4j/descriptor/proposal/impl/Libnd4jArgDescriptorSource.java
index fb234c1fa..34c576474 100644
--- a/contrib/codegen-tools/libnd4j-gen/src/main/java/org/nd4j/descriptor/proposal/impl/Libnd4jArgDescriptorSource.java
+++ b/contrib/codegen-tools/libnd4j-gen/src/main/java/org/nd4j/descriptor/proposal/impl/Libnd4jArgDescriptorSource.java
@@ -913,6 +913,8 @@ public class Libnd4jArgDescriptorSource implements ArgDescriptorSource {
}
+
+
if(name.equals("lin_space")) {
argDescriptorProposals.add(ArgDescriptorProposal.builder()
.sourceOfProposal("start")
@@ -1224,6 +1226,19 @@ public class Libnd4jArgDescriptorSource implements ArgDescriptorSource {
.build());
}
+ if(name.contains("scatter_update")) {
+ argDescriptorProposals.add(ArgDescriptorProposal.builder()
+ .sourceOfProposal("cpp")
+ .proposalWeight(Double.MAX_VALUE)
+ .descriptor(OpNamespace.ArgDescriptor.newBuilder()
+ .setArgType(OpNamespace.ArgDescriptor.ArgType.INPUT_TENSOR)
+ .setName("indices")
+ .setIsArray(false)
+ .setArgIndex(2)
+ .build()).build());
+
+ }
+
} else if(line.contains(REDUCTION_OP_IMPL)) {
//REDUCTION_OP_IMPL(NAME, NIN, NOUT, INPLACEABLE, TARGS, IARGS)
foundOp = true;
diff --git a/nd4j/nd4j-backends/nd4j-tests/ops-added-new.txt b/nd4j/nd4j-backends/nd4j-tests/ops-added-new.txt
index 11eb696f1..ea6bc3c53 100644
--- a/nd4j/nd4j-backends/nd4j-tests/ops-added-new.txt
+++ b/nd4j/nd4j-backends/nd4j-tests/ops-added-new.txt
@@ -1,7 +1,704 @@
-Const,Variable
-Const,Variable_1
-Identity,Variable/read
-Identity,Variable_1/read
-Pack,floordiv/x
-Pack,floordiv/y
-FloorDiv,floordiv
+Placeholder,input_tensor
+Const,transpose/perm
+Const,Pad/paddings
+Const,conv2d/kernel
+Const,batch_normalization/gamma
+Const,batch_normalization/beta
+Const,batch_normalization/moving_mean
+Const,batch_normalization/moving_variance
+Const,conv2d_1/kernel
+Const,conv2d_2/kernel
+Const,batch_normalization_1/gamma
+Const,batch_normalization_1/beta
+Const,batch_normalization_1/moving_mean
+Const,batch_normalization_1/moving_variance
+Const,conv2d_3/kernel
+Const,batch_normalization_2/gamma
+Const,batch_normalization_2/beta
+Const,batch_normalization_2/moving_mean
+Const,batch_normalization_2/moving_variance
+Const,conv2d_4/kernel
+Const,batch_normalization_3/gamma
+Const,batch_normalization_3/beta
+Const,batch_normalization_3/moving_mean
+Const,batch_normalization_3/moving_variance
+Const,conv2d_5/kernel
+Const,batch_normalization_4/gamma
+Const,batch_normalization_4/beta
+Const,batch_normalization_4/moving_mean
+Const,batch_normalization_4/moving_variance
+Const,conv2d_6/kernel
+Const,batch_normalization_5/gamma
+Const,batch_normalization_5/beta
+Const,batch_normalization_5/moving_mean
+Const,batch_normalization_5/moving_variance
+Const,conv2d_7/kernel
+Const,batch_normalization_6/gamma
+Const,batch_normalization_6/beta
+Const,batch_normalization_6/moving_mean
+Const,batch_normalization_6/moving_variance
+Const,conv2d_8/kernel
+Const,batch_normalization_7/gamma
+Const,batch_normalization_7/beta
+Const,batch_normalization_7/moving_mean
+Const,batch_normalization_7/moving_variance
+Const,conv2d_9/kernel
+Const,batch_normalization_8/gamma
+Const,batch_normalization_8/beta
+Const,batch_normalization_8/moving_mean
+Const,batch_normalization_8/moving_variance
+Const,conv2d_10/kernel
+Const,batch_normalization_9/gamma
+Const,batch_normalization_9/beta
+Const,batch_normalization_9/moving_mean
+Const,batch_normalization_9/moving_variance
+Const,Pad_1/paddings
+Const,conv2d_11/kernel
+Const,conv2d_12/kernel
+Const,batch_normalization_10/gamma
+Const,batch_normalization_10/beta
+Const,batch_normalization_10/moving_mean
+Const,batch_normalization_10/moving_variance
+Const,Pad_2/paddings
+Const,conv2d_13/kernel
+Const,batch_normalization_11/gamma
+Const,batch_normalization_11/beta
+Const,batch_normalization_11/moving_mean
+Const,batch_normalization_11/moving_variance
+Const,conv2d_14/kernel
+Const,batch_normalization_12/gamma
+Const,batch_normalization_12/beta
+Const,batch_normalization_12/moving_mean
+Const,batch_normalization_12/moving_variance
+Const,conv2d_15/kernel
+Const,batch_normalization_13/gamma
+Const,batch_normalization_13/beta
+Const,batch_normalization_13/moving_mean
+Const,batch_normalization_13/moving_variance
+Const,conv2d_16/kernel
+Const,batch_normalization_14/gamma
+Const,batch_normalization_14/beta
+Const,batch_normalization_14/moving_mean
+Const,batch_normalization_14/moving_variance
+Const,conv2d_17/kernel
+Const,batch_normalization_15/gamma
+Const,batch_normalization_15/beta
+Const,batch_normalization_15/moving_mean
+Const,batch_normalization_15/moving_variance
+Const,conv2d_18/kernel
+Const,batch_normalization_16/gamma
+Const,batch_normalization_16/beta
+Const,batch_normalization_16/moving_mean
+Const,batch_normalization_16/moving_variance
+Const,conv2d_19/kernel
+Const,batch_normalization_17/gamma
+Const,batch_normalization_17/beta
+Const,batch_normalization_17/moving_mean
+Const,batch_normalization_17/moving_variance
+Const,conv2d_20/kernel
+Const,batch_normalization_18/gamma
+Const,batch_normalization_18/beta
+Const,batch_normalization_18/moving_mean
+Const,batch_normalization_18/moving_variance
+Const,conv2d_21/kernel
+Const,batch_normalization_19/gamma
+Const,batch_normalization_19/beta
+Const,batch_normalization_19/moving_mean
+Const,batch_normalization_19/moving_variance
+Const,conv2d_22/kernel
+Const,batch_normalization_20/gamma
+Const,batch_normalization_20/beta
+Const,batch_normalization_20/moving_mean
+Const,batch_normalization_20/moving_variance
+Const,conv2d_23/kernel
+Const,batch_normalization_21/gamma
+Const,batch_normalization_21/beta
+Const,batch_normalization_21/moving_mean
+Const,batch_normalization_21/moving_variance
+Const,Pad_3/paddings
+Const,conv2d_24/kernel
+Const,conv2d_25/kernel
+Const,batch_normalization_22/gamma
+Const,batch_normalization_22/beta
+Const,batch_normalization_22/moving_mean
+Const,batch_normalization_22/moving_variance
+Const,Pad_4/paddings
+Const,conv2d_26/kernel
+Const,batch_normalization_23/gamma
+Const,batch_normalization_23/beta
+Const,batch_normalization_23/moving_mean
+Const,batch_normalization_23/moving_variance
+Const,conv2d_27/kernel
+Const,batch_normalization_24/gamma
+Const,batch_normalization_24/beta
+Const,batch_normalization_24/moving_mean
+Const,batch_normalization_24/moving_variance
+Const,conv2d_28/kernel
+Const,batch_normalization_25/gamma
+Const,batch_normalization_25/beta
+Const,batch_normalization_25/moving_mean
+Const,batch_normalization_25/moving_variance
+Const,conv2d_29/kernel
+Const,batch_normalization_26/gamma
+Const,batch_normalization_26/beta
+Const,batch_normalization_26/moving_mean
+Const,batch_normalization_26/moving_variance
+Const,conv2d_30/kernel
+Const,batch_normalization_27/gamma
+Const,batch_normalization_27/beta
+Const,batch_normalization_27/moving_mean
+Const,batch_normalization_27/moving_variance
+Const,conv2d_31/kernel
+Const,batch_normalization_28/gamma
+Const,batch_normalization_28/beta
+Const,batch_normalization_28/moving_mean
+Const,batch_normalization_28/moving_variance
+Const,conv2d_32/kernel
+Const,batch_normalization_29/gamma
+Const,batch_normalization_29/beta
+Const,batch_normalization_29/moving_mean
+Const,batch_normalization_29/moving_variance
+Const,conv2d_33/kernel
+Const,batch_normalization_30/gamma
+Const,batch_normalization_30/beta
+Const,batch_normalization_30/moving_mean
+Const,batch_normalization_30/moving_variance
+Const,conv2d_34/kernel
+Const,batch_normalization_31/gamma
+Const,batch_normalization_31/beta
+Const,batch_normalization_31/moving_mean
+Const,batch_normalization_31/moving_variance
+Const,conv2d_35/kernel
+Const,batch_normalization_32/gamma
+Const,batch_normalization_32/beta
+Const,batch_normalization_32/moving_mean
+Const,batch_normalization_32/moving_variance
+Const,conv2d_36/kernel
+Const,batch_normalization_33/gamma
+Const,batch_normalization_33/beta
+Const,batch_normalization_33/moving_mean
+Const,batch_normalization_33/moving_variance
+Const,conv2d_37/kernel
+Const,batch_normalization_34/gamma
+Const,batch_normalization_34/beta
+Const,batch_normalization_34/moving_mean
+Const,batch_normalization_34/moving_variance
+Const,conv2d_38/kernel
+Const,batch_normalization_35/gamma
+Const,batch_normalization_35/beta
+Const,batch_normalization_35/moving_mean
+Const,batch_normalization_35/moving_variance
+Const,conv2d_39/kernel
+Const,batch_normalization_36/gamma
+Const,batch_normalization_36/beta
+Const,batch_normalization_36/moving_mean
+Const,batch_normalization_36/moving_variance
+Const,conv2d_40/kernel
+Const,batch_normalization_37/gamma
+Const,batch_normalization_37/beta
+Const,batch_normalization_37/moving_mean
+Const,batch_normalization_37/moving_variance
+Const,conv2d_41/kernel
+Const,batch_normalization_38/gamma
+Const,batch_normalization_38/beta
+Const,batch_normalization_38/moving_mean
+Const,batch_normalization_38/moving_variance
+Const,conv2d_42/kernel
+Const,batch_normalization_39/gamma
+Const,batch_normalization_39/beta
+Const,batch_normalization_39/moving_mean
+Const,batch_normalization_39/moving_variance
+Const,Pad_5/paddings
+Const,conv2d_43/kernel
+Const,conv2d_44/kernel
+Const,batch_normalization_40/gamma
+Const,batch_normalization_40/beta
+Const,batch_normalization_40/moving_mean
+Const,batch_normalization_40/moving_variance
+Const,Pad_6/paddings
+Const,conv2d_45/kernel
+Const,batch_normalization_41/gamma
+Const,batch_normalization_41/beta
+Const,batch_normalization_41/moving_mean
+Const,batch_normalization_41/moving_variance
+Const,conv2d_46/kernel
+Const,batch_normalization_42/gamma
+Const,batch_normalization_42/beta
+Const,batch_normalization_42/moving_mean
+Const,batch_normalization_42/moving_variance
+Const,conv2d_47/kernel
+Const,batch_normalization_43/gamma
+Const,batch_normalization_43/beta
+Const,batch_normalization_43/moving_mean
+Const,batch_normalization_43/moving_variance
+Const,conv2d_48/kernel
+Const,batch_normalization_44/gamma
+Const,batch_normalization_44/beta
+Const,batch_normalization_44/moving_mean
+Const,batch_normalization_44/moving_variance
+Const,conv2d_49/kernel
+Const,batch_normalization_45/gamma
+Const,batch_normalization_45/beta
+Const,batch_normalization_45/moving_mean
+Const,batch_normalization_45/moving_variance
+Const,conv2d_50/kernel
+Const,batch_normalization_46/gamma
+Const,batch_normalization_46/beta
+Const,batch_normalization_46/moving_mean
+Const,batch_normalization_46/moving_variance
+Const,conv2d_51/kernel
+Const,batch_normalization_47/gamma
+Const,batch_normalization_47/beta
+Const,batch_normalization_47/moving_mean
+Const,batch_normalization_47/moving_variance
+Const,conv2d_52/kernel
+Const,batch_normalization_48/gamma
+Const,batch_normalization_48/beta
+Const,batch_normalization_48/moving_mean
+Const,batch_normalization_48/moving_variance
+Const,Mean/reduction_indices
+Const,Reshape/shape
+Const,dense/kernel
+Const,dense/bias
+Const,ArgMax/dimension
+Transpose,transpose
+Identity,conv2d/kernel/read
+Identity,batch_normalization/gamma/read
+Identity,batch_normalization/beta/read
+Identity,batch_normalization/moving_mean/read
+Identity,batch_normalization/moving_variance/read
+Identity,conv2d_1/kernel/read
+Identity,conv2d_2/kernel/read
+Identity,batch_normalization_1/gamma/read
+Identity,batch_normalization_1/beta/read
+Identity,batch_normalization_1/moving_mean/read
+Identity,batch_normalization_1/moving_variance/read
+Identity,conv2d_3/kernel/read
+Identity,batch_normalization_2/gamma/read
+Identity,batch_normalization_2/beta/read
+Identity,batch_normalization_2/moving_mean/read
+Identity,batch_normalization_2/moving_variance/read
+Identity,conv2d_4/kernel/read
+Identity,batch_normalization_3/gamma/read
+Identity,batch_normalization_3/beta/read
+Identity,batch_normalization_3/moving_mean/read
+Identity,batch_normalization_3/moving_variance/read
+Identity,conv2d_5/kernel/read
+Identity,batch_normalization_4/gamma/read
+Identity,batch_normalization_4/beta/read
+Identity,batch_normalization_4/moving_mean/read
+Identity,batch_normalization_4/moving_variance/read
+Identity,conv2d_6/kernel/read
+Identity,batch_normalization_5/gamma/read
+Identity,batch_normalization_5/beta/read
+Identity,batch_normalization_5/moving_mean/read
+Identity,batch_normalization_5/moving_variance/read
+Identity,conv2d_7/kernel/read
+Identity,batch_normalization_6/gamma/read
+Identity,batch_normalization_6/beta/read
+Identity,batch_normalization_6/moving_mean/read
+Identity,batch_normalization_6/moving_variance/read
+Identity,conv2d_8/kernel/read
+Identity,batch_normalization_7/gamma/read
+Identity,batch_normalization_7/beta/read
+Identity,batch_normalization_7/moving_mean/read
+Identity,batch_normalization_7/moving_variance/read
+Identity,conv2d_9/kernel/read
+Identity,batch_normalization_8/gamma/read
+Identity,batch_normalization_8/beta/read
+Identity,batch_normalization_8/moving_mean/read
+Identity,batch_normalization_8/moving_variance/read
+Identity,conv2d_10/kernel/read
+Identity,batch_normalization_9/gamma/read
+Identity,batch_normalization_9/beta/read
+Identity,batch_normalization_9/moving_mean/read
+Identity,batch_normalization_9/moving_variance/read
+Identity,conv2d_11/kernel/read
+Identity,conv2d_12/kernel/read
+Identity,batch_normalization_10/gamma/read
+Identity,batch_normalization_10/beta/read
+Identity,batch_normalization_10/moving_mean/read
+Identity,batch_normalization_10/moving_variance/read
+Identity,conv2d_13/kernel/read
+Identity,batch_normalization_11/gamma/read
+Identity,batch_normalization_11/beta/read
+Identity,batch_normalization_11/moving_mean/read
+Identity,batch_normalization_11/moving_variance/read
+Identity,conv2d_14/kernel/read
+Identity,batch_normalization_12/gamma/read
+Identity,batch_normalization_12/beta/read
+Identity,batch_normalization_12/moving_mean/read
+Identity,batch_normalization_12/moving_variance/read
+Identity,conv2d_15/kernel/read
+Identity,batch_normalization_13/gamma/read
+Identity,batch_normalization_13/beta/read
+Identity,batch_normalization_13/moving_mean/read
+Identity,batch_normalization_13/moving_variance/read
+Identity,conv2d_16/kernel/read
+Identity,batch_normalization_14/gamma/read
+Identity,batch_normalization_14/beta/read
+Identity,batch_normalization_14/moving_mean/read
+Identity,batch_normalization_14/moving_variance/read
+Identity,conv2d_17/kernel/read
+Identity,batch_normalization_15/gamma/read
+Identity,batch_normalization_15/beta/read
+Identity,batch_normalization_15/moving_mean/read
+Identity,batch_normalization_15/moving_variance/read
+Identity,conv2d_18/kernel/read
+Identity,batch_normalization_16/gamma/read
+Identity,batch_normalization_16/beta/read
+Identity,batch_normalization_16/moving_mean/read
+Identity,batch_normalization_16/moving_variance/read
+Identity,conv2d_19/kernel/read
+Identity,batch_normalization_17/gamma/read
+Identity,batch_normalization_17/beta/read
+Identity,batch_normalization_17/moving_mean/read
+Identity,batch_normalization_17/moving_variance/read
+Identity,conv2d_20/kernel/read
+Identity,batch_normalization_18/gamma/read
+Identity,batch_normalization_18/beta/read
+Identity,batch_normalization_18/moving_mean/read
+Identity,batch_normalization_18/moving_variance/read
+Identity,conv2d_21/kernel/read
+Identity,batch_normalization_19/gamma/read
+Identity,batch_normalization_19/beta/read
+Identity,batch_normalization_19/moving_mean/read
+Identity,batch_normalization_19/moving_variance/read
+Identity,conv2d_22/kernel/read
+Identity,batch_normalization_20/gamma/read
+Identity,batch_normalization_20/beta/read
+Identity,batch_normalization_20/moving_mean/read
+Identity,batch_normalization_20/moving_variance/read
+Identity,conv2d_23/kernel/read
+Identity,batch_normalization_21/gamma/read
+Identity,batch_normalization_21/beta/read
+Identity,batch_normalization_21/moving_mean/read
+Identity,batch_normalization_21/moving_variance/read
+Identity,conv2d_24/kernel/read
+Identity,conv2d_25/kernel/read
+Identity,batch_normalization_22/gamma/read
+Identity,batch_normalization_22/beta/read
+Identity,batch_normalization_22/moving_mean/read
+Identity,batch_normalization_22/moving_variance/read
+Identity,conv2d_26/kernel/read
+Identity,batch_normalization_23/gamma/read
+Identity,batch_normalization_23/beta/read
+Identity,batch_normalization_23/moving_mean/read
+Identity,batch_normalization_23/moving_variance/read
+Identity,conv2d_27/kernel/read
+Identity,batch_normalization_24/gamma/read
+Identity,batch_normalization_24/beta/read
+Identity,batch_normalization_24/moving_mean/read
+Identity,batch_normalization_24/moving_variance/read
+Identity,conv2d_28/kernel/read
+Identity,batch_normalization_25/gamma/read
+Identity,batch_normalization_25/beta/read
+Identity,batch_normalization_25/moving_mean/read
+Identity,batch_normalization_25/moving_variance/read
+Identity,conv2d_29/kernel/read
+Identity,batch_normalization_26/gamma/read
+Identity,batch_normalization_26/beta/read
+Identity,batch_normalization_26/moving_mean/read
+Identity,batch_normalization_26/moving_variance/read
+Identity,conv2d_30/kernel/read
+Identity,batch_normalization_27/gamma/read
+Identity,batch_normalization_27/beta/read
+Identity,batch_normalization_27/moving_mean/read
+Identity,batch_normalization_27/moving_variance/read
+Identity,conv2d_31/kernel/read
+Identity,batch_normalization_28/gamma/read
+Identity,batch_normalization_28/beta/read
+Identity,batch_normalization_28/moving_mean/read
+Identity,batch_normalization_28/moving_variance/read
+Identity,conv2d_32/kernel/read
+Identity,batch_normalization_29/gamma/read
+Identity,batch_normalization_29/beta/read
+Identity,batch_normalization_29/moving_mean/read
+Identity,batch_normalization_29/moving_variance/read
+Identity,conv2d_33/kernel/read
+Identity,batch_normalization_30/gamma/read
+Identity,batch_normalization_30/beta/read
+Identity,batch_normalization_30/moving_mean/read
+Identity,batch_normalization_30/moving_variance/read
+Identity,conv2d_34/kernel/read
+Identity,batch_normalization_31/gamma/read
+Identity,batch_normalization_31/beta/read
+Identity,batch_normalization_31/moving_mean/read
+Identity,batch_normalization_31/moving_variance/read
+Identity,conv2d_35/kernel/read
+Identity,batch_normalization_32/gamma/read
+Identity,batch_normalization_32/beta/read
+Identity,batch_normalization_32/moving_mean/read
+Identity,batch_normalization_32/moving_variance/read
+Identity,conv2d_36/kernel/read
+Identity,batch_normalization_33/gamma/read
+Identity,batch_normalization_33/beta/read
+Identity,batch_normalization_33/moving_mean/read
+Identity,batch_normalization_33/moving_variance/read
+Identity,conv2d_37/kernel/read
+Identity,batch_normalization_34/gamma/read
+Identity,batch_normalization_34/beta/read
+Identity,batch_normalization_34/moving_mean/read
+Identity,batch_normalization_34/moving_variance/read
+Identity,conv2d_38/kernel/read
+Identity,batch_normalization_35/gamma/read
+Identity,batch_normalization_35/beta/read
+Identity,batch_normalization_35/moving_mean/read
+Identity,batch_normalization_35/moving_variance/read
+Identity,conv2d_39/kernel/read
+Identity,batch_normalization_36/gamma/read
+Identity,batch_normalization_36/beta/read
+Identity,batch_normalization_36/moving_mean/read
+Identity,batch_normalization_36/moving_variance/read
+Identity,conv2d_40/kernel/read
+Identity,batch_normalization_37/gamma/read
+Identity,batch_normalization_37/beta/read
+Identity,batch_normalization_37/moving_mean/read
+Identity,batch_normalization_37/moving_variance/read
+Identity,conv2d_41/kernel/read
+Identity,batch_normalization_38/gamma/read
+Identity,batch_normalization_38/beta/read
+Identity,batch_normalization_38/moving_mean/read
+Identity,batch_normalization_38/moving_variance/read
+Identity,conv2d_42/kernel/read
+Identity,batch_normalization_39/gamma/read
+Identity,batch_normalization_39/beta/read
+Identity,batch_normalization_39/moving_mean/read
+Identity,batch_normalization_39/moving_variance/read
+Identity,conv2d_43/kernel/read
+Identity,conv2d_44/kernel/read
+Identity,batch_normalization_40/gamma/read
+Identity,batch_normalization_40/beta/read
+Identity,batch_normalization_40/moving_mean/read
+Identity,batch_normalization_40/moving_variance/read
+Identity,conv2d_45/kernel/read
+Identity,batch_normalization_41/gamma/read
+Identity,batch_normalization_41/beta/read
+Identity,batch_normalization_41/moving_mean/read
+Identity,batch_normalization_41/moving_variance/read
+Identity,conv2d_46/kernel/read
+Identity,batch_normalization_42/gamma/read
+Identity,batch_normalization_42/beta/read
+Identity,batch_normalization_42/moving_mean/read
+Identity,batch_normalization_42/moving_variance/read
+Identity,conv2d_47/kernel/read
+Identity,batch_normalization_43/gamma/read
+Identity,batch_normalization_43/beta/read
+Identity,batch_normalization_43/moving_mean/read
+Identity,batch_normalization_43/moving_variance/read
+Identity,conv2d_48/kernel/read
+Identity,batch_normalization_44/gamma/read
+Identity,batch_normalization_44/beta/read
+Identity,batch_normalization_44/moving_mean/read
+Identity,batch_normalization_44/moving_variance/read
+Identity,conv2d_49/kernel/read
+Identity,batch_normalization_45/gamma/read
+Identity,batch_normalization_45/beta/read
+Identity,batch_normalization_45/moving_mean/read
+Identity,batch_normalization_45/moving_variance/read
+Identity,conv2d_50/kernel/read
+Identity,batch_normalization_46/gamma/read
+Identity,batch_normalization_46/beta/read
+Identity,batch_normalization_46/moving_mean/read
+Identity,batch_normalization_46/moving_variance/read
+Identity,conv2d_51/kernel/read
+Identity,batch_normalization_47/gamma/read
+Identity,batch_normalization_47/beta/read
+Identity,batch_normalization_47/moving_mean/read
+Identity,batch_normalization_47/moving_variance/read
+Identity,conv2d_52/kernel/read
+Identity,batch_normalization_48/gamma/read
+Identity,batch_normalization_48/beta/read
+Identity,batch_normalization_48/moving_mean/read
+Identity,batch_normalization_48/moving_variance/read
+Identity,dense/kernel/read
+Identity,dense/bias/read
+Pad,Pad
+Conv2D,conv2d/Conv2D
+Identity,initial_conv
+MaxPool,max_pooling2d/MaxPool
+Identity,initial_max_pool
+FusedBatchNorm,batch_normalization/FusedBatchNorm
+Relu,Relu
+Conv2D,conv2d_1/Conv2D
+Conv2D,conv2d_2/Conv2D
+FusedBatchNorm,batch_normalization_1/FusedBatchNorm
+Relu,Relu_1
+Conv2D,conv2d_3/Conv2D
+FusedBatchNorm,batch_normalization_2/FusedBatchNorm
+Relu,Relu_2
+Conv2D,conv2d_4/Conv2D
+Add,add
+FusedBatchNorm,batch_normalization_3/FusedBatchNorm
+Relu,Relu_3
+Conv2D,conv2d_5/Conv2D
+FusedBatchNorm,batch_normalization_4/FusedBatchNorm
+Relu,Relu_4
+Conv2D,conv2d_6/Conv2D
+FusedBatchNorm,batch_normalization_5/FusedBatchNorm
+Relu,Relu_5
+Conv2D,conv2d_7/Conv2D
+Add,add_1
+FusedBatchNorm,batch_normalization_6/FusedBatchNorm
+Relu,Relu_6
+Conv2D,conv2d_8/Conv2D
+FusedBatchNorm,batch_normalization_7/FusedBatchNorm
+Relu,Relu_7
+Conv2D,conv2d_9/Conv2D
+FusedBatchNorm,batch_normalization_8/FusedBatchNorm
+Relu,Relu_8
+Conv2D,conv2d_10/Conv2D
+Add,add_2
+Identity,block_layer1
+FusedBatchNorm,batch_normalization_9/FusedBatchNorm
+Relu,Relu_9
+Pad,Pad_1
+Conv2D,conv2d_12/Conv2D
+Conv2D,conv2d_11/Conv2D
+FusedBatchNorm,batch_normalization_10/FusedBatchNorm
+Relu,Relu_10
+Pad,Pad_2
+Conv2D,conv2d_13/Conv2D
+FusedBatchNorm,batch_normalization_11/FusedBatchNorm
+Relu,Relu_11
+Conv2D,conv2d_14/Conv2D
+Add,add_3
+FusedBatchNorm,batch_normalization_12/FusedBatchNorm
+Relu,Relu_12
+Conv2D,conv2d_15/Conv2D
+FusedBatchNorm,batch_normalization_13/FusedBatchNorm
+Relu,Relu_13
+Conv2D,conv2d_16/Conv2D
+FusedBatchNorm,batch_normalization_14/FusedBatchNorm
+Relu,Relu_14
+Conv2D,conv2d_17/Conv2D
+Add,add_4
+FusedBatchNorm,batch_normalization_15/FusedBatchNorm
+Relu,Relu_15
+Conv2D,conv2d_18/Conv2D
+FusedBatchNorm,batch_normalization_16/FusedBatchNorm
+Relu,Relu_16
+Conv2D,conv2d_19/Conv2D
+FusedBatchNorm,batch_normalization_17/FusedBatchNorm
+Relu,Relu_17
+Conv2D,conv2d_20/Conv2D
+Add,add_5
+FusedBatchNorm,batch_normalization_18/FusedBatchNorm
+Relu,Relu_18
+Conv2D,conv2d_21/Conv2D
+FusedBatchNorm,batch_normalization_19/FusedBatchNorm
+Relu,Relu_19
+Conv2D,conv2d_22/Conv2D
+FusedBatchNorm,batch_normalization_20/FusedBatchNorm
+Relu,Relu_20
+Conv2D,conv2d_23/Conv2D
+Add,add_6
+Identity,block_layer2
+FusedBatchNorm,batch_normalization_21/FusedBatchNorm
+Relu,Relu_21
+Pad,Pad_3
+Conv2D,conv2d_25/Conv2D
+Conv2D,conv2d_24/Conv2D
+FusedBatchNorm,batch_normalization_22/FusedBatchNorm
+Relu,Relu_22
+Pad,Pad_4
+Conv2D,conv2d_26/Conv2D
+FusedBatchNorm,batch_normalization_23/FusedBatchNorm
+Relu,Relu_23
+Conv2D,conv2d_27/Conv2D
+Add,add_7
+FusedBatchNorm,batch_normalization_24/FusedBatchNorm
+Relu,Relu_24
+Conv2D,conv2d_28/Conv2D
+FusedBatchNorm,batch_normalization_25/FusedBatchNorm
+Relu,Relu_25
+Conv2D,conv2d_29/Conv2D
+FusedBatchNorm,batch_normalization_26/FusedBatchNorm
+Relu,Relu_26
+Conv2D,conv2d_30/Conv2D
+Add,add_8
+FusedBatchNorm,batch_normalization_27/FusedBatchNorm
+Relu,Relu_27
+Conv2D,conv2d_31/Conv2D
+FusedBatchNorm,batch_normalization_28/FusedBatchNorm
+Relu,Relu_28
+Conv2D,conv2d_32/Conv2D
+FusedBatchNorm,batch_normalization_29/FusedBatchNorm
+Relu,Relu_29
+Conv2D,conv2d_33/Conv2D
+Add,add_9
+FusedBatchNorm,batch_normalization_30/FusedBatchNorm
+Relu,Relu_30
+Conv2D,conv2d_34/Conv2D
+FusedBatchNorm,batch_normalization_31/FusedBatchNorm
+Relu,Relu_31
+Conv2D,conv2d_35/Conv2D
+FusedBatchNorm,batch_normalization_32/FusedBatchNorm
+Relu,Relu_32
+Conv2D,conv2d_36/Conv2D
+Add,add_10
+FusedBatchNorm,batch_normalization_33/FusedBatchNorm
+Relu,Relu_33
+Conv2D,conv2d_37/Conv2D
+FusedBatchNorm,batch_normalization_34/FusedBatchNorm
+Relu,Relu_34
+Conv2D,conv2d_38/Conv2D
+FusedBatchNorm,batch_normalization_35/FusedBatchNorm
+Relu,Relu_35
+Conv2D,conv2d_39/Conv2D
+Add,add_11
+FusedBatchNorm,batch_normalization_36/FusedBatchNorm
+Relu,Relu_36
+Conv2D,conv2d_40/Conv2D
+FusedBatchNorm,batch_normalization_37/FusedBatchNorm
+Relu,Relu_37
+Conv2D,conv2d_41/Conv2D
+FusedBatchNorm,batch_normalization_38/FusedBatchNorm
+Relu,Relu_38
+Conv2D,conv2d_42/Conv2D
+Add,add_12
+Identity,block_layer3
+FusedBatchNorm,batch_normalization_39/FusedBatchNorm
+Relu,Relu_39
+Pad,Pad_5
+Conv2D,conv2d_44/Conv2D
+Conv2D,conv2d_43/Conv2D
+FusedBatchNorm,batch_normalization_40/FusedBatchNorm
+Relu,Relu_40
+Pad,Pad_6
+Conv2D,conv2d_45/Conv2D
+FusedBatchNorm,batch_normalization_41/FusedBatchNorm
+Relu,Relu_41
+Conv2D,conv2d_46/Conv2D
+Add,add_13
+FusedBatchNorm,batch_normalization_42/FusedBatchNorm
+Relu,Relu_42
+Conv2D,conv2d_47/Conv2D
+FusedBatchNorm,batch_normalization_43/FusedBatchNorm
+Relu,Relu_43
+Conv2D,conv2d_48/Conv2D
+FusedBatchNorm,batch_normalization_44/FusedBatchNorm
+Relu,Relu_44
+Conv2D,conv2d_49/Conv2D
+Add,add_14
+FusedBatchNorm,batch_normalization_45/FusedBatchNorm
+Relu,Relu_45
+Conv2D,conv2d_50/Conv2D
+FusedBatchNorm,batch_normalization_46/FusedBatchNorm
+Relu,Relu_46
+Conv2D,conv2d_51/Conv2D
+FusedBatchNorm,batch_normalization_47/FusedBatchNorm
+Relu,Relu_47
+Conv2D,conv2d_52/Conv2D
+Add,add_15
+Identity,block_layer4
+FusedBatchNorm,batch_normalization_48/FusedBatchNorm
+Relu,Relu_48
+Mean,Mean
+Identity,final_reduce_mean
+Reshape,Reshape
+MatMul,dense/MatMul
+BiasAdd,dense/BiasAdd
+Identity,final_dense
+ArgMax,ArgMax
+Softmax,softmax_tensor
diff --git a/nd4j/nd4j-backends/nd4j-tests/src/test/java/org/nd4j/imports/TFGraphs/TFGraphTestAllSameDiff.java b/nd4j/nd4j-backends/nd4j-tests/src/test/java/org/nd4j/imports/TFGraphs/TFGraphTestAllSameDiff.java
index d9f5de304..6e4c999a5 100644
--- a/nd4j/nd4j-backends/nd4j-tests/src/test/java/org/nd4j/imports/TFGraphs/TFGraphTestAllSameDiff.java
+++ b/nd4j/nd4j-backends/nd4j-tests/src/test/java/org/nd4j/imports/TFGraphs/TFGraphTestAllSameDiff.java
@@ -91,6 +91,7 @@ public class TFGraphTestAllSameDiff { //Note: Can't extend BaseNd4jTest here a
"bincount/rank2_weights",
"slogdet/.*",
"fused_batch_norm/float16_nhwc",
+ "emptyArrayTests/scatter_update/rank2_emptyIndices_emptyUpdates",
//Don't bother to test RNG. We can test subsets of ops with dropout to make sure they are consistent
//These tests have random uniform and other RNG in them that don't need to be perfectly compatible to be acceptable.
//We need different test cases here.
diff --git a/nd4j/nd4j-backends/nd4j-tests/src/test/java/org/nd4j/imports/TFGraphs/TFGraphTestZooModels.java b/nd4j/nd4j-backends/nd4j-tests/src/test/java/org/nd4j/imports/TFGraphs/TFGraphTestZooModels.java
index c9bb119a5..2f0b6a2f6 100644
--- a/nd4j/nd4j-backends/nd4j-tests/src/test/java/org/nd4j/imports/TFGraphs/TFGraphTestZooModels.java
+++ b/nd4j/nd4j-backends/nd4j-tests/src/test/java/org/nd4j/imports/TFGraphs/TFGraphTestZooModels.java
@@ -24,10 +24,7 @@ import lombok.extern.slf4j.Slf4j;
import org.apache.commons.io.FileUtils;
import org.apache.commons.io.FilenameUtils;
import org.apache.commons.lang3.ArrayUtils;
-import org.junit.BeforeClass;
-import org.junit.ClassRule;
-import org.junit.Rule;
-import org.junit.Test;
+import org.junit.*;
import org.junit.rules.TemporaryFolder;
import org.junit.runner.RunWith;
import org.junit.runners.Parameterized;
@@ -53,6 +50,7 @@ import java.util.Map;
@RunWith(Parameterized.class)
@Slf4j
+@Ignore
public class TFGraphTestZooModels { //Note: Can't extend BaseNd4jTest here as we need no-arg constructor for parameterized tests
@ClassRule
@@ -176,7 +174,7 @@ public class TFGraphTestZooModels { //Note: Can't extend BaseNd4jTest here as we
} else {
//Multiple files... try to find "frozen_inference_graph.pb"
for(String str : pbFiles){
- if(str.endsWith("frozen_inference_graph.pb")){
+ if(str.endsWith("frozen_inference_graph.pb")) {
toExtract = str;
}
}
diff --git a/nd4j/nd4j-backends/nd4j-tests/src/test/java/org/nd4j/smoketests/SmokeTest.java b/nd4j/nd4j-backends/nd4j-tests/src/test/java/org/nd4j/smoketests/SmokeTest.java
index 91419ebef..74831083d 100644
--- a/nd4j/nd4j-backends/nd4j-tests/src/test/java/org/nd4j/smoketests/SmokeTest.java
+++ b/nd4j/nd4j-backends/nd4j-tests/src/test/java/org/nd4j/smoketests/SmokeTest.java
@@ -47,6 +47,9 @@ public class SmokeTest {
INDArray arr = Nd4j.randn(2,2);
INDArray arr2 = Nd4j.randn(2,2);
for(DataType dataType : DataType.values()) {
+ if(!dataType.isFPType()) {
+ continue;
+ }
log.info("Testing matrix multiply on data type {}",dataType);
INDArray casted = arr.castTo(dataType);
INDArray casted2 = arr2.castTo(dataType);
diff --git a/nd4j/samediff-import/pom.xml b/nd4j/samediff-import/pom.xml
index 6ddc995d7..7ce829081 100644
--- a/nd4j/samediff-import/pom.xml
+++ b/nd4j/samediff-import/pom.xml
@@ -37,11 +37,7 @@
pom
-
-
- testresources
-
-
+
samediff-import-api
diff --git a/nd4j/samediff-import/samediff-import-api/src/main/kotlin/org/nd4j/samediff/frameworkimport/opdefs/OpDescriptorLoader.kt b/nd4j/samediff-import/samediff-import-api/src/main/kotlin/org/nd4j/samediff/frameworkimport/opdefs/OpDescriptorLoader.kt
index b093a3dd6..1eff8b813 100644
--- a/nd4j/samediff-import/samediff-import-api/src/main/kotlin/org/nd4j/samediff/frameworkimport/opdefs/OpDescriptorLoader.kt
+++ b/nd4j/samediff-import/samediff-import-api/src/main/kotlin/org/nd4j/samediff/frameworkimport/opdefs/OpDescriptorLoader.kt
@@ -26,7 +26,7 @@ import org.nd4j.samediff.frameworkimport.registry.OpMappingRegistry
import org.nd4j.shade.protobuf.GeneratedMessageV3
import org.nd4j.shade.protobuf.ProtocolMessageEnum
-val nd4jFileNameTextDefault = "nd4j-op-def.pbtxt"
+val nd4jFileNameTextDefault = "/nd4j-op-def.pbtxt"
val nd4jFileSpecifierProperty = "samediff.import.nd4jdescriptors"
interface OpDescriptorLoader {
diff --git a/nd4j/samediff-import/samediff-import-api/src/main/resources/nd4j-op-def.pbtxt b/nd4j/samediff-import/samediff-import-api/src/main/resources/nd4j-op-def.pbtxt
index b7ee920f0..9cbb9c962 100644
--- a/nd4j/samediff-import/samediff-import-api/src/main/resources/nd4j-op-def.pbtxt
+++ b/nd4j/samediff-import/samediff-import-api/src/main/resources/nd4j-op-def.pbtxt
@@ -12,23 +12,14 @@ opList {
}
opList {
name: "BinaryMinimalRelativeError"
- argDescriptor {
- name: "outputs"
- argType: OUTPUT_TENSOR
- }
- argDescriptor {
- name: "input"
- argType: INPUT_TENSOR
- }
- argDescriptor {
- name: "y"
- argType: INPUT_TENSOR
- argIndex: 1
- }
argDescriptor {
name: "dataType"
argType: DATA_TYPE
}
+ argDescriptor {
+ name: "outputs"
+ argType: OUTPUT_TENSOR
+ }
argDescriptor {
name: "thresholdRelative"
argType: DOUBLE
@@ -38,18 +29,6 @@ opList {
argType: DOUBLE
argIndex: 1
}
- opDeclarationType: LEGACY_XYZ
-}
-opList {
- name: "BinaryRelativeError"
- argDescriptor {
- name: "outputs"
- argType: OUTPUT_TENSOR
- }
- argDescriptor {
- name: "inPlace"
- argType: BOOL
- }
argDescriptor {
name: "input"
argType: INPUT_TENSOR
@@ -59,14 +38,35 @@ opList {
argType: INPUT_TENSOR
argIndex: 1
}
+ opDeclarationType: LEGACY_XYZ
+}
+opList {
+ name: "BinaryRelativeError"
argDescriptor {
name: "dataType"
argType: DATA_TYPE
}
+ argDescriptor {
+ name: "outputs"
+ argType: OUTPUT_TENSOR
+ }
+ argDescriptor {
+ name: "inPlace"
+ argType: BOOL
+ }
argDescriptor {
name: "threshold"
argType: DOUBLE
}
+ argDescriptor {
+ name: "input"
+ argType: INPUT_TENSOR
+ }
+ argDescriptor {
+ name: "y"
+ argType: INPUT_TENSOR
+ argIndex: 1
+ }
opDeclarationType: LEGACY_XYZ
}
opList {
@@ -79,10 +79,6 @@ opList {
name: "inPlace"
argType: BOOL
}
- argDescriptor {
- name: "input"
- argType: INPUT_TENSOR
- }
argDescriptor {
name: "clipValueMin"
argType: DOUBLE
@@ -92,6 +88,10 @@ opList {
argType: DOUBLE
argIndex: 1
}
+ argDescriptor {
+ name: "input"
+ argType: INPUT_TENSOR
+ }
}
opList {
name: "Conditional"
@@ -158,6 +158,10 @@ opList {
}
opList {
name: "Pow_bp"
+ argDescriptor {
+ name: "dtype"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "dLdx"
argType: OUTPUT_TENSOR
@@ -195,13 +199,13 @@ opList {
argType: INPUT_TENSOR
argIndex: 4
}
- argDescriptor {
- name: "dtype"
- argType: DATA_TYPE
- }
}
opList {
name: "Reciprocal"
+ argDescriptor {
+ name: "dataType"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -210,14 +214,14 @@ opList {
name: "input"
argType: INPUT_TENSOR
}
- argDescriptor {
- name: "dataType"
- argType: DATA_TYPE
- }
opDeclarationType: LEGACY_XYZ
}
opList {
name: "RelativeError"
+ argDescriptor {
+ name: "dataType"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -235,10 +239,6 @@ opList {
argType: INPUT_TENSOR
argIndex: 1
}
- argDescriptor {
- name: "dataType"
- argType: DATA_TYPE
- }
opDeclarationType: LEGACY_XYZ
}
opList {
@@ -276,6 +276,10 @@ opList {
}
opList {
name: "Where"
+ argDescriptor {
+ name: "dtype"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -294,10 +298,6 @@ opList {
argType: INPUT_TENSOR
argIndex: 2
}
- argDescriptor {
- name: "dtype"
- argType: DATA_TYPE
- }
}
opList {
name: "While"
@@ -305,6 +305,10 @@ opList {
name: "outputs"
argType: OUTPUT_TENSOR
}
+ argDescriptor {
+ name: "frameName"
+ argType: STRING
+ }
argDescriptor {
name: "isConstant"
argType: BOOL
@@ -313,14 +317,14 @@ opList {
name: "condition"
argType: INPUT_TENSOR
}
- argDescriptor {
- name: "frameName"
- argType: STRING
- }
opDeclarationType: LOGIC_OP_IMPL
}
opList {
name: "_geluderivative"
+ argDescriptor {
+ name: "dataType"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -333,14 +337,14 @@ opList {
name: "input"
argType: INPUT_TENSOR
}
- argDescriptor {
- name: "dataType"
- argType: DATA_TYPE
- }
opDeclarationType: LEGACY_XYZ
}
opList {
name: "_mishderivative"
+ argDescriptor {
+ name: "dataType"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -358,10 +362,6 @@ opList {
argType: INPUT_TENSOR
argIndex: 1
}
- argDescriptor {
- name: "dataType"
- argType: DATA_TYPE
- }
opDeclarationType: LEGACY_XYZ
}
opList {
@@ -374,18 +374,22 @@ opList {
name: "inPlace"
argType: BOOL
}
- argDescriptor {
- name: "input"
- argType: INPUT_TENSOR
- }
argDescriptor {
name: "pow"
argType: DOUBLE
}
+ argDescriptor {
+ name: "input"
+ argType: INPUT_TENSOR
+ }
opDeclarationType: LEGACY_XYZ
}
opList {
name: "_precise_geluderivative"
+ argDescriptor {
+ name: "dataType"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -403,14 +407,14 @@ opList {
name: "input"
argType: INPUT_TENSOR
}
- argDescriptor {
- name: "dataType"
- argType: DATA_TYPE
- }
opDeclarationType: LEGACY_XYZ
}
opList {
name: "_sigmoidderivative"
+ argDescriptor {
+ name: "dataType"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -428,14 +432,14 @@ opList {
argType: INPUT_TENSOR
argIndex: 1
}
- argDescriptor {
- name: "dataType"
- argType: DATA_TYPE
- }
opDeclarationType: LEGACY_XYZ
}
opList {
name: "_swishderivative"
+ argDescriptor {
+ name: "dataType"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -453,14 +457,14 @@ opList {
argType: INPUT_TENSOR
argIndex: 1
}
- argDescriptor {
- name: "dataType"
- argType: DATA_TYPE
- }
opDeclarationType: LEGACY_XYZ
}
opList {
name: "_tanhderivative"
+ argDescriptor {
+ name: "dataType"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -473,14 +477,14 @@ opList {
name: "input"
argType: INPUT_TENSOR
}
- argDescriptor {
- name: "dataType"
- argType: DATA_TYPE
- }
opDeclarationType: LEGACY_XYZ
}
opList {
name: "abs"
+ argDescriptor {
+ name: "dataType"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -493,14 +497,18 @@ opList {
name: "input"
argType: INPUT_TENSOR
}
- argDescriptor {
- name: "dataType"
- argType: DATA_TYPE
- }
opDeclarationType: LEGACY_XYZ
}
opList {
name: "absolute_difference_loss"
+ argDescriptor {
+ name: "reductionMode"
+ argType: INT64
+ }
+ argDescriptor {
+ name: "dtype"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -519,6 +527,9 @@ opList {
argType: INPUT_TENSOR
argIndex: 2
}
+}
+opList {
+ name: "absolute_difference_loss_grad"
argDescriptor {
name: "reductionMode"
argType: INT64
@@ -527,9 +538,6 @@ opList {
name: "dtype"
argType: DATA_TYPE
}
-}
-opList {
- name: "absolute_difference_loss_grad"
argDescriptor {
name: "dLdp"
argType: OUTPUT_TENSOR
@@ -558,17 +566,13 @@ opList {
argType: INPUT_TENSOR
argIndex: 2
}
- argDescriptor {
- name: "reductionMode"
- argType: INT64
- }
- argDescriptor {
- name: "dtype"
- argType: DATA_TYPE
- }
}
opList {
name: "acos"
+ argDescriptor {
+ name: "dataType"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -581,14 +585,14 @@ opList {
name: "input"
argType: INPUT_TENSOR
}
- argDescriptor {
- name: "dataType"
- argType: DATA_TYPE
- }
opDeclarationType: LEGACY_XYZ
}
opList {
name: "acosh"
+ argDescriptor {
+ name: "dataType"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -601,10 +605,6 @@ opList {
name: "input"
argType: INPUT_TENSOR
}
- argDescriptor {
- name: "dataType"
- argType: DATA_TYPE
- }
opDeclarationType: LEGACY_XYZ
}
opList {
@@ -623,6 +623,15 @@ opList {
argType: OUTPUT_TENSOR
argIndex: 2
}
+ argDescriptor {
+ name: "dRho"
+ argType: DOUBLE
+ }
+ argDescriptor {
+ name: "dEpsilon"
+ argType: DOUBLE
+ argIndex: 1
+ }
argDescriptor {
name: "gradient"
argType: INPUT_TENSOR
@@ -652,15 +661,6 @@ opList {
argType: INPUT_TENSOR
argIndex: 5
}
- argDescriptor {
- name: "dRho"
- argType: DOUBLE
- }
- argDescriptor {
- name: "dEpsilon"
- argType: DOUBLE
- argIndex: 1
- }
opDeclarationType: CONFIGURABLE_OP_IMPL
}
opList {
@@ -674,6 +674,15 @@ opList {
argType: OUTPUT_TENSOR
argIndex: 1
}
+ argDescriptor {
+ name: "dLr"
+ argType: DOUBLE
+ }
+ argDescriptor {
+ name: "dEpsilon"
+ argType: DOUBLE
+ argIndex: 1
+ }
argDescriptor {
name: "gradient"
argType: INPUT_TENSOR
@@ -693,19 +702,14 @@ opList {
argType: INPUT_TENSOR
argIndex: 3
}
- argDescriptor {
- name: "dLr"
- argType: DOUBLE
- }
- argDescriptor {
- name: "dEpsilon"
- argType: DOUBLE
- argIndex: 1
- }
opDeclarationType: CONFIGURABLE_OP_IMPL
}
opList {
name: "ada_max_updater"
+ argDescriptor {
+ name: "iteration"
+ argType: INT64
+ }
argDescriptor {
name: "update"
argType: OUTPUT_TENSOR
@@ -720,6 +724,25 @@ opList {
argType: OUTPUT_TENSOR
argIndex: 2
}
+ argDescriptor {
+ name: "dLr"
+ argType: DOUBLE
+ }
+ argDescriptor {
+ name: "dBeta1"
+ argType: DOUBLE
+ argIndex: 1
+ }
+ argDescriptor {
+ name: "dBeta2"
+ argType: DOUBLE
+ argIndex: 2
+ }
+ argDescriptor {
+ name: "dEpsilon"
+ argType: DOUBLE
+ argIndex: 3
+ }
argDescriptor {
name: "gradient"
argType: INPUT_TENSOR
@@ -754,10 +777,28 @@ opList {
argType: INPUT_TENSOR
argIndex: 6
}
+ opDeclarationType: CONFIGURABLE_OP_IMPL
+}
+opList {
+ name: "adabelief_updater"
argDescriptor {
name: "iteration"
argType: INT64
}
+ argDescriptor {
+ name: "update"
+ argType: OUTPUT_TENSOR
+ }
+ argDescriptor {
+ name: "stateU"
+ argType: OUTPUT_TENSOR
+ argIndex: 1
+ }
+ argDescriptor {
+ name: "stateM"
+ argType: OUTPUT_TENSOR
+ argIndex: 2
+ }
argDescriptor {
name: "dLr"
argType: DOUBLE
@@ -777,10 +818,48 @@ opList {
argType: DOUBLE
argIndex: 3
}
+ argDescriptor {
+ name: "gradient"
+ argType: INPUT_TENSOR
+ }
+ argDescriptor {
+ name: "initStateU"
+ argType: INPUT_TENSOR
+ argIndex: 1
+ }
+ argDescriptor {
+ name: "initStateM"
+ argType: INPUT_TENSOR
+ argIndex: 2
+ }
+ argDescriptor {
+ name: "lr"
+ argType: INPUT_TENSOR
+ argIndex: 3
+ }
+ argDescriptor {
+ name: "beta1"
+ argType: INPUT_TENSOR
+ argIndex: 4
+ }
+ argDescriptor {
+ name: "beta2"
+ argType: INPUT_TENSOR
+ argIndex: 5
+ }
+ argDescriptor {
+ name: "epsilon"
+ argType: INPUT_TENSOR
+ argIndex: 6
+ }
opDeclarationType: CONFIGURABLE_OP_IMPL
}
opList {
name: "adam_updater"
+ argDescriptor {
+ name: "iteration"
+ argType: INT64
+ }
argDescriptor {
name: "update"
argType: OUTPUT_TENSOR
@@ -795,6 +874,25 @@ opList {
argType: OUTPUT_TENSOR
argIndex: 2
}
+ argDescriptor {
+ name: "dLr"
+ argType: DOUBLE
+ }
+ argDescriptor {
+ name: "dBeta1"
+ argType: DOUBLE
+ argIndex: 1
+ }
+ argDescriptor {
+ name: "dBeta2"
+ argType: DOUBLE
+ argIndex: 2
+ }
+ argDescriptor {
+ name: "dEpsilon"
+ argType: DOUBLE
+ argIndex: 3
+ }
argDescriptor {
name: "gradient"
argType: INPUT_TENSOR
@@ -829,33 +927,14 @@ opList {
argType: INPUT_TENSOR
argIndex: 6
}
- argDescriptor {
- name: "iteration"
- argType: INT64
- }
- argDescriptor {
- name: "dLr"
- argType: DOUBLE
- }
- argDescriptor {
- name: "dBeta1"
- argType: DOUBLE
- argIndex: 1
- }
- argDescriptor {
- name: "dBeta2"
- argType: DOUBLE
- argIndex: 2
- }
- argDescriptor {
- name: "dEpsilon"
- argType: DOUBLE
- argIndex: 3
- }
opDeclarationType: CONFIGURABLE_OP_IMPL
}
opList {
name: "add"
+ argDescriptor {
+ name: "dataType"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -873,14 +952,14 @@ opList {
argType: INPUT_TENSOR
argIndex: 1
}
- argDescriptor {
- name: "dataType"
- argType: DATA_TYPE
- }
opDeclarationType: BROADCASTABLE_OP_IMPL
}
opList {
name: "add_bp"
+ argDescriptor {
+ name: "dtype"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "gradX"
argType: OUTPUT_TENSOR
@@ -908,10 +987,6 @@ opList {
argType: INPUT_TENSOR
argIndex: 2
}
- argDescriptor {
- name: "dtype"
- argType: DATA_TYPE
- }
}
opList {
name: "add_scalar"
@@ -952,6 +1027,10 @@ opList {
name: "outputs"
argType: OUTPUT_TENSOR
}
+ argDescriptor {
+ name: "factor"
+ argType: DOUBLE
+ }
argDescriptor {
name: "input"
argType: INPUT_TENSOR
@@ -961,14 +1040,14 @@ opList {
argType: INPUT_TENSOR
argIndex: 1
}
- argDescriptor {
- name: "factor"
- argType: DOUBLE
- }
opDeclarationType: CONFIGURABLE_OP_IMPL
}
opList {
name: "adjust_hue"
+ argDescriptor {
+ name: "dimC"
+ argType: INT64
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -982,22 +1061,22 @@ opList {
argType: INPUT_TENSOR
argIndex: 1
}
- argDescriptor {
- name: "dimC"
- argType: INT64
- }
- argDescriptor {
- name: "factor"
- argType: DOUBLE
- }
opDeclarationType: CONFIGURABLE_OP_IMPL
}
opList {
name: "adjust_saturation"
+ argDescriptor {
+ name: "dimC"
+ argType: INT64
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
}
+ argDescriptor {
+ name: "factor"
+ argType: DOUBLE
+ }
argDescriptor {
name: "input"
argType: INPUT_TENSOR
@@ -1007,18 +1086,14 @@ opList {
argType: INPUT_TENSOR
argIndex: 1
}
- argDescriptor {
- name: "dimC"
- argType: INT64
- }
- argDescriptor {
- name: "factor"
- argType: DOUBLE
- }
opDeclarationType: CONFIGURABLE_OP_IMPL
}
opList {
name: "all"
+ argDescriptor {
+ name: "dimensions"
+ argType: INT64
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -1036,10 +1111,6 @@ opList {
argType: INPUT_TENSOR
argIndex: 1
}
- argDescriptor {
- name: "dimensions"
- argType: INT64
- }
opDeclarationType: LEGACY_XYZ
}
opList {
@@ -1048,10 +1119,6 @@ opList {
name: "outputs"
argType: OUTPUT_TENSOR
}
- argDescriptor {
- name: "input"
- argType: INPUT_TENSOR
- }
argDescriptor {
name: "a"
argType: DOUBLE
@@ -1071,32 +1138,22 @@ opList {
argType: DOUBLE
argIndex: 3
}
+ argDescriptor {
+ name: "input"
+ argType: INPUT_TENSOR
+ }
opDeclarationType: LEGACY_XYZ
}
opList {
name: "alpha_dropout_bp"
- argDescriptor {
- name: "outputs"
- argType: OUTPUT_TENSOR
- }
- argDescriptor {
- name: "input"
- argType: INPUT_TENSOR
- }
- argDescriptor {
- name: "gradOut"
- argType: INPUT_TENSOR
- argIndex: 1
- }
- argDescriptor {
- name: "reduceShape"
- argType: INPUT_TENSOR
- argIndex: 2
- }
argDescriptor {
name: "seed"
argType: INT64
}
+ argDescriptor {
+ name: "outputs"
+ argType: OUTPUT_TENSOR
+ }
argDescriptor {
name: "probValue"
argType: DOUBLE
@@ -1116,10 +1173,28 @@ opList {
argType: DOUBLE
argIndex: 3
}
+ argDescriptor {
+ name: "input"
+ argType: INPUT_TENSOR
+ }
+ argDescriptor {
+ name: "gradOut"
+ argType: INPUT_TENSOR
+ argIndex: 1
+ }
+ argDescriptor {
+ name: "reduceShape"
+ argType: INPUT_TENSOR
+ argIndex: 2
+ }
opDeclarationType: CONFIGURABLE_OP_IMPL
}
opList {
name: "amax"
+ argDescriptor {
+ name: "dimensions"
+ argType: INT64
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -1137,14 +1212,14 @@ opList {
argType: INPUT_TENSOR
argIndex: 1
}
- argDescriptor {
- name: "dimensions"
- argType: INT64
- }
opDeclarationType: LEGACY_XYZ
}
opList {
name: "amax_pairwise"
+ argDescriptor {
+ name: "dataType"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -1158,14 +1233,14 @@ opList {
argType: INPUT_TENSOR
argIndex: 1
}
- argDescriptor {
- name: "dataType"
- argType: DATA_TYPE
- }
opDeclarationType: LEGACY_XYZ
}
opList {
name: "amean"
+ argDescriptor {
+ name: "dimensions"
+ argType: INT64
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -1183,14 +1258,14 @@ opList {
argType: INPUT_TENSOR
argIndex: 1
}
- argDescriptor {
- name: "dimensions"
- argType: INT64
- }
opDeclarationType: LEGACY_XYZ
}
opList {
name: "amin"
+ argDescriptor {
+ name: "dimensions"
+ argType: INT64
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -1208,14 +1283,14 @@ opList {
argType: INPUT_TENSOR
argIndex: 1
}
- argDescriptor {
- name: "dimensions"
- argType: INT64
- }
opDeclarationType: LEGACY_XYZ
}
opList {
name: "amin_pairwise"
+ argDescriptor {
+ name: "dataType"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -1229,14 +1304,14 @@ opList {
argType: INPUT_TENSOR
argIndex: 1
}
- argDescriptor {
- name: "dataType"
- argType: DATA_TYPE
- }
opDeclarationType: LEGACY_XYZ
}
opList {
name: "ams_grad_updater"
+ argDescriptor {
+ name: "iteration"
+ argType: INT64
+ }
argDescriptor {
name: "update"
argType: OUTPUT_TENSOR
@@ -1256,6 +1331,25 @@ opList {
argType: OUTPUT_TENSOR
argIndex: 3
}
+ argDescriptor {
+ name: "dLr"
+ argType: DOUBLE
+ }
+ argDescriptor {
+ name: "dBeta1"
+ argType: DOUBLE
+ argIndex: 1
+ }
+ argDescriptor {
+ name: "dBeta2"
+ argType: DOUBLE
+ argIndex: 2
+ }
+ argDescriptor {
+ name: "dEpsilon"
+ argType: DOUBLE
+ argIndex: 3
+ }
argDescriptor {
name: "gradient"
argType: INPUT_TENSOR
@@ -1295,52 +1389,30 @@ opList {
argType: INPUT_TENSOR
argIndex: 7
}
- argDescriptor {
- name: "iteration"
- argType: INT64
- }
- argDescriptor {
- name: "dLr"
- argType: DOUBLE
- }
- argDescriptor {
- name: "dBeta1"
- argType: DOUBLE
- argIndex: 1
- }
- argDescriptor {
- name: "dBeta2"
- argType: DOUBLE
- argIndex: 2
- }
- argDescriptor {
- name: "dEpsilon"
- argType: DOUBLE
- argIndex: 3
- }
opDeclarationType: CONFIGURABLE_OP_IMPL
}
opList {
name: "and"
+ argDescriptor {
+ name: "dataType"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
}
argDescriptor {
- name: "inPlace"
- argType: BOOL
+ name: "comparable"
+ argType: DOUBLE
}
argDescriptor {
name: "input"
argType: INPUT_TENSOR
}
argDescriptor {
- name: "dataType"
- argType: DATA_TYPE
- }
- argDescriptor {
- name: "comparable"
- argType: DOUBLE
+ name: "y"
+ argType: INPUT_TENSOR
+ argIndex: 1
}
opDeclarationType: LEGACY_XYZ
}
@@ -1362,6 +1434,10 @@ opList {
}
opList {
name: "any"
+ argDescriptor {
+ name: "dimensions"
+ argType: INT64
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -1379,10 +1455,6 @@ opList {
argType: INPUT_TENSOR
argIndex: 1
}
- argDescriptor {
- name: "dimensions"
- argType: INT64
- }
opDeclarationType: LEGACY_XYZ
}
opList {
@@ -1391,6 +1463,10 @@ opList {
name: "Z"
argType: OUTPUT_TENSOR
}
+ argDescriptor {
+ name: "lr"
+ argType: DOUBLE
+ }
argDescriptor {
name: "parameters"
argType: INPUT_TENSOR
@@ -1405,10 +1481,6 @@ opList {
argType: INPUT_TENSOR
argIndex: 2
}
- argDescriptor {
- name: "lr"
- argType: DOUBLE
- }
opDeclarationType: CONFIGURABLE_OP_IMPL
}
opList {
@@ -1420,6 +1492,14 @@ opList {
}
opList {
name: "argamax"
+ argDescriptor {
+ name: "dimensions"
+ argType: INT64
+ }
+ argDescriptor {
+ name: "dtype"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -1437,17 +1517,17 @@ opList {
argType: INPUT_TENSOR
argIndex: 1
}
- argDescriptor {
- name: "dimensions"
- argType: INT64
- }
- argDescriptor {
- name: "dtype"
- argType: DATA_TYPE
- }
}
opList {
name: "argamin"
+ argDescriptor {
+ name: "dimensions"
+ argType: INT64
+ }
+ argDescriptor {
+ name: "dtype"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -1465,17 +1545,17 @@ opList {
argType: INPUT_TENSOR
argIndex: 1
}
- argDescriptor {
- name: "dimensions"
- argType: INT64
- }
- argDescriptor {
- name: "dtype"
- argType: DATA_TYPE
- }
}
opList {
name: "argmax"
+ argDescriptor {
+ name: "dimensions"
+ argType: INT64
+ }
+ argDescriptor {
+ name: "dtype"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -1493,17 +1573,17 @@ opList {
argType: INPUT_TENSOR
argIndex: 1
}
- argDescriptor {
- name: "dimensions"
- argType: INT64
- }
- argDescriptor {
- name: "dtype"
- argType: DATA_TYPE
- }
}
opList {
name: "argmin"
+ argDescriptor {
+ name: "dimensions"
+ argType: INT64
+ }
+ argDescriptor {
+ name: "dtype"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -1521,17 +1601,13 @@ opList {
argType: INPUT_TENSOR
argIndex: 1
}
- argDescriptor {
- name: "dimensions"
- argType: INT64
- }
- argDescriptor {
- name: "dtype"
- argType: DATA_TYPE
- }
}
opList {
name: "asin"
+ argDescriptor {
+ name: "dataType"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -1544,14 +1620,14 @@ opList {
name: "input"
argType: INPUT_TENSOR
}
- argDescriptor {
- name: "dataType"
- argType: DATA_TYPE
- }
opDeclarationType: LEGACY_XYZ
}
opList {
name: "asinh"
+ argDescriptor {
+ name: "dataType"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -1564,10 +1640,6 @@ opList {
name: "input"
argType: INPUT_TENSOR
}
- argDescriptor {
- name: "dataType"
- argType: DATA_TYPE
- }
opDeclarationType: LEGACY_XYZ
}
opList {
@@ -1589,6 +1661,10 @@ opList {
}
opList {
name: "assign_bp"
+ argDescriptor {
+ name: "dtype"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "gradX"
argType: OUTPUT_TENSOR
@@ -1612,13 +1688,13 @@ opList {
argType: INPUT_TENSOR
argIndex: 2
}
- argDescriptor {
- name: "dtype"
- argType: DATA_TYPE
- }
}
opList {
name: "asum"
+ argDescriptor {
+ name: "dimensions"
+ argType: INT64
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -1636,14 +1712,14 @@ opList {
argType: INPUT_TENSOR
argIndex: 1
}
- argDescriptor {
- name: "dimensions"
- argType: INT64
- }
opDeclarationType: LEGACY_XYZ
}
opList {
name: "atan"
+ argDescriptor {
+ name: "dataType"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -1656,14 +1732,14 @@ opList {
name: "input"
argType: INPUT_TENSOR
}
- argDescriptor {
- name: "dataType"
- argType: DATA_TYPE
- }
opDeclarationType: LEGACY_XYZ
}
opList {
name: "atanh"
+ argDescriptor {
+ name: "dataType"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -1676,22 +1752,10 @@ opList {
name: "input"
argType: INPUT_TENSOR
}
- argDescriptor {
- name: "dataType"
- argType: DATA_TYPE
- }
opDeclarationType: LEGACY_XYZ
}
opList {
name: "avgpool2d"
- argDescriptor {
- name: "outputs"
- argType: OUTPUT_TENSOR
- }
- argDescriptor {
- name: "input"
- argType: INPUT_TENSOR
- }
argDescriptor {
name: "kH"
argType: INT64
@@ -1750,22 +1814,17 @@ opList {
name: "dtype"
argType: DATA_TYPE
}
+ argDescriptor {
+ name: "outputs"
+ argType: OUTPUT_TENSOR
+ }
+ argDescriptor {
+ name: "input"
+ argType: INPUT_TENSOR
+ }
}
opList {
name: "avgpool2d_bp"
- argDescriptor {
- name: "gradI"
- argType: OUTPUT_TENSOR
- }
- argDescriptor {
- name: "input"
- argType: INPUT_TENSOR
- }
- argDescriptor {
- name: "gradO"
- argType: INPUT_TENSOR
- argIndex: 1
- }
argDescriptor {
name: "kH"
argType: INT64
@@ -1824,98 +1883,6 @@ opList {
name: "dtype"
argType: DATA_TYPE
}
-}
-opList {
- name: "avgpool3dnew"
- argDescriptor {
- name: "outputs"
- argType: OUTPUT_TENSOR
- }
- argDescriptor {
- name: "input"
- argType: INPUT_TENSOR
- }
- argDescriptor {
- name: "kD"
- argType: INT64
- }
- argDescriptor {
- name: "kH"
- argType: INT64
- argIndex: 1
- }
- argDescriptor {
- name: "kW"
- argType: INT64
- argIndex: 2
- }
- argDescriptor {
- name: "sD"
- argType: INT64
- argIndex: 3
- }
- argDescriptor {
- name: "sH"
- argType: INT64
- argIndex: 4
- }
- argDescriptor {
- name: "sW"
- argType: INT64
- argIndex: 5
- }
- argDescriptor {
- name: "pD"
- argType: INT64
- argIndex: 6
- }
- argDescriptor {
- name: "pH"
- argType: INT64
- argIndex: 7
- }
- argDescriptor {
- name: "pW"
- argType: INT64
- argIndex: 8
- }
- argDescriptor {
- name: "dD"
- argType: INT64
- argIndex: 9
- }
- argDescriptor {
- name: "dH"
- argType: INT64
- argIndex: 10
- }
- argDescriptor {
- name: "dW"
- argType: INT64
- argIndex: 11
- }
- argDescriptor {
- name: "isSameMode"
- argType: INT64
- argIndex: 12
- }
- argDescriptor {
- name: "extraParam0"
- argType: INT64
- argIndex: 13
- }
- argDescriptor {
- name: "isNCDHW"
- argType: INT64
- argIndex: 14
- }
- argDescriptor {
- name: "dtype"
- argType: DATA_TYPE
- }
-}
-opList {
- name: "avgpool3dnew_bp"
argDescriptor {
name: "gradI"
argType: OUTPUT_TENSOR
@@ -1929,6 +1896,9 @@ opList {
argType: INPUT_TENSOR
argIndex: 1
}
+}
+opList {
+ name: "avgpool3dnew"
argDescriptor {
name: "kD"
argType: INT64
@@ -2007,13 +1977,127 @@ opList {
name: "dtype"
argType: DATA_TYPE
}
+ argDescriptor {
+ name: "outputs"
+ argType: OUTPUT_TENSOR
+ }
+ argDescriptor {
+ name: "input"
+ argType: INPUT_TENSOR
+ }
+}
+opList {
+ name: "avgpool3dnew_bp"
+ argDescriptor {
+ name: "kD"
+ argType: INT64
+ }
+ argDescriptor {
+ name: "kH"
+ argType: INT64
+ argIndex: 1
+ }
+ argDescriptor {
+ name: "kW"
+ argType: INT64
+ argIndex: 2
+ }
+ argDescriptor {
+ name: "sD"
+ argType: INT64
+ argIndex: 3
+ }
+ argDescriptor {
+ name: "sH"
+ argType: INT64
+ argIndex: 4
+ }
+ argDescriptor {
+ name: "sW"
+ argType: INT64
+ argIndex: 5
+ }
+ argDescriptor {
+ name: "pD"
+ argType: INT64
+ argIndex: 6
+ }
+ argDescriptor {
+ name: "pH"
+ argType: INT64
+ argIndex: 7
+ }
+ argDescriptor {
+ name: "pW"
+ argType: INT64
+ argIndex: 8
+ }
+ argDescriptor {
+ name: "dD"
+ argType: INT64
+ argIndex: 9
+ }
+ argDescriptor {
+ name: "dH"
+ argType: INT64
+ argIndex: 10
+ }
+ argDescriptor {
+ name: "dW"
+ argType: INT64
+ argIndex: 11
+ }
+ argDescriptor {
+ name: "isSameMode"
+ argType: INT64
+ argIndex: 12
+ }
+ argDescriptor {
+ name: "extraParam0"
+ argType: INT64
+ argIndex: 13
+ }
+ argDescriptor {
+ name: "isNCDHW"
+ argType: INT64
+ argIndex: 14
+ }
+ argDescriptor {
+ name: "dtype"
+ argType: DATA_TYPE
+ }
+ argDescriptor {
+ name: "gradI"
+ argType: OUTPUT_TENSOR
+ }
+ argDescriptor {
+ name: "input"
+ argType: INPUT_TENSOR
+ }
+ argDescriptor {
+ name: "gradO"
+ argType: INPUT_TENSOR
+ argIndex: 1
+ }
}
opList {
name: "axpy"
+ argDescriptor {
+ name: "n"
+ argType: INT64
+ }
+ argDescriptor {
+ name: "dataType"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
}
+ argDescriptor {
+ name: "a"
+ argType: DOUBLE
+ }
argDescriptor {
name: "input"
argType: INPUT_TENSOR
@@ -2028,22 +2112,18 @@ opList {
argType: INPUT_TENSOR
argIndex: 2
}
- argDescriptor {
- name: "n"
- argType: INT64
- }
- argDescriptor {
- name: "dataType"
- argType: DATA_TYPE
- }
- argDescriptor {
- name: "a"
- argType: DOUBLE
- }
opDeclarationType: CONFIGURABLE_OP_IMPL
}
opList {
name: "barnes_edge_forces"
+ argDescriptor {
+ name: "N"
+ argType: INT64
+ }
+ argDescriptor {
+ name: "dtype"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -2067,14 +2147,6 @@ opList {
argType: INPUT_TENSOR
argIndex: 3
}
- argDescriptor {
- name: "N"
- argType: INT64
- }
- argDescriptor {
- name: "dtype"
- argType: DATA_TYPE
- }
}
opList {
name: "barnes_gains"
@@ -2100,6 +2172,14 @@ opList {
}
opList {
name: "barnes_symmetrized"
+ argDescriptor {
+ name: "N"
+ argType: INT64
+ }
+ argDescriptor {
+ name: "dtype"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputRows"
argType: OUTPUT_TENSOR
@@ -2133,30 +2213,9 @@ opList {
argType: INPUT_TENSOR
argIndex: 3
}
- argDescriptor {
- name: "N"
- argType: INT64
- }
- argDescriptor {
- name: "dtype"
- argType: DATA_TYPE
- }
}
opList {
name: "batch_to_space"
- argDescriptor {
- name: "outputs"
- argType: OUTPUT_TENSOR
- }
- argDescriptor {
- name: "input"
- argType: INPUT_TENSOR
- }
- argDescriptor {
- name: "crop"
- argType: INPUT_TENSOR
- argIndex: 1
- }
argDescriptor {
name: "blockSize"
argType: INT64
@@ -2175,9 +2234,30 @@ opList {
name: "dtype"
argType: DATA_TYPE
}
+ argDescriptor {
+ name: "outputs"
+ argType: OUTPUT_TENSOR
+ }
+ argDescriptor {
+ name: "input"
+ argType: INPUT_TENSOR
+ }
+ argDescriptor {
+ name: "crop"
+ argType: INPUT_TENSOR
+ argIndex: 1
+ }
}
opList {
name: "batch_to_space_nd"
+ argDescriptor {
+ name: "blocks"
+ argType: INT64
+ }
+ argDescriptor {
+ name: "dtype"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -2200,49 +2280,9 @@ opList {
argType: INPUT_TENSOR
argIndex: 2
}
- argDescriptor {
- name: "blocks"
- argType: INT64
- }
- argDescriptor {
- name: "dtype"
- argType: DATA_TYPE
- }
}
opList {
name: "batched_gemm"
- argDescriptor {
- name: "vC"
- argType: OUTPUT_TENSOR
- }
- argDescriptor {
- name: "transposeA"
- argType: BOOL
- }
- argDescriptor {
- name: "transposeB"
- argType: BOOL
- argIndex: 1
- }
- argDescriptor {
- name: "alpha"
- argType: INPUT_TENSOR
- }
- argDescriptor {
- name: "beta"
- argType: INPUT_TENSOR
- argIndex: 1
- }
- argDescriptor {
- name: "vA"
- argType: INPUT_TENSOR
- argIndex: 2
- }
- argDescriptor {
- name: "vB"
- argType: INPUT_TENSOR
- argIndex: 3
- }
argDescriptor {
name: "transA"
argType: INT64
@@ -2291,13 +2331,62 @@ opList {
name: "dtype"
argType: DATA_TYPE
}
+ argDescriptor {
+ name: "vC"
+ argType: OUTPUT_TENSOR
+ }
+ argDescriptor {
+ name: "transposeA"
+ argType: BOOL
+ }
+ argDescriptor {
+ name: "transposeB"
+ argType: BOOL
+ argIndex: 1
+ }
+ argDescriptor {
+ name: "alpha"
+ argType: INPUT_TENSOR
+ }
+ argDescriptor {
+ name: "beta"
+ argType: INPUT_TENSOR
+ argIndex: 1
+ }
+ argDescriptor {
+ name: "vA"
+ argType: INPUT_TENSOR
+ argIndex: 2
+ }
+ argDescriptor {
+ name: "vB"
+ argType: INPUT_TENSOR
+ argIndex: 3
+ }
}
opList {
name: "batchnorm"
+ argDescriptor {
+ name: "applyScale"
+ argType: INT64
+ }
+ argDescriptor {
+ name: "applyOffset"
+ argType: INT64
+ argIndex: 1
+ }
+ argDescriptor {
+ name: "dtype"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
}
+ argDescriptor {
+ name: "epsilon"
+ argType: DOUBLE
+ }
argDescriptor {
name: "inPlace"
argType: BOOL
@@ -2331,6 +2420,9 @@ opList {
argType: INPUT_TENSOR
argIndex: 3
}
+}
+opList {
+ name: "batchnorm_bp"
argDescriptor {
name: "applyScale"
argType: INT64
@@ -2344,13 +2436,6 @@ opList {
name: "dtype"
argType: DATA_TYPE
}
- argDescriptor {
- name: "epsilon"
- argType: DOUBLE
- }
-}
-opList {
- name: "batchnorm_bp"
argDescriptor {
name: "dLdI"
argType: OUTPUT_TENSOR
@@ -2370,6 +2455,10 @@ opList {
argType: OUTPUT_TENSOR
argIndex: 3
}
+ argDescriptor {
+ name: "epsilon"
+ argType: DOUBLE
+ }
argDescriptor {
name: "inPlace"
argType: BOOL
@@ -2403,23 +2492,6 @@ opList {
argType: INPUT_TENSOR
argIndex: 3
}
- argDescriptor {
- name: "applyScale"
- argType: INT64
- }
- argDescriptor {
- name: "applyOffset"
- argType: INT64
- argIndex: 1
- }
- argDescriptor {
- name: "dtype"
- argType: DATA_TYPE
- }
- argDescriptor {
- name: "epsilon"
- argType: DOUBLE
- }
}
opList {
name: "betainc"
@@ -2445,6 +2517,10 @@ opList {
}
opList {
name: "biasadd"
+ argDescriptor {
+ name: "dtype"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -2462,13 +2538,13 @@ opList {
argType: INPUT_TENSOR
argIndex: 1
}
+}
+opList {
+ name: "biasadd_bp"
argDescriptor {
name: "dtype"
argType: DATA_TYPE
}
-}
-opList {
- name: "biasadd_bp"
argDescriptor {
name: "gradI"
argType: OUTPUT_TENSOR
@@ -2496,13 +2572,27 @@ opList {
argType: INPUT_TENSOR
argIndex: 2
}
- argDescriptor {
- name: "dtype"
- argType: DATA_TYPE
- }
}
opList {
name: "bincount"
+ argDescriptor {
+ name: "minLength"
+ argType: INT64
+ }
+ argDescriptor {
+ name: "maxLength"
+ argType: INT64
+ argIndex: 1
+ }
+ argDescriptor {
+ name: "outputType"
+ argType: INT64
+ argIndex: 2
+ }
+ argDescriptor {
+ name: "outputType"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -2526,35 +2616,9 @@ opList {
argType: INPUT_TENSOR
argIndex: 3
}
- argDescriptor {
- name: "minLength"
- argType: INT64
- }
- argDescriptor {
- name: "maxLength"
- argType: INT64
- argIndex: 1
- }
- argDescriptor {
- name: "outputType"
- argType: INT64
- argIndex: 2
- }
- argDescriptor {
- name: "outputType"
- argType: DATA_TYPE
- }
}
opList {
name: "bitcast"
- argDescriptor {
- name: "outputs"
- argType: OUTPUT_TENSOR
- }
- argDescriptor {
- name: "input"
- argType: INPUT_TENSOR
- }
argDescriptor {
name: "newType"
argType: INT64
@@ -2563,9 +2627,26 @@ opList {
name: "dtype"
argType: DATA_TYPE
}
+ argDescriptor {
+ name: "outputs"
+ argType: OUTPUT_TENSOR
+ }
+ argDescriptor {
+ name: "input"
+ argType: INPUT_TENSOR
+ }
+ argDescriptor {
+ name: "dataType"
+ argType: INPUT_TENSOR
+ argIndex: 1
+ }
}
opList {
name: "bits_hamming_distance"
+ argDescriptor {
+ name: "dtype"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -2579,13 +2660,13 @@ opList {
argType: INPUT_TENSOR
argIndex: 1
}
- argDescriptor {
- name: "dtype"
- argType: DATA_TYPE
- }
}
opList {
name: "bitwise_and"
+ argDescriptor {
+ name: "dataType"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -2603,14 +2684,14 @@ opList {
argType: INPUT_TENSOR
argIndex: 1
}
- argDescriptor {
- name: "dataType"
- argType: DATA_TYPE
- }
opDeclarationType: BROADCASTABLE_OP_IMPL
}
opList {
name: "bitwise_or"
+ argDescriptor {
+ name: "dataType"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -2628,14 +2709,14 @@ opList {
argType: INPUT_TENSOR
argIndex: 1
}
- argDescriptor {
- name: "dataType"
- argType: DATA_TYPE
- }
opDeclarationType: BROADCASTABLE_OP_IMPL
}
opList {
name: "bitwise_xor"
+ argDescriptor {
+ name: "dataType"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -2653,14 +2734,14 @@ opList {
argType: INPUT_TENSOR
argIndex: 1
}
- argDescriptor {
- name: "dataType"
- argType: DATA_TYPE
- }
opDeclarationType: BROADCASTABLE_OP_IMPL
}
opList {
name: "bool_not"
+ argDescriptor {
+ name: "dataType"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -2669,10 +2750,6 @@ opList {
name: "input"
argType: INPUT_TENSOR
}
- argDescriptor {
- name: "dataType"
- argType: DATA_TYPE
- }
opDeclarationType: LEGACY_XYZ
}
opList {
@@ -2745,6 +2822,15 @@ opList {
}
opList {
name: "broadcast_amax"
+ argDescriptor {
+ name: "shape"
+ argType: INT64
+ }
+ argDescriptor {
+ name: "dimension"
+ argType: INT64
+ argIndex: 1
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -2757,19 +2843,19 @@ opList {
name: "input"
argType: INPUT_TENSOR
}
- argDescriptor {
- name: "shape"
- argType: INT64
- }
- argDescriptor {
- name: "dimension"
- argType: INT64
- argIndex: 1
- }
opDeclarationType: LEGACY_XYZ
}
opList {
name: "broadcast_amin"
+ argDescriptor {
+ name: "shape"
+ argType: INT64
+ }
+ argDescriptor {
+ name: "dimension"
+ argType: INT64
+ argIndex: 1
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -2782,19 +2868,14 @@ opList {
name: "input"
argType: INPUT_TENSOR
}
- argDescriptor {
- name: "shape"
- argType: INT64
- }
- argDescriptor {
- name: "dimension"
- argType: INT64
- argIndex: 1
- }
opDeclarationType: LEGACY_XYZ
}
opList {
name: "broadcast_dynamic_shape"
+ argDescriptor {
+ name: "dtype"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -2808,13 +2889,13 @@ opList {
argType: INPUT_TENSOR
argIndex: 1
}
- argDescriptor {
- name: "dtype"
- argType: DATA_TYPE
- }
}
opList {
name: "broadcast_equalto"
+ argDescriptor {
+ name: "dimension"
+ argType: INT64
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -2832,14 +2913,19 @@ opList {
argType: INPUT_TENSOR
argIndex: 1
}
- argDescriptor {
- name: "dimension"
- argType: INT64
- }
opDeclarationType: LEGACY_XYZ
}
opList {
name: "broadcast_greaterthan"
+ argDescriptor {
+ name: "shape"
+ argType: INT64
+ }
+ argDescriptor {
+ name: "dimension"
+ argType: INT64
+ argIndex: 1
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -2852,19 +2938,19 @@ opList {
name: "input"
argType: INPUT_TENSOR
}
- argDescriptor {
- name: "shape"
- argType: INT64
- }
- argDescriptor {
- name: "dimension"
- argType: INT64
- argIndex: 1
- }
opDeclarationType: LEGACY_XYZ
}
opList {
name: "broadcast_greaterthanorequal"
+ argDescriptor {
+ name: "shape"
+ argType: INT64
+ }
+ argDescriptor {
+ name: "dimension"
+ argType: INT64
+ argIndex: 1
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -2877,19 +2963,19 @@ opList {
name: "input"
argType: INPUT_TENSOR
}
- argDescriptor {
- name: "shape"
- argType: INT64
- }
- argDescriptor {
- name: "dimension"
- argType: INT64
- argIndex: 1
- }
opDeclarationType: LEGACY_XYZ
}
opList {
name: "broadcast_lessthan"
+ argDescriptor {
+ name: "shape"
+ argType: INT64
+ }
+ argDescriptor {
+ name: "dimension"
+ argType: INT64
+ argIndex: 1
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -2902,19 +2988,19 @@ opList {
name: "input"
argType: INPUT_TENSOR
}
- argDescriptor {
- name: "shape"
- argType: INT64
- }
- argDescriptor {
- name: "dimension"
- argType: INT64
- argIndex: 1
- }
opDeclarationType: LEGACY_XYZ
}
opList {
name: "broadcast_lessthanorequal"
+ argDescriptor {
+ name: "shape"
+ argType: INT64
+ }
+ argDescriptor {
+ name: "dimension"
+ argType: INT64
+ argIndex: 1
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -2927,19 +3013,19 @@ opList {
name: "input"
argType: INPUT_TENSOR
}
- argDescriptor {
- name: "shape"
- argType: INT64
- }
- argDescriptor {
- name: "dimension"
- argType: INT64
- argIndex: 1
- }
opDeclarationType: LEGACY_XYZ
}
opList {
name: "broadcast_max"
+ argDescriptor {
+ name: "shape"
+ argType: INT64
+ }
+ argDescriptor {
+ name: "dimension"
+ argType: INT64
+ argIndex: 1
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -2952,19 +3038,19 @@ opList {
name: "input"
argType: INPUT_TENSOR
}
- argDescriptor {
- name: "shape"
- argType: INT64
- }
- argDescriptor {
- name: "dimension"
- argType: INT64
- argIndex: 1
- }
opDeclarationType: LEGACY_XYZ
}
opList {
name: "broadcast_min"
+ argDescriptor {
+ name: "shape"
+ argType: INT64
+ }
+ argDescriptor {
+ name: "dimension"
+ argType: INT64
+ argIndex: 1
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -2977,19 +3063,19 @@ opList {
name: "input"
argType: INPUT_TENSOR
}
- argDescriptor {
- name: "shape"
- argType: INT64
- }
- argDescriptor {
- name: "dimension"
- argType: INT64
- argIndex: 1
- }
opDeclarationType: LEGACY_XYZ
}
opList {
name: "broadcast_notequal"
+ argDescriptor {
+ name: "shape"
+ argType: INT64
+ }
+ argDescriptor {
+ name: "dimension"
+ argType: INT64
+ argIndex: 1
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -3002,19 +3088,14 @@ opList {
name: "input"
argType: INPUT_TENSOR
}
- argDescriptor {
- name: "shape"
- argType: INT64
- }
- argDescriptor {
- name: "dimension"
- argType: INT64
- argIndex: 1
- }
opDeclarationType: LEGACY_XYZ
}
opList {
name: "broadcast_to"
+ argDescriptor {
+ name: "dtype"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -3028,13 +3109,13 @@ opList {
argType: INPUT_TENSOR
argIndex: 1
}
- argDescriptor {
- name: "dtype"
- argType: DATA_TYPE
- }
}
opList {
name: "broadcastadd"
+ argDescriptor {
+ name: "dimension"
+ argType: INT64
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -3048,14 +3129,19 @@ opList {
argType: INPUT_TENSOR
argIndex: 1
}
- argDescriptor {
- name: "dimension"
- argType: INT64
- }
opDeclarationType: LEGACY_XYZ
}
opList {
name: "broadcastcopy"
+ argDescriptor {
+ name: "shape"
+ argType: INT64
+ }
+ argDescriptor {
+ name: "dimension"
+ argType: INT64
+ argIndex: 1
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -3068,19 +3154,19 @@ opList {
name: "input"
argType: INPUT_TENSOR
}
- argDescriptor {
- name: "shape"
- argType: INT64
- }
- argDescriptor {
- name: "dimension"
- argType: INT64
- argIndex: 1
- }
opDeclarationType: LEGACY_XYZ
}
opList {
name: "broadcastdiv"
+ argDescriptor {
+ name: "shape"
+ argType: INT64
+ }
+ argDescriptor {
+ name: "dimension"
+ argType: INT64
+ argIndex: 1
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -3093,26 +3179,17 @@ opList {
name: "input"
argType: INPUT_TENSOR
}
- argDescriptor {
- name: "shape"
- argType: INT64
- }
- argDescriptor {
- name: "dimension"
- argType: INT64
- argIndex: 1
- }
opDeclarationType: LEGACY_XYZ
}
opList {
name: "broadcastgradientargs"
argDescriptor {
- name: "outputs"
- argType: OUTPUT_TENSOR
+ name: "dimension"
+ argType: INT64
}
argDescriptor {
- name: "inPlace"
- argType: BOOL
+ name: "outputs"
+ argType: OUTPUT_TENSOR
}
argDescriptor {
name: "input"
@@ -3123,14 +3200,19 @@ opList {
argType: INPUT_TENSOR
argIndex: 1
}
- argDescriptor {
- name: "dimension"
- argType: INT64
- }
opDeclarationType: OP_IMPL
}
opList {
name: "broadcastmul"
+ argDescriptor {
+ name: "shape"
+ argType: INT64
+ }
+ argDescriptor {
+ name: "dimension"
+ argType: INT64
+ argIndex: 1
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -3143,19 +3225,19 @@ opList {
name: "input"
argType: INPUT_TENSOR
}
- argDescriptor {
- name: "shape"
- argType: INT64
- }
- argDescriptor {
- name: "dimension"
- argType: INT64
- argIndex: 1
- }
opDeclarationType: LEGACY_XYZ
}
opList {
name: "broadcastrdiv"
+ argDescriptor {
+ name: "shape"
+ argType: INT64
+ }
+ argDescriptor {
+ name: "dimension"
+ argType: INT64
+ argIndex: 1
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -3168,19 +3250,19 @@ opList {
name: "input"
argType: INPUT_TENSOR
}
- argDescriptor {
- name: "shape"
- argType: INT64
- }
- argDescriptor {
- name: "dimension"
- argType: INT64
- argIndex: 1
- }
opDeclarationType: LEGACY_XYZ
}
opList {
name: "broadcastrsub"
+ argDescriptor {
+ name: "shape"
+ argType: INT64
+ }
+ argDescriptor {
+ name: "dimension"
+ argType: INT64
+ argIndex: 1
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -3193,31 +3275,10 @@ opList {
name: "input"
argType: INPUT_TENSOR
}
- argDescriptor {
- name: "shape"
- argType: INT64
- }
- argDescriptor {
- name: "dimension"
- argType: INT64
- argIndex: 1
- }
opDeclarationType: LEGACY_XYZ
}
opList {
name: "broadcastsub"
- argDescriptor {
- name: "outputs"
- argType: OUTPUT_TENSOR
- }
- argDescriptor {
- name: "inPlace"
- argType: BOOL
- }
- argDescriptor {
- name: "input"
- argType: INPUT_TENSOR
- }
argDescriptor {
name: "shape"
argType: INT64
@@ -3227,83 +3288,6 @@ opList {
argType: INT64
argIndex: 1
}
- opDeclarationType: LEGACY_XYZ
-}
-opList {
- name: "car"
- argDescriptor {
- name: "outputs"
- argType: OUTPUT_TENSOR
- }
- argDescriptor {
- name: "to"
- argType: INPUT_TENSOR
- }
- argDescriptor {
- name: "from"
- argType: INPUT_TENSOR
- argIndex: 1
- }
- argDescriptor {
- name: "mode"
- argType: INT64
- }
- argDescriptor {
- name: "dataType"
- argType: DATA_TYPE
- }
- argDescriptor {
- name: "compare"
- argType: DOUBLE
- }
- argDescriptor {
- name: "set"
- argType: DOUBLE
- argIndex: 1
- }
- argDescriptor {
- name: "eps"
- argType: DOUBLE
- argIndex: 2
- }
- opDeclarationType: LEGACY_XYZ
-}
-opList {
- name: "cas"
- argDescriptor {
- name: "outputs"
- argType: OUTPUT_TENSOR
- }
- argDescriptor {
- name: "input"
- argType: INPUT_TENSOR
- }
- argDescriptor {
- name: "mode"
- argType: INT64
- }
- argDescriptor {
- name: "dataType"
- argType: DATA_TYPE
- }
- argDescriptor {
- name: "compare"
- argType: DOUBLE
- }
- argDescriptor {
- name: "set"
- argType: DOUBLE
- argIndex: 1
- }
- argDescriptor {
- name: "eps"
- argType: DOUBLE
- argIndex: 2
- }
- opDeclarationType: LEGACY_XYZ
-}
-opList {
- name: "cast"
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -3316,6 +3300,83 @@ opList {
name: "input"
argType: INPUT_TENSOR
}
+ opDeclarationType: LEGACY_XYZ
+}
+opList {
+ name: "car"
+ argDescriptor {
+ name: "mode"
+ argType: INT64
+ }
+ argDescriptor {
+ name: "dataType"
+ argType: DATA_TYPE
+ }
+ argDescriptor {
+ name: "outputs"
+ argType: OUTPUT_TENSOR
+ }
+ argDescriptor {
+ name: "compare"
+ argType: DOUBLE
+ }
+ argDescriptor {
+ name: "set"
+ argType: DOUBLE
+ argIndex: 1
+ }
+ argDescriptor {
+ name: "eps"
+ argType: DOUBLE
+ argIndex: 2
+ }
+ argDescriptor {
+ name: "input"
+ argType: INPUT_TENSOR
+ }
+ argDescriptor {
+ name: "y"
+ argType: INPUT_TENSOR
+ argIndex: 1
+ }
+ opDeclarationType: LEGACY_XYZ
+}
+opList {
+ name: "cas"
+ argDescriptor {
+ name: "mode"
+ argType: INT64
+ }
+ argDescriptor {
+ name: "dataType"
+ argType: DATA_TYPE
+ }
+ argDescriptor {
+ name: "outputs"
+ argType: OUTPUT_TENSOR
+ }
+ argDescriptor {
+ name: "compare"
+ argType: DOUBLE
+ }
+ argDescriptor {
+ name: "set"
+ argType: DOUBLE
+ argIndex: 1
+ }
+ argDescriptor {
+ name: "eps"
+ argType: DOUBLE
+ argIndex: 2
+ }
+ argDescriptor {
+ name: "input"
+ argType: INPUT_TENSOR
+ }
+ opDeclarationType: LEGACY_XYZ
+}
+opList {
+ name: "cast"
argDescriptor {
name: "dst"
argType: INT64
@@ -3324,9 +3385,30 @@ opList {
name: "dtype"
argType: DATA_TYPE
}
+ argDescriptor {
+ name: "outputs"
+ argType: OUTPUT_TENSOR
+ }
+ argDescriptor {
+ name: "inPlace"
+ argType: BOOL
+ }
+ argDescriptor {
+ name: "input"
+ argType: INPUT_TENSOR
+ }
}
opList {
name: "cbow"
+ argDescriptor {
+ name: "numWorkers"
+ argType: INT64
+ }
+ argDescriptor {
+ name: "nsRounds"
+ argType: INT64
+ argIndex: 1
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -3414,19 +3496,14 @@ opList {
argType: INPUT_TENSOR
argIndex: 14
}
- argDescriptor {
- name: "numWorkers"
- argType: INT64
- }
- argDescriptor {
- name: "nsRounds"
- argType: INT64
- argIndex: 1
- }
opDeclarationType: CONFIGURABLE_OP_IMPL
}
opList {
name: "ceil"
+ argDescriptor {
+ name: "dataType"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -3439,14 +3516,18 @@ opList {
name: "input"
argType: INPUT_TENSOR
}
- argDescriptor {
- name: "dataType"
- argType: DATA_TYPE
- }
opDeclarationType: LEGACY_XYZ
}
opList {
name: "cell_contains"
+ argDescriptor {
+ name: "dimension"
+ argType: INT64
+ }
+ argDescriptor {
+ name: "dtype"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -3469,17 +3550,13 @@ opList {
argType: INPUT_TENSOR
argIndex: 2
}
- argDescriptor {
- name: "dimension"
- argType: INT64
- }
+}
+opList {
+ name: "check_numerics"
argDescriptor {
name: "dtype"
argType: DATA_TYPE
}
-}
-opList {
- name: "check_numerics"
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -3493,10 +3570,6 @@ opList {
argType: INPUT_TENSOR
argIndex: 1
}
- argDescriptor {
- name: "dtype"
- argType: DATA_TYPE
- }
}
opList {
name: "choice"
@@ -3529,6 +3602,14 @@ opList {
}
opList {
name: "choose"
+ argDescriptor {
+ name: "mode"
+ argType: INT64
+ }
+ argDescriptor {
+ name: "dtype"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -3538,6 +3619,10 @@ opList {
argType: OUTPUT_TENSOR
argIndex: 1
}
+ argDescriptor {
+ name: "scalar"
+ argType: DOUBLE
+ }
argDescriptor {
name: "arg"
argType: INPUT_TENSOR
@@ -3547,64 +3632,60 @@ opList {
argType: INPUT_TENSOR
argIndex: 1
}
- argDescriptor {
- name: "mode"
- argType: INT64
- }
+}
+opList {
+ name: "clip_by_global_norm"
argDescriptor {
name: "dtype"
argType: DATA_TYPE
}
- argDescriptor {
- name: "scalar"
- argType: DOUBLE
- }
-}
-opList {
- name: "clip_by_global_norm"
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
}
- argDescriptor {
- name: "input"
- argType: INPUT_TENSOR
- }
- argDescriptor {
- name: "dtype"
- argType: DATA_TYPE
- }
argDescriptor {
name: "clipNorm"
argType: DOUBLE
}
+ argDescriptor {
+ name: "input"
+ argType: INPUT_TENSOR
+ }
}
opList {
name: "clipbyavgnorm"
- argDescriptor {
- name: "outputs"
- argType: OUTPUT_TENSOR
- }
- argDescriptor {
- name: "input"
- argType: INPUT_TENSOR
- }
argDescriptor {
name: "dimensions"
argType: INT64
}
+ argDescriptor {
+ name: "outputs"
+ argType: OUTPUT_TENSOR
+ }
argDescriptor {
name: "clipNorm"
argType: DOUBLE
}
+ argDescriptor {
+ name: "input"
+ argType: INPUT_TENSOR
+ }
opDeclarationType: CONFIGURABLE_OP_IMPL
}
opList {
name: "clipbyavgnorm_bp"
+ argDescriptor {
+ name: "dtype"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "gradI"
argType: OUTPUT_TENSOR
}
+ argDescriptor {
+ name: "clipNorm"
+ argType: DOUBLE
+ }
argDescriptor {
name: "input"
argType: INPUT_TENSOR
@@ -3614,41 +3695,45 @@ opList {
argType: INPUT_TENSOR
argIndex: 1
}
- argDescriptor {
- name: "dtype"
- argType: DATA_TYPE
- }
- argDescriptor {
- name: "clipNorm"
- argType: DOUBLE
- }
}
opList {
name: "clipbynorm"
- argDescriptor {
- name: "outputs"
- argType: OUTPUT_TENSOR
- }
- argDescriptor {
- name: "input"
- argType: INPUT_TENSOR
- }
argDescriptor {
name: "dimensions"
argType: INT64
}
+ argDescriptor {
+ name: "outputs"
+ argType: OUTPUT_TENSOR
+ }
argDescriptor {
name: "clipValue"
argType: DOUBLE
}
+ argDescriptor {
+ name: "input"
+ argType: INPUT_TENSOR
+ }
opDeclarationType: CONFIGURABLE_OP_IMPL
}
opList {
name: "clipbynorm_bp"
+ argDescriptor {
+ name: "dimensions"
+ argType: INT64
+ }
+ argDescriptor {
+ name: "dtype"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "gradI"
argType: OUTPUT_TENSOR
}
+ argDescriptor {
+ name: "clipValue"
+ argType: DOUBLE
+ }
argDescriptor {
name: "input"
argType: INPUT_TENSOR
@@ -3658,18 +3743,6 @@ opList {
argType: INPUT_TENSOR
argIndex: 1
}
- argDescriptor {
- name: "dimensions"
- argType: INT64
- }
- argDescriptor {
- name: "dtype"
- argType: DATA_TYPE
- }
- argDescriptor {
- name: "clipValue"
- argType: DOUBLE
- }
}
opList {
name: "clipbyvalue"
@@ -3677,10 +3750,6 @@ opList {
name: "outputs"
argType: OUTPUT_TENSOR
}
- argDescriptor {
- name: "input"
- argType: INPUT_TENSOR
- }
argDescriptor {
name: "left"
argType: DOUBLE
@@ -3690,6 +3759,10 @@ opList {
argType: DOUBLE
argIndex: 1
}
+ argDescriptor {
+ name: "input"
+ argType: INPUT_TENSOR
+ }
opDeclarationType: CONFIGURABLE_OP_IMPL
}
opList {
@@ -3706,19 +3779,6 @@ opList {
}
opList {
name: "col2im"
- argDescriptor {
- name: "outputs"
- argType: OUTPUT_TENSOR
- }
- argDescriptor {
- name: "input"
- argType: INPUT_TENSOR
- }
- argDescriptor {
- name: "inputArrays"
- argType: INPUT_TENSOR
- argIndex: 1
- }
argDescriptor {
name: "strideY"
argType: INT64
@@ -3762,13 +3822,34 @@ opList {
name: "dtype"
argType: DATA_TYPE
}
+ argDescriptor {
+ name: "outputs"
+ argType: OUTPUT_TENSOR
+ }
+ argDescriptor {
+ name: "input"
+ argType: INPUT_TENSOR
+ }
+ argDescriptor {
+ name: "inputArrays"
+ argType: INPUT_TENSOR
+ argIndex: 1
+ }
}
opList {
name: "compare_and_bitpack"
+ argDescriptor {
+ name: "dtype"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
}
+ argDescriptor {
+ name: "threshold"
+ argType: DOUBLE
+ }
argDescriptor {
name: "input"
argType: INPUT_TENSOR
@@ -3778,17 +3859,13 @@ opList {
argType: INPUT_TENSOR
argIndex: 1
}
+}
+opList {
+ name: "compat_sparse_to_dense"
argDescriptor {
name: "dtype"
argType: DATA_TYPE
}
- argDescriptor {
- name: "threshold"
- argType: DOUBLE
- }
-}
-opList {
- name: "compat_sparse_to_dense"
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -3812,13 +3889,13 @@ opList {
argType: INPUT_TENSOR
argIndex: 3
}
+}
+opList {
+ name: "compat_string_split"
argDescriptor {
name: "dtype"
argType: DATA_TYPE
}
-}
-opList {
- name: "compat_string_split"
argDescriptor {
name: "indices"
argType: OUTPUT_TENSOR
@@ -3847,13 +3924,17 @@ opList {
argType: INPUT_TENSOR
argIndex: 3
}
+}
+opList {
+ name: "concat"
+ argDescriptor {
+ name: "concatDimension"
+ argType: INT64
+ }
argDescriptor {
name: "dtype"
argType: DATA_TYPE
}
-}
-opList {
- name: "concat"
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -3871,6 +3952,9 @@ opList {
argType: INPUT_TENSOR
argIndex: 1
}
+}
+opList {
+ name: "concat_bp"
argDescriptor {
name: "concatDimension"
argType: INT64
@@ -3879,9 +3963,6 @@ opList {
name: "dtype"
argType: DATA_TYPE
}
-}
-opList {
- name: "concat_bp"
argDescriptor {
name: "epsilonChunk"
argType: OUTPUT_TENSOR
@@ -3899,17 +3980,22 @@ opList {
argType: INPUT_TENSOR
argIndex: 1
}
+}
+opList {
+ name: "confusion_matrix"
argDescriptor {
- name: "concatDimension"
+ name: "numClasses"
argType: INT64
}
+ argDescriptor {
+ name: "dataType"
+ argType: INT64
+ argIndex: 1
+ }
argDescriptor {
name: "dtype"
argType: DATA_TYPE
}
-}
-opList {
- name: "confusion_matrix"
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -3928,35 +4014,9 @@ opList {
argType: INPUT_TENSOR
argIndex: 2
}
- argDescriptor {
- name: "dataType"
- argType: INT64
- }
- argDescriptor {
- name: "dtype"
- argType: DATA_TYPE
- }
}
opList {
name: "conv1d"
- argDescriptor {
- name: "outputs"
- argType: OUTPUT_TENSOR
- }
- argDescriptor {
- name: "input"
- argType: INPUT_TENSOR
- }
- argDescriptor {
- name: "weights"
- argType: INPUT_TENSOR
- argIndex: 1
- }
- argDescriptor {
- name: "bias"
- argType: INPUT_TENSOR
- argIndex: 2
- }
argDescriptor {
name: "kW"
argType: INT64
@@ -3995,42 +4055,27 @@ opList {
name: "dtype"
argType: DATA_TYPE
}
+ argDescriptor {
+ name: "outputs"
+ argType: OUTPUT_TENSOR
+ }
+ argDescriptor {
+ name: "input"
+ argType: INPUT_TENSOR
+ }
+ argDescriptor {
+ name: "weights"
+ argType: INPUT_TENSOR
+ argIndex: 1
+ }
+ argDescriptor {
+ name: "bias"
+ argType: INPUT_TENSOR
+ argIndex: 2
+ }
}
opList {
name: "conv1d_bp"
- argDescriptor {
- name: "gradI"
- argType: OUTPUT_TENSOR
- }
- argDescriptor {
- name: "gradW"
- argType: OUTPUT_TENSOR
- argIndex: 1
- }
- argDescriptor {
- name: "gradB"
- argType: OUTPUT_TENSOR
- argIndex: 2
- }
- argDescriptor {
- name: "input"
- argType: INPUT_TENSOR
- }
- argDescriptor {
- name: "weights"
- argType: INPUT_TENSOR
- argIndex: 1
- }
- argDescriptor {
- name: "bias"
- argType: INPUT_TENSOR
- argIndex: 2
- }
- argDescriptor {
- name: "gradO"
- argType: INPUT_TENSOR
- argIndex: 3
- }
argDescriptor {
name: "kW"
argType: INT64
@@ -4069,88 +4114,6 @@ opList {
name: "dtype"
argType: DATA_TYPE
}
-}
-opList {
- name: "conv2d"
- argDescriptor {
- name: "outputs"
- argType: OUTPUT_TENSOR
- }
- argDescriptor {
- name: "input"
- argType: INPUT_TENSOR
- }
- argDescriptor {
- name: "weights"
- argType: INPUT_TENSOR
- argIndex: 1
- }
- argDescriptor {
- name: "bias"
- argType: INPUT_TENSOR
- argIndex: 2
- }
- argDescriptor {
- name: "kH"
- argType: INT64
- }
- argDescriptor {
- name: "kW"
- argType: INT64
- argIndex: 1
- }
- argDescriptor {
- name: "sH"
- argType: INT64
- argIndex: 2
- }
- argDescriptor {
- name: "sW"
- argType: INT64
- argIndex: 3
- }
- argDescriptor {
- name: "pH"
- argType: INT64
- argIndex: 4
- }
- argDescriptor {
- name: "pW"
- argType: INT64
- argIndex: 5
- }
- argDescriptor {
- name: "dH"
- argType: INT64
- argIndex: 6
- }
- argDescriptor {
- name: "dW"
- argType: INT64
- argIndex: 7
- }
- argDescriptor {
- name: "isSameMode"
- argType: INT64
- argIndex: 8
- }
- argDescriptor {
- name: "isNCHW"
- argType: INT64
- argIndex: 9
- }
- argDescriptor {
- name: "wFormat"
- argType: INT64
- argIndex: 10
- }
- argDescriptor {
- name: "dtype"
- argType: DATA_TYPE
- }
-}
-opList {
- name: "conv2d_bp"
argDescriptor {
name: "gradI"
argType: OUTPUT_TENSOR
@@ -4184,6 +4147,9 @@ opList {
argType: INPUT_TENSOR
argIndex: 3
}
+}
+opList {
+ name: "conv2d"
argDescriptor {
name: "kH"
argType: INT64
@@ -4242,9 +4208,179 @@ opList {
name: "dtype"
argType: DATA_TYPE
}
+ argDescriptor {
+ name: "outputs"
+ argType: OUTPUT_TENSOR
+ }
+ argDescriptor {
+ name: "input"
+ argType: INPUT_TENSOR
+ }
+ argDescriptor {
+ name: "weights"
+ argType: INPUT_TENSOR
+ argIndex: 1
+ }
+ argDescriptor {
+ name: "bias"
+ argType: INPUT_TENSOR
+ argIndex: 2
+ }
+}
+opList {
+ name: "conv2d_bp"
+ argDescriptor {
+ name: "kH"
+ argType: INT64
+ }
+ argDescriptor {
+ name: "kW"
+ argType: INT64
+ argIndex: 1
+ }
+ argDescriptor {
+ name: "sH"
+ argType: INT64
+ argIndex: 2
+ }
+ argDescriptor {
+ name: "sW"
+ argType: INT64
+ argIndex: 3
+ }
+ argDescriptor {
+ name: "pH"
+ argType: INT64
+ argIndex: 4
+ }
+ argDescriptor {
+ name: "pW"
+ argType: INT64
+ argIndex: 5
+ }
+ argDescriptor {
+ name: "dH"
+ argType: INT64
+ argIndex: 6
+ }
+ argDescriptor {
+ name: "dW"
+ argType: INT64
+ argIndex: 7
+ }
+ argDescriptor {
+ name: "isSameMode"
+ argType: INT64
+ argIndex: 8
+ }
+ argDescriptor {
+ name: "isNCHW"
+ argType: INT64
+ argIndex: 9
+ }
+ argDescriptor {
+ name: "wFormat"
+ argType: INT64
+ argIndex: 10
+ }
+ argDescriptor {
+ name: "dtype"
+ argType: DATA_TYPE
+ }
+ argDescriptor {
+ name: "gradI"
+ argType: OUTPUT_TENSOR
+ }
+ argDescriptor {
+ name: "gradW"
+ argType: OUTPUT_TENSOR
+ argIndex: 1
+ }
+ argDescriptor {
+ name: "gradB"
+ argType: OUTPUT_TENSOR
+ argIndex: 2
+ }
+ argDescriptor {
+ name: "input"
+ argType: INPUT_TENSOR
+ }
+ argDescriptor {
+ name: "weights"
+ argType: INPUT_TENSOR
+ argIndex: 1
+ }
+ argDescriptor {
+ name: "bias"
+ argType: INPUT_TENSOR
+ argIndex: 2
+ }
+ argDescriptor {
+ name: "gradO"
+ argType: INPUT_TENSOR
+ argIndex: 3
+ }
}
opList {
name: "conv2d_input_bp"
+ argDescriptor {
+ name: "kH"
+ argType: INT64
+ }
+ argDescriptor {
+ name: "kW"
+ argType: INT64
+ argIndex: 1
+ }
+ argDescriptor {
+ name: "sH"
+ argType: INT64
+ argIndex: 2
+ }
+ argDescriptor {
+ name: "sW"
+ argType: INT64
+ argIndex: 3
+ }
+ argDescriptor {
+ name: "pH"
+ argType: INT64
+ argIndex: 4
+ }
+ argDescriptor {
+ name: "pW"
+ argType: INT64
+ argIndex: 5
+ }
+ argDescriptor {
+ name: "dH"
+ argType: INT64
+ argIndex: 6
+ }
+ argDescriptor {
+ name: "dW"
+ argType: INT64
+ argIndex: 7
+ }
+ argDescriptor {
+ name: "isSameMode"
+ argType: INT64
+ argIndex: 8
+ }
+ argDescriptor {
+ name: "isNCHW"
+ argType: INT64
+ argIndex: 9
+ }
+ argDescriptor {
+ name: "wFormat"
+ argType: INT64
+ argIndex: 10
+ }
+ argDescriptor {
+ name: "dtype"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "gradI"
argType: OUTPUT_TENSOR
@@ -4263,85 +4399,9 @@ opList {
argType: INPUT_TENSOR
argIndex: 2
}
- argDescriptor {
- name: "kH"
- argType: INT64
- }
- argDescriptor {
- name: "kW"
- argType: INT64
- argIndex: 1
- }
- argDescriptor {
- name: "sH"
- argType: INT64
- argIndex: 2
- }
- argDescriptor {
- name: "sW"
- argType: INT64
- argIndex: 3
- }
- argDescriptor {
- name: "pH"
- argType: INT64
- argIndex: 4
- }
- argDescriptor {
- name: "pW"
- argType: INT64
- argIndex: 5
- }
- argDescriptor {
- name: "dH"
- argType: INT64
- argIndex: 6
- }
- argDescriptor {
- name: "dW"
- argType: INT64
- argIndex: 7
- }
- argDescriptor {
- name: "isSameMode"
- argType: INT64
- argIndex: 8
- }
- argDescriptor {
- name: "isNCHW"
- argType: INT64
- argIndex: 9
- }
- argDescriptor {
- name: "wFormat"
- argType: INT64
- argIndex: 10
- }
- argDescriptor {
- name: "dtype"
- argType: DATA_TYPE
- }
}
opList {
name: "conv3dnew"
- argDescriptor {
- name: "outputs"
- argType: OUTPUT_TENSOR
- }
- argDescriptor {
- name: "input"
- argType: INPUT_TENSOR
- }
- argDescriptor {
- name: "weights"
- argType: INPUT_TENSOR
- argIndex: 1
- }
- argDescriptor {
- name: "bias"
- argType: INPUT_TENSOR
- argIndex: 2
- }
argDescriptor {
name: "kD"
argType: INT64
@@ -4420,9 +4480,105 @@ opList {
name: "dtype"
argType: DATA_TYPE
}
+ argDescriptor {
+ name: "outputs"
+ argType: OUTPUT_TENSOR
+ }
+ argDescriptor {
+ name: "input"
+ argType: INPUT_TENSOR
+ }
+ argDescriptor {
+ name: "weights"
+ argType: INPUT_TENSOR
+ argIndex: 1
+ }
+ argDescriptor {
+ name: "bias"
+ argType: INPUT_TENSOR
+ argIndex: 2
+ }
}
opList {
name: "conv3dnew_bp"
+ argDescriptor {
+ name: "kD"
+ argType: INT64
+ }
+ argDescriptor {
+ name: "kH"
+ argType: INT64
+ argIndex: 1
+ }
+ argDescriptor {
+ name: "kW"
+ argType: INT64
+ argIndex: 2
+ }
+ argDescriptor {
+ name: "sD"
+ argType: INT64
+ argIndex: 3
+ }
+ argDescriptor {
+ name: "sH"
+ argType: INT64
+ argIndex: 4
+ }
+ argDescriptor {
+ name: "sW"
+ argType: INT64
+ argIndex: 5
+ }
+ argDescriptor {
+ name: "pD"
+ argType: INT64
+ argIndex: 6
+ }
+ argDescriptor {
+ name: "pH"
+ argType: INT64
+ argIndex: 7
+ }
+ argDescriptor {
+ name: "pW"
+ argType: INT64
+ argIndex: 8
+ }
+ argDescriptor {
+ name: "dD"
+ argType: INT64
+ argIndex: 9
+ }
+ argDescriptor {
+ name: "dH"
+ argType: INT64
+ argIndex: 10
+ }
+ argDescriptor {
+ name: "dW"
+ argType: INT64
+ argIndex: 11
+ }
+ argDescriptor {
+ name: "paddingMode"
+ argType: INT64
+ argIndex: 12
+ }
+ argDescriptor {
+ name: "isNCDHW"
+ argType: INT64
+ argIndex: 13
+ }
+ argDescriptor {
+ name: "wFormat"
+ argType: INT64
+ argIndex: 14
+ }
+ argDescriptor {
+ name: "dtype"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "gradI"
argType: OUTPUT_TENSOR
@@ -4456,87 +4612,13 @@ opList {
argType: INPUT_TENSOR
argIndex: 3
}
- argDescriptor {
- name: "kD"
- argType: INT64
- }
- argDescriptor {
- name: "kH"
- argType: INT64
- argIndex: 1
- }
- argDescriptor {
- name: "kW"
- argType: INT64
- argIndex: 2
- }
- argDescriptor {
- name: "sD"
- argType: INT64
- argIndex: 3
- }
- argDescriptor {
- name: "sH"
- argType: INT64
- argIndex: 4
- }
- argDescriptor {
- name: "sW"
- argType: INT64
- argIndex: 5
- }
- argDescriptor {
- name: "pD"
- argType: INT64
- argIndex: 6
- }
- argDescriptor {
- name: "pH"
- argType: INT64
- argIndex: 7
- }
- argDescriptor {
- name: "pW"
- argType: INT64
- argIndex: 8
- }
- argDescriptor {
- name: "dD"
- argType: INT64
- argIndex: 9
- }
- argDescriptor {
- name: "dH"
- argType: INT64
- argIndex: 10
- }
- argDescriptor {
- name: "dW"
- argType: INT64
- argIndex: 11
- }
- argDescriptor {
- name: "paddingMode"
- argType: INT64
- argIndex: 12
- }
- argDescriptor {
- name: "isNCDHW"
- argType: INT64
- argIndex: 13
- }
- argDescriptor {
- name: "wFormat"
- argType: INT64
- argIndex: 14
- }
- argDescriptor {
- name: "dtype"
- argType: DATA_TYPE
- }
}
opList {
name: "copy"
+ argDescriptor {
+ name: "dataType"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -4554,14 +4636,14 @@ opList {
argType: INPUT_TENSOR
argIndex: 1
}
- argDescriptor {
- name: "dataType"
- argType: DATA_TYPE
- }
opDeclarationType: LEGACY_XYZ
}
opList {
name: "cos"
+ argDescriptor {
+ name: "dataType"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -4574,14 +4656,14 @@ opList {
name: "input"
argType: INPUT_TENSOR
}
- argDescriptor {
- name: "dataType"
- argType: DATA_TYPE
- }
opDeclarationType: LEGACY_XYZ
}
opList {
name: "cosh"
+ argDescriptor {
+ name: "dataType"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -4594,14 +4676,23 @@ opList {
name: "input"
argType: INPUT_TENSOR
}
- argDescriptor {
- name: "dataType"
- argType: DATA_TYPE
- }
opDeclarationType: LEGACY_XYZ
}
opList {
name: "cosine_distance_loss"
+ argDescriptor {
+ name: "reductionMode"
+ argType: INT64
+ }
+ argDescriptor {
+ name: "dimensions"
+ argType: INT64
+ argIndex: 1
+ }
+ argDescriptor {
+ name: "dtype"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -4620,6 +4711,9 @@ opList {
argType: INPUT_TENSOR
argIndex: 2
}
+}
+opList {
+ name: "cosine_distance_loss_grad"
argDescriptor {
name: "reductionMode"
argType: INT64
@@ -4633,9 +4727,6 @@ opList {
name: "dtype"
argType: DATA_TYPE
}
-}
-opList {
- name: "cosine_distance_loss_grad"
argDescriptor {
name: "dLdp"
argType: OUTPUT_TENSOR
@@ -4664,22 +4755,13 @@ opList {
argType: INPUT_TENSOR
argIndex: 2
}
- argDescriptor {
- name: "reductionMode"
- argType: INT64
- }
- argDescriptor {
- name: "dimensions"
- argType: INT64
- argIndex: 1
- }
- argDescriptor {
- name: "dtype"
- argType: DATA_TYPE
- }
}
opList {
name: "cosinedistance"
+ argDescriptor {
+ name: "dimensions"
+ argType: INT64
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -4702,14 +4784,14 @@ opList {
argType: INPUT_TENSOR
argIndex: 1
}
- argDescriptor {
- name: "dimensions"
- argType: INT64
- }
opDeclarationType: LEGACY_XYZ
}
opList {
name: "cosinesimilarity"
+ argDescriptor {
+ name: "dimensions"
+ argType: INT64
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -4732,14 +4814,14 @@ opList {
argType: INPUT_TENSOR
argIndex: 1
}
- argDescriptor {
- name: "dimensions"
- argType: INT64
- }
opDeclarationType: LEGACY_XYZ
}
opList {
name: "countNonZero"
+ argDescriptor {
+ name: "dimensions"
+ argType: INT64
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -4757,14 +4839,14 @@ opList {
argType: INPUT_TENSOR
argIndex: 1
}
- argDescriptor {
- name: "dimensions"
- argType: INT64
- }
opDeclarationType: LEGACY_XYZ
}
opList {
name: "countZero"
+ argDescriptor {
+ name: "dimensions"
+ argType: INT64
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -4782,26 +4864,10 @@ opList {
argType: INPUT_TENSOR
argIndex: 1
}
- argDescriptor {
- name: "dimensions"
- argType: INT64
- }
opDeclarationType: LEGACY_XYZ
}
opList {
name: "create"
- argDescriptor {
- name: "outputs"
- argType: OUTPUT_TENSOR
- }
- argDescriptor {
- name: "init"
- argType: BOOL
- }
- argDescriptor {
- name: "input"
- argType: INPUT_TENSOR
- }
argDescriptor {
name: "order"
argType: INT64
@@ -4815,17 +4881,21 @@ opList {
name: "outputType"
argType: DATA_TYPE
}
-}
-opList {
- name: "create_list"
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
}
+ argDescriptor {
+ name: "init"
+ argType: BOOL
+ }
argDescriptor {
name: "input"
argType: INPUT_TENSOR
}
+}
+opList {
+ name: "create_list"
argDescriptor {
name: "height"
argType: INT64
@@ -4835,10 +4905,6 @@ opList {
argType: INT64
argIndex: 1
}
- opDeclarationType: LIST_OP_IMPL
-}
-opList {
- name: "crelu"
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -4847,13 +4913,29 @@ opList {
name: "input"
argType: INPUT_TENSOR
}
+ opDeclarationType: LIST_OP_IMPL
+}
+opList {
+ name: "crelu"
argDescriptor {
name: "dtype"
argType: DATA_TYPE
}
+ argDescriptor {
+ name: "outputs"
+ argType: OUTPUT_TENSOR
+ }
+ argDescriptor {
+ name: "input"
+ argType: INPUT_TENSOR
+ }
}
opList {
name: "crelu_bp"
+ argDescriptor {
+ name: "dtype"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "epsilon"
argType: OUTPUT_TENSOR
@@ -4867,17 +4949,25 @@ opList {
argType: INPUT_TENSOR
argIndex: 1
}
- argDescriptor {
- name: "dtype"
- argType: DATA_TYPE
- }
}
opList {
name: "crop_and_resize"
+ argDescriptor {
+ name: "method"
+ argType: INT64
+ }
+ argDescriptor {
+ name: "dtype"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
}
+ argDescriptor {
+ name: "extrapolationVal"
+ argType: DOUBLE
+ }
argDescriptor {
name: "image"
argType: INPUT_TENSOR
@@ -4897,18 +4987,6 @@ opList {
argType: INPUT_TENSOR
argIndex: 3
}
- argDescriptor {
- name: "method"
- argType: INT64
- }
- argDescriptor {
- name: "dtype"
- argType: DATA_TYPE
- }
- argDescriptor {
- name: "extrapolationVal"
- argType: DOUBLE
- }
}
opList {
name: "cross"
@@ -4927,8 +5005,80 @@ opList {
}
opDeclarationType: OP_IMPL
}
+opList {
+ name: "ctc_loss"
+ argDescriptor {
+ name: "blankIndex"
+ argType: INT64
+ }
+ argDescriptor {
+ name: "dtype"
+ argType: DATA_TYPE
+ }
+ argDescriptor {
+ name: "outputLosses"
+ argType: OUTPUT_TENSOR
+ }
+ argDescriptor {
+ name: "targetLabels"
+ argType: INPUT_TENSOR
+ }
+ argDescriptor {
+ name: "logitInput"
+ argType: INPUT_TENSOR
+ argIndex: 1
+ }
+ argDescriptor {
+ name: "targetLabelLengths"
+ argType: INPUT_TENSOR
+ argIndex: 2
+ }
+ argDescriptor {
+ name: "logitInputLengths"
+ argType: INPUT_TENSOR
+ argIndex: 3
+ }
+}
+opList {
+ name: "ctc_loss_grad"
+ argDescriptor {
+ name: "blankIndex"
+ argType: INT64
+ }
+ argDescriptor {
+ name: "dtype"
+ argType: DATA_TYPE
+ }
+ argDescriptor {
+ name: "outputGradients"
+ argType: OUTPUT_TENSOR
+ }
+ argDescriptor {
+ name: "targetLabels"
+ argType: INPUT_TENSOR
+ }
+ argDescriptor {
+ name: "logitInput"
+ argType: INPUT_TENSOR
+ argIndex: 1
+ }
+ argDescriptor {
+ name: "targetLabelLengths"
+ argType: INPUT_TENSOR
+ argIndex: 2
+ }
+ argDescriptor {
+ name: "logitInputLengths"
+ argType: INPUT_TENSOR
+ argIndex: 3
+ }
+}
opList {
name: "cube"
+ argDescriptor {
+ name: "dataType"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -4941,10 +5091,6 @@ opList {
name: "input"
argType: INPUT_TENSOR
}
- argDescriptor {
- name: "dataType"
- argType: DATA_TYPE
- }
opDeclarationType: CONFIGURABLE_OP_IMPL
}
opList {
@@ -4966,6 +5112,10 @@ opList {
}
opList {
name: "cubederivative"
+ argDescriptor {
+ name: "dataType"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -4978,14 +5128,24 @@ opList {
name: "input"
argType: INPUT_TENSOR
}
- argDescriptor {
- name: "dataType"
- argType: DATA_TYPE
- }
opDeclarationType: LEGACY_XYZ
}
opList {
name: "cumprod"
+ argDescriptor {
+ name: "exclusive"
+ argType: INT64
+ }
+ argDescriptor {
+ name: "reverse"
+ argType: INT64
+ argIndex: 1
+ }
+ argDescriptor {
+ name: "dimensions"
+ argType: INT64
+ argIndex: 2
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -5008,24 +5168,28 @@ opList {
argType: INPUT_TENSOR
argIndex: 1
}
- argDescriptor {
- name: "exclusive"
- argType: INT64
- }
- argDescriptor {
- name: "reverse"
- argType: INT64
- argIndex: 1
- }
- argDescriptor {
- name: "dimensions"
- argType: INT64
- argIndex: 2
- }
opDeclarationType: CONFIGURABLE_OP_IMPL
}
opList {
name: "cumprod_bp"
+ argDescriptor {
+ name: "exclusive"
+ argType: INT64
+ }
+ argDescriptor {
+ name: "reverse"
+ argType: INT64
+ argIndex: 1
+ }
+ argDescriptor {
+ name: "dimensions"
+ argType: INT64
+ argIndex: 2
+ }
+ argDescriptor {
+ name: "dtype"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -5053,6 +5217,9 @@ opList {
argType: INPUT_TENSOR
argIndex: 2
}
+}
+opList {
+ name: "cumsum"
argDescriptor {
name: "exclusive"
argType: INT64
@@ -5067,13 +5234,6 @@ opList {
argType: INT64
argIndex: 2
}
- argDescriptor {
- name: "dtype"
- argType: DATA_TYPE
- }
-}
-opList {
- name: "cumsum"
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -5096,24 +5256,28 @@ opList {
argType: INPUT_TENSOR
argIndex: 1
}
- argDescriptor {
- name: "exclusive"
- argType: INT64
- }
- argDescriptor {
- name: "reverse"
- argType: INT64
- argIndex: 1
- }
- argDescriptor {
- name: "dimensions"
- argType: INT64
- argIndex: 2
- }
opDeclarationType: CONFIGURABLE_OP_IMPL
}
opList {
name: "cumsum_bp"
+ argDescriptor {
+ name: "exclusive"
+ argType: INT64
+ }
+ argDescriptor {
+ name: "reverse"
+ argType: INT64
+ argIndex: 1
+ }
+ argDescriptor {
+ name: "dimensions"
+ argType: INT64
+ argIndex: 2
+ }
+ argDescriptor {
+ name: "dtype"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -5141,27 +5305,13 @@ opList {
argType: INPUT_TENSOR
argIndex: 2
}
- argDescriptor {
- name: "exclusive"
- argType: INT64
- }
- argDescriptor {
- name: "reverse"
- argType: INT64
- argIndex: 1
- }
- argDescriptor {
- name: "dimensions"
- argType: INT64
- argIndex: 2
- }
- argDescriptor {
- name: "dtype"
- argType: DATA_TYPE
- }
}
opList {
name: "cyclic_rshift_bits"
+ argDescriptor {
+ name: "dataType"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -5179,14 +5329,14 @@ opList {
argType: INPUT_TENSOR
argIndex: 1
}
- argDescriptor {
- name: "dataType"
- argType: DATA_TYPE
- }
opDeclarationType: BROADCASTABLE_OP_IMPL
}
opList {
name: "cyclic_shift_bits"
+ argDescriptor {
+ name: "dataType"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -5204,14 +5354,14 @@ opList {
argType: INPUT_TENSOR
argIndex: 1
}
- argDescriptor {
- name: "dataType"
- argType: DATA_TYPE
- }
opDeclarationType: BROADCASTABLE_OP_IMPL
}
opList {
name: "decode_bitmap"
+ argDescriptor {
+ name: "dtype"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "updates"
argType: OUTPUT_TENSOR
@@ -5225,13 +5375,13 @@ opList {
argType: INPUT_TENSOR
argIndex: 1
}
+}
+opList {
+ name: "decode_threshold"
argDescriptor {
name: "dtype"
argType: DATA_TYPE
}
-}
-opList {
- name: "decode_threshold"
argDescriptor {
name: "updates"
argType: OUTPUT_TENSOR
@@ -5245,31 +5395,9 @@ opList {
argType: INPUT_TENSOR
argIndex: 1
}
- argDescriptor {
- name: "dtype"
- argType: DATA_TYPE
- }
}
opList {
name: "deconv2d"
- argDescriptor {
- name: "outputs"
- argType: OUTPUT_TENSOR
- }
- argDescriptor {
- name: "input"
- argType: INPUT_TENSOR
- }
- argDescriptor {
- name: "weights"
- argType: INPUT_TENSOR
- argIndex: 1
- }
- argDescriptor {
- name: "bias"
- argType: INPUT_TENSOR
- argIndex: 2
- }
argDescriptor {
name: "kH"
argType: INT64
@@ -5328,9 +5456,85 @@ opList {
name: "dtype"
argType: DATA_TYPE
}
+ argDescriptor {
+ name: "outputs"
+ argType: OUTPUT_TENSOR
+ }
+ argDescriptor {
+ name: "input"
+ argType: INPUT_TENSOR
+ }
+ argDescriptor {
+ name: "weights"
+ argType: INPUT_TENSOR
+ argIndex: 1
+ }
+ argDescriptor {
+ name: "bias"
+ argType: INPUT_TENSOR
+ argIndex: 2
+ }
}
opList {
name: "deconv2d_bp"
+ argDescriptor {
+ name: "kH"
+ argType: INT64
+ }
+ argDescriptor {
+ name: "kW"
+ argType: INT64
+ argIndex: 1
+ }
+ argDescriptor {
+ name: "sH"
+ argType: INT64
+ argIndex: 2
+ }
+ argDescriptor {
+ name: "sW"
+ argType: INT64
+ argIndex: 3
+ }
+ argDescriptor {
+ name: "pH"
+ argType: INT64
+ argIndex: 4
+ }
+ argDescriptor {
+ name: "pW"
+ argType: INT64
+ argIndex: 5
+ }
+ argDescriptor {
+ name: "dH"
+ argType: INT64
+ argIndex: 6
+ }
+ argDescriptor {
+ name: "dW"
+ argType: INT64
+ argIndex: 7
+ }
+ argDescriptor {
+ name: "isSameMode"
+ argType: INT64
+ argIndex: 8
+ }
+ argDescriptor {
+ name: "isNCHW"
+ argType: INT64
+ argIndex: 9
+ }
+ argDescriptor {
+ name: "wFormat"
+ argType: INT64
+ argIndex: 10
+ }
+ argDescriptor {
+ name: "dtype"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "gradI"
argType: OUTPUT_TENSOR
@@ -5364,6 +5568,9 @@ opList {
argType: INPUT_TENSOR
argIndex: 3
}
+}
+opList {
+ name: "deconv2d_tf"
argDescriptor {
name: "kH"
argType: INT64
@@ -5422,9 +5629,6 @@ opList {
name: "dtype"
argType: DATA_TYPE
}
-}
-opList {
- name: "deconv2d_tf"
argDescriptor {
name: "gradI"
argType: OUTPUT_TENSOR
@@ -5443,85 +5647,9 @@ opList {
argType: INPUT_TENSOR
argIndex: 2
}
- argDescriptor {
- name: "kH"
- argType: INT64
- }
- argDescriptor {
- name: "kW"
- argType: INT64
- argIndex: 1
- }
- argDescriptor {
- name: "sH"
- argType: INT64
- argIndex: 2
- }
- argDescriptor {
- name: "sW"
- argType: INT64
- argIndex: 3
- }
- argDescriptor {
- name: "pH"
- argType: INT64
- argIndex: 4
- }
- argDescriptor {
- name: "pW"
- argType: INT64
- argIndex: 5
- }
- argDescriptor {
- name: "dH"
- argType: INT64
- argIndex: 6
- }
- argDescriptor {
- name: "dW"
- argType: INT64
- argIndex: 7
- }
- argDescriptor {
- name: "isSameMode"
- argType: INT64
- argIndex: 8
- }
- argDescriptor {
- name: "isNCHW"
- argType: INT64
- argIndex: 9
- }
- argDescriptor {
- name: "wFormat"
- argType: INT64
- argIndex: 10
- }
- argDescriptor {
- name: "dtype"
- argType: DATA_TYPE
- }
}
opList {
name: "deconv3d"
- argDescriptor {
- name: "outputs"
- argType: OUTPUT_TENSOR
- }
- argDescriptor {
- name: "input"
- argType: INPUT_TENSOR
- }
- argDescriptor {
- name: "weights"
- argType: INPUT_TENSOR
- argIndex: 1
- }
- argDescriptor {
- name: "bias"
- argType: INPUT_TENSOR
- argIndex: 2
- }
argDescriptor {
name: "kD"
argType: INT64
@@ -5600,9 +5728,105 @@ opList {
name: "dtype"
argType: DATA_TYPE
}
+ argDescriptor {
+ name: "outputs"
+ argType: OUTPUT_TENSOR
+ }
+ argDescriptor {
+ name: "input"
+ argType: INPUT_TENSOR
+ }
+ argDescriptor {
+ name: "weights"
+ argType: INPUT_TENSOR
+ argIndex: 1
+ }
+ argDescriptor {
+ name: "bias"
+ argType: INPUT_TENSOR
+ argIndex: 2
+ }
}
opList {
name: "deconv3d_bp"
+ argDescriptor {
+ name: "kD"
+ argType: INT64
+ }
+ argDescriptor {
+ name: "kH"
+ argType: INT64
+ argIndex: 1
+ }
+ argDescriptor {
+ name: "kW"
+ argType: INT64
+ argIndex: 2
+ }
+ argDescriptor {
+ name: "sD"
+ argType: INT64
+ argIndex: 3
+ }
+ argDescriptor {
+ name: "sH"
+ argType: INT64
+ argIndex: 4
+ }
+ argDescriptor {
+ name: "sW"
+ argType: INT64
+ argIndex: 5
+ }
+ argDescriptor {
+ name: "pD"
+ argType: INT64
+ argIndex: 6
+ }
+ argDescriptor {
+ name: "pH"
+ argType: INT64
+ argIndex: 7
+ }
+ argDescriptor {
+ name: "pW"
+ argType: INT64
+ argIndex: 8
+ }
+ argDescriptor {
+ name: "dD"
+ argType: INT64
+ argIndex: 9
+ }
+ argDescriptor {
+ name: "dH"
+ argType: INT64
+ argIndex: 10
+ }
+ argDescriptor {
+ name: "dW"
+ argType: INT64
+ argIndex: 11
+ }
+ argDescriptor {
+ name: "isSameMode"
+ argType: INT64
+ argIndex: 12
+ }
+ argDescriptor {
+ name: "isNCDHW"
+ argType: INT64
+ argIndex: 13
+ }
+ argDescriptor {
+ name: "wFormat"
+ argType: INT64
+ argIndex: 14
+ }
+ argDescriptor {
+ name: "dtype"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "gradI"
argType: OUTPUT_TENSOR
@@ -5636,84 +5860,6 @@ opList {
argType: INPUT_TENSOR
argIndex: 3
}
- argDescriptor {
- name: "kD"
- argType: INT64
- }
- argDescriptor {
- name: "kH"
- argType: INT64
- argIndex: 1
- }
- argDescriptor {
- name: "kW"
- argType: INT64
- argIndex: 2
- }
- argDescriptor {
- name: "sD"
- argType: INT64
- argIndex: 3
- }
- argDescriptor {
- name: "sH"
- argType: INT64
- argIndex: 4
- }
- argDescriptor {
- name: "sW"
- argType: INT64
- argIndex: 5
- }
- argDescriptor {
- name: "pD"
- argType: INT64
- argIndex: 6
- }
- argDescriptor {
- name: "pH"
- argType: INT64
- argIndex: 7
- }
- argDescriptor {
- name: "pW"
- argType: INT64
- argIndex: 8
- }
- argDescriptor {
- name: "dD"
- argType: INT64
- argIndex: 9
- }
- argDescriptor {
- name: "dH"
- argType: INT64
- argIndex: 10
- }
- argDescriptor {
- name: "dW"
- argType: INT64
- argIndex: 11
- }
- argDescriptor {
- name: "isSameMode"
- argType: INT64
- argIndex: 12
- }
- argDescriptor {
- name: "isNCDHW"
- argType: INT64
- argIndex: 13
- }
- argDescriptor {
- name: "wFormat"
- argType: INT64
- argIndex: 14
- }
- argDescriptor {
- name: "dtype"
- argType: DATA_TYPE
- }
}
opList {
name: "deconv3d_tf"
@@ -5738,14 +5884,6 @@ opList {
}
opList {
name: "depth_to_space"
- argDescriptor {
- name: "outputs"
- argType: OUTPUT_TENSOR
- }
- argDescriptor {
- name: "input"
- argType: INPUT_TENSOR
- }
argDescriptor {
name: "block_size"
argType: INT64
@@ -5759,9 +5897,6 @@ opList {
name: "dtype"
argType: DATA_TYPE
}
-}
-opList {
- name: "depthwise_conv2d"
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -5770,16 +5905,9 @@ opList {
name: "input"
argType: INPUT_TENSOR
}
- argDescriptor {
- name: "weights"
- argType: INPUT_TENSOR
- argIndex: 1
- }
- argDescriptor {
- name: "bias"
- argType: INPUT_TENSOR
- argIndex: 2
- }
+}
+opList {
+ name: "depthwise_conv2d"
argDescriptor {
name: "kH"
argType: INT64
@@ -5838,9 +5966,85 @@ opList {
name: "dtype"
argType: DATA_TYPE
}
+ argDescriptor {
+ name: "outputs"
+ argType: OUTPUT_TENSOR
+ }
+ argDescriptor {
+ name: "input"
+ argType: INPUT_TENSOR
+ }
+ argDescriptor {
+ name: "weights"
+ argType: INPUT_TENSOR
+ argIndex: 1
+ }
+ argDescriptor {
+ name: "bias"
+ argType: INPUT_TENSOR
+ argIndex: 2
+ }
}
opList {
name: "depthwise_conv2d_bp"
+ argDescriptor {
+ name: "kH"
+ argType: INT64
+ }
+ argDescriptor {
+ name: "kW"
+ argType: INT64
+ argIndex: 1
+ }
+ argDescriptor {
+ name: "sH"
+ argType: INT64
+ argIndex: 2
+ }
+ argDescriptor {
+ name: "sW"
+ argType: INT64
+ argIndex: 3
+ }
+ argDescriptor {
+ name: "pH"
+ argType: INT64
+ argIndex: 4
+ }
+ argDescriptor {
+ name: "pW"
+ argType: INT64
+ argIndex: 5
+ }
+ argDescriptor {
+ name: "dH"
+ argType: INT64
+ argIndex: 6
+ }
+ argDescriptor {
+ name: "dW"
+ argType: INT64
+ argIndex: 7
+ }
+ argDescriptor {
+ name: "isSameMode"
+ argType: INT64
+ argIndex: 8
+ }
+ argDescriptor {
+ name: "isNCHW"
+ argType: INT64
+ argIndex: 9
+ }
+ argDescriptor {
+ name: "wFormat"
+ argType: INT64
+ argIndex: 10
+ }
+ argDescriptor {
+ name: "dtype"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "gradI"
argType: OUTPUT_TENSOR
@@ -5874,67 +6078,13 @@ opList {
argType: INPUT_TENSOR
argIndex: 3
}
- argDescriptor {
- name: "kH"
- argType: INT64
- }
- argDescriptor {
- name: "kW"
- argType: INT64
- argIndex: 1
- }
- argDescriptor {
- name: "sH"
- argType: INT64
- argIndex: 2
- }
- argDescriptor {
- name: "sW"
- argType: INT64
- argIndex: 3
- }
- argDescriptor {
- name: "pH"
- argType: INT64
- argIndex: 4
- }
- argDescriptor {
- name: "pW"
- argType: INT64
- argIndex: 5
- }
- argDescriptor {
- name: "dH"
- argType: INT64
- argIndex: 6
- }
- argDescriptor {
- name: "dW"
- argType: INT64
- argIndex: 7
- }
- argDescriptor {
- name: "isSameMode"
- argType: INT64
- argIndex: 8
- }
- argDescriptor {
- name: "isNCHW"
- argType: INT64
- argIndex: 9
- }
- argDescriptor {
- name: "wFormat"
- argType: INT64
- argIndex: 10
- }
+}
+opList {
+ name: "diag"
argDescriptor {
name: "dtype"
argType: DATA_TYPE
}
-}
-opList {
- name: "diag"
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -5947,13 +6097,13 @@ opList {
name: "input"
argType: INPUT_TENSOR
}
- argDescriptor {
- name: "dtype"
- argType: DATA_TYPE
- }
}
opList {
name: "diag_part"
+ argDescriptor {
+ name: "dtype"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -5966,10 +6116,6 @@ opList {
name: "input"
argType: INPUT_TENSOR
}
- argDescriptor {
- name: "dtype"
- argType: DATA_TYPE
- }
}
opList {
name: "digamma"
@@ -5985,6 +6131,24 @@ opList {
}
opList {
name: "dilation2d"
+ argDescriptor {
+ name: "isSameMode"
+ argType: INT64
+ }
+ argDescriptor {
+ name: "rates"
+ argType: INT64
+ argIndex: 1
+ }
+ argDescriptor {
+ name: "strides"
+ argType: INT64
+ argIndex: 2
+ }
+ argDescriptor {
+ name: "dtype"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -6017,31 +6181,9 @@ opList {
argType: INPUT_TENSOR
argIndex: 3
}
- argDescriptor {
- name: "isSameMode"
- argType: INT64
- }
- argDescriptor {
- name: "rates"
- argType: INT64
- argIndex: 1
- }
- argDescriptor {
- name: "strides"
- argType: INT64
- argIndex: 2
- }
- argDescriptor {
- name: "dtype"
- argType: DATA_TYPE
- }
}
opList {
name: "distribution_bernoulli"
- argDescriptor {
- name: "outputs"
- argType: OUTPUT_TENSOR
- }
argDescriptor {
name: "dataType"
argType: INT64
@@ -6051,6 +6193,10 @@ opList {
argType: INT64
argIndex: 1
}
+ argDescriptor {
+ name: "outputs"
+ argType: OUTPUT_TENSOR
+ }
argDescriptor {
name: "prob"
argType: DOUBLE
@@ -6059,10 +6205,6 @@ opList {
}
opList {
name: "distribution_binomial"
- argDescriptor {
- name: "outputs"
- argType: OUTPUT_TENSOR
- }
argDescriptor {
name: "trials"
argType: INT64
@@ -6077,6 +6219,10 @@ opList {
argType: INT64
argIndex: 2
}
+ argDescriptor {
+ name: "outputs"
+ argType: OUTPUT_TENSOR
+ }
argDescriptor {
name: "probability"
argType: DOUBLE
@@ -6085,18 +6231,14 @@ opList {
}
opList {
name: "distribution_binomial_ex"
- argDescriptor {
- name: "outputs"
- argType: OUTPUT_TENSOR
- }
- argDescriptor {
- name: "probabilities"
- argType: INPUT_TENSOR
- }
argDescriptor {
name: "trials"
argType: INT64
}
+ argDescriptor {
+ name: "outputs"
+ argType: OUTPUT_TENSOR
+ }
argDescriptor {
name: "probability"
argType: DOUBLE
@@ -6105,10 +6247,6 @@ opList {
}
opList {
name: "distribution_gaussian"
- argDescriptor {
- name: "outputs"
- argType: OUTPUT_TENSOR
- }
argDescriptor {
name: "dataType"
argType: INT64
@@ -6118,6 +6256,10 @@ opList {
argType: INT64
argIndex: 1
}
+ argDescriptor {
+ name: "outputs"
+ argType: OUTPUT_TENSOR
+ }
argDescriptor {
name: "mean"
argType: DOUBLE
@@ -6131,10 +6273,6 @@ opList {
}
opList {
name: "distribution_lognormal"
- argDescriptor {
- name: "outputs"
- argType: OUTPUT_TENSOR
- }
argDescriptor {
name: "dataType"
argType: INT64
@@ -6144,6 +6282,10 @@ opList {
argType: INT64
argIndex: 1
}
+ argDescriptor {
+ name: "outputs"
+ argType: OUTPUT_TENSOR
+ }
argDescriptor {
name: "mean"
argType: DOUBLE
@@ -6157,10 +6299,6 @@ opList {
}
opList {
name: "distribution_truncated"
- argDescriptor {
- name: "outputs"
- argType: OUTPUT_TENSOR
- }
argDescriptor {
name: "dataType"
argType: INT64
@@ -6170,6 +6308,10 @@ opList {
argType: INT64
argIndex: 1
}
+ argDescriptor {
+ name: "outputs"
+ argType: OUTPUT_TENSOR
+ }
argDescriptor {
name: "mean"
argType: DOUBLE
@@ -6183,10 +6325,6 @@ opList {
}
opList {
name: "distribution_uniform"
- argDescriptor {
- name: "outputs"
- argType: OUTPUT_TENSOR
- }
argDescriptor {
name: "dataType"
argType: INT64
@@ -6196,6 +6334,10 @@ opList {
argType: INT64
argIndex: 1
}
+ argDescriptor {
+ name: "outputs"
+ argType: OUTPUT_TENSOR
+ }
argDescriptor {
name: "from"
argType: DOUBLE
@@ -6226,6 +6368,10 @@ opList {
}
opList {
name: "divide"
+ argDescriptor {
+ name: "dataType"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -6243,14 +6389,14 @@ opList {
argType: INPUT_TENSOR
argIndex: 1
}
- argDescriptor {
- name: "dataType"
- argType: DATA_TYPE
- }
opDeclarationType: BROADCASTABLE_OP_IMPL
}
opList {
name: "divide_bp"
+ argDescriptor {
+ name: "dtype"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "gradX"
argType: OUTPUT_TENSOR
@@ -6278,10 +6424,6 @@ opList {
argType: INPUT_TENSOR
argIndex: 2
}
- argDescriptor {
- name: "dtype"
- argType: DATA_TYPE
- }
}
opList {
name: "divide_no_nan"
@@ -6302,6 +6444,10 @@ opList {
}
opList {
name: "dot"
+ argDescriptor {
+ name: "dimensions"
+ argType: INT64
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -6324,14 +6470,23 @@ opList {
argType: INPUT_TENSOR
argIndex: 1
}
- argDescriptor {
- name: "dimensions"
- argType: INT64
- }
opDeclarationType: LEGACY_XYZ
}
opList {
name: "dot_product_attention"
+ argDescriptor {
+ name: "normalization"
+ argType: INT64
+ }
+ argDescriptor {
+ name: "outputWeights"
+ argType: INT64
+ argIndex: 1
+ }
+ argDescriptor {
+ name: "dtype"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -6369,22 +6524,17 @@ opList {
argType: INPUT_TENSOR
argIndex: 3
}
+}
+opList {
+ name: "dot_product_attention_bp"
argDescriptor {
name: "normalization"
argType: INT64
}
- argDescriptor {
- name: "outputWeights"
- argType: INT64
- argIndex: 1
- }
argDescriptor {
name: "dtype"
argType: DATA_TYPE
}
-}
-opList {
- name: "dot_product_attention_bp"
argDescriptor {
name: "dLdq"
argType: OUTPUT_TENSOR
@@ -6427,14 +6577,6 @@ opList {
argType: INPUT_TENSOR
argIndex: 4
}
- argDescriptor {
- name: "normalization"
- argType: INT64
- }
- argDescriptor {
- name: "dtype"
- argType: DATA_TYPE
- }
}
opList {
name: "draw_bounding_boxes"
@@ -6460,10 +6602,18 @@ opList {
}
opList {
name: "dropout"
+ argDescriptor {
+ name: "seed"
+ argType: INT64
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
}
+ argDescriptor {
+ name: "probValue"
+ argType: DOUBLE
+ }
argDescriptor {
name: "input"
argType: INPUT_TENSOR
@@ -6473,22 +6623,22 @@ opList {
argType: INPUT_TENSOR
argIndex: 1
}
- argDescriptor {
- name: "seed"
- argType: INT64
- }
- argDescriptor {
- name: "probValue"
- argType: DOUBLE
- }
opDeclarationType: CONFIGURABLE_OP_IMPL
}
opList {
name: "dropout_bp"
+ argDescriptor {
+ name: "seed"
+ argType: INT64
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
}
+ argDescriptor {
+ name: "probValue"
+ argType: DOUBLE
+ }
argDescriptor {
name: "input"
argType: INPUT_TENSOR
@@ -6503,14 +6653,6 @@ opList {
argType: INPUT_TENSOR
argIndex: 2
}
- argDescriptor {
- name: "seed"
- argType: INT64
- }
- argDescriptor {
- name: "probValue"
- argType: DOUBLE
- }
opDeclarationType: CONFIGURABLE_OP_IMPL
}
opList {
@@ -6519,18 +6661,26 @@ opList {
name: "outputs"
argType: OUTPUT_TENSOR
}
- argDescriptor {
- name: "input"
- argType: INPUT_TENSOR
- }
argDescriptor {
name: "p"
argType: DOUBLE
}
+ argDescriptor {
+ name: "input"
+ argType: INPUT_TENSOR
+ }
opDeclarationType: LEGACY_XYZ
}
opList {
name: "dynamic_bidirectional_rnn"
+ argDescriptor {
+ name: "timeMajor"
+ argType: INT64
+ }
+ argDescriptor {
+ name: "dtype"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "hFW"
argType: OUTPUT_TENSOR
@@ -6594,17 +6744,17 @@ opList {
argType: INPUT_TENSOR
argIndex: 8
}
+}
+opList {
+ name: "dynamic_partition"
argDescriptor {
- name: "timeMajor"
+ name: "numPartitions"
argType: INT64
}
argDescriptor {
name: "dtype"
argType: DATA_TYPE
}
-}
-opList {
- name: "dynamic_partition"
argDescriptor {
name: "outputList"
argType: OUTPUT_TENSOR
@@ -6618,17 +6768,17 @@ opList {
argType: INPUT_TENSOR
argIndex: 1
}
+}
+opList {
+ name: "dynamic_partition_bp"
argDescriptor {
- name: "numPartitions"
+ name: "numPartition"
argType: INT64
}
argDescriptor {
name: "dtype"
argType: DATA_TYPE
}
-}
-opList {
- name: "dynamic_partition_bp"
argDescriptor {
name: "outputList"
argType: OUTPUT_TENSOR
@@ -6647,17 +6797,17 @@ opList {
argType: INPUT_TENSOR
argIndex: 2
}
+}
+opList {
+ name: "dynamic_rnn"
argDescriptor {
- name: "numPartition"
+ name: "timeMajor"
argType: INT64
}
argDescriptor {
name: "dtype"
argType: DATA_TYPE
}
-}
-opList {
- name: "dynamic_rnn"
argDescriptor {
name: "h"
argType: OUTPUT_TENSOR
@@ -6691,17 +6841,17 @@ opList {
argType: INPUT_TENSOR
argIndex: 4
}
+}
+opList {
+ name: "dynamic_stitch"
argDescriptor {
- name: "timeMajor"
+ name: "numPartitions"
argType: INT64
}
argDescriptor {
name: "dtype"
argType: DATA_TYPE
}
-}
-opList {
- name: "dynamic_stitch"
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -6715,14 +6865,6 @@ opList {
argType: INPUT_TENSOR
argIndex: 1
}
- argDescriptor {
- name: "numPartitions"
- argType: INT64
- }
- argDescriptor {
- name: "dtype"
- argType: DATA_TYPE
- }
}
opList {
name: "elu"
@@ -6730,14 +6872,14 @@ opList {
name: "outputs"
argType: OUTPUT_TENSOR
}
- argDescriptor {
- name: "input"
- argType: INPUT_TENSOR
- }
argDescriptor {
name: "alpha"
argType: DOUBLE
}
+ argDescriptor {
+ name: "input"
+ argType: INPUT_TENSOR
+ }
opDeclarationType: CONFIGURABLE_OP_IMPL
}
opList {
@@ -6746,6 +6888,10 @@ opList {
name: "outputs"
argType: OUTPUT_TENSOR
}
+ argDescriptor {
+ name: "alpha"
+ argType: DOUBLE
+ }
argDescriptor {
name: "input"
argType: INPUT_TENSOR
@@ -6755,14 +6901,18 @@ opList {
argType: INPUT_TENSOR
argIndex: 1
}
- argDescriptor {
- name: "alpha"
- argType: DOUBLE
- }
opDeclarationType: CONFIGURABLE_OP_IMPL
}
opList {
name: "embedding_lookup"
+ argDescriptor {
+ name: "partition_mode"
+ argType: INT64
+ }
+ argDescriptor {
+ name: "dtype"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -6776,17 +6926,13 @@ opList {
argType: INPUT_TENSOR
argIndex: 1
}
- argDescriptor {
- name: "partition_mode"
- argType: INT64
- }
+}
+opList {
+ name: "encode_bitmap"
argDescriptor {
name: "dtype"
argType: DATA_TYPE
}
-}
-opList {
- name: "encode_bitmap"
argDescriptor {
name: "encoded"
argType: OUTPUT_TENSOR
@@ -6796,6 +6942,10 @@ opList {
argType: OUTPUT_TENSOR
argIndex: 1
}
+ argDescriptor {
+ name: "threshold"
+ argType: DOUBLE
+ }
argDescriptor {
name: "input"
argType: INPUT_TENSOR
@@ -6810,17 +6960,17 @@ opList {
argType: INPUT_TENSOR
argIndex: 2
}
+}
+opList {
+ name: "encode_threshold"
+ argDescriptor {
+ name: "boundary"
+ argType: INT64
+ }
argDescriptor {
name: "dtype"
argType: DATA_TYPE
}
- argDescriptor {
- name: "threshold"
- argType: DOUBLE
- }
-}
-opList {
- name: "encode_threshold"
argDescriptor {
name: "updated"
argType: OUTPUT_TENSOR
@@ -6830,6 +6980,10 @@ opList {
argType: OUTPUT_TENSOR
argIndex: 1
}
+ argDescriptor {
+ name: "threshold"
+ argType: DOUBLE
+ }
argDescriptor {
name: "input"
argType: INPUT_TENSOR
@@ -6839,18 +6993,6 @@ opList {
argType: INPUT_TENSOR
argIndex: 1
}
- argDescriptor {
- name: "boundary"
- argType: INT64
- }
- argDescriptor {
- name: "dtype"
- argType: DATA_TYPE
- }
- argDescriptor {
- name: "threshold"
- argType: DOUBLE
- }
}
opList {
name: "enter"
@@ -6858,6 +7000,10 @@ opList {
name: "outputs"
argType: OUTPUT_TENSOR
}
+ argDescriptor {
+ name: "frameName"
+ argType: STRING
+ }
argDescriptor {
name: "isConstant"
argType: BOOL
@@ -6866,13 +7012,13 @@ opList {
name: "input"
argType: INPUT_TENSOR
}
- argDescriptor {
- name: "frameName"
- argType: STRING
- }
}
opList {
name: "entropy"
+ argDescriptor {
+ name: "dimensions"
+ argType: INT64
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -6890,21 +7036,17 @@ opList {
argType: INPUT_TENSOR
argIndex: 1
}
- argDescriptor {
- name: "dimensions"
- argType: INT64
- }
opDeclarationType: LEGACY_XYZ
}
opList {
name: "eps"
argDescriptor {
- name: "outputs"
- argType: OUTPUT_TENSOR
+ name: "dataType"
+ argType: DATA_TYPE
}
argDescriptor {
- name: "inPlace"
- argType: BOOL
+ name: "outputs"
+ argType: OUTPUT_TENSOR
}
argDescriptor {
name: "input"
@@ -6915,10 +7057,6 @@ opList {
argType: INPUT_TENSOR
argIndex: 1
}
- argDescriptor {
- name: "dataType"
- argType: DATA_TYPE
- }
opDeclarationType: LEGACY_XYZ
}
opList {
@@ -6939,6 +7077,10 @@ opList {
}
opList {
name: "equals"
+ argDescriptor {
+ name: "dataType"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -6956,10 +7098,6 @@ opList {
argType: INPUT_TENSOR
argIndex: 1
}
- argDescriptor {
- name: "dataType"
- argType: DATA_TYPE
- }
opDeclarationType: BROADCASTABLE_BOOL_OP_IMPL
}
opList {
@@ -6980,6 +7118,10 @@ opList {
}
opList {
name: "equals_with_eps"
+ argDescriptor {
+ name: "dimensions"
+ argType: INT64
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -6988,6 +7130,10 @@ opList {
name: "keepDims"
argType: BOOL
}
+ argDescriptor {
+ name: "eps"
+ argType: DOUBLE
+ }
argDescriptor {
name: "input"
argType: INPUT_TENSOR
@@ -6997,18 +7143,14 @@ opList {
argType: INPUT_TENSOR
argIndex: 1
}
- argDescriptor {
- name: "dimensions"
- argType: INT64
- }
- argDescriptor {
- name: "eps"
- argType: DOUBLE
- }
opDeclarationType: LEGACY_XYZ
}
opList {
name: "erf"
+ argDescriptor {
+ name: "dataType"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -7021,14 +7163,14 @@ opList {
name: "input"
argType: INPUT_TENSOR
}
- argDescriptor {
- name: "dataType"
- argType: DATA_TYPE
- }
opDeclarationType: LEGACY_XYZ
}
opList {
name: "erfc"
+ argDescriptor {
+ name: "dataType"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -7041,14 +7183,14 @@ opList {
name: "input"
argType: INPUT_TENSOR
}
- argDescriptor {
- name: "dataType"
- argType: DATA_TYPE
- }
opDeclarationType: LEGACY_XYZ
}
opList {
name: "euclidean"
+ argDescriptor {
+ name: "dimensions"
+ argType: INT64
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -7071,14 +7213,14 @@ opList {
argType: INPUT_TENSOR
argIndex: 1
}
- argDescriptor {
- name: "dimensions"
- argType: INT64
- }
opDeclarationType: LEGACY_XYZ
}
opList {
name: "evaluate_reduction_shape"
+ argDescriptor {
+ name: "dtype"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -7101,10 +7243,6 @@ opList {
argType: INPUT_TENSOR
argIndex: 1
}
- argDescriptor {
- name: "dtype"
- argType: DATA_TYPE
- }
}
opList {
name: "exit"
@@ -7112,17 +7250,21 @@ opList {
name: "outputs"
argType: OUTPUT_TENSOR
}
- argDescriptor {
- name: "input"
- argType: INPUT_TENSOR
- }
argDescriptor {
name: "frameName"
argType: STRING
}
+ argDescriptor {
+ name: "input"
+ argType: INPUT_TENSOR
+ }
}
opList {
name: "exp"
+ argDescriptor {
+ name: "dataType"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -7135,22 +7277,10 @@ opList {
name: "input"
argType: INPUT_TENSOR
}
- argDescriptor {
- name: "dataType"
- argType: DATA_TYPE
- }
opDeclarationType: LEGACY_XYZ
}
opList {
name: "expand_dims"
- argDescriptor {
- name: "outputs"
- argType: OUTPUT_TENSOR
- }
- argDescriptor {
- name: "input"
- argType: INPUT_TENSOR
- }
argDescriptor {
name: "dimensions"
argType: INT64
@@ -7159,9 +7289,6 @@ opList {
name: "dtype"
argType: DATA_TYPE
}
-}
-opList {
- name: "expm1"
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -7174,41 +7301,44 @@ opList {
name: "input"
argType: INPUT_TENSOR
}
+}
+opList {
+ name: "expm1"
argDescriptor {
name: "dataType"
argType: DATA_TYPE
}
- opDeclarationType: LEGACY_XYZ
-}
-opList {
- name: "expose"
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
}
argDescriptor {
- name: "input"
- argType: INPUT_TENSOR
- }
- argDescriptor {
- name: "dtype"
- argType: DATA_TYPE
- }
-}
-opList {
- name: "extract_image_patches"
- argDescriptor {
- name: "outputs"
- argType: OUTPUT_TENSOR
- }
- argDescriptor {
- name: "sameMode"
+ name: "inPlace"
argType: BOOL
}
argDescriptor {
name: "input"
argType: INPUT_TENSOR
}
+ opDeclarationType: LEGACY_XYZ
+}
+opList {
+ name: "expose"
+ argDescriptor {
+ name: "dtype"
+ argType: DATA_TYPE
+ }
+ argDescriptor {
+ name: "outputs"
+ argType: OUTPUT_TENSOR
+ }
+ argDescriptor {
+ name: "input"
+ argType: INPUT_TENSOR
+ }
+}
+opList {
+ name: "extract_image_patches"
argDescriptor {
name: "ksizeRows"
argType: INT64
@@ -7247,13 +7377,21 @@ opList {
name: "dtype"
argType: DATA_TYPE
}
-}
-opList {
- name: "eye"
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
}
+ argDescriptor {
+ name: "sameMode"
+ argType: BOOL
+ }
+ argDescriptor {
+ name: "input"
+ argType: INPUT_TENSOR
+ }
+}
+opList {
+ name: "eye"
argDescriptor {
name: "numRows"
argType: INT64
@@ -7277,13 +7415,30 @@ opList {
name: "dtype"
argType: DATA_TYPE
}
+ argDescriptor {
+ name: "outputs"
+ argType: OUTPUT_TENSOR
+ }
argDescriptor {
name: "dataType"
argType: DOUBLE
}
+ argDescriptor {
+ name: "numRows"
+ argType: INPUT_TENSOR
+ }
+ argDescriptor {
+ name: "numCols"
+ argType: INPUT_TENSOR
+ argIndex: 1
+ }
}
opList {
name: "fake_quant_with_min_max_args"
+ argDescriptor {
+ name: "numBits"
+ argType: INT64
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -7292,14 +7447,6 @@ opList {
name: "narrowRange"
argType: BOOL
}
- argDescriptor {
- name: "input"
- argType: INPUT_TENSOR
- }
- argDescriptor {
- name: "numBits"
- argType: INT64
- }
argDescriptor {
name: "min"
argType: DOUBLE
@@ -7309,9 +7456,17 @@ opList {
argType: DOUBLE
argIndex: 1
}
+ argDescriptor {
+ name: "input"
+ argType: INPUT_TENSOR
+ }
}
opList {
name: "fake_quant_with_min_max_vars"
+ argDescriptor {
+ name: "numBits"
+ argType: INT64
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -7320,24 +7475,6 @@ opList {
name: "narrowed"
argType: BOOL
}
- argDescriptor {
- name: "input"
- argType: INPUT_TENSOR
- }
- argDescriptor {
- name: "min"
- argType: INPUT_TENSOR
- argIndex: 1
- }
- argDescriptor {
- name: "max"
- argType: INPUT_TENSOR
- argIndex: 2
- }
- argDescriptor {
- name: "numBits"
- argType: INT64
- }
argDescriptor {
name: "m"
argType: DOUBLE
@@ -7347,10 +7484,28 @@ opList {
argType: DOUBLE
argIndex: 1
}
+ argDescriptor {
+ name: "input"
+ argType: INPUT_TENSOR
+ }
+ argDescriptor {
+ name: "min"
+ argType: INPUT_TENSOR
+ argIndex: 1
+ }
+ argDescriptor {
+ name: "max"
+ argType: INPUT_TENSOR
+ argIndex: 2
+ }
opDeclarationType: CONFIGURABLE_OP_IMPL
}
opList {
name: "fake_quant_with_min_max_vars_per_channel"
+ argDescriptor {
+ name: "numBits"
+ argType: INT64
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -7373,27 +7528,10 @@ opList {
argType: INPUT_TENSOR
argIndex: 2
}
- argDescriptor {
- name: "numBits"
- argType: INT64
- }
opDeclarationType: CONFIGURABLE_OP_IMPL
}
opList {
name: "fill"
- argDescriptor {
- name: "outputs"
- argType: OUTPUT_TENSOR
- }
- argDescriptor {
- name: "shape"
- argType: INPUT_TENSOR
- }
- argDescriptor {
- name: "outputs"
- argType: INPUT_TENSOR
- argIndex: 1
- }
argDescriptor {
name: "dtype"
argType: INT64
@@ -7402,10 +7540,23 @@ opList {
name: "dtype"
argType: DATA_TYPE
}
+ argDescriptor {
+ name: "outputs"
+ argType: OUTPUT_TENSOR
+ }
argDescriptor {
name: "value"
argType: DOUBLE
}
+ argDescriptor {
+ name: "shape"
+ argType: INPUT_TENSOR
+ }
+ argDescriptor {
+ name: "outputs"
+ argType: INPUT_TENSOR
+ argIndex: 1
+ }
}
opList {
name: "fill_as"
@@ -7421,6 +7572,10 @@ opList {
}
opList {
name: "firas_sparse"
+ argDescriptor {
+ name: "dtype"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -7429,13 +7584,13 @@ opList {
name: "input"
argType: INPUT_TENSOR
}
- argDescriptor {
- name: "dtype"
- argType: DATA_TYPE
- }
}
opList {
name: "first_index"
+ argDescriptor {
+ name: "dimensions"
+ argType: INT64
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -7444,14 +7599,6 @@ opList {
name: "keepDims"
argType: BOOL
}
- argDescriptor {
- name: "input"
- argType: INPUT_TENSOR
- }
- argDescriptor {
- name: "dimensions"
- argType: INT64
- }
argDescriptor {
name: "compare"
argType: DOUBLE
@@ -7461,18 +7608,14 @@ opList {
argType: DOUBLE
argIndex: 1
}
- opDeclarationType: LEGACY_XYZ
-}
-opList {
- name: "flatten"
- argDescriptor {
- name: "outputs"
- argType: OUTPUT_TENSOR
- }
argDescriptor {
name: "input"
argType: INPUT_TENSOR
}
+ opDeclarationType: LEGACY_XYZ
+}
+opList {
+ name: "flatten"
argDescriptor {
name: "order"
argType: INT64
@@ -7481,9 +7624,6 @@ opList {
name: "dtype"
argType: DATA_TYPE
}
-}
-opList {
- name: "flatten_2d"
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -7492,17 +7632,32 @@ opList {
name: "input"
argType: INPUT_TENSOR
}
+}
+opList {
+ name: "flatten_2d"
argDescriptor {
- name: "dimensions"
+ name: "flattenDimension"
argType: INT64
}
argDescriptor {
name: "dtype"
argType: DATA_TYPE
}
+ argDescriptor {
+ name: "outputs"
+ argType: OUTPUT_TENSOR
+ }
+ argDescriptor {
+ name: "input"
+ argType: INPUT_TENSOR
+ }
}
opList {
name: "floor"
+ argDescriptor {
+ name: "dataType"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -7515,14 +7670,14 @@ opList {
name: "input"
argType: INPUT_TENSOR
}
- argDescriptor {
- name: "dataType"
- argType: DATA_TYPE
- }
opDeclarationType: LEGACY_XYZ
}
opList {
name: "floordiv"
+ argDescriptor {
+ name: "dataType"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -7540,14 +7695,14 @@ opList {
argType: INPUT_TENSOR
argIndex: 1
}
- argDescriptor {
- name: "dataType"
- argType: DATA_TYPE
- }
opDeclarationType: BROADCASTABLE_OP_IMPL
}
opList {
name: "floordiv_bp"
+ argDescriptor {
+ name: "dtype"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "gradX"
argType: OUTPUT_TENSOR
@@ -7575,13 +7730,13 @@ opList {
argType: INPUT_TENSOR
argIndex: 2
}
- argDescriptor {
- name: "dtype"
- argType: DATA_TYPE
- }
}
opList {
name: "floormod"
+ argDescriptor {
+ name: "dataType"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -7599,14 +7754,14 @@ opList {
argType: INPUT_TENSOR
argIndex: 1
}
- argDescriptor {
- name: "dataType"
- argType: DATA_TYPE
- }
opDeclarationType: BROADCASTABLE_OP_IMPL
}
opList {
name: "floormod_bp"
+ argDescriptor {
+ name: "dtype"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "gradX"
argType: OUTPUT_TENSOR
@@ -7634,13 +7789,13 @@ opList {
argType: INPUT_TENSOR
argIndex: 2
}
- argDescriptor {
- name: "dtype"
- argType: DATA_TYPE
- }
}
opList {
name: "fmod"
+ argDescriptor {
+ name: "dataType"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -7658,10 +7813,6 @@ opList {
argType: INPUT_TENSOR
argIndex: 1
}
- argDescriptor {
- name: "dataType"
- argType: DATA_TYPE
- }
opDeclarationType: LEGACY_XYZ
}
opList {
@@ -7683,6 +7834,19 @@ opList {
}
opList {
name: "fused_batch_norm"
+ argDescriptor {
+ name: "dataFormat"
+ argType: INT64
+ }
+ argDescriptor {
+ name: "isTraining"
+ argType: INT64
+ argIndex: 1
+ }
+ argDescriptor {
+ name: "dtype"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "y"
argType: OUTPUT_TENSOR
@@ -7697,6 +7861,10 @@ opList {
argType: OUTPUT_TENSOR
argIndex: 2
}
+ argDescriptor {
+ name: "epsilon"
+ argType: DOUBLE
+ }
argDescriptor {
name: "input"
argType: INPUT_TENSOR
@@ -7726,26 +7894,17 @@ opList {
argType: INPUT_TENSOR
argIndex: 5
}
+}
+opList {
+ name: "gather"
argDescriptor {
- name: "dataFormat"
+ name: "dimensions"
argType: INT64
}
- argDescriptor {
- name: "isTraining"
- argType: INT64
- argIndex: 1
- }
argDescriptor {
name: "dtype"
argType: DATA_TYPE
}
- argDescriptor {
- name: "epsilon"
- argType: DOUBLE
- }
-}
-opList {
- name: "gather"
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -7768,17 +7927,13 @@ opList {
argType: INPUT_TENSOR
argIndex: 2
}
- argDescriptor {
- name: "dimensions"
- argType: INT64
- }
+}
+opList {
+ name: "gather_list"
argDescriptor {
name: "dtype"
argType: DATA_TYPE
}
-}
-opList {
- name: "gather_list"
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -7792,14 +7947,14 @@ opList {
argType: INPUT_TENSOR
argIndex: 1
}
- argDescriptor {
- name: "dtype"
- argType: DATA_TYPE
- }
opDeclarationType: LIST_OP_IMPL
}
opList {
name: "gather_nd"
+ argDescriptor {
+ name: "dtype"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -7817,13 +7972,13 @@ opList {
argType: INPUT_TENSOR
argIndex: 1
}
- argDescriptor {
- name: "dtype"
- argType: DATA_TYPE
- }
}
opList {
name: "gelu"
+ argDescriptor {
+ name: "dataType"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -7841,22 +7996,18 @@ opList {
name: "input"
argType: INPUT_TENSOR
}
- argDescriptor {
- name: "dataType"
- argType: DATA_TYPE
- }
opDeclarationType: LEGACY_XYZ
}
opList {
name: "get_seed"
- argDescriptor {
- name: "outputs"
- argType: OUTPUT_TENSOR
- }
argDescriptor {
name: "dtype"
argType: DATA_TYPE
}
+ argDescriptor {
+ name: "outputs"
+ argType: OUTPUT_TENSOR
+ }
}
opList {
name: "gradientbackwards"
@@ -7876,6 +8027,10 @@ opList {
}
opList {
name: "greater"
+ argDescriptor {
+ name: "dataType"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -7893,14 +8048,14 @@ opList {
argType: INPUT_TENSOR
argIndex: 1
}
- argDescriptor {
- name: "dataType"
- argType: DATA_TYPE
- }
opDeclarationType: BROADCASTABLE_BOOL_OP_IMPL
}
opList {
name: "greater_equal"
+ argDescriptor {
+ name: "dataType"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -7918,10 +8073,6 @@ opList {
argType: INPUT_TENSOR
argIndex: 1
}
- argDescriptor {
- name: "dataType"
- argType: DATA_TYPE
- }
opDeclarationType: BROADCASTABLE_BOOL_OP_IMPL
}
opList {
@@ -7975,6 +8126,10 @@ opList {
}
opList {
name: "gru"
+ argDescriptor {
+ name: "dtype"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "h"
argType: OUTPUT_TENSOR
@@ -8003,13 +8158,13 @@ opList {
argType: INPUT_TENSOR
argIndex: 4
}
+}
+opList {
+ name: "gruCell"
argDescriptor {
name: "dtype"
argType: DATA_TYPE
}
-}
-opList {
- name: "gruCell"
argDescriptor {
name: "r"
argType: OUTPUT_TENSOR
@@ -8058,13 +8213,13 @@ opList {
argType: INPUT_TENSOR
argIndex: 5
}
+}
+opList {
+ name: "gruCell_bp"
argDescriptor {
name: "dtype"
argType: DATA_TYPE
}
-}
-opList {
- name: "gruCell_bp"
argDescriptor {
name: "dLdx"
argType: OUTPUT_TENSOR
@@ -8143,13 +8298,13 @@ opList {
argType: INPUT_TENSOR
argIndex: 9
}
+}
+opList {
+ name: "gru_bp"
argDescriptor {
name: "dtype"
argType: DATA_TYPE
}
-}
-opList {
- name: "gru_bp"
argDescriptor {
name: "dLdx"
argType: OUTPUT_TENSOR
@@ -8203,13 +8358,13 @@ opList {
argType: INPUT_TENSOR
argIndex: 5
}
- argDescriptor {
- name: "dtype"
- argType: DATA_TYPE
- }
}
opList {
name: "hammingdistance"
+ argDescriptor {
+ name: "dimensions"
+ argType: INT64
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -8232,14 +8387,14 @@ opList {
argType: INPUT_TENSOR
argIndex: 1
}
- argDescriptor {
- name: "dimensions"
- argType: INT64
- }
opDeclarationType: LEGACY_XYZ
}
opList {
name: "hard_sigmoid"
+ argDescriptor {
+ name: "dataType"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -8252,14 +8407,14 @@ opList {
name: "input"
argType: INPUT_TENSOR
}
- argDescriptor {
- name: "dataType"
- argType: DATA_TYPE
- }
opDeclarationType: LEGACY_XYZ
}
opList {
name: "hard_sigmoidderivative"
+ argDescriptor {
+ name: "dataType"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -8272,10 +8427,6 @@ opList {
name: "input"
argType: INPUT_TENSOR
}
- argDescriptor {
- name: "dataType"
- argType: DATA_TYPE
- }
opDeclarationType: LEGACY_XYZ
}
opList {
@@ -8309,6 +8460,10 @@ opList {
}
opList {
name: "hardtanh"
+ argDescriptor {
+ name: "dataType"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -8321,10 +8476,6 @@ opList {
name: "input"
argType: INPUT_TENSOR
}
- argDescriptor {
- name: "dataType"
- argType: DATA_TYPE
- }
opDeclarationType: CONFIGURABLE_OP_IMPL
}
opList {
@@ -8346,6 +8497,10 @@ opList {
}
opList {
name: "hardtanhderivative"
+ argDescriptor {
+ name: "dataType"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -8358,14 +8513,14 @@ opList {
name: "input"
argType: INPUT_TENSOR
}
- argDescriptor {
- name: "dataType"
- argType: DATA_TYPE
- }
opDeclarationType: LEGACY_XYZ
}
opList {
name: "hashcode"
+ argDescriptor {
+ name: "dtype"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -8374,13 +8529,13 @@ opList {
name: "input"
argType: INPUT_TENSOR
}
- argDescriptor {
- name: "dtype"
- argType: DATA_TYPE
- }
}
opList {
name: "hasinf"
+ argDescriptor {
+ name: "dimensions"
+ argType: INT64
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -8393,14 +8548,14 @@ opList {
name: "input"
argType: INPUT_TENSOR
}
- argDescriptor {
- name: "dimensions"
- argType: INT64
- }
opDeclarationType: LEGACY_XYZ
}
opList {
name: "hasnan"
+ argDescriptor {
+ name: "dimensions"
+ argType: INT64
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -8413,14 +8568,18 @@ opList {
name: "input"
argType: INPUT_TENSOR
}
- argDescriptor {
- name: "dimensions"
- argType: INT64
- }
opDeclarationType: LEGACY_XYZ
}
opList {
name: "hinge_loss"
+ argDescriptor {
+ name: "reductionMode"
+ argType: INT64
+ }
+ argDescriptor {
+ name: "dtype"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -8439,6 +8598,9 @@ opList {
argType: INPUT_TENSOR
argIndex: 2
}
+}
+opList {
+ name: "hinge_loss_grad"
argDescriptor {
name: "reductionMode"
argType: INT64
@@ -8447,9 +8609,6 @@ opList {
name: "dtype"
argType: DATA_TYPE
}
-}
-opList {
- name: "hinge_loss_grad"
argDescriptor {
name: "dLdp"
argType: OUTPUT_TENSOR
@@ -8478,25 +8637,9 @@ opList {
argType: INPUT_TENSOR
argIndex: 2
}
- argDescriptor {
- name: "reductionMode"
- argType: INT64
- }
- argDescriptor {
- name: "dtype"
- argType: DATA_TYPE
- }
}
opList {
name: "histogram"
- argDescriptor {
- name: "outputs"
- argType: OUTPUT_TENSOR
- }
- argDescriptor {
- name: "input"
- argType: INPUT_TENSOR
- }
argDescriptor {
name: "numBins"
argType: INT64
@@ -8505,9 +8648,25 @@ opList {
name: "dtype"
argType: DATA_TYPE
}
+ argDescriptor {
+ name: "outputs"
+ argType: OUTPUT_TENSOR
+ }
+ argDescriptor {
+ name: "input"
+ argType: INPUT_TENSOR
+ }
}
opList {
name: "histogram_fixed_width"
+ argDescriptor {
+ name: "nbins"
+ argType: INT64
+ }
+ argDescriptor {
+ name: "dtype"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -8526,17 +8685,13 @@ opList {
argType: INPUT_TENSOR
argIndex: 2
}
- argDescriptor {
- name: "nbins"
- argType: INT64
- }
- argDescriptor {
- name: "dtype"
- argType: DATA_TYPE
- }
}
opList {
name: "hsv_to_rgb"
+ argDescriptor {
+ name: "dimC"
+ argType: INT64
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -8545,18 +8700,26 @@ opList {
name: "input"
argType: INPUT_TENSOR
}
- argDescriptor {
- name: "dimC"
- argType: INT64
- }
opDeclarationType: CONFIGURABLE_OP_IMPL
}
opList {
name: "huber_loss"
+ argDescriptor {
+ name: "reductionMode"
+ argType: INT64
+ }
+ argDescriptor {
+ name: "dtype"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
}
+ argDescriptor {
+ name: "delta"
+ argType: DOUBLE
+ }
argDescriptor {
name: "predictions"
argType: INPUT_TENSOR
@@ -8571,6 +8734,9 @@ opList {
argType: INPUT_TENSOR
argIndex: 2
}
+}
+opList {
+ name: "huber_loss_grad"
argDescriptor {
name: "reductionMode"
argType: INT64
@@ -8579,13 +8745,6 @@ opList {
name: "dtype"
argType: DATA_TYPE
}
- argDescriptor {
- name: "delta"
- argType: DOUBLE
- }
-}
-opList {
- name: "huber_loss_grad"
argDescriptor {
name: "dLdp"
argType: OUTPUT_TENSOR
@@ -8600,6 +8759,10 @@ opList {
argType: OUTPUT_TENSOR
argIndex: 2
}
+ argDescriptor {
+ name: "delta"
+ argType: DOUBLE
+ }
argDescriptor {
name: "predictions"
argType: INPUT_TENSOR
@@ -8614,21 +8777,13 @@ opList {
argType: INPUT_TENSOR
argIndex: 2
}
- argDescriptor {
- name: "reductionMode"
- argType: INT64
- }
- argDescriptor {
- name: "dtype"
- argType: DATA_TYPE
- }
- argDescriptor {
- name: "delta"
- argType: DOUBLE
- }
}
opList {
name: "identity"
+ argDescriptor {
+ name: "dataType"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -8641,10 +8796,6 @@ opList {
name: "input"
argType: INPUT_TENSOR
}
- argDescriptor {
- name: "dataType"
- argType: DATA_TYPE
- }
opDeclarationType: OP_IMPL
}
opList {
@@ -8666,6 +8817,10 @@ opList {
}
opList {
name: "identity_n"
+ argDescriptor {
+ name: "dtype"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -8674,10 +8829,6 @@ opList {
name: "input"
argType: INPUT_TENSOR
}
- argDescriptor {
- name: "dtype"
- argType: DATA_TYPE
- }
}
opList {
name: "igamma"
@@ -8715,19 +8866,6 @@ opList {
}
opList {
name: "im2col"
- argDescriptor {
- name: "outputs"
- argType: OUTPUT_TENSOR
- }
- argDescriptor {
- name: "input"
- argType: INPUT_TENSOR
- }
- argDescriptor {
- name: "inputArrays"
- argType: INPUT_TENSOR
- argIndex: 1
- }
argDescriptor {
name: "kernelHeight"
argType: INT64
@@ -8771,26 +8909,26 @@ opList {
name: "dtype"
argType: DATA_TYPE
}
- argDescriptor {
- name: "zeroPadVal"
- argType: DOUBLE
- }
-}
-opList {
- name: "im2col_bp"
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
}
+ argDescriptor {
+ name: "zeroPadVal"
+ argType: DOUBLE
+ }
argDescriptor {
name: "input"
argType: INPUT_TENSOR
}
argDescriptor {
- name: "gradAtOutput"
+ name: "inputArrays"
argType: INPUT_TENSOR
argIndex: 1
}
+}
+opList {
+ name: "im2col_bp"
argDescriptor {
name: "kernelHeight"
argType: INT64
@@ -8834,13 +8972,34 @@ opList {
name: "dtype"
argType: DATA_TYPE
}
+ argDescriptor {
+ name: "outputs"
+ argType: OUTPUT_TENSOR
+ }
argDescriptor {
name: "zeroPadVal"
argType: DOUBLE
}
+ argDescriptor {
+ name: "input"
+ argType: INPUT_TENSOR
+ }
+ argDescriptor {
+ name: "gradAtOutput"
+ argType: INPUT_TENSOR
+ argIndex: 1
+ }
}
opList {
name: "image_resize"
+ argDescriptor {
+ name: "method"
+ argType: INT64
+ }
+ argDescriptor {
+ name: "dtype"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -8863,17 +9022,17 @@ opList {
argType: INPUT_TENSOR
argIndex: 1
}
+}
+opList {
+ name: "in_top_k"
argDescriptor {
- name: "method"
+ name: "k"
argType: INT64
}
argDescriptor {
name: "dtype"
argType: DATA_TYPE
}
-}
-opList {
- name: "in_top_k"
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -8891,17 +9050,13 @@ opList {
argType: INPUT_TENSOR
argIndex: 1
}
- argDescriptor {
- name: "k"
- argType: INT64
- }
- argDescriptor {
- name: "dtype"
- argType: DATA_TYPE
- }
}
opList {
name: "invert_permutation"
+ argDescriptor {
+ name: "dataType"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -8914,10 +9069,6 @@ opList {
name: "input"
argType: INPUT_TENSOR
}
- argDescriptor {
- name: "dataType"
- argType: DATA_TYPE
- }
opDeclarationType: CONFIGURABLE_OP_IMPL
}
opList {
@@ -8970,6 +9121,10 @@ opList {
}
opList {
name: "isfinite"
+ argDescriptor {
+ name: "dataType"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -8982,14 +9137,14 @@ opList {
name: "input"
argType: INPUT_TENSOR
}
- argDescriptor {
- name: "dataType"
- argType: DATA_TYPE
- }
opDeclarationType: LEGACY_XYZ
}
opList {
name: "isinf"
+ argDescriptor {
+ name: "dataType"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -9002,14 +9157,14 @@ opList {
name: "input"
argType: INPUT_TENSOR
}
- argDescriptor {
- name: "dataType"
- argType: DATA_TYPE
- }
opDeclarationType: LEGACY_XYZ
}
opList {
name: "ismax"
+ argDescriptor {
+ name: "dimensions"
+ argType: INT64
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -9018,14 +9173,14 @@ opList {
name: "input"
argType: INPUT_TENSOR
}
- argDescriptor {
- name: "dimensions"
- argType: INT64
- }
opDeclarationType: CONFIGURABLE_OP_IMPL
}
opList {
name: "isnan"
+ argDescriptor {
+ name: "dataType"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -9038,14 +9193,14 @@ opList {
name: "input"
argType: INPUT_TENSOR
}
- argDescriptor {
- name: "dataType"
- argType: DATA_TYPE
- }
opDeclarationType: LEGACY_XYZ
}
opList {
name: "jaccarddistance"
+ argDescriptor {
+ name: "dimensions"
+ argType: INT64
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -9068,14 +9223,14 @@ opList {
argType: INPUT_TENSOR
argIndex: 1
}
- argDescriptor {
- name: "dimensions"
- argType: INT64
- }
opDeclarationType: LEGACY_XYZ
}
opList {
name: "knn_mindistance"
+ argDescriptor {
+ name: "dtype"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -9099,13 +9254,13 @@ opList {
argType: INPUT_TENSOR
argIndex: 3
}
+}
+opList {
+ name: "l2_loss"
argDescriptor {
name: "dtype"
argType: DATA_TYPE
}
-}
-opList {
- name: "l2_loss"
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -9114,13 +9269,13 @@ opList {
name: "input"
argType: INPUT_TENSOR
}
- argDescriptor {
- name: "dtype"
- argType: DATA_TYPE
- }
}
opList {
name: "last_index"
+ argDescriptor {
+ name: "dimensions"
+ argType: INT64
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -9129,14 +9284,6 @@ opList {
name: "keepDims"
argType: BOOL
}
- argDescriptor {
- name: "input"
- argType: INPUT_TENSOR
- }
- argDescriptor {
- name: "dimensions"
- argType: INT64
- }
argDescriptor {
name: "compare"
argType: DOUBLE
@@ -9146,16 +9293,24 @@ opList {
argType: DOUBLE
argIndex: 1
}
+ argDescriptor {
+ name: "input"
+ argType: INPUT_TENSOR
+ }
opDeclarationType: LEGACY_XYZ
}
opList {
name: "layer_norm"
+ argDescriptor {
+ name: "dimensions"
+ argType: INT64
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
}
argDescriptor {
- name: "channelsFirst"
+ name: "noBias"
argType: BOOL
}
argDescriptor {
@@ -9172,14 +9327,18 @@ opList {
argType: INPUT_TENSOR
argIndex: 2
}
- argDescriptor {
- name: "dimensions"
- argType: INT64
- }
opDeclarationType: CONFIGURABLE_OP_IMPL
}
opList {
name: "layer_norm_bp"
+ argDescriptor {
+ name: "dimensions"
+ argType: INT64
+ }
+ argDescriptor {
+ name: "dtype"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "dLdx"
argType: OUTPUT_TENSOR
@@ -9195,7 +9354,7 @@ opList {
argIndex: 2
}
argDescriptor {
- name: "channelsFirst"
+ name: "noBias"
argType: BOOL
}
argDescriptor {
@@ -9232,14 +9391,6 @@ opList {
argType: INPUT_TENSOR
argIndex: 6
}
- argDescriptor {
- name: "dimensions"
- argType: INT64
- }
- argDescriptor {
- name: "dtype"
- argType: DATA_TYPE
- }
}
opList {
name: "leakyrelu"
@@ -9248,17 +9399,13 @@ opList {
argType: OUTPUT_TENSOR
}
argDescriptor {
- name: "inPlace"
- argType: BOOL
+ name: "alpha"
+ argType: DOUBLE
}
argDescriptor {
name: "input"
argType: INPUT_TENSOR
}
- argDescriptor {
- name: "alpha"
- argType: DOUBLE
- }
opDeclarationType: LEGACY_XYZ
}
opList {
@@ -9268,21 +9415,21 @@ opList {
argType: OUTPUT_TENSOR
}
argDescriptor {
- name: "inPlace"
- argType: BOOL
+ name: "alpha"
+ argType: DOUBLE
}
argDescriptor {
name: "input"
argType: INPUT_TENSOR
}
- argDescriptor {
- name: "alpha"
- argType: DOUBLE
- }
opDeclarationType: LEGACY_XYZ
}
opList {
name: "less"
+ argDescriptor {
+ name: "dataType"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -9300,14 +9447,14 @@ opList {
argType: INPUT_TENSOR
argIndex: 1
}
- argDescriptor {
- name: "dataType"
- argType: DATA_TYPE
- }
opDeclarationType: BROADCASTABLE_BOOL_OP_IMPL
}
opList {
name: "less_equal"
+ argDescriptor {
+ name: "dataType"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -9325,10 +9472,6 @@ opList {
argType: INPUT_TENSOR
argIndex: 1
}
- argDescriptor {
- name: "dataType"
- argType: DATA_TYPE
- }
opDeclarationType: BROADCASTABLE_BOOL_OP_IMPL
}
opList {
@@ -9377,10 +9520,27 @@ opList {
}
opList {
name: "lin_space"
+ argDescriptor {
+ name: "dataType"
+ argType: INT64
+ }
+ argDescriptor {
+ name: "dataType"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
}
+ argDescriptor {
+ name: "start"
+ argType: DOUBLE
+ }
+ argDescriptor {
+ name: "stop"
+ argType: DOUBLE
+ argIndex: 1
+ }
argDescriptor {
name: "start"
argType: INPUT_TENSOR
@@ -9395,45 +9555,23 @@ opList {
argType: INPUT_TENSOR
argIndex: 2
}
- argDescriptor {
- name: "dataType"
- argType: INT64
- }
- argDescriptor {
- name: "dataType"
- argType: DATA_TYPE
- }
- argDescriptor {
- name: "start"
- argType: DOUBLE
- }
- argDescriptor {
- name: "stop"
- argType: DOUBLE
- argIndex: 1
- }
}
opList {
name: "linspace_random"
- argDescriptor {
- name: "outputs"
- argType: OUTPUT_TENSOR
- }
argDescriptor {
name: "length"
argType: INT64
}
argDescriptor {
- name: "dataType"
- argType: INT64
- argIndex: 1
+ name: "outputs"
+ argType: OUTPUT_TENSOR
}
argDescriptor {
name: "from"
argType: DOUBLE
}
argDescriptor {
- name: "step"
+ name: "to"
argType: DOUBLE
argIndex: 1
}
@@ -9441,6 +9579,10 @@ opList {
}
opList {
name: "listdiff"
+ argDescriptor {
+ name: "dtype"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "output1"
argType: OUTPUT_TENSOR
@@ -9459,13 +9601,13 @@ opList {
argType: INPUT_TENSOR
argIndex: 1
}
- argDescriptor {
- name: "dtype"
- argType: DATA_TYPE
- }
}
opList {
name: "log"
+ argDescriptor {
+ name: "dataType"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -9478,14 +9620,14 @@ opList {
name: "input"
argType: INPUT_TENSOR
}
- argDescriptor {
- name: "dataType"
- argType: DATA_TYPE
- }
opDeclarationType: LEGACY_XYZ
}
opList {
name: "log1p"
+ argDescriptor {
+ name: "dataType"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -9498,18 +9640,26 @@ opList {
name: "input"
argType: INPUT_TENSOR
}
- argDescriptor {
- name: "dataType"
- argType: DATA_TYPE
- }
opDeclarationType: LEGACY_XYZ
}
opList {
name: "log_loss"
+ argDescriptor {
+ name: "reductionMode"
+ argType: INT64
+ }
+ argDescriptor {
+ name: "dtype"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
}
+ argDescriptor {
+ name: "epsilon"
+ argType: DOUBLE
+ }
argDescriptor {
name: "predictions"
argType: INPUT_TENSOR
@@ -9524,6 +9674,9 @@ opList {
argType: INPUT_TENSOR
argIndex: 2
}
+}
+opList {
+ name: "log_loss_grad"
argDescriptor {
name: "reductionMode"
argType: INT64
@@ -9532,13 +9685,6 @@ opList {
name: "dtype"
argType: DATA_TYPE
}
- argDescriptor {
- name: "epsilon"
- argType: DOUBLE
- }
-}
-opList {
- name: "log_loss_grad"
argDescriptor {
name: "dLdp"
argType: OUTPUT_TENSOR
@@ -9553,6 +9699,10 @@ opList {
argType: OUTPUT_TENSOR
argIndex: 2
}
+ argDescriptor {
+ name: "epsilon"
+ argType: DOUBLE
+ }
argDescriptor {
name: "predictions"
argType: INPUT_TENSOR
@@ -9567,21 +9717,13 @@ opList {
argType: INPUT_TENSOR
argIndex: 2
}
- argDescriptor {
- name: "reductionMode"
- argType: INT64
- }
+}
+opList {
+ name: "log_matrix_determinant"
argDescriptor {
name: "dtype"
argType: DATA_TYPE
}
- argDescriptor {
- name: "epsilon"
- argType: DOUBLE
- }
-}
-opList {
- name: "log_matrix_determinant"
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -9594,13 +9736,17 @@ opList {
name: "input"
argType: INPUT_TENSOR
}
+}
+opList {
+ name: "log_poisson_loss"
+ argDescriptor {
+ name: "reductionMode"
+ argType: INT64
+ }
argDescriptor {
name: "dtype"
argType: DATA_TYPE
}
-}
-opList {
- name: "log_poisson_loss"
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -9623,6 +9769,9 @@ opList {
argType: INPUT_TENSOR
argIndex: 2
}
+}
+opList {
+ name: "log_poisson_loss_grad"
argDescriptor {
name: "reductionMode"
argType: INT64
@@ -9631,9 +9780,6 @@ opList {
name: "dtype"
argType: DATA_TYPE
}
-}
-opList {
- name: "log_poisson_loss_grad"
argDescriptor {
name: "dLdp"
argType: OUTPUT_TENSOR
@@ -9666,17 +9812,13 @@ opList {
argType: INPUT_TENSOR
argIndex: 2
}
- argDescriptor {
- name: "reductionMode"
- argType: INT64
- }
- argDescriptor {
- name: "dtype"
- argType: DATA_TYPE
- }
}
opList {
name: "log_softmax"
+ argDescriptor {
+ name: "dimension"
+ argType: INT64
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -9685,14 +9827,14 @@ opList {
name: "input"
argType: INPUT_TENSOR
}
- argDescriptor {
- name: "dimension"
- argType: INT64
- }
opDeclarationType: CONFIGURABLE_OP_IMPL
}
opList {
name: "log_softmax_bp"
+ argDescriptor {
+ name: "dimension"
+ argType: INT64
+ }
argDescriptor {
name: "gradI"
argType: OUTPUT_TENSOR
@@ -9706,10 +9848,6 @@ opList {
argType: INPUT_TENSOR
argIndex: 1
}
- argDescriptor {
- name: "dimension"
- argType: INT64
- }
opDeclarationType: CONFIGURABLE_OP_IMPL
}
opList {
@@ -9718,18 +9856,22 @@ opList {
name: "outputs"
argType: OUTPUT_TENSOR
}
- argDescriptor {
- name: "input"
- argType: INPUT_TENSOR
- }
argDescriptor {
name: "base"
argType: DOUBLE
}
+ argDescriptor {
+ name: "input"
+ argType: INPUT_TENSOR
+ }
opDeclarationType: LEGACY_XYZ
}
opList {
name: "logdet"
+ argDescriptor {
+ name: "dtype"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -9738,13 +9880,13 @@ opList {
name: "input"
argType: INPUT_TENSOR
}
- argDescriptor {
- name: "dtype"
- argType: DATA_TYPE
- }
}
opList {
name: "logentropy"
+ argDescriptor {
+ name: "dimensions"
+ argType: INT64
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -9762,14 +9904,14 @@ opList {
argType: INPUT_TENSOR
argIndex: 1
}
- argDescriptor {
- name: "dimensions"
- argType: INT64
- }
opDeclarationType: LEGACY_XYZ
}
opList {
name: "logsigmoid"
+ argDescriptor {
+ name: "dataType"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -9782,10 +9924,6 @@ opList {
name: "input"
argType: INPUT_TENSOR
}
- argDescriptor {
- name: "dataType"
- argType: DATA_TYPE
- }
opDeclarationType: LEGACY_XYZ
}
opList {
@@ -9794,14 +9932,14 @@ opList {
name: "outputs"
argType: OUTPUT_TENSOR
}
- argDescriptor {
- name: "input"
- argType: INPUT_TENSOR
- }
argDescriptor {
name: "frameName"
argType: STRING
}
+ argDescriptor {
+ name: "input"
+ argType: INPUT_TENSOR
+ }
}
opList {
name: "lrelu"
@@ -9809,14 +9947,14 @@ opList {
name: "outputs"
argType: OUTPUT_TENSOR
}
- argDescriptor {
- name: "input"
- argType: INPUT_TENSOR
- }
argDescriptor {
name: "alpha"
argType: DOUBLE
}
+ argDescriptor {
+ name: "input"
+ argType: INPUT_TENSOR
+ }
opDeclarationType: CONFIGURABLE_OP_IMPL
}
opList {
@@ -9825,6 +9963,10 @@ opList {
name: "outputs"
argType: OUTPUT_TENSOR
}
+ argDescriptor {
+ name: "alpha"
+ argType: DOUBLE
+ }
argDescriptor {
name: "input"
argType: INPUT_TENSOR
@@ -9834,14 +9976,14 @@ opList {
argType: INPUT_TENSOR
argIndex: 1
}
- argDescriptor {
- name: "alpha"
- argType: DOUBLE
- }
opDeclarationType: CONFIGURABLE_OP_IMPL
}
opList {
name: "lrn"
+ argDescriptor {
+ name: "depth"
+ argType: INT64
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -9850,14 +9992,6 @@ opList {
name: "inPlace"
argType: BOOL
}
- argDescriptor {
- name: "input"
- argType: INPUT_TENSOR
- }
- argDescriptor {
- name: "depth"
- argType: INT64
- }
argDescriptor {
name: "bias"
argType: DOUBLE
@@ -9872,10 +10006,18 @@ opList {
argType: DOUBLE
argIndex: 2
}
+ argDescriptor {
+ name: "input"
+ argType: INPUT_TENSOR
+ }
opDeclarationType: CONFIGURABLE_OP_IMPL
}
opList {
name: "lrn_bp"
+ argDescriptor {
+ name: "depth"
+ argType: INT64
+ }
argDescriptor {
name: "gradI"
argType: OUTPUT_TENSOR
@@ -9884,6 +10026,20 @@ opList {
name: "inPlace"
argType: BOOL
}
+ argDescriptor {
+ name: "bias"
+ argType: DOUBLE
+ }
+ argDescriptor {
+ name: "alpha"
+ argType: DOUBLE
+ argIndex: 1
+ }
+ argDescriptor {
+ name: "beta"
+ argType: DOUBLE
+ argIndex: 2
+ }
argDescriptor {
name: "input"
argType: INPUT_TENSOR
@@ -9893,28 +10049,23 @@ opList {
argType: INPUT_TENSOR
argIndex: 1
}
- argDescriptor {
- name: "depth"
- argType: INT64
- }
- argDescriptor {
- name: "bias"
- argType: DOUBLE
- }
- argDescriptor {
- name: "alpha"
- argType: DOUBLE
- argIndex: 1
- }
- argDescriptor {
- name: "beta"
- argType: DOUBLE
- argIndex: 2
- }
opDeclarationType: CONFIGURABLE_OP_IMPL
}
opList {
name: "lstm"
+ argDescriptor {
+ name: "peephole"
+ argType: INT64
+ }
+ argDescriptor {
+ name: "projection"
+ argType: INT64
+ argIndex: 1
+ }
+ argDescriptor {
+ name: "dtype"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "h"
argType: OUTPUT_TENSOR
@@ -9924,6 +10075,20 @@ opList {
argType: OUTPUT_TENSOR
argIndex: 1
}
+ argDescriptor {
+ name: "clippingCellValue"
+ argType: DOUBLE
+ }
+ argDescriptor {
+ name: "clippingProjValue"
+ argType: DOUBLE
+ argIndex: 1
+ }
+ argDescriptor {
+ name: "forgetBias"
+ argType: DOUBLE
+ argIndex: 2
+ }
argDescriptor {
name: "input"
argType: INPUT_TENSOR
@@ -9963,12 +10128,15 @@ opList {
argType: INPUT_TENSOR
argIndex: 7
}
+}
+opList {
+ name: "lstmBlock"
argDescriptor {
name: "peephole"
argType: INT64
}
argDescriptor {
- name: "projection"
+ name: "dataFormat"
argType: INT64
argIndex: 1
}
@@ -9976,23 +10144,6 @@ opList {
name: "dtype"
argType: DATA_TYPE
}
- argDescriptor {
- name: "clippingCellValue"
- argType: DOUBLE
- }
- argDescriptor {
- name: "clippingProjValue"
- argType: DOUBLE
- argIndex: 1
- }
- argDescriptor {
- name: "forgetBias"
- argType: DOUBLE
- argIndex: 2
- }
-}
-opList {
- name: "lstmBlock"
argDescriptor {
name: "i"
argType: OUTPUT_TENSOR
@@ -10027,6 +10178,15 @@ opList {
argType: OUTPUT_TENSOR
argIndex: 6
}
+ argDescriptor {
+ name: "forgetBias"
+ argType: DOUBLE
+ }
+ argDescriptor {
+ name: "clippingCellValue"
+ argType: DOUBLE
+ argIndex: 1
+ }
argDescriptor {
name: "maxTSLength"
argType: INPUT_TENSOR
@@ -10071,31 +10231,17 @@ opList {
argType: INPUT_TENSOR
argIndex: 8
}
+}
+opList {
+ name: "lstmBlockCell"
argDescriptor {
name: "peephole"
argType: INT64
}
- argDescriptor {
- name: "dataFormat"
- argType: INT64
- argIndex: 1
- }
argDescriptor {
name: "dtype"
argType: DATA_TYPE
}
- argDescriptor {
- name: "forgetBias"
- argType: DOUBLE
- }
- argDescriptor {
- name: "clippingCellValue"
- argType: DOUBLE
- argIndex: 1
- }
-}
-opList {
- name: "lstmBlockCell"
argDescriptor {
name: "i"
argType: OUTPUT_TENSOR
@@ -10130,6 +10276,15 @@ opList {
argType: OUTPUT_TENSOR
argIndex: 6
}
+ argDescriptor {
+ name: "forgetBias"
+ argType: DOUBLE
+ }
+ argDescriptor {
+ name: "clippingCellValue"
+ argType: DOUBLE
+ argIndex: 1
+ }
argDescriptor {
name: "xt"
argType: INPUT_TENSOR
@@ -10169,26 +10324,22 @@ opList {
argType: INPUT_TENSOR
argIndex: 7
}
+}
+opList {
+ name: "lstmCell"
argDescriptor {
name: "peephole"
argType: INT64
}
+ argDescriptor {
+ name: "projection"
+ argType: INT64
+ argIndex: 1
+ }
argDescriptor {
name: "dtype"
argType: DATA_TYPE
}
- argDescriptor {
- name: "forgetBias"
- argType: DOUBLE
- }
- argDescriptor {
- name: "clippingCellValue"
- argType: DOUBLE
- argIndex: 1
- }
-}
-opList {
- name: "lstmCell"
argDescriptor {
name: "ht"
argType: OUTPUT_TENSOR
@@ -10198,6 +10349,20 @@ opList {
argType: OUTPUT_TENSOR
argIndex: 1
}
+ argDescriptor {
+ name: "clippingCellValue"
+ argType: DOUBLE
+ }
+ argDescriptor {
+ name: "clippingProjValue"
+ argType: DOUBLE
+ argIndex: 1
+ }
+ argDescriptor {
+ name: "forgetBias"
+ argType: DOUBLE
+ argIndex: 2
+ }
argDescriptor {
name: "xt"
argType: INPUT_TENSOR
@@ -10237,36 +10402,37 @@ opList {
argType: INPUT_TENSOR
argIndex: 7
}
+}
+opList {
+ name: "lstmLayer"
argDescriptor {
- name: "peephole"
+ name: "dataFormat"
argType: INT64
}
argDescriptor {
- name: "projection"
+ name: "directionMode"
argType: INT64
argIndex: 1
}
+ argDescriptor {
+ name: "gateAct"
+ argType: INT64
+ argIndex: 2
+ }
+ argDescriptor {
+ name: "cellAct"
+ argType: INT64
+ argIndex: 3
+ }
+ argDescriptor {
+ name: "outAct"
+ argType: INT64
+ argIndex: 4
+ }
argDescriptor {
name: "dtype"
argType: DATA_TYPE
}
- argDescriptor {
- name: "clippingCellValue"
- argType: DOUBLE
- }
- argDescriptor {
- name: "clippingProjValue"
- argType: DOUBLE
- argIndex: 1
- }
- argDescriptor {
- name: "forgetBias"
- argType: DOUBLE
- argIndex: 2
- }
-}
-opList {
- name: "lstmLayer"
argDescriptor {
name: "h"
argType: OUTPUT_TENSOR
@@ -10320,6 +10486,40 @@ opList {
argType: BOOL
argIndex: 7
}
+ argDescriptor {
+ name: "cellClip"
+ argType: DOUBLE
+ }
+ argDescriptor {
+ name: "gateAlpha"
+ argType: DOUBLE
+ argIndex: 1
+ }
+ argDescriptor {
+ name: "gateBeta"
+ argType: DOUBLE
+ argIndex: 2
+ }
+ argDescriptor {
+ name: "cellAlpha"
+ argType: DOUBLE
+ argIndex: 3
+ }
+ argDescriptor {
+ name: "cellBeta"
+ argType: DOUBLE
+ argIndex: 4
+ }
+ argDescriptor {
+ name: "outAlpha"
+ argType: DOUBLE
+ argIndex: 5
+ }
+ argDescriptor {
+ name: "outBeta"
+ argType: DOUBLE
+ argIndex: 6
+ }
argDescriptor {
name: "input"
argType: INPUT_TENSOR
@@ -10359,34 +10559,45 @@ opList {
argType: INPUT_TENSOR
argIndex: 7
}
- argDescriptor {
- name: "dataFormat"
- argType: INT64
- }
- argDescriptor {
- name: "directionMode"
- argType: INT64
- argIndex: 1
- }
+}
+opList {
+ name: "lstmLayerCell"
argDescriptor {
name: "gateAct"
argType: INT64
- argIndex: 2
}
argDescriptor {
name: "cellAct"
argType: INT64
- argIndex: 3
+ argIndex: 1
}
argDescriptor {
name: "outAct"
argType: INT64
- argIndex: 4
+ argIndex: 2
}
argDescriptor {
name: "dtype"
argType: DATA_TYPE
}
+ argDescriptor {
+ name: "h"
+ argType: OUTPUT_TENSOR
+ }
+ argDescriptor {
+ name: "c"
+ argType: OUTPUT_TENSOR
+ argIndex: 1
+ }
+ argDescriptor {
+ name: "hasBiases"
+ argType: BOOL
+ }
+ argDescriptor {
+ name: "hasPH"
+ argType: BOOL
+ argIndex: 1
+ }
argDescriptor {
name: "cellClip"
argType: DOUBLE
@@ -10421,27 +10632,6 @@ opList {
argType: DOUBLE
argIndex: 6
}
-}
-opList {
- name: "lstmLayerCell"
- argDescriptor {
- name: "h"
- argType: OUTPUT_TENSOR
- }
- argDescriptor {
- name: "c"
- argType: OUTPUT_TENSOR
- argIndex: 1
- }
- argDescriptor {
- name: "hasBiases"
- argType: BOOL
- }
- argDescriptor {
- name: "hasPH"
- argType: BOOL
- argIndex: 1
- }
argDescriptor {
name: "input"
argType: INPUT_TENSOR
@@ -10476,6 +10666,9 @@ opList {
argType: INPUT_TENSOR
argIndex: 6
}
+}
+opList {
+ name: "lstmLayerCellBp"
argDescriptor {
name: "gateAct"
argType: INT64
@@ -10494,43 +10687,6 @@ opList {
name: "dtype"
argType: DATA_TYPE
}
- argDescriptor {
- name: "cellClip"
- argType: DOUBLE
- }
- argDescriptor {
- name: "gateAlpha"
- argType: DOUBLE
- argIndex: 1
- }
- argDescriptor {
- name: "gateBeta"
- argType: DOUBLE
- argIndex: 2
- }
- argDescriptor {
- name: "cellAlpha"
- argType: DOUBLE
- argIndex: 3
- }
- argDescriptor {
- name: "cellBeta"
- argType: DOUBLE
- argIndex: 4
- }
- argDescriptor {
- name: "outAlpha"
- argType: DOUBLE
- argIndex: 5
- }
- argDescriptor {
- name: "outBeta"
- argType: DOUBLE
- argIndex: 6
- }
-}
-opList {
- name: "lstmLayerCellBp"
argDescriptor {
name: "dLdx"
argType: OUTPUT_TENSOR
@@ -10574,6 +10730,40 @@ opList {
argType: BOOL
argIndex: 1
}
+ argDescriptor {
+ name: "cellClip"
+ argType: DOUBLE
+ }
+ argDescriptor {
+ name: "gateAlpha"
+ argType: DOUBLE
+ argIndex: 1
+ }
+ argDescriptor {
+ name: "gateBeta"
+ argType: DOUBLE
+ argIndex: 2
+ }
+ argDescriptor {
+ name: "cellAlpha"
+ argType: DOUBLE
+ argIndex: 3
+ }
+ argDescriptor {
+ name: "cellBeta"
+ argType: DOUBLE
+ argIndex: 4
+ }
+ argDescriptor {
+ name: "outAlpha"
+ argType: DOUBLE
+ argIndex: 5
+ }
+ argDescriptor {
+ name: "outBeta"
+ argType: DOUBLE
+ argIndex: 6
+ }
argDescriptor {
name: "input"
argType: INPUT_TENSOR
@@ -10613,61 +10803,37 @@ opList {
argType: INPUT_TENSOR
argIndex: 7
}
+}
+opList {
+ name: "lstmLayer_bp"
argDescriptor {
- name: "gateAct"
+ name: "dataFormat"
argType: INT64
}
argDescriptor {
- name: "cellAct"
+ name: "directionMode"
argType: INT64
argIndex: 1
}
argDescriptor {
- name: "outAct"
+ name: "gateAct"
argType: INT64
argIndex: 2
}
+ argDescriptor {
+ name: "cellAct"
+ argType: INT64
+ argIndex: 3
+ }
+ argDescriptor {
+ name: "outAct"
+ argType: INT64
+ argIndex: 4
+ }
argDescriptor {
name: "dtype"
argType: DATA_TYPE
}
- argDescriptor {
- name: "cellClip"
- argType: DOUBLE
- }
- argDescriptor {
- name: "gateAlpha"
- argType: DOUBLE
- argIndex: 1
- }
- argDescriptor {
- name: "gateBeta"
- argType: DOUBLE
- argIndex: 2
- }
- argDescriptor {
- name: "cellAlpha"
- argType: DOUBLE
- argIndex: 3
- }
- argDescriptor {
- name: "cellBeta"
- argType: DOUBLE
- argIndex: 4
- }
- argDescriptor {
- name: "outAlpha"
- argType: DOUBLE
- argIndex: 5
- }
- argDescriptor {
- name: "outBeta"
- argType: DOUBLE
- argIndex: 6
- }
-}
-opList {
- name: "lstmLayer_bp"
argDescriptor {
name: "dLdx"
argType: OUTPUT_TENSOR
@@ -10741,6 +10907,40 @@ opList {
argType: BOOL
argIndex: 7
}
+ argDescriptor {
+ name: "cellClip"
+ argType: DOUBLE
+ }
+ argDescriptor {
+ name: "gateAlpha"
+ argType: DOUBLE
+ argIndex: 1
+ }
+ argDescriptor {
+ name: "gateBeta"
+ argType: DOUBLE
+ argIndex: 2
+ }
+ argDescriptor {
+ name: "cellAlpha"
+ argType: DOUBLE
+ argIndex: 3
+ }
+ argDescriptor {
+ name: "cellBeta"
+ argType: DOUBLE
+ argIndex: 4
+ }
+ argDescriptor {
+ name: "outAlpha"
+ argType: DOUBLE
+ argIndex: 5
+ }
+ argDescriptor {
+ name: "outBeta"
+ argType: DOUBLE
+ argIndex: 6
+ }
argDescriptor {
name: "input"
argType: INPUT_TENSOR
@@ -10800,71 +11000,13 @@ opList {
argType: INPUT_TENSOR
argIndex: 11
}
- argDescriptor {
- name: "dataFormat"
- argType: INT64
- }
- argDescriptor {
- name: "directionMode"
- argType: INT64
- argIndex: 1
- }
- argDescriptor {
- name: "gateAct"
- argType: INT64
- argIndex: 2
- }
- argDescriptor {
- name: "cellAct"
- argType: INT64
- argIndex: 3
- }
- argDescriptor {
- name: "outAct"
- argType: INT64
- argIndex: 4
- }
+}
+opList {
+ name: "lstsq"
argDescriptor {
name: "dtype"
argType: DATA_TYPE
}
- argDescriptor {
- name: "cellClip"
- argType: DOUBLE
- }
- argDescriptor {
- name: "gateAlpha"
- argType: DOUBLE
- argIndex: 1
- }
- argDescriptor {
- name: "gateBeta"
- argType: DOUBLE
- argIndex: 2
- }
- argDescriptor {
- name: "cellAlpha"
- argType: DOUBLE
- argIndex: 3
- }
- argDescriptor {
- name: "cellBeta"
- argType: DOUBLE
- argIndex: 4
- }
- argDescriptor {
- name: "outAlpha"
- argType: DOUBLE
- argIndex: 5
- }
- argDescriptor {
- name: "outBeta"
- argType: DOUBLE
- argIndex: 6
- }
-}
-opList {
- name: "lstsq"
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -10873,6 +11015,10 @@ opList {
name: "fastFlag"
argType: BOOL
}
+ argDescriptor {
+ name: "l2_factor"
+ argType: DOUBLE
+ }
argDescriptor {
name: "a"
argType: INPUT_TENSOR
@@ -10882,17 +11028,13 @@ opList {
argType: INPUT_TENSOR
argIndex: 1
}
+}
+opList {
+ name: "lu"
argDescriptor {
name: "dtype"
argType: DATA_TYPE
}
- argDescriptor {
- name: "l2_factor"
- argType: DOUBLE
- }
-}
-opList {
- name: "lu"
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -10906,13 +11048,13 @@ opList {
name: "input"
argType: INPUT_TENSOR
}
- argDescriptor {
- name: "dtype"
- argType: DATA_TYPE
- }
}
opList {
name: "manhattan"
+ argDescriptor {
+ name: "dimensions"
+ argType: INT64
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -10935,14 +11077,14 @@ opList {
argType: INPUT_TENSOR
argIndex: 1
}
- argDescriptor {
- name: "dimensions"
- argType: INT64
- }
opDeclarationType: LEGACY_XYZ
}
opList {
name: "match_condition"
+ argDescriptor {
+ name: "dimensions"
+ argType: INT64
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -10951,14 +11093,6 @@ opList {
name: "keepDims"
argType: BOOL
}
- argDescriptor {
- name: "input"
- argType: INPUT_TENSOR
- }
- argDescriptor {
- name: "dimensions"
- argType: INT64
- }
argDescriptor {
name: "compare"
argType: DOUBLE
@@ -10968,18 +11102,14 @@ opList {
argType: DOUBLE
argIndex: 1
}
- opDeclarationType: LEGACY_XYZ
-}
-opList {
- name: "match_condition_transform"
- argDescriptor {
- name: "outputs"
- argType: OUTPUT_TENSOR
- }
argDescriptor {
name: "input"
argType: INPUT_TENSOR
}
+ opDeclarationType: LEGACY_XYZ
+}
+opList {
+ name: "match_condition_transform"
argDescriptor {
name: "mode"
argType: INT64
@@ -10988,32 +11118,19 @@ opList {
name: "dataType"
argType: DATA_TYPE
}
- argDescriptor {
- name: "compare"
- argType: DOUBLE
- }
- opDeclarationType: LEGACY_XYZ
-}
-opList {
- name: "matmul"
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
}
argDescriptor {
- name: "transposeX"
- argType: BOOL
+ name: "compare"
+ argType: DOUBLE
}
argDescriptor {
- name: "transposeY"
- argType: BOOL
+ name: "eps"
+ argType: DOUBLE
argIndex: 1
}
- argDescriptor {
- name: "transposeZ"
- argType: BOOL
- argIndex: 2
- }
argDescriptor {
name: "input"
argType: INPUT_TENSOR
@@ -11023,6 +11140,10 @@ opList {
argType: INPUT_TENSOR
argIndex: 1
}
+ opDeclarationType: LEGACY_XYZ
+}
+opList {
+ name: "matmul"
argDescriptor {
name: "transX"
argType: INT64
@@ -11041,6 +11162,24 @@ opList {
name: "dtype"
argType: DATA_TYPE
}
+ argDescriptor {
+ name: "outputs"
+ argType: OUTPUT_TENSOR
+ }
+ argDescriptor {
+ name: "transposeX"
+ argType: BOOL
+ }
+ argDescriptor {
+ name: "transposeY"
+ argType: BOOL
+ argIndex: 1
+ }
+ argDescriptor {
+ name: "transposeZ"
+ argType: BOOL
+ argIndex: 2
+ }
argDescriptor {
name: "alpha"
argType: DOUBLE
@@ -11050,9 +11189,36 @@ opList {
argType: DOUBLE
argIndex: 1
}
+ argDescriptor {
+ name: "input"
+ argType: INPUT_TENSOR
+ }
+ argDescriptor {
+ name: "y"
+ argType: INPUT_TENSOR
+ argIndex: 1
+ }
}
opList {
name: "matmul_bp"
+ argDescriptor {
+ name: "transX"
+ argType: INT64
+ }
+ argDescriptor {
+ name: "transY"
+ argType: INT64
+ argIndex: 1
+ }
+ argDescriptor {
+ name: "transZ"
+ argType: INT64
+ argIndex: 2
+ }
+ argDescriptor {
+ name: "dtype"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "dldx"
argType: OUTPUT_TENSOR
@@ -11062,6 +11228,15 @@ opList {
argType: OUTPUT_TENSOR
argIndex: 1
}
+ argDescriptor {
+ name: "alpha"
+ argType: DOUBLE
+ }
+ argDescriptor {
+ name: "beta"
+ argType: DOUBLE
+ argIndex: 1
+ }
argDescriptor {
name: "input"
argType: INPUT_TENSOR
@@ -11086,36 +11261,18 @@ opList {
argType: INPUT_TENSOR
argIndex: 4
}
- argDescriptor {
- name: "transX"
- argType: INT64
- }
- argDescriptor {
- name: "transY"
- argType: INT64
- argIndex: 1
- }
- argDescriptor {
- name: "transZ"
- argType: INT64
- argIndex: 2
- }
- argDescriptor {
- name: "dtype"
- argType: DATA_TYPE
- }
- argDescriptor {
- name: "alpha"
- argType: DOUBLE
- }
- argDescriptor {
- name: "beta"
- argType: DOUBLE
- argIndex: 1
- }
}
opList {
name: "matrix_band_part"
+ argDescriptor {
+ name: "minLower"
+ argType: INT64
+ }
+ argDescriptor {
+ name: "maxUpper"
+ argType: INT64
+ argIndex: 1
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -11134,19 +11291,14 @@ opList {
argType: INPUT_TENSOR
argIndex: 2
}
- argDescriptor {
- name: "minLower"
- argType: INT64
- }
- argDescriptor {
- name: "maxUpper"
- argType: INT64
- argIndex: 1
- }
opDeclarationType: CONFIGURABLE_OP_IMPL
}
opList {
name: "matrix_determinant"
+ argDescriptor {
+ name: "dtype"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -11159,13 +11311,13 @@ opList {
name: "input"
argType: INPUT_TENSOR
}
+}
+opList {
+ name: "matrix_diag"
argDescriptor {
name: "dtype"
argType: DATA_TYPE
}
-}
-opList {
- name: "matrix_diag"
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -11178,13 +11330,13 @@ opList {
name: "diagonal"
argType: INPUT_TENSOR
}
+}
+opList {
+ name: "matrix_diag_part"
argDescriptor {
name: "dtype"
argType: DATA_TYPE
}
-}
-opList {
- name: "matrix_diag_part"
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -11197,10 +11349,6 @@ opList {
name: "input"
argType: INPUT_TENSOR
}
- argDescriptor {
- name: "dtype"
- argType: DATA_TYPE
- }
}
opList {
name: "matrix_inverse"
@@ -11241,6 +11389,10 @@ opList {
}
opList {
name: "max_pairwise"
+ argDescriptor {
+ name: "dataType"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -11254,32 +11406,10 @@ opList {
argType: INPUT_TENSOR
argIndex: 1
}
- argDescriptor {
- name: "dataType"
- argType: DATA_TYPE
- }
opDeclarationType: LEGACY_XYZ
}
opList {
name: "max_pool_with_argmax"
- argDescriptor {
- name: "outputs"
- argType: OUTPUT_TENSOR
- }
- argDescriptor {
- name: "indices"
- argType: OUTPUT_TENSOR
- argIndex: 1
- }
- argDescriptor {
- name: "input"
- argType: INPUT_TENSOR
- }
- argDescriptor {
- name: "outArgMax"
- argType: INPUT_TENSOR
- argIndex: 1
- }
argDescriptor {
name: "kH"
argType: INT64
@@ -11338,6 +11468,24 @@ opList {
name: "dtype"
argType: DATA_TYPE
}
+ argDescriptor {
+ name: "outputs"
+ argType: OUTPUT_TENSOR
+ }
+ argDescriptor {
+ name: "indices"
+ argType: OUTPUT_TENSOR
+ argIndex: 1
+ }
+ argDescriptor {
+ name: "input"
+ argType: INPUT_TENSOR
+ }
+ argDescriptor {
+ name: "outArgMax"
+ argType: INPUT_TENSOR
+ argIndex: 1
+ }
}
opList {
name: "max_scalar"
@@ -11358,6 +11506,10 @@ opList {
}
opList {
name: "maximum"
+ argDescriptor {
+ name: "dataType"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -11375,14 +11527,14 @@ opList {
argType: INPUT_TENSOR
argIndex: 1
}
- argDescriptor {
- name: "dataType"
- argType: DATA_TYPE
- }
opDeclarationType: BROADCASTABLE_OP_IMPL
}
opList {
name: "maximum_bp"
+ argDescriptor {
+ name: "dtype"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "gradX"
argType: OUTPUT_TENSOR
@@ -11406,41 +11558,25 @@ opList {
argType: INPUT_TENSOR
argIndex: 2
}
- argDescriptor {
- name: "dtype"
- argType: DATA_TYPE
- }
}
opList {
name: "maxout"
+ argDescriptor {
+ name: "dataType"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
}
- argDescriptor {
- name: "inPlace"
- argType: BOOL
- }
argDescriptor {
name: "input"
argType: INPUT_TENSOR
}
- argDescriptor {
- name: "dataType"
- argType: DATA_TYPE
- }
opDeclarationType: LEGACY_XYZ
}
opList {
name: "maxpool2d"
- argDescriptor {
- name: "outputs"
- argType: OUTPUT_TENSOR
- }
- argDescriptor {
- name: "input"
- argType: INPUT_TENSOR
- }
argDescriptor {
name: "kH"
argType: INT64
@@ -11499,22 +11635,17 @@ opList {
name: "dtype"
argType: DATA_TYPE
}
+ argDescriptor {
+ name: "outputs"
+ argType: OUTPUT_TENSOR
+ }
+ argDescriptor {
+ name: "input"
+ argType: INPUT_TENSOR
+ }
}
opList {
name: "maxpool2d_bp"
- argDescriptor {
- name: "gradI"
- argType: OUTPUT_TENSOR
- }
- argDescriptor {
- name: "input"
- argType: INPUT_TENSOR
- }
- argDescriptor {
- name: "gradO"
- argType: INPUT_TENSOR
- argIndex: 1
- }
argDescriptor {
name: "kH"
argType: INT64
@@ -11573,98 +11704,6 @@ opList {
name: "dtype"
argType: DATA_TYPE
}
-}
-opList {
- name: "maxpool3dnew"
- argDescriptor {
- name: "outputs"
- argType: OUTPUT_TENSOR
- }
- argDescriptor {
- name: "input"
- argType: INPUT_TENSOR
- }
- argDescriptor {
- name: "kD"
- argType: INT64
- }
- argDescriptor {
- name: "kH"
- argType: INT64
- argIndex: 1
- }
- argDescriptor {
- name: "kW"
- argType: INT64
- argIndex: 2
- }
- argDescriptor {
- name: "sD"
- argType: INT64
- argIndex: 3
- }
- argDescriptor {
- name: "sH"
- argType: INT64
- argIndex: 4
- }
- argDescriptor {
- name: "sW"
- argType: INT64
- argIndex: 5
- }
- argDescriptor {
- name: "pD"
- argType: INT64
- argIndex: 6
- }
- argDescriptor {
- name: "pH"
- argType: INT64
- argIndex: 7
- }
- argDescriptor {
- name: "pW"
- argType: INT64
- argIndex: 8
- }
- argDescriptor {
- name: "dD"
- argType: INT64
- argIndex: 9
- }
- argDescriptor {
- name: "dH"
- argType: INT64
- argIndex: 10
- }
- argDescriptor {
- name: "dW"
- argType: INT64
- argIndex: 11
- }
- argDescriptor {
- name: "isSameMode"
- argType: INT64
- argIndex: 12
- }
- argDescriptor {
- name: "extraParam0"
- argType: INT64
- argIndex: 13
- }
- argDescriptor {
- name: "isNCDHW"
- argType: INT64
- argIndex: 14
- }
- argDescriptor {
- name: "dtype"
- argType: DATA_TYPE
- }
-}
-opList {
- name: "maxpool3dnew_bp"
argDescriptor {
name: "gradI"
argType: OUTPUT_TENSOR
@@ -11678,6 +11717,9 @@ opList {
argType: INPUT_TENSOR
argIndex: 1
}
+}
+opList {
+ name: "maxpool3dnew"
argDescriptor {
name: "kD"
argType: INT64
@@ -11756,9 +11798,124 @@ opList {
name: "dtype"
argType: DATA_TYPE
}
+ argDescriptor {
+ name: "outputs"
+ argType: OUTPUT_TENSOR
+ }
+ argDescriptor {
+ name: "input"
+ argType: INPUT_TENSOR
+ }
+ argDescriptor {
+ name: "arrayOutput"
+ argType: INPUT_TENSOR
+ argIndex: 1
+ }
+}
+opList {
+ name: "maxpool3dnew_bp"
+ argDescriptor {
+ name: "kD"
+ argType: INT64
+ }
+ argDescriptor {
+ name: "kH"
+ argType: INT64
+ argIndex: 1
+ }
+ argDescriptor {
+ name: "kW"
+ argType: INT64
+ argIndex: 2
+ }
+ argDescriptor {
+ name: "sD"
+ argType: INT64
+ argIndex: 3
+ }
+ argDescriptor {
+ name: "sH"
+ argType: INT64
+ argIndex: 4
+ }
+ argDescriptor {
+ name: "sW"
+ argType: INT64
+ argIndex: 5
+ }
+ argDescriptor {
+ name: "pD"
+ argType: INT64
+ argIndex: 6
+ }
+ argDescriptor {
+ name: "pH"
+ argType: INT64
+ argIndex: 7
+ }
+ argDescriptor {
+ name: "pW"
+ argType: INT64
+ argIndex: 8
+ }
+ argDescriptor {
+ name: "dD"
+ argType: INT64
+ argIndex: 9
+ }
+ argDescriptor {
+ name: "dH"
+ argType: INT64
+ argIndex: 10
+ }
+ argDescriptor {
+ name: "dW"
+ argType: INT64
+ argIndex: 11
+ }
+ argDescriptor {
+ name: "isSameMode"
+ argType: INT64
+ argIndex: 12
+ }
+ argDescriptor {
+ name: "extraParam0"
+ argType: INT64
+ argIndex: 13
+ }
+ argDescriptor {
+ name: "isNCDHW"
+ argType: INT64
+ argIndex: 14
+ }
+ argDescriptor {
+ name: "dtype"
+ argType: DATA_TYPE
+ }
+ argDescriptor {
+ name: "gradI"
+ argType: OUTPUT_TENSOR
+ }
+ argDescriptor {
+ name: "input"
+ argType: INPUT_TENSOR
+ }
+ argDescriptor {
+ name: "gradO"
+ argType: INPUT_TENSOR
+ argIndex: 1
+ }
}
opList {
name: "mean_pairwssqerr_loss"
+ argDescriptor {
+ name: "reductionMode"
+ argType: INT64
+ }
+ argDescriptor {
+ name: "dtype"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -11777,17 +11934,17 @@ opList {
argType: INPUT_TENSOR
argIndex: 2
}
- argDescriptor {
- name: "reductionMode"
- argType: INT64
- }
- argDescriptor {
- name: "dtype"
- argType: DATA_TYPE
- }
}
opList {
name: "mean_pairwssqerr_loss_grad"
+ argDescriptor {
+ name: "reductionMode"
+ argType: INT64
+ }
+ argDescriptor {
+ name: "dtype"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "dLdp"
argType: OUTPUT_TENSOR
@@ -11816,6 +11973,9 @@ opList {
argType: INPUT_TENSOR
argIndex: 2
}
+}
+opList {
+ name: "mean_sqerr_loss"
argDescriptor {
name: "reductionMode"
argType: INT64
@@ -11824,9 +11984,6 @@ opList {
name: "dtype"
argType: DATA_TYPE
}
-}
-opList {
- name: "mean_sqerr_loss"
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -11845,6 +12002,9 @@ opList {
argType: INPUT_TENSOR
argIndex: 2
}
+}
+opList {
+ name: "mean_sqerr_loss_grad"
argDescriptor {
name: "reductionMode"
argType: INT64
@@ -11853,9 +12013,6 @@ opList {
name: "dtype"
argType: DATA_TYPE
}
-}
-opList {
- name: "mean_sqerr_loss_grad"
argDescriptor {
name: "dLdp"
argType: OUTPUT_TENSOR
@@ -11884,14 +12041,6 @@ opList {
argType: INPUT_TENSOR
argIndex: 2
}
- argDescriptor {
- name: "reductionMode"
- argType: INT64
- }
- argDescriptor {
- name: "dtype"
- argType: DATA_TYPE
- }
}
opList {
name: "merge"
@@ -11906,6 +12055,10 @@ opList {
}
opList {
name: "mergeadd"
+ argDescriptor {
+ name: "dataType"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -11918,14 +12071,14 @@ opList {
name: "inArrs"
argType: INPUT_TENSOR
}
- argDescriptor {
- name: "dataType"
- argType: DATA_TYPE
- }
opDeclarationType: OP_IMPL
}
opList {
name: "mergeadd_bp"
+ argDescriptor {
+ name: "dtype"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outArrs"
argType: OUTPUT_TENSOR
@@ -11939,10 +12092,6 @@ opList {
argType: INPUT_TENSOR
argIndex: 1
}
- argDescriptor {
- name: "dtype"
- argType: DATA_TYPE
- }
}
opList {
name: "mergeavg"
@@ -11958,6 +12107,10 @@ opList {
}
opList {
name: "mergeavg_bp"
+ argDescriptor {
+ name: "dtype"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outArrs"
argType: OUTPUT_TENSOR
@@ -11971,10 +12124,6 @@ opList {
argType: INPUT_TENSOR
argIndex: 1
}
- argDescriptor {
- name: "dtype"
- argType: DATA_TYPE
- }
}
opList {
name: "mergemax"
@@ -11990,6 +12139,10 @@ opList {
}
opList {
name: "mergemax_bp"
+ argDescriptor {
+ name: "dtype"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outArrs"
argType: OUTPUT_TENSOR
@@ -12003,21 +12156,9 @@ opList {
argType: INPUT_TENSOR
argIndex: 1
}
- argDescriptor {
- name: "dtype"
- argType: DATA_TYPE
- }
}
opList {
name: "mergemaxindex"
- argDescriptor {
- name: "outputs"
- argType: OUTPUT_TENSOR
- }
- argDescriptor {
- name: "inArrs"
- argType: INPUT_TENSOR
- }
argDescriptor {
name: "dataType"
argType: INT64
@@ -12026,6 +12167,14 @@ opList {
name: "dtype"
argType: DATA_TYPE
}
+ argDescriptor {
+ name: "outputs"
+ argType: OUTPUT_TENSOR
+ }
+ argDescriptor {
+ name: "inArrs"
+ argType: INPUT_TENSOR
+ }
}
opList {
name: "mergesum"
@@ -12040,6 +12189,14 @@ opList {
}
opList {
name: "meshgrid"
+ argDescriptor {
+ name: "swapFirst2Dims"
+ argType: INT64
+ }
+ argDescriptor {
+ name: "dtype"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outArrs"
argType: OUTPUT_TENSOR
@@ -12052,14 +12209,6 @@ opList {
name: "inArrs"
argType: INPUT_TENSOR
}
- argDescriptor {
- name: "swapFirst2Dims"
- argType: INT64
- }
- argDescriptor {
- name: "dtype"
- argType: DATA_TYPE
- }
}
opList {
name: "meta_postulate"
@@ -12067,6 +12216,15 @@ opList {
name: "outputs"
argType: OUTPUT_TENSOR
}
+ argDescriptor {
+ name: "input"
+ argType: INPUT_TENSOR
+ }
+ argDescriptor {
+ name: "y"
+ argType: INPUT_TENSOR
+ argIndex: 1
+ }
opDeclarationType: LEGACY_XYZ
}
opList {
@@ -12083,31 +12241,26 @@ opList {
name: "outputs"
argType: OUTPUT_TENSOR
}
- argDescriptor {
- name: "input"
- argType: INPUT_TENSOR
- }
- argDescriptor {
- name: "y"
- argType: INPUT_TENSOR
- argIndex: 1
- }
opDeclarationType: LEGACY_XYZ
}
opList {
name: "meta_reduce"
- argDescriptor {
- name: "outputs"
- argType: OUTPUT_TENSOR
- }
argDescriptor {
name: "dimensions"
argType: INT64
}
+ argDescriptor {
+ name: "outputs"
+ argType: OUTPUT_TENSOR
+ }
opDeclarationType: LEGACY_XYZ
}
opList {
name: "min_pairwise"
+ argDescriptor {
+ name: "dataType"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -12121,14 +12274,14 @@ opList {
argType: INPUT_TENSOR
argIndex: 1
}
- argDescriptor {
- name: "dataType"
- argType: DATA_TYPE
- }
opDeclarationType: LEGACY_XYZ
}
opList {
name: "minimum"
+ argDescriptor {
+ name: "dataType"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -12146,14 +12299,14 @@ opList {
argType: INPUT_TENSOR
argIndex: 1
}
- argDescriptor {
- name: "dataType"
- argType: DATA_TYPE
- }
opDeclarationType: BROADCASTABLE_OP_IMPL
}
opList {
name: "minimum_bp"
+ argDescriptor {
+ name: "dtype"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "gradX"
argType: OUTPUT_TENSOR
@@ -12177,13 +12330,17 @@ opList {
argType: INPUT_TENSOR
argIndex: 2
}
+}
+opList {
+ name: "mirror_pad"
+ argDescriptor {
+ name: "mode"
+ argType: INT64
+ }
argDescriptor {
name: "dtype"
argType: DATA_TYPE
}
-}
-opList {
- name: "mirror_pad"
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -12201,17 +12358,13 @@ opList {
argType: INPUT_TENSOR
argIndex: 1
}
- argDescriptor {
- name: "mode"
- argType: INT64
- }
- argDescriptor {
- name: "dtype"
- argType: DATA_TYPE
- }
}
opList {
name: "mish"
+ argDescriptor {
+ name: "dataType"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -12224,14 +12377,14 @@ opList {
name: "input"
argType: INPUT_TENSOR
}
- argDescriptor {
- name: "dataType"
- argType: DATA_TYPE
- }
opDeclarationType: LEGACY_XYZ
}
opList {
name: "mod"
+ argDescriptor {
+ name: "dataType"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -12249,14 +12402,14 @@ opList {
argType: INPUT_TENSOR
argIndex: 1
}
- argDescriptor {
- name: "dataType"
- argType: DATA_TYPE
- }
opDeclarationType: BROADCASTABLE_OP_IMPL
}
opList {
name: "mod_bp"
+ argDescriptor {
+ name: "dtype"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "gradX"
argType: OUTPUT_TENSOR
@@ -12284,13 +12437,17 @@ opList {
argType: INPUT_TENSOR
argIndex: 2
}
+}
+opList {
+ name: "moments"
+ argDescriptor {
+ name: "dimensions"
+ argType: INT64
+ }
argDescriptor {
name: "dtype"
argType: DATA_TYPE
}
-}
-opList {
- name: "moments"
argDescriptor {
name: "means"
argType: OUTPUT_TENSOR
@@ -12300,6 +12457,10 @@ opList {
argType: OUTPUT_TENSOR
argIndex: 1
}
+ argDescriptor {
+ name: "keepDims"
+ argType: DOUBLE
+ }
argDescriptor {
name: "input"
argType: INPUT_TENSOR
@@ -12314,18 +12475,6 @@ opList {
argType: INPUT_TENSOR
argIndex: 2
}
- argDescriptor {
- name: "dimensions"
- argType: INT64
- }
- argDescriptor {
- name: "dtype"
- argType: DATA_TYPE
- }
- argDescriptor {
- name: "keepDims"
- argType: DOUBLE
- }
}
opList {
name: "mul_scalar"
@@ -12334,18 +12483,30 @@ opList {
argType: OUTPUT_TENSOR
}
argDescriptor {
- name: "input"
- argType: INPUT_TENSOR
+ name: "inPlace"
+ argType: BOOL
}
argDescriptor {
- name: "y"
+ name: "input"
argType: INPUT_TENSOR
- argIndex: 1
}
opDeclarationType: LEGACY_XYZ
}
opList {
name: "multi_head_dot_product_attention"
+ argDescriptor {
+ name: "normalization"
+ argType: INT64
+ }
+ argDescriptor {
+ name: "weights"
+ argType: INT64
+ argIndex: 1
+ }
+ argDescriptor {
+ name: "dtype"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -12398,22 +12559,17 @@ opList {
argType: INPUT_TENSOR
argIndex: 7
}
+}
+opList {
+ name: "multi_head_dot_product_attention_bp"
argDescriptor {
name: "normalization"
argType: INT64
}
- argDescriptor {
- name: "weights"
- argType: INT64
- argIndex: 1
- }
argDescriptor {
name: "dtype"
argType: DATA_TYPE
}
-}
-opList {
- name: "multi_head_dot_product_attention_bp"
argDescriptor {
name: "dLdq"
argType: OUTPUT_TENSOR
@@ -12496,17 +12652,13 @@ opList {
argType: INPUT_TENSOR
argIndex: 8
}
- argDescriptor {
- name: "normalization"
- argType: INT64
- }
- argDescriptor {
- name: "dtype"
- argType: DATA_TYPE
- }
}
opList {
name: "multiply"
+ argDescriptor {
+ name: "dataType"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -12524,14 +12676,14 @@ opList {
argType: INPUT_TENSOR
argIndex: 1
}
- argDescriptor {
- name: "dataType"
- argType: DATA_TYPE
- }
opDeclarationType: BROADCASTABLE_OP_IMPL
}
opList {
name: "multiply_bp"
+ argDescriptor {
+ name: "dtype"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "gradX"
argType: OUTPUT_TENSOR
@@ -12559,13 +12711,13 @@ opList {
argType: INPUT_TENSOR
argIndex: 2
}
- argDescriptor {
- name: "dtype"
- argType: DATA_TYPE
- }
}
opList {
name: "nadam_updater"
+ argDescriptor {
+ name: "iteration"
+ argType: INT64
+ }
argDescriptor {
name: "update"
argType: OUTPUT_TENSOR
@@ -12580,6 +12732,25 @@ opList {
argType: OUTPUT_TENSOR
argIndex: 2
}
+ argDescriptor {
+ name: "dLr"
+ argType: DOUBLE
+ }
+ argDescriptor {
+ name: "dBeta1"
+ argType: DOUBLE
+ argIndex: 1
+ }
+ argDescriptor {
+ name: "dBeta2"
+ argType: DOUBLE
+ argIndex: 2
+ }
+ argDescriptor {
+ name: "dEpsilon"
+ argType: DOUBLE
+ argIndex: 3
+ }
argDescriptor {
name: "gradient"
argType: INPUT_TENSOR
@@ -12614,33 +12785,14 @@ opList {
argType: INPUT_TENSOR
argIndex: 6
}
- argDescriptor {
- name: "iteration"
- argType: INT64
- }
- argDescriptor {
- name: "dLr"
- argType: DOUBLE
- }
- argDescriptor {
- name: "dBeta1"
- argType: DOUBLE
- argIndex: 1
- }
- argDescriptor {
- name: "dBeta2"
- argType: DOUBLE
- argIndex: 2
- }
- argDescriptor {
- name: "dEpsilon"
- argType: DOUBLE
- argIndex: 3
- }
opDeclarationType: CONFIGURABLE_OP_IMPL
}
opList {
name: "neg"
+ argDescriptor {
+ name: "dataType"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -12653,10 +12805,6 @@ opList {
name: "input"
argType: INPUT_TENSOR
}
- argDescriptor {
- name: "dataType"
- argType: DATA_TYPE
- }
opDeclarationType: LEGACY_XYZ
}
opList {
@@ -12670,6 +12818,15 @@ opList {
argType: OUTPUT_TENSOR
argIndex: 1
}
+ argDescriptor {
+ name: "dLr"
+ argType: DOUBLE
+ }
+ argDescriptor {
+ name: "dMomentum"
+ argType: DOUBLE
+ argIndex: 1
+ }
argDescriptor {
name: "gradient"
argType: INPUT_TENSOR
@@ -12689,15 +12846,6 @@ opList {
argType: INPUT_TENSOR
argIndex: 3
}
- argDescriptor {
- name: "dLr"
- argType: DOUBLE
- }
- argDescriptor {
- name: "dMomentum"
- argType: DOUBLE
- argIndex: 1
- }
opDeclarationType: CONFIGURABLE_OP_IMPL
}
opList {
@@ -12706,17 +12854,25 @@ opList {
name: "outputs"
argType: OUTPUT_TENSOR
}
- argDescriptor {
- name: "input"
- argType: INPUT_TENSOR
- }
argDescriptor {
name: "frameName"
argType: STRING
}
+ argDescriptor {
+ name: "input"
+ argType: INPUT_TENSOR
+ }
}
opList {
name: "non_max_suppression"
+ argDescriptor {
+ name: "maxOutputSize"
+ argType: INT64
+ }
+ argDescriptor {
+ name: "dtype"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -12736,7 +12892,7 @@ opList {
argIndex: 2
}
argDescriptor {
- name: "overlayThreshold"
+ name: "iouThreshold"
argType: INPUT_TENSOR
argIndex: 3
}
@@ -12745,6 +12901,9 @@ opList {
argType: INPUT_TENSOR
argIndex: 4
}
+}
+opList {
+ name: "non_max_suppression_overlaps"
argDescriptor {
name: "maxOutputSize"
argType: INT64
@@ -12754,7 +12913,11 @@ opList {
argType: DATA_TYPE
}
argDescriptor {
- name: "iouThreshold"
+ name: "outputs"
+ argType: OUTPUT_TENSOR
+ }
+ argDescriptor {
+ name: "overlapThreshold"
argType: DOUBLE
}
argDescriptor {
@@ -12762,13 +12925,6 @@ opList {
argType: DOUBLE
argIndex: 1
}
-}
-opList {
- name: "non_max_suppression_overlaps"
- argDescriptor {
- name: "outputs"
- argType: OUTPUT_TENSOR
- }
argDescriptor {
name: "boxes"
argType: INPUT_TENSOR
@@ -12778,6 +12934,24 @@ opList {
argType: INPUT_TENSOR
argIndex: 1
}
+ argDescriptor {
+ name: "maxOutSize"
+ argType: INPUT_TENSOR
+ argIndex: 2
+ }
+ argDescriptor {
+ name: "iouThreshold"
+ argType: INPUT_TENSOR
+ argIndex: 3
+ }
+ argDescriptor {
+ name: "scoreThreshold"
+ argType: INPUT_TENSOR
+ argIndex: 4
+ }
+}
+opList {
+ name: "non_max_suppression_v3"
argDescriptor {
name: "maxOutputSize"
argType: INT64
@@ -12786,18 +12960,6 @@ opList {
name: "dtype"
argType: DATA_TYPE
}
- argDescriptor {
- name: "overlapThreshold"
- argType: DOUBLE
- }
- argDescriptor {
- name: "scoreThreshold"
- argType: DOUBLE
- argIndex: 1
- }
-}
-opList {
- name: "non_max_suppression_v3"
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -12826,14 +12988,6 @@ opList {
argType: INPUT_TENSOR
argIndex: 4
}
- argDescriptor {
- name: "maxOutputSize"
- argType: INT64
- }
- argDescriptor {
- name: "dtype"
- argType: DATA_TYPE
- }
}
opList {
name: "noop"
@@ -12850,9 +13004,13 @@ opList {
opList {
name: "norm"
argDescriptor {
- name: "outputs"
+ name: "*output"
argType: OUTPUT_TENSOR
}
+ argDescriptor {
+ name: "mode"
+ argType: DOUBLE
+ }
argDescriptor {
name: "input"
argType: INPUT_TENSOR
@@ -12862,14 +13020,14 @@ opList {
argType: INPUT_TENSOR
argIndex: 1
}
- argDescriptor {
- name: "mode"
- argType: DOUBLE
- }
opDeclarationType: REDUCTION_OP_IMPL
}
opList {
name: "normalize_moments"
+ argDescriptor {
+ name: "dtype"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "resMeans"
argType: OUTPUT_TENSOR
@@ -12879,6 +13037,10 @@ opList {
argType: OUTPUT_TENSOR
argIndex: 1
}
+ argDescriptor {
+ name: "shift"
+ argType: DOUBLE
+ }
argDescriptor {
name: "counts"
argType: INPUT_TENSOR
@@ -12903,21 +13065,21 @@ opList {
argType: INPUT_TENSOR
argIndex: 4
}
- argDescriptor {
- name: "dtype"
- argType: DATA_TYPE
- }
- argDescriptor {
- name: "shift"
- argType: DOUBLE
- }
}
opList {
name: "not"
+ argDescriptor {
+ name: "dataType"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
}
+ argDescriptor {
+ name: "comparable"
+ argType: DOUBLE
+ }
argDescriptor {
name: "input"
argType: INPUT_TENSOR
@@ -12927,18 +13089,14 @@ opList {
argType: INPUT_TENSOR
argIndex: 1
}
- argDescriptor {
- name: "dataType"
- argType: DATA_TYPE
- }
- argDescriptor {
- name: "comparable"
- argType: DOUBLE
- }
opDeclarationType: LEGACY_XYZ
}
opList {
name: "not_equals"
+ argDescriptor {
+ name: "dataType"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -12956,10 +13114,6 @@ opList {
argType: INPUT_TENSOR
argIndex: 1
}
- argDescriptor {
- name: "dataType"
- argType: DATA_TYPE
- }
opDeclarationType: BROADCASTABLE_BOOL_OP_IMPL
}
opList {
@@ -12996,6 +13150,14 @@ opList {
}
opList {
name: "nth_element"
+ argDescriptor {
+ name: "reverse"
+ argType: INT64
+ }
+ argDescriptor {
+ name: "dtype"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -13013,17 +13175,13 @@ opList {
argType: INPUT_TENSOR
argIndex: 1
}
- argDescriptor {
- name: "reverse"
- argType: INT64
- }
- argDescriptor {
- name: "dtype"
- argType: DATA_TYPE
- }
}
opList {
name: "old_assign"
+ argDescriptor {
+ name: "dataType"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -13036,18 +13194,41 @@ opList {
name: "input"
argType: INPUT_TENSOR
}
- argDescriptor {
- name: "dataType"
- argType: DATA_TYPE
- }
opDeclarationType: LEGACY_XYZ
}
opList {
name: "onehot"
+ argDescriptor {
+ name: "dimensions"
+ argType: INT64
+ }
+ argDescriptor {
+ name: "depth"
+ argType: INT64
+ argIndex: 1
+ }
+ argDescriptor {
+ name: "dataType"
+ argType: INT64
+ argIndex: 2
+ }
+ argDescriptor {
+ name: "dtype"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
}
+ argDescriptor {
+ name: "on"
+ argType: DOUBLE
+ }
+ argDescriptor {
+ name: "off"
+ argType: DOUBLE
+ argIndex: 1
+ }
argDescriptor {
name: "input"
argType: INPUT_TENSOR
@@ -13067,36 +13248,13 @@ opList {
argType: INPUT_TENSOR
argIndex: 3
}
- argDescriptor {
- name: "dimensions"
- argType: INT64
- }
- argDescriptor {
- name: "depth"
- argType: INT64
- argIndex: 1
- }
- argDescriptor {
- name: "dataType"
- argType: INT64
- argIndex: 2
- }
- argDescriptor {
- name: "dtype"
- argType: DATA_TYPE
- }
- argDescriptor {
- name: "on"
- argType: DOUBLE
- }
- argDescriptor {
- name: "off"
- argType: DOUBLE
- argIndex: 1
- }
}
opList {
name: "oneminus"
+ argDescriptor {
+ name: "dataType"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -13109,22 +13267,10 @@ opList {
name: "input"
argType: INPUT_TENSOR
}
- argDescriptor {
- name: "dataType"
- argType: DATA_TYPE
- }
opDeclarationType: LEGACY_XYZ
}
opList {
name: "ones_as"
- argDescriptor {
- name: "outputs"
- argType: OUTPUT_TENSOR
- }
- argDescriptor {
- name: "input"
- argType: INPUT_TENSOR
- }
argDescriptor {
name: "dataType"
argType: INT64
@@ -13133,9 +13279,21 @@ opList {
name: "dtype"
argType: DATA_TYPE
}
+ argDescriptor {
+ name: "outputs"
+ argType: OUTPUT_TENSOR
+ }
+ argDescriptor {
+ name: "input"
+ argType: INPUT_TENSOR
+ }
}
opList {
name: "or"
+ argDescriptor {
+ name: "dataType"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -13144,23 +13302,14 @@ opList {
name: "inPlace"
argType: BOOL
}
- argDescriptor {
- name: "input"
- argType: INPUT_TENSOR
- }
- argDescriptor {
- name: "y"
- argType: INPUT_TENSOR
- argIndex: 1
- }
- argDescriptor {
- name: "dataType"
- argType: DATA_TYPE
- }
argDescriptor {
name: "comparable"
argType: DOUBLE
}
+ argDescriptor {
+ name: "input"
+ argType: INPUT_TENSOR
+ }
opDeclarationType: LEGACY_XYZ
}
opList {
@@ -13181,6 +13330,10 @@ opList {
}
opList {
name: "order"
+ argDescriptor {
+ name: "dtype"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -13189,17 +13342,25 @@ opList {
name: "input"
argType: INPUT_TENSOR
}
- argDescriptor {
- name: "dtype"
- argType: DATA_TYPE
- }
}
opList {
name: "pad"
+ argDescriptor {
+ name: "mode"
+ argType: INT64
+ }
+ argDescriptor {
+ name: "dtype"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
}
+ argDescriptor {
+ name: "padValue"
+ argType: DOUBLE
+ }
argDescriptor {
name: "input"
argType: INPUT_TENSOR
@@ -13209,21 +13370,13 @@ opList {
argType: INPUT_TENSOR
argIndex: 1
}
- argDescriptor {
- name: "mode"
- argType: INT64
- }
+}
+opList {
+ name: "parallel_stack"
argDescriptor {
name: "dtype"
argType: DATA_TYPE
}
- argDescriptor {
- name: "padValue"
- argType: DOUBLE
- }
-}
-opList {
- name: "parallel_stack"
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -13237,21 +13390,9 @@ opList {
argType: INPUT_TENSOR
argIndex: 1
}
- argDescriptor {
- name: "dtype"
- argType: DATA_TYPE
- }
}
opList {
name: "percentile"
- argDescriptor {
- name: "outputs"
- argType: OUTPUT_TENSOR
- }
- argDescriptor {
- name: "input"
- argType: INPUT_TENSOR
- }
argDescriptor {
name: "dimensions"
argType: INT64
@@ -13260,6 +13401,10 @@ opList {
name: "dtype"
argType: DATA_TYPE
}
+ argDescriptor {
+ name: "outputs"
+ argType: OUTPUT_TENSOR
+ }
argDescriptor {
name: "q"
argType: DOUBLE
@@ -13274,9 +13419,21 @@ opList {
argType: DOUBLE
argIndex: 2
}
+ argDescriptor {
+ name: "input"
+ argType: INPUT_TENSOR
+ }
}
opList {
name: "permute"
+ argDescriptor {
+ name: "reverseDims"
+ argType: INT64
+ }
+ argDescriptor {
+ name: "dtype"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -13286,18 +13443,10 @@ opList {
argType: INPUT_TENSOR
}
argDescriptor {
- name: "permuteDims"
+ name: "permutationVector"
argType: INPUT_TENSOR
argIndex: 1
}
- argDescriptor {
- name: "reverseDims"
- argType: INT64
- }
- argDescriptor {
- name: "dtype"
- argType: DATA_TYPE
- }
}
opList {
name: "pick_list"
@@ -13318,14 +13467,6 @@ opList {
}
opList {
name: "pnormpool2d"
- argDescriptor {
- name: "outputs"
- argType: OUTPUT_TENSOR
- }
- argDescriptor {
- name: "input"
- argType: INPUT_TENSOR
- }
argDescriptor {
name: "kY"
argType: INT64
@@ -13384,22 +13525,17 @@ opList {
name: "dtype"
argType: DATA_TYPE
}
-}
-opList {
- name: "pnormpool2d_bp"
argDescriptor {
- name: "gradI"
+ name: "outputs"
argType: OUTPUT_TENSOR
}
argDescriptor {
name: "input"
argType: INPUT_TENSOR
}
- argDescriptor {
- name: "gradO"
- argType: INPUT_TENSOR
- argIndex: 1
- }
+}
+opList {
+ name: "pnormpool2d_bp"
argDescriptor {
name: "kH"
argType: INT64
@@ -13458,13 +13594,39 @@ opList {
name: "dtype"
argType: DATA_TYPE
}
+ argDescriptor {
+ name: "gradI"
+ argType: OUTPUT_TENSOR
+ }
argDescriptor {
name: "eps"
argType: DOUBLE
}
+ argDescriptor {
+ name: "input"
+ argType: INPUT_TENSOR
+ }
+ argDescriptor {
+ name: "gradO"
+ argType: INPUT_TENSOR
+ argIndex: 1
+ }
}
opList {
name: "pointwise_conv2d"
+ argDescriptor {
+ name: "isNCHW"
+ argType: INT64
+ }
+ argDescriptor {
+ name: "wFormat"
+ argType: INT64
+ argIndex: 1
+ }
+ argDescriptor {
+ name: "dtype"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -13483,19 +13645,6 @@ opList {
argType: INPUT_TENSOR
argIndex: 2
}
- argDescriptor {
- name: "isNCHW"
- argType: INT64
- }
- argDescriptor {
- name: "wFormat"
- argType: INT64
- argIndex: 1
- }
- argDescriptor {
- name: "dtype"
- argType: DATA_TYPE
- }
}
opList {
name: "polygamma"
@@ -13544,6 +13693,10 @@ opList {
name: "inPlace"
argType: BOOL
}
+ argDescriptor {
+ name: "pow"
+ argType: DOUBLE
+ }
argDescriptor {
name: "input"
argType: INPUT_TENSOR
@@ -13553,14 +13706,14 @@ opList {
argType: INPUT_TENSOR
argIndex: 1
}
- argDescriptor {
- name: "pow"
- argType: DOUBLE
- }
opDeclarationType: LEGACY_XYZ
}
opList {
name: "pow_pairwise"
+ argDescriptor {
+ name: "dataType"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -13578,14 +13731,14 @@ opList {
argType: INPUT_TENSOR
argIndex: 1
}
- argDescriptor {
- name: "dataType"
- argType: DATA_TYPE
- }
opDeclarationType: LEGACY_XYZ
}
opList {
name: "precise_gelu"
+ argDescriptor {
+ name: "dataType"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -13603,14 +13756,14 @@ opList {
name: "input"
argType: INPUT_TENSOR
}
- argDescriptor {
- name: "dataType"
- argType: DATA_TYPE
- }
opDeclarationType: LEGACY_XYZ
}
opList {
name: "prelu"
+ argDescriptor {
+ name: "sharedAxes"
+ argType: INT64
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -13624,14 +13777,14 @@ opList {
argType: INPUT_TENSOR
argIndex: 1
}
- argDescriptor {
- name: "sharedAxes"
- argType: INT64
- }
opDeclarationType: CONFIGURABLE_OP_IMPL
}
opList {
name: "prelu_bp"
+ argDescriptor {
+ name: "sharedAxes"
+ argType: INT64
+ }
argDescriptor {
name: "dLdI"
argType: OUTPUT_TENSOR
@@ -13665,14 +13818,14 @@ opList {
argType: INPUT_TENSOR
argIndex: 4
}
- argDescriptor {
- name: "sharedAxes"
- argType: INT64
- }
opDeclarationType: CONFIGURABLE_OP_IMPL
}
opList {
name: "print_affinity"
+ argDescriptor {
+ name: "dtype"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -13681,13 +13834,13 @@ opList {
name: "input"
argType: INPUT_TENSOR
}
+}
+opList {
+ name: "print_variable"
argDescriptor {
name: "dtype"
argType: DATA_TYPE
}
-}
-opList {
- name: "print_variable"
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -13705,10 +13858,6 @@ opList {
argType: INPUT_TENSOR
argIndex: 1
}
- argDescriptor {
- name: "dtype"
- argType: DATA_TYPE
- }
}
opList {
name: "probablistic_merge"
@@ -13716,6 +13865,10 @@ opList {
name: "outputs"
argType: OUTPUT_TENSOR
}
+ argDescriptor {
+ name: "probability"
+ argType: DOUBLE
+ }
argDescriptor {
name: "input"
argType: INPUT_TENSOR
@@ -13725,14 +13878,14 @@ opList {
argType: INPUT_TENSOR
argIndex: 1
}
- argDescriptor {
- name: "probability"
- argType: DOUBLE
- }
opDeclarationType: LEGACY_XYZ
}
opList {
name: "qr"
+ argDescriptor {
+ name: "dtype"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputQ"
argType: OUTPUT_TENSOR
@@ -13750,32 +13903,36 @@ opList {
name: "input"
argType: INPUT_TENSOR
}
+}
+opList {
+ name: "random_bernoulli"
argDescriptor {
name: "dtype"
argType: DATA_TYPE
}
-}
-opList {
- name: "random_bernoulli"
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
}
+ argDescriptor {
+ name: "f"
+ argType: DOUBLE
+ }
argDescriptor {
name: "input"
argType: INPUT_TENSOR
}
+}
+opList {
+ name: "random_crop"
+ argDescriptor {
+ name: "seed"
+ argType: INT64
+ }
argDescriptor {
name: "dtype"
argType: DATA_TYPE
}
- argDescriptor {
- name: "f"
- argType: DOUBLE
- }
-}
-opList {
- name: "random_crop"
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -13789,21 +13946,9 @@ opList {
argType: INPUT_TENSOR
argIndex: 1
}
- argDescriptor {
- name: "seed"
- argType: INT64
- }
- argDescriptor {
- name: "dtype"
- argType: DATA_TYPE
- }
}
opList {
name: "random_exponential"
- argDescriptor {
- name: "outputs"
- argType: OUTPUT_TENSOR
- }
argDescriptor {
name: "shape"
argType: INT64
@@ -13812,6 +13957,10 @@ opList {
name: "dtype"
argType: DATA_TYPE
}
+ argDescriptor {
+ name: "outputs"
+ argType: OUTPUT_TENSOR
+ }
argDescriptor {
name: "lambda"
argType: DOUBLE
@@ -13819,6 +13968,14 @@ opList {
}
opList {
name: "random_gamma"
+ argDescriptor {
+ name: "seed"
+ argType: INT64
+ }
+ argDescriptor {
+ name: "dtype"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -13837,17 +13994,17 @@ opList {
argType: INPUT_TENSOR
argIndex: 2
}
+}
+opList {
+ name: "random_multinomial"
argDescriptor {
- name: "seed"
+ name: "dimC"
argType: INT64
}
argDescriptor {
name: "dtype"
argType: DATA_TYPE
}
-}
-opList {
- name: "random_multinomial"
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -13861,17 +14018,13 @@ opList {
argType: INPUT_TENSOR
argIndex: 1
}
- argDescriptor {
- name: "dimC"
- argType: INT64
- }
+}
+opList {
+ name: "random_normal"
argDescriptor {
name: "dtype"
argType: DATA_TYPE
}
-}
-opList {
- name: "random_normal"
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -13880,13 +14033,17 @@ opList {
name: "input"
argType: INPUT_TENSOR
}
+}
+opList {
+ name: "random_poisson"
+ argDescriptor {
+ name: "seed"
+ argType: INT64
+ }
argDescriptor {
name: "dtype"
argType: DATA_TYPE
}
-}
-opList {
- name: "random_poisson"
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -13900,17 +14057,13 @@ opList {
argType: INPUT_TENSOR
argIndex: 1
}
- argDescriptor {
- name: "seed"
- argType: INT64
- }
- argDescriptor {
- name: "dtype"
- argType: DATA_TYPE
- }
}
opList {
name: "random_shuffle"
+ argDescriptor {
+ name: "seeds"
+ argType: INT64
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -13919,10 +14072,6 @@ opList {
name: "input"
argType: INPUT_TENSOR
}
- argDescriptor {
- name: "seeds"
- argType: INT64
- }
opDeclarationType: OP_IMPL
}
opList {
@@ -13931,10 +14080,6 @@ opList {
name: "outputs"
argType: OUTPUT_TENSOR
}
- argDescriptor {
- name: "shape"
- argType: INPUT_TENSOR
- }
argDescriptor {
name: "mean"
argType: DOUBLE
@@ -13944,27 +14089,13 @@ opList {
argType: DOUBLE
argIndex: 1
}
-}
-opList {
- name: "randomuniform"
- argDescriptor {
- name: "outputs"
- argType: OUTPUT_TENSOR
- }
argDescriptor {
name: "shape"
argType: INPUT_TENSOR
}
- argDescriptor {
- name: "min"
- argType: INPUT_TENSOR
- argIndex: 1
- }
- argDescriptor {
- name: "max"
- argType: INPUT_TENSOR
- argIndex: 2
- }
+}
+opList {
+ name: "randomuniform"
argDescriptor {
name: "dtype"
argType: INT64
@@ -13978,6 +14109,10 @@ opList {
name: "dataType"
argType: DATA_TYPE
}
+ argDescriptor {
+ name: "outputs"
+ argType: OUTPUT_TENSOR
+ }
argDescriptor {
name: "min"
argType: DOUBLE
@@ -13987,27 +14122,23 @@ opList {
argType: DOUBLE
argIndex: 1
}
-}
-opList {
- name: "range"
argDescriptor {
- name: "outputs"
- argType: OUTPUT_TENSOR
- }
- argDescriptor {
- name: "from"
+ name: "shape"
argType: INPUT_TENSOR
}
argDescriptor {
- name: "to"
+ name: "min"
argType: INPUT_TENSOR
argIndex: 1
}
argDescriptor {
- name: "step"
+ name: "max"
argType: INPUT_TENSOR
argIndex: 2
}
+}
+opList {
+ name: "range"
argDescriptor {
name: "from"
argType: INT64
@@ -14026,6 +14157,10 @@ opList {
name: "dtype"
argType: DATA_TYPE
}
+ argDescriptor {
+ name: "outputs"
+ argType: OUTPUT_TENSOR
+ }
argDescriptor {
name: "from"
argType: DOUBLE
@@ -14040,9 +14175,27 @@ opList {
argType: DOUBLE
argIndex: 2
}
+ argDescriptor {
+ name: "from"
+ argType: INPUT_TENSOR
+ }
+ argDescriptor {
+ name: "to"
+ argType: INPUT_TENSOR
+ argIndex: 1
+ }
+ argDescriptor {
+ name: "step"
+ argType: INPUT_TENSOR
+ argIndex: 2
+ }
}
opList {
name: "rank"
+ argDescriptor {
+ name: "dtype"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -14055,13 +14208,13 @@ opList {
name: "input"
argType: INPUT_TENSOR
}
- argDescriptor {
- name: "dtype"
- argType: DATA_TYPE
- }
}
opList {
name: "rational_tanh"
+ argDescriptor {
+ name: "dataType"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -14074,14 +14227,14 @@ opList {
name: "input"
argType: INPUT_TENSOR
}
- argDescriptor {
- name: "dataType"
- argType: DATA_TYPE
- }
opDeclarationType: LEGACY_XYZ
}
opList {
name: "rational_tanh_derivative"
+ argDescriptor {
+ name: "dataType"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -14094,10 +14247,6 @@ opList {
name: "input"
argType: INPUT_TENSOR
}
- argDescriptor {
- name: "dataType"
- argType: DATA_TYPE
- }
opDeclarationType: LEGACY_XYZ
}
opList {
@@ -14135,18 +14284,27 @@ opList {
name: "outputs"
argType: OUTPUT_TENSOR
}
- argDescriptor {
- name: "inPlace"
- argType: BOOL
- }
argDescriptor {
name: "input"
argType: INPUT_TENSOR
}
+ argDescriptor {
+ name: "y"
+ argType: INPUT_TENSOR
+ argIndex: 1
+ }
opDeclarationType: LEGACY_XYZ
}
opList {
name: "read_list"
+ argDescriptor {
+ name: "index"
+ argType: INT64
+ }
+ argDescriptor {
+ name: "importDataType"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -14160,18 +14318,14 @@ opList {
argType: INPUT_TENSOR
argIndex: 1
}
- argDescriptor {
- name: "index"
- argType: INT64
- }
- argDescriptor {
- name: "importDataType"
- argType: DATA_TYPE
- }
opDeclarationType: LIST_OP_IMPL
}
opList {
name: "realdiv"
+ argDescriptor {
+ name: "dataType"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -14189,14 +14343,14 @@ opList {
argType: INPUT_TENSOR
argIndex: 1
}
- argDescriptor {
- name: "dataType"
- argType: DATA_TYPE
- }
opDeclarationType: BROADCASTABLE_OP_IMPL
}
opList {
name: "realdiv_bp"
+ argDescriptor {
+ name: "dtype"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "gradX"
argType: OUTPUT_TENSOR
@@ -14220,13 +14374,13 @@ opList {
argType: INPUT_TENSOR
argIndex: 2
}
- argDescriptor {
- name: "dtype"
- argType: DATA_TYPE
- }
}
opList {
name: "rectified_tanh"
+ argDescriptor {
+ name: "dataType"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -14239,14 +14393,14 @@ opList {
name: "input"
argType: INPUT_TENSOR
}
- argDescriptor {
- name: "dataType"
- argType: DATA_TYPE
- }
opDeclarationType: LEGACY_XYZ
}
opList {
name: "rectified_tanh_derivative"
+ argDescriptor {
+ name: "dataType"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -14259,10 +14413,6 @@ opList {
name: "input"
argType: INPUT_TENSOR
}
- argDescriptor {
- name: "dataType"
- argType: DATA_TYPE
- }
opDeclarationType: LEGACY_XYZ
}
opList {
@@ -14296,6 +14446,14 @@ opList {
}
opList {
name: "reduce_dot_bp"
+ argDescriptor {
+ name: "dimensions"
+ argType: INT64
+ }
+ argDescriptor {
+ name: "dtype"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "gradX"
argType: OUTPUT_TENSOR
@@ -14333,34 +14491,9 @@ opList {
argType: INPUT_TENSOR
argIndex: 4
}
- argDescriptor {
- name: "dimensions"
- argType: INT64
- }
- argDescriptor {
- name: "dtype"
- argType: DATA_TYPE
- }
}
opList {
name: "reduce_logsumexp"
- argDescriptor {
- name: "outputs"
- argType: OUTPUT_TENSOR
- }
- argDescriptor {
- name: "keepDim"
- argType: BOOL
- }
- argDescriptor {
- name: "input"
- argType: INPUT_TENSOR
- }
- argDescriptor {
- name: "dimensions"
- argType: INPUT_TENSOR
- argIndex: 1
- }
argDescriptor {
name: "dimensions"
argType: INT64
@@ -14369,13 +14502,38 @@ opList {
name: "dtype"
argType: DATA_TYPE
}
+ argDescriptor {
+ name: "outputs"
+ argType: OUTPUT_TENSOR
+ }
argDescriptor {
name: "keepDims"
argType: DOUBLE
}
+ argDescriptor {
+ name: "keepDims"
+ argType: BOOL
+ }
+ argDescriptor {
+ name: "input"
+ argType: INPUT_TENSOR
+ }
+ argDescriptor {
+ name: "dimensions"
+ argType: INPUT_TENSOR
+ argIndex: 1
+ }
}
opList {
name: "reduce_max"
+ argDescriptor {
+ name: "dimensions"
+ argType: INT64
+ }
+ argDescriptor {
+ name: "dtype"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -14393,17 +14551,17 @@ opList {
argType: INPUT_TENSOR
argIndex: 1
}
- argDescriptor {
- name: "dimensions"
- argType: INT64
- }
- argDescriptor {
- name: "dtype"
- argType: DATA_TYPE
- }
}
opList {
name: "reduce_max_bp"
+ argDescriptor {
+ name: "dimensions"
+ argType: INT64
+ }
+ argDescriptor {
+ name: "dtype"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "gradI"
argType: OUTPUT_TENSOR
@@ -14426,17 +14584,17 @@ opList {
argType: INPUT_TENSOR
argIndex: 2
}
- argDescriptor {
- name: "dimensions"
- argType: INT64
- }
- argDescriptor {
- name: "dtype"
- argType: DATA_TYPE
- }
}
opList {
name: "reduce_mean"
+ argDescriptor {
+ name: "dimensions"
+ argType: INT64
+ }
+ argDescriptor {
+ name: "dtype"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -14454,17 +14612,17 @@ opList {
argType: INPUT_TENSOR
argIndex: 1
}
- argDescriptor {
- name: "dimensions"
- argType: INT64
- }
- argDescriptor {
- name: "dtype"
- argType: DATA_TYPE
- }
}
opList {
name: "reduce_mean_bp"
+ argDescriptor {
+ name: "dimensions"
+ argType: INT64
+ }
+ argDescriptor {
+ name: "dtype"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "gradI"
argType: OUTPUT_TENSOR
@@ -14487,17 +14645,17 @@ opList {
argType: INPUT_TENSOR
argIndex: 2
}
- argDescriptor {
- name: "dimensions"
- argType: INT64
- }
- argDescriptor {
- name: "dtype"
- argType: DATA_TYPE
- }
}
opList {
name: "reduce_min"
+ argDescriptor {
+ name: "dimensions"
+ argType: INT64
+ }
+ argDescriptor {
+ name: "dtype"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -14515,17 +14673,17 @@ opList {
argType: INPUT_TENSOR
argIndex: 1
}
- argDescriptor {
- name: "dimensions"
- argType: INT64
- }
- argDescriptor {
- name: "dtype"
- argType: DATA_TYPE
- }
}
opList {
name: "reduce_min_bp"
+ argDescriptor {
+ name: "dimensions"
+ argType: INT64
+ }
+ argDescriptor {
+ name: "dtype"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "gradI"
argType: OUTPUT_TENSOR
@@ -14548,17 +14706,17 @@ opList {
argType: INPUT_TENSOR
argIndex: 2
}
- argDescriptor {
- name: "dimensions"
- argType: INT64
- }
- argDescriptor {
- name: "dtype"
- argType: DATA_TYPE
- }
}
opList {
name: "reduce_norm1"
+ argDescriptor {
+ name: "dimensions"
+ argType: INT64
+ }
+ argDescriptor {
+ name: "dtype"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -14576,17 +14734,17 @@ opList {
argType: INPUT_TENSOR
argIndex: 1
}
- argDescriptor {
- name: "dimensions"
- argType: INT64
- }
- argDescriptor {
- name: "dtype"
- argType: DATA_TYPE
- }
}
opList {
name: "reduce_norm1_bp"
+ argDescriptor {
+ name: "dimensions"
+ argType: INT64
+ }
+ argDescriptor {
+ name: "dtype"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "gradI"
argType: OUTPUT_TENSOR
@@ -14609,17 +14767,17 @@ opList {
argType: INPUT_TENSOR
argIndex: 2
}
- argDescriptor {
- name: "dimensions"
- argType: INT64
- }
- argDescriptor {
- name: "dtype"
- argType: DATA_TYPE
- }
}
opList {
name: "reduce_norm2"
+ argDescriptor {
+ name: "dimensions"
+ argType: INT64
+ }
+ argDescriptor {
+ name: "dtype"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -14637,17 +14795,17 @@ opList {
argType: INPUT_TENSOR
argIndex: 1
}
- argDescriptor {
- name: "dimensions"
- argType: INT64
- }
- argDescriptor {
- name: "dtype"
- argType: DATA_TYPE
- }
}
opList {
name: "reduce_norm2_bp"
+ argDescriptor {
+ name: "dimensions"
+ argType: INT64
+ }
+ argDescriptor {
+ name: "dtype"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "gradI"
argType: OUTPUT_TENSOR
@@ -14670,17 +14828,13 @@ opList {
argType: INPUT_TENSOR
argIndex: 2
}
- argDescriptor {
- name: "dimensions"
- argType: INT64
- }
- argDescriptor {
- name: "dtype"
- argType: DATA_TYPE
- }
}
opList {
name: "reduce_norm_max"
+ argDescriptor {
+ name: "dtype"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -14698,13 +14852,17 @@ opList {
argType: INPUT_TENSOR
argIndex: 1
}
+}
+opList {
+ name: "reduce_norm_max_bp"
+ argDescriptor {
+ name: "dimensions"
+ argType: INT64
+ }
argDescriptor {
name: "dtype"
argType: DATA_TYPE
}
-}
-opList {
- name: "reduce_norm_max_bp"
argDescriptor {
name: "gradI"
argType: OUTPUT_TENSOR
@@ -14727,17 +14885,13 @@ opList {
argType: INPUT_TENSOR
argIndex: 2
}
+}
+opList {
+ name: "reduce_normmax"
argDescriptor {
name: "dimensions"
argType: INT64
}
- argDescriptor {
- name: "dtype"
- argType: DATA_TYPE
- }
-}
-opList {
- name: "reduce_normmax"
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -14755,14 +14909,18 @@ opList {
argType: INPUT_TENSOR
argIndex: 1
}
- argDescriptor {
- name: "dimensions"
- argType: INT64
- }
opDeclarationType: LEGACY_XYZ
}
opList {
name: "reduce_prod"
+ argDescriptor {
+ name: "dimensions"
+ argType: INT64
+ }
+ argDescriptor {
+ name: "dtype"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -14780,17 +14938,17 @@ opList {
argType: INPUT_TENSOR
argIndex: 1
}
- argDescriptor {
- name: "dimensions"
- argType: INT64
- }
- argDescriptor {
- name: "dtype"
- argType: DATA_TYPE
- }
}
opList {
name: "reduce_prod_bp"
+ argDescriptor {
+ name: "dimensions"
+ argType: INT64
+ }
+ argDescriptor {
+ name: "dtype"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "gradI"
argType: OUTPUT_TENSOR
@@ -14813,17 +14971,17 @@ opList {
argType: INPUT_TENSOR
argIndex: 2
}
- argDescriptor {
- name: "dimensions"
- argType: INT64
- }
- argDescriptor {
- name: "dtype"
- argType: DATA_TYPE
- }
}
opList {
name: "reduce_sqnorm"
+ argDescriptor {
+ name: "dimensions"
+ argType: INT64
+ }
+ argDescriptor {
+ name: "dtype"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "gradI"
argType: OUTPUT_TENSOR
@@ -14841,6 +14999,9 @@ opList {
argType: INPUT_TENSOR
argIndex: 1
}
+}
+opList {
+ name: "reduce_sqnorm_bp"
argDescriptor {
name: "dimensions"
argType: INT64
@@ -14849,9 +15010,6 @@ opList {
name: "dtype"
argType: DATA_TYPE
}
-}
-opList {
- name: "reduce_sqnorm_bp"
argDescriptor {
name: "gradI"
argType: OUTPUT_TENSOR
@@ -14874,34 +15032,17 @@ opList {
argType: INPUT_TENSOR
argIndex: 2
}
- argDescriptor {
- name: "dimensions"
- argType: INT64
- }
- argDescriptor {
- name: "dtype"
- argType: DATA_TYPE
- }
}
opList {
name: "reduce_stdev"
- argDescriptor {
- name: "outputs"
- argType: OUTPUT_TENSOR
- }
- argDescriptor {
- name: "input"
- argType: INPUT_TENSOR
- }
- argDescriptor {
- name: "dimensions"
- argType: INPUT_TENSOR
- argIndex: 1
- }
argDescriptor {
name: "dtype"
argType: DATA_TYPE
}
+ argDescriptor {
+ name: "outputs"
+ argType: OUTPUT_TENSOR
+ }
argDescriptor {
name: "keepDims"
argType: DOUBLE
@@ -14911,9 +15052,26 @@ opList {
argType: DOUBLE
argIndex: 1
}
+ argDescriptor {
+ name: "input"
+ argType: INPUT_TENSOR
+ }
+ argDescriptor {
+ name: "dimensions"
+ argType: INPUT_TENSOR
+ argIndex: 1
+ }
}
opList {
name: "reduce_stdev_bp"
+ argDescriptor {
+ name: "dimensions"
+ argType: INT64
+ }
+ argDescriptor {
+ name: "dtype"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "gradI"
argType: OUTPUT_TENSOR
@@ -14927,6 +15085,15 @@ opList {
argType: BOOL
argIndex: 1
}
+ argDescriptor {
+ name: "keepDims"
+ argType: DOUBLE
+ }
+ argDescriptor {
+ name: "biasCorrected"
+ argType: DOUBLE
+ argIndex: 1
+ }
argDescriptor {
name: "input"
argType: INPUT_TENSOR
@@ -14941,26 +15108,17 @@ opList {
argType: INPUT_TENSOR
argIndex: 2
}
- argDescriptor {
- name: "dimensions"
- argType: INT64
- }
- argDescriptor {
- name: "dtype"
- argType: DATA_TYPE
- }
- argDescriptor {
- name: "keepDims"
- argType: DOUBLE
- }
- argDescriptor {
- name: "biasCorrected"
- argType: DOUBLE
- argIndex: 1
- }
}
opList {
name: "reduce_sum"
+ argDescriptor {
+ name: "dimensions"
+ argType: INT64
+ }
+ argDescriptor {
+ name: "dtype"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -14978,6 +15136,9 @@ opList {
argType: INPUT_TENSOR
argIndex: 1
}
+}
+opList {
+ name: "reduce_sum_bp"
argDescriptor {
name: "dimensions"
argType: INT64
@@ -14986,9 +15147,6 @@ opList {
name: "dtype"
argType: DATA_TYPE
}
-}
-opList {
- name: "reduce_sum_bp"
argDescriptor {
name: "gradI"
argType: OUTPUT_TENSOR
@@ -15011,34 +15169,17 @@ opList {
argType: INPUT_TENSOR
argIndex: 2
}
- argDescriptor {
- name: "dimensions"
- argType: INT64
- }
- argDescriptor {
- name: "dtype"
- argType: DATA_TYPE
- }
}
opList {
name: "reduce_variance"
- argDescriptor {
- name: "outputs"
- argType: OUTPUT_TENSOR
- }
- argDescriptor {
- name: "input"
- argType: INPUT_TENSOR
- }
- argDescriptor {
- name: "dimensions"
- argType: INPUT_TENSOR
- argIndex: 1
- }
argDescriptor {
name: "dtype"
argType: DATA_TYPE
}
+ argDescriptor {
+ name: "outputs"
+ argType: OUTPUT_TENSOR
+ }
argDescriptor {
name: "keepDims"
argType: DOUBLE
@@ -15048,9 +15189,26 @@ opList {
argType: DOUBLE
argIndex: 1
}
+ argDescriptor {
+ name: "input"
+ argType: INPUT_TENSOR
+ }
+ argDescriptor {
+ name: "dimensions"
+ argType: INPUT_TENSOR
+ argIndex: 1
+ }
}
opList {
name: "reduce_variance_bp"
+ argDescriptor {
+ name: "dimensions"
+ argType: INT64
+ }
+ argDescriptor {
+ name: "dtype"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "gradI"
argType: OUTPUT_TENSOR
@@ -15064,6 +15222,15 @@ opList {
argType: BOOL
argIndex: 1
}
+ argDescriptor {
+ name: "keepDims"
+ argType: DOUBLE
+ }
+ argDescriptor {
+ name: "biasCorrected"
+ argType: DOUBLE
+ argIndex: 1
+ }
argDescriptor {
name: "input"
argType: INPUT_TENSOR
@@ -15078,23 +15245,6 @@ opList {
argType: INPUT_TENSOR
argIndex: 2
}
- argDescriptor {
- name: "dimensions"
- argType: INT64
- }
- argDescriptor {
- name: "dtype"
- argType: DATA_TYPE
- }
- argDescriptor {
- name: "keepDims"
- argType: DOUBLE
- }
- argDescriptor {
- name: "biasCorrected"
- argType: DOUBLE
- argIndex: 1
- }
}
opList {
name: "relu"
@@ -15106,18 +15256,22 @@ opList {
name: "inPlace"
argType: BOOL
}
- argDescriptor {
- name: "input"
- argType: INPUT_TENSOR
- }
argDescriptor {
name: "cutoff"
argType: DOUBLE
}
+ argDescriptor {
+ name: "input"
+ argType: INPUT_TENSOR
+ }
opDeclarationType: CONFIGURABLE_OP_IMPL
}
opList {
name: "relu6"
+ argDescriptor {
+ name: "dtype"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -15126,18 +15280,14 @@ opList {
name: "inPlace"
argType: BOOL
}
- argDescriptor {
- name: "input"
- argType: INPUT_TENSOR
- }
- argDescriptor {
- name: "dtype"
- argType: DATA_TYPE
- }
argDescriptor {
name: "cutoff"
argType: DOUBLE
}
+ argDescriptor {
+ name: "input"
+ argType: INPUT_TENSOR
+ }
opDeclarationType: CONFIGURABLE_OP_IMPL
}
opList {
@@ -15146,6 +15296,10 @@ opList {
name: "gradI"
argType: OUTPUT_TENSOR
}
+ argDescriptor {
+ name: "cutoff"
+ argType: DOUBLE
+ }
argDescriptor {
name: "input"
argType: INPUT_TENSOR
@@ -15155,10 +15309,6 @@ opList {
argType: INPUT_TENSOR
argIndex: 1
}
- argDescriptor {
- name: "cutoff"
- argType: DOUBLE
- }
opDeclarationType: CONFIGURABLE_OP_IMPL
}
opList {
@@ -15167,6 +15317,10 @@ opList {
name: "outputs"
argType: OUTPUT_TENSOR
}
+ argDescriptor {
+ name: "scalar"
+ argType: DOUBLE
+ }
argDescriptor {
name: "input"
argType: INPUT_TENSOR
@@ -15176,14 +15330,14 @@ opList {
argType: INPUT_TENSOR
argIndex: 1
}
- argDescriptor {
- name: "scalar"
- argType: DOUBLE
- }
opDeclarationType: CONFIGURABLE_OP_IMPL
}
opList {
name: "relu_layer"
+ argDescriptor {
+ name: "dtype"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -15202,13 +15356,13 @@ opList {
argType: INPUT_TENSOR
argIndex: 2
}
- argDescriptor {
- name: "dtype"
- argType: DATA_TYPE
- }
}
opList {
name: "remainder"
+ argDescriptor {
+ name: "dataType"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -15226,10 +15380,6 @@ opList {
argType: INPUT_TENSOR
argIndex: 1
}
- argDescriptor {
- name: "dataType"
- argType: DATA_TYPE
- }
opDeclarationType: LEGACY_XYZ
}
opList {
@@ -15250,14 +15400,6 @@ opList {
}
opList {
name: "repeat"
- argDescriptor {
- name: "outputs"
- argType: OUTPUT_TENSOR
- }
- argDescriptor {
- name: "input"
- argType: INPUT_TENSOR
- }
argDescriptor {
name: "dimensions"
argType: INT64
@@ -15266,6 +15408,14 @@ opList {
name: "dtype"
argType: DATA_TYPE
}
+ argDescriptor {
+ name: "outputs"
+ argType: OUTPUT_TENSOR
+ }
+ argDescriptor {
+ name: "input"
+ argType: INPUT_TENSOR
+ }
}
opList {
name: "replace_nans"
@@ -15274,21 +15424,25 @@ opList {
argType: OUTPUT_TENSOR
}
argDescriptor {
- name: "inPlace"
- argType: BOOL
+ name: "set"
+ argType: DOUBLE
}
argDescriptor {
name: "input"
argType: INPUT_TENSOR
}
- argDescriptor {
- name: "set"
- argType: DOUBLE
- }
opDeclarationType: LEGACY_XYZ
}
opList {
name: "reshape"
+ argDescriptor {
+ name: "shapeArr"
+ argType: INT64
+ }
+ argDescriptor {
+ name: "dtype"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -15302,17 +15456,13 @@ opList {
argType: INPUT_TENSOR
argIndex: 1
}
- argDescriptor {
- name: "shapeArr"
- argType: INT64
- }
+}
+opList {
+ name: "reshapeas"
argDescriptor {
name: "dtype"
argType: DATA_TYPE
}
-}
-opList {
- name: "reshapeas"
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -15326,13 +15476,22 @@ opList {
argType: INPUT_TENSOR
argIndex: 1
}
+}
+opList {
+ name: "resize_area"
+ argDescriptor {
+ name: "height"
+ argType: INT64
+ }
+ argDescriptor {
+ name: "width"
+ argType: INT64
+ argIndex: 1
+ }
argDescriptor {
name: "dtype"
argType: DATA_TYPE
}
-}
-opList {
- name: "resize_area"
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -15350,22 +15509,13 @@ opList {
argType: INPUT_TENSOR
argIndex: 1
}
- argDescriptor {
- name: "height"
- argType: INT64
- }
- argDescriptor {
- name: "width"
- argType: INT64
- argIndex: 1
- }
+}
+opList {
+ name: "resize_bicubic"
argDescriptor {
name: "dtype"
argType: DATA_TYPE
}
-}
-opList {
- name: "resize_bicubic"
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -15388,35 +15538,9 @@ opList {
argType: INPUT_TENSOR
argIndex: 1
}
- argDescriptor {
- name: "dtype"
- argType: DATA_TYPE
- }
}
opList {
name: "resize_bilinear"
- argDescriptor {
- name: "outputs"
- argType: OUTPUT_TENSOR
- }
- argDescriptor {
- name: "alignCorners"
- argType: BOOL
- }
- argDescriptor {
- name: "halfPixelCenters"
- argType: BOOL
- argIndex: 1
- }
- argDescriptor {
- name: "image"
- argType: INPUT_TENSOR
- }
- argDescriptor {
- name: "newImageSize"
- argType: INPUT_TENSOR
- argIndex: 1
- }
argDescriptor {
name: "height"
argType: INT64
@@ -15430,9 +15554,35 @@ opList {
name: "dtype"
argType: DATA_TYPE
}
+ argDescriptor {
+ name: "outputs"
+ argType: OUTPUT_TENSOR
+ }
+ argDescriptor {
+ name: "alignCorners"
+ argType: BOOL
+ }
+ argDescriptor {
+ name: "halfPixelCenter"
+ argType: BOOL
+ argIndex: 1
+ }
+ argDescriptor {
+ name: "image"
+ argType: INPUT_TENSOR
+ }
+ argDescriptor {
+ name: "newImageSize"
+ argType: INPUT_TENSOR
+ argIndex: 1
+ }
}
opList {
name: "resize_images"
+ argDescriptor {
+ name: "dtype"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -15460,13 +15610,22 @@ opList {
argType: INPUT_TENSOR
argIndex: 2
}
+}
+opList {
+ name: "resize_nearest_neighbor"
+ argDescriptor {
+ name: "height"
+ argType: INT64
+ }
+ argDescriptor {
+ name: "width"
+ argType: INT64
+ argIndex: 1
+ }
argDescriptor {
name: "dtype"
argType: DATA_TYPE
}
-}
-opList {
- name: "resize_nearest_neighbor"
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -15489,19 +15648,6 @@ opList {
argType: INPUT_TENSOR
argIndex: 1
}
- argDescriptor {
- name: "height"
- argType: INT64
- }
- argDescriptor {
- name: "width"
- argType: INT64
- argIndex: 1
- }
- argDescriptor {
- name: "dtype"
- argType: DATA_TYPE
- }
}
opList {
name: "restorev2"
@@ -15512,6 +15658,10 @@ opList {
}
opList {
name: "reverse"
+ argDescriptor {
+ name: "dimensions"
+ argType: INT64
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -15525,14 +15675,18 @@ opList {
argType: INPUT_TENSOR
argIndex: 1
}
- argDescriptor {
- name: "dimensions"
- argType: INT64
- }
opDeclarationType: CONFIGURABLE_OP_IMPL
}
opList {
name: "reverse_bp"
+ argDescriptor {
+ name: "dimensions"
+ argType: INT64
+ }
+ argDescriptor {
+ name: "dtype"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -15551,30 +15705,9 @@ opList {
argType: INPUT_TENSOR
argIndex: 2
}
- argDescriptor {
- name: "dimensions"
- argType: INT64
- }
- argDescriptor {
- name: "dtype"
- argType: DATA_TYPE
- }
}
opList {
name: "reverse_sequence"
- argDescriptor {
- name: "outputs"
- argType: OUTPUT_TENSOR
- }
- argDescriptor {
- name: "input"
- argType: INPUT_TENSOR
- }
- argDescriptor {
- name: "seqLengths"
- argType: INPUT_TENSOR
- argIndex: 1
- }
argDescriptor {
name: "seqDim"
argType: INT64
@@ -15588,6 +15721,19 @@ opList {
name: "dtype"
argType: DATA_TYPE
}
+ argDescriptor {
+ name: "outputs"
+ argType: OUTPUT_TENSOR
+ }
+ argDescriptor {
+ name: "input"
+ argType: INPUT_TENSOR
+ }
+ argDescriptor {
+ name: "seqLengths"
+ argType: INPUT_TENSOR
+ argIndex: 1
+ }
}
opList {
name: "reverse_v2"
@@ -15602,6 +15748,10 @@ opList {
}
opList {
name: "reversedivide"
+ argDescriptor {
+ name: "dataType"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -15619,14 +15769,14 @@ opList {
argType: INPUT_TENSOR
argIndex: 1
}
- argDescriptor {
- name: "dataType"
- argType: DATA_TYPE
- }
opDeclarationType: BROADCASTABLE_OP_IMPL
}
opList {
name: "reversedivide_bp"
+ argDescriptor {
+ name: "dtype"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "gradX"
argType: OUTPUT_TENSOR
@@ -15654,10 +15804,6 @@ opList {
argType: INPUT_TENSOR
argIndex: 2
}
- argDescriptor {
- name: "dtype"
- argType: DATA_TYPE
- }
}
opList {
name: "reversemod"
@@ -15678,6 +15824,10 @@ opList {
}
opList {
name: "reversemod_bp"
+ argDescriptor {
+ name: "dtype"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "gradX"
argType: OUTPUT_TENSOR
@@ -15701,13 +15851,13 @@ opList {
argType: INPUT_TENSOR
argIndex: 2
}
- argDescriptor {
- name: "dtype"
- argType: DATA_TYPE
- }
}
opList {
name: "reversesubtract"
+ argDescriptor {
+ name: "dataType"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -15725,14 +15875,14 @@ opList {
argType: INPUT_TENSOR
argIndex: 1
}
- argDescriptor {
- name: "dataType"
- argType: DATA_TYPE
- }
opDeclarationType: BROADCASTABLE_OP_IMPL
}
opList {
name: "reversesubtract_bp"
+ argDescriptor {
+ name: "dtype"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "gradX"
argType: OUTPUT_TENSOR
@@ -15760,21 +15910,9 @@ opList {
argType: INPUT_TENSOR
argIndex: 2
}
- argDescriptor {
- name: "dtype"
- argType: DATA_TYPE
- }
}
opList {
name: "rgb_to_grs"
- argDescriptor {
- name: "outputs"
- argType: OUTPUT_TENSOR
- }
- argDescriptor {
- name: "input"
- argType: INPUT_TENSOR
- }
argDescriptor {
name: "dimC"
argType: INT64
@@ -15783,9 +15921,21 @@ opList {
name: "dtype"
argType: DATA_TYPE
}
+ argDescriptor {
+ name: "outputs"
+ argType: OUTPUT_TENSOR
+ }
+ argDescriptor {
+ name: "input"
+ argType: INPUT_TENSOR
+ }
}
opList {
name: "rgb_to_hsv"
+ argDescriptor {
+ name: "dimC"
+ argType: INT64
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -15794,14 +15944,14 @@ opList {
name: "input"
argType: INPUT_TENSOR
}
- argDescriptor {
- name: "dimC"
- argType: INT64
- }
opDeclarationType: CONFIGURABLE_OP_IMPL
}
opList {
name: "rgb_to_yiq"
+ argDescriptor {
+ name: "dimC"
+ argType: INT64
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -15810,14 +15960,14 @@ opList {
name: "input"
argType: INPUT_TENSOR
}
- argDescriptor {
- name: "dimC"
- argType: INT64
- }
opDeclarationType: CONFIGURABLE_OP_IMPL
}
opList {
name: "rgb_to_yuv"
+ argDescriptor {
+ name: "dimC"
+ argType: INT64
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -15826,14 +15976,14 @@ opList {
name: "input"
argType: INPUT_TENSOR
}
- argDescriptor {
- name: "dimC"
- argType: INT64
- }
opDeclarationType: CONFIGURABLE_OP_IMPL
}
opList {
name: "rint"
+ argDescriptor {
+ name: "dataType"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -15846,10 +15996,6 @@ opList {
name: "input"
argType: INPUT_TENSOR
}
- argDescriptor {
- name: "dataType"
- argType: DATA_TYPE
- }
opDeclarationType: OP_IMPL
}
opList {
@@ -15863,6 +16009,20 @@ opList {
argType: OUTPUT_TENSOR
argIndex: 1
}
+ argDescriptor {
+ name: "dLr"
+ argType: DOUBLE
+ }
+ argDescriptor {
+ name: "dRmsDecay"
+ argType: DOUBLE
+ argIndex: 1
+ }
+ argDescriptor {
+ name: "dEpsilon"
+ argType: DOUBLE
+ argIndex: 2
+ }
argDescriptor {
name: "gradient"
argType: INPUT_TENSOR
@@ -15887,24 +16047,14 @@ opList {
argType: INPUT_TENSOR
argIndex: 4
}
- argDescriptor {
- name: "dLr"
- argType: DOUBLE
- }
- argDescriptor {
- name: "dRmsDecay"
- argType: DOUBLE
- argIndex: 1
- }
- argDescriptor {
- name: "dEpsilon"
- argType: DOUBLE
- argIndex: 2
- }
opDeclarationType: CONFIGURABLE_OP_IMPL
}
opList {
name: "roll"
+ argDescriptor {
+ name: "shift"
+ argType: INT64
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -15923,14 +16073,14 @@ opList {
argType: INPUT_TENSOR
argIndex: 2
}
- argDescriptor {
- name: "shift"
- argType: INT64
- }
opDeclarationType: CONFIGURABLE_OP_IMPL
}
opList {
name: "round"
+ argDescriptor {
+ name: "dataType"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -15943,14 +16093,14 @@ opList {
name: "input"
argType: INPUT_TENSOR
}
- argDescriptor {
- name: "dataType"
- argType: DATA_TYPE
- }
opDeclarationType: LEGACY_XYZ
}
opList {
name: "rshift_bits"
+ argDescriptor {
+ name: "dataType"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -15968,14 +16118,14 @@ opList {
argType: INPUT_TENSOR
argIndex: 1
}
- argDescriptor {
- name: "dataType"
- argType: DATA_TYPE
- }
opDeclarationType: BROADCASTABLE_OP_IMPL
}
opList {
name: "rsqrt"
+ argDescriptor {
+ name: "dataType"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -15988,10 +16138,6 @@ opList {
name: "input"
argType: INPUT_TENSOR
}
- argDescriptor {
- name: "dataType"
- argType: DATA_TYPE
- }
opDeclarationType: LEGACY_XYZ
}
opList {
@@ -16098,6 +16244,10 @@ opList {
}
opList {
name: "scatter_list"
+ argDescriptor {
+ name: "dtype"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -16116,10 +16266,6 @@ opList {
argType: INPUT_TENSOR
argIndex: 2
}
- argDescriptor {
- name: "dtype"
- argType: DATA_TYPE
- }
opDeclarationType: LIST_OP_IMPL
}
opList {
@@ -16217,6 +16363,10 @@ opList {
}
opList {
name: "scatter_nd"
+ argDescriptor {
+ name: "dtype"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -16244,10 +16394,6 @@ opList {
argType: INPUT_TENSOR
argIndex: 2
}
- argDescriptor {
- name: "dtype"
- argType: DATA_TYPE
- }
}
opList {
name: "scatter_nd_add"
@@ -16421,35 +16567,13 @@ opList {
}
argDescriptor {
name: "indices"
- argType: INT64
- }
- argDescriptor {
- name: "dimension"
- argType: INT64
- argIndex: 1
+ argType: INPUT_TENSOR
+ argIndex: 2
}
opDeclarationType: CONFIGURABLE_OP_IMPL
}
opList {
name: "sconv2d"
- argDescriptor {
- name: "*output"
- argType: OUTPUT_TENSOR
- }
- argDescriptor {
- name: "*input"
- argType: INPUT_TENSOR
- }
- argDescriptor {
- name: "*weightsDepth"
- argType: INPUT_TENSOR
- argIndex: 1
- }
- argDescriptor {
- name: "bias"
- argType: INPUT_TENSOR
- argIndex: 2
- }
argDescriptor {
name: "kH"
argType: INT64
@@ -16508,9 +16632,85 @@ opList {
name: "dtype"
argType: DATA_TYPE
}
+ argDescriptor {
+ name: "*output"
+ argType: OUTPUT_TENSOR
+ }
+ argDescriptor {
+ name: "*input"
+ argType: INPUT_TENSOR
+ }
+ argDescriptor {
+ name: "*weightsDepth"
+ argType: INPUT_TENSOR
+ argIndex: 1
+ }
+ argDescriptor {
+ name: "bias"
+ argType: INPUT_TENSOR
+ argIndex: 2
+ }
}
opList {
name: "sconv2d_bp"
+ argDescriptor {
+ name: "kH"
+ argType: INT64
+ }
+ argDescriptor {
+ name: "kW"
+ argType: INT64
+ argIndex: 1
+ }
+ argDescriptor {
+ name: "sH"
+ argType: INT64
+ argIndex: 2
+ }
+ argDescriptor {
+ name: "sW"
+ argType: INT64
+ argIndex: 3
+ }
+ argDescriptor {
+ name: "pH"
+ argType: INT64
+ argIndex: 4
+ }
+ argDescriptor {
+ name: "pW"
+ argType: INT64
+ argIndex: 5
+ }
+ argDescriptor {
+ name: "dH"
+ argType: INT64
+ argIndex: 6
+ }
+ argDescriptor {
+ name: "dW"
+ argType: INT64
+ argIndex: 7
+ }
+ argDescriptor {
+ name: "isSameMode"
+ argType: INT64
+ argIndex: 8
+ }
+ argDescriptor {
+ name: "isNCHW"
+ argType: INT64
+ argIndex: 9
+ }
+ argDescriptor {
+ name: "wFormat"
+ argType: INT64
+ argIndex: 10
+ }
+ argDescriptor {
+ name: "dtype"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "*gradI"
argType: OUTPUT_TENSOR
@@ -16545,71 +16745,17 @@ opList {
argIndex: 2
}
argDescriptor {
- name: "weightsPoint"
+ name: "bias"
argType: INPUT_TENSOR
argIndex: 3
}
- argDescriptor {
- name: "kH"
- argType: INT64
- }
- argDescriptor {
- name: "kW"
- argType: INT64
- argIndex: 1
- }
- argDescriptor {
- name: "sH"
- argType: INT64
- argIndex: 2
- }
- argDescriptor {
- name: "sW"
- argType: INT64
- argIndex: 3
- }
- argDescriptor {
- name: "pH"
- argType: INT64
- argIndex: 4
- }
- argDescriptor {
- name: "pW"
- argType: INT64
- argIndex: 5
- }
- argDescriptor {
- name: "dH"
- argType: INT64
- argIndex: 6
- }
- argDescriptor {
- name: "dW"
- argType: INT64
- argIndex: 7
- }
- argDescriptor {
- name: "isSameMode"
- argType: INT64
- argIndex: 8
- }
- argDescriptor {
- name: "isNCHW"
- argType: INT64
- argIndex: 9
- }
- argDescriptor {
- name: "wFormat"
- argType: INT64
- argIndex: 10
- }
- argDescriptor {
- name: "dtype"
- argType: DATA_TYPE
- }
}
opList {
name: "segment_max"
+ argDescriptor {
+ name: "dtype"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "segmentedOutput"
argType: OUTPUT_TENSOR
@@ -16623,13 +16769,13 @@ opList {
argType: INPUT_TENSOR
argIndex: 1
}
- argDescriptor {
- name: "dtype"
- argType: DATA_TYPE
- }
}
opList {
name: "segment_max_bp"
+ argDescriptor {
+ name: "dtype"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -16653,13 +16799,13 @@ opList {
argType: INPUT_TENSOR
argIndex: 2
}
- argDescriptor {
- name: "dtype"
- argType: DATA_TYPE
- }
}
opList {
name: "segment_mean"
+ argDescriptor {
+ name: "dtype"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "segmentedOutput"
argType: OUTPUT_TENSOR
@@ -16673,13 +16819,13 @@ opList {
argType: INPUT_TENSOR
argIndex: 1
}
- argDescriptor {
- name: "dtype"
- argType: DATA_TYPE
- }
}
opList {
name: "segment_mean_bp"
+ argDescriptor {
+ name: "dtype"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -16703,13 +16849,13 @@ opList {
argType: INPUT_TENSOR
argIndex: 2
}
- argDescriptor {
- name: "dtype"
- argType: DATA_TYPE
- }
}
opList {
name: "segment_min"
+ argDescriptor {
+ name: "dtype"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "segmentedOutput"
argType: OUTPUT_TENSOR
@@ -16723,13 +16869,13 @@ opList {
argType: INPUT_TENSOR
argIndex: 1
}
- argDescriptor {
- name: "dtype"
- argType: DATA_TYPE
- }
}
opList {
name: "segment_min_bp"
+ argDescriptor {
+ name: "dtype"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -16753,13 +16899,13 @@ opList {
argType: INPUT_TENSOR
argIndex: 2
}
- argDescriptor {
- name: "dtype"
- argType: DATA_TYPE
- }
}
opList {
name: "segment_prod"
+ argDescriptor {
+ name: "dtype"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "segmentedOutput"
argType: OUTPUT_TENSOR
@@ -16773,13 +16919,13 @@ opList {
argType: INPUT_TENSOR
argIndex: 1
}
+}
+opList {
+ name: "segment_prod_bp"
argDescriptor {
name: "dtype"
argType: DATA_TYPE
}
-}
-opList {
- name: "segment_prod_bp"
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -16803,13 +16949,13 @@ opList {
argType: INPUT_TENSOR
argIndex: 2
}
+}
+opList {
+ name: "segment_sum"
argDescriptor {
name: "dtype"
argType: DATA_TYPE
}
-}
-opList {
- name: "segment_sum"
argDescriptor {
name: "segmentedOutput"
argType: OUTPUT_TENSOR
@@ -16823,13 +16969,13 @@ opList {
argType: INPUT_TENSOR
argIndex: 1
}
+}
+opList {
+ name: "segment_sum_bp"
argDescriptor {
name: "dtype"
argType: DATA_TYPE
}
-}
-opList {
- name: "segment_sum_bp"
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -16848,13 +16994,13 @@ opList {
argType: INPUT_TENSOR
argIndex: 2
}
+}
+opList {
+ name: "select"
argDescriptor {
name: "dtype"
argType: DATA_TYPE
}
-}
-opList {
- name: "select"
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -16873,13 +17019,13 @@ opList {
argType: INPUT_TENSOR
argIndex: 2
}
- argDescriptor {
- name: "dtype"
- argType: DATA_TYPE
- }
}
opList {
name: "selu"
+ argDescriptor {
+ name: "dataType"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -16892,10 +17038,6 @@ opList {
name: "input"
argType: INPUT_TENSOR
}
- argDescriptor {
- name: "dataType"
- argType: DATA_TYPE
- }
opDeclarationType: CONFIGURABLE_OP_IMPL
}
opList {
@@ -16917,6 +17059,10 @@ opList {
}
opList {
name: "seluderivative"
+ argDescriptor {
+ name: "dataType"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -16929,14 +17075,18 @@ opList {
name: "input"
argType: INPUT_TENSOR
}
- argDescriptor {
- name: "dataType"
- argType: DATA_TYPE
- }
opDeclarationType: LEGACY_XYZ
}
opList {
name: "sequence_mask"
+ argDescriptor {
+ name: "maxInd"
+ argType: INT64
+ }
+ argDescriptor {
+ name: "dtype"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -16954,22 +17104,13 @@ opList {
argType: INPUT_TENSOR
argIndex: 1
}
- argDescriptor {
- name: "maxInd"
- argType: INT64
- }
- argDescriptor {
- name: "dataType"
- argType: INT64
- argIndex: 1
- }
- argDescriptor {
- name: "dtype"
- argType: DATA_TYPE
- }
}
opList {
name: "set"
+ argDescriptor {
+ name: "dataType"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -16983,10 +17124,6 @@ opList {
argType: INPUT_TENSOR
argIndex: 1
}
- argDescriptor {
- name: "dataType"
- argType: DATA_TYPE
- }
opDeclarationType: LEGACY_XYZ
}
opList {
@@ -17008,14 +17145,6 @@ opList {
}
opList {
name: "set_seed"
- argDescriptor {
- name: "outputs"
- argType: OUTPUT_TENSOR
- }
- argDescriptor {
- name: "input"
- argType: INPUT_TENSOR
- }
argDescriptor {
name: "seed"
argType: INT64
@@ -17024,9 +17153,21 @@ opList {
name: "dtype"
argType: DATA_TYPE
}
+ argDescriptor {
+ name: "outputs"
+ argType: OUTPUT_TENSOR
+ }
+ argDescriptor {
+ name: "input"
+ argType: INPUT_TENSOR
+ }
}
opList {
name: "setrange"
+ argDescriptor {
+ name: "dataType"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -17035,14 +17176,6 @@ opList {
name: "inPlace"
argType: BOOL
}
- argDescriptor {
- name: "input"
- argType: INPUT_TENSOR
- }
- argDescriptor {
- name: "dataType"
- argType: DATA_TYPE
- }
argDescriptor {
name: "min"
argType: DOUBLE
@@ -17052,6 +17185,10 @@ opList {
argType: DOUBLE
argIndex: 1
}
+ argDescriptor {
+ name: "input"
+ argType: INPUT_TENSOR
+ }
opDeclarationType: LEGACY_XYZ
}
opList {
@@ -17076,6 +17213,10 @@ opList {
name: "outputs"
argType: OUTPUT_TENSOR
}
+ argDescriptor {
+ name: "lr"
+ argType: DOUBLE
+ }
argDescriptor {
name: "input"
argType: INPUT_TENSOR
@@ -17085,14 +17226,14 @@ opList {
argType: INPUT_TENSOR
argIndex: 1
}
- argDescriptor {
- name: "lr"
- argType: DOUBLE
- }
opDeclarationType: CONFIGURABLE_OP_IMPL
}
opList {
name: "shannonentropy"
+ argDescriptor {
+ name: "dimensions"
+ argType: INT64
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -17110,14 +17251,14 @@ opList {
argType: INPUT_TENSOR
argIndex: 1
}
- argDescriptor {
- name: "dimensions"
- argType: INT64
- }
opDeclarationType: LEGACY_XYZ
}
opList {
name: "shape_of"
+ argDescriptor {
+ name: "dtype"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -17130,13 +17271,13 @@ opList {
name: "input"
argType: INPUT_TENSOR
}
- argDescriptor {
- name: "dtype"
- argType: DATA_TYPE
- }
}
opList {
name: "shapes_of"
+ argDescriptor {
+ name: "dtype"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -17149,13 +17290,13 @@ opList {
name: "input"
argType: INPUT_TENSOR
}
- argDescriptor {
- name: "dtype"
- argType: DATA_TYPE
- }
}
opList {
name: "shift_bits"
+ argDescriptor {
+ name: "dataType"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -17173,18 +17314,26 @@ opList {
argType: INPUT_TENSOR
argIndex: 1
}
- argDescriptor {
- name: "dataType"
- argType: DATA_TYPE
- }
opDeclarationType: BROADCASTABLE_OP_IMPL
}
opList {
name: "sigm_cross_entropy_loss"
+ argDescriptor {
+ name: "reductionMode"
+ argType: INT64
+ }
+ argDescriptor {
+ name: "dtype"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
}
+ argDescriptor {
+ name: "labelsSmoothing"
+ argType: DOUBLE
+ }
argDescriptor {
name: "logits"
argType: INPUT_TENSOR
@@ -17199,6 +17348,9 @@ opList {
argType: INPUT_TENSOR
argIndex: 2
}
+}
+opList {
+ name: "sigm_cross_entropy_loss_grad"
argDescriptor {
name: "reductionMode"
argType: INT64
@@ -17207,13 +17359,6 @@ opList {
name: "dtype"
argType: DATA_TYPE
}
- argDescriptor {
- name: "labelsSmoothing"
- argType: DOUBLE
- }
-}
-opList {
- name: "sigm_cross_entropy_loss_grad"
argDescriptor {
name: "dLdp"
argType: OUTPUT_TENSOR
@@ -17228,6 +17373,10 @@ opList {
argType: OUTPUT_TENSOR
argIndex: 2
}
+ argDescriptor {
+ name: "labelSmoothing"
+ argType: DOUBLE
+ }
argDescriptor {
name: "logits"
argType: INPUT_TENSOR
@@ -17242,21 +17391,13 @@ opList {
argType: INPUT_TENSOR
argIndex: 2
}
- argDescriptor {
- name: "reductionMode"
- argType: INT64
- }
- argDescriptor {
- name: "dtype"
- argType: DATA_TYPE
- }
- argDescriptor {
- name: "labelSmoothing"
- argType: DOUBLE
- }
}
opList {
name: "sigmoid"
+ argDescriptor {
+ name: "dataType"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -17269,10 +17410,6 @@ opList {
name: "input"
argType: INPUT_TENSOR
}
- argDescriptor {
- name: "dataType"
- argType: DATA_TYPE
- }
opDeclarationType: CONFIGURABLE_OP_IMPL
}
opList {
@@ -17294,6 +17431,10 @@ opList {
}
opList {
name: "sign"
+ argDescriptor {
+ name: "dataType"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -17306,14 +17447,14 @@ opList {
name: "input"
argType: INPUT_TENSOR
}
- argDescriptor {
- name: "dataType"
- argType: DATA_TYPE
- }
opDeclarationType: LEGACY_XYZ
}
opList {
name: "sin"
+ argDescriptor {
+ name: "dataType"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -17326,14 +17467,14 @@ opList {
name: "input"
argType: INPUT_TENSOR
}
- argDescriptor {
- name: "dataType"
- argType: DATA_TYPE
- }
opDeclarationType: LEGACY_XYZ
}
opList {
name: "sinh"
+ argDescriptor {
+ name: "dataType"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -17346,14 +17487,14 @@ opList {
name: "input"
argType: INPUT_TENSOR
}
- argDescriptor {
- name: "dataType"
- argType: DATA_TYPE
- }
opDeclarationType: LEGACY_XYZ
}
opList {
name: "size"
+ argDescriptor {
+ name: "dtype"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -17362,21 +17503,9 @@ opList {
name: "input"
argType: INPUT_TENSOR
}
- argDescriptor {
- name: "dtype"
- argType: DATA_TYPE
- }
}
opList {
name: "size_at"
- argDescriptor {
- name: "outputs"
- argType: OUTPUT_TENSOR
- }
- argDescriptor {
- name: "input"
- argType: INPUT_TENSOR
- }
argDescriptor {
name: "dimensions"
argType: INT64
@@ -17385,6 +17514,14 @@ opList {
name: "dtype"
argType: DATA_TYPE
}
+ argDescriptor {
+ name: "outputs"
+ argType: OUTPUT_TENSOR
+ }
+ argDescriptor {
+ name: "input"
+ argType: INPUT_TENSOR
+ }
}
opList {
name: "size_list"
@@ -17400,6 +17537,15 @@ opList {
}
opList {
name: "skipgram"
+ argDescriptor {
+ name: "numWorkers"
+ argType: INT64
+ }
+ argDescriptor {
+ name: "nsRounds"
+ argType: INT64
+ argIndex: 1
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -17472,19 +17618,18 @@ opList {
argType: INPUT_TENSOR
argIndex: 11
}
- argDescriptor {
- name: "numWorkers"
- argType: INT64
- }
- argDescriptor {
- name: "nsRounds"
- argType: INT64
- argIndex: 1
- }
opDeclarationType: CONFIGURABLE_OP_IMPL
}
opList {
name: "slice"
+ argDescriptor {
+ name: "size"
+ argType: INT64
+ }
+ argDescriptor {
+ name: "dtype"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -17503,6 +17648,9 @@ opList {
argType: INPUT_TENSOR
argIndex: 2
}
+}
+opList {
+ name: "slice_bp"
argDescriptor {
name: "size"
argType: INT64
@@ -17511,9 +17659,6 @@ opList {
name: "dtype"
argType: DATA_TYPE
}
-}
-opList {
- name: "slice_bp"
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -17537,17 +17682,17 @@ opList {
argType: INPUT_TENSOR
argIndex: 3
}
- argDescriptor {
- name: "size"
- argType: INT64
- }
- argDescriptor {
- name: "dtype"
- argType: DATA_TYPE
- }
}
opList {
name: "softmax"
+ argDescriptor {
+ name: "dimension"
+ argType: INT64
+ }
+ argDescriptor {
+ name: "dataType"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -17560,18 +17705,14 @@ opList {
name: "input"
argType: INPUT_TENSOR
}
- argDescriptor {
- name: "dimension"
- argType: INT64
- }
- argDescriptor {
- name: "dataType"
- argType: DATA_TYPE
- }
opDeclarationType: CONFIGURABLE_OP_IMPL
}
opList {
name: "softmax_bp"
+ argDescriptor {
+ name: "dimension"
+ argType: INT64
+ }
argDescriptor {
name: "gradI"
argType: OUTPUT_TENSOR
@@ -17585,18 +17726,26 @@ opList {
argType: INPUT_TENSOR
argIndex: 1
}
- argDescriptor {
- name: "dimension"
- argType: INT64
- }
opDeclarationType: CONFIGURABLE_OP_IMPL
}
opList {
name: "softmax_cross_entropy_loss"
+ argDescriptor {
+ name: "reductionMode"
+ argType: INT64
+ }
+ argDescriptor {
+ name: "dtype"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
}
+ argDescriptor {
+ name: "labelsSmoothing"
+ argType: DOUBLE
+ }
argDescriptor {
name: "logits"
argType: INPUT_TENSOR
@@ -17611,6 +17760,9 @@ opList {
argType: INPUT_TENSOR
argIndex: 2
}
+}
+opList {
+ name: "softmax_cross_entropy_loss_grad"
argDescriptor {
name: "reductionMode"
argType: INT64
@@ -17619,13 +17771,6 @@ opList {
name: "dtype"
argType: DATA_TYPE
}
- argDescriptor {
- name: "labelsSmoothing"
- argType: DOUBLE
- }
-}
-opList {
- name: "softmax_cross_entropy_loss_grad"
argDescriptor {
name: "dLdp"
argType: OUTPUT_TENSOR
@@ -17640,6 +17785,10 @@ opList {
argType: OUTPUT_TENSOR
argIndex: 2
}
+ argDescriptor {
+ name: "labelsSmoothing"
+ argType: DOUBLE
+ }
argDescriptor {
name: "logits"
argType: INPUT_TENSOR
@@ -17654,21 +17803,17 @@ opList {
argType: INPUT_TENSOR
argIndex: 2
}
+}
+opList {
+ name: "softmax_cross_entropy_loss_with_logits"
argDescriptor {
- name: "reductionMode"
+ name: "classesDim"
argType: INT64
}
argDescriptor {
name: "dtype"
argType: DATA_TYPE
}
- argDescriptor {
- name: "labelsSmoothing"
- argType: DOUBLE
- }
-}
-opList {
- name: "softmax_cross_entropy_loss_with_logits"
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -17682,6 +17827,9 @@ opList {
argType: INPUT_TENSOR
argIndex: 1
}
+}
+opList {
+ name: "softmax_cross_entropy_loss_with_logits_grad"
argDescriptor {
name: "classesDim"
argType: INT64
@@ -17690,9 +17838,6 @@ opList {
name: "dtype"
argType: DATA_TYPE
}
-}
-opList {
- name: "softmax_cross_entropy_loss_with_logits_grad"
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -17711,17 +17856,13 @@ opList {
argType: INPUT_TENSOR
argIndex: 1
}
- argDescriptor {
- name: "classesDim"
- argType: INT64
- }
- argDescriptor {
- name: "dtype"
- argType: DATA_TYPE
- }
}
opList {
name: "softplus"
+ argDescriptor {
+ name: "dataType"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -17734,10 +17875,6 @@ opList {
name: "input"
argType: INPUT_TENSOR
}
- argDescriptor {
- name: "dataType"
- argType: DATA_TYPE
- }
opDeclarationType: CONFIGURABLE_OP_IMPL
}
opList {
@@ -17759,6 +17896,10 @@ opList {
}
opList {
name: "softsign"
+ argDescriptor {
+ name: "dataType"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -17771,10 +17912,6 @@ opList {
name: "input"
argType: INPUT_TENSOR
}
- argDescriptor {
- name: "dataType"
- argType: DATA_TYPE
- }
opDeclarationType: CONFIGURABLE_OP_IMPL
}
opList {
@@ -17796,6 +17933,10 @@ opList {
}
opList {
name: "softsignderivative"
+ argDescriptor {
+ name: "dataType"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -17808,14 +17949,14 @@ opList {
name: "input"
argType: INPUT_TENSOR
}
- argDescriptor {
- name: "dataType"
- argType: DATA_TYPE
- }
opDeclarationType: LEGACY_XYZ
}
opList {
name: "solve"
+ argDescriptor {
+ name: "dtype"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -17838,13 +17979,13 @@ opList {
argType: INPUT_TENSOR
argIndex: 2
}
+}
+opList {
+ name: "solve_ls"
argDescriptor {
name: "dtype"
argType: DATA_TYPE
}
-}
-opList {
- name: "solve_ls"
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -17853,6 +17994,10 @@ opList {
name: "fastFlag"
argType: BOOL
}
+ argDescriptor {
+ name: "l2_factor"
+ argType: DOUBLE
+ }
argDescriptor {
name: "a"
argType: INPUT_TENSOR
@@ -17862,14 +18007,6 @@ opList {
argType: INPUT_TENSOR
argIndex: 1
}
- argDescriptor {
- name: "dtype"
- argType: DATA_TYPE
- }
- argDescriptor {
- name: "l2_factor"
- argType: DOUBLE
- }
}
opList {
name: "somepoolingpool2d"
@@ -17901,12 +18038,26 @@ opList {
opList {
name: "space_to_batch"
argDescriptor {
- name: "outputs"
- argType: OUTPUT_TENSOR
+ name: "blockSize"
+ argType: INT64
}
argDescriptor {
- name: "inPlace"
- argType: BOOL
+ name: "paddingTop"
+ argType: INT64
+ argIndex: 1
+ }
+ argDescriptor {
+ name: "paddingBottom"
+ argType: INT64
+ argIndex: 2
+ }
+ argDescriptor {
+ name: "dtype"
+ argType: DATA_TYPE
+ }
+ argDescriptor {
+ name: "outputs"
+ argType: OUTPUT_TENSOR
}
argDescriptor {
name: "input"
@@ -17917,17 +18068,17 @@ opList {
argType: INPUT_TENSOR
argIndex: 1
}
+}
+opList {
+ name: "space_to_batch_nd"
argDescriptor {
- name: "blockSize"
+ name: "blocks"
argType: INT64
}
argDescriptor {
name: "dtype"
argType: DATA_TYPE
}
-}
-opList {
- name: "space_to_batch_nd"
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -17950,25 +18101,9 @@ opList {
argType: INPUT_TENSOR
argIndex: 2
}
- argDescriptor {
- name: "blocks"
- argType: INT64
- }
- argDescriptor {
- name: "dtype"
- argType: DATA_TYPE
- }
}
opList {
name: "space_to_depth"
- argDescriptor {
- name: "outputs"
- argType: OUTPUT_TENSOR
- }
- argDescriptor {
- name: "input"
- argType: INPUT_TENSOR
- }
argDescriptor {
name: "block_size"
argType: INT64
@@ -17982,9 +18117,21 @@ opList {
name: "dtype"
argType: DATA_TYPE
}
+ argDescriptor {
+ name: "outputs"
+ argType: OUTPUT_TENSOR
+ }
+ argDescriptor {
+ name: "input"
+ argType: INPUT_TENSOR
+ }
}
opList {
name: "sparse_softmax_cross_entropy_loss_with_logits"
+ argDescriptor {
+ name: "dtype"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -17998,13 +18145,13 @@ opList {
argType: INPUT_TENSOR
argIndex: 1
}
+}
+opList {
+ name: "sparse_softmax_cross_entropy_loss_with_logits_grad"
argDescriptor {
name: "dtype"
argType: DATA_TYPE
}
-}
-opList {
- name: "sparse_softmax_cross_entropy_loss_with_logits_grad"
argDescriptor {
name: "dLdp"
argType: OUTPUT_TENSOR
@@ -18018,26 +18165,9 @@ opList {
argType: INPUT_TENSOR
argIndex: 1
}
- argDescriptor {
- name: "dtype"
- argType: DATA_TYPE
- }
}
opList {
name: "split"
- argDescriptor {
- name: "outArrs"
- argType: OUTPUT_TENSOR
- }
- argDescriptor {
- name: "a"
- argType: INPUT_TENSOR
- }
- argDescriptor {
- name: "b"
- argType: INPUT_TENSOR
- argIndex: 1
- }
argDescriptor {
name: "numSplit"
argType: INT64
@@ -18051,9 +18181,26 @@ opList {
name: "dtype"
argType: DATA_TYPE
}
+ argDescriptor {
+ name: "outArrs"
+ argType: OUTPUT_TENSOR
+ }
+ argDescriptor {
+ name: "a"
+ argType: INPUT_TENSOR
+ }
+ argDescriptor {
+ name: "b"
+ argType: INPUT_TENSOR
+ argIndex: 1
+ }
}
opList {
name: "split_list"
+ argDescriptor {
+ name: "dtype"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -18072,14 +18219,14 @@ opList {
argType: INPUT_TENSOR
argIndex: 2
}
- argDescriptor {
- name: "dtype"
- argType: DATA_TYPE
- }
opDeclarationType: LIST_OP_IMPL
}
opList {
name: "split_string"
+ argDescriptor {
+ name: "dtype"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -18093,13 +18240,22 @@ opList {
argType: INPUT_TENSOR
argIndex: 1
}
+}
+opList {
+ name: "split_v"
+ argDescriptor {
+ name: "dimensions"
+ argType: INT64
+ }
+ argDescriptor {
+ name: "numSplit"
+ argType: INT64
+ argIndex: 1
+ }
argDescriptor {
name: "dtype"
argType: DATA_TYPE
}
-}
-opList {
- name: "split_v"
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -18118,22 +18274,13 @@ opList {
argType: INPUT_TENSOR
argIndex: 2
}
- argDescriptor {
- name: "dimensions"
- argType: INT64
- }
- argDescriptor {
- name: "numSplit"
- argType: INT64
- argIndex: 1
- }
- argDescriptor {
- name: "dtype"
- argType: DATA_TYPE
- }
}
opList {
name: "sqrt"
+ argDescriptor {
+ name: "dataType"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -18146,10 +18293,6 @@ opList {
name: "input"
argType: INPUT_TENSOR
}
- argDescriptor {
- name: "dataType"
- argType: DATA_TYPE
- }
opDeclarationType: LEGACY_XYZ
}
opList {
@@ -18166,6 +18309,10 @@ opList {
}
opList {
name: "square"
+ argDescriptor {
+ name: "dataType"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -18178,14 +18325,14 @@ opList {
name: "input"
argType: INPUT_TENSOR
}
- argDescriptor {
- name: "dataType"
- argType: DATA_TYPE
- }
opDeclarationType: OP_IMPL
}
opList {
name: "squaredsubtract"
+ argDescriptor {
+ name: "dataType"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -18203,14 +18350,14 @@ opList {
argType: INPUT_TENSOR
argIndex: 1
}
- argDescriptor {
- name: "dataType"
- argType: DATA_TYPE
- }
opDeclarationType: BROADCASTABLE_OP_IMPL
}
opList {
name: "squaredsubtract_bp"
+ argDescriptor {
+ name: "dtype"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "gradX"
argType: OUTPUT_TENSOR
@@ -18234,13 +18381,17 @@ opList {
argType: INPUT_TENSOR
argIndex: 2
}
+}
+opList {
+ name: "squeeze"
+ argDescriptor {
+ name: "_a"
+ argType: INT64
+ }
argDescriptor {
name: "dtype"
argType: DATA_TYPE
}
-}
-opList {
- name: "squeeze"
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -18254,17 +18405,13 @@ opList {
argType: INPUT_TENSOR
argIndex: 1
}
- argDescriptor {
- name: "_a"
- argType: INT64
- }
+}
+opList {
+ name: "sru"
argDescriptor {
name: "dtype"
argType: DATA_TYPE
}
-}
-opList {
- name: "sru"
argDescriptor {
name: "h"
argType: OUTPUT_TENSOR
@@ -18298,13 +18445,13 @@ opList {
argType: INPUT_TENSOR
argIndex: 4
}
+}
+opList {
+ name: "sruCell"
argDescriptor {
name: "dtype"
argType: DATA_TYPE
}
-}
-opList {
- name: "sruCell"
argDescriptor {
name: "ht"
argType: OUTPUT_TENSOR
@@ -18333,13 +18480,13 @@ opList {
argType: INPUT_TENSOR
argIndex: 3
}
+}
+opList {
+ name: "sru_bi"
argDescriptor {
name: "dtype"
argType: DATA_TYPE
}
-}
-opList {
- name: "sru_bi"
argDescriptor {
name: "ht"
argType: OUTPUT_TENSOR
@@ -18373,13 +18520,13 @@ opList {
argType: INPUT_TENSOR
argIndex: 4
}
+}
+opList {
+ name: "sru_bi_bp"
argDescriptor {
name: "dtype"
argType: DATA_TYPE
}
-}
-opList {
- name: "sru_bi_bp"
argDescriptor {
name: "gradI"
argType: OUTPUT_TENSOR
@@ -18438,13 +18585,13 @@ opList {
argType: INPUT_TENSOR
argIndex: 7
}
+}
+opList {
+ name: "sru_bp"
argDescriptor {
name: "dtype"
argType: DATA_TYPE
}
-}
-opList {
- name: "sru_bp"
argDescriptor {
name: "gradX"
argType: OUTPUT_TENSOR
@@ -18503,13 +18650,13 @@ opList {
argType: INPUT_TENSOR
argIndex: 7
}
- argDescriptor {
- name: "dtype"
- argType: DATA_TYPE
- }
}
opList {
name: "stabilize"
+ argDescriptor {
+ name: "dataType"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -18518,14 +18665,6 @@ opList {
name: "inPlace"
argType: BOOL
}
- argDescriptor {
- name: "input"
- argType: INPUT_TENSOR
- }
- argDescriptor {
- name: "dataType"
- argType: DATA_TYPE
- }
argDescriptor {
name: "realMin"
argType: DOUBLE
@@ -18540,10 +18679,22 @@ opList {
argType: DOUBLE
argIndex: 2
}
+ argDescriptor {
+ name: "input"
+ argType: INPUT_TENSOR
+ }
opDeclarationType: LEGACY_XYZ
}
opList {
name: "stack"
+ argDescriptor {
+ name: "dimensions"
+ argType: INT64
+ }
+ argDescriptor {
+ name: "dtype"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -18557,17 +18708,13 @@ opList {
argType: INPUT_TENSOR
argIndex: 1
}
- argDescriptor {
- name: "dimensions"
- argType: INT64
- }
+}
+opList {
+ name: "stack_list"
argDescriptor {
name: "dtype"
argType: DATA_TYPE
}
-}
-opList {
- name: "stack_list"
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -18576,14 +18723,14 @@ opList {
name: "list"
argType: INPUT_TENSOR
}
- argDescriptor {
- name: "dtype"
- argType: DATA_TYPE
- }
opDeclarationType: LIST_OP_IMPL
}
opList {
name: "standardize"
+ argDescriptor {
+ name: "dimensions"
+ argType: INT64
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -18597,14 +18744,18 @@ opList {
argType: INPUT_TENSOR
argIndex: 1
}
- argDescriptor {
- name: "dimensions"
- argType: INT64
- }
opDeclarationType: CONFIGURABLE_OP_IMPL
}
opList {
name: "standardize_bp"
+ argDescriptor {
+ name: "dimensions"
+ argType: INT64
+ }
+ argDescriptor {
+ name: "dtype"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -18623,17 +18774,13 @@ opList {
argType: INPUT_TENSOR
argIndex: 2
}
- argDescriptor {
- name: "dimensions"
- argType: INT64
- }
+}
+opList {
+ name: "static_bidirectional_rnn"
argDescriptor {
name: "dtype"
argType: DATA_TYPE
}
-}
-opList {
- name: "static_bidirectional_rnn"
argDescriptor {
name: "h"
argType: OUTPUT_TENSOR
@@ -18692,13 +18839,13 @@ opList {
argType: INPUT_TENSOR
argIndex: 8
}
+}
+opList {
+ name: "static_rnn"
argDescriptor {
name: "dtype"
argType: DATA_TYPE
}
-}
-opList {
- name: "static_rnn"
argDescriptor {
name: "h"
argType: OUTPUT_TENSOR
@@ -18732,19 +18879,19 @@ opList {
argType: INPUT_TENSOR
argIndex: 4
}
- argDescriptor {
- name: "dtype"
- argType: DATA_TYPE
- }
}
opList {
name: "std"
+ argDescriptor {
+ name: "dimensions"
+ argType: INT64
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
}
argDescriptor {
- name: "newFormat"
+ name: "biasCorrected"
argType: BOOL
}
argDescriptor {
@@ -18756,10 +18903,6 @@ opList {
name: "input"
argType: INPUT_TENSOR
}
- argDescriptor {
- name: "dimensions"
- argType: INT64
- }
opDeclarationType: LEGACY_XYZ
}
opList {
@@ -18772,18 +18915,22 @@ opList {
name: "inPlace"
argType: BOOL
}
- argDescriptor {
- name: "input"
- argType: INPUT_TENSOR
- }
argDescriptor {
name: "cutoff"
argType: DOUBLE
}
+ argDescriptor {
+ name: "input"
+ argType: INPUT_TENSOR
+ }
opDeclarationType: LEGACY_XYZ
}
opList {
name: "stop_gradient"
+ argDescriptor {
+ name: "dataType"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -18796,37 +18943,10 @@ opList {
name: "input"
argType: INPUT_TENSOR
}
- argDescriptor {
- name: "dataType"
- argType: DATA_TYPE
- }
opDeclarationType: OP_IMPL
}
opList {
name: "strided_slice"
- argDescriptor {
- name: "outputs"
- argType: OUTPUT_TENSOR
- }
- argDescriptor {
- name: "input"
- argType: INPUT_TENSOR
- }
- argDescriptor {
- name: "v_begin"
- argType: INPUT_TENSOR
- argIndex: 1
- }
- argDescriptor {
- name: "v_end"
- argType: INPUT_TENSOR
- argIndex: 2
- }
- argDescriptor {
- name: "v_stride"
- argType: INPUT_TENSOR
- argIndex: 3
- }
argDescriptor {
name: "begin_mask"
argType: INT64
@@ -18855,9 +18975,60 @@ opList {
name: "dtype"
argType: DATA_TYPE
}
+ argDescriptor {
+ name: "outputs"
+ argType: OUTPUT_TENSOR
+ }
+ argDescriptor {
+ name: "input"
+ argType: INPUT_TENSOR
+ }
+ argDescriptor {
+ name: "v_begin"
+ argType: INPUT_TENSOR
+ argIndex: 1
+ }
+ argDescriptor {
+ name: "v_end"
+ argType: INPUT_TENSOR
+ argIndex: 2
+ }
+ argDescriptor {
+ name: "v_stride"
+ argType: INPUT_TENSOR
+ argIndex: 3
+ }
}
opList {
name: "strided_slice_bp"
+ argDescriptor {
+ name: "begin_mask"
+ argType: INT64
+ }
+ argDescriptor {
+ name: "ellipsis_mask"
+ argType: INT64
+ argIndex: 1
+ }
+ argDescriptor {
+ name: "end_mask"
+ argType: INT64
+ argIndex: 2
+ }
+ argDescriptor {
+ name: "new_axis_mask"
+ argType: INT64
+ argIndex: 3
+ }
+ argDescriptor {
+ name: "shrink_axis_mask"
+ argType: INT64
+ argIndex: 4
+ }
+ argDescriptor {
+ name: "dtype"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -18886,34 +19057,6 @@ opList {
argType: INPUT_TENSOR
argIndex: 4
}
- argDescriptor {
- name: "begin_mask"
- argType: INT64
- }
- argDescriptor {
- name: "ellipsis_mask"
- argType: INT64
- argIndex: 1
- }
- argDescriptor {
- name: "end_mask"
- argType: INT64
- argIndex: 2
- }
- argDescriptor {
- name: "new_axis_mask"
- argType: INT64
- argIndex: 3
- }
- argDescriptor {
- name: "shrink_axis_mask"
- argType: INT64
- argIndex: 4
- }
- argDescriptor {
- name: "dtype"
- argType: DATA_TYPE
- }
}
opList {
name: "sub_scalar"
@@ -18921,18 +19064,23 @@ opList {
name: "outputs"
argType: OUTPUT_TENSOR
}
- argDescriptor {
- name: "inPlace"
- argType: BOOL
- }
argDescriptor {
name: "input"
argType: INPUT_TENSOR
}
+ argDescriptor {
+ name: "y"
+ argType: INPUT_TENSOR
+ argIndex: 1
+ }
opDeclarationType: LEGACY_XYZ
}
opList {
name: "subtract"
+ argDescriptor {
+ name: "dataType"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -18950,14 +19098,14 @@ opList {
argType: INPUT_TENSOR
argIndex: 1
}
- argDescriptor {
- name: "dataType"
- argType: DATA_TYPE
- }
opDeclarationType: BROADCASTABLE_OP_IMPL
}
opList {
name: "subtract_bp"
+ argDescriptor {
+ name: "dtype"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "gradX"
argType: OUTPUT_TENSOR
@@ -18985,13 +19133,13 @@ opList {
argType: INPUT_TENSOR
argIndex: 2
}
+}
+opList {
+ name: "sufficient_statistics"
argDescriptor {
name: "dtype"
argType: DATA_TYPE
}
-}
-opList {
- name: "sufficient_statistics"
argDescriptor {
name: "dataCount"
argType: OUTPUT_TENSOR
@@ -19025,19 +19173,33 @@ opList {
argType: INPUT_TENSOR
argIndex: 2
}
+}
+opList {
+ name: "svd"
+ argDescriptor {
+ name: "fullUV"
+ argType: INT64
+ }
+ argDescriptor {
+ name: "calcUV"
+ argType: INT64
+ argIndex: 1
+ }
+ argDescriptor {
+ name: "switchNum"
+ argType: INT64
+ argIndex: 2
+ }
argDescriptor {
name: "dtype"
argType: DATA_TYPE
}
-}
-opList {
- name: "svd"
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
}
argDescriptor {
- name: "fullUV"
+ name: "full_matrices"
argType: BOOL
}
argDescriptor {
@@ -19064,27 +19226,13 @@ opList {
argType: INPUT_TENSOR
argIndex: 3
}
- argDescriptor {
- name: "fullUV"
- argType: INT64
- }
- argDescriptor {
- name: "calcUV"
- argType: INT64
- argIndex: 1
- }
- argDescriptor {
- name: "switchNum"
- argType: INT64
- argIndex: 2
- }
- argDescriptor {
- name: "dtype"
- argType: DATA_TYPE
- }
}
opList {
name: "swish"
+ argDescriptor {
+ name: "dataType"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -19097,10 +19245,6 @@ opList {
name: "input"
argType: INPUT_TENSOR
}
- argDescriptor {
- name: "dataType"
- argType: DATA_TYPE
- }
opDeclarationType: LEGACY_XYZ
}
opList {
@@ -19109,6 +19253,10 @@ opList {
name: "outputs"
argType: OUTPUT_TENSOR
}
+ argDescriptor {
+ name: "frameName"
+ argType: STRING
+ }
argDescriptor {
name: "input"
argType: INPUT_TENSOR
@@ -19118,13 +19266,13 @@ opList {
argType: INPUT_TENSOR
argIndex: 1
}
- argDescriptor {
- name: "frameName"
- argType: STRING
- }
}
opList {
name: "tan"
+ argDescriptor {
+ name: "dataType"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -19137,14 +19285,14 @@ opList {
name: "input"
argType: INPUT_TENSOR
}
- argDescriptor {
- name: "dataType"
- argType: DATA_TYPE
- }
opDeclarationType: LEGACY_XYZ
}
opList {
name: "tanderivative"
+ argDescriptor {
+ name: "dataType"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -19153,14 +19301,14 @@ opList {
name: "input"
argType: INPUT_TENSOR
}
- argDescriptor {
- name: "dataType"
- argType: DATA_TYPE
- }
opDeclarationType: LEGACY_XYZ
}
opList {
name: "tanh"
+ argDescriptor {
+ name: "dataType"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -19173,10 +19321,6 @@ opList {
name: "input"
argType: INPUT_TENSOR
}
- argDescriptor {
- name: "dataType"
- argType: DATA_TYPE
- }
opDeclarationType: CONFIGURABLE_OP_IMPL
}
opList {
@@ -19198,6 +19342,10 @@ opList {
}
opList {
name: "tear"
+ argDescriptor {
+ name: "dtype"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outE"
argType: OUTPUT_TENSOR
@@ -19206,21 +19354,17 @@ opList {
name: "input"
argType: INPUT_TENSOR
}
- argDescriptor {
- name: "dtype"
- argType: DATA_TYPE
- }
}
opList {
name: "tensorarrayv3"
- argDescriptor {
- name: "outputs"
- argType: OUTPUT_TENSOR
- }
argDescriptor {
name: "dataType"
argType: INT64
}
+ argDescriptor {
+ name: "outputs"
+ argType: OUTPUT_TENSOR
+ }
}
opList {
name: "tensorarraywritev3"
@@ -19235,6 +19379,10 @@ opList {
}
opList {
name: "tensordot"
+ argDescriptor {
+ name: "dimensionsY"
+ argType: INT64
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -19262,26 +19410,9 @@ opList {
argType: INPUT_TENSOR
argIndex: 1
}
- argDescriptor {
- name: "dimensionsY"
- argType: INT64
- }
}
opList {
name: "tensormmul"
- argDescriptor {
- name: "c"
- argType: OUTPUT_TENSOR
- }
- argDescriptor {
- name: "a"
- argType: INPUT_TENSOR
- }
- argDescriptor {
- name: "b"
- argType: INPUT_TENSOR
- argIndex: 1
- }
argDescriptor {
name: "axe0_size"
argType: INT64
@@ -19295,9 +19426,30 @@ opList {
name: "dtype"
argType: DATA_TYPE
}
+ argDescriptor {
+ name: "c"
+ argType: OUTPUT_TENSOR
+ }
+ argDescriptor {
+ name: "a"
+ argType: INPUT_TENSOR
+ }
+ argDescriptor {
+ name: "b"
+ argType: INPUT_TENSOR
+ argIndex: 1
+ }
}
opList {
name: "tensormmul_bp"
+ argDescriptor {
+ name: "axe0Size"
+ argType: INT64
+ }
+ argDescriptor {
+ name: "dtype"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "dLdA"
argType: OUTPUT_TENSOR
@@ -19321,14 +19473,6 @@ opList {
argType: INPUT_TENSOR
argIndex: 2
}
- argDescriptor {
- name: "axe0Size"
- argType: INT64
- }
- argDescriptor {
- name: "dtype"
- argType: DATA_TYPE
- }
}
opList {
name: "test_output_reshape"
@@ -19344,6 +19488,10 @@ opList {
}
opList {
name: "test_scalar"
+ argDescriptor {
+ name: "dtype"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -19352,21 +19500,17 @@ opList {
name: "input"
argType: INPUT_TENSOR
}
- argDescriptor {
- name: "dtype"
- argType: DATA_TYPE
- }
}
opList {
name: "testcustom"
- argDescriptor {
- name: "outputs"
- argType: OUTPUT_TENSOR
- }
argDescriptor {
name: "dtype"
argType: DATA_TYPE
}
+ argDescriptor {
+ name: "outputs"
+ argType: OUTPUT_TENSOR
+ }
}
opList {
name: "testop2i2o"
@@ -19400,6 +19544,10 @@ opList {
}
opList {
name: "tf_atan2"
+ argDescriptor {
+ name: "dataType"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -19417,10 +19565,6 @@ opList {
argType: INPUT_TENSOR
argIndex: 1
}
- argDescriptor {
- name: "dataType"
- argType: DATA_TYPE
- }
opDeclarationType: BROADCASTABLE_OP_IMPL
}
opList {
@@ -19433,14 +19577,14 @@ opList {
name: "inPlace"
argType: BOOL
}
- argDescriptor {
- name: "input"
- argType: INPUT_TENSOR
- }
argDescriptor {
name: "cutoff"
argType: DOUBLE
}
+ argDescriptor {
+ name: "input"
+ argType: INPUT_TENSOR
+ }
opDeclarationType: CONFIGURABLE_OP_IMPL
}
opList {
@@ -19449,6 +19593,10 @@ opList {
name: "dLdI"
argType: OUTPUT_TENSOR
}
+ argDescriptor {
+ name: "cutoff"
+ argType: DOUBLE
+ }
argDescriptor {
name: "input"
argType: INPUT_TENSOR
@@ -19458,14 +19606,18 @@ opList {
argType: INPUT_TENSOR
argIndex: 1
}
- argDescriptor {
- name: "cutoff"
- argType: DOUBLE
- }
opDeclarationType: CONFIGURABLE_OP_IMPL
}
opList {
name: "tile"
+ argDescriptor {
+ name: "dimensions"
+ argType: INT64
+ }
+ argDescriptor {
+ name: "dtype"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -19483,17 +19635,17 @@ opList {
argType: INPUT_TENSOR
argIndex: 1
}
+}
+opList {
+ name: "tile_bp"
argDescriptor {
- name: "dimensions"
+ name: "repeat"
argType: INT64
}
argDescriptor {
name: "dtype"
argType: DATA_TYPE
}
-}
-opList {
- name: "tile_bp"
argDescriptor {
name: "gradI"
argType: OUTPUT_TENSOR
@@ -19507,17 +19659,13 @@ opList {
argType: INPUT_TENSOR
argIndex: 1
}
- argDescriptor {
- name: "repeat"
- argType: INT64
- }
+}
+opList {
+ name: "tile_to_shape"
argDescriptor {
name: "dtype"
argType: DATA_TYPE
}
-}
-opList {
- name: "tile_to_shape"
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -19526,13 +19674,13 @@ opList {
name: "input"
argType: INPUT_TENSOR
}
+}
+opList {
+ name: "tile_to_shape_bp"
argDescriptor {
name: "dtype"
argType: DATA_TYPE
}
-}
-opList {
- name: "tile_to_shape_bp"
argDescriptor {
name: "gradX"
argType: OUTPUT_TENSOR
@@ -19546,13 +19694,13 @@ opList {
argType: INPUT_TENSOR
argIndex: 1
}
- argDescriptor {
- name: "dtype"
- argType: DATA_TYPE
- }
}
opList {
name: "timesoneminus"
+ argDescriptor {
+ name: "dataType"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -19565,14 +19713,14 @@ opList {
name: "input"
argType: INPUT_TENSOR
}
- argDescriptor {
- name: "dataType"
- argType: DATA_TYPE
- }
opDeclarationType: LEGACY_XYZ
}
opList {
name: "to_double"
+ argDescriptor {
+ name: "dtype"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -19581,13 +19729,13 @@ opList {
name: "input"
argType: INPUT_TENSOR
}
- argDescriptor {
- name: "dtype"
- argType: DATA_TYPE
- }
}
opList {
name: "to_float16"
+ argDescriptor {
+ name: "dtype"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -19596,13 +19744,13 @@ opList {
name: "input"
argType: INPUT_TENSOR
}
- argDescriptor {
- name: "dtype"
- argType: DATA_TYPE
- }
}
opList {
name: "to_float32"
+ argDescriptor {
+ name: "dtype"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -19611,13 +19759,13 @@ opList {
name: "input"
argType: INPUT_TENSOR
}
- argDescriptor {
- name: "dtype"
- argType: DATA_TYPE
- }
}
opList {
name: "to_int32"
+ argDescriptor {
+ name: "dtype"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -19626,13 +19774,13 @@ opList {
name: "input"
argType: INPUT_TENSOR
}
- argDescriptor {
- name: "dtype"
- argType: DATA_TYPE
- }
}
opList {
name: "to_int64"
+ argDescriptor {
+ name: "dtype"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -19641,13 +19789,13 @@ opList {
name: "input"
argType: INPUT_TENSOR
}
- argDescriptor {
- name: "dtype"
- argType: DATA_TYPE
- }
}
opList {
name: "to_uint32"
+ argDescriptor {
+ name: "dtype"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -19656,13 +19804,13 @@ opList {
name: "input"
argType: INPUT_TENSOR
}
- argDescriptor {
- name: "dtype"
- argType: DATA_TYPE
- }
}
opList {
name: "to_uint64"
+ argDescriptor {
+ name: "dtype"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -19671,10 +19819,6 @@ opList {
name: "input"
argType: INPUT_TENSOR
}
- argDescriptor {
- name: "dtype"
- argType: DATA_TYPE
- }
}
opList {
name: "toggle_bits"
@@ -19690,6 +19834,14 @@ opList {
}
opList {
name: "top_k"
+ argDescriptor {
+ name: "k"
+ argType: INT64
+ }
+ argDescriptor {
+ name: "dtype"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "values"
argType: OUTPUT_TENSOR
@@ -19707,17 +19859,13 @@ opList {
name: "input"
argType: INPUT_TENSOR
}
- argDescriptor {
- name: "k"
- argType: INT64
- }
+}
+opList {
+ name: "trace"
argDescriptor {
name: "dtype"
argType: DATA_TYPE
}
-}
-opList {
- name: "trace"
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -19726,13 +19874,13 @@ opList {
name: "input"
argType: INPUT_TENSOR
}
+}
+opList {
+ name: "transpose"
argDescriptor {
name: "dtype"
argType: DATA_TYPE
}
-}
-opList {
- name: "transpose"
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -19746,17 +19894,9 @@ opList {
argType: INPUT_TENSOR
argIndex: 1
}
- argDescriptor {
- name: "dtype"
- argType: DATA_TYPE
- }
}
opList {
name: "tri"
- argDescriptor {
- name: "outputs"
- argType: OUTPUT_TENSOR
- }
argDescriptor {
name: "row"
argType: INT64
@@ -19775,9 +19915,17 @@ opList {
name: "dtype"
argType: DATA_TYPE
}
+ argDescriptor {
+ name: "outputs"
+ argType: OUTPUT_TENSOR
+ }
}
opList {
name: "triangular_solve"
+ argDescriptor {
+ name: "dtype"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -19810,21 +19958,9 @@ opList {
argType: INPUT_TENSOR
argIndex: 3
}
- argDescriptor {
- name: "dtype"
- argType: DATA_TYPE
- }
}
opList {
name: "triu"
- argDescriptor {
- name: "outputs"
- argType: OUTPUT_TENSOR
- }
- argDescriptor {
- name: "input"
- argType: INPUT_TENSOR
- }
argDescriptor {
name: "diag"
argType: INT64
@@ -19833,9 +19969,25 @@ opList {
name: "dtype"
argType: DATA_TYPE
}
+ argDescriptor {
+ name: "outputs"
+ argType: OUTPUT_TENSOR
+ }
+ argDescriptor {
+ name: "input"
+ argType: INPUT_TENSOR
+ }
}
opList {
name: "triu_bp"
+ argDescriptor {
+ name: "diag"
+ argType: INT64
+ }
+ argDescriptor {
+ name: "dtype"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "gradI"
argType: OUTPUT_TENSOR
@@ -19849,17 +20001,13 @@ opList {
argType: INPUT_TENSOR
argIndex: 1
}
- argDescriptor {
- name: "diag"
- argType: INT64
- }
- argDescriptor {
- name: "dtype"
- argType: DATA_TYPE
- }
}
opList {
name: "truncatediv"
+ argDescriptor {
+ name: "dataType"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -19877,14 +20025,14 @@ opList {
argType: INPUT_TENSOR
argIndex: 1
}
- argDescriptor {
- name: "dataType"
- argType: DATA_TYPE
- }
opDeclarationType: BROADCASTABLE_OP_IMPL
}
opList {
name: "unique"
+ argDescriptor {
+ name: "dtype"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "values"
argType: OUTPUT_TENSOR
@@ -19898,13 +20046,13 @@ opList {
name: "input"
argType: INPUT_TENSOR
}
+}
+opList {
+ name: "unique_with_counts"
argDescriptor {
name: "dtype"
argType: DATA_TYPE
}
-}
-opList {
- name: "unique_with_counts"
argDescriptor {
name: "values"
argType: OUTPUT_TENSOR
@@ -19923,13 +20071,17 @@ opList {
name: "input"
argType: INPUT_TENSOR
}
+}
+opList {
+ name: "unsorted_segment_max"
+ argDescriptor {
+ name: "numSegments"
+ argType: INT64
+ }
argDescriptor {
name: "dtype"
argType: DATA_TYPE
}
-}
-opList {
- name: "unsorted_segment_max"
argDescriptor {
name: "segmentedOutput"
argType: OUTPUT_TENSOR
@@ -19948,17 +20100,17 @@ opList {
argType: INPUT_TENSOR
argIndex: 2
}
- argDescriptor {
- name: "numSegments"
- argType: INT64
- }
- argDescriptor {
- name: "dtype"
- argType: DATA_TYPE
- }
}
opList {
name: "unsorted_segment_max_bp"
+ argDescriptor {
+ name: "numSegments"
+ argType: INT64
+ }
+ argDescriptor {
+ name: "dtype"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -19977,17 +20129,17 @@ opList {
argType: INPUT_TENSOR
argIndex: 2
}
- argDescriptor {
- name: "numSegments"
- argType: INT64
- }
- argDescriptor {
- name: "dtype"
- argType: DATA_TYPE
- }
}
opList {
name: "unsorted_segment_mean"
+ argDescriptor {
+ name: "numSegments"
+ argType: INT64
+ }
+ argDescriptor {
+ name: "dtype"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "segmentedOutput"
argType: OUTPUT_TENSOR
@@ -20006,17 +20158,17 @@ opList {
argType: INPUT_TENSOR
argIndex: 2
}
- argDescriptor {
- name: "numSegments"
- argType: INT64
- }
- argDescriptor {
- name: "dtype"
- argType: DATA_TYPE
- }
}
opList {
name: "unsorted_segment_mean_bp"
+ argDescriptor {
+ name: "numSegments"
+ argType: INT64
+ }
+ argDescriptor {
+ name: "dtype"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -20035,17 +20187,17 @@ opList {
argType: INPUT_TENSOR
argIndex: 2
}
- argDescriptor {
- name: "numSegments"
- argType: INT64
- }
- argDescriptor {
- name: "dtype"
- argType: DATA_TYPE
- }
}
opList {
name: "unsorted_segment_min"
+ argDescriptor {
+ name: "numSegments"
+ argType: INT64
+ }
+ argDescriptor {
+ name: "dtype"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "segmentedOutput"
argType: OUTPUT_TENSOR
@@ -20064,17 +20216,17 @@ opList {
argType: INPUT_TENSOR
argIndex: 2
}
- argDescriptor {
- name: "numSegments"
- argType: INT64
- }
- argDescriptor {
- name: "dtype"
- argType: DATA_TYPE
- }
}
opList {
name: "unsorted_segment_min_bp"
+ argDescriptor {
+ name: "numSegments"
+ argType: INT64
+ }
+ argDescriptor {
+ name: "dtype"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -20093,17 +20245,17 @@ opList {
argType: INPUT_TENSOR
argIndex: 2
}
- argDescriptor {
- name: "numSegments"
- argType: INT64
- }
- argDescriptor {
- name: "dtype"
- argType: DATA_TYPE
- }
}
opList {
name: "unsorted_segment_prod"
+ argDescriptor {
+ name: "numSegments"
+ argType: INT64
+ }
+ argDescriptor {
+ name: "dtype"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "segmentedOutput"
argType: OUTPUT_TENSOR
@@ -20122,17 +20274,17 @@ opList {
argType: INPUT_TENSOR
argIndex: 2
}
- argDescriptor {
- name: "numSegments"
- argType: INT64
- }
- argDescriptor {
- name: "dtype"
- argType: DATA_TYPE
- }
}
opList {
name: "unsorted_segment_prod_bp"
+ argDescriptor {
+ name: "numSegments"
+ argType: INT64
+ }
+ argDescriptor {
+ name: "dtype"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -20151,17 +20303,17 @@ opList {
argType: INPUT_TENSOR
argIndex: 2
}
- argDescriptor {
- name: "numSegments"
- argType: INT64
- }
- argDescriptor {
- name: "dtype"
- argType: DATA_TYPE
- }
}
opList {
name: "unsorted_segment_sqrt_n"
+ argDescriptor {
+ name: "numSegments"
+ argType: INT64
+ }
+ argDescriptor {
+ name: "dtype"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "segmentedOutput"
argType: OUTPUT_TENSOR
@@ -20180,17 +20332,17 @@ opList {
argType: INPUT_TENSOR
argIndex: 2
}
- argDescriptor {
- name: "numSegments"
- argType: INT64
- }
- argDescriptor {
- name: "dtype"
- argType: DATA_TYPE
- }
}
opList {
name: "unsorted_segment_sqrt_n_bp"
+ argDescriptor {
+ name: "numSegments"
+ argType: INT64
+ }
+ argDescriptor {
+ name: "dtype"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -20209,6 +20361,9 @@ opList {
argType: INPUT_TENSOR
argIndex: 2
}
+}
+opList {
+ name: "unsorted_segment_sum"
argDescriptor {
name: "numSegments"
argType: INT64
@@ -20217,9 +20372,6 @@ opList {
name: "dtype"
argType: DATA_TYPE
}
-}
-opList {
- name: "unsorted_segment_sum"
argDescriptor {
name: "segmentedOutput"
argType: OUTPUT_TENSOR
@@ -20238,6 +20390,9 @@ opList {
argType: INPUT_TENSOR
argIndex: 2
}
+}
+opList {
+ name: "unsorted_segment_sum_bp"
argDescriptor {
name: "numSegments"
argType: INT64
@@ -20246,9 +20401,6 @@ opList {
name: "dtype"
argType: DATA_TYPE
}
-}
-opList {
- name: "unsorted_segment_sum_bp"
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -20267,25 +20419,9 @@ opList {
argType: INPUT_TENSOR
argIndex: 2
}
- argDescriptor {
- name: "numSegments"
- argType: INT64
- }
- argDescriptor {
- name: "dtype"
- argType: DATA_TYPE
- }
}
opList {
name: "unstack"
- argDescriptor {
- name: "outArrs"
- argType: OUTPUT_TENSOR
- }
- argDescriptor {
- name: "input"
- argType: INPUT_TENSOR
- }
argDescriptor {
name: "dimensions"
argType: INT64
@@ -20299,6 +20435,14 @@ opList {
name: "dtype"
argType: DATA_TYPE
}
+ argDescriptor {
+ name: "outArrs"
+ argType: OUTPUT_TENSOR
+ }
+ argDescriptor {
+ name: "input"
+ argType: INPUT_TENSOR
+ }
}
opList {
name: "unstack_list"
@@ -20314,18 +20458,6 @@ opList {
}
opList {
name: "upsampling2d"
- argDescriptor {
- name: "outputs"
- argType: OUTPUT_TENSOR
- }
- argDescriptor {
- name: "nchw"
- argType: BOOL
- }
- argDescriptor {
- name: "input"
- argType: INPUT_TENSOR
- }
argDescriptor {
name: "factorH"
argType: INT64
@@ -20344,9 +20476,29 @@ opList {
name: "dtype"
argType: DATA_TYPE
}
+ argDescriptor {
+ name: "outputs"
+ argType: OUTPUT_TENSOR
+ }
+ argDescriptor {
+ name: "nchw"
+ argType: BOOL
+ }
+ argDescriptor {
+ name: "input"
+ argType: INPUT_TENSOR
+ }
}
opList {
name: "upsampling2d_bp"
+ argDescriptor {
+ name: "scaleW"
+ argType: INT64
+ }
+ argDescriptor {
+ name: "dtype"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "gradI"
argType: OUTPUT_TENSOR
@@ -20364,29 +20516,9 @@ opList {
argType: INPUT_TENSOR
argIndex: 1
}
- argDescriptor {
- name: "scaleW"
- argType: INT64
- }
- argDescriptor {
- name: "dtype"
- argType: DATA_TYPE
- }
}
opList {
name: "upsampling3d"
- argDescriptor {
- name: "outputs"
- argType: OUTPUT_TENSOR
- }
- argDescriptor {
- name: "ncdhw"
- argType: BOOL
- }
- argDescriptor {
- name: "input"
- argType: INPUT_TENSOR
- }
argDescriptor {
name: "factorD"
argType: INT64
@@ -20410,9 +20542,29 @@ opList {
name: "dtype"
argType: DATA_TYPE
}
+ argDescriptor {
+ name: "outputs"
+ argType: OUTPUT_TENSOR
+ }
+ argDescriptor {
+ name: "ncdhw"
+ argType: BOOL
+ }
+ argDescriptor {
+ name: "input"
+ argType: INPUT_TENSOR
+ }
}
opList {
name: "upsampling3d_bp"
+ argDescriptor {
+ name: "isNCDHW"
+ argType: INT64
+ }
+ argDescriptor {
+ name: "dtype"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "gradI"
argType: OUTPUT_TENSOR
@@ -20430,17 +20582,13 @@ opList {
argType: INPUT_TENSOR
argIndex: 1
}
- argDescriptor {
- name: "isNCDHW"
- argType: INT64
- }
- argDescriptor {
- name: "dtype"
- argType: DATA_TYPE
- }
}
opList {
name: "var"
+ argDescriptor {
+ name: "dimensions"
+ argType: INT64
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -20458,10 +20606,6 @@ opList {
name: "input"
argType: INPUT_TENSOR
}
- argDescriptor {
- name: "dimensions"
- argType: INT64
- }
opDeclarationType: LEGACY_XYZ
}
opList {
@@ -20488,6 +20632,10 @@ opList {
}
opList {
name: "where_np"
+ argDescriptor {
+ name: "dtype"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -20506,13 +20654,13 @@ opList {
argType: INPUT_TENSOR
argIndex: 2
}
- argDescriptor {
- name: "dtype"
- argType: DATA_TYPE
- }
}
opList {
name: "write_list"
+ argDescriptor {
+ name: "idx"
+ argType: INT64
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -20526,18 +20674,22 @@ opList {
argType: INPUT_TENSOR
argIndex: 1
}
- argDescriptor {
- name: "idx"
- argType: INT64
- }
opDeclarationType: LIST_OP_IMPL
}
opList {
name: "xor"
+ argDescriptor {
+ name: "dataType"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
}
+ argDescriptor {
+ name: "comparable"
+ argType: DOUBLE
+ }
argDescriptor {
name: "input"
argType: INPUT_TENSOR
@@ -20547,14 +20699,6 @@ opList {
argType: INPUT_TENSOR
argIndex: 1
}
- argDescriptor {
- name: "dataType"
- argType: DATA_TYPE
- }
- argDescriptor {
- name: "comparable"
- argType: DOUBLE
- }
opDeclarationType: LEGACY_XYZ
}
opList {
@@ -20575,6 +20719,14 @@ opList {
}
opList {
name: "xw_plus_b"
+ argDescriptor {
+ name: "bTranspose"
+ argType: INT64
+ }
+ argDescriptor {
+ name: "dtype"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -20593,6 +20745,9 @@ opList {
argType: INPUT_TENSOR
argIndex: 2
}
+}
+opList {
+ name: "xw_plus_b_bp"
argDescriptor {
name: "bTranspose"
argType: INT64
@@ -20601,9 +20756,6 @@ opList {
name: "dtype"
argType: DATA_TYPE
}
-}
-opList {
- name: "xw_plus_b_bp"
argDescriptor {
name: "dLdx"
argType: OUTPUT_TENSOR
@@ -20637,17 +20789,13 @@ opList {
argType: INPUT_TENSOR
argIndex: 3
}
- argDescriptor {
- name: "bTranspose"
- argType: INT64
- }
- argDescriptor {
- name: "dtype"
- argType: DATA_TYPE
- }
}
opList {
name: "yiq_to_rgb"
+ argDescriptor {
+ name: "dimC"
+ argType: INT64
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -20656,14 +20804,14 @@ opList {
name: "input"
argType: INPUT_TENSOR
}
- argDescriptor {
- name: "dimC"
- argType: INT64
- }
opDeclarationType: CONFIGURABLE_OP_IMPL
}
opList {
name: "yuv_to_rgb"
+ argDescriptor {
+ name: "dimC"
+ argType: INT64
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -20672,14 +20820,14 @@ opList {
name: "input"
argType: INPUT_TENSOR
}
- argDescriptor {
- name: "dimC"
- argType: INT64
- }
opDeclarationType: CONFIGURABLE_OP_IMPL
}
opList {
name: "zero_fraction"
+ argDescriptor {
+ name: "dtype"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -20688,21 +20836,17 @@ opList {
name: "input"
argType: INPUT_TENSOR
}
- argDescriptor {
- name: "dtype"
- argType: DATA_TYPE
- }
}
opList {
name: "zeros_as"
- argDescriptor {
- name: "outputs"
- argType: OUTPUT_TENSOR
- }
argDescriptor {
name: "dtype"
argType: DATA_TYPE
}
+ argDescriptor {
+ name: "outputs"
+ argType: OUTPUT_TENSOR
+ }
}
opList {
name: "zeros_like"
@@ -20717,6 +20861,10 @@ opList {
}
opList {
name: "zeroslike"
+ argDescriptor {
+ name: "dataType"
+ argType: INT64
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -20729,13 +20877,13 @@ opList {
name: "input"
argType: INPUT_TENSOR
}
- argDescriptor {
- name: "dataType"
- argType: INT64
- }
}
opList {
name: "zeta"
+ argDescriptor {
+ name: "dataType"
+ argType: DATA_TYPE
+ }
argDescriptor {
name: "outputs"
argType: OUTPUT_TENSOR
@@ -20753,10 +20901,6 @@ opList {
argType: INPUT_TENSOR
argIndex: 1
}
- argDescriptor {
- name: "dataType"
- argType: DATA_TYPE
- }
opDeclarationType: CONFIGURABLE_OP_IMPL
}
opList {
diff --git a/nd4j/samediff-import/samediff-import-onnx/src/main/kotlin/org/nd4j/samediff/frameworkimport/onnx/opdefs/OnnxOpDescriptorLoader.kt b/nd4j/samediff-import/samediff-import-onnx/src/main/kotlin/org/nd4j/samediff/frameworkimport/onnx/opdefs/OnnxOpDescriptorLoader.kt
index 9a27836bb..972d67667 100644
--- a/nd4j/samediff-import/samediff-import-onnx/src/main/kotlin/org/nd4j/samediff/frameworkimport/onnx/opdefs/OnnxOpDescriptorLoader.kt
+++ b/nd4j/samediff-import/samediff-import-onnx/src/main/kotlin/org/nd4j/samediff/frameworkimport/onnx/opdefs/OnnxOpDescriptorLoader.kt
@@ -21,6 +21,7 @@ package org.nd4j.samediff.frameworkimport.onnx.opdefs
import onnx.Onnx
import org.apache.commons.io.IOUtils
+import org.nd4j.common.config.ND4JClassLoading
import org.nd4j.common.io.ClassPathResource
import org.nd4j.ir.MapperNamespace
import org.nd4j.ir.OpNamespace
@@ -37,11 +38,11 @@ import java.nio.charset.Charset
class OnnxOpDescriptorLoader: OpDescriptorLoader {
- val onnxFileNameTextDefault = "onnx-op-defs.pb"
+ val onnxFileNameTextDefault = "/onnx-op-defs.pb"
val onnxFileSpecifierProperty = "samediff.import.onnxdescriptors"
- val onnxMappingRulSetDefaultFile = "onnx-mapping-ruleset.pbtxt"
+ val onnxMappingRulSetDefaultFile = "/onnx-mapping-ruleset.pbtxt"
val onnxRulesetSpecifierProperty = "samediff.import.onnxmappingrules"
val nd4jOpDescriptors = nd4jOpList()
var mapperDefSet: MapperNamespace.MappingDefinitionSet? = mappingProcessDefinitionSet()
@@ -55,7 +56,7 @@ class OnnxOpDescriptorLoader: OpDescriptorLoader {
override fun nd4jOpList(): OpNamespace.OpDescriptorList {
val fileName = System.getProperty(nd4jFileSpecifierProperty, nd4jFileNameTextDefault)
- val nd4jOpDescriptorResourceStream = ClassPathResource(fileName).inputStream
+ val nd4jOpDescriptorResourceStream = ClassPathResource(fileName, ND4JClassLoading.getNd4jClassloader()).inputStream
val resourceString = IOUtils.toString(nd4jOpDescriptorResourceStream, Charset.defaultCharset())
val descriptorListBuilder = OpNamespace.OpDescriptorList.newBuilder()
TextFormat.merge(resourceString,descriptorListBuilder)
@@ -72,7 +73,7 @@ class OnnxOpDescriptorLoader: OpDescriptorLoader {
if(cachedOpDefs != null)
return cachedOpDefs!!
val fileName = System.getProperty(onnxFileSpecifierProperty, onnxFileNameTextDefault)
- val stream = ClassPathResource(fileName).inputStream
+ val stream = ClassPathResource(fileName,ND4JClassLoading.getNd4jClassloader()).inputStream
val ret = HashMap()
val graphProto = Onnx.GraphProto.parseFrom(stream)
@@ -88,7 +89,7 @@ class OnnxOpDescriptorLoader: OpDescriptorLoader {
if(mapperDefSet != null)
return mapperDefSet!!
val fileName = System.getProperty(onnxRulesetSpecifierProperty, onnxMappingRulSetDefaultFile)
- val string = IOUtils.toString(ClassPathResource(fileName).inputStream, Charset.defaultCharset())
+ val string = IOUtils.toString(ClassPathResource(fileName,ND4JClassLoading.getNd4jClassloader()).inputStream, Charset.defaultCharset())
val declarationBuilder = MapperNamespace.MappingDefinitionSet.newBuilder()
TextFormat.merge(string,declarationBuilder)
val ret = declarationBuilder.build()
diff --git a/nd4j/samediff-import/samediff-import-tensorflow/src/main/kotlin/org/nd4j/samediff/frameworkimport/tensorflow/definitions/TensorflowOpDeclarations.kt b/nd4j/samediff-import/samediff-import-tensorflow/src/main/kotlin/org/nd4j/samediff/frameworkimport/tensorflow/definitions/TensorflowOpDeclarations.kt
index a8a13be46..94a6a416a 100644
--- a/nd4j/samediff-import/samediff-import-tensorflow/src/main/kotlin/org/nd4j/samediff/frameworkimport/tensorflow/definitions/TensorflowOpDeclarations.kt
+++ b/nd4j/samediff-import/samediff-import-tensorflow/src/main/kotlin/org/nd4j/samediff/frameworkimport/tensorflow/definitions/TensorflowOpDeclarations.kt
@@ -1461,7 +1461,7 @@ val nonMaxSuppressionV1 = multipleNameMapping(inputFrameworkOpNames = listOf("No
argIndex = 1
}
)),
- valueMapping(mutableMapOf("overlayThreshold" to "iou_threshold")),
+ valueMapping(mutableMapOf("iouThreshold" to "iou_threshold")),
convertNDArrayInputToNumericalAttr(mutableMapOf("maxOutputSize" to "max_output_size")))
,tensorflowOpRegistry = tensorflowOpRegistry)
@@ -1470,7 +1470,7 @@ val nonMaxSuppressionV1 = multipleNameMapping(inputFrameworkOpNames = listOf("No
val nonMaxSuppressionV2 = multipleNameMapping(inputFrameworkOpNames = listOf("NonMaxSuppressionV2"),
opName = "non_max_suppression",
tensorNames = mutableMapOf("boxes" to "boxes","scales" to "scores",
- "overlayThreshold" to "iou_threshold","maxOutputSize" to "max_output_size"),
+ "iouThreshold" to "iou_threshold","maxOutputSize" to "max_output_size"),
attributeMappingRules = listOf(
argDescriptorConstant(listOf(
ArgDescriptor {
@@ -1763,7 +1763,7 @@ val resizeBiCubic = multipleNameMapping(inputFrameworkOpNames = listOf("ResizeBi
tensorNames = mutableMapOf("image" to "images","size" to "size"),tensorflowOpRegistry = tensorflowOpRegistry)
val resizeBiLinear = multipleNameMapping(inputFrameworkOpNames = listOf("ResizeBilinear"),opName = "resize_bilinear",
- attributeMappingRules = listOf(valueMapping(mutableMapOf("alignCorners" to "align_corners","halfPixelCenters" to "half_pixel_centers"))),
+ attributeMappingRules = listOf(valueMapping(mutableMapOf("alignCorners" to "align_corners","halfPixelCenter" to "half_pixel_centers"))),
tensorNames = mutableMapOf("image" to "images","newImageSize" to "size"),tensorflowOpRegistry = tensorflowOpRegistry)
val resizeNearestNeighbor = multipleNameMapping(inputFrameworkOpNames = listOf("ResizeNearestNeighbor"),opName = "resize_nearest_neighbor",
@@ -1861,12 +1861,12 @@ val scatterSub = multipleNameMapping(inputFrameworkOpNames = listOf("ScatterSub"
//TODO: note: TF expects indices, we don't support them?
val scatterUpdate = multipleNameMapping(inputFrameworkOpNames = listOf("ScatterUpdate"),opName = "scatter_update",
- attributeMappingRules = listOf(ndarrayToIntList(mutableMapOf("indices" to "indices"))),
- tensorNames = mutableMapOf("operand" to "ref","updates" to "updates"),tensorflowOpRegistry = tensorflowOpRegistry)
+ attributeMappingRules = listOf(),
+ tensorNames = mutableMapOf("operand" to "ref","updates" to "updates","indices" to "indices"),tensorflowOpRegistry = tensorflowOpRegistry)
val tensorScatterUpdate = multipleNameMapping(inputFrameworkOpNames = listOf("TensorScatterUpdate"),opName = "scatter_update",
- attributeMappingRules = listOf(ndarrayToIntList(mutableMapOf("indices" to "indices"))),
- tensorNames = mutableMapOf("operand" to "tensor","updates" to "updates"),tensorflowOpRegistry = tensorflowOpRegistry)
+ attributeMappingRules = listOf(),
+ tensorNames = mutableMapOf("operand" to "tensor","updates" to "updates","indices" to "indices"),tensorflowOpRegistry = tensorflowOpRegistry)
//L2Loss
val l2Loss = multipleNameMapping(inputFrameworkOpNames = listOf("L2Loss"),opName = "l2_loss",
attributeMappingRules = listOf(valueMapping(mutableMapOf("dtype" to "T"))),
diff --git a/nd4j/samediff-import/samediff-import-tensorflow/src/main/kotlin/org/nd4j/samediff/frameworkimport/tensorflow/opdefs/TensorflowOpDescriptorLoader.kt b/nd4j/samediff-import/samediff-import-tensorflow/src/main/kotlin/org/nd4j/samediff/frameworkimport/tensorflow/opdefs/TensorflowOpDescriptorLoader.kt
index 0c269630c..5d8258745 100644
--- a/nd4j/samediff-import/samediff-import-tensorflow/src/main/kotlin/org/nd4j/samediff/frameworkimport/tensorflow/opdefs/TensorflowOpDescriptorLoader.kt
+++ b/nd4j/samediff-import/samediff-import-tensorflow/src/main/kotlin/org/nd4j/samediff/frameworkimport/tensorflow/opdefs/TensorflowOpDescriptorLoader.kt
@@ -20,6 +20,7 @@
package org.nd4j.samediff.frameworkimport.tensorflow.opdefs
import org.apache.commons.io.IOUtils
+import org.nd4j.common.config.ND4JClassLoading
import org.nd4j.common.io.ClassPathResource
import org.nd4j.ir.MapperNamespace
import org.nd4j.ir.OpNamespace
@@ -37,10 +38,10 @@ import java.nio.charset.Charset
class TensorflowOpDescriptorLoader: OpDescriptorLoader {
- val tensorflowFileNameTextDefault = "tensorflow-op-def.pbtxt"
+ val tensorflowFileNameTextDefault = "/tensorflow-op-def.pbtxt"
val tensorflowFileSpecifierProperty = "samediff.import.tensorflowdescriptors"
- val tensorflowMappingRulSetDefaultFile = "tensorflow-mapping-ruleset.pbtxt"
+ val tensorflowMappingRulSetDefaultFile = "/tensorflow-mapping-ruleset.pbtxt"
val tensorflowRulesetSpecifierProperty = "samediff.import.tensorflowmappingrules"
val nd4jOpDescriptors = nd4jOpList()
var mapperDefSet: MapperNamespace.MappingDefinitionSet? = mappingProcessDefinitionSet()
@@ -51,7 +52,7 @@ class TensorflowOpDescriptorLoader: OpDescriptorLoader {
override fun nd4jOpList(): OpNamespace.OpDescriptorList {
val fileName = System.getProperty(nd4jFileSpecifierProperty, nd4jFileNameTextDefault)
- val nd4jOpDescriptorResourceStream = ClassPathResource(fileName).inputStream
+ val nd4jOpDescriptorResourceStream = ClassPathResource(fileName,ND4JClassLoading.getNd4jClassloader()).inputStream
val resourceString = IOUtils.toString(nd4jOpDescriptorResourceStream, Charset.defaultCharset())
val descriptorListBuilder = OpNamespace.OpDescriptorList.newBuilder()
TextFormat.merge(resourceString,descriptorListBuilder)
@@ -69,7 +70,7 @@ class TensorflowOpDescriptorLoader: OpDescriptorLoader {
return cachedOpList!!
}
val fileName = System.getProperty(tensorflowFileSpecifierProperty, tensorflowFileNameTextDefault)
- val string = IOUtils.toString(ClassPathResource(fileName).inputStream, Charset.defaultCharset())
+ val string = IOUtils.toString(ClassPathResource(fileName,ND4JClassLoading.getNd4jClassloader()).inputStream, Charset.defaultCharset())
val tfListBuilder = OpList.newBuilder()
TextFormat.merge(string, tfListBuilder)
val ret = HashMap()
@@ -86,7 +87,7 @@ class TensorflowOpDescriptorLoader: OpDescriptorLoader {
if(mapperDefSet != null)
return mapperDefSet!!
val fileName = System.getProperty(tensorflowRulesetSpecifierProperty, tensorflowMappingRulSetDefaultFile)
- val string = IOUtils.toString(ClassPathResource(fileName).inputStream, Charset.defaultCharset())
+ val string = IOUtils.toString(ClassPathResource(fileName,ND4JClassLoading.getNd4jClassloader()).inputStream, Charset.defaultCharset())
val declarationBuilder = MapperNamespace.MappingDefinitionSet.newBuilder()
try {
TextFormat.merge(string,declarationBuilder)
diff --git a/nd4j/samediff-import/samediff-import-tensorflow/src/main/resources/tensorflow-mapping-ruleset.pbtxt b/nd4j/samediff-import/samediff-import-tensorflow/src/main/resources/tensorflow-mapping-ruleset.pbtxt
index 400b9233a..96303b8f3 100644
--- a/nd4j/samediff-import/samediff-import-tensorflow/src/main/resources/tensorflow-mapping-ruleset.pbtxt
+++ b/nd4j/samediff-import/samediff-import-tensorflow/src/main/resources/tensorflow-mapping-ruleset.pbtxt
@@ -1706,6 +1706,7 @@ mappings {
rule {
ruleName: "ndarrayinputtonumericalattribute"
functionName: "ndarrayinputtonumericalattribute"
+ outputIntName: "numSegments"
inputToOutput {
key: "numSegments"
value: "num_segments"
@@ -1951,6 +1952,7 @@ mappings {
rule {
ruleName: "ndarraytointattributevalue"
functionName: "ndarraytointattributevalue"
+ outputIntName: "dimensions"
inputToOutput {
key: "dimensions"
value: "reduction_indices"
@@ -2084,6 +2086,7 @@ mappings {
rule {
ruleName: "ndarraytointattributevalue"
functionName: "ndarraytointattributevalue"
+ outputIntName: "dimensions"
inputToOutput {
key: "dimensions"
value: "reduction_indices"
@@ -2372,10 +2375,10 @@ mappings {
rule {
ruleName: "valuemapping"
functionName: "valuemapping"
+ outputIntName: "fullUV"
inputBooleanName: "compute_uv"
inputBooleanName: "full_matrices"
outputBooleanName: "computeUv"
- outputBooleanName: "fullUV"
inputToOutput {
key: "computeUv"
value: "compute_uv"
@@ -2391,9 +2394,9 @@ mappings {
ruleName: "invertbooleannumber"
functionName: "invertbooleannumber"
outputIntName: "calcUV"
+ outputIntName: "fullUV"
inputBooleanName: "compute_uv"
inputBooleanName: "full_matrices"
- outputBooleanName: "fullUV"
inputToOutput {
key: "calcUV"
value: "compute_uv"
@@ -4823,6 +4826,7 @@ mappings {
rule {
ruleName: "ndarraytointattributevalue"
functionName: "ndarraytointattributevalue"
+ outputIntName: "dimensions"
inputToOutput {
key: "dimensions"
value: "axis"
@@ -5298,7 +5302,7 @@ mappings {
functionName: "stringequals"
inputStringAttrName: "padding"
inputStringAttrName: "padding"
- outputBooleanName: "isSameMode"
+ outputIntName: "isSameMode"
inputToOutput {
key: "isSameMode"
value: "padding"
@@ -6140,10 +6144,10 @@ mappings {
rule {
ruleName: "invertbooleannumber"
functionName: "invertbooleannumber"
+ outputIntName: "exclusive"
+ outputIntName: "reverse"
inputBooleanName: "exclusive"
inputBooleanName: "reverse"
- outputBooleanName: "exclusive"
- outputBooleanName: "reverse"
inputToOutput {
key: "exclusive"
value: "exclusive"
@@ -6815,6 +6819,7 @@ mappings {
rule {
ruleName: "ndarrayinputtonumericalattribute"
functionName: "ndarrayinputtonumericalattribute"
+ outputIntName: "numSegments"
inputToOutput {
key: "numSegments"
value: "num_segments"
@@ -7065,6 +7070,7 @@ mappings {
rule {
ruleName: "ndarrayinputtonumericalattribute"
functionName: "ndarrayinputtonumericalattribute"
+ outputDoubleName: "start"
outputDoubleName: "stop"
inputToOutput {
key: "start"
@@ -7227,6 +7233,7 @@ mappings {
rule {
ruleName: "ndarrayinputtonumericalattribute"
functionName: "ndarrayinputtonumericalattribute"
+ outputIntName: "numSegments"
inputToOutput {
key: "numSegments"
value: "num_segments"
@@ -8895,7 +8902,7 @@ mappings {
inputTensorName: "max_output_size"
outputTensorName: "boxes"
outputTensorName: "scales"
- outputTensorName: "overlayThreshold"
+ outputTensorName: "iouThreshold"
outputTensorName: "maxOutputSize"
inputToOutput {
key: "boxes"
@@ -8906,7 +8913,7 @@ mappings {
value: "scores"
}
inputToOutput {
- key: "overlayThreshold"
+ key: "iouThreshold"
value: "iou_threshold"
}
inputToOutput {
@@ -8935,6 +8942,7 @@ mappings {
rule {
ruleName: "ndarrayinputtonumericalattribute"
functionName: "ndarrayinputtonumericalattribute"
+ outputIntName: "maxOutputSize"
inputToOutput {
key: "maxOutputSize"
value: "max_output_size"
@@ -9014,6 +9022,9 @@ mappings {
rule {
ruleName: "ndarrayinputtonumericalattribute"
functionName: "ndarrayinputtonumericalattribute"
+ outputIntName: "depth"
+ outputDoubleName: "on"
+ outputDoubleName: "off"
inputToOutput {
key: "on"
value: "on_value"
@@ -9226,6 +9237,7 @@ mappings {
rule {
ruleName: "ndarrayinputtonumericalattribute"
functionName: "ndarrayinputtonumericalattribute"
+ outputIntName: "numSegments"
inputToOutput {
key: "numSegments"
value: "num_segments"
@@ -9285,13 +9297,13 @@ mappings {
inputBooleanName: "align_corners"
inputBooleanName: "half_pixel_centers"
outputBooleanName: "alignCorners"
- outputBooleanName: "halfPixelCenters"
+ outputBooleanName: "halfPixelCenter"
inputToOutput {
key: "alignCorners"
value: "align_corners"
}
inputToOutput {
- key: "halfPixelCenters"
+ key: "halfPixelCenter"
value: "half_pixel_centers"
}
ruleType: "attribute"
@@ -9689,7 +9701,7 @@ mappings {
functionName: "valuemapping"
inputFloatName: "iou_threshold"
inputToOutput {
- key: "overlayThreshold"
+ key: "iouThreshold"
value: "iou_threshold"
}
ruleType: "attribute"
@@ -9698,6 +9710,7 @@ mappings {
rule {
ruleName: "ndarrayinputtonumericalattribute"
functionName: "ndarrayinputtonumericalattribute"
+ outputIntName: "maxOutputSize"
inputToOutput {
key: "maxOutputSize"
value: "max_output_size"
@@ -9923,6 +9936,7 @@ mappings {
rule {
ruleName: "ndarraytointattributevalue"
functionName: "ndarraytointattributevalue"
+ outputIntName: "dimensions"
inputToOutput {
key: "dimensions"
value: "reduction_indices"
@@ -10163,6 +10177,7 @@ mappings {
rule {
ruleName: "ndarraytointattributevalue"
functionName: "ndarraytointattributevalue"
+ outputIntName: "dimensions"
inputToOutput {
key: "dimensions"
value: "reduction_indices"
@@ -10323,6 +10338,8 @@ mappings {
rule {
ruleName: "ndarrayinputtonumericalattribute"
functionName: "ndarrayinputtonumericalattribute"
+ outputDoubleName: "min"
+ outputDoubleName: "max"
inputToOutput {
key: "min"
value: "minval"
@@ -11779,6 +11796,7 @@ mappings {
rule {
ruleName: "ndarraytointattributevalue"
functionName: "ndarraytointattributevalue"
+ outputIntName: "dimensions"
inputToOutput {
key: "dimensions"
value: "reduction_indices"
@@ -12463,6 +12481,7 @@ mappings {
rule {
ruleName: "ndarrayinputtonumericalattribute"
functionName: "ndarrayinputtonumericalattribute"
+ outputIntName: "concatDimension"
inputToOutput {
key: "concatDimension"
value: "concat_dim"
@@ -12958,8 +12977,8 @@ mappings {
rule {
ruleName: "invertbooleannumber"
functionName: "invertbooleannumber"
+ outputIntName: "reverse"
inputBooleanName: "reverse"
- outputBooleanName: "reverse"
inputToOutput {
key: "reverse"
value: "reverse"
@@ -13194,6 +13213,9 @@ mappings {
rule {
ruleName: "ndarrayinputtonumericalattribute"
functionName: "ndarrayinputtonumericalattribute"
+ outputIntName: "from"
+ outputIntName: "to"
+ outputIntName: "step"
inputToOutput {
key: "from"
value: "start"
@@ -14358,6 +14380,7 @@ mappings {
rule {
ruleName: "ndarraytointattributevalue"
functionName: "ndarraytointattributevalue"
+ outputIntName: "dimensions"
inputToOutput {
key: "dimensions"
value: "reduction_indices"
@@ -14580,6 +14603,7 @@ mappings {
rule {
ruleName: "ndarrayinputtonumericalattribute"
functionName: "ndarrayinputtonumericalattribute"
+ outputIntName: "concatDimension"
inputToOutput {
key: "concatDimension"
value: "axis"
@@ -14689,6 +14713,7 @@ mappings {
rule {
ruleName: "ndarraytointattributevalue"
functionName: "ndarraytointattributevalue"
+ outputIntName: "dimensions"
inputToOutput {
key: "dimensions"
value: "reduction_indices"
@@ -15391,8 +15416,10 @@ mappings {
functionName: "ndarraymapping"
inputTensorName: "ref"
inputTensorName: "updates"
+ inputTensorName: "indices"
outputTensorName: "operand"
outputTensorName: "updates"
+ outputTensorName: "indices"
inputToOutput {
key: "operand"
value: "ref"
@@ -15401,18 +15428,11 @@ mappings {
key: "updates"
value: "updates"
}
- ruleType: "tensor"
- inputFrameworkOpName: "ScatterUpdate"
- }
- rule {
- ruleName: "ndarraytointattributevalue"
- functionName: "ndarraytointattributevalue"
- outputIntName: "indices"
inputToOutput {
key: "indices"
value: "indices"
}
- ruleType: "attribute"
+ ruleType: "tensor"
inputFrameworkOpName: "ScatterUpdate"
}
}
@@ -15539,8 +15559,10 @@ mappings {
functionName: "ndarraymapping"
inputTensorName: "tensor"
inputTensorName: "updates"
+ inputTensorName: "indices"
outputTensorName: "operand"
outputTensorName: "updates"
+ outputTensorName: "indices"
inputToOutput {
key: "operand"
value: "tensor"
@@ -15549,18 +15571,11 @@ mappings {
key: "updates"
value: "updates"
}
- ruleType: "tensor"
- inputFrameworkOpName: "TensorScatterUpdate"
- }
- rule {
- ruleName: "ndarraytointattributevalue"
- functionName: "ndarraytointattributevalue"
- outputIntName: "indices"
inputToOutput {
key: "indices"
value: "indices"
}
- ruleType: "attribute"
+ ruleType: "tensor"
inputFrameworkOpName: "TensorScatterUpdate"
}
}
@@ -15647,10 +15662,10 @@ mappings {
rule {
ruleName: "invertbooleannumber"
functionName: "invertbooleannumber"
+ outputIntName: "exclusive"
+ outputIntName: "reverse"
inputBooleanName: "exclusive"
inputBooleanName: "reverse"
- outputBooleanName: "exclusive"
- outputBooleanName: "reverse"
inputToOutput {
key: "exclusive"
value: "exclusive"
diff --git a/nd4j/samediff-import/samediff-import-tensorflow/tensorflow-processes.pbtxt b/nd4j/samediff-import/samediff-import-tensorflow/tensorflow-processes.pbtxt
index 400b9233a..96303b8f3 100644
--- a/nd4j/samediff-import/samediff-import-tensorflow/tensorflow-processes.pbtxt
+++ b/nd4j/samediff-import/samediff-import-tensorflow/tensorflow-processes.pbtxt
@@ -1706,6 +1706,7 @@ mappings {
rule {
ruleName: "ndarrayinputtonumericalattribute"
functionName: "ndarrayinputtonumericalattribute"
+ outputIntName: "numSegments"
inputToOutput {
key: "numSegments"
value: "num_segments"
@@ -1951,6 +1952,7 @@ mappings {
rule {
ruleName: "ndarraytointattributevalue"
functionName: "ndarraytointattributevalue"
+ outputIntName: "dimensions"
inputToOutput {
key: "dimensions"
value: "reduction_indices"
@@ -2084,6 +2086,7 @@ mappings {
rule {
ruleName: "ndarraytointattributevalue"
functionName: "ndarraytointattributevalue"
+ outputIntName: "dimensions"
inputToOutput {
key: "dimensions"
value: "reduction_indices"
@@ -2372,10 +2375,10 @@ mappings {
rule {
ruleName: "valuemapping"
functionName: "valuemapping"
+ outputIntName: "fullUV"
inputBooleanName: "compute_uv"
inputBooleanName: "full_matrices"
outputBooleanName: "computeUv"
- outputBooleanName: "fullUV"
inputToOutput {
key: "computeUv"
value: "compute_uv"
@@ -2391,9 +2394,9 @@ mappings {
ruleName: "invertbooleannumber"
functionName: "invertbooleannumber"
outputIntName: "calcUV"
+ outputIntName: "fullUV"
inputBooleanName: "compute_uv"
inputBooleanName: "full_matrices"
- outputBooleanName: "fullUV"
inputToOutput {
key: "calcUV"
value: "compute_uv"
@@ -4823,6 +4826,7 @@ mappings {
rule {
ruleName: "ndarraytointattributevalue"
functionName: "ndarraytointattributevalue"
+ outputIntName: "dimensions"
inputToOutput {
key: "dimensions"
value: "axis"
@@ -5298,7 +5302,7 @@ mappings {
functionName: "stringequals"
inputStringAttrName: "padding"
inputStringAttrName: "padding"
- outputBooleanName: "isSameMode"
+ outputIntName: "isSameMode"
inputToOutput {
key: "isSameMode"
value: "padding"
@@ -6140,10 +6144,10 @@ mappings {
rule {
ruleName: "invertbooleannumber"
functionName: "invertbooleannumber"
+ outputIntName: "exclusive"
+ outputIntName: "reverse"
inputBooleanName: "exclusive"
inputBooleanName: "reverse"
- outputBooleanName: "exclusive"
- outputBooleanName: "reverse"
inputToOutput {
key: "exclusive"
value: "exclusive"
@@ -6815,6 +6819,7 @@ mappings {
rule {
ruleName: "ndarrayinputtonumericalattribute"
functionName: "ndarrayinputtonumericalattribute"
+ outputIntName: "numSegments"
inputToOutput {
key: "numSegments"
value: "num_segments"
@@ -7065,6 +7070,7 @@ mappings {
rule {
ruleName: "ndarrayinputtonumericalattribute"
functionName: "ndarrayinputtonumericalattribute"
+ outputDoubleName: "start"
outputDoubleName: "stop"
inputToOutput {
key: "start"
@@ -7227,6 +7233,7 @@ mappings {
rule {
ruleName: "ndarrayinputtonumericalattribute"
functionName: "ndarrayinputtonumericalattribute"
+ outputIntName: "numSegments"
inputToOutput {
key: "numSegments"
value: "num_segments"
@@ -8895,7 +8902,7 @@ mappings {
inputTensorName: "max_output_size"
outputTensorName: "boxes"
outputTensorName: "scales"
- outputTensorName: "overlayThreshold"
+ outputTensorName: "iouThreshold"
outputTensorName: "maxOutputSize"
inputToOutput {
key: "boxes"
@@ -8906,7 +8913,7 @@ mappings {
value: "scores"
}
inputToOutput {
- key: "overlayThreshold"
+ key: "iouThreshold"
value: "iou_threshold"
}
inputToOutput {
@@ -8935,6 +8942,7 @@ mappings {
rule {
ruleName: "ndarrayinputtonumericalattribute"
functionName: "ndarrayinputtonumericalattribute"
+ outputIntName: "maxOutputSize"
inputToOutput {
key: "maxOutputSize"
value: "max_output_size"
@@ -9014,6 +9022,9 @@ mappings {
rule {
ruleName: "ndarrayinputtonumericalattribute"
functionName: "ndarrayinputtonumericalattribute"
+ outputIntName: "depth"
+ outputDoubleName: "on"
+ outputDoubleName: "off"
inputToOutput {
key: "on"
value: "on_value"
@@ -9226,6 +9237,7 @@ mappings {
rule {
ruleName: "ndarrayinputtonumericalattribute"
functionName: "ndarrayinputtonumericalattribute"
+ outputIntName: "numSegments"
inputToOutput {
key: "numSegments"
value: "num_segments"
@@ -9285,13 +9297,13 @@ mappings {
inputBooleanName: "align_corners"
inputBooleanName: "half_pixel_centers"
outputBooleanName: "alignCorners"
- outputBooleanName: "halfPixelCenters"
+ outputBooleanName: "halfPixelCenter"
inputToOutput {
key: "alignCorners"
value: "align_corners"
}
inputToOutput {
- key: "halfPixelCenters"
+ key: "halfPixelCenter"
value: "half_pixel_centers"
}
ruleType: "attribute"
@@ -9689,7 +9701,7 @@ mappings {
functionName: "valuemapping"
inputFloatName: "iou_threshold"
inputToOutput {
- key: "overlayThreshold"
+ key: "iouThreshold"
value: "iou_threshold"
}
ruleType: "attribute"
@@ -9698,6 +9710,7 @@ mappings {
rule {
ruleName: "ndarrayinputtonumericalattribute"
functionName: "ndarrayinputtonumericalattribute"
+ outputIntName: "maxOutputSize"
inputToOutput {
key: "maxOutputSize"
value: "max_output_size"
@@ -9923,6 +9936,7 @@ mappings {
rule {
ruleName: "ndarraytointattributevalue"
functionName: "ndarraytointattributevalue"
+ outputIntName: "dimensions"
inputToOutput {
key: "dimensions"
value: "reduction_indices"
@@ -10163,6 +10177,7 @@ mappings {
rule {
ruleName: "ndarraytointattributevalue"
functionName: "ndarraytointattributevalue"
+ outputIntName: "dimensions"
inputToOutput {
key: "dimensions"
value: "reduction_indices"
@@ -10323,6 +10338,8 @@ mappings {
rule {
ruleName: "ndarrayinputtonumericalattribute"
functionName: "ndarrayinputtonumericalattribute"
+ outputDoubleName: "min"
+ outputDoubleName: "max"
inputToOutput {
key: "min"
value: "minval"
@@ -11779,6 +11796,7 @@ mappings {
rule {
ruleName: "ndarraytointattributevalue"
functionName: "ndarraytointattributevalue"
+ outputIntName: "dimensions"
inputToOutput {
key: "dimensions"
value: "reduction_indices"
@@ -12463,6 +12481,7 @@ mappings {
rule {
ruleName: "ndarrayinputtonumericalattribute"
functionName: "ndarrayinputtonumericalattribute"
+ outputIntName: "concatDimension"
inputToOutput {
key: "concatDimension"
value: "concat_dim"
@@ -12958,8 +12977,8 @@ mappings {
rule {
ruleName: "invertbooleannumber"
functionName: "invertbooleannumber"
+ outputIntName: "reverse"
inputBooleanName: "reverse"
- outputBooleanName: "reverse"
inputToOutput {
key: "reverse"
value: "reverse"
@@ -13194,6 +13213,9 @@ mappings {
rule {
ruleName: "ndarrayinputtonumericalattribute"
functionName: "ndarrayinputtonumericalattribute"
+ outputIntName: "from"
+ outputIntName: "to"
+ outputIntName: "step"
inputToOutput {
key: "from"
value: "start"
@@ -14358,6 +14380,7 @@ mappings {
rule {
ruleName: "ndarraytointattributevalue"
functionName: "ndarraytointattributevalue"
+ outputIntName: "dimensions"
inputToOutput {
key: "dimensions"
value: "reduction_indices"
@@ -14580,6 +14603,7 @@ mappings {
rule {
ruleName: "ndarrayinputtonumericalattribute"
functionName: "ndarrayinputtonumericalattribute"
+ outputIntName: "concatDimension"
inputToOutput {
key: "concatDimension"
value: "axis"
@@ -14689,6 +14713,7 @@ mappings {
rule {
ruleName: "ndarraytointattributevalue"
functionName: "ndarraytointattributevalue"
+ outputIntName: "dimensions"
inputToOutput {
key: "dimensions"
value: "reduction_indices"
@@ -15391,8 +15416,10 @@ mappings {
functionName: "ndarraymapping"
inputTensorName: "ref"
inputTensorName: "updates"
+ inputTensorName: "indices"
outputTensorName: "operand"
outputTensorName: "updates"
+ outputTensorName: "indices"
inputToOutput {
key: "operand"
value: "ref"
@@ -15401,18 +15428,11 @@ mappings {
key: "updates"
value: "updates"
}
- ruleType: "tensor"
- inputFrameworkOpName: "ScatterUpdate"
- }
- rule {
- ruleName: "ndarraytointattributevalue"
- functionName: "ndarraytointattributevalue"
- outputIntName: "indices"
inputToOutput {
key: "indices"
value: "indices"
}
- ruleType: "attribute"
+ ruleType: "tensor"
inputFrameworkOpName: "ScatterUpdate"
}
}
@@ -15539,8 +15559,10 @@ mappings {
functionName: "ndarraymapping"
inputTensorName: "tensor"
inputTensorName: "updates"
+ inputTensorName: "indices"
outputTensorName: "operand"
outputTensorName: "updates"
+ outputTensorName: "indices"
inputToOutput {
key: "operand"
value: "tensor"
@@ -15549,18 +15571,11 @@ mappings {
key: "updates"
value: "updates"
}
- ruleType: "tensor"
- inputFrameworkOpName: "TensorScatterUpdate"
- }
- rule {
- ruleName: "ndarraytointattributevalue"
- functionName: "ndarraytointattributevalue"
- outputIntName: "indices"
inputToOutput {
key: "indices"
value: "indices"
}
- ruleType: "attribute"
+ ruleType: "tensor"
inputFrameworkOpName: "TensorScatterUpdate"
}
}
@@ -15647,10 +15662,10 @@ mappings {
rule {
ruleName: "invertbooleannumber"
functionName: "invertbooleannumber"
+ outputIntName: "exclusive"
+ outputIntName: "reverse"
inputBooleanName: "exclusive"
inputBooleanName: "reverse"
- outputBooleanName: "exclusive"
- outputBooleanName: "reverse"
inputToOutput {
key: "exclusive"
value: "exclusive"