Searched refs:kNumCategories (Results 1 – 3 of 3) sorted by relevance
21 constexpr int kNumCategories = 1001; variable43 CHECK_EQ(kNumCategories, ground_truth_labels.size()); in ImagenetTopKAccuracy()53 if (!output.shape().IsSameSize({1, kNumCategories})) { in ComputeEval()64 probabilities.reserve(kNumCategories); in ComputeEval()84 CHECK_EQ(kNumCategories, probabilities.size()); in ComputeEval()
23 const int kNumCategories = 1001; variable32 Tensor tensor(DT_FLOAT, TensorShape({1, kNumCategories})); in CreateOutputTensor()33 for (int i = 0; i < kNumCategories; i++) { in CreateOutputTensor()41 ground_truth.reserve(kNumCategories); in CreateGroundTruth()42 for (int i = 0; i < kNumCategories; i++) { in CreateGroundTruth()
42 kNumCategories = 3 // sentinel - keep last enumerator45 return static_cast<unsigned>(EventCategory::kNumCategories); in GetNumEventCategories()