experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

TFLite GPU Metal: Internal change to fix TFLite GPU Metal backend compilation.

Browse Source 
PiperOrigin-RevId: 242223007
This commit is contained in:
Juhyun Lee 2019-04-05 17:22:08 -07:00 committed by TensorFlower Gardener
parent 22b32624a2
commit 088e9ef4b6
2 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
tensorflow/lite/delegates/gpu/metal/inference_context.mm
View File

@ -18,7 +18,7 @@ limitations under the License.
#include <map> #include <map>
#include <vector> #include <vector>
#include "third_party/absl/strings/substitute.h" #include "absl/strings/substitute.h"
#include "tensorflow/lite/delegates/gpu/common/memory_management.h" #include "tensorflow/lite/delegates/gpu/common/memory_management.h"
#include "tensorflow/lite/delegates/gpu/common/model.h" #include "tensorflow/lite/delegates/gpu/common/model.h"
#include "tensorflow/lite/delegates/gpu/common/shape.h" #include "tensorflow/lite/delegates/gpu/common/shape.h"

2
tensorflow/lite/delegates/gpu/metal_delegate.mm
View File

@ -23,7 +23,7 @@ limitations under the License.
#include <string> #include <string>
#include <vector> #include <vector>
#include "third_party/absl/types/span.h" #include "absl/types/span.h"
#include "tensorflow/lite/builtin_ops.h" #include "tensorflow/lite/builtin_ops.h"
#include "tensorflow/lite/c/c_api_internal.h" #include "tensorflow/lite/c/c_api_internal.h"
#include "tensorflow/lite/delegates/gpu/common/convert.h" #include "tensorflow/lite/delegates/gpu/common/convert.h"