aboutsummaryrefslogtreecommitdiffhomepage
path: root/tensorflow/core/platform/platform.h
diff options
context:
space:
mode:
Diffstat (limited to 'tensorflow/core/platform/platform.h')
-rw-r--r--tensorflow/core/platform/platform.h5
1 files changed, 3 insertions, 2 deletions
diff --git a/tensorflow/core/platform/platform.h b/tensorflow/core/platform/platform.h
index 7c44b22eff..12120c4ab9 100644
--- a/tensorflow/core/platform/platform.h
+++ b/tensorflow/core/platform/platform.h
@@ -43,9 +43,10 @@ limitations under the License.
#elif defined(__arm__)
#define PLATFORM_POSIX
-// Since there's no macro for the Raspberry Pi, assume we're on a mobile
-// platform if we're compiling for the ARM CPU.
+// Require an outside macro to tell us if we're building for Raspberry Pi.
+#if !defined(RASPBERRY_PI)
#define IS_MOBILE_PLATFORM
+#endif // !defined(RASPBERRY_PI)
#else
// If no platform specified, use: