diff options
author | A. Unique TensorFlower <gardener@tensorflow.org> | 2018-05-30 10:40:39 -0700 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2018-05-30 10:45:09 -0700 |
commit | 81755953863f36f13d1c70a108469b0c3f5fa697 (patch) | |
tree | 208c244cd48aaea158b09464598bb348fe9d71f2 /tensorflow/compiler/xla/service/cpu/shape_partition.cc | |
parent | 2bb9fe8d202b2400219d45a8a2185a02dd070fb5 (diff) |
Internal change
PiperOrigin-RevId: 198582954
Diffstat (limited to 'tensorflow/compiler/xla/service/cpu/shape_partition.cc')
-rw-r--r-- | tensorflow/compiler/xla/service/cpu/shape_partition.cc | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/tensorflow/compiler/xla/service/cpu/shape_partition.cc b/tensorflow/compiler/xla/service/cpu/shape_partition.cc index 42fe955f19..d12c539614 100644 --- a/tensorflow/compiler/xla/service/cpu/shape_partition.cc +++ b/tensorflow/compiler/xla/service/cpu/shape_partition.cc @@ -115,7 +115,7 @@ ShapePartitionIterator::ShapePartitionIterator( for (int i = 0; i < dimension_partition_sizes_.size(); ++i) { const int64 dim_size = shape_.dimensions(dimensions_[i]); dimension_partition_sizes_[i] = - std::max(1LL, dim_size / dimension_partition_counts_[i]); + std::max(int64{1}, dim_size / dimension_partition_counts_[i]); } // Calculate the partition strides for each dimension. |