• Home
  • Raw
  • Download

Lines Matching refs:kResultElementCount

77     static const int kResultElementCount = Generator::kResultElementCount;
83 typedef Array<Eigen::half, kResultElementCount> ResultType;
90 for (int i = 0; i < kResultElementCount; ++i) { in operator()
101 static const int kResultElementCount = Generator::kResultElementCount;
107 typedef Array<bfloat16, kResultElementCount> ResultType;
114 for (int i = 0; i < kResultElementCount; ++i) { in operator()
125 static const int kResultElementCount = Generator::kResultElementCount;
131 typedef Array<float, kResultElementCount> ResultType;
138 for (int i = 0; i < kResultElementCount; ++i) { in operator()
149 static const int kResultElementCount = Generator::kResultElementCount / 2;
155 typedef Array<double, kResultElementCount> ResultType;
162 for (int i = 0; i < kResultElementCount; ++i) { in operator()
173 static const int kResultElementCount = Generator::kResultElementCount;
179 typedef Array<int32, kResultElementCount> ResultType;
190 for (int i = 0; i < kResultElementCount; ++i) { in operator()
208 static const int kResultElementCount = Generator::kResultElementCount / 2;
214 typedef Array<int64, kResultElementCount> ResultType;
225 for (int i = 0; i < kResultElementCount; ++i) { in operator()
246 static const int kResultElementCount = 1;
248 static const int kNativeElementCount = Generator::kResultElementCount;
254 : generator_(gen), used_result_index_(Generator::kResultElementCount) {} in SingleSampleAdapter()
258 if (used_result_index_ == Generator::kResultElementCount) { in operator()
329 static const int kResultElementCount = Generator::kResultElementCount;
335 typedef Array<Eigen::half, kResultElementCount> ResultType;
342 for (int i = 0; i < kResultElementCount; i += 2) { in operator()
356 static const int kResultElementCount = Generator::kResultElementCount;
362 typedef Array<bfloat16, kResultElementCount> ResultType;
369 static_assert(kResultElementCount % 2 == 0, "kResultElementCount should be an even number"); in operator()
370 for (int i = 0; i < kResultElementCount; i += 2) { in operator()
385 static const int kResultElementCount = Generator::kResultElementCount;
391 typedef Array<float, kResultElementCount> ResultType;
398 for (int i = 0; i < kResultElementCount; i += 2) { in operator()
409 static const int kResultElementCount = Generator::kResultElementCount / 2;
415 typedef Array<double, kResultElementCount> ResultType;
422 for (int i = 0; i < kResultElementCount; i += 2) { in operator()
452 static const int kResultElementCount = SingleSampleGenerator::kNativeElementCount;
461 typedef Array<Eigen::half, kResultElementCount> ResultType;
480 if (index >= kResultElementCount) { in operator()
493 static const int kResultElementCount = SingleSampleGenerator::kNativeElementCount;
502 typedef Array<bfloat16, kResultElementCount> ResultType;
521 if (index >= kResultElementCount) { in operator()
535 static const int kResultElementCount = SingleSampleGenerator::kNativeElementCount;
544 typedef Array<float, kResultElementCount> ResultType;
563 if (index >= kResultElementCount) { in operator()
577 static const int kResultElementCount = (SingleSampleGenerator::kNativeElementCount > 1)
585 typedef Array<double, kResultElementCount> ResultType;
604 if (index >= kResultElementCount) { in operator()