Searched refs:kResultElementCount (Results 1 – 11 of 11) sorted by relevance
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;[all …]
38 static constexpr int kResultElementCount = PhiloxRandom::kResultElementCount; member in tensorflow::random::__anon305595b30111::TrivialPhiloxDistribution
36 const int granularity = Distribution::kResultElementCount; in FillRandoms()
50 int granularity = Distribution::kResultElementCount; in FillRandomsWithSingles()310 static const int kResultElementCount = 1; member in tensorflow::random::__anon552513520111::MockGenerator
108 static const int kResultElementCount = 4; variable
68 const int kDistSize = Uniform::kResultElementCount; in TruncatedNormalKernel()139 remaining_samples -= gen.kResultElementCount; in TruncatedNormalKernel()169 remaining_samples -= gen.kResultElementCount; in TruncatedNormalKernel()178 remaining_samples -= gen.kResultElementCount; in TruncatedNormalKernel()205 remaining_samples -= gen.kResultElementCount; in TruncatedNormalKernel()
82 const int kGroupSize = Distribution::kResultElementCount; in Run()113 const int kGroupSize = Distribution::kResultElementCount; in Run()117 PhiloxRandom::kResultElementCount; in Run()158 const int kGroupSize = Distribution::kResultElementCount; in operator ()()165 random::PhiloxRandom::kResultElementCount * in operator ()()310 uniform_remaining = Uniform::kResultElementCount; \ in Compute()402 norm_remaining = Normal::kResultElementCount; in Compute()586 const size_t kGroupSize = Distribution::kResultElementCount; in operator ()()637 const size_t kGroupSize = Distribution::kResultElementCount; in operator ()()640 PhiloxRandom::kResultElementCount; in operator ()()
133 const int kGroupSize = Distribution::kResultElementCount;169 const int kGroupSize = Distribution::kResultElementCount;172 PhiloxRandom::kResultElementCount;
70 const int kGroupSize = Distribution::kResultElementCount; in operator ()()
48 uniform_remaining = Uniform::kResultElementCount; \
50 size_t num_samples = Distribution::kResultElementCount; in ComputeRandomUniformArray()