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

Reland "Fix the TFMobile op stripping build target"

Browse Source 
PiperOrigin-RevId: 247041021
This commit is contained in:
Jared Duke 2019-05-07 10:08:50 -07:00 committed by TensorFlower Gardener
parent bf1d138211
commit 94187bc3f6
4 changed files with 10 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
tensorflow/core/kernels/BUILD
View File

@ -6149,6 +6149,7 @@ filegroup(
) )
ANDROID_TEXTUAL_HDRS = [ ANDROID_TEXTUAL_HDRS = [
"eigen_spatial_convolutions-inl.h",
"gather_nd_op_cpu_impl.h", "gather_nd_op_cpu_impl.h",
"gemm_functors.h", "gemm_functors.h",
"mirror_pad_op_cpu_impl.h", "mirror_pad_op_cpu_impl.h",

4
tensorflow/core/kernels/redux_functor.h
View File

@ -16,7 +16,11 @@ limitations under the License.
#ifndef TENSORFLOW_CORE_KERNELS_REDUX_FUNCTOR_H_ #ifndef TENSORFLOW_CORE_KERNELS_REDUX_FUNCTOR_H_
#define TENSORFLOW_CORE_KERNELS_REDUX_FUNCTOR_H_ #define TENSORFLOW_CORE_KERNELS_REDUX_FUNCTOR_H_
#define EIGEN_USE_THREADS
#include "third_party/eigen3/Eigen/Core" #include "third_party/eigen3/Eigen/Core"
#include "third_party/eigen3/unsupported/Eigen/CXX11/Tensor"
#include "tensorflow/core/framework/tensor.h"
#include "tensorflow/core/framework/tensor_types.h" #include "tensorflow/core/framework/tensor_types.h"
#include "tensorflow/core/platform/types.h" #include "tensorflow/core/platform/types.h"

1
tensorflow/examples/android/jni/object_tracking/jni_utils.h
View File

@ -16,6 +16,7 @@ limitations under the License.
#ifndef TENSORFLOW_EXAMPLES_ANDROID_JNI_OBJECT_TRACKING_JNI_UTILS_H_ #ifndef TENSORFLOW_EXAMPLES_ANDROID_JNI_OBJECT_TRACKING_JNI_UTILS_H_
#define TENSORFLOW_EXAMPLES_ANDROID_JNI_OBJECT_TRACKING_JNI_UTILS_H_ #define TENSORFLOW_EXAMPLES_ANDROID_JNI_OBJECT_TRACKING_JNI_UTILS_H_
#include <jni.h>
#include <stdint.h> #include <stdint.h>
#include "tensorflow/examples/android/jni/object_tracking/utils.h" #include "tensorflow/examples/android/jni/object_tracking/utils.h"

8
tensorflow/examples/android/jni/object_tracking/sprite.h
View File

@ -16,16 +16,14 @@ limitations under the License.
#ifndef TENSORFLOW_EXAMPLES_ANDROID_JNI_OBJECT_TRACKING_SPRITE_H_ #ifndef TENSORFLOW_EXAMPLES_ANDROID_JNI_OBJECT_TRACKING_SPRITE_H_
#define TENSORFLOW_EXAMPLES_ANDROID_JNI_OBJECT_TRACKING_SPRITE_H_ #define TENSORFLOW_EXAMPLES_ANDROID_JNI_OBJECT_TRACKING_SPRITE_H_
#ifdef __RENDER_OPENGL__
#include <GLES/gl.h> #include <GLES/gl.h>
#include <GLES/glext.h> #include <GLES/glext.h>
#include "tensorflow/examples/android/jni/object_tracking/image-inl.h" #include "tensorflow/examples/android/jni/object_tracking/image-inl.h"
#include "tensorflow/examples/android/jni/object_tracking/image.h" #include "tensorflow/examples/android/jni/object_tracking/image.h"
#ifndef __RENDER_OPENGL__
#error sprite.h should not included if OpenGL is not enabled by platform.h
#endif
namespace tf_tracking { namespace tf_tracking {
// This class encapsulates the logic necessary to load an render image data // This class encapsulates the logic necessary to load an render image data
@ -199,4 +197,6 @@ class Sprite {
} // namespace tf_tracking } // namespace tf_tracking
#endif // __RENDER_OPENGL__
#endif // TENSORFLOW_EXAMPLES_ANDROID_JNI_OBJECT_TRACKING_SPRITE_H_ #endif // TENSORFLOW_EXAMPLES_ANDROID_JNI_OBJECT_TRACKING_SPRITE_H_