Searched refs:test_dynamic_dim (Results 1 – 1 of 1) sorted by relevance
| /external/pytorch/test/dynamo/ |
| D | test_subclasses.py | 868 def test_dynamic_dim(f, x, dim_dynamic, exp_frame_count, exp_op_count): function
899 test_dynamic_dim(f, x, DimDynamic.DYNAMIC, 1, 1)
900 test_dynamic_dim(f, x, DimDynamic.DUCK, 1, 1)
901 test_dynamic_dim(f, x, DimDynamic.STATIC, 1, 1)
|