IntenseWebs/openvino
3
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity
Files
e14d1b2a070426ced0525f5a63a87f3a955d79a9
openvino/tests/layer_tests/pytorch_tests
History
Surya Siddharth Pemmaraju e14d1b2a07 Torchscript backend (#17132) 
* Added torch script backend

* Added ts_backend to pytorch layer tests

* Added use_ts_backend fixture to the test suite to activate the
torchscript backend

* Fixed failing test_dict layer test

* Added USE_TS_BACKEND as an env variable

* Removed use_ts_backend fixture

* Added more tests for ts backend

* Added more information in the comments about usage

* Removed convolution3d test from precommit_ts_backend

* Added some torchscript backend tests to ci

* Removed tests from CI as torch.compile doesn't support 3.11 currently

* Fixed linter issues

* Addressed PR comments and linter issues
2023-05-17 14:58:38 +02:00
..
conftest.py
Add PyTorch Frontend (#15069)
2023-01-18 18:16:57 +04:00
pytorch_layer_test_class.py
Torchscript backend (#17132)
2023-05-17 14:58:38 +02:00
test_adaptive_avg_pool3d.py
Torchscript backend (#17132)
2023-05-17 14:58:38 +02:00
test_adaptive_max_pool_2d.py
Torchscript backend (#17132)
2023-05-17 14:58:38 +02:00
test_add.py
Torchscript backend (#17132)
2023-05-17 14:58:38 +02:00
test_addcmul.py
[PT FE]: improve integration into mo.convert_model (#16243)
2023-03-24 16:55:07 +01:00
test_addmm.py
Fix sporadic fails when beta==0 in baddmm (#16610)
2023-03-28 18:47:35 +00:00
test_all.py
[PT FE] Add aten::all implementation (#16901)
2023-05-02 21:12:12 +02:00
test_and.py
[PT FE] Add aten::__and__ operator with layer test (#15365)
2023-01-28 10:06:06 +03:00
test_arange.py
[PT FE]: fix arange dtype if it is not provided (#16084)
2023-03-04 10:17:09 +01:00
test_argsort.py
[PT FE] Enable stable tests for sort & argsort (#16415)
2023-04-25 01:21:16 +04:00
test_batch_norm.py
Torchscript backend (#17132)
2023-05-17 14:58:38 +02:00
test_bitwise_not.py
[PT FE]: enable dtype in softmax and constant filling, extend logical ops support (#16276)
2023-03-20 14:12:12 +00:00
test_bool.py
Add support for concatenation in Loop (#15899)
2023-02-28 21:31:33 +01:00
test_cat.py
Support more complicated cases of list concatenation (#16139)
2023-03-10 07:51:10 +01:00
test_ceil.py
Add PyTorch Frontend (#15069)
2023-01-18 18:16:57 +04:00
test_chunk.py
[PT FE] Fix aten::chunk for dynamic shapes (#16902)
2023-05-01 09:32:10 +00:00
test_clamp.py
Add PyTorch Frontend (#15069)
2023-01-18 18:16:57 +04:00
test_clone.py
Add PyTorch Frontend (#15069)
2023-01-18 18:16:57 +04:00
test_comparision.py
Add eltwise types resolving. Support big int constants. (#15415)
2023-02-02 02:15:33 +01:00
test_conv_transposend.py
[PT FE]: support conv transpose (#15191)
2023-01-22 07:03:54 +00:00
test_convnd.py
[PT FE] Fix GroupConvolution weights reshape (#15560)
2023-02-08 14:11:35 +01:00
test_convolution_mode.py
Add PyTorch Frontend (#15069)
2023-01-18 18:16:57 +04:00
test_convolution.py
Torchscript backend (#17132)
2023-05-17 14:58:38 +02:00
test_cumsum.py
[PT FE]: add translation for aten::cumsum (#16092)
2023-03-08 11:07:29 +01:00
test_deformable_convolution.py
[PT FE] Add torchvision::deform_conv2d translation (#16450)
2023-03-27 11:13:32 +02:00
test_device.py
[PT FE] Add prim::device and prim::type, improvements to aten::to, aten::eq, aten::ne (#15881)
2023-03-01 09:46:55 +01:00
test_dict.py
[PT FE] Support prim::DictConstruct on the output (#16894)
2023-04-13 16:42:17 +00:00
test_div.py
Add eltwise types resolving. Support big int constants. (#15415)
2023-02-02 02:15:33 +01:00
test_einsum.py
[PT FE]: support aten::einsum (#15844)
2023-02-23 11:39:28 +04:00
test_embedding_bag.py
[Pt FE]: aten::embedding_bag (#17098)
2023-05-10 09:44:08 +02:00
test_embedding.py
[Pt FE]: aten::embedding_bag (#17098)
2023-05-10 09:44:08 +02:00
test_empty.py
[PT FE] Add aten::new_empty (#16312)
2023-03-20 10:03:33 +01:00
test_eq.py
add tests for aten::eq (#15222)
2023-01-21 21:07:47 +03:00
test_exp.py
Add PyTorch Frontend (#15069)
2023-01-18 18:16:57 +04:00
test_expand.py
[PT FE]: add more list construct transformations (#15614)
2023-02-15 16:39:27 +00:00
test_eye.py
[PT FE]: supports eye, tril, triu, tensor.fill_ and implicit scalars (#15187)
2023-01-25 21:04:56 +03:00
test_flatten.py
[PT FE] Fix aten::flatten, add more tests (#15576)
2023-02-10 14:23:27 +01:00
test_floor_divide.py
[PT FE]: support mixed precision in floor divide (#16362)
2023-03-20 11:00:26 +01:00
test_floor.py
Add PyTorch Frontend (#15069)
2023-01-18 18:16:57 +04:00
test_full.py
[PT FE]: supports eye, tril, triu, tensor.fill_ and implicit scalars (#15187)
2023-01-25 21:04:56 +03:00
test_gather.py
[PT FE]: aten::gather (#16784)
2023-04-11 14:28:05 +04:00
test_gelu.py
[PT FE]: support aten::gelu with tanh approximate and tests (#15715)
2023-02-15 15:11:37 +01:00
test_getitem.py
Support more complicated cases of list concatenation (#16139)
2023-03-10 07:51:10 +01:00
test_glu.py
[PT FE]: support aten::glu and aten::sigmoid_ (#15185)
2023-01-21 21:05:20 +03:00
test_grid_sampler.py
[PT FE]: support grid sampler (#15243)
2023-01-31 14:04:37 +04:00
test_group_norm.py
Add PyTorch Frontend (#15069)
2023-01-18 18:16:57 +04:00
test_if.py
Improve detectron2 support (#16011)
2023-03-23 22:30:03 +00:00
test_im2col.py
Add PyTorch Frontend (#15069)
2023-01-18 18:16:57 +04:00
test_index_put_.py
[PT FE] Fix failing translation of aten::index_put_ (#16140)
2023-03-09 20:14:58 +00:00
test_index_select.py
[PT FE] fix invalid reshape shape after aten::index (#16821)
2023-04-11 12:41:59 +03:00
test_index.py
[PT FE] fix invalid reshape shape after aten::index (#16821)
2023-04-11 12:41:59 +03:00
test_instance_norm.py
[PT FE]: improve integration into mo.convert_model (#16243)
2023-03-24 16:55:07 +01:00
test_leaky_relu.py
Add PyTorch Frontend (#15069)
2023-01-18 18:16:57 +04:00
test_len.py
[PT FE] Fix aten::len for empty lists (#15820)
2023-02-20 21:34:04 +00:00
test_linear.py
[PT FE] Fix aten::flatten, add more tests (#15576)
2023-02-10 14:23:27 +01:00
test_listunpack.py
[PT FE]: improve integration into mo.convert_model (#16243)
2023-03-24 16:55:07 +01:00
test_log.py
Remove None at outputs of the model, improve types handling in frontend (#15258)
2023-01-25 13:28:47 +03:00
test_masked_fill.py
[PT FE]: update masked_fill support for multiple precisions (#15722)
2023-02-15 15:32:06 +04:00
test_mean.py
[PT FE]: fix aten::mean behaviour for provided dtype (#16790)
2023-04-11 14:29:29 +04:00
test_min_max.py
[PT FE]: improve integration into mo.convert_model (#16243)
2023-03-24 16:55:07 +01:00
test_mm.py
Add PyTorch Frontend (#15069)
2023-01-18 18:16:57 +04:00
test_mul.py
Add eltwise types resolving. Support big int constants. (#15415)
2023-02-02 02:15:33 +01:00
test_narrow.py
Add aten::narrow operator with layer test (#15788)
2023-02-20 15:47:25 +01:00
test_nms.py
Add PyTorch Frontend (#15069)
2023-01-18 18:16:57 +04:00
test_nonzero.py
Add PyTorch Frontend (#15069)
2023-01-18 18:16:57 +04:00
test_norm.py
[PT FE]: support aten::linalng_vector_norm (#16109)
2023-03-24 21:00:17 +01:00
test_numel.py
Add PyTorch Frontend (#15069)
2023-01-18 18:16:57 +04:00
test_pad.py
[PT FE]: fix padding vaqlue dtype (#16087)
2023-03-04 11:08:56 +00:00
test_permute.py
[PT FE]: add more list construct transformations (#15614)
2023-02-15 16:39:27 +00:00
test_pooling.py
[PT FE] Support default strides for avg and max pooling (#17337)
2023-05-04 09:27:10 +02:00
test_pow.py
Fix issue with Pow when both inputs are scalars (#17305)
2023-05-02 16:49:42 +02:00
test_pythonop.py
[PT FE] Add prim::PythonOp (#15714)
2023-03-07 17:20:13 +01:00
test_relu.py
Torchscript backend (#17132)
2023-05-17 14:58:38 +02:00
test_remainder.py
[PT FE]: support aten::var_mean and aten::remainder (#15193)
2023-01-26 14:30:42 +00:00
test_repeat_interleave.py
[PT FE]: improve integration into mo.convert_model (#16243)
2023-03-24 16:55:07 +01:00
test_repeat.py
[PT FE]: add more list construct transformations (#15614)
2023-02-15 16:39:27 +00:00
test_reshape_as.py
Add PyTorch Frontend (#15069)
2023-01-18 18:16:57 +04:00
test_reshape.py
Add PyTorch Frontend (#15069)
2023-01-18 18:16:57 +04:00
test_roi_align.py
[PT FE] Add torchvision::roi_align operator with layer test (#15821)
2023-02-23 09:26:17 +01:00
test_roll.py
Add PyTorch Frontend (#15069)
2023-01-18 18:16:57 +04:00
test_rsqrt.py
Add PyTorch Frontend (#15069)
2023-01-18 18:16:57 +04:00
test_rsub.py
[PT FE] Support set/get element type, shape and value in PyTorch FE InputModel (#16100)
2023-03-07 15:45:29 +01:00
test_scaled_dot_product_attention.py
[PT FE]: implement scaled dot product attention (#17178)
2023-04-26 12:51:02 +04:00
test_select.py
Add PyTorch Frontend (#15069)
2023-01-18 18:16:57 +04:00
test_selu.py
Add PyTorch Frontend (#15069)
2023-01-18 18:16:57 +04:00
test_set_item.py
Add support for aten::_set_item (#15643)
2023-02-14 14:11:51 +01:00
test_sigmoid.py
[PT FE]: support aten::glu and aten::sigmoid_ (#15185)
2023-01-21 21:05:20 +03:00
test_sign.py
[PT FE]: aten::gather (#16784)
2023-04-11 14:28:05 +04:00
test_silu.py
Add PyTorch Frontend (#15069)
2023-01-18 18:16:57 +04:00
test_size.py
[PT FE]: support prim::min, prim::ListConstruct as output and fix aten::getitem for prim::ListConstruct (#15470)
2023-02-02 22:36:07 +00:00
test_softmax.py
[PT FE]: enable dtype in softmax and constant filling, extend logical ops support (#16276)
2023-03-20 14:12:12 +00:00
test_sort.py
[PT FE] Enable stable sort layer tests (#17229)
2023-04-26 18:24:38 +02:00
test_split.py
Improve detectron2 support (#16011)
2023-03-23 22:30:03 +00:00
test_sqrt.py
Add PyTorch Frontend (#15069)
2023-01-18 18:16:57 +04:00
test_squeeze.py
Add PyTorch Frontend (#15069)
2023-01-18 18:16:57 +04:00
test_stack.py
Add aten stack transformation (#15311)
2023-02-02 20:15:28 +01:00
test_strided_const.py
Add PyTorch Frontend (#15069)
2023-01-18 18:16:57 +04:00
test_sub.py
Add eltwise types resolving. Support big int constants. (#15415)
2023-02-02 02:15:33 +01:00
test_sum.py
Try to fix sporadic sum layer test fail (#15304)
2023-01-26 16:08:30 +04:00
test_to.py
[PyOV] Remove deprecated (#17512)
2023-05-16 21:48:39 +02:00
test_topk.py
add aten::topk (#15221)
2023-01-27 09:34:55 +01:00
test_transpose.py
[PT FE] Fix aten::flatten, add more tests (#15576)
2023-02-10 14:23:27 +01:00
test_trigonometry.py
Add PyTorch Frontend (#15069)
2023-01-18 18:16:57 +04:00
test_trilu.py
[PT FE]: supports eye, tril, triu, tensor.fill_ and implicit scalars (#15187)
2023-01-25 21:04:56 +03:00
test_tuple_construct.py
[PT FE]: support prim::min, prim::ListConstruct as output and fix aten::getitem for prim::ListConstruct (#15470)
2023-02-02 22:36:07 +00:00
test_type_as.py
Add PyTorch Frontend (#15069)
2023-01-18 18:16:57 +04:00
test_unbind.py
Add PyTorch Frontend (#15069)
2023-01-18 18:16:57 +04:00
test_unfold.py
[PT FE] Add aten::unfold operator with layer test (#15316)
2023-01-31 18:29:32 +01:00
test_unsqueeze.py
Add PyTorch Frontend (#15069)
2023-01-18 18:16:57 +04:00
test_upsample.py
Support sizes input as list to upsample ops (#17350)
2023-05-07 19:23:51 +02:00
test_var_mean.py
[PT FE]: support aten::var_mean and aten::remainder (#15193)
2023-01-26 14:30:42 +00:00
test_view.py
[PT FE]: Add aten::view transformations (#15339)
2023-02-01 12:14:17 +01:00
test_where.py
Add PyTorch Frontend (#15069)
2023-01-18 18:16:57 +04:00