Lines Matching refs:kResultElementCount
76 static const int kResultElementCount = Generator::kResultElementCount;
82 typedef Array<Eigen::half, kResultElementCount> ResultType;
89 for (int i = 0; i < kResultElementCount; ++i) { in operator()
100 static const int kResultElementCount = Generator::kResultElementCount;
106 typedef Array<bfloat16, kResultElementCount> ResultType;
113 for (int i = 0; i < kResultElementCount; ++i) { in operator()
124 static const int kResultElementCount = Generator::kResultElementCount;
130 typedef Array<float, kResultElementCount> ResultType;
137 for (int i = 0; i < kResultElementCount; ++i) { in operator()
148 static const int kResultElementCount = Generator::kResultElementCount / 2;
154 typedef Array<double, kResultElementCount> ResultType;
161 for (int i = 0; i < kResultElementCount; ++i) { in operator()
172 static const int kResultElementCount = Generator::kResultElementCount;
178 typedef Array<int32, kResultElementCount> ResultType;
189 for (int i = 0; i < kResultElementCount; ++i) { in operator()
207 static const int kResultElementCount = Generator::kResultElementCount / 2;
213 typedef Array<int64, kResultElementCount> ResultType;
224 for (int i = 0; i < kResultElementCount; ++i) { in operator()
245 static const int kResultElementCount = 1;
247 static const int kNativeElementCount = Generator::kResultElementCount;
253 : generator_(gen), used_result_index_(Generator::kResultElementCount) {} in SingleSampleAdapter()
257 if (used_result_index_ == Generator::kResultElementCount) { in operator()
328 static const int kResultElementCount = Generator::kResultElementCount;
334 typedef Array<Eigen::half, kResultElementCount> ResultType;
341 for (int i = 0; i < kResultElementCount; i += 2) { in operator()
355 static const int kResultElementCount = Generator::kResultElementCount;
361 typedef Array<bfloat16, kResultElementCount> ResultType;
368 static_assert(kResultElementCount % 2 == 0, "kResultElementCount should be an even number"); in operator()
369 for (int i = 0; i < kResultElementCount; i += 2) { in operator()
384 static const int kResultElementCount = Generator::kResultElementCount;
390 typedef Array<float, kResultElementCount> ResultType;
397 for (int i = 0; i < kResultElementCount; i += 2) { in operator()
408 static const int kResultElementCount = Generator::kResultElementCount / 2;
414 typedef Array<double, kResultElementCount> ResultType;
421 for (int i = 0; i < kResultElementCount; i += 2) { in operator()
451 static const int kResultElementCount = SingleSampleGenerator::kNativeElementCount;
460 typedef Array<Eigen::half, kResultElementCount> ResultType;
479 if (index >= kResultElementCount) { in operator()
492 static const int kResultElementCount = SingleSampleGenerator::kNativeElementCount;
501 typedef Array<bfloat16, kResultElementCount> ResultType;
520 if (index >= kResultElementCount) { in operator()
534 static const int kResultElementCount = SingleSampleGenerator::kNativeElementCount;
543 typedef Array<float, kResultElementCount> ResultType;
562 if (index >= kResultElementCount) { in operator()
576 static const int kResultElementCount = (SingleSampleGenerator::kNativeElementCount > 1)
584 typedef Array<double, kResultElementCount> ResultType;
603 if (index >= kResultElementCount) { in operator()