Lines Matching refs:kResultElementCount
86 static const int kResultElementCount = Generator::kResultElementCount;
92 typedef Array<Eigen::half, kResultElementCount> ResultType;
99 for (int i = 0; i < kResultElementCount; ++i) { in operator()
110 static const int kResultElementCount = Generator::kResultElementCount;
116 typedef Array<bfloat16, kResultElementCount> ResultType;
123 for (int i = 0; i < kResultElementCount; ++i) { in operator()
134 static const int kResultElementCount = Generator::kResultElementCount;
140 typedef Array<float, kResultElementCount> ResultType;
147 for (int i = 0; i < kResultElementCount; ++i) { in operator()
158 static const int kResultElementCount = Generator::kResultElementCount / 2;
164 typedef Array<double, kResultElementCount> ResultType;
171 for (int i = 0; i < kResultElementCount; ++i) { in operator()
182 static const int kResultElementCount = Generator::kResultElementCount;
188 typedef Array<int32, kResultElementCount> ResultType;
199 for (int i = 0; i < kResultElementCount; ++i) { in operator()
217 static const int kResultElementCount = Generator::kResultElementCount / 2;
223 typedef Array<int64, kResultElementCount> ResultType;
234 for (int i = 0; i < kResultElementCount; ++i) { in operator()
255 static const int kResultElementCount = 1;
257 static const int kNativeElementCount = Generator::kResultElementCount;
263 : generator_(gen), used_result_index_(Generator::kResultElementCount) {} in SingleSampleAdapter()
267 if (used_result_index_ == Generator::kResultElementCount) { in operator()
338 static const int kResultElementCount = Generator::kResultElementCount;
344 typedef Array<Eigen::half, kResultElementCount> ResultType;
351 for (int i = 0; i < kResultElementCount; i += 2) { in operator()
365 static const int kResultElementCount = Generator::kResultElementCount;
371 typedef Array<bfloat16, kResultElementCount> ResultType;
378 static_assert(kResultElementCount % 2 == 0, "kResultElementCount should be an even number"); in operator()
379 for (int i = 0; i < kResultElementCount; i += 2) { in operator()
394 static const int kResultElementCount = Generator::kResultElementCount;
400 typedef Array<float, kResultElementCount> ResultType;
407 for (int i = 0; i < kResultElementCount; i += 2) { in operator()
418 static const int kResultElementCount = Generator::kResultElementCount / 2;
424 typedef Array<double, kResultElementCount> ResultType;
431 for (int i = 0; i < kResultElementCount; i += 2) { in operator()
461 static const int kResultElementCount = SingleSampleGenerator::kNativeElementCount;
470 typedef Array<Eigen::half, kResultElementCount> ResultType;
489 if (index >= kResultElementCount) { in operator()
502 static const int kResultElementCount = SingleSampleGenerator::kNativeElementCount;
511 typedef Array<bfloat16, kResultElementCount> ResultType;
530 if (index >= kResultElementCount) { in operator()
544 static const int kResultElementCount = SingleSampleGenerator::kNativeElementCount;
553 typedef Array<float, kResultElementCount> ResultType;
572 if (index >= kResultElementCount) { in operator()
586 static const int kResultElementCount = (SingleSampleGenerator::kNativeElementCount > 1)
594 typedef Array<double, kResultElementCount> ResultType;
613 if (index >= kResultElementCount) { in operator()