Skip to content

Commit

Permalink
wip covering tflite (270 ok/3690 ign)
Browse files Browse the repository at this point in the history
  • Loading branch information
kali committed Jul 31, 2023
1 parent 15562eb commit 2cbd7c8
Showing 1 changed file with 10 additions and 2 deletions.
12 changes: 10 additions & 2 deletions test-rt/test-tflite/suite.rs
Original file line number Diff line number Diff line change
Expand Up @@ -14,15 +14,23 @@ pub fn suite() -> infra::TestSuite {

fn ignore_onnx(t: &[String]) -> bool {
let name = t.last().unwrap();
!name.contains("_conv_") || name == "test_conv_with_strides_and_asymmetric_padding"
let included = "_conv_ Conv1d Conv2d squeeze _transpose_ test_reshape test_flatten";
let excluded = "
test_Conv1d_groups
test_Conv2d_groups
test_Conv1d_depthwise_with_multiplier
test_Conv2d_depthwise_with_multiplier
test_Conv2d_groups_thnn
test_reshape_allowzero_reordered";
!included.split_whitespace().any(|s| name.contains(s))
|| excluded.split_whitespace().any(|s| s == name)
}

fn ignore_conv(t: &[String]) -> bool {
let unit: &str = t.last().map(|s| &**s).unwrap();
t[0] == "q"
|| unit == "proptest"
// grouping and depthwise
|| unit == "depthwise_0"
|| unit.starts_with("group")
// conv 3D
|| unit == "lazy_im2col_big"
Expand Down

0 comments on commit 2cbd7c8

Please sign in to comment.