Lines Matching refs:kResultElementCount

75     static const int kResultElementCount = Generator::kResultElementCount;
81 typedef Array<Eigen::half, kResultElementCount> ResultType;
88 for (int i = 0; i < kResultElementCount; ++i) { in operator()
99 static const int kResultElementCount = Generator::kResultElementCount;
105 typedef Array<bfloat16, kResultElementCount> ResultType;
112 for (int i = 0; i < kResultElementCount; ++i) { in operator()
123 static const int kResultElementCount = Generator::kResultElementCount;
129 typedef Array<float, kResultElementCount> ResultType;
136 for (int i = 0; i < kResultElementCount; ++i) { in operator()
147 static const int kResultElementCount = Generator::kResultElementCount / 2;
153 typedef Array<double, kResultElementCount> ResultType;
160 for (int i = 0; i < kResultElementCount; ++i) { in operator()
171 static const int kResultElementCount = Generator::kResultElementCount;
177 typedef Array<int32, kResultElementCount> ResultType;
188 for (int i = 0; i < kResultElementCount; ++i) { in operator()
206 static const int kResultElementCount = Generator::kResultElementCount / 2;
212 typedef Array<int64, kResultElementCount> ResultType;
223 for (int i = 0; i < kResultElementCount; ++i) { in operator()
244 static const int kResultElementCount = 1;
246 static const int kNativeElementCount = Generator::kResultElementCount;
252 : generator_(gen), used_result_index_(Generator::kResultElementCount) {} in SingleSampleAdapter()
256 if (used_result_index_ == Generator::kResultElementCount) { in operator()
327 static const int kResultElementCount = Generator::kResultElementCount;
333 typedef Array<Eigen::half, kResultElementCount> ResultType;
340 for (int i = 0; i < kResultElementCount; i += 2) { in operator()
354 static const int kResultElementCount = Generator::kResultElementCount;
360 typedef Array<bfloat16, kResultElementCount> ResultType;
367 static_assert(kResultElementCount % 2 == 0, "kResultElementCount should be an even number"); in operator()
368 for (int i = 0; i < kResultElementCount; i += 2) { in operator()
383 static const int kResultElementCount = Generator::kResultElementCount;
389 typedef Array<float, kResultElementCount> ResultType;
396 for (int i = 0; i < kResultElementCount; i += 2) { in operator()
407 static const int kResultElementCount = Generator::kResultElementCount / 2;
413 typedef Array<double, kResultElementCount> ResultType;
420 for (int i = 0; i < kResultElementCount; i += 2) { in operator()
450 static const int kResultElementCount = SingleSampleGenerator::kNativeElementCount;
459 typedef Array<Eigen::half, kResultElementCount> ResultType;
478 if (index >= kResultElementCount) { in operator()
491 static const int kResultElementCount = SingleSampleGenerator::kNativeElementCount;
500 typedef Array<bfloat16, kResultElementCount> ResultType;
519 if (index >= kResultElementCount) { in operator()
533 static const int kResultElementCount = SingleSampleGenerator::kNativeElementCount;
542 typedef Array<float, kResultElementCount> ResultType;
561 if (index >= kResultElementCount) { in operator()
575 static const int kResultElementCount = (SingleSampleGenerator::kNativeElementCount > 1)
583 typedef Array<double, kResultElementCount> ResultType;
602 if (index >= kResultElementCount) { in operator()