Tuning of FullyConnected for Intel.

PiperOrigin-RevId: 331808322 Change-Id: I950680fc58523112490ec2b9450f65f4731ba72f
2020-09-15 11:03:03 -07:00 · 2020-09-15 11:03:03 -07:00 · 876386c45d
commit 876386c45d
parent 3cbb507689
1 changed files with 2 additions and 0 deletions
--- a/tensorflow/lite/delegates/gpu/cl/kernels/fully_connected.cc
+++ b/tensorflow/lite/delegates/gpu/cl/kernels/fully_connected.cc
@ -35,6 +35,8 @@ FullyConnected::FullyConnected(const OperationDef& definition,
    } else {
      work_group_size_ = int3(32, 4, 1);
    }
+  } else if (device_info.IsIntel()) {
+    work_group_size_ = int3(8, 4, 1);
  } else {
    work_group_size_ = int3(16, 4, 1);
  }