changed long type to fixed integer
@@ -72,7 +72,7 @@ using tensorflow::Tensor;
using tensorflow::Status;
using tensorflow::string;
using tensorflow::int32;
-
+using tensorflow::int64;
void x_clock_start(int timer) {
#ifdef XOPENME
@@ -327,8 +327,9 @@ Status GetTopLabels(const std::vector<Tensor>& outputs, int how_many_labels,
return left.second > right.second;
});
scores.resize(how_many_labels);
- Tensor sorted_indices(tensorflow::DT_INT32, {(long long)scores.size()});
- Tensor sorted_scores(tensorflow::DT_FLOAT, {(long long)scores.size()});
+ int64 scores_size = scores.size();
+ Tensor sorted_indices(tensorflow::DT_INT32, {scores_size});
+ Tensor sorted_scores(tensorflow::DT_FLOAT, {scores_size});
for (int i = 0; i < scores.size(); ++i) {
sorted_indices.flat<int>()(i) = scores[i].first;
sorted_scores.flat<float>()(i) = scores[i].second;
