onnx_diagnostic.reference.torch_evaluator¶
- onnx_diagnostic.reference.torch_evaluator.get_kernels() Dict[Tuple[str, str, int], type[OpRun]] [source][source]¶
Retrieves all the available kernels class
TorchOnnxEvaluator
can use. The full list is the following.<<<
from onnx_diagnostic.reference.torch_evaluator import get_kernels for k, v in sorted(get_kernels().items()): domain, name, version = k f = f"{name}({version})" if domain == "" else f"{name}[{domain}]({version})" add = " " * max(25 - len(f), 0) dd = " -- device dependent" if v.device_dependent() else "" print(f"{f}{add} -- {v.__name__}{dd}")
>>>
Abs(1) -- Abs_1 Add(1) -- Add_1 And(1) -- And_1 AveragePool(11) -- AveragePool_11 Cast(6) -- Cast_6 CastLike(15) -- CastLike_15 Concat(1) -- Concat_1 ConcatFromSequence(11) -- ConcatFromSequence_11 ConstantOfShape(9) -- ConstantOfShape_9 -- device dependent Conv(11) -- Conv_11 Cos(1) -- Cos_1 Div(1) -- Div_1 Equal(1) -- Equal_1 Erf(9) -- Erf_9 Exp(1) -- Exp_1 Expand(8) -- Expand_8 Gather(1) -- Gather_1 Greater(1) -- Greater_1 GreaterOrEqual(1) -- GreaterOrEqual_1 Identity(1) -- Identity_1 If(1) -- If_1 LayerNormalization(17) -- LayerNormalization_17 Less(1) -- Less_1 LessOrEqual(1) -- LessOrEqual_1 Log(1) -- Log_1 Loop(16) -- Loop_16 MatMul(1) -- MatMul_1 Mul(1) -- Mul_1 Neg(1) -- Neg_1 NonZero(13) -- NonZero_13 Not(1) -- Not_1 Or(1) -- Or_1 Pow(12) -- Pow_12 Range(11) -- Range_11 -- device dependent Reciprocal(1) -- Reciprocal_1 ReduceMax(18) -- ReduceMax_18 ReduceMean(18) -- ReduceMean_18 ReduceMin(17) -- ReduceMin_17 ReduceMin(18) -- ReduceMin_18 ReduceSum(13) -- ReduceSum_13 Reshape(14) -- Reshape_14 ScatterND(16) -- ScatterND_16 SequenceEmpty(11) -- SequenceEmpty_11 SequenceInsert(11) -- SequenceInsert_11 Shape(15) -- Shape_15 Sigmoid(6) -- Sigmoid_6 Sin(1) -- Sin_1 Slice(13) -- Slice_13 Softmax(13) -- Softmax_13 Split(18) -- Split_18 Sqrt(1) -- Sqrt_1 Squeeze(13) -- Squeeze_13 Sub(1) -- Sub_1 Tanh(6) -- Tanh_6 Tile(6) -- Tile_6 Transpose(1) -- Transpose_1 Trilu(14) -- Trilu_14 Unsqueeze(13) -- Unsqueeze_13 Where(9) -- Where_9