diff options
| author |  Anton Dmitriev <dmitrievanthony@gmail.com> | 2018-09-24 12:14:45 +0300 |
|---|---|---|
| committer | Anton Dmitriev <dmitrievanthony@gmail.com> | 2018-09-24 12:14:45 +0300 |
| commit | 90c68770467701a23d23a85c5d769f6f4fa39f0f (patch) | |
| tree | 796bb6d2c2415f94f1c15eff2eb9ee933b007653 /tensorflow/contrib/ignite | |
| parent | 8f4ded5884684f40b4912d95c717b185340996b8 (diff) | |
Fix byte-order issue.
Diffstat (limited to 'tensorflow/contrib/ignite')
| -rw-r--r-- | tensorflow/contrib/ignite/kernels/ignite_byte_swapper.h | 5 |
1 files changed, 2 insertions, 3 deletions
diff --git a/tensorflow/contrib/ignite/kernels/ignite_byte_swapper.h b/tensorflow/contrib/ignite/kernels/ignite_byte_swapper.h index 6753c67701..46df3e39dc 100644 --- a/tensorflow/contrib/ignite/kernels/ignite_byte_swapper.h +++ b/tensorflow/contrib/ignite/kernels/ignite_byte_swapper.h @@ -17,14 +17,13 @@ limitations under the License. #define TENSORFLOW_CONTRIB_IGNITE_KERNELS_IGNITE_BYTE_SWAPPER_H_ #include <stdint.h> +#include "tensorflow/core/platform/byte_order.h" namespace tensorflow { -constexpr bool kLittleEndian = __BYTE_ORDER__ == __ORDER_LITTLE_ENDIAN__; - class ByteSwapper { public: - ByteSwapper(bool big_endian) { swap_ = big_endian == kLittleEndian; } + ByteSwapper(bool big_endian) { swap_ = big_endian == port::kLittleEndian; } inline void SwapIfRequiredInt16(int16_t *x) const { if (swap_) { |