mirror of https://github.com/llvm/torch-mlir
[torchdynamo] Add default decomposition for ops in the dynamo backend
Signed-Off By: Vivek Khandelwal <vivek@nod-labs.com>pull/1830/head snapshot-20230123.727
parent
bf4c77cbc5
commit
23aa6903f7
|
@ -52,6 +52,15 @@ def _get_decomposition_table():
|
|||
aten.native_group_norm,
|
||||
aten.split.Tensor,
|
||||
aten.split_with_sizes,
|
||||
aten.norm.ScalarOpt_dim,
|
||||
aten.embedding_dense_backward,
|
||||
aten.native_layer_norm_backward,
|
||||
aten.slice_backward,
|
||||
aten.select_backward,
|
||||
aten.upsample_bilinear2d.vec,
|
||||
aten.mse_loss_backward,
|
||||
aten.native_group_norm_backward,
|
||||
aten.sigmoid_backward,
|
||||
])
|
||||
|
||||
|
||||
|
|
Loading…
Reference in New Issue