aboutsummaryrefslogtreecommitdiffhomepage
path: root/tensorflow/core/kernels/queue_base.cc
diff options
context:
space:
mode:
authorGravatar David G. Andersen <dga@google.com>2016-03-11 17:14:43 -0800
committerGravatar TensorFlower Gardener <gardener@tensorflow.org>2016-03-11 20:45:51 -0800
commit08332fc3e535c4c31c1969349b8b02fd6486cbe2 (patch)
tree9b13bed15e580980698d60e60927bc815b8ec5d9 /tensorflow/core/kernels/queue_base.cc
parent9849c30bb441d8bc07b5a5bb2efe0640ce46c97c (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.cc7
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)); \