opList { name: "Assert" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "input" argType: INPUT_TENSOR } opDeclarationType: OP_IMPL } 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: "thresholdRelative" argType: DOUBLE } argDescriptor { name: "thresholdAbsolute" 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 } argDescriptor { name: "y" argType: INPUT_TENSOR argIndex: 1 } argDescriptor { name: "dataType" argType: DATA_TYPE } argDescriptor { name: "threshold" argType: DOUBLE } opDeclarationType: LEGACY_XYZ } opList { name: "ClipByValue" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "inPlace" argType: BOOL } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "clipValueMin" argType: DOUBLE } argDescriptor { name: "clipValueMax" argType: DOUBLE argIndex: 1 } } opList { name: "Conditional" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } opDeclarationType: LOGIC_OP_IMPL } opList { name: "ExternalErrorsFn" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } } opList { name: "Floor" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "input" argType: INPUT_TENSOR } opDeclarationType: OP_IMPL } opList { name: "Log1p" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "input" argType: INPUT_TENSOR } opDeclarationType: OP_IMPL } opList { name: "ParallelConcat" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } } opList { name: "Pow" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "y" argType: INPUT_TENSOR argIndex: 1 } opDeclarationType: BROADCASTABLE_OP_IMPL } opList { name: "Pow_bp" argDescriptor { name: "dLdx" argType: OUTPUT_TENSOR } argDescriptor { name: "dLdy" argType: OUTPUT_TENSOR argIndex: 1 } argDescriptor { name: "inPlace" argType: BOOL } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "y" argType: INPUT_TENSOR argIndex: 1 } argDescriptor { name: "dLdz" argType: INPUT_TENSOR argIndex: 2 } argDescriptor { name: "dLdx" argType: INPUT_TENSOR argIndex: 3 } argDescriptor { name: "dLdy" argType: INPUT_TENSOR argIndex: 4 } argDescriptor { name: "dtype" argType: DATA_TYPE } } opList { name: "Reciprocal" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "dataType" argType: DATA_TYPE } opDeclarationType: LEGACY_XYZ } opList { name: "RelativeError" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "inPlace" argType: BOOL } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "y" argType: INPUT_TENSOR argIndex: 1 } argDescriptor { name: "dataType" argType: DATA_TYPE } opDeclarationType: LEGACY_XYZ } opList { name: "Return" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } opDeclarationType: LOGIC_OP_IMPL } opList { name: "Scope" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } opDeclarationType: LOGIC_OP_IMPL } opList { name: "Switch" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "condition" argType: INPUT_TENSOR argIndex: 1 } opDeclarationType: DIVERGENT_OP_IMPL } opList { name: "Where" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "condition" argType: INPUT_TENSOR } argDescriptor { name: "input" argType: INPUT_TENSOR argIndex: 1 } argDescriptor { name: "y" argType: INPUT_TENSOR argIndex: 2 } argDescriptor { name: "dtype" argType: DATA_TYPE } } opList { name: "While" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "isConstant" argType: BOOL } argDescriptor { name: "condition" argType: INPUT_TENSOR } argDescriptor { name: "frameName" argType: STRING } opDeclarationType: LOGIC_OP_IMPL } opList { name: "_geluderivative" 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: "_mishderivative" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "inPlace" argType: BOOL } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "y" argType: INPUT_TENSOR argIndex: 1 } argDescriptor { name: "dataType" argType: DATA_TYPE } opDeclarationType: LEGACY_XYZ } opList { name: "_powderivative" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "inPlace" argType: BOOL } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "pow" argType: DOUBLE } opDeclarationType: LEGACY_XYZ } opList { name: "_precise_geluderivative" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "inPlace" argType: BOOL } argDescriptor { name: "precise" argType: BOOL argIndex: 1 } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "dataType" argType: DATA_TYPE } opDeclarationType: LEGACY_XYZ } opList { name: "_sigmoidderivative" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "inPlace" argType: BOOL } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "y" argType: INPUT_TENSOR argIndex: 1 } argDescriptor { name: "dataType" argType: DATA_TYPE } opDeclarationType: LEGACY_XYZ } opList { name: "_swishderivative" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "inPlace" argType: BOOL } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "y" argType: INPUT_TENSOR argIndex: 1 } argDescriptor { name: "dataType" argType: DATA_TYPE } opDeclarationType: LEGACY_XYZ } opList { name: "_tanhderivative" 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: "abs" 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: "absolute_difference_loss" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "predictions" argType: INPUT_TENSOR } argDescriptor { name: "weights" argType: INPUT_TENSOR argIndex: 1 } argDescriptor { name: "labels" argType: INPUT_TENSOR argIndex: 2 } argDescriptor { name: "reductionMode" argType: INT64 } argDescriptor { name: "dtype" argType: DATA_TYPE } } opList { name: "absolute_difference_loss_grad" argDescriptor { name: "dLdp" argType: OUTPUT_TENSOR } argDescriptor { name: "dLdw" argType: OUTPUT_TENSOR argIndex: 1 } argDescriptor { name: "dLdl" argType: OUTPUT_TENSOR argIndex: 2 } argDescriptor { name: "predictions" argType: INPUT_TENSOR } argDescriptor { name: "weights" argType: INPUT_TENSOR argIndex: 1 } argDescriptor { name: "labels" argType: INPUT_TENSOR argIndex: 2 } argDescriptor { name: "reductionMode" argType: INT64 } argDescriptor { name: "dtype" argType: DATA_TYPE } } opList { name: "acos" 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: "acosh" 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: "ada_delta_updater" argDescriptor { name: "update" argType: OUTPUT_TENSOR } argDescriptor { name: "stateMsg" argType: OUTPUT_TENSOR argIndex: 1 } argDescriptor { name: "stateMsdx" argType: OUTPUT_TENSOR argIndex: 2 } argDescriptor { name: "gradient" argType: INPUT_TENSOR } argDescriptor { name: "initStateMsg" argType: INPUT_TENSOR argIndex: 1 } argDescriptor { name: "initStateMsdx" argType: INPUT_TENSOR argIndex: 2 } argDescriptor { name: "rho" argType: INPUT_TENSOR argIndex: 3 } argDescriptor { name: "epsilon" argType: INPUT_TENSOR argIndex: 4 } argDescriptor { name: "updatedStateMsdx" argType: INPUT_TENSOR argIndex: 5 } argDescriptor { name: "dRho" argType: DOUBLE } argDescriptor { name: "dEpsilon" argType: DOUBLE argIndex: 1 } opDeclarationType: CONFIGURABLE_OP_IMPL } opList { name: "ada_grad_updater" argDescriptor { name: "update" argType: OUTPUT_TENSOR } argDescriptor { name: "stateH" argType: OUTPUT_TENSOR argIndex: 1 } argDescriptor { name: "gradient" argType: INPUT_TENSOR } argDescriptor { name: "initState" argType: INPUT_TENSOR argIndex: 1 } argDescriptor { name: "lr" argType: INPUT_TENSOR argIndex: 2 } argDescriptor { name: "epsilon" 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: "update" argType: OUTPUT_TENSOR } argDescriptor { name: "stateU" argType: OUTPUT_TENSOR argIndex: 1 } argDescriptor { name: "stateM" argType: OUTPUT_TENSOR argIndex: 2 } 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 } 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: "adam_updater" argDescriptor { name: "update" argType: OUTPUT_TENSOR } argDescriptor { name: "stateU" argType: OUTPUT_TENSOR argIndex: 1 } argDescriptor { name: "stateM" argType: OUTPUT_TENSOR argIndex: 2 } 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 } 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: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "inPlace" argType: BOOL } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "y" argType: INPUT_TENSOR argIndex: 1 } argDescriptor { name: "dataType" argType: DATA_TYPE } opDeclarationType: BROADCASTABLE_OP_IMPL } opList { name: "add_bp" argDescriptor { name: "gradX" argType: OUTPUT_TENSOR } argDescriptor { name: "gradY" argType: OUTPUT_TENSOR argIndex: 1 } argDescriptor { name: "inPlace" argType: BOOL } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "y" argType: INPUT_TENSOR argIndex: 1 } argDescriptor { name: "epsNext" argType: INPUT_TENSOR argIndex: 2 } argDescriptor { name: "dtype" argType: DATA_TYPE } } opList { name: "add_scalar" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "inPlace" argType: BOOL } argDescriptor { name: "input" argType: INPUT_TENSOR } opDeclarationType: LEGACY_XYZ } opList { name: "adjust_contrast" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "factor" argType: INPUT_TENSOR argIndex: 1 } opDeclarationType: CONFIGURABLE_OP_IMPL } opList { name: "adjust_contrast_v2" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "factor" argType: INPUT_TENSOR argIndex: 1 } argDescriptor { name: "factor" argType: DOUBLE } opDeclarationType: CONFIGURABLE_OP_IMPL } opList { name: "adjust_hue" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "delta" 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: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "factor" argType: INPUT_TENSOR argIndex: 1 } argDescriptor { name: "dimC" argType: INT64 } argDescriptor { name: "factor" argType: DOUBLE } opDeclarationType: CONFIGURABLE_OP_IMPL } opList { name: "all" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "keepDims" argType: BOOL } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "dimensions" argType: INPUT_TENSOR argIndex: 1 } argDescriptor { name: "dimensions" argType: INT64 } opDeclarationType: LEGACY_XYZ } opList { name: "alpha_dropout" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "a" argType: DOUBLE } argDescriptor { name: "b" argType: DOUBLE argIndex: 1 } argDescriptor { name: "alphaPrime" argType: DOUBLE argIndex: 2 } argDescriptor { name: "beta" argType: DOUBLE argIndex: 3 } 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: "probValue" argType: DOUBLE } argDescriptor { name: "alphaValue" argType: DOUBLE argIndex: 1 } argDescriptor { name: "alpha1Value" argType: DOUBLE argIndex: 2 } argDescriptor { name: "betaValue" argType: DOUBLE argIndex: 3 } opDeclarationType: CONFIGURABLE_OP_IMPL } opList { name: "amax" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "keepDims" argType: BOOL } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "dimensions" argType: INPUT_TENSOR argIndex: 1 } argDescriptor { name: "dimensions" argType: INT64 } opDeclarationType: LEGACY_XYZ } opList { name: "amax_pairwise" 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 } opDeclarationType: LEGACY_XYZ } opList { name: "amean" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "keepDims" argType: BOOL } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "dimensions" argType: INPUT_TENSOR argIndex: 1 } argDescriptor { name: "dimensions" argType: INT64 } opDeclarationType: LEGACY_XYZ } opList { name: "amin" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "keepDims" argType: BOOL } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "dimensions" argType: INPUT_TENSOR argIndex: 1 } argDescriptor { name: "dimensions" argType: INT64 } opDeclarationType: LEGACY_XYZ } opList { name: "amin_pairwise" 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 } opDeclarationType: LEGACY_XYZ } opList { name: "ams_grad_updater" argDescriptor { name: "update" argType: OUTPUT_TENSOR } argDescriptor { name: "stateV" argType: OUTPUT_TENSOR argIndex: 1 } argDescriptor { name: "stateM" argType: OUTPUT_TENSOR argIndex: 2 } argDescriptor { name: "stateH" argType: OUTPUT_TENSOR argIndex: 3 } argDescriptor { name: "gradient" argType: INPUT_TENSOR } argDescriptor { name: "initStateV" argType: INPUT_TENSOR argIndex: 1 } argDescriptor { name: "initStateM" argType: INPUT_TENSOR argIndex: 2 } argDescriptor { name: "initStateH" argType: INPUT_TENSOR argIndex: 3 } argDescriptor { name: "lr" argType: INPUT_TENSOR argIndex: 4 } argDescriptor { name: "beta1" argType: INPUT_TENSOR argIndex: 5 } argDescriptor { name: "beta2" argType: INPUT_TENSOR argIndex: 6 } argDescriptor { name: "epsilon" 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: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "inPlace" argType: BOOL } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "dataType" argType: DATA_TYPE } argDescriptor { name: "comparable" argType: DOUBLE } opDeclarationType: LEGACY_XYZ } opList { name: "and_scalar" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "inPlace" argType: BOOL } argDescriptor { name: "input" argType: INPUT_TENSOR } opDeclarationType: LEGACY_XYZ } opList { name: "any" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "keepDims" argType: BOOL } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "dimensions" argType: INPUT_TENSOR argIndex: 1 } argDescriptor { name: "dimensions" argType: INT64 } opDeclarationType: LEGACY_XYZ } opList { name: "apply_sgd" argDescriptor { name: "Z" argType: OUTPUT_TENSOR } argDescriptor { name: "parameters" argType: INPUT_TENSOR } argDescriptor { name: "gradients" argType: INPUT_TENSOR argIndex: 1 } argDescriptor { name: "tarr" argType: INPUT_TENSOR argIndex: 2 } argDescriptor { name: "lr" argType: DOUBLE } opDeclarationType: CONFIGURABLE_OP_IMPL } opList { name: "applygradientdescent" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } } opList { name: "argamax" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "keepDims" argType: BOOL } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "dimensions" argType: INPUT_TENSOR argIndex: 1 } argDescriptor { name: "dimensions" argType: INT64 } argDescriptor { name: "dtype" argType: DATA_TYPE } } opList { name: "argamin" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "keepDims" argType: BOOL } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "dimensions" argType: INPUT_TENSOR argIndex: 1 } argDescriptor { name: "dimensions" argType: INT64 } argDescriptor { name: "dtype" argType: DATA_TYPE } } opList { name: "argmax" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "keepDims" argType: BOOL } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "dimensions" argType: INPUT_TENSOR argIndex: 1 } argDescriptor { name: "dimensions" argType: INT64 } argDescriptor { name: "dtype" argType: DATA_TYPE } } opList { name: "argmin" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "keepDims" argType: BOOL } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "dimensions" argType: INPUT_TENSOR argIndex: 1 } argDescriptor { name: "dimensions" argType: INT64 } argDescriptor { name: "dtype" argType: DATA_TYPE } } opList { name: "asin" 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: "asinh" 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: "assign" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "y" argType: INPUT_TENSOR argIndex: 1 } opDeclarationType: BROADCASTABLE_OP_IMPL } opList { name: "assign_bp" argDescriptor { name: "gradX" argType: OUTPUT_TENSOR } argDescriptor { name: "gradY" argType: OUTPUT_TENSOR argIndex: 1 } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "y" argType: INPUT_TENSOR argIndex: 1 } argDescriptor { name: "epsNext" argType: INPUT_TENSOR argIndex: 2 } argDescriptor { name: "dtype" argType: DATA_TYPE } } opList { name: "asum" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "keepDims" argType: BOOL } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "dimensions" argType: INPUT_TENSOR argIndex: 1 } argDescriptor { name: "dimensions" argType: INT64 } opDeclarationType: LEGACY_XYZ } opList { name: "atan" 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: "atanh" 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: "avgpool2d" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "input" argType: INPUT_TENSOR } 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: "extraParam0" argType: INT64 argIndex: 9 } argDescriptor { name: "isNCHW" argType: INT64 argIndex: 10 } argDescriptor { name: "dtype" argType: DATA_TYPE } } 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 } 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: "extraParam0" argType: INT64 argIndex: 9 } argDescriptor { name: "isNCHW" argType: INT64 argIndex: 10 } argDescriptor { 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 } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "gradO" argType: INPUT_TENSOR argIndex: 1 } 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: "axpy" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "y" argType: INPUT_TENSOR argIndex: 1 } argDescriptor { name: "alpha" 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: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "rowP" argType: INPUT_TENSOR } argDescriptor { name: "colP" argType: INPUT_TENSOR argIndex: 1 } argDescriptor { name: "valP" argType: INPUT_TENSOR argIndex: 2 } argDescriptor { name: "dataP" argType: INPUT_TENSOR argIndex: 3 } argDescriptor { name: "N" argType: INT64 } argDescriptor { name: "dtype" argType: DATA_TYPE } } opList { name: "barnes_gains" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "gradX" argType: INPUT_TENSOR argIndex: 1 } argDescriptor { name: "epsilon" argType: INPUT_TENSOR argIndex: 2 } opDeclarationType: OP_IMPL } opList { name: "barnes_symmetrized" argDescriptor { name: "outputRows" argType: OUTPUT_TENSOR } argDescriptor { name: "outputCols" argType: OUTPUT_TENSOR argIndex: 1 } argDescriptor { name: "outputVals" argType: OUTPUT_TENSOR argIndex: 2 } argDescriptor { name: "rowP" argType: INPUT_TENSOR } argDescriptor { name: "colP" argType: INPUT_TENSOR argIndex: 1 } argDescriptor { name: "valP" argType: INPUT_TENSOR argIndex: 2 } argDescriptor { name: "outRows" 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 } argDescriptor { name: "croppingTop" argType: INT64 argIndex: 1 } argDescriptor { name: "croppingBottom" argType: INT64 argIndex: 2 } argDescriptor { name: "dtype" argType: DATA_TYPE } } opList { name: "batch_to_space_nd" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "inPlace" argType: BOOL } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "blockShape" argType: INPUT_TENSOR argIndex: 1 } argDescriptor { name: "crop" 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 } argDescriptor { name: "transB" argType: INT64 argIndex: 1 } argDescriptor { name: "M" argType: INT64 argIndex: 2 } argDescriptor { name: "N" argType: INT64 argIndex: 3 } argDescriptor { name: "K" argType: INT64 argIndex: 4 } argDescriptor { name: "ldA" argType: INT64 argIndex: 5 } argDescriptor { name: "ldB" argType: INT64 argIndex: 6 } argDescriptor { name: "ldC" argType: INT64 argIndex: 7 } argDescriptor { name: "batchSize" argType: INT64 argIndex: 8 } argDescriptor { name: "dtype" argType: DATA_TYPE } } opList { name: "batchnorm" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "inPlace" argType: BOOL } argDescriptor { name: "applyGamma" argType: BOOL argIndex: 1 } argDescriptor { name: "applyBeta" argType: BOOL argIndex: 2 } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "mean" argType: INPUT_TENSOR argIndex: 1 } argDescriptor { name: "variance" argType: INPUT_TENSOR argIndex: 2 } argDescriptor { name: "gamma" 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: "batchnorm_bp" argDescriptor { name: "dLdI" argType: OUTPUT_TENSOR } argDescriptor { name: "dLdM" argType: OUTPUT_TENSOR argIndex: 1 } argDescriptor { name: "dLdV" argType: OUTPUT_TENSOR argIndex: 2 } argDescriptor { name: "dLdG" argType: OUTPUT_TENSOR argIndex: 3 } argDescriptor { name: "inPlace" argType: BOOL } argDescriptor { name: "applyGamma" argType: BOOL argIndex: 1 } argDescriptor { name: "applyBeta" argType: BOOL argIndex: 2 } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "mean" argType: INPUT_TENSOR argIndex: 1 } argDescriptor { name: "variance" argType: INPUT_TENSOR argIndex: 2 } argDescriptor { name: "gamma" 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" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "a" argType: INPUT_TENSOR } argDescriptor { name: "b" argType: INPUT_TENSOR argIndex: 1 } argDescriptor { name: "input" argType: INPUT_TENSOR argIndex: 2 } opDeclarationType: CONFIGURABLE_OP_IMPL } opList { name: "biasadd" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "nchw" argType: BOOL } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "bias" argType: INPUT_TENSOR argIndex: 1 } argDescriptor { name: "dtype" argType: DATA_TYPE } } opList { name: "biasadd_bp" argDescriptor { name: "gradI" argType: OUTPUT_TENSOR } argDescriptor { name: "gradB" argType: OUTPUT_TENSOR argIndex: 1 } argDescriptor { name: "nchw" argType: BOOL } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "bias" argType: INPUT_TENSOR argIndex: 1 } argDescriptor { name: "gradO" argType: INPUT_TENSOR argIndex: 2 } argDescriptor { name: "dtype" argType: DATA_TYPE } } opList { name: "bincount" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "values" argType: INPUT_TENSOR } argDescriptor { name: "weights" argType: INPUT_TENSOR argIndex: 1 } argDescriptor { name: "min" argType: INPUT_TENSOR argIndex: 2 } argDescriptor { name: "max" 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 } argDescriptor { name: "dtype" argType: DATA_TYPE } } opList { name: "bits_hamming_distance" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "y" argType: INPUT_TENSOR argIndex: 1 } argDescriptor { name: "dtype" argType: DATA_TYPE } } opList { name: "bitwise_and" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "inPlace" argType: BOOL } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "y" argType: INPUT_TENSOR argIndex: 1 } argDescriptor { name: "dataType" argType: DATA_TYPE } opDeclarationType: BROADCASTABLE_OP_IMPL } opList { name: "bitwise_or" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "inPlace" argType: BOOL } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "y" argType: INPUT_TENSOR argIndex: 1 } argDescriptor { name: "dataType" argType: DATA_TYPE } opDeclarationType: BROADCASTABLE_OP_IMPL } opList { name: "bitwise_xor" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "inPlace" argType: BOOL } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "y" argType: INPUT_TENSOR argIndex: 1 } argDescriptor { name: "dataType" argType: DATA_TYPE } opDeclarationType: BROADCASTABLE_OP_IMPL } opList { name: "bool_not" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "dataType" argType: DATA_TYPE } opDeclarationType: LEGACY_XYZ } opList { name: "boolean_and" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "y" argType: INPUT_TENSOR argIndex: 1 } opDeclarationType: BROADCASTABLE_OP_IMPL } opList { name: "boolean_not" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "y" argType: INPUT_TENSOR argIndex: 1 } opDeclarationType: OP_IMPL } opList { name: "boolean_or" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "y" argType: INPUT_TENSOR argIndex: 1 } opDeclarationType: BROADCASTABLE_OP_IMPL } opList { name: "boolean_xor" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "y" argType: INPUT_TENSOR argIndex: 1 } opDeclarationType: BROADCASTABLE_OP_IMPL } opList { name: "broadcast_amax" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "inPlace" argType: BOOL } argDescriptor { 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: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "inPlace" argType: BOOL } argDescriptor { 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: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "y" argType: INPUT_TENSOR argIndex: 1 } argDescriptor { name: "dtype" argType: DATA_TYPE } } opList { name: "broadcast_equalto" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "inPlace" argType: BOOL } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "y" argType: INPUT_TENSOR argIndex: 1 } argDescriptor { name: "dimension" argType: INT64 } opDeclarationType: LEGACY_XYZ } opList { name: "broadcast_greaterthan" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "inPlace" argType: BOOL } argDescriptor { 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: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "inPlace" argType: BOOL } argDescriptor { 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: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "inPlace" argType: BOOL } argDescriptor { 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: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "inPlace" argType: BOOL } argDescriptor { 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: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "inPlace" argType: BOOL } argDescriptor { 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: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "inPlace" argType: BOOL } argDescriptor { 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: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "inPlace" argType: BOOL } argDescriptor { 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: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "shape" argType: INPUT_TENSOR argIndex: 1 } argDescriptor { name: "dtype" argType: DATA_TYPE } } opList { name: "broadcastadd" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "y" argType: INPUT_TENSOR argIndex: 1 } argDescriptor { name: "dimension" argType: INT64 } opDeclarationType: LEGACY_XYZ } opList { name: "broadcastcopy" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "inPlace" argType: BOOL } argDescriptor { 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: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "inPlace" argType: BOOL } argDescriptor { 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 } argDescriptor { name: "inPlace" argType: BOOL } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "y" argType: INPUT_TENSOR argIndex: 1 } argDescriptor { name: "dimension" argType: INT64 } opDeclarationType: OP_IMPL } opList { name: "broadcastmul" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "inPlace" argType: BOOL } argDescriptor { 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: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "inPlace" argType: BOOL } argDescriptor { 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: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "inPlace" argType: BOOL } argDescriptor { 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 } argDescriptor { name: "dimension" 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 } argDescriptor { name: "inPlace" argType: BOOL } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "dst" argType: INT64 } argDescriptor { name: "dtype" argType: DATA_TYPE } } opList { name: "cbow" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "trainWords" argType: BOOL } argDescriptor { name: "isInference" argType: BOOL argIndex: 1 } argDescriptor { name: "target" argType: INPUT_TENSOR } argDescriptor { name: "ngStarter" argType: INPUT_TENSOR argIndex: 1 } argDescriptor { name: "context" argType: INPUT_TENSOR argIndex: 2 } argDescriptor { name: "indices" argType: INPUT_TENSOR argIndex: 3 } argDescriptor { name: "codes" argType: INPUT_TENSOR argIndex: 4 } argDescriptor { name: "syn0" argType: INPUT_TENSOR argIndex: 5 } argDescriptor { name: "syn1" argType: INPUT_TENSOR argIndex: 6 } argDescriptor { name: "syn1neg" argType: INPUT_TENSOR argIndex: 7 } argDescriptor { name: "expTable" argType: INPUT_TENSOR argIndex: 8 } argDescriptor { name: "negTable" argType: INPUT_TENSOR argIndex: 9 } argDescriptor { name: "alpha" argType: INPUT_TENSOR argIndex: 10 } argDescriptor { name: "randomValue" argType: INPUT_TENSOR argIndex: 11 } argDescriptor { name: "numLabels" argType: INPUT_TENSOR argIndex: 12 } argDescriptor { name: "lockedWords" argType: INPUT_TENSOR argIndex: 13 } argDescriptor { name: "inferenceVector" 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: "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: "cell_contains" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "contains" argType: BOOL } argDescriptor { name: "corner" argType: INPUT_TENSOR } argDescriptor { name: "width" argType: INPUT_TENSOR argIndex: 1 } argDescriptor { name: "point" argType: INPUT_TENSOR argIndex: 2 } argDescriptor { name: "dimension" argType: INT64 } argDescriptor { name: "dtype" argType: DATA_TYPE } } opList { name: "check_numerics" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "message" argType: INPUT_TENSOR argIndex: 1 } argDescriptor { name: "dtype" argType: DATA_TYPE } } opList { name: "choice" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "source" argType: INPUT_TENSOR } argDescriptor { name: "probabilities" argType: INPUT_TENSOR argIndex: 1 } opDeclarationType: LEGACY_XYZ } opList { name: "cholesky" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "input" argType: INPUT_TENSOR } opDeclarationType: OP_IMPL } opList { name: "choose" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "numResults" argType: OUTPUT_TENSOR argIndex: 1 } argDescriptor { name: "arg" argType: INPUT_TENSOR } argDescriptor { name: "comp" argType: INPUT_TENSOR argIndex: 1 } argDescriptor { name: "mode" argType: INT64 } 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 } } opList { name: "clipbyavgnorm" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "dimensions" argType: INT64 } argDescriptor { name: "clipNorm" argType: DOUBLE } opDeclarationType: CONFIGURABLE_OP_IMPL } opList { name: "clipbyavgnorm_bp" argDescriptor { name: "gradI" argType: OUTPUT_TENSOR } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "gradO" 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: "clipValue" argType: DOUBLE } opDeclarationType: CONFIGURABLE_OP_IMPL } opList { name: "clipbynorm_bp" argDescriptor { name: "gradI" argType: OUTPUT_TENSOR } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "gradO" argType: INPUT_TENSOR argIndex: 1 } argDescriptor { name: "dimensions" argType: INT64 } argDescriptor { name: "dtype" argType: DATA_TYPE } argDescriptor { name: "clipValue" argType: DOUBLE } } opList { name: "clipbyvalue" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "left" argType: DOUBLE } argDescriptor { name: "right" argType: DOUBLE argIndex: 1 } opDeclarationType: CONFIGURABLE_OP_IMPL } opList { name: "clone_list" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "list" argType: INPUT_TENSOR } opDeclarationType: LIST_OP_IMPL } 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 } argDescriptor { name: "strideX" argType: INT64 argIndex: 1 } argDescriptor { name: "padHeight" argType: INT64 argIndex: 2 } argDescriptor { name: "padWidth" argType: INT64 argIndex: 3 } argDescriptor { name: "imgHeight" argType: INT64 argIndex: 4 } argDescriptor { name: "imgWidth" argType: INT64 argIndex: 5 } argDescriptor { name: "dY" argType: INT64 argIndex: 6 } argDescriptor { name: "dX" argType: INT64 argIndex: 7 } argDescriptor { name: "dtype" argType: DATA_TYPE } } opList { name: "compare_and_bitpack" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "y" argType: INPUT_TENSOR argIndex: 1 } argDescriptor { name: "dtype" argType: DATA_TYPE } argDescriptor { name: "threshold" argType: DOUBLE } } opList { name: "compat_sparse_to_dense" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "indices" argType: INPUT_TENSOR } argDescriptor { name: "shape" argType: INPUT_TENSOR argIndex: 1 } argDescriptor { name: "values" argType: INPUT_TENSOR argIndex: 2 } argDescriptor { name: "def" argType: INPUT_TENSOR argIndex: 3 } argDescriptor { name: "dtype" argType: DATA_TYPE } } opList { name: "compat_string_split" argDescriptor { name: "indices" argType: OUTPUT_TENSOR } argDescriptor { name: "values" argType: OUTPUT_TENSOR argIndex: 1 } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "delim" argType: INPUT_TENSOR argIndex: 1 } argDescriptor { name: "indices" argType: INPUT_TENSOR argIndex: 2 } argDescriptor { name: "values" argType: INPUT_TENSOR argIndex: 3 } argDescriptor { name: "dtype" argType: DATA_TYPE } } opList { name: "concat" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "isDynamicAxis" argType: BOOL } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "concatDimension" argType: INPUT_TENSOR argIndex: 1 } argDescriptor { name: "concatDimension" argType: INT64 } argDescriptor { name: "dtype" argType: DATA_TYPE } } opList { name: "concat_bp" argDescriptor { name: "epsilonChunk" argType: OUTPUT_TENSOR } argDescriptor { name: "dynamicAxis" argType: BOOL } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "originalChunk" argType: INPUT_TENSOR argIndex: 1 } argDescriptor { name: "concatDimension" argType: INT64 } argDescriptor { name: "dtype" argType: DATA_TYPE } } opList { name: "confusion_matrix" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "labels" argType: INPUT_TENSOR } argDescriptor { name: "predictions" argType: INPUT_TENSOR argIndex: 1 } argDescriptor { name: "weights" 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 } argDescriptor { name: "sW" argType: INT64 argIndex: 1 } argDescriptor { name: "pW" argType: INT64 argIndex: 2 } argDescriptor { name: "dW" argType: INT64 argIndex: 3 } argDescriptor { name: "paddingMode" argType: INT64 argIndex: 4 } argDescriptor { name: "isNCW" argType: INT64 argIndex: 5 } argDescriptor { name: "wFormat" argType: INT64 argIndex: 6 } argDescriptor { name: "dtype" argType: DATA_TYPE } } 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 } argDescriptor { name: "sW" argType: INT64 argIndex: 1 } argDescriptor { name: "pW" argType: INT64 argIndex: 2 } argDescriptor { name: "dW" argType: INT64 argIndex: 3 } argDescriptor { name: "paddingMode" argType: INT64 argIndex: 4 } argDescriptor { name: "isNCW" argType: INT64 argIndex: 5 } argDescriptor { name: "wFormat" argType: INT64 argIndex: 6 } argDescriptor { 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 } 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: "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_input_bp" argDescriptor { name: "gradI" argType: OUTPUT_TENSOR } argDescriptor { name: "gradIShape" argType: INPUT_TENSOR } argDescriptor { name: "weights" argType: INPUT_TENSOR argIndex: 1 } argDescriptor { name: "gradO" 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 } 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: "conv3dnew_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: "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: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "inPlace" argType: BOOL } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "y" argType: INPUT_TENSOR argIndex: 1 } argDescriptor { name: "dataType" argType: DATA_TYPE } opDeclarationType: LEGACY_XYZ } opList { name: "cos" 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: "cosh" 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: "cosine_distance_loss" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "predictions" argType: INPUT_TENSOR } argDescriptor { name: "weights" argType: INPUT_TENSOR argIndex: 1 } argDescriptor { name: "labels" 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: "cosine_distance_loss_grad" argDescriptor { name: "dLdp" argType: OUTPUT_TENSOR } argDescriptor { name: "dLdw" argType: OUTPUT_TENSOR argIndex: 1 } argDescriptor { name: "dLdl" argType: OUTPUT_TENSOR argIndex: 2 } argDescriptor { name: "predictions" argType: INPUT_TENSOR } argDescriptor { name: "weights" argType: INPUT_TENSOR argIndex: 1 } argDescriptor { name: "labels" 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: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "keepDims" argType: BOOL } argDescriptor { name: "allDistances" argType: BOOL argIndex: 1 } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "y" argType: INPUT_TENSOR argIndex: 1 } argDescriptor { name: "dimensions" argType: INT64 } opDeclarationType: LEGACY_XYZ } opList { name: "cosinesimilarity" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "keepDims" argType: BOOL } argDescriptor { name: "allDistances" argType: BOOL argIndex: 1 } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "y" argType: INPUT_TENSOR argIndex: 1 } argDescriptor { name: "dimensions" argType: INT64 } opDeclarationType: LEGACY_XYZ } opList { name: "countNonZero" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "keepDims" argType: BOOL } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "dimensions" argType: INPUT_TENSOR argIndex: 1 } argDescriptor { name: "dimensions" argType: INT64 } opDeclarationType: LEGACY_XYZ } opList { name: "countZero" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "keepDims" argType: BOOL } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "dimensions" 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 } argDescriptor { name: "outputType" argType: INT64 argIndex: 1 } argDescriptor { name: "outputType" argType: DATA_TYPE } } opList { name: "create_list" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "height" argType: INT64 } argDescriptor { name: "expandable" argType: INT64 argIndex: 1 } opDeclarationType: LIST_OP_IMPL } opList { name: "crelu" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "dtype" argType: DATA_TYPE } } opList { name: "crelu_bp" argDescriptor { name: "epsilon" argType: OUTPUT_TENSOR } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "epsilonNext" argType: INPUT_TENSOR argIndex: 1 } argDescriptor { name: "dtype" argType: DATA_TYPE } } opList { name: "crop_and_resize" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "image" argType: INPUT_TENSOR } argDescriptor { name: "boxes" argType: INPUT_TENSOR argIndex: 1 } argDescriptor { name: "boxIndexes" argType: INPUT_TENSOR argIndex: 2 } argDescriptor { name: "newImageSize" argType: INPUT_TENSOR argIndex: 3 } argDescriptor { name: "method" argType: INT64 } argDescriptor { name: "dtype" argType: DATA_TYPE } argDescriptor { name: "extrapolationVal" argType: DOUBLE } } opList { name: "cross" argDescriptor { name: "o" argType: OUTPUT_TENSOR } argDescriptor { name: "a" argType: INPUT_TENSOR } argDescriptor { name: "b" argType: INPUT_TENSOR argIndex: 1 } opDeclarationType: OP_IMPL } opList { name: "cube" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "inPlace" argType: BOOL } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "dataType" argType: DATA_TYPE } opDeclarationType: CONFIGURABLE_OP_IMPL } opList { name: "cube_bp" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "epsilon" argType: INPUT_TENSOR argIndex: 1 } opDeclarationType: CONFIGURABLE_OP_IMPL } opList { name: "cubederivative" 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: "cumprod" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "exclusive" argType: BOOL } argDescriptor { name: "reverse" argType: BOOL argIndex: 1 } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "dimensions" 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: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "exclusive" argType: BOOL } argDescriptor { name: "reverse" argType: BOOL argIndex: 1 } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "dimensions" argType: INPUT_TENSOR argIndex: 1 } argDescriptor { name: "gradOut" 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: "cumsum" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "exclusive" argType: BOOL } argDescriptor { name: "reverse" argType: BOOL argIndex: 1 } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "dimensions" 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: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "exclusive" argType: BOOL } argDescriptor { name: "reverse" argType: BOOL argIndex: 1 } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "dimensions" argType: INPUT_TENSOR argIndex: 1 } argDescriptor { name: "gradOut" 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: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "inPlace" argType: BOOL } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "y" argType: INPUT_TENSOR argIndex: 1 } argDescriptor { name: "dataType" argType: DATA_TYPE } opDeclarationType: BROADCASTABLE_OP_IMPL } opList { name: "cyclic_shift_bits" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "inPlace" argType: BOOL } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "y" argType: INPUT_TENSOR argIndex: 1 } argDescriptor { name: "dataType" argType: DATA_TYPE } opDeclarationType: BROADCASTABLE_OP_IMPL } opList { name: "decode_bitmap" argDescriptor { name: "updates" argType: OUTPUT_TENSOR } argDescriptor { name: "start" argType: INPUT_TENSOR } argDescriptor { name: "encoded" argType: INPUT_TENSOR argIndex: 1 } argDescriptor { name: "dtype" argType: DATA_TYPE } } opList { name: "decode_threshold" argDescriptor { name: "updates" argType: OUTPUT_TENSOR } argDescriptor { name: "weights" argType: INPUT_TENSOR } argDescriptor { name: "encoded" 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 } 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: "deconv2d_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: "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: "deconv2d_tf" argDescriptor { name: "gradI" argType: OUTPUT_TENSOR } argDescriptor { name: "gradIShape" argType: INPUT_TENSOR } argDescriptor { name: "weights" argType: INPUT_TENSOR argIndex: 1 } argDescriptor { name: "gradO" 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 } 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_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: "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" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "shape" argType: INPUT_TENSOR } argDescriptor { name: "weights" argType: INPUT_TENSOR argIndex: 1 } argDescriptor { name: "input" argType: INPUT_TENSOR argIndex: 2 } } opList { name: "depth_to_space" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "block_size" argType: INT64 } argDescriptor { name: "isNHWC" argType: INT64 argIndex: 1 } argDescriptor { name: "dtype" argType: DATA_TYPE } } opList { name: "depthwise_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: "depthwise_conv2d_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: "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: "diag" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "inPlace" argType: BOOL } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "dtype" argType: DATA_TYPE } } opList { name: "diag_part" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "inPlace" argType: BOOL } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "dtype" argType: DATA_TYPE } } opList { name: "digamma" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "input" argType: INPUT_TENSOR } opDeclarationType: CONFIGURABLE_OP_IMPL } opList { name: "dilation2d" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "isSameMode" argType: BOOL } argDescriptor { name: "inPlace" argType: BOOL argIndex: 1 } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "weights" argType: INPUT_TENSOR argIndex: 1 } argDescriptor { name: "r" argType: INPUT_TENSOR argIndex: 2 } argDescriptor { name: "s" 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 } argDescriptor { name: "shape" argType: INT64 argIndex: 1 } argDescriptor { name: "prob" argType: DOUBLE } opDeclarationType: LEGACY_XYZ } opList { name: "distribution_binomial" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "trials" argType: INT64 } argDescriptor { name: "dataType" argType: INT64 argIndex: 1 } argDescriptor { name: "shape" argType: INT64 argIndex: 2 } argDescriptor { name: "probability" argType: DOUBLE } opDeclarationType: LEGACY_XYZ } opList { name: "distribution_binomial_ex" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "probabilities" argType: INPUT_TENSOR } argDescriptor { name: "trials" argType: INT64 } argDescriptor { name: "probability" argType: DOUBLE } opDeclarationType: LEGACY_XYZ } opList { name: "distribution_gaussian" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "dataType" argType: INT64 } argDescriptor { name: "shape" argType: INT64 argIndex: 1 } argDescriptor { name: "mean" argType: DOUBLE } argDescriptor { name: "stddev" argType: DOUBLE argIndex: 1 } opDeclarationType: LEGACY_XYZ } opList { name: "distribution_lognormal" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "dataType" argType: INT64 } argDescriptor { name: "shape" argType: INT64 argIndex: 1 } argDescriptor { name: "mean" argType: DOUBLE } argDescriptor { name: "stdev" argType: DOUBLE argIndex: 1 } opDeclarationType: LEGACY_XYZ } opList { name: "distribution_truncated" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "dataType" argType: INT64 } argDescriptor { name: "shape" argType: INT64 argIndex: 1 } argDescriptor { name: "mean" argType: DOUBLE } argDescriptor { name: "stddev" argType: DOUBLE argIndex: 1 } opDeclarationType: LEGACY_XYZ } opList { name: "distribution_uniform" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "dataType" argType: INT64 } argDescriptor { name: "shape" argType: INT64 argIndex: 1 } argDescriptor { name: "from" argType: DOUBLE } argDescriptor { name: "to" argType: DOUBLE argIndex: 1 } opDeclarationType: LEGACY_XYZ } opList { name: "div_scalar" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "y" argType: INPUT_TENSOR argIndex: 1 } opDeclarationType: LEGACY_XYZ } opList { name: "divide" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "inPlace" argType: BOOL } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "y" argType: INPUT_TENSOR argIndex: 1 } argDescriptor { name: "dataType" argType: DATA_TYPE } opDeclarationType: BROADCASTABLE_OP_IMPL } opList { name: "divide_bp" argDescriptor { name: "gradX" argType: OUTPUT_TENSOR } argDescriptor { name: "gradY" argType: OUTPUT_TENSOR argIndex: 1 } argDescriptor { name: "inPlace" argType: BOOL } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "y" argType: INPUT_TENSOR argIndex: 1 } argDescriptor { name: "epsNext" argType: INPUT_TENSOR argIndex: 2 } argDescriptor { name: "dtype" argType: DATA_TYPE } } opList { name: "divide_no_nan" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "y" argType: INPUT_TENSOR argIndex: 1 } opDeclarationType: BROADCASTABLE_OP_IMPL } opList { name: "dot" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "newFormat" argType: BOOL } argDescriptor { name: "keepDims" argType: BOOL argIndex: 1 } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "y" argType: INPUT_TENSOR argIndex: 1 } argDescriptor { name: "dimensions" argType: INT64 } opDeclarationType: LEGACY_XYZ } opList { name: "dot_product_attention" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "weights" argType: OUTPUT_TENSOR argIndex: 1 } argDescriptor { name: "scaled" argType: BOOL } argDescriptor { name: "withWeights" argType: BOOL argIndex: 1 } argDescriptor { name: "queries" argType: INPUT_TENSOR } argDescriptor { name: "keys" argType: INPUT_TENSOR argIndex: 1 } argDescriptor { name: "values" argType: INPUT_TENSOR argIndex: 2 } argDescriptor { name: "mask" argType: INPUT_TENSOR argIndex: 3 } 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 } argDescriptor { name: "dLdk" argType: OUTPUT_TENSOR argIndex: 1 } argDescriptor { name: "dLdv" argType: OUTPUT_TENSOR argIndex: 2 } argDescriptor { name: "scaled" argType: BOOL } argDescriptor { name: "queries" argType: INPUT_TENSOR } argDescriptor { name: "keys" argType: INPUT_TENSOR argIndex: 1 } argDescriptor { name: "values" argType: INPUT_TENSOR argIndex: 2 } argDescriptor { name: "eps" argType: INPUT_TENSOR argIndex: 3 } argDescriptor { name: "mask" argType: INPUT_TENSOR argIndex: 4 } argDescriptor { name: "normalization" argType: INT64 } argDescriptor { name: "dtype" argType: DATA_TYPE } } opList { name: "draw_bounding_boxes" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "images" argType: INPUT_TENSOR } argDescriptor { name: "boxes" argType: INPUT_TENSOR argIndex: 1 } argDescriptor { name: "colors" argType: INPUT_TENSOR argIndex: 2 } opDeclarationType: OP_IMPL } opList { name: "dropout" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "reduceShape" 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: "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: "probValue" argType: DOUBLE } opDeclarationType: CONFIGURABLE_OP_IMPL } opList { name: "dropout_inverted" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "p" argType: DOUBLE } opDeclarationType: LEGACY_XYZ } opList { name: "dynamic_bidirectional_rnn" argDescriptor { name: "hFW" argType: OUTPUT_TENSOR } argDescriptor { name: "hBW" argType: OUTPUT_TENSOR argIndex: 1 } argDescriptor { name: "hFWFinal" argType: OUTPUT_TENSOR argIndex: 2 } argDescriptor { name: "hBWFinal" argType: OUTPUT_TENSOR argIndex: 3 } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "WxFW" argType: INPUT_TENSOR argIndex: 1 } argDescriptor { name: "WhFW" argType: INPUT_TENSOR argIndex: 2 } argDescriptor { name: "bFW" argType: INPUT_TENSOR argIndex: 3 } argDescriptor { name: "WxBW" argType: INPUT_TENSOR argIndex: 4 } argDescriptor { name: "WhBW" argType: INPUT_TENSOR argIndex: 5 } argDescriptor { name: "bBW" argType: INPUT_TENSOR argIndex: 6 } argDescriptor { name: "h0FW" argType: INPUT_TENSOR argIndex: 7 } argDescriptor { name: "h0BW" argType: INPUT_TENSOR argIndex: 8 } argDescriptor { name: "timeMajor" argType: INT64 } argDescriptor { name: "dtype" argType: DATA_TYPE } } opList { name: "dynamic_partition" argDescriptor { name: "outputList" argType: OUTPUT_TENSOR } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "indices" argType: INPUT_TENSOR argIndex: 1 } argDescriptor { name: "numPartitions" argType: INT64 } argDescriptor { name: "dtype" argType: DATA_TYPE } } opList { name: "dynamic_partition_bp" argDescriptor { name: "outputList" argType: OUTPUT_TENSOR } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "indices" argType: INPUT_TENSOR argIndex: 1 } argDescriptor { name: "gradsAtOutput" argType: INPUT_TENSOR argIndex: 2 } argDescriptor { name: "numPartition" argType: INT64 } argDescriptor { name: "dtype" argType: DATA_TYPE } } opList { name: "dynamic_rnn" argDescriptor { name: "h" argType: OUTPUT_TENSOR } argDescriptor { name: "hFinal" argType: OUTPUT_TENSOR argIndex: 1 } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "Wx" argType: INPUT_TENSOR argIndex: 1 } argDescriptor { name: "Wh" argType: INPUT_TENSOR argIndex: 2 } argDescriptor { name: "b" argType: INPUT_TENSOR argIndex: 3 } argDescriptor { name: "h0" argType: INPUT_TENSOR argIndex: 4 } argDescriptor { name: "timeMajor" argType: INT64 } argDescriptor { name: "dtype" argType: DATA_TYPE } } opList { name: "dynamic_stitch" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "index" argType: INPUT_TENSOR } argDescriptor { name: "input" argType: INPUT_TENSOR argIndex: 1 } argDescriptor { name: "numPartitions" argType: INT64 } argDescriptor { name: "dtype" argType: DATA_TYPE } } opList { name: "elu" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "alpha" argType: DOUBLE } opDeclarationType: CONFIGURABLE_OP_IMPL } opList { name: "elu_bp" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "epsilon" argType: INPUT_TENSOR argIndex: 1 } argDescriptor { name: "alpha" argType: DOUBLE } opDeclarationType: CONFIGURABLE_OP_IMPL } opList { name: "embedding_lookup" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "indices" argType: INPUT_TENSOR argIndex: 1 } argDescriptor { name: "partition_mode" argType: INT64 } argDescriptor { name: "dtype" argType: DATA_TYPE } } opList { name: "encode_bitmap" argDescriptor { name: "encoded" argType: OUTPUT_TENSOR } argDescriptor { name: "counter" argType: OUTPUT_TENSOR argIndex: 1 } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "encoded" argType: INPUT_TENSOR argIndex: 1 } argDescriptor { name: "counter" argType: INPUT_TENSOR argIndex: 2 } argDescriptor { name: "dtype" argType: DATA_TYPE } argDescriptor { name: "threshold" argType: DOUBLE } } opList { name: "encode_threshold" argDescriptor { name: "updated" argType: OUTPUT_TENSOR } argDescriptor { name: "encoded" argType: OUTPUT_TENSOR argIndex: 1 } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "encoded" argType: INPUT_TENSOR argIndex: 1 } argDescriptor { name: "boundary" argType: INT64 } argDescriptor { name: "dtype" argType: DATA_TYPE } argDescriptor { name: "threshold" argType: DOUBLE } } opList { name: "enter" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "isConstant" argType: BOOL } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "frameName" argType: STRING } } opList { name: "entropy" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "keepDims" argType: BOOL } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "dimensions" argType: INPUT_TENSOR argIndex: 1 } argDescriptor { name: "dimensions" argType: INT64 } opDeclarationType: LEGACY_XYZ } opList { name: "eps" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "inPlace" argType: BOOL } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "y" argType: INPUT_TENSOR argIndex: 1 } argDescriptor { name: "dataType" argType: DATA_TYPE } opDeclarationType: LEGACY_XYZ } opList { name: "eps_scalar" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "inPlace" argType: BOOL } argDescriptor { name: "input" argType: INPUT_TENSOR } opDeclarationType: LEGACY_XYZ } opList { name: "equals" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "inPlace" argType: BOOL } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "y" argType: INPUT_TENSOR argIndex: 1 } argDescriptor { name: "dataType" argType: DATA_TYPE } opDeclarationType: BROADCASTABLE_BOOL_OP_IMPL } opList { name: "equals_scalar" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "inPlace" argType: BOOL } argDescriptor { name: "input" argType: INPUT_TENSOR } opDeclarationType: LEGACY_XYZ } opList { name: "equals_with_eps" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "keepDims" argType: BOOL } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "dimensions" argType: INPUT_TENSOR argIndex: 1 } argDescriptor { name: "dimensions" argType: INT64 } argDescriptor { name: "eps" argType: DOUBLE } opDeclarationType: LEGACY_XYZ } opList { name: "erf" 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: "erfc" 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: "euclidean" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "keepDims" argType: BOOL } argDescriptor { name: "allDistances" argType: BOOL argIndex: 1 } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "y" argType: INPUT_TENSOR argIndex: 1 } argDescriptor { name: "dimensions" argType: INT64 } opDeclarationType: LEGACY_XYZ } opList { name: "evaluate_reduction_shape" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "keepDims" argType: BOOL } argDescriptor { name: "oldFormat" argType: BOOL argIndex: 1 } argDescriptor { name: "inputShape" argType: INPUT_TENSOR } argDescriptor { name: "dimensions" argType: INPUT_TENSOR argIndex: 1 } argDescriptor { name: "dtype" argType: DATA_TYPE } } opList { name: "exit" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "frameName" argType: STRING } } opList { name: "exp" 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: "expand_dims" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "dimensions" argType: INT64 } argDescriptor { name: "dtype" argType: DATA_TYPE } } opList { name: "expm1" 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: "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" argType: BOOL } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "ksizeRows" argType: INT64 } argDescriptor { name: "ksizeCols" argType: INT64 argIndex: 1 } argDescriptor { name: "kstrideRows" argType: INT64 argIndex: 2 } argDescriptor { name: "kstrideCols" argType: INT64 argIndex: 3 } argDescriptor { name: "krateRows" argType: INT64 argIndex: 4 } argDescriptor { name: "krateCols" argType: INT64 argIndex: 5 } argDescriptor { name: "isSameMode" argType: INT64 argIndex: 6 } argDescriptor { name: "dtype" argType: DATA_TYPE } } opList { name: "eye" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "numRows" argType: INT64 } argDescriptor { name: "numCols" argType: INT64 argIndex: 1 } argDescriptor { name: "batchDimension" argType: INT64 argIndex: 2 } argDescriptor { name: "dataType" argType: INT64 argIndex: 3 } argDescriptor { name: "dtype" argType: DATA_TYPE } argDescriptor { name: "dataType" argType: DOUBLE } } opList { name: "fake_quant_with_min_max_args" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "narrowRange" argType: BOOL } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "numBits" argType: INT64 } argDescriptor { name: "min" argType: DOUBLE } argDescriptor { name: "max" argType: DOUBLE argIndex: 1 } } opList { name: "fake_quant_with_min_max_vars" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { 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 } argDescriptor { name: "m2" argType: DOUBLE argIndex: 1 } opDeclarationType: CONFIGURABLE_OP_IMPL } opList { name: "fake_quant_with_min_max_vars_per_channel" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { 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 } 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 } argDescriptor { name: "dtype" argType: DATA_TYPE } argDescriptor { name: "value" argType: DOUBLE } } opList { name: "fill_as" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "s" argType: INPUT_TENSOR } opDeclarationType: CONFIGURABLE_OP_IMPL } opList { name: "firas_sparse" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "dtype" argType: DATA_TYPE } } opList { name: "first_index" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "keepDims" argType: BOOL } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "dimensions" argType: INT64 } argDescriptor { name: "compare" argType: DOUBLE } argDescriptor { name: "eps" argType: DOUBLE argIndex: 1 } opDeclarationType: LEGACY_XYZ } opList { name: "flatten" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "order" argType: INT64 } argDescriptor { name: "dtype" argType: DATA_TYPE } } opList { name: "flatten_2d" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "dimensions" argType: INT64 } argDescriptor { name: "dtype" argType: DATA_TYPE } } opList { name: "floor" 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: "floordiv" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "inPlace" argType: BOOL } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "y" argType: INPUT_TENSOR argIndex: 1 } argDescriptor { name: "dataType" argType: DATA_TYPE } opDeclarationType: BROADCASTABLE_OP_IMPL } opList { name: "floordiv_bp" argDescriptor { name: "gradX" argType: OUTPUT_TENSOR } argDescriptor { name: "gradY" argType: OUTPUT_TENSOR argIndex: 1 } argDescriptor { name: "inPlace" argType: BOOL } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "y" argType: INPUT_TENSOR argIndex: 1 } argDescriptor { name: "epsNext" argType: INPUT_TENSOR argIndex: 2 } argDescriptor { name: "dtype" argType: DATA_TYPE } } opList { name: "floormod" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "inPlace" argType: BOOL } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "y" argType: INPUT_TENSOR argIndex: 1 } argDescriptor { name: "dataType" argType: DATA_TYPE } opDeclarationType: BROADCASTABLE_OP_IMPL } opList { name: "floormod_bp" argDescriptor { name: "gradX" argType: OUTPUT_TENSOR } argDescriptor { name: "gradY" argType: OUTPUT_TENSOR argIndex: 1 } argDescriptor { name: "inPlace" argType: BOOL } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "y" argType: INPUT_TENSOR argIndex: 1 } argDescriptor { name: "epsNext" argType: INPUT_TENSOR argIndex: 2 } argDescriptor { name: "dtype" argType: DATA_TYPE } } opList { name: "fmod" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "inPlace" argType: BOOL } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "y" argType: INPUT_TENSOR argIndex: 1 } argDescriptor { name: "dataType" argType: DATA_TYPE } opDeclarationType: LEGACY_XYZ } opList { name: "fmod_scalar" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "y" argType: INPUT_TENSOR argIndex: 1 } opDeclarationType: LEGACY_XYZ } opList { name: "fused_batch_norm" argDescriptor { name: "y" argType: OUTPUT_TENSOR } argDescriptor { name: "batchMean" argType: OUTPUT_TENSOR argIndex: 1 } argDescriptor { name: "batchVar" argType: OUTPUT_TENSOR argIndex: 2 } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "scale" argType: INPUT_TENSOR argIndex: 1 } argDescriptor { name: "offset" argType: INPUT_TENSOR argIndex: 2 } argDescriptor { name: "mean" argType: INPUT_TENSOR argIndex: 3 } argDescriptor { name: "variance" argType: INPUT_TENSOR argIndex: 4 } argDescriptor { name: "batchMeanVar" argType: INPUT_TENSOR argIndex: 5 } argDescriptor { name: "dataFormat" 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 } argDescriptor { name: "inPlace" argType: BOOL } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "indices" argType: INPUT_TENSOR argIndex: 1 } argDescriptor { name: "intArgs" argType: INPUT_TENSOR argIndex: 2 } argDescriptor { name: "dimensions" argType: INT64 } argDescriptor { name: "dtype" argType: DATA_TYPE } } opList { name: "gather_list" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "list" argType: INPUT_TENSOR } argDescriptor { name: "indices" argType: INPUT_TENSOR argIndex: 1 } argDescriptor { name: "dtype" argType: DATA_TYPE } opDeclarationType: LIST_OP_IMPL } opList { name: "gather_nd" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "checkIndices" argType: BOOL } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "indices" argType: INPUT_TENSOR argIndex: 1 } argDescriptor { name: "dtype" argType: DATA_TYPE } } opList { name: "gelu" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "inPlace" argType: BOOL } argDescriptor { name: "precise" argType: BOOL argIndex: 1 } argDescriptor { 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 } } opList { name: "gradientbackwards" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "y" argType: INPUT_TENSOR argIndex: 1 } } opList { name: "greater" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "inPlace" argType: BOOL } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "y" argType: INPUT_TENSOR argIndex: 1 } argDescriptor { name: "dataType" argType: DATA_TYPE } opDeclarationType: BROADCASTABLE_BOOL_OP_IMPL } opList { name: "greater_equal" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "inPlace" argType: BOOL } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "y" argType: INPUT_TENSOR argIndex: 1 } argDescriptor { name: "dataType" argType: DATA_TYPE } opDeclarationType: BROADCASTABLE_BOOL_OP_IMPL } opList { name: "greaterthan_scalar" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "inPlace" argType: BOOL } argDescriptor { name: "input" argType: INPUT_TENSOR } opDeclarationType: LEGACY_XYZ } opList { name: "greaterthanorequal_scalar" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "inPlace" argType: BOOL } argDescriptor { name: "input" argType: INPUT_TENSOR } opDeclarationType: LEGACY_XYZ } opList { name: "grid_free" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "y" argType: INPUT_TENSOR argIndex: 1 } opDeclarationType: LEGACY_XYZ } opList { name: "gru" argDescriptor { name: "h" argType: OUTPUT_TENSOR } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "hI" argType: INPUT_TENSOR argIndex: 1 } argDescriptor { name: "Wx" argType: INPUT_TENSOR argIndex: 2 } argDescriptor { name: "Wh" argType: INPUT_TENSOR argIndex: 3 } argDescriptor { name: "b" argType: INPUT_TENSOR argIndex: 4 } argDescriptor { name: "dtype" argType: DATA_TYPE } } opList { name: "gruCell" argDescriptor { name: "r" argType: OUTPUT_TENSOR } argDescriptor { name: "u" argType: OUTPUT_TENSOR argIndex: 1 } argDescriptor { name: "c" argType: OUTPUT_TENSOR argIndex: 2 } argDescriptor { name: "h" argType: OUTPUT_TENSOR argIndex: 3 } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "hLast" argType: INPUT_TENSOR argIndex: 1 } argDescriptor { name: "Wru" argType: INPUT_TENSOR argIndex: 2 } argDescriptor { name: "Wc" argType: INPUT_TENSOR argIndex: 3 } argDescriptor { name: "bru" argType: INPUT_TENSOR argIndex: 4 } argDescriptor { name: "bc" argType: INPUT_TENSOR argIndex: 5 } argDescriptor { name: "dtype" argType: DATA_TYPE } } opList { name: "gruCell_bp" argDescriptor { name: "dLdx" argType: OUTPUT_TENSOR } argDescriptor { name: "dLdhi" argType: OUTPUT_TENSOR argIndex: 1 } argDescriptor { name: "dLdW" argType: OUTPUT_TENSOR argIndex: 2 } argDescriptor { name: "dLdWc" argType: OUTPUT_TENSOR argIndex: 3 } argDescriptor { name: "dLdb" argType: OUTPUT_TENSOR argIndex: 4 } argDescriptor { name: "dLdbc" argType: OUTPUT_TENSOR argIndex: 5 } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "hi" argType: INPUT_TENSOR argIndex: 1 } argDescriptor { name: "W" argType: INPUT_TENSOR argIndex: 2 } argDescriptor { name: "Wc" argType: INPUT_TENSOR argIndex: 3 } argDescriptor { name: "b" argType: INPUT_TENSOR argIndex: 4 } argDescriptor { name: "bc" argType: INPUT_TENSOR argIndex: 5 } argDescriptor { name: "dLdr" argType: INPUT_TENSOR argIndex: 6 } argDescriptor { name: "dLdu" argType: INPUT_TENSOR argIndex: 7 } argDescriptor { name: "dLdc" argType: INPUT_TENSOR argIndex: 8 } argDescriptor { name: "dLdh" argType: INPUT_TENSOR argIndex: 9 } argDescriptor { name: "dtype" argType: DATA_TYPE } } opList { name: "gru_bp" argDescriptor { name: "dLdx" argType: OUTPUT_TENSOR } argDescriptor { name: "dLdhI" argType: OUTPUT_TENSOR argIndex: 1 } argDescriptor { name: "dLdWx" argType: OUTPUT_TENSOR argIndex: 2 } argDescriptor { name: "dLdWh" argType: OUTPUT_TENSOR argIndex: 3 } argDescriptor { name: "dLdb" argType: OUTPUT_TENSOR argIndex: 4 } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "hI" argType: INPUT_TENSOR argIndex: 1 } argDescriptor { name: "Wx" argType: INPUT_TENSOR argIndex: 2 } argDescriptor { name: "Wh" argType: INPUT_TENSOR argIndex: 3 } argDescriptor { name: "b" argType: INPUT_TENSOR argIndex: 4 } argDescriptor { name: "dLdh" argType: INPUT_TENSOR argIndex: 5 } argDescriptor { name: "dtype" argType: DATA_TYPE } } opList { name: "hammingdistance" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "keepDims" argType: BOOL } argDescriptor { name: "allDistances" argType: BOOL argIndex: 1 } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "y" argType: INPUT_TENSOR argIndex: 1 } argDescriptor { name: "dimensions" argType: INT64 } opDeclarationType: LEGACY_XYZ } opList { name: "hard_sigmoid" 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: "hard_sigmoidderivative" 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: "hardsigmoid" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "input" argType: INPUT_TENSOR } opDeclarationType: CONFIGURABLE_OP_IMPL } opList { name: "hardsigmoid_bp" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "epsilon" argType: INPUT_TENSOR argIndex: 1 } opDeclarationType: CONFIGURABLE_OP_IMPL } opList { name: "hardtanh" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "inPlace" argType: BOOL } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "dataType" argType: DATA_TYPE } opDeclarationType: CONFIGURABLE_OP_IMPL } opList { name: "hardtanh_bp" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "epsilon" argType: INPUT_TENSOR argIndex: 1 } opDeclarationType: CONFIGURABLE_OP_IMPL } opList { name: "hardtanhderivative" 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: "hashcode" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "dtype" argType: DATA_TYPE } } opList { name: "hasinf" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "keepDims" argType: BOOL } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "dimensions" argType: INT64 } opDeclarationType: LEGACY_XYZ } opList { name: "hasnan" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "keepDims" argType: BOOL } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "dimensions" argType: INT64 } opDeclarationType: LEGACY_XYZ } opList { name: "hinge_loss" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "logits" argType: INPUT_TENSOR } argDescriptor { name: "weights" argType: INPUT_TENSOR argIndex: 1 } argDescriptor { name: "labels" argType: INPUT_TENSOR argIndex: 2 } argDescriptor { name: "reductionMode" argType: INT64 } argDescriptor { name: "dtype" argType: DATA_TYPE } } opList { name: "hinge_loss_grad" argDescriptor { name: "dLdp" argType: OUTPUT_TENSOR } argDescriptor { name: "dLdw" argType: OUTPUT_TENSOR argIndex: 1 } argDescriptor { name: "dLdl" argType: OUTPUT_TENSOR argIndex: 2 } argDescriptor { name: "logits" argType: INPUT_TENSOR } argDescriptor { name: "weights" argType: INPUT_TENSOR argIndex: 1 } argDescriptor { name: "labels" 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 } argDescriptor { name: "dtype" argType: DATA_TYPE } } opList { name: "histogram_fixed_width" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "range" argType: INPUT_TENSOR argIndex: 1 } argDescriptor { name: "numBins" argType: INPUT_TENSOR argIndex: 2 } argDescriptor { name: "nbins" argType: INT64 } argDescriptor { name: "dtype" argType: DATA_TYPE } } opList { name: "hsv_to_rgb" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "dimC" argType: INT64 } opDeclarationType: CONFIGURABLE_OP_IMPL } opList { name: "huber_loss" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "predictions" argType: INPUT_TENSOR } argDescriptor { name: "weights" argType: INPUT_TENSOR argIndex: 1 } argDescriptor { name: "labels" argType: INPUT_TENSOR argIndex: 2 } argDescriptor { name: "reductionMode" argType: INT64 } argDescriptor { name: "dtype" argType: DATA_TYPE } argDescriptor { name: "delta" argType: DOUBLE } } opList { name: "huber_loss_grad" argDescriptor { name: "dLdp" argType: OUTPUT_TENSOR } argDescriptor { name: "dLdw" argType: OUTPUT_TENSOR argIndex: 1 } argDescriptor { name: "dLdl" argType: OUTPUT_TENSOR argIndex: 2 } argDescriptor { name: "predictions" argType: INPUT_TENSOR } argDescriptor { name: "weights" argType: INPUT_TENSOR argIndex: 1 } argDescriptor { name: "labels" 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: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "inPlace" argType: BOOL } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "dataType" argType: DATA_TYPE } opDeclarationType: OP_IMPL } opList { name: "identity_bp" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "epsilon" argType: INPUT_TENSOR argIndex: 1 } opDeclarationType: OP_IMPL } opList { name: "identity_n" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "dtype" argType: DATA_TYPE } } opList { name: "igamma" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "y" argType: INPUT_TENSOR argIndex: 1 } opDeclarationType: BROADCASTABLE_OP_IMPL } opList { name: "igammac" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "y" argType: INPUT_TENSOR argIndex: 1 } opDeclarationType: BROADCASTABLE_OP_IMPL } 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 } argDescriptor { name: "kernelWidth" argType: INT64 argIndex: 1 } argDescriptor { name: "strideY" argType: INT64 argIndex: 2 } argDescriptor { name: "strideX" argType: INT64 argIndex: 3 } argDescriptor { name: "padHeight" argType: INT64 argIndex: 4 } argDescriptor { name: "padWidth" argType: INT64 argIndex: 5 } argDescriptor { name: "dY" argType: INT64 argIndex: 6 } argDescriptor { name: "dX" argType: INT64 argIndex: 7 } argDescriptor { name: "dtype" argType: DATA_TYPE } argDescriptor { name: "zeroPadVal" argType: DOUBLE } } opList { name: "im2col_bp" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "gradAtOutput" argType: INPUT_TENSOR argIndex: 1 } argDescriptor { name: "kernelHeight" argType: INT64 } argDescriptor { name: "kernelWidth" argType: INT64 argIndex: 1 } argDescriptor { name: "strideY" argType: INT64 argIndex: 2 } argDescriptor { name: "strideX" argType: INT64 argIndex: 3 } argDescriptor { name: "pH" argType: INT64 argIndex: 4 } argDescriptor { name: "pW" argType: INT64 argIndex: 5 } argDescriptor { name: "dY" argType: INT64 argIndex: 6 } argDescriptor { name: "dX" argType: INT64 argIndex: 7 } argDescriptor { name: "dtype" argType: DATA_TYPE } argDescriptor { name: "zeroPadVal" argType: DOUBLE } } opList { name: "image_resize" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "preserveAspectRatio" argType: BOOL } argDescriptor { name: "antialias" argType: BOOL argIndex: 1 } argDescriptor { name: "image" argType: INPUT_TENSOR } argDescriptor { name: "size" argType: INPUT_TENSOR argIndex: 1 } argDescriptor { name: "method" argType: INT64 } argDescriptor { name: "dtype" argType: DATA_TYPE } } opList { name: "in_top_k" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "sorted" argType: BOOL } argDescriptor { name: "predictions" argType: INPUT_TENSOR } argDescriptor { name: "target" argType: INPUT_TENSOR argIndex: 1 } argDescriptor { name: "k" argType: INT64 } argDescriptor { name: "dtype" argType: DATA_TYPE } } opList { name: "invert_permutation" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "inPlace" argType: BOOL } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "dataType" argType: DATA_TYPE } opDeclarationType: CONFIGURABLE_OP_IMPL } opList { name: "is_non_decreasing" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "inPlace" argType: BOOL } argDescriptor { name: "input" argType: INPUT_TENSOR } opDeclarationType: BOOLEAN_OP_IMPL } opList { name: "is_numeric_tensor" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "inPlace" argType: BOOL } argDescriptor { name: "input" argType: INPUT_TENSOR } opDeclarationType: BOOLEAN_OP_IMPL } opList { name: "is_strictly_increasing" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "inPlace" argType: BOOL } argDescriptor { name: "input" argType: INPUT_TENSOR } opDeclarationType: BOOLEAN_OP_IMPL } opList { name: "isfinite" 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: "isinf" 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: "ismax" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "dimensions" argType: INT64 } opDeclarationType: CONFIGURABLE_OP_IMPL } opList { name: "isnan" 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: "jaccarddistance" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "keepDims" argType: BOOL } argDescriptor { name: "allDistances" argType: BOOL argIndex: 1 } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "y" argType: INPUT_TENSOR argIndex: 1 } argDescriptor { name: "dimensions" argType: INT64 } opDeclarationType: LEGACY_XYZ } opList { name: "knn_mindistance" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "lowest" argType: INPUT_TENSOR argIndex: 1 } argDescriptor { name: "highest" argType: INPUT_TENSOR argIndex: 2 } argDescriptor { name: "distance" argType: INPUT_TENSOR argIndex: 3 } argDescriptor { name: "dtype" argType: DATA_TYPE } } opList { name: "l2_loss" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "dtype" argType: DATA_TYPE } } opList { name: "last_index" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "keepDims" argType: BOOL } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "dimensions" argType: INT64 } argDescriptor { name: "compare" argType: DOUBLE } argDescriptor { name: "eps" argType: DOUBLE argIndex: 1 } opDeclarationType: LEGACY_XYZ } opList { name: "layer_norm" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "channelsFirst" argType: BOOL } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "gain" argType: INPUT_TENSOR argIndex: 1 } argDescriptor { name: "bias" argType: INPUT_TENSOR argIndex: 2 } argDescriptor { name: "dimensions" argType: INT64 } opDeclarationType: CONFIGURABLE_OP_IMPL } opList { name: "layer_norm_bp" argDescriptor { name: "dLdx" argType: OUTPUT_TENSOR } argDescriptor { name: "dLdg" argType: OUTPUT_TENSOR argIndex: 1 } argDescriptor { name: "dLdb" argType: OUTPUT_TENSOR argIndex: 2 } argDescriptor { name: "channelsFirst" argType: BOOL } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "gain" argType: INPUT_TENSOR argIndex: 1 } argDescriptor { name: "bias" argType: INPUT_TENSOR argIndex: 2 } argDescriptor { name: "eps" argType: INPUT_TENSOR argIndex: 3 } argDescriptor { name: "dLdx" argType: INPUT_TENSOR argIndex: 4 } argDescriptor { name: "dLdg" argType: INPUT_TENSOR argIndex: 5 } argDescriptor { name: "dLdb" argType: INPUT_TENSOR argIndex: 6 } argDescriptor { name: "dimensions" argType: INT64 } argDescriptor { name: "dtype" argType: DATA_TYPE } } opList { name: "leakyrelu" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "inPlace" argType: BOOL } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "alpha" argType: DOUBLE } opDeclarationType: LEGACY_XYZ } opList { name: "leakyreluderivative" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "inPlace" argType: BOOL } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "alpha" argType: DOUBLE } opDeclarationType: LEGACY_XYZ } opList { name: "less" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "inPlace" argType: BOOL } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "y" argType: INPUT_TENSOR argIndex: 1 } argDescriptor { name: "dataType" argType: DATA_TYPE } opDeclarationType: BROADCASTABLE_BOOL_OP_IMPL } opList { name: "less_equal" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "inPlace" argType: BOOL } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "y" argType: INPUT_TENSOR argIndex: 1 } argDescriptor { name: "dataType" argType: DATA_TYPE } opDeclarationType: BROADCASTABLE_BOOL_OP_IMPL } opList { name: "lessthan_scalar" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "inPlace" argType: BOOL } argDescriptor { name: "input" argType: INPUT_TENSOR } opDeclarationType: LEGACY_XYZ } opList { name: "lessthanorequal_scalar" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "inPlace" argType: BOOL } argDescriptor { name: "input" argType: INPUT_TENSOR } opDeclarationType: LEGACY_XYZ } opList { name: "lgamma" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "input" argType: INPUT_TENSOR } opDeclarationType: OP_IMPL } opList { name: "lin_space" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "start" argType: INPUT_TENSOR } argDescriptor { name: "finish" argType: INPUT_TENSOR argIndex: 1 } argDescriptor { name: "numOfElements" 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 } argDescriptor { name: "from" argType: DOUBLE } argDescriptor { name: "step" argType: DOUBLE argIndex: 1 } opDeclarationType: LEGACY_XYZ } opList { name: "listdiff" argDescriptor { name: "output1" argType: OUTPUT_TENSOR } argDescriptor { name: "output2" argType: OUTPUT_TENSOR argIndex: 1 } argDescriptor { name: "values" argType: INPUT_TENSOR } argDescriptor { name: "keep" argType: INPUT_TENSOR argIndex: 1 } argDescriptor { name: "dtype" argType: DATA_TYPE } } opList { name: "log" 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: "log1p" 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: "log_loss" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "predictions" argType: INPUT_TENSOR } argDescriptor { name: "weights" argType: INPUT_TENSOR argIndex: 1 } argDescriptor { name: "labels" argType: INPUT_TENSOR argIndex: 2 } argDescriptor { name: "reductionMode" argType: INT64 } argDescriptor { name: "dtype" argType: DATA_TYPE } argDescriptor { name: "epsilon" argType: DOUBLE } } opList { name: "log_loss_grad" argDescriptor { name: "dLdp" argType: OUTPUT_TENSOR } argDescriptor { name: "dLdw" argType: OUTPUT_TENSOR argIndex: 1 } argDescriptor { name: "dLdl" argType: OUTPUT_TENSOR argIndex: 2 } argDescriptor { name: "predictions" argType: INPUT_TENSOR } argDescriptor { name: "weights" argType: INPUT_TENSOR argIndex: 1 } argDescriptor { name: "labels" argType: INPUT_TENSOR argIndex: 2 } argDescriptor { name: "reductionMode" argType: INT64 } argDescriptor { name: "dtype" argType: DATA_TYPE } argDescriptor { name: "epsilon" argType: DOUBLE } } opList { name: "log_matrix_determinant" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "inPlace" argType: BOOL } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "dtype" argType: DATA_TYPE } } opList { name: "log_poisson_loss" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "full" argType: BOOL } argDescriptor { name: "log_predictions" argType: INPUT_TENSOR } argDescriptor { name: "weights" argType: INPUT_TENSOR argIndex: 1 } argDescriptor { name: "labels" argType: INPUT_TENSOR argIndex: 2 } argDescriptor { name: "reductionMode" argType: INT64 } argDescriptor { name: "dtype" argType: DATA_TYPE } } opList { name: "log_poisson_loss_grad" argDescriptor { name: "dLdp" argType: OUTPUT_TENSOR } argDescriptor { name: "dLdw" argType: OUTPUT_TENSOR argIndex: 1 } argDescriptor { name: "dLdl" argType: OUTPUT_TENSOR argIndex: 2 } argDescriptor { name: "full" argType: BOOL } argDescriptor { name: "log_predictions" argType: INPUT_TENSOR } argDescriptor { name: "weights" argType: INPUT_TENSOR argIndex: 1 } argDescriptor { name: "labels" argType: INPUT_TENSOR argIndex: 2 } argDescriptor { name: "reductionMode" argType: INT64 } argDescriptor { name: "dtype" argType: DATA_TYPE } } opList { name: "log_softmax" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "dimension" argType: INT64 } opDeclarationType: CONFIGURABLE_OP_IMPL } opList { name: "log_softmax_bp" argDescriptor { name: "gradI" argType: OUTPUT_TENSOR } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "gradO" argType: INPUT_TENSOR argIndex: 1 } argDescriptor { name: "dimension" argType: INT64 } opDeclarationType: CONFIGURABLE_OP_IMPL } opList { name: "log_x" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "base" argType: DOUBLE } opDeclarationType: LEGACY_XYZ } opList { name: "logdet" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "dtype" argType: DATA_TYPE } } opList { name: "logentropy" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "keepDims" argType: BOOL } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "dimensions" argType: INPUT_TENSOR argIndex: 1 } argDescriptor { name: "dimensions" argType: INT64 } opDeclarationType: LEGACY_XYZ } opList { name: "logsigmoid" 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: "loop_cond" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "frameName" argType: STRING } } opList { name: "lrelu" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "alpha" argType: DOUBLE } opDeclarationType: CONFIGURABLE_OP_IMPL } opList { name: "lrelu_bp" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "epsilon" argType: INPUT_TENSOR argIndex: 1 } argDescriptor { name: "alpha" argType: DOUBLE } opDeclarationType: CONFIGURABLE_OP_IMPL } opList { name: "lrn" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "inPlace" argType: BOOL } argDescriptor { name: "input" argType: INPUT_TENSOR } 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: "lrn_bp" argDescriptor { name: "gradI" argType: OUTPUT_TENSOR } argDescriptor { name: "inPlace" argType: BOOL } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "gradO" 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: "h" argType: OUTPUT_TENSOR } argDescriptor { name: "c" argType: OUTPUT_TENSOR argIndex: 1 } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "h0" argType: INPUT_TENSOR argIndex: 1 } argDescriptor { name: "c0" argType: INPUT_TENSOR argIndex: 2 } argDescriptor { name: "Wx" argType: INPUT_TENSOR argIndex: 3 } argDescriptor { name: "Wh" argType: INPUT_TENSOR argIndex: 4 } argDescriptor { name: "Wc" argType: INPUT_TENSOR argIndex: 5 } argDescriptor { name: "Wp" argType: INPUT_TENSOR argIndex: 6 } argDescriptor { name: "b" argType: INPUT_TENSOR argIndex: 7 } argDescriptor { name: "peephole" argType: INT64 } argDescriptor { name: "projection" argType: INT64 argIndex: 1 } 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: "lstmBlock" argDescriptor { name: "i" argType: OUTPUT_TENSOR } argDescriptor { name: "c" argType: OUTPUT_TENSOR argIndex: 1 } argDescriptor { name: "f" argType: OUTPUT_TENSOR argIndex: 2 } argDescriptor { name: "o" argType: OUTPUT_TENSOR argIndex: 3 } argDescriptor { name: "outputs" argType: OUTPUT_TENSOR argIndex: 4 } argDescriptor { name: "h" argType: OUTPUT_TENSOR argIndex: 5 } argDescriptor { name: "y" argType: OUTPUT_TENSOR argIndex: 6 } argDescriptor { name: "maxTSLength" argType: INPUT_TENSOR } argDescriptor { name: "input" argType: INPUT_TENSOR argIndex: 1 } argDescriptor { name: "cLast" argType: INPUT_TENSOR argIndex: 2 } argDescriptor { name: "yLast" argType: INPUT_TENSOR argIndex: 3 } argDescriptor { name: "W" argType: INPUT_TENSOR argIndex: 4 } argDescriptor { name: "Wci" argType: INPUT_TENSOR argIndex: 5 } argDescriptor { name: "Wcf" argType: INPUT_TENSOR argIndex: 6 } argDescriptor { name: "Wco" argType: INPUT_TENSOR argIndex: 7 } argDescriptor { name: "b" argType: INPUT_TENSOR argIndex: 8 } 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 } argDescriptor { name: "c" argType: OUTPUT_TENSOR argIndex: 1 } argDescriptor { name: "f" argType: OUTPUT_TENSOR argIndex: 2 } argDescriptor { name: "o" argType: OUTPUT_TENSOR argIndex: 3 } argDescriptor { name: "outputs" argType: OUTPUT_TENSOR argIndex: 4 } argDescriptor { name: "h" argType: OUTPUT_TENSOR argIndex: 5 } argDescriptor { name: "y" argType: OUTPUT_TENSOR argIndex: 6 } argDescriptor { name: "xt" argType: INPUT_TENSOR } argDescriptor { name: "cLast" argType: INPUT_TENSOR argIndex: 1 } argDescriptor { name: "yLast" argType: INPUT_TENSOR argIndex: 2 } argDescriptor { name: "W" argType: INPUT_TENSOR argIndex: 3 } argDescriptor { name: "Wci" argType: INPUT_TENSOR argIndex: 4 } argDescriptor { name: "Wcf" argType: INPUT_TENSOR argIndex: 5 } argDescriptor { name: "Wco" argType: INPUT_TENSOR argIndex: 6 } argDescriptor { name: "b" argType: INPUT_TENSOR argIndex: 7 } argDescriptor { name: "peephole" argType: INT64 } 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 } argDescriptor { name: "ct" argType: OUTPUT_TENSOR argIndex: 1 } argDescriptor { name: "xt" argType: INPUT_TENSOR } argDescriptor { name: "ht_1" argType: INPUT_TENSOR argIndex: 1 } argDescriptor { name: "ct_1" argType: INPUT_TENSOR argIndex: 2 } argDescriptor { name: "Wx" argType: INPUT_TENSOR argIndex: 3 } argDescriptor { name: "Wh" argType: INPUT_TENSOR argIndex: 4 } argDescriptor { name: "Wc" argType: INPUT_TENSOR argIndex: 5 } argDescriptor { name: "Wp" argType: INPUT_TENSOR argIndex: 6 } argDescriptor { name: "b" argType: INPUT_TENSOR argIndex: 7 } argDescriptor { name: "peephole" argType: INT64 } argDescriptor { name: "projection" argType: INT64 argIndex: 1 } 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 } argDescriptor { name: "hL" argType: OUTPUT_TENSOR argIndex: 1 } argDescriptor { name: "cL" argType: OUTPUT_TENSOR argIndex: 2 } argDescriptor { name: "hasBiases" argType: BOOL } argDescriptor { name: "hasSeqLen" argType: BOOL argIndex: 1 } argDescriptor { name: "hasInitH" argType: BOOL argIndex: 2 } argDescriptor { name: "hasInitC" argType: BOOL argIndex: 3 } argDescriptor { name: "hasPH" argType: BOOL argIndex: 4 } argDescriptor { name: "retFullSeq" argType: BOOL argIndex: 5 } argDescriptor { name: "retLastH" argType: BOOL argIndex: 6 } argDescriptor { name: "retLastC" argType: BOOL argIndex: 7 } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "Wx" argType: INPUT_TENSOR argIndex: 1 } argDescriptor { name: "Wr" argType: INPUT_TENSOR argIndex: 2 } argDescriptor { name: "b" argType: INPUT_TENSOR argIndex: 3 } argDescriptor { name: "seqLen" argType: INPUT_TENSOR argIndex: 4 } argDescriptor { name: "hI" argType: INPUT_TENSOR argIndex: 5 } argDescriptor { name: "cI" argType: INPUT_TENSOR argIndex: 6 } argDescriptor { name: "Wp" argType: INPUT_TENSOR argIndex: 7 } 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 } 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: "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 } argDescriptor { name: "Wx" argType: INPUT_TENSOR argIndex: 1 } argDescriptor { name: "Wr" argType: INPUT_TENSOR argIndex: 2 } argDescriptor { name: "b" argType: INPUT_TENSOR argIndex: 3 } argDescriptor { name: "hI" argType: INPUT_TENSOR argIndex: 4 } argDescriptor { name: "cI" argType: INPUT_TENSOR argIndex: 5 } argDescriptor { name: "Wp" argType: INPUT_TENSOR argIndex: 6 } argDescriptor { name: "gateAct" argType: INT64 } argDescriptor { name: "cellAct" argType: INT64 argIndex: 1 } argDescriptor { name: "outAct" argType: INT64 argIndex: 2 } 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: "lstmLayerCellBp" argDescriptor { name: "dLdx" argType: OUTPUT_TENSOR } argDescriptor { name: "dLdWx" argType: OUTPUT_TENSOR argIndex: 1 } argDescriptor { name: "dLdWr" argType: OUTPUT_TENSOR argIndex: 2 } argDescriptor { name: "dLdb" argType: OUTPUT_TENSOR argIndex: 3 } argDescriptor { name: "dLdhI" argType: OUTPUT_TENSOR argIndex: 4 } argDescriptor { name: "dLdcI" argType: OUTPUT_TENSOR argIndex: 5 } argDescriptor { name: "dLdWp" argType: OUTPUT_TENSOR argIndex: 6 } argDescriptor { name: "hasBiases" argType: BOOL } argDescriptor { name: "hasPH" argType: BOOL argIndex: 1 } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "Wx" argType: INPUT_TENSOR argIndex: 1 } argDescriptor { name: "Wr" argType: INPUT_TENSOR argIndex: 2 } argDescriptor { name: "b" argType: INPUT_TENSOR argIndex: 3 } argDescriptor { name: "hI" argType: INPUT_TENSOR argIndex: 4 } argDescriptor { name: "cI" argType: INPUT_TENSOR argIndex: 5 } argDescriptor { name: "Wp" argType: INPUT_TENSOR argIndex: 6 } argDescriptor { name: "dLdh" argType: INPUT_TENSOR argIndex: 7 } argDescriptor { name: "gateAct" argType: INT64 } argDescriptor { name: "cellAct" argType: INT64 argIndex: 1 } argDescriptor { name: "outAct" argType: INT64 argIndex: 2 } 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 } argDescriptor { name: "dLdWx" argType: OUTPUT_TENSOR argIndex: 1 } argDescriptor { name: "dLdWr" argType: OUTPUT_TENSOR argIndex: 2 } argDescriptor { name: "dLdb" argType: OUTPUT_TENSOR argIndex: 3 } argDescriptor { name: "dLdhI" argType: OUTPUT_TENSOR argIndex: 4 } argDescriptor { name: "dLdcI" argType: OUTPUT_TENSOR argIndex: 5 } argDescriptor { name: "dLdWp" argType: OUTPUT_TENSOR argIndex: 6 } argDescriptor { name: "hasBiases" argType: BOOL } argDescriptor { name: "hasSeqLen" argType: BOOL argIndex: 1 } argDescriptor { name: "hasInitH" argType: BOOL argIndex: 2 } argDescriptor { name: "hasInitC" argType: BOOL argIndex: 3 } argDescriptor { name: "hasPH" argType: BOOL argIndex: 4 } argDescriptor { name: "retFullSeq" argType: BOOL argIndex: 5 } argDescriptor { name: "retLastH" argType: BOOL argIndex: 6 } argDescriptor { name: "retLastC" argType: BOOL argIndex: 7 } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "Wx" argType: INPUT_TENSOR argIndex: 1 } argDescriptor { name: "Wr" argType: INPUT_TENSOR argIndex: 2 } argDescriptor { name: "b" argType: INPUT_TENSOR argIndex: 3 } argDescriptor { name: "seqLen" argType: INPUT_TENSOR argIndex: 4 } argDescriptor { name: "hI" argType: INPUT_TENSOR argIndex: 5 } argDescriptor { name: "cI" argType: INPUT_TENSOR argIndex: 6 } argDescriptor { name: "Wp" argType: INPUT_TENSOR argIndex: 7 } argDescriptor { name: "dLdh" argType: INPUT_TENSOR argIndex: 8 } argDescriptor { name: "dLdhL" argType: INPUT_TENSOR argIndex: 9 } argDescriptor { name: "dLdcL" argType: INPUT_TENSOR argIndex: 10 } argDescriptor { name: "dLdsL" 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 } 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 } argDescriptor { name: "fastFlag" argType: BOOL } argDescriptor { name: "a" argType: INPUT_TENSOR } argDescriptor { name: "b" argType: INPUT_TENSOR argIndex: 1 } argDescriptor { name: "dtype" argType: DATA_TYPE } argDescriptor { name: "l2_factor" argType: DOUBLE } } opList { name: "lu" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "p" argType: OUTPUT_TENSOR argIndex: 1 } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "dtype" argType: DATA_TYPE } } opList { name: "manhattan" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "keepDims" argType: BOOL } argDescriptor { name: "allDistances" argType: BOOL argIndex: 1 } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "y" argType: INPUT_TENSOR argIndex: 1 } argDescriptor { name: "dimensions" argType: INT64 } opDeclarationType: LEGACY_XYZ } opList { name: "match_condition" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "keepDims" argType: BOOL } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "dimensions" argType: INT64 } argDescriptor { name: "compare" argType: DOUBLE } argDescriptor { name: "eps" argType: DOUBLE argIndex: 1 } opDeclarationType: LEGACY_XYZ } opList { name: "match_condition_transform" 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 } opDeclarationType: LEGACY_XYZ } opList { name: "matmul" 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: "input" argType: INPUT_TENSOR } argDescriptor { name: "y" argType: INPUT_TENSOR argIndex: 1 } 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: "matmul_bp" argDescriptor { name: "dldx" argType: OUTPUT_TENSOR } argDescriptor { name: "dldy" argType: OUTPUT_TENSOR argIndex: 1 } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "y" argType: INPUT_TENSOR argIndex: 1 } argDescriptor { name: "eps" argType: INPUT_TENSOR argIndex: 2 } argDescriptor { name: "dldx" argType: INPUT_TENSOR argIndex: 3 } argDescriptor { name: "dldy" 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: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "minLowerT" argType: INPUT_TENSOR argIndex: 1 } argDescriptor { name: "maxUpperT" 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: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "inPlace" argType: BOOL } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "dtype" argType: DATA_TYPE } } opList { name: "matrix_diag" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "inPlace" argType: BOOL } argDescriptor { name: "diagonal" argType: INPUT_TENSOR } argDescriptor { name: "dtype" argType: DATA_TYPE } } opList { name: "matrix_diag_part" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "inPlace" argType: BOOL } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "dtype" argType: DATA_TYPE } } opList { name: "matrix_inverse" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "inPlace" argType: BOOL } argDescriptor { name: "input" argType: INPUT_TENSOR } opDeclarationType: OP_IMPL } opList { name: "matrix_set_diag" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "inPlace" argType: BOOL } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "diagonal" argType: INPUT_TENSOR argIndex: 1 } opDeclarationType: CONFIGURABLE_OP_IMPL } opList { name: "max_pairwise" 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 } 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 } 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: "sameMode" argType: INT64 argIndex: 8 } argDescriptor { name: "extraParam0" argType: INT64 argIndex: 9 } argDescriptor { name: "isNHWC" argType: INT64 argIndex: 10 } argDescriptor { name: "dtype" argType: DATA_TYPE } } opList { name: "max_scalar" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "y" argType: INPUT_TENSOR argIndex: 1 } opDeclarationType: LEGACY_XYZ } opList { name: "maximum" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "inPlace" argType: BOOL } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "y" argType: INPUT_TENSOR argIndex: 1 } argDescriptor { name: "dataType" argType: DATA_TYPE } opDeclarationType: BROADCASTABLE_OP_IMPL } opList { name: "maximum_bp" argDescriptor { name: "gradX" argType: OUTPUT_TENSOR } argDescriptor { name: "gradY" argType: OUTPUT_TENSOR argIndex: 1 } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "y" argType: INPUT_TENSOR argIndex: 1 } argDescriptor { name: "epsNext" argType: INPUT_TENSOR argIndex: 2 } argDescriptor { name: "dtype" argType: DATA_TYPE } } opList { name: "maxout" 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 } 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: "extraParam0" argType: INT64 argIndex: 9 } argDescriptor { name: "isNCHW" argType: INT64 argIndex: 10 } argDescriptor { name: "dtype" argType: DATA_TYPE } } 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 } 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: "extraParam0" argType: INT64 argIndex: 9 } argDescriptor { name: "isNCHW" argType: INT64 argIndex: 10 } argDescriptor { 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 } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "gradO" argType: INPUT_TENSOR argIndex: 1 } 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: "mean_pairwssqerr_loss" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "predictions" argType: INPUT_TENSOR } argDescriptor { name: "weights" argType: INPUT_TENSOR argIndex: 1 } argDescriptor { name: "labels" argType: INPUT_TENSOR argIndex: 2 } argDescriptor { name: "reductionMode" argType: INT64 } argDescriptor { name: "dtype" argType: DATA_TYPE } } opList { name: "mean_pairwssqerr_loss_grad" argDescriptor { name: "dLdp" argType: OUTPUT_TENSOR } argDescriptor { name: "dLdw" argType: OUTPUT_TENSOR argIndex: 1 } argDescriptor { name: "dLdl" argType: OUTPUT_TENSOR argIndex: 2 } argDescriptor { name: "predictions" argType: INPUT_TENSOR } argDescriptor { name: "weights" argType: INPUT_TENSOR argIndex: 1 } argDescriptor { name: "labels" argType: INPUT_TENSOR argIndex: 2 } argDescriptor { name: "reductionMode" argType: INT64 } argDescriptor { name: "dtype" argType: DATA_TYPE } } opList { name: "mean_sqerr_loss" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "predictions" argType: INPUT_TENSOR } argDescriptor { name: "weights" argType: INPUT_TENSOR argIndex: 1 } argDescriptor { name: "labels" argType: INPUT_TENSOR argIndex: 2 } argDescriptor { name: "reductionMode" argType: INT64 } argDescriptor { name: "dtype" argType: DATA_TYPE } } opList { name: "mean_sqerr_loss_grad" argDescriptor { name: "dLdp" argType: OUTPUT_TENSOR } argDescriptor { name: "dLdw" argType: OUTPUT_TENSOR argIndex: 1 } argDescriptor { name: "dLdl" argType: OUTPUT_TENSOR argIndex: 2 } argDescriptor { name: "predictions" argType: INPUT_TENSOR } argDescriptor { name: "weights" argType: INPUT_TENSOR argIndex: 1 } argDescriptor { name: "labels" argType: INPUT_TENSOR argIndex: 2 } argDescriptor { name: "reductionMode" argType: INT64 } argDescriptor { name: "dtype" argType: DATA_TYPE } } opList { name: "merge" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "input" argType: INPUT_TENSOR } } opList { name: "mergeadd" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "inPlace" argType: BOOL } argDescriptor { name: "inArrs" argType: INPUT_TENSOR } argDescriptor { name: "dataType" argType: DATA_TYPE } opDeclarationType: OP_IMPL } opList { name: "mergeadd_bp" argDescriptor { name: "outArrs" argType: OUTPUT_TENSOR } argDescriptor { name: "gradient" argType: INPUT_TENSOR } argDescriptor { name: "gradO" argType: INPUT_TENSOR argIndex: 1 } argDescriptor { name: "dtype" argType: DATA_TYPE } } opList { name: "mergeavg" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "inArrs" argType: INPUT_TENSOR } opDeclarationType: OP_IMPL } opList { name: "mergeavg_bp" argDescriptor { name: "outArrs" argType: OUTPUT_TENSOR } argDescriptor { name: "gradient" argType: INPUT_TENSOR } argDescriptor { name: "gradO" argType: INPUT_TENSOR argIndex: 1 } argDescriptor { name: "dtype" argType: DATA_TYPE } } opList { name: "mergemax" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "inArrs" argType: INPUT_TENSOR } opDeclarationType: OP_IMPL } opList { name: "mergemax_bp" argDescriptor { name: "outArrs" argType: OUTPUT_TENSOR } argDescriptor { name: "inArrs" argType: INPUT_TENSOR } argDescriptor { name: "gradO" 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 } argDescriptor { name: "dtype" argType: DATA_TYPE } } opList { name: "mergesum" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "input" argType: INPUT_TENSOR } } opList { name: "meshgrid" argDescriptor { name: "outArrs" argType: OUTPUT_TENSOR } argDescriptor { name: "cartesian" argType: BOOL } argDescriptor { name: "inArrs" argType: INPUT_TENSOR } argDescriptor { name: "swapFirst2Dims" argType: INT64 } argDescriptor { name: "dtype" argType: DATA_TYPE } } opList { name: "meta_postulate" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } opDeclarationType: LEGACY_XYZ } opList { name: "meta_predicate" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } opDeclarationType: LEGACY_XYZ } opList { name: "meta_predicate_inverted" argDescriptor { 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 } opDeclarationType: LEGACY_XYZ } opList { name: "min_pairwise" 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 } opDeclarationType: LEGACY_XYZ } opList { name: "minimum" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "inPlace" argType: BOOL } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "y" argType: INPUT_TENSOR argIndex: 1 } argDescriptor { name: "dataType" argType: DATA_TYPE } opDeclarationType: BROADCASTABLE_OP_IMPL } opList { name: "minimum_bp" argDescriptor { name: "gradX" argType: OUTPUT_TENSOR } argDescriptor { name: "gradY" argType: OUTPUT_TENSOR argIndex: 1 } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "y" argType: INPUT_TENSOR argIndex: 1 } argDescriptor { name: "epsNext" argType: INPUT_TENSOR argIndex: 2 } argDescriptor { name: "dtype" argType: DATA_TYPE } } opList { name: "mirror_pad" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "isSymmetric" argType: BOOL } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "paddings" argType: INPUT_TENSOR argIndex: 1 } argDescriptor { name: "mode" argType: INT64 } argDescriptor { name: "dtype" argType: DATA_TYPE } } opList { name: "mish" 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: "mod" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "inPlace" argType: BOOL } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "y" argType: INPUT_TENSOR argIndex: 1 } argDescriptor { name: "dataType" argType: DATA_TYPE } opDeclarationType: BROADCASTABLE_OP_IMPL } opList { name: "mod_bp" argDescriptor { name: "gradX" argType: OUTPUT_TENSOR } argDescriptor { name: "gradY" argType: OUTPUT_TENSOR argIndex: 1 } argDescriptor { name: "inPlace" argType: BOOL } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "y" argType: INPUT_TENSOR argIndex: 1 } argDescriptor { name: "epsNext" argType: INPUT_TENSOR argIndex: 2 } argDescriptor { name: "dtype" argType: DATA_TYPE } } opList { name: "moments" argDescriptor { name: "means" argType: OUTPUT_TENSOR } argDescriptor { name: "variances" argType: OUTPUT_TENSOR argIndex: 1 } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "dimensions" argType: INPUT_TENSOR argIndex: 1 } argDescriptor { name: "outStd" 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" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "y" argType: INPUT_TENSOR argIndex: 1 } opDeclarationType: LEGACY_XYZ } opList { name: "multi_head_dot_product_attention" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "scaled" argType: BOOL } argDescriptor { name: "withWeights" argType: BOOL argIndex: 1 } argDescriptor { name: "queries" argType: INPUT_TENSOR } argDescriptor { name: "keys" argType: INPUT_TENSOR argIndex: 1 } argDescriptor { name: "values" argType: INPUT_TENSOR argIndex: 2 } argDescriptor { name: "Wq" argType: INPUT_TENSOR argIndex: 3 } argDescriptor { name: "Wk" argType: INPUT_TENSOR argIndex: 4 } argDescriptor { name: "Wv" argType: INPUT_TENSOR argIndex: 5 } argDescriptor { name: "Wo" argType: INPUT_TENSOR argIndex: 6 } argDescriptor { name: "mask" argType: INPUT_TENSOR argIndex: 7 } 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 } argDescriptor { name: "dLdk" argType: OUTPUT_TENSOR argIndex: 1 } argDescriptor { name: "dLdv" argType: OUTPUT_TENSOR argIndex: 2 } argDescriptor { name: "dLdWq" argType: OUTPUT_TENSOR argIndex: 3 } argDescriptor { name: "dLdWk" argType: OUTPUT_TENSOR argIndex: 4 } argDescriptor { name: "dLdWv" argType: OUTPUT_TENSOR argIndex: 5 } argDescriptor { name: "dLdWo" argType: OUTPUT_TENSOR argIndex: 6 } argDescriptor { name: "scaled" argType: BOOL } argDescriptor { name: "queries" argType: INPUT_TENSOR } argDescriptor { name: "keys" argType: INPUT_TENSOR argIndex: 1 } argDescriptor { name: "values" argType: INPUT_TENSOR argIndex: 2 } argDescriptor { name: "Wq" argType: INPUT_TENSOR argIndex: 3 } argDescriptor { name: "Wk" argType: INPUT_TENSOR argIndex: 4 } argDescriptor { name: "Wv" argType: INPUT_TENSOR argIndex: 5 } argDescriptor { name: "Wo" argType: INPUT_TENSOR argIndex: 6 } argDescriptor { name: "eps" argType: INPUT_TENSOR argIndex: 7 } argDescriptor { name: "mask" argType: INPUT_TENSOR argIndex: 8 } argDescriptor { name: "normalization" argType: INT64 } argDescriptor { name: "dtype" argType: DATA_TYPE } } opList { name: "multiply" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "inPlace" argType: BOOL } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "y" argType: INPUT_TENSOR argIndex: 1 } argDescriptor { name: "dataType" argType: DATA_TYPE } opDeclarationType: BROADCASTABLE_OP_IMPL } opList { name: "multiply_bp" argDescriptor { name: "gradX" argType: OUTPUT_TENSOR } argDescriptor { name: "gradY" argType: OUTPUT_TENSOR argIndex: 1 } argDescriptor { name: "inPlace" argType: BOOL } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "y" argType: INPUT_TENSOR argIndex: 1 } argDescriptor { name: "epsNext" argType: INPUT_TENSOR argIndex: 2 } argDescriptor { name: "dtype" argType: DATA_TYPE } } opList { name: "nadam_updater" argDescriptor { name: "update" argType: OUTPUT_TENSOR } argDescriptor { name: "stateV" argType: OUTPUT_TENSOR argIndex: 1 } argDescriptor { name: "stateM" argType: OUTPUT_TENSOR argIndex: 2 } argDescriptor { name: "gradient" argType: INPUT_TENSOR } argDescriptor { name: "initStateV" 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 } 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: "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: "nesterovs_updater" argDescriptor { name: "update" argType: OUTPUT_TENSOR } argDescriptor { name: "stateV" argType: OUTPUT_TENSOR argIndex: 1 } argDescriptor { name: "gradient" argType: INPUT_TENSOR } argDescriptor { name: "initState" argType: INPUT_TENSOR argIndex: 1 } argDescriptor { name: "lr" argType: INPUT_TENSOR argIndex: 2 } argDescriptor { name: "momentum" argType: INPUT_TENSOR argIndex: 3 } argDescriptor { name: "dLr" argType: DOUBLE } argDescriptor { name: "dMomentum" argType: DOUBLE argIndex: 1 } opDeclarationType: CONFIGURABLE_OP_IMPL } opList { name: "next_iteration" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "frameName" argType: STRING } } opList { name: "non_max_suppression" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "boxes" argType: INPUT_TENSOR } argDescriptor { name: "scales" argType: INPUT_TENSOR argIndex: 1 } argDescriptor { name: "maxOutputSize" argType: INPUT_TENSOR argIndex: 2 } argDescriptor { name: "overlayThreshold" argType: INPUT_TENSOR argIndex: 3 } argDescriptor { name: "scoreThreshold" argType: INPUT_TENSOR argIndex: 4 } argDescriptor { name: "maxOutputSize" argType: INT64 } argDescriptor { name: "dtype" argType: DATA_TYPE } argDescriptor { name: "iouThreshold" argType: DOUBLE } argDescriptor { name: "scoreThreshold" argType: DOUBLE argIndex: 1 } } opList { name: "non_max_suppression_overlaps" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "boxes" argType: INPUT_TENSOR } argDescriptor { name: "scales" argType: INPUT_TENSOR argIndex: 1 } argDescriptor { name: "maxOutputSize" argType: INT64 } argDescriptor { 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 } argDescriptor { name: "boxes" argType: INPUT_TENSOR } argDescriptor { name: "scales" 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 } argDescriptor { name: "maxOutputSize" argType: INT64 } argDescriptor { name: "dtype" argType: DATA_TYPE } } opList { name: "noop" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "input" argType: INPUT_TENSOR } opDeclarationType: OP_IMPL } opList { name: "norm" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "dimensions" argType: INPUT_TENSOR argIndex: 1 } argDescriptor { name: "mode" argType: DOUBLE } opDeclarationType: REDUCTION_OP_IMPL } opList { name: "normalize_moments" argDescriptor { name: "resMeans" argType: OUTPUT_TENSOR } argDescriptor { name: "resVariances" argType: OUTPUT_TENSOR argIndex: 1 } argDescriptor { name: "counts" argType: INPUT_TENSOR } argDescriptor { name: "means" argType: INPUT_TENSOR argIndex: 1 } argDescriptor { name: "variances" argType: INPUT_TENSOR argIndex: 2 } argDescriptor { name: "outMean" argType: INPUT_TENSOR argIndex: 3 } argDescriptor { name: "outVar" argType: INPUT_TENSOR argIndex: 4 } argDescriptor { name: "dtype" argType: DATA_TYPE } argDescriptor { name: "shift" argType: DOUBLE } } opList { name: "not" 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: "comparable" argType: DOUBLE } opDeclarationType: LEGACY_XYZ } opList { name: "not_equals" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "inPlace" argType: BOOL } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "y" argType: INPUT_TENSOR argIndex: 1 } argDescriptor { name: "dataType" argType: DATA_TYPE } opDeclarationType: BROADCASTABLE_BOOL_OP_IMPL } opList { name: "not_scalar" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "inPlace" argType: BOOL } argDescriptor { name: "input" argType: INPUT_TENSOR } opDeclarationType: LEGACY_XYZ } opList { name: "notequals_scalar" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "inPlace" argType: BOOL } argDescriptor { name: "input" argType: INPUT_TENSOR } opDeclarationType: LEGACY_XYZ } opList { name: "nth_element" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "reverse" argType: BOOL } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "n" argType: INPUT_TENSOR argIndex: 1 } argDescriptor { name: "reverse" argType: INT64 } argDescriptor { name: "dtype" argType: DATA_TYPE } } opList { name: "old_assign" 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: "onehot" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "depth" argType: INPUT_TENSOR argIndex: 1 } argDescriptor { name: "on" argType: INPUT_TENSOR argIndex: 2 } argDescriptor { name: "off" 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: "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: "ones_as" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "dataType" argType: INT64 } argDescriptor { name: "dtype" argType: DATA_TYPE } } opList { name: "or" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { 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 } opDeclarationType: LEGACY_XYZ } opList { name: "or_scalar" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "inPlace" argType: BOOL } argDescriptor { name: "input" argType: INPUT_TENSOR } opDeclarationType: LEGACY_XYZ } opList { name: "order" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "dtype" argType: DATA_TYPE } } opList { name: "pad" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "paddings" argType: INPUT_TENSOR argIndex: 1 } argDescriptor { name: "mode" argType: INT64 } argDescriptor { name: "dtype" argType: DATA_TYPE } argDescriptor { name: "padValue" argType: DOUBLE } } opList { name: "parallel_stack" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "inArrs" 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 } argDescriptor { name: "dtype" argType: DATA_TYPE } argDescriptor { name: "q" argType: DOUBLE } argDescriptor { name: "interpolation" argType: DOUBLE argIndex: 1 } argDescriptor { name: "keepDims" argType: DOUBLE argIndex: 2 } } opList { name: "permute" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "permuteDims" argType: INPUT_TENSOR argIndex: 1 } argDescriptor { name: "reverseDims" argType: INT64 } argDescriptor { name: "dtype" argType: DATA_TYPE } } opList { name: "pick_list" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "list" argType: INPUT_TENSOR } argDescriptor { name: "ia" argType: INPUT_TENSOR argIndex: 1 } opDeclarationType: LIST_OP_IMPL } opList { name: "pnormpool2d" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "kY" argType: INT64 } argDescriptor { name: "kX" argType: INT64 argIndex: 1 } argDescriptor { name: "sY" argType: INT64 argIndex: 2 } argDescriptor { name: "sX" argType: INT64 argIndex: 3 } argDescriptor { name: "pY" argType: INT64 argIndex: 4 } argDescriptor { name: "pX" argType: INT64 argIndex: 5 } argDescriptor { name: "dY" argType: INT64 argIndex: 6 } argDescriptor { name: "dX" argType: INT64 argIndex: 7 } argDescriptor { name: "isSameMode" argType: INT64 argIndex: 8 } argDescriptor { name: "extraParam0" argType: INT64 argIndex: 9 } argDescriptor { name: "isNCHW" argType: INT64 argIndex: 10 } argDescriptor { name: "dtype" argType: DATA_TYPE } } opList { name: "pnormpool2d_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 } 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: "pnorm" argType: INT64 argIndex: 9 } argDescriptor { name: "isNCHW" argType: INT64 argIndex: 10 } argDescriptor { name: "dtype" argType: DATA_TYPE } argDescriptor { name: "eps" argType: DOUBLE } } opList { name: "pointwise_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: "isNCHW" argType: INT64 } argDescriptor { name: "wFormat" argType: INT64 argIndex: 1 } argDescriptor { name: "dtype" argType: DATA_TYPE } } opList { name: "polygamma" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "n" argType: INPUT_TENSOR } argDescriptor { name: "input" argType: INPUT_TENSOR argIndex: 1 } opDeclarationType: CONFIGURABLE_OP_IMPL } opList { name: "pooling3dpool3dnew_bp" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "inPlace" argType: BOOL } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "inputArrays" argType: INPUT_TENSOR argIndex: 1 } } opList { name: "pow" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "inPlace" argType: BOOL } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "pow" argType: INPUT_TENSOR argIndex: 1 } argDescriptor { name: "pow" argType: DOUBLE } opDeclarationType: LEGACY_XYZ } opList { name: "pow_pairwise" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "inPlace" argType: BOOL } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "y" argType: INPUT_TENSOR argIndex: 1 } argDescriptor { name: "dataType" argType: DATA_TYPE } opDeclarationType: LEGACY_XYZ } opList { name: "precise_gelu" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "inPlace" argType: BOOL } argDescriptor { name: "precise" argType: BOOL argIndex: 1 } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "dataType" argType: DATA_TYPE } opDeclarationType: LEGACY_XYZ } opList { name: "prelu" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "alpha" argType: INPUT_TENSOR argIndex: 1 } argDescriptor { name: "sharedAxes" argType: INT64 } opDeclarationType: CONFIGURABLE_OP_IMPL } opList { name: "prelu_bp" argDescriptor { name: "dLdI" argType: OUTPUT_TENSOR } argDescriptor { name: "dLdA" argType: OUTPUT_TENSOR argIndex: 1 } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "alpha" argType: INPUT_TENSOR argIndex: 1 } argDescriptor { name: "dLdO" argType: INPUT_TENSOR argIndex: 2 } argDescriptor { name: "dLdI" argType: INPUT_TENSOR argIndex: 3 } argDescriptor { name: "dLdA" argType: INPUT_TENSOR argIndex: 4 } argDescriptor { name: "sharedAxes" argType: INT64 } opDeclarationType: CONFIGURABLE_OP_IMPL } opList { name: "print_affinity" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "dtype" argType: DATA_TYPE } } opList { name: "print_variable" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "printSpecial" argType: BOOL } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "message" argType: INPUT_TENSOR argIndex: 1 } argDescriptor { name: "dtype" argType: DATA_TYPE } } opList { name: "probablistic_merge" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "y" argType: INPUT_TENSOR argIndex: 1 } argDescriptor { name: "probability" argType: DOUBLE } opDeclarationType: LEGACY_XYZ } opList { name: "qr" argDescriptor { name: "outputQ" argType: OUTPUT_TENSOR } argDescriptor { name: "outputR" argType: OUTPUT_TENSOR argIndex: 1 } argDescriptor { name: "fullMatricies" argType: BOOL } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "dtype" argType: DATA_TYPE } } opList { name: "random_bernoulli" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "dtype" argType: DATA_TYPE } argDescriptor { name: "f" argType: DOUBLE } } opList { name: "random_crop" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "shape" 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 } argDescriptor { name: "dtype" argType: DATA_TYPE } argDescriptor { name: "lambda" argType: DOUBLE } } opList { name: "random_gamma" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "shape" argType: INPUT_TENSOR } argDescriptor { name: "alpha" argType: INPUT_TENSOR argIndex: 1 } argDescriptor { name: "beta" argType: INPUT_TENSOR argIndex: 2 } argDescriptor { name: "seed" argType: INT64 } argDescriptor { name: "dtype" argType: DATA_TYPE } } opList { name: "random_multinomial" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "inputSamples" argType: INPUT_TENSOR argIndex: 1 } argDescriptor { name: "dimC" argType: INT64 } argDescriptor { name: "dtype" argType: DATA_TYPE } } opList { name: "random_normal" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "dtype" argType: DATA_TYPE } } opList { name: "random_poisson" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "shape" argType: INPUT_TENSOR } argDescriptor { name: "lambda" argType: INPUT_TENSOR argIndex: 1 } argDescriptor { name: "seed" argType: INT64 } argDescriptor { name: "dtype" argType: DATA_TYPE } } opList { name: "random_shuffle" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "seeds" argType: INT64 } opDeclarationType: OP_IMPL } opList { name: "randomnormal" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "shape" argType: INPUT_TENSOR } argDescriptor { name: "mean" argType: DOUBLE } argDescriptor { name: "stdev" 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 } argDescriptor { name: "dtype" argType: INT64 } argDescriptor { name: "seed" argType: INT64 argIndex: 1 } argDescriptor { name: "dataType" argType: DATA_TYPE } argDescriptor { name: "min" argType: DOUBLE } argDescriptor { name: "max" argType: DOUBLE argIndex: 1 } } opList { name: "range" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "from" argType: INPUT_TENSOR } argDescriptor { name: "to" argType: INPUT_TENSOR argIndex: 1 } argDescriptor { name: "step" argType: INPUT_TENSOR argIndex: 2 } argDescriptor { name: "from" argType: INT64 } argDescriptor { name: "to" argType: INT64 argIndex: 1 } argDescriptor { name: "step" argType: INT64 argIndex: 2 } argDescriptor { name: "dtype" argType: DATA_TYPE } argDescriptor { name: "from" argType: DOUBLE } argDescriptor { name: "to" argType: DOUBLE argIndex: 1 } argDescriptor { name: "step" argType: DOUBLE argIndex: 2 } } opList { name: "rank" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "inPlace" argType: BOOL } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "dtype" argType: DATA_TYPE } } opList { name: "rational_tanh" 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: "rational_tanh_derivative" 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: "rationaltanh" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "input" argType: INPUT_TENSOR } opDeclarationType: CONFIGURABLE_OP_IMPL } opList { name: "rationaltanh_bp" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "epsilon" argType: INPUT_TENSOR argIndex: 1 } opDeclarationType: CONFIGURABLE_OP_IMPL } opList { name: "rdiv_scalar" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "inPlace" argType: BOOL } argDescriptor { name: "input" argType: INPUT_TENSOR } opDeclarationType: LEGACY_XYZ } opList { name: "read_list" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "list" argType: INPUT_TENSOR } argDescriptor { name: "vec" 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: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "inPlace" argType: BOOL } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "y" argType: INPUT_TENSOR argIndex: 1 } argDescriptor { name: "dataType" argType: DATA_TYPE } opDeclarationType: BROADCASTABLE_OP_IMPL } opList { name: "realdiv_bp" argDescriptor { name: "gradX" argType: OUTPUT_TENSOR } argDescriptor { name: "gradY" argType: OUTPUT_TENSOR argIndex: 1 } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "y" argType: INPUT_TENSOR argIndex: 1 } argDescriptor { name: "epsNext" argType: INPUT_TENSOR argIndex: 2 } argDescriptor { name: "dtype" argType: DATA_TYPE } } opList { name: "rectified_tanh" 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: "rectified_tanh_derivative" 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: "rectifiedtanh" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "input" argType: INPUT_TENSOR } opDeclarationType: CONFIGURABLE_OP_IMPL } opList { name: "rectifiedtanh_bp" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "epsilon" argType: INPUT_TENSOR argIndex: 1 } opDeclarationType: CONFIGURABLE_OP_IMPL } opList { name: "reduce_dot_bp" argDescriptor { name: "gradX" argType: OUTPUT_TENSOR } argDescriptor { name: "gradY" argType: OUTPUT_TENSOR argIndex: 1 } argDescriptor { name: "keepDims" argType: BOOL } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "y" argType: INPUT_TENSOR argIndex: 1 } argDescriptor { name: "gradO" argType: INPUT_TENSOR argIndex: 2 } argDescriptor { name: "dimensions" argType: INPUT_TENSOR argIndex: 3 } argDescriptor { name: "outputY" 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 } argDescriptor { name: "dtype" argType: DATA_TYPE } argDescriptor { name: "keepDims" argType: DOUBLE } } opList { name: "reduce_max" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "keepDims" argType: BOOL } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "dimensions" argType: INPUT_TENSOR argIndex: 1 } argDescriptor { name: "dimensions" argType: INT64 } argDescriptor { name: "dtype" argType: DATA_TYPE } } opList { name: "reduce_max_bp" argDescriptor { name: "gradI" argType: OUTPUT_TENSOR } argDescriptor { name: "keepDims" argType: BOOL } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "gradO" argType: INPUT_TENSOR argIndex: 1 } argDescriptor { name: "dimensions" argType: INPUT_TENSOR argIndex: 2 } argDescriptor { name: "dimensions" argType: INT64 } argDescriptor { name: "dtype" argType: DATA_TYPE } } opList { name: "reduce_mean" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "keepDims" argType: BOOL } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "dimensions" argType: INPUT_TENSOR argIndex: 1 } argDescriptor { name: "dimensions" argType: INT64 } argDescriptor { name: "dtype" argType: DATA_TYPE } } opList { name: "reduce_mean_bp" argDescriptor { name: "gradI" argType: OUTPUT_TENSOR } argDescriptor { name: "keepDims" argType: BOOL } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "gradO" argType: INPUT_TENSOR argIndex: 1 } argDescriptor { name: "dimensions" argType: INPUT_TENSOR argIndex: 2 } argDescriptor { name: "dimensions" argType: INT64 } argDescriptor { name: "dtype" argType: DATA_TYPE } } opList { name: "reduce_min" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "keepDims" argType: BOOL } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "dimensions" argType: INPUT_TENSOR argIndex: 1 } argDescriptor { name: "dimensions" argType: INT64 } argDescriptor { name: "dtype" argType: DATA_TYPE } } opList { name: "reduce_min_bp" argDescriptor { name: "gradI" argType: OUTPUT_TENSOR } argDescriptor { name: "keepDims" argType: BOOL } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "gradO" argType: INPUT_TENSOR argIndex: 1 } argDescriptor { name: "dimensions" argType: INPUT_TENSOR argIndex: 2 } argDescriptor { name: "dimensions" argType: INT64 } argDescriptor { name: "dtype" argType: DATA_TYPE } } opList { name: "reduce_norm1" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "keepDims" argType: BOOL } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "dimensions" argType: INPUT_TENSOR argIndex: 1 } argDescriptor { name: "dimensions" argType: INT64 } argDescriptor { name: "dtype" argType: DATA_TYPE } } opList { name: "reduce_norm1_bp" argDescriptor { name: "gradI" argType: OUTPUT_TENSOR } argDescriptor { name: "keepDims" argType: BOOL } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "gradO" argType: INPUT_TENSOR argIndex: 1 } argDescriptor { name: "dimensions" argType: INPUT_TENSOR argIndex: 2 } argDescriptor { name: "dimensions" argType: INT64 } argDescriptor { name: "dtype" argType: DATA_TYPE } } opList { name: "reduce_norm2" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "keepDims" argType: BOOL } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "dimensions" argType: INPUT_TENSOR argIndex: 1 } argDescriptor { name: "dimensions" argType: INT64 } argDescriptor { name: "dtype" argType: DATA_TYPE } } opList { name: "reduce_norm2_bp" argDescriptor { name: "gradI" argType: OUTPUT_TENSOR } argDescriptor { name: "keepDims" argType: BOOL } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "gradO" argType: INPUT_TENSOR argIndex: 1 } argDescriptor { name: "dimensions" argType: INPUT_TENSOR argIndex: 2 } argDescriptor { name: "dimensions" argType: INT64 } argDescriptor { name: "dtype" argType: DATA_TYPE } } opList { name: "reduce_norm_max" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "keepDims" argType: BOOL } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "dimensions" argType: INPUT_TENSOR argIndex: 1 } argDescriptor { name: "dtype" argType: DATA_TYPE } } opList { name: "reduce_norm_max_bp" argDescriptor { name: "gradI" argType: OUTPUT_TENSOR } argDescriptor { name: "keepDims" argType: BOOL } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "gradO" argType: INPUT_TENSOR argIndex: 1 } argDescriptor { name: "dimensions" argType: INPUT_TENSOR argIndex: 2 } argDescriptor { name: "dimensions" argType: INT64 } argDescriptor { name: "dtype" argType: DATA_TYPE } } opList { name: "reduce_normmax" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "keepDims" argType: BOOL } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "dimensions" argType: INPUT_TENSOR argIndex: 1 } argDescriptor { name: "dimensions" argType: INT64 } opDeclarationType: LEGACY_XYZ } opList { name: "reduce_prod" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "keepDims" argType: BOOL } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "dimensions" argType: INPUT_TENSOR argIndex: 1 } argDescriptor { name: "dimensions" argType: INT64 } argDescriptor { name: "dtype" argType: DATA_TYPE } } opList { name: "reduce_prod_bp" argDescriptor { name: "gradI" argType: OUTPUT_TENSOR } argDescriptor { name: "keepDims" argType: BOOL } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "gradO" argType: INPUT_TENSOR argIndex: 1 } argDescriptor { name: "dimensions" argType: INPUT_TENSOR argIndex: 2 } argDescriptor { name: "dimensions" argType: INT64 } argDescriptor { name: "dtype" argType: DATA_TYPE } } opList { name: "reduce_sqnorm" argDescriptor { name: "gradI" argType: OUTPUT_TENSOR } argDescriptor { name: "keepDims" argType: BOOL } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "dimensions" argType: INPUT_TENSOR argIndex: 1 } argDescriptor { name: "dimensions" argType: INT64 } argDescriptor { name: "dtype" argType: DATA_TYPE } } opList { name: "reduce_sqnorm_bp" argDescriptor { name: "gradI" argType: OUTPUT_TENSOR } argDescriptor { name: "keepDims" argType: BOOL } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "gradO" argType: INPUT_TENSOR argIndex: 1 } argDescriptor { name: "dimensions" 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: "keepDims" argType: DOUBLE } argDescriptor { name: "biasCorrected" argType: DOUBLE argIndex: 1 } } opList { name: "reduce_stdev_bp" argDescriptor { name: "gradI" argType: OUTPUT_TENSOR } argDescriptor { name: "biasCorrected" argType: BOOL } argDescriptor { name: "keepDims" argType: BOOL argIndex: 1 } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "gradO" argType: INPUT_TENSOR argIndex: 1 } argDescriptor { name: "dimensions" 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: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "keepDims" argType: BOOL } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "dimensions" argType: INPUT_TENSOR argIndex: 1 } argDescriptor { name: "dimensions" argType: INT64 } argDescriptor { name: "dtype" argType: DATA_TYPE } } opList { name: "reduce_sum_bp" argDescriptor { name: "gradI" argType: OUTPUT_TENSOR } argDescriptor { name: "keepDims" argType: BOOL } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "gradO" argType: INPUT_TENSOR argIndex: 1 } argDescriptor { name: "dimensions" 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: "keepDims" argType: DOUBLE } argDescriptor { name: "biasCorrected" argType: DOUBLE argIndex: 1 } } opList { name: "reduce_variance_bp" argDescriptor { name: "gradI" argType: OUTPUT_TENSOR } argDescriptor { name: "biasCorrected" argType: BOOL } argDescriptor { name: "keepDims" argType: BOOL argIndex: 1 } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "gradO" argType: INPUT_TENSOR argIndex: 1 } argDescriptor { name: "dimensions" 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" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "inPlace" argType: BOOL } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "cutoff" argType: DOUBLE } opDeclarationType: CONFIGURABLE_OP_IMPL } opList { name: "relu6" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "inPlace" argType: BOOL } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "dtype" argType: DATA_TYPE } argDescriptor { name: "cutoff" argType: DOUBLE } opDeclarationType: CONFIGURABLE_OP_IMPL } opList { name: "relu6_bp" argDescriptor { name: "gradI" argType: OUTPUT_TENSOR } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "gradO" argType: INPUT_TENSOR argIndex: 1 } argDescriptor { name: "cutoff" argType: DOUBLE } opDeclarationType: CONFIGURABLE_OP_IMPL } opList { name: "relu_bp" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "epsilon" argType: INPUT_TENSOR argIndex: 1 } argDescriptor { name: "scalar" argType: DOUBLE } opDeclarationType: CONFIGURABLE_OP_IMPL } opList { name: "relu_layer" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "w" argType: INPUT_TENSOR argIndex: 1 } argDescriptor { name: "b" argType: INPUT_TENSOR argIndex: 2 } argDescriptor { name: "dtype" argType: DATA_TYPE } } opList { name: "remainder" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "inPlace" argType: BOOL } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "y" argType: INPUT_TENSOR argIndex: 1 } argDescriptor { name: "dataType" argType: DATA_TYPE } opDeclarationType: LEGACY_XYZ } opList { name: "remainder_scalar" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "inPlace" argType: BOOL } argDescriptor { name: "input" argType: INPUT_TENSOR } opDeclarationType: LEGACY_XYZ } opList { name: "repeat" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "dimensions" argType: INT64 } argDescriptor { name: "dtype" argType: DATA_TYPE } } opList { name: "replace_nans" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "inPlace" argType: BOOL } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "set" argType: DOUBLE } opDeclarationType: LEGACY_XYZ } opList { name: "reshape" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "shape" argType: INPUT_TENSOR argIndex: 1 } argDescriptor { name: "shapeArr" argType: INT64 } argDescriptor { name: "dtype" argType: DATA_TYPE } } opList { name: "reshapeas" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "y" argType: INPUT_TENSOR argIndex: 1 } argDescriptor { name: "dtype" argType: DATA_TYPE } } opList { name: "resize_area" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "alignCorners" argType: BOOL } argDescriptor { name: "image" argType: INPUT_TENSOR } argDescriptor { name: "size" 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: "resize_bicubic" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "alignCorners" argType: BOOL } argDescriptor { name: "alignPixelCenters" argType: BOOL argIndex: 1 } argDescriptor { name: "image" argType: INPUT_TENSOR } argDescriptor { name: "size" 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 } argDescriptor { name: "width" argType: INT64 argIndex: 1 } argDescriptor { name: "dtype" argType: DATA_TYPE } } opList { name: "resize_images" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "alignCorners" argType: BOOL } argDescriptor { name: "preserveAspectRatio" argType: BOOL argIndex: 1 } argDescriptor { name: "image" argType: INPUT_TENSOR } argDescriptor { name: "size" argType: INPUT_TENSOR argIndex: 1 } argDescriptor { name: "methodT" argType: INPUT_TENSOR argIndex: 2 } argDescriptor { name: "dtype" argType: DATA_TYPE } } opList { name: "resize_nearest_neighbor" 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 } argDescriptor { name: "height" argType: INT64 } argDescriptor { name: "width" argType: INT64 argIndex: 1 } argDescriptor { name: "dtype" argType: DATA_TYPE } } opList { name: "restorev2" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } } opList { name: "reverse" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "dimensions" argType: INPUT_TENSOR argIndex: 1 } argDescriptor { name: "dimensions" argType: INT64 } opDeclarationType: CONFIGURABLE_OP_IMPL } opList { name: "reverse_bp" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "grad" argType: INPUT_TENSOR argIndex: 1 } argDescriptor { name: "eps" 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 } argDescriptor { name: "batchDim" argType: INT64 argIndex: 1 } argDescriptor { name: "dtype" argType: DATA_TYPE } } opList { name: "reverse_v2" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "isLegacy" argType: BOOL } } opList { name: "reversedivide" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "inPlace" argType: BOOL } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "y" argType: INPUT_TENSOR argIndex: 1 } argDescriptor { name: "dataType" argType: DATA_TYPE } opDeclarationType: BROADCASTABLE_OP_IMPL } opList { name: "reversedivide_bp" argDescriptor { name: "gradX" argType: OUTPUT_TENSOR } argDescriptor { name: "gradY" argType: OUTPUT_TENSOR argIndex: 1 } argDescriptor { name: "inPlace" argType: BOOL } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "y" argType: INPUT_TENSOR argIndex: 1 } argDescriptor { name: "epsNext" argType: INPUT_TENSOR argIndex: 2 } argDescriptor { name: "dtype" argType: DATA_TYPE } } opList { name: "reversemod" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "y" argType: INPUT_TENSOR argIndex: 1 } opDeclarationType: BROADCASTABLE_OP_IMPL } opList { name: "reversemod_bp" argDescriptor { name: "gradX" argType: OUTPUT_TENSOR } argDescriptor { name: "gradY" argType: OUTPUT_TENSOR argIndex: 1 } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "y" argType: INPUT_TENSOR argIndex: 1 } argDescriptor { name: "epsNext" argType: INPUT_TENSOR argIndex: 2 } argDescriptor { name: "dtype" argType: DATA_TYPE } } opList { name: "reversesubtract" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "inPlace" argType: BOOL } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "y" argType: INPUT_TENSOR argIndex: 1 } argDescriptor { name: "dataType" argType: DATA_TYPE } opDeclarationType: BROADCASTABLE_OP_IMPL } opList { name: "reversesubtract_bp" argDescriptor { name: "gradX" argType: OUTPUT_TENSOR } argDescriptor { name: "gradY" argType: OUTPUT_TENSOR argIndex: 1 } argDescriptor { name: "inPlace" argType: BOOL } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "y" argType: INPUT_TENSOR argIndex: 1 } argDescriptor { name: "epsNext" 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 } argDescriptor { name: "dtype" argType: DATA_TYPE } } opList { name: "rgb_to_hsv" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "dimC" argType: INT64 } opDeclarationType: CONFIGURABLE_OP_IMPL } opList { name: "rgb_to_yiq" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "dimC" argType: INT64 } opDeclarationType: CONFIGURABLE_OP_IMPL } opList { name: "rgb_to_yuv" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "dimC" argType: INT64 } opDeclarationType: CONFIGURABLE_OP_IMPL } opList { name: "rint" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "inPlace" argType: BOOL } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "dataType" argType: DATA_TYPE } opDeclarationType: OP_IMPL } opList { name: "rms_prop_updater" argDescriptor { name: "update" argType: OUTPUT_TENSOR } argDescriptor { name: "stateG" argType: OUTPUT_TENSOR argIndex: 1 } argDescriptor { name: "gradient" argType: INPUT_TENSOR } argDescriptor { name: "initState" argType: INPUT_TENSOR argIndex: 1 } argDescriptor { name: "lr" argType: INPUT_TENSOR argIndex: 2 } argDescriptor { name: "rmsDecay" argType: INPUT_TENSOR argIndex: 3 } argDescriptor { name: "epsilon" 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: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "shiftsI" argType: INPUT_TENSOR argIndex: 1 } argDescriptor { name: "dimensions" argType: INPUT_TENSOR argIndex: 2 } argDescriptor { name: "shift" argType: INT64 } opDeclarationType: CONFIGURABLE_OP_IMPL } opList { name: "round" 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: "rshift_bits" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "inPlace" argType: BOOL } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "y" argType: INPUT_TENSOR argIndex: 1 } argDescriptor { name: "dataType" argType: DATA_TYPE } opDeclarationType: BROADCASTABLE_OP_IMPL } opList { name: "rsqrt" 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: "rsub_scalar" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "inPlace" argType: BOOL } argDescriptor { name: "input" argType: INPUT_TENSOR } opDeclarationType: LEGACY_XYZ } opList { name: "savev2" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } } opList { name: "scalar_min" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "y" argType: INPUT_TENSOR argIndex: 1 } opDeclarationType: LEGACY_XYZ } opList { name: "scatter_add" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "lock" argType: BOOL } argDescriptor { name: "checkIndices" argType: BOOL argIndex: 1 } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "indices" argType: INPUT_TENSOR argIndex: 1 } argDescriptor { name: "updates" argType: INPUT_TENSOR argIndex: 2 } opDeclarationType: OP_IMPL } opList { name: "scatter_div" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "lock" argType: BOOL } argDescriptor { name: "checkIndices" argType: BOOL argIndex: 1 } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "indices" argType: INPUT_TENSOR argIndex: 1 } argDescriptor { name: "updates" argType: INPUT_TENSOR argIndex: 2 } opDeclarationType: OP_IMPL } opList { name: "scatter_list" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "list" argType: INPUT_TENSOR } argDescriptor { name: "array" argType: INPUT_TENSOR argIndex: 1 } argDescriptor { name: "sizes" argType: INPUT_TENSOR argIndex: 2 } argDescriptor { name: "dtype" argType: DATA_TYPE } opDeclarationType: LIST_OP_IMPL } opList { name: "scatter_max" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "lock" argType: BOOL } argDescriptor { name: "checkIndices" argType: BOOL argIndex: 1 } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "indices" argType: INPUT_TENSOR argIndex: 1 } argDescriptor { name: "updates" argType: INPUT_TENSOR argIndex: 2 } opDeclarationType: OP_IMPL } opList { name: "scatter_min" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "lock" argType: BOOL } argDescriptor { name: "checkIndices" argType: BOOL argIndex: 1 } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "indices" argType: INPUT_TENSOR argIndex: 1 } argDescriptor { name: "updates" argType: INPUT_TENSOR argIndex: 2 } opDeclarationType: OP_IMPL } opList { name: "scatter_mul" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "lock" argType: BOOL } argDescriptor { name: "checkIndices" argType: BOOL argIndex: 1 } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "indices" argType: INPUT_TENSOR argIndex: 1 } argDescriptor { name: "updates" argType: INPUT_TENSOR argIndex: 2 } opDeclarationType: OP_IMPL } opList { name: "scatter_nd" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "lock" argType: BOOL } argDescriptor { name: "checkIndices" argType: BOOL argIndex: 1 } argDescriptor { name: "indices" argType: INPUT_TENSOR } argDescriptor { name: "updates" argType: INPUT_TENSOR argIndex: 1 } argDescriptor { name: "shape" argType: INPUT_TENSOR argIndex: 2 } argDescriptor { name: "dtype" argType: DATA_TYPE } } opList { name: "scatter_nd_add" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "lock" argType: BOOL } argDescriptor { name: "checkIndices" argType: BOOL argIndex: 1 } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "indices" argType: INPUT_TENSOR argIndex: 1 } argDescriptor { name: "updates" argType: INPUT_TENSOR argIndex: 2 } opDeclarationType: OP_IMPL } opList { name: "scatter_nd_sub" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "lock" argType: BOOL } argDescriptor { name: "checkIndices" argType: BOOL argIndex: 1 } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "indices" argType: INPUT_TENSOR argIndex: 1 } argDescriptor { name: "updates" argType: INPUT_TENSOR argIndex: 2 } opDeclarationType: OP_IMPL } opList { name: "scatter_nd_update" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "lock" argType: BOOL } argDescriptor { name: "checkIndices" argType: BOOL argIndex: 1 } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "indices" argType: INPUT_TENSOR argIndex: 1 } argDescriptor { name: "updates" argType: INPUT_TENSOR argIndex: 2 } opDeclarationType: OP_IMPL } opList { name: "scatter_sub" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "lock" argType: BOOL } argDescriptor { name: "checkIndices" argType: BOOL argIndex: 1 } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "indices" argType: INPUT_TENSOR argIndex: 1 } argDescriptor { name: "updates" argType: INPUT_TENSOR argIndex: 2 } opDeclarationType: OP_IMPL } opList { name: "scatter_upd" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "lock" argType: BOOL } argDescriptor { name: "checkIndices" argType: BOOL argIndex: 1 } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "indices" argType: INPUT_TENSOR argIndex: 1 } argDescriptor { name: "updates" argType: INPUT_TENSOR argIndex: 2 } opDeclarationType: OP_IMPL } opList { name: "scatter_update" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "operand" argType: INPUT_TENSOR } argDescriptor { name: "updates" argType: INPUT_TENSOR argIndex: 1 } argDescriptor { name: "indices" argType: INT64 } argDescriptor { name: "dimension" argType: INT64 argIndex: 1 } 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 } 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: "sconv2d_bp" argDescriptor { name: "*gradI" argType: OUTPUT_TENSOR } argDescriptor { name: "*gradWD" argType: OUTPUT_TENSOR argIndex: 1 } argDescriptor { name: "gradWP" argType: OUTPUT_TENSOR argIndex: 2 } argDescriptor { name: "gradB" argType: OUTPUT_TENSOR argIndex: 3 } argDescriptor { name: "*input" argType: INPUT_TENSOR } argDescriptor { name: "*gradO" argType: INPUT_TENSOR argIndex: 1 } argDescriptor { name: "*weightsDepth" argType: INPUT_TENSOR argIndex: 2 } argDescriptor { name: "weightsPoint" 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: "segmentedOutput" argType: OUTPUT_TENSOR } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "idxSegments" argType: INPUT_TENSOR argIndex: 1 } argDescriptor { name: "dtype" argType: DATA_TYPE } } opList { name: "segment_max_bp" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "outIndices" argType: OUTPUT_TENSOR argIndex: 1 } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "indices" argType: INPUT_TENSOR argIndex: 1 } argDescriptor { name: "gradOut" argType: INPUT_TENSOR argIndex: 2 } argDescriptor { name: "dtype" argType: DATA_TYPE } } opList { name: "segment_mean" argDescriptor { name: "segmentedOutput" argType: OUTPUT_TENSOR } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "idxSegments" argType: INPUT_TENSOR argIndex: 1 } argDescriptor { name: "dtype" argType: DATA_TYPE } } opList { name: "segment_mean_bp" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "outIndices" argType: OUTPUT_TENSOR argIndex: 1 } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "indices" argType: INPUT_TENSOR argIndex: 1 } argDescriptor { name: "gradOut" argType: INPUT_TENSOR argIndex: 2 } argDescriptor { name: "dtype" argType: DATA_TYPE } } opList { name: "segment_min" argDescriptor { name: "segmentedOutput" argType: OUTPUT_TENSOR } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "idxSegments" argType: INPUT_TENSOR argIndex: 1 } argDescriptor { name: "dtype" argType: DATA_TYPE } } opList { name: "segment_min_bp" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "outIndices" argType: OUTPUT_TENSOR argIndex: 1 } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "indices" argType: INPUT_TENSOR argIndex: 1 } argDescriptor { name: "gradOut" argType: INPUT_TENSOR argIndex: 2 } argDescriptor { name: "dtype" argType: DATA_TYPE } } opList { name: "segment_prod" argDescriptor { name: "segmentedOutput" argType: OUTPUT_TENSOR } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "idxSegments" argType: INPUT_TENSOR argIndex: 1 } argDescriptor { name: "dtype" argType: DATA_TYPE } } opList { name: "segment_prod_bp" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "outIndices" argType: OUTPUT_TENSOR argIndex: 1 } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "indices" argType: INPUT_TENSOR argIndex: 1 } argDescriptor { name: "gradOut" argType: INPUT_TENSOR argIndex: 2 } argDescriptor { name: "dtype" argType: DATA_TYPE } } opList { name: "segment_sum" argDescriptor { name: "segmentedOutput" argType: OUTPUT_TENSOR } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "idxSegments" argType: INPUT_TENSOR argIndex: 1 } argDescriptor { name: "dtype" argType: DATA_TYPE } } opList { name: "segment_sum_bp" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "data" argType: INPUT_TENSOR } argDescriptor { name: "segmentIds" argType: INPUT_TENSOR argIndex: 1 } argDescriptor { name: "gradient" argType: INPUT_TENSOR argIndex: 2 } argDescriptor { name: "dtype" argType: DATA_TYPE } } opList { name: "select" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "cond" argType: INPUT_TENSOR } argDescriptor { name: "input" argType: INPUT_TENSOR argIndex: 1 } argDescriptor { name: "y" argType: INPUT_TENSOR argIndex: 2 } argDescriptor { name: "dtype" argType: DATA_TYPE } } opList { name: "selu" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "inPlace" argType: BOOL } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "dataType" argType: DATA_TYPE } opDeclarationType: CONFIGURABLE_OP_IMPL } opList { name: "selu_bp" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "epsilon" argType: INPUT_TENSOR argIndex: 1 } opDeclarationType: CONFIGURABLE_OP_IMPL } opList { name: "seluderivative" 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: "sequence_mask" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "is_static_maxlen" argType: BOOL } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "maxlen" 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: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "y" argType: INPUT_TENSOR argIndex: 1 } argDescriptor { name: "dataType" argType: DATA_TYPE } opDeclarationType: LEGACY_XYZ } opList { name: "set_scalar" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "y" argType: INPUT_TENSOR argIndex: 1 } opDeclarationType: LEGACY_XYZ } opList { name: "set_seed" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "seed" argType: INT64 } argDescriptor { name: "dtype" argType: DATA_TYPE } } opList { name: "setrange" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "inPlace" argType: BOOL } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "dataType" argType: DATA_TYPE } argDescriptor { name: "min" argType: DOUBLE } argDescriptor { name: "max" argType: DOUBLE argIndex: 1 } opDeclarationType: LEGACY_XYZ } opList { name: "setvalorless_scalar" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "inPlace" argType: BOOL } argDescriptor { name: "input" argType: INPUT_TENSOR } opDeclarationType: LEGACY_XYZ } opList { name: "sgd_updater" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "lr" argType: INPUT_TENSOR argIndex: 1 } argDescriptor { name: "lr" argType: DOUBLE } opDeclarationType: CONFIGURABLE_OP_IMPL } opList { name: "shannonentropy" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "keepDims" argType: BOOL } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "dimensions" argType: INPUT_TENSOR argIndex: 1 } argDescriptor { name: "dimensions" argType: INT64 } opDeclarationType: LEGACY_XYZ } opList { name: "shape_of" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "inPlace" argType: BOOL } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "dtype" argType: DATA_TYPE } } opList { name: "shapes_of" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "inPlace" argType: BOOL } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "dtype" argType: DATA_TYPE } } opList { name: "shift_bits" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "inPlace" argType: BOOL } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "y" argType: INPUT_TENSOR argIndex: 1 } argDescriptor { name: "dataType" argType: DATA_TYPE } opDeclarationType: BROADCASTABLE_OP_IMPL } opList { name: "sigm_cross_entropy_loss" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "logits" argType: INPUT_TENSOR } argDescriptor { name: "weights" argType: INPUT_TENSOR argIndex: 1 } argDescriptor { name: "labels" argType: INPUT_TENSOR argIndex: 2 } argDescriptor { name: "reductionMode" argType: INT64 } argDescriptor { name: "dtype" argType: DATA_TYPE } argDescriptor { name: "labelsSmoothing" argType: DOUBLE } } opList { name: "sigm_cross_entropy_loss_grad" argDescriptor { name: "dLdp" argType: OUTPUT_TENSOR } argDescriptor { name: "dLdw" argType: OUTPUT_TENSOR argIndex: 1 } argDescriptor { name: "dLdl" argType: OUTPUT_TENSOR argIndex: 2 } argDescriptor { name: "logits" argType: INPUT_TENSOR } argDescriptor { name: "weights" argType: INPUT_TENSOR argIndex: 1 } argDescriptor { name: "labels" 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: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "inPlace" argType: BOOL } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "dataType" argType: DATA_TYPE } opDeclarationType: CONFIGURABLE_OP_IMPL } opList { name: "sigmoid_bp" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "epsilon" argType: INPUT_TENSOR argIndex: 1 } opDeclarationType: CONFIGURABLE_OP_IMPL } opList { name: "sign" 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: "sin" 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: "sinh" 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: "size" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { 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 } argDescriptor { name: "dtype" argType: DATA_TYPE } } opList { name: "size_list" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "list" argType: INPUT_TENSOR } opDeclarationType: LIST_OP_IMPL } opList { name: "skipgram" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "isInference" argType: BOOL } argDescriptor { name: "isPreciseMode" argType: BOOL argIndex: 1 } argDescriptor { name: "target" argType: INPUT_TENSOR } argDescriptor { name: "ngStarter" argType: INPUT_TENSOR argIndex: 1 } argDescriptor { name: "indices" argType: INPUT_TENSOR argIndex: 2 } argDescriptor { name: "codes" argType: INPUT_TENSOR argIndex: 3 } argDescriptor { name: "syn0" argType: INPUT_TENSOR argIndex: 4 } argDescriptor { name: "syn1" argType: INPUT_TENSOR argIndex: 5 } argDescriptor { name: "syn1neg" argType: INPUT_TENSOR argIndex: 6 } argDescriptor { name: "expTable" argType: INPUT_TENSOR argIndex: 7 } argDescriptor { name: "negTable" argType: INPUT_TENSOR argIndex: 8 } argDescriptor { name: "alpha" argType: INPUT_TENSOR argIndex: 9 } argDescriptor { name: "randomValue" argType: INPUT_TENSOR argIndex: 10 } argDescriptor { name: "inferenceVector" 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: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "b" argType: INPUT_TENSOR argIndex: 1 } argDescriptor { name: "e" argType: INPUT_TENSOR argIndex: 2 } argDescriptor { name: "size" argType: INT64 } argDescriptor { name: "dtype" argType: DATA_TYPE } } opList { name: "slice_bp" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "b" argType: INPUT_TENSOR argIndex: 1 } argDescriptor { name: "e" argType: INPUT_TENSOR argIndex: 2 } argDescriptor { name: "epsNext" argType: INPUT_TENSOR argIndex: 3 } argDescriptor { name: "size" argType: INT64 } argDescriptor { name: "dtype" argType: DATA_TYPE } } opList { name: "softmax" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "inPlace" argType: BOOL } argDescriptor { 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: "gradI" argType: OUTPUT_TENSOR } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "gradO" argType: INPUT_TENSOR argIndex: 1 } argDescriptor { name: "dimension" argType: INT64 } opDeclarationType: CONFIGURABLE_OP_IMPL } opList { name: "softmax_cross_entropy_loss" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "logits" argType: INPUT_TENSOR } argDescriptor { name: "weights" argType: INPUT_TENSOR argIndex: 1 } argDescriptor { name: "labels" argType: INPUT_TENSOR argIndex: 2 } argDescriptor { name: "reductionMode" argType: INT64 } argDescriptor { name: "dtype" argType: DATA_TYPE } argDescriptor { name: "labelsSmoothing" argType: DOUBLE } } opList { name: "softmax_cross_entropy_loss_grad" argDescriptor { name: "dLdp" argType: OUTPUT_TENSOR } argDescriptor { name: "dLdw" argType: OUTPUT_TENSOR argIndex: 1 } argDescriptor { name: "dLdl" argType: OUTPUT_TENSOR argIndex: 2 } argDescriptor { name: "logits" argType: INPUT_TENSOR } argDescriptor { name: "weights" argType: INPUT_TENSOR argIndex: 1 } argDescriptor { name: "labels" argType: INPUT_TENSOR argIndex: 2 } argDescriptor { name: "reductionMode" 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 } argDescriptor { name: "logits" argType: INPUT_TENSOR } argDescriptor { name: "labels" argType: INPUT_TENSOR argIndex: 1 } argDescriptor { name: "classesDim" argType: INT64 } argDescriptor { name: "dtype" argType: DATA_TYPE } } opList { name: "softmax_cross_entropy_loss_with_logits_grad" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "dLdl" argType: OUTPUT_TENSOR argIndex: 1 } argDescriptor { name: "logits" argType: INPUT_TENSOR } argDescriptor { name: "labels" argType: INPUT_TENSOR argIndex: 1 } argDescriptor { name: "classesDim" argType: INT64 } argDescriptor { name: "dtype" argType: DATA_TYPE } } opList { name: "softplus" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "inPlace" argType: BOOL } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "dataType" argType: DATA_TYPE } opDeclarationType: CONFIGURABLE_OP_IMPL } opList { name: "softplus_bp" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "epsilon" argType: INPUT_TENSOR argIndex: 1 } opDeclarationType: CONFIGURABLE_OP_IMPL } opList { name: "softsign" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "inPlace" argType: BOOL } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "dataType" argType: DATA_TYPE } opDeclarationType: CONFIGURABLE_OP_IMPL } opList { name: "softsign_bp" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "epsilon" argType: INPUT_TENSOR argIndex: 1 } opDeclarationType: CONFIGURABLE_OP_IMPL } opList { name: "softsignderivative" 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: "solve" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "useAdjoint" argType: BOOL } argDescriptor { name: "a" argType: INPUT_TENSOR } argDescriptor { name: "b" argType: INPUT_TENSOR argIndex: 1 } argDescriptor { name: "adjoint" argType: INPUT_TENSOR argIndex: 2 } argDescriptor { name: "dtype" argType: DATA_TYPE } } opList { name: "solve_ls" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "fastFlag" argType: BOOL } argDescriptor { name: "a" argType: INPUT_TENSOR } argDescriptor { name: "b" argType: INPUT_TENSOR argIndex: 1 } argDescriptor { name: "dtype" argType: DATA_TYPE } argDescriptor { name: "l2_factor" argType: DOUBLE } } opList { name: "somepoolingpool2d" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "input" argType: INPUT_TENSOR } } opList { name: "somepoolingpool2d_bp" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "grad" argType: INPUT_TENSOR argIndex: 1 } } opList { name: "space_to_batch" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "inPlace" argType: BOOL } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "padding" argType: INPUT_TENSOR argIndex: 1 } argDescriptor { name: "blockSize" argType: INT64 } argDescriptor { name: "dtype" argType: DATA_TYPE } } opList { name: "space_to_batch_nd" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "inPlace" argType: BOOL } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "blockShape" argType: INPUT_TENSOR argIndex: 1 } argDescriptor { name: "padding" 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 } argDescriptor { name: "isNHWC" argType: INT64 argIndex: 1 } argDescriptor { name: "dtype" argType: DATA_TYPE } } opList { name: "sparse_softmax_cross_entropy_loss_with_logits" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "labels" argType: INPUT_TENSOR } argDescriptor { name: "logits" argType: INPUT_TENSOR argIndex: 1 } argDescriptor { name: "dtype" argType: DATA_TYPE } } opList { name: "sparse_softmax_cross_entropy_loss_with_logits_grad" argDescriptor { name: "dLdp" argType: OUTPUT_TENSOR } argDescriptor { name: "labels" argType: INPUT_TENSOR } argDescriptor { name: "logits" 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 } argDescriptor { name: "dimensions" argType: INT64 argIndex: 1 } argDescriptor { name: "dtype" argType: DATA_TYPE } } opList { name: "split_list" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "list" argType: INPUT_TENSOR } argDescriptor { name: "array" argType: INPUT_TENSOR argIndex: 1 } argDescriptor { name: "sizes" argType: INPUT_TENSOR argIndex: 2 } argDescriptor { name: "dtype" argType: DATA_TYPE } opDeclarationType: LIST_OP_IMPL } opList { name: "split_string" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "delim" argType: INPUT_TENSOR argIndex: 1 } argDescriptor { name: "dtype" argType: DATA_TYPE } } opList { name: "split_v" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "sizes" argType: INPUT_TENSOR argIndex: 1 } argDescriptor { name: "_a" 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: "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: "sqrtm" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "input" argType: INPUT_TENSOR } opDeclarationType: CONFIGURABLE_OP_IMPL } opList { name: "square" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "inPlace" argType: BOOL } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "dataType" argType: DATA_TYPE } opDeclarationType: OP_IMPL } opList { name: "squaredsubtract" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "inPlace" argType: BOOL } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "y" argType: INPUT_TENSOR argIndex: 1 } argDescriptor { name: "dataType" argType: DATA_TYPE } opDeclarationType: BROADCASTABLE_OP_IMPL } opList { name: "squaredsubtract_bp" argDescriptor { name: "gradX" argType: OUTPUT_TENSOR } argDescriptor { name: "gradY" argType: OUTPUT_TENSOR argIndex: 1 } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "y" argType: INPUT_TENSOR argIndex: 1 } argDescriptor { name: "epsNext" argType: INPUT_TENSOR argIndex: 2 } argDescriptor { name: "dtype" argType: DATA_TYPE } } opList { name: "squeeze" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "a" argType: INPUT_TENSOR argIndex: 1 } argDescriptor { name: "_a" argType: INT64 } argDescriptor { name: "dtype" argType: DATA_TYPE } } opList { name: "sru" argDescriptor { name: "h" argType: OUTPUT_TENSOR } argDescriptor { name: "c" argType: OUTPUT_TENSOR argIndex: 1 } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "w" argType: INPUT_TENSOR argIndex: 1 } argDescriptor { name: "b" argType: INPUT_TENSOR argIndex: 2 } argDescriptor { name: "c0" argType: INPUT_TENSOR argIndex: 3 } argDescriptor { name: "mask" argType: INPUT_TENSOR argIndex: 4 } argDescriptor { name: "dtype" argType: DATA_TYPE } } opList { name: "sruCell" argDescriptor { name: "ht" argType: OUTPUT_TENSOR } argDescriptor { name: "ct" argType: OUTPUT_TENSOR argIndex: 1 } argDescriptor { name: "xt" argType: INPUT_TENSOR } argDescriptor { name: "ct_1" argType: INPUT_TENSOR argIndex: 1 } argDescriptor { name: "w" argType: INPUT_TENSOR argIndex: 2 } argDescriptor { name: "b" argType: INPUT_TENSOR argIndex: 3 } argDescriptor { name: "dtype" argType: DATA_TYPE } } opList { name: "sru_bi" argDescriptor { name: "ht" argType: OUTPUT_TENSOR } argDescriptor { name: "ct" argType: OUTPUT_TENSOR argIndex: 1 } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "w" argType: INPUT_TENSOR argIndex: 1 } argDescriptor { name: "b" argType: INPUT_TENSOR argIndex: 2 } argDescriptor { name: "c0" argType: INPUT_TENSOR argIndex: 3 } argDescriptor { name: "mask" argType: INPUT_TENSOR argIndex: 4 } argDescriptor { name: "dtype" argType: DATA_TYPE } } opList { name: "sru_bi_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: "gradC0" argType: OUTPUT_TENSOR argIndex: 3 } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "w" argType: INPUT_TENSOR argIndex: 1 } argDescriptor { name: "b" argType: INPUT_TENSOR argIndex: 2 } argDescriptor { name: "c0" argType: INPUT_TENSOR argIndex: 3 } argDescriptor { name: "ct" argType: INPUT_TENSOR argIndex: 4 } argDescriptor { name: "inGradC0" argType: INPUT_TENSOR argIndex: 5 } argDescriptor { name: "inGradHt" argType: INPUT_TENSOR argIndex: 6 } argDescriptor { name: "mask" argType: INPUT_TENSOR argIndex: 7 } argDescriptor { name: "dtype" argType: DATA_TYPE } } opList { name: "sru_bp" argDescriptor { name: "gradX" argType: OUTPUT_TENSOR } argDescriptor { name: "gradW" argType: OUTPUT_TENSOR argIndex: 1 } argDescriptor { name: "gradB" argType: OUTPUT_TENSOR argIndex: 2 } argDescriptor { name: "gradInit" argType: OUTPUT_TENSOR argIndex: 3 } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "w" argType: INPUT_TENSOR argIndex: 1 } argDescriptor { name: "b" argType: INPUT_TENSOR argIndex: 2 } argDescriptor { name: "c0" argType: INPUT_TENSOR argIndex: 3 } argDescriptor { name: "c" argType: INPUT_TENSOR argIndex: 4 } argDescriptor { name: "inGradCt" argType: INPUT_TENSOR argIndex: 5 } argDescriptor { name: "inGradH" argType: INPUT_TENSOR argIndex: 6 } argDescriptor { name: "mask" argType: INPUT_TENSOR argIndex: 7 } argDescriptor { name: "dtype" argType: DATA_TYPE } } opList { name: "stabilize" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "inPlace" argType: BOOL } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "dataType" argType: DATA_TYPE } argDescriptor { name: "realMin" argType: DOUBLE } argDescriptor { name: "cutOff" argType: DOUBLE argIndex: 1 } argDescriptor { name: "k" argType: DOUBLE argIndex: 2 } opDeclarationType: LEGACY_XYZ } opList { name: "stack" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "inArrs" argType: INPUT_TENSOR argIndex: 1 } argDescriptor { name: "dimensions" argType: INT64 } argDescriptor { name: "dtype" argType: DATA_TYPE } } opList { name: "stack_list" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "list" argType: INPUT_TENSOR } argDescriptor { name: "dtype" argType: DATA_TYPE } opDeclarationType: LIST_OP_IMPL } opList { name: "standardize" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "dimensions" argType: INPUT_TENSOR argIndex: 1 } argDescriptor { name: "dimensions" argType: INT64 } opDeclarationType: CONFIGURABLE_OP_IMPL } opList { name: "standardize_bp" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "dimensions" argType: INPUT_TENSOR argIndex: 1 } argDescriptor { name: "eps" argType: INPUT_TENSOR argIndex: 2 } argDescriptor { name: "dimensions" argType: INT64 } argDescriptor { name: "dtype" argType: DATA_TYPE } } opList { name: "static_bidirectional_rnn" argDescriptor { name: "h" argType: OUTPUT_TENSOR } argDescriptor { name: "hFWFinal" argType: OUTPUT_TENSOR argIndex: 1 } argDescriptor { name: "hBWFinal" argType: OUTPUT_TENSOR argIndex: 2 } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "WxFW" argType: INPUT_TENSOR argIndex: 1 } argDescriptor { name: "WhFW" argType: INPUT_TENSOR argIndex: 2 } argDescriptor { name: "bFW" argType: INPUT_TENSOR argIndex: 3 } argDescriptor { name: "WxBW" argType: INPUT_TENSOR argIndex: 4 } argDescriptor { name: "WhBW" argType: INPUT_TENSOR argIndex: 5 } argDescriptor { name: "bBW" argType: INPUT_TENSOR argIndex: 6 } argDescriptor { name: "h0FW" argType: INPUT_TENSOR argIndex: 7 } argDescriptor { name: "h0BW" argType: INPUT_TENSOR argIndex: 8 } argDescriptor { name: "dtype" argType: DATA_TYPE } } opList { name: "static_rnn" argDescriptor { name: "h" argType: OUTPUT_TENSOR } argDescriptor { name: "hFinal" argType: OUTPUT_TENSOR argIndex: 1 } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "Wx" argType: INPUT_TENSOR argIndex: 1 } argDescriptor { name: "Wh" argType: INPUT_TENSOR argIndex: 2 } argDescriptor { name: "b" argType: INPUT_TENSOR argIndex: 3 } argDescriptor { name: "h0" argType: INPUT_TENSOR argIndex: 4 } argDescriptor { name: "dtype" argType: DATA_TYPE } } opList { name: "std" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "newFormat" argType: BOOL } argDescriptor { name: "keepDims" argType: BOOL argIndex: 1 } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "dimensions" argType: INT64 } opDeclarationType: LEGACY_XYZ } opList { name: "step" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "inPlace" argType: BOOL } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "cutoff" argType: DOUBLE } opDeclarationType: LEGACY_XYZ } opList { name: "stop_gradient" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "inPlace" argType: BOOL } argDescriptor { 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 } 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: "strided_slice_bp" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "epsNext" argType: INPUT_TENSOR argIndex: 1 } argDescriptor { name: "v_begin" argType: INPUT_TENSOR argIndex: 2 } argDescriptor { name: "v_end" argType: INPUT_TENSOR argIndex: 3 } argDescriptor { name: "v_stride" 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" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "inPlace" argType: BOOL } argDescriptor { name: "input" argType: INPUT_TENSOR } opDeclarationType: LEGACY_XYZ } opList { name: "subtract" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "inPlace" argType: BOOL } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "y" argType: INPUT_TENSOR argIndex: 1 } argDescriptor { name: "dataType" argType: DATA_TYPE } opDeclarationType: BROADCASTABLE_OP_IMPL } opList { name: "subtract_bp" argDescriptor { name: "gradX" argType: OUTPUT_TENSOR } argDescriptor { name: "gradY" argType: OUTPUT_TENSOR argIndex: 1 } argDescriptor { name: "inPlace" argType: BOOL } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "y" argType: INPUT_TENSOR argIndex: 1 } argDescriptor { name: "epsNext" argType: INPUT_TENSOR argIndex: 2 } argDescriptor { name: "dtype" argType: DATA_TYPE } } opList { name: "sufficient_statistics" argDescriptor { name: "dataCount" argType: OUTPUT_TENSOR } argDescriptor { name: "sum" argType: OUTPUT_TENSOR argIndex: 1 } argDescriptor { name: "squares" argType: OUTPUT_TENSOR argIndex: 2 } argDescriptor { name: "shift" argType: OUTPUT_TENSOR argIndex: 3 } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "dimensions" argType: INPUT_TENSOR argIndex: 1 } argDescriptor { name: "shift" argType: INPUT_TENSOR argIndex: 2 } argDescriptor { name: "dtype" argType: DATA_TYPE } } opList { name: "svd" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "fullUV" argType: BOOL } argDescriptor { name: "computeUv" argType: BOOL argIndex: 1 } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "s" argType: INPUT_TENSOR argIndex: 1 } argDescriptor { name: "u" argType: INPUT_TENSOR argIndex: 2 } argDescriptor { name: "v" 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: "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: "switch" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "predicate" argType: INPUT_TENSOR argIndex: 1 } argDescriptor { name: "frameName" argType: STRING } } opList { name: "tan" 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: "tanderivative" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "dataType" argType: DATA_TYPE } opDeclarationType: LEGACY_XYZ } opList { name: "tanh" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "inPlace" argType: BOOL } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "dataType" argType: DATA_TYPE } opDeclarationType: CONFIGURABLE_OP_IMPL } opList { name: "tanh_bp" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "epsilon" argType: INPUT_TENSOR argIndex: 1 } opDeclarationType: CONFIGURABLE_OP_IMPL } opList { name: "tear" argDescriptor { name: "outE" argType: OUTPUT_TENSOR } argDescriptor { 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 } } opList { name: "tensorarraywritev3" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "input" argType: INPUT_TENSOR } } opList { name: "tensordot" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "addedEdges" argType: BOOL } argDescriptor { name: "transposeY" argType: BOOL argIndex: 1 } argDescriptor { name: "transposeZ" argType: BOOL argIndex: 2 } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "y" 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 } argDescriptor { name: "axe1_size" argType: INT64 argIndex: 1 } argDescriptor { name: "dtype" argType: DATA_TYPE } } opList { name: "tensormmul_bp" argDescriptor { name: "dLdA" argType: OUTPUT_TENSOR } argDescriptor { name: "dLdB" argType: OUTPUT_TENSOR argIndex: 1 } argDescriptor { name: "A" argType: INPUT_TENSOR } argDescriptor { name: "B" argType: INPUT_TENSOR argIndex: 1 } argDescriptor { name: "dLdC" argType: INPUT_TENSOR argIndex: 2 } argDescriptor { name: "axe0Size" argType: INT64 } argDescriptor { name: "dtype" argType: DATA_TYPE } } opList { name: "test_output_reshape" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "input" argType: INPUT_TENSOR } opDeclarationType: OP_IMPL } opList { name: "test_scalar" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { 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 } } opList { name: "testop2i2o" argDescriptor { name: "xO" argType: OUTPUT_TENSOR } argDescriptor { name: "yO" argType: OUTPUT_TENSOR argIndex: 1 } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "y" argType: INPUT_TENSOR argIndex: 1 } opDeclarationType: OP_IMPL } opList { name: "testreduction" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } opDeclarationType: REDUCTION_OP_IMPL } opList { name: "tf_atan2" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "inPlace" argType: BOOL } argDescriptor { name: "y" argType: INPUT_TENSOR } argDescriptor { name: "input" argType: INPUT_TENSOR argIndex: 1 } argDescriptor { name: "dataType" argType: DATA_TYPE } opDeclarationType: BROADCASTABLE_OP_IMPL } opList { name: "thresholdedrelu" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "inPlace" argType: BOOL } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "cutoff" argType: DOUBLE } opDeclarationType: CONFIGURABLE_OP_IMPL } opList { name: "thresholdedrelu_bp" argDescriptor { name: "dLdI" argType: OUTPUT_TENSOR } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "dLdO" argType: INPUT_TENSOR argIndex: 1 } argDescriptor { name: "cutoff" argType: DOUBLE } opDeclarationType: CONFIGURABLE_OP_IMPL } opList { name: "tile" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "is_static_reps" argType: BOOL } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "reps_vector" argType: INPUT_TENSOR argIndex: 1 } argDescriptor { name: "dimensions" argType: INT64 } argDescriptor { name: "dtype" argType: DATA_TYPE } } opList { name: "tile_bp" argDescriptor { name: "gradI" argType: OUTPUT_TENSOR } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "gradO" argType: INPUT_TENSOR argIndex: 1 } argDescriptor { name: "repeat" argType: INT64 } argDescriptor { name: "dtype" argType: DATA_TYPE } } opList { name: "tile_to_shape" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "dtype" argType: DATA_TYPE } } opList { name: "tile_to_shape_bp" argDescriptor { name: "gradX" argType: OUTPUT_TENSOR } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "epsNext" argType: INPUT_TENSOR argIndex: 1 } argDescriptor { name: "dtype" argType: DATA_TYPE } } opList { name: "timesoneminus" 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: "to_double" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "dtype" argType: DATA_TYPE } } opList { name: "to_float16" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "dtype" argType: DATA_TYPE } } opList { name: "to_float32" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "dtype" argType: DATA_TYPE } } opList { name: "to_int32" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "dtype" argType: DATA_TYPE } } opList { name: "to_int64" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "dtype" argType: DATA_TYPE } } opList { name: "to_uint32" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "dtype" argType: DATA_TYPE } } opList { name: "to_uint64" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "dtype" argType: DATA_TYPE } } opList { name: "toggle_bits" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "input" argType: INPUT_TENSOR } opDeclarationType: OP_IMPL } opList { name: "top_k" argDescriptor { name: "values" argType: OUTPUT_TENSOR } argDescriptor { name: "indices" argType: OUTPUT_TENSOR argIndex: 1 } argDescriptor { name: "needSort" argType: BOOL } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "k" argType: INT64 } argDescriptor { name: "dtype" argType: DATA_TYPE } } opList { name: "trace" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "dtype" argType: DATA_TYPE } } opList { name: "transpose" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "permuteDims" 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 } argDescriptor { name: "column" argType: INT64 argIndex: 1 } argDescriptor { name: "diag" argType: INT64 argIndex: 2 } argDescriptor { name: "dtype" argType: DATA_TYPE } } opList { name: "triangular_solve" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "isLower" argType: BOOL } argDescriptor { name: "useAdjoint" argType: BOOL argIndex: 1 } argDescriptor { name: "a" argType: INPUT_TENSOR } argDescriptor { name: "b" argType: INPUT_TENSOR argIndex: 1 } argDescriptor { name: "lower" argType: INPUT_TENSOR argIndex: 2 } argDescriptor { name: "adjoint" 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 } argDescriptor { name: "dtype" argType: DATA_TYPE } } opList { name: "triu_bp" argDescriptor { name: "gradI" argType: OUTPUT_TENSOR } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "gradO" argType: INPUT_TENSOR argIndex: 1 } argDescriptor { name: "diag" argType: INT64 } argDescriptor { name: "dtype" argType: DATA_TYPE } } opList { name: "truncatediv" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "inPlace" argType: BOOL } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "y" argType: INPUT_TENSOR argIndex: 1 } argDescriptor { name: "dataType" argType: DATA_TYPE } opDeclarationType: BROADCASTABLE_OP_IMPL } opList { name: "unique" argDescriptor { name: "values" argType: OUTPUT_TENSOR } argDescriptor { name: "indices" argType: OUTPUT_TENSOR argIndex: 1 } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "dtype" argType: DATA_TYPE } } opList { name: "unique_with_counts" argDescriptor { name: "values" argType: OUTPUT_TENSOR } argDescriptor { name: "indices" argType: OUTPUT_TENSOR argIndex: 1 } argDescriptor { name: "counts" argType: OUTPUT_TENSOR argIndex: 2 } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "dtype" argType: DATA_TYPE } } opList { name: "unsorted_segment_max" argDescriptor { name: "segmentedOutput" argType: OUTPUT_TENSOR } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "idxSegments" argType: INPUT_TENSOR argIndex: 1 } argDescriptor { name: "numSegments" argType: INPUT_TENSOR argIndex: 2 } argDescriptor { name: "numSegments" argType: INT64 } argDescriptor { name: "dtype" argType: DATA_TYPE } } opList { name: "unsorted_segment_max_bp" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "idxSegments" argType: INPUT_TENSOR argIndex: 1 } argDescriptor { name: "numSegments" argType: INPUT_TENSOR argIndex: 2 } argDescriptor { name: "numSegments" argType: INT64 } argDescriptor { name: "dtype" argType: DATA_TYPE } } opList { name: "unsorted_segment_mean" argDescriptor { name: "segmentedOutput" argType: OUTPUT_TENSOR } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "idxSegments" argType: INPUT_TENSOR argIndex: 1 } argDescriptor { name: "numSegments" argType: INPUT_TENSOR argIndex: 2 } argDescriptor { name: "numSegments" argType: INT64 } argDescriptor { name: "dtype" argType: DATA_TYPE } } opList { name: "unsorted_segment_mean_bp" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "idxSegments" argType: INPUT_TENSOR argIndex: 1 } argDescriptor { name: "numSegments" argType: INPUT_TENSOR argIndex: 2 } argDescriptor { name: "numSegments" argType: INT64 } argDescriptor { name: "dtype" argType: DATA_TYPE } } opList { name: "unsorted_segment_min" argDescriptor { name: "segmentedOutput" argType: OUTPUT_TENSOR } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "idxSegments" argType: INPUT_TENSOR argIndex: 1 } argDescriptor { name: "numSegments" argType: INPUT_TENSOR argIndex: 2 } argDescriptor { name: "numSegments" argType: INT64 } argDescriptor { name: "dtype" argType: DATA_TYPE } } opList { name: "unsorted_segment_min_bp" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "idxSegments" argType: INPUT_TENSOR argIndex: 1 } argDescriptor { name: "numSegments" argType: INPUT_TENSOR argIndex: 2 } argDescriptor { name: "numSegments" argType: INT64 } argDescriptor { name: "dtype" argType: DATA_TYPE } } opList { name: "unsorted_segment_prod" argDescriptor { name: "segmentedOutput" argType: OUTPUT_TENSOR } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "idxSegments" argType: INPUT_TENSOR argIndex: 1 } argDescriptor { name: "numSegments" argType: INPUT_TENSOR argIndex: 2 } argDescriptor { name: "numSegments" argType: INT64 } argDescriptor { name: "dtype" argType: DATA_TYPE } } opList { name: "unsorted_segment_prod_bp" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "idxSegments" argType: INPUT_TENSOR argIndex: 1 } argDescriptor { name: "numSegments" argType: INPUT_TENSOR argIndex: 2 } argDescriptor { name: "numSegments" argType: INT64 } argDescriptor { name: "dtype" argType: DATA_TYPE } } opList { name: "unsorted_segment_sqrt_n" argDescriptor { name: "segmentedOutput" argType: OUTPUT_TENSOR } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "idxSegments" argType: INPUT_TENSOR argIndex: 1 } argDescriptor { name: "numSegments" 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: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "idxSegments" argType: INPUT_TENSOR argIndex: 1 } argDescriptor { name: "numSegments" argType: INPUT_TENSOR argIndex: 2 } argDescriptor { name: "numSegments" argType: INT64 } argDescriptor { name: "dtype" argType: DATA_TYPE } } opList { name: "unsorted_segment_sum" argDescriptor { name: "segmentedOutput" argType: OUTPUT_TENSOR } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "idxSegments" argType: INPUT_TENSOR argIndex: 1 } argDescriptor { name: "numSegments" argType: INPUT_TENSOR argIndex: 2 } argDescriptor { name: "numSegments" argType: INT64 } argDescriptor { name: "dtype" argType: DATA_TYPE } } opList { name: "unsorted_segment_sum_bp" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "idxSegments" argType: INPUT_TENSOR argIndex: 1 } argDescriptor { name: "numSegments" 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 } argDescriptor { name: "num" argType: INT64 argIndex: 1 } argDescriptor { name: "dtype" argType: DATA_TYPE } } opList { name: "unstack_list" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "outputList" argType: INPUT_TENSOR } opDeclarationType: LIST_OP_IMPL } 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 } argDescriptor { name: "factorW" argType: INT64 argIndex: 1 } argDescriptor { name: "isNCHW" argType: INT64 argIndex: 2 } argDescriptor { name: "dtype" argType: DATA_TYPE } } opList { name: "upsampling2d_bp" argDescriptor { name: "gradI" argType: OUTPUT_TENSOR } argDescriptor { name: "nchw" argType: BOOL } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "gradO" 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 } argDescriptor { name: "factorH" argType: INT64 argIndex: 1 } argDescriptor { name: "factorW" argType: INT64 argIndex: 2 } argDescriptor { name: "isNCDHW" argType: INT64 argIndex: 3 } argDescriptor { name: "dtype" argType: DATA_TYPE } } opList { name: "upsampling3d_bp" argDescriptor { name: "gradI" argType: OUTPUT_TENSOR } argDescriptor { name: "ncdhw" argType: BOOL } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "gradO" argType: INPUT_TENSOR argIndex: 1 } argDescriptor { name: "isNCDHW" argType: INT64 } argDescriptor { name: "dtype" argType: DATA_TYPE } } opList { name: "var" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "biasCorrected" argType: BOOL } argDescriptor { name: "keepDims" argType: BOOL argIndex: 1 } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "dimensions" argType: INT64 } opDeclarationType: LEGACY_XYZ } opList { name: "weighted_cross_entropy_with_logits" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "targets" argType: INPUT_TENSOR } argDescriptor { name: "input" argType: INPUT_TENSOR argIndex: 1 } argDescriptor { name: "weights" argType: INPUT_TENSOR argIndex: 2 } opDeclarationType: OP_IMPL } opList { name: "where_np" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "condition" argType: INPUT_TENSOR } argDescriptor { name: "input" argType: INPUT_TENSOR argIndex: 1 } argDescriptor { name: "y" argType: INPUT_TENSOR argIndex: 2 } argDescriptor { name: "dtype" argType: DATA_TYPE } } opList { name: "write_list" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "list" argType: INPUT_TENSOR } argDescriptor { name: "input" argType: INPUT_TENSOR argIndex: 1 } argDescriptor { name: "idx" argType: INT64 } opDeclarationType: LIST_OP_IMPL } opList { name: "xor" 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: "comparable" argType: DOUBLE } opDeclarationType: LEGACY_XYZ } opList { name: "xor_scalar" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "inPlace" argType: BOOL } argDescriptor { name: "input" argType: INPUT_TENSOR } opDeclarationType: LEGACY_XYZ } opList { name: "xw_plus_b" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "weights" argType: INPUT_TENSOR argIndex: 1 } argDescriptor { name: "b" argType: INPUT_TENSOR argIndex: 2 } argDescriptor { name: "bTranspose" argType: INT64 } argDescriptor { name: "dtype" argType: DATA_TYPE } } opList { name: "xw_plus_b_bp" argDescriptor { name: "dLdx" argType: OUTPUT_TENSOR } argDescriptor { name: "dLdw" argType: OUTPUT_TENSOR argIndex: 1 } argDescriptor { name: "dLdb" argType: OUTPUT_TENSOR argIndex: 2 } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "w" argType: INPUT_TENSOR argIndex: 1 } argDescriptor { name: "b" argType: INPUT_TENSOR argIndex: 2 } argDescriptor { name: "dLdz" argType: INPUT_TENSOR argIndex: 3 } argDescriptor { name: "bTranspose" argType: INT64 } argDescriptor { name: "dtype" argType: DATA_TYPE } } opList { name: "yiq_to_rgb" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "dimC" argType: INT64 } opDeclarationType: CONFIGURABLE_OP_IMPL } opList { name: "yuv_to_rgb" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "dimC" argType: INT64 } opDeclarationType: CONFIGURABLE_OP_IMPL } opList { name: "zero_fraction" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { 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 } } opList { name: "zeros_like" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "input" argType: INPUT_TENSOR } } opList { name: "zeroslike" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "inPlace" argType: BOOL } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "dataType" argType: INT64 } } opList { name: "zeta" argDescriptor { name: "outputs" argType: OUTPUT_TENSOR } argDescriptor { name: "inPlace" argType: BOOL } argDescriptor { name: "input" argType: INPUT_TENSOR } argDescriptor { name: "q" argType: INPUT_TENSOR argIndex: 1 } argDescriptor { name: "dataType" argType: DATA_TYPE } opDeclarationType: CONFIGURABLE_OP_IMPL } opList { name: "placeholder" opDeclarationType: LOGIC_OP_IMPL }