diff options
author | 2016-03-11 17:14:43 -0800 | |
---|---|---|
committer | 2016-03-11 20:45:51 -0800 | |
commit | 08332fc3e535c4c31c1969349b8b02fd6486cbe2 (patch) | |
tree | 9b13bed15e580980698d60e60927bc815b8ec5d9 /tensorflow/core/kernels/queue_base.cc | |
parent | 9849c30bb441d8bc07b5a5bb2efe0640ce46c97c (diff) |
Cleaning up int64 implicit cast warnings from fifo queue.
Change: 117023195
Diffstat (limited to 'tensorflow/core/kernels/queue_base.cc')
-rw-r--r-- | tensorflow/core/kernels/queue_base.cc | 7 |
1 files changed, 4 insertions, 3 deletions
diff --git a/tensorflow/core/kernels/queue_base.cc b/tensorflow/core/kernels/queue_base.cc index 05fccc6d2c..6886165f98 100644 --- a/tensorflow/core/kernels/queue_base.cc +++ b/tensorflow/core/kernels/queue_base.cc @@ -26,7 +26,8 @@ namespace tensorflow { namespace { template <DataType DT> -Status HandleSliceToElement(const Tensor& parent, Tensor* element, int index) { +Status HandleSliceToElement(const Tensor& parent, Tensor* element, + int64 index) { typedef typename EnumToDataType<DT>::Type T; DCHECK_NE(parent.dim_size(0), 0); if (element->NumElements() != (parent.NumElements() / parent.dim_size(0))) { @@ -349,7 +350,7 @@ void QueueBase::FlushUnlocked() { } Status QueueBase::CopySliceToElement(const Tensor& parent, Tensor* element, - int index) { + int64 index) { #define HANDLE_TYPE(DT) \ if (parent.dtype() == DT) { \ TF_RETURN_IF_ERROR(HandleSliceToElement<DT>(parent, element, index)); \ @@ -377,7 +378,7 @@ Status QueueBase::CopySliceToElement(const Tensor& parent, Tensor* element, // Static method Status QueueBase::CopyElementToSlice(const Tensor& element, Tensor* parent, - int index) { + int64 index) { #define HANDLE_TYPE(DT) \ if (element.dtype() == DT) { \ TF_RETURN_IF_ERROR(HandleElementToSlice<DT>(element, parent, index)); \ |