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

KernelInfo moved to gpu/common.

Browse Source 
PiperOrigin-RevId: 342928157
Change-Id: I1654ab624e22e57e206c6de595a7d3c1fdd710d9
This commit is contained in:
Raman Sarokin 2020-11-17 13:03:00 -08:00 committed by TensorFlower Gardener
parent aec80b2420
commit 0c22c09ed7
6 changed files with 39 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
tensorflow/lite/delegates/gpu/cl/BUILD
View File

@ -218,6 +218,7 @@ cc_library(
":cl_program",
":opencl_wrapper",
":util",
"//tensorflow/lite/delegates/gpu/common:kernel_info",
"//tensorflow/lite/delegates/gpu/common:status",
"@com_google_absl//absl/strings",
],

6
tensorflow/lite/delegates/gpu/cl/cl_kernel.h
View File

@ -22,17 +22,13 @@ limitations under the License.
#include "tensorflow/lite/delegates/gpu/cl/cl_device.h"
#include "tensorflow/lite/delegates/gpu/cl/cl_program.h"
#include "tensorflow/lite/delegates/gpu/cl/opencl_wrapper.h"
#include "tensorflow/lite/delegates/gpu/common/kernel_info.h"
#include "tensorflow/lite/delegates/gpu/common/status.h"
namespace tflite {
namespace gpu {
namespace cl {
struct KernelInfo {
int private_memory_size;
int max_work_group_size;
};
// Arguments binding to CLKernel can be manual or automatic
// In manual you specify binding index explicitly
// In automatic binding, index auto-incremented with every binding call

1
tensorflow/lite/delegates/gpu/cl/kernels/BUILD
View File

@ -1299,6 +1299,7 @@ cc_library(
deps = [
"//tensorflow/lite/delegates/gpu/cl:cl_kernel",
"//tensorflow/lite/delegates/gpu/cl:device_info",
"//tensorflow/lite/delegates/gpu/common:kernel_info",
"//tensorflow/lite/delegates/gpu/common:types",
"//tensorflow/lite/delegates/gpu/common:util",
"//tensorflow/lite/delegates/gpu/common:workgroup_selection",

2
tensorflow/lite/delegates/gpu/cl/kernels/work_group_picking.h
View File

@ -18,8 +18,8 @@ limitations under the License.
#include <vector>
#include "tensorflow/lite/delegates/gpu/cl/cl_kernel.h"
#include "tensorflow/lite/delegates/gpu/cl/device_info.h"
#include "tensorflow/lite/delegates/gpu/common/kernel_info.h"
#include "tensorflow/lite/delegates/gpu/common/task/tuning_type.h"
#include "tensorflow/lite/delegates/gpu/common/types.h"
#include "tensorflow/lite/delegates/gpu/common/workgroup_selection.h"

5
tensorflow/lite/delegates/gpu/common/BUILD
View File

@ -44,6 +44,11 @@ cc_library(
],
)
cc_library(
name = "kernel_info",
hdrs = ["kernel_info.h"],
)
cc_library(
name = "data_type",
srcs = ["data_type.cc"],

30
tensorflow/lite/delegates/gpu/common/kernel_info.h Normal file
View File

@ -0,0 +1,30 @@
/* Copyright 2020 The TensorFlow Authors. All Rights Reserved.
Licensed under the Apache License, Version 2.0 (the "License");
you may not use this file except in compliance with the License.
You may obtain a copy of the License at
http://www.apache.org/licenses/LICENSE-2.0
Unless required by applicable law or agreed to in writing, software
distributed under the License is distributed on an "AS IS" BASIS,
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
See the License for the specific language governing permissions and
limitations under the License.
==============================================================================*/
#ifndef TENSORFLOW_LITE_DELEGATES_GPU_COMMON_KERNEL_INFO_H_
#define TENSORFLOW_LITE_DELEGATES_GPU_COMMON_KERNEL_INFO_H_
namespace tflite {
namespace gpu {
struct KernelInfo {
int private_memory_size;
int max_work_group_size;
};
} // namespace gpu
} // namespace tflite
#endif // TENSORFLOW_LITE_DELEGATES_GPU_COMMON_KERNEL_INFO_H_