Home
last modified time | relevance | path

Searched refs:rng_ (Results 1 – 25 of 30) sorted by relevance

12

/external/libaom/libaom/test/
Dblend_a64_mask_test.cc56 switch (this->rng_(3)) { in GetSources()
83 uint8_t Rand1() { return this->rng_.Rand8() & 1; } in Rand1()
93 dst_offset_ = this->rng_(33); in RunOneTest()
94 dst_stride_ = this->rng_(kMaxWidth + 1 - w_) + w_; in RunOneTest()
96 src0_offset_ = this->rng_(33); in RunOneTest()
97 src0_stride_ = this->rng_(kMaxWidth + 1 - w_) + w_; in RunOneTest()
99 src1_offset_ = this->rng_(33); in RunOneTest()
100 src1_stride_ = this->rng_(kMaxWidth + 1 - w_) + w_; in RunOneTest()
103 this->rng_(kMaxWidth + 1 - w_ * (subx_ ? 2 : 1)) + w_ * (subx_ ? 2 : 1); in RunOneTest()
196 int bsize = rng_.Rand8() % BLOCK_SIZES_ALL; in TEST_P()
[all …]
Dpickrst_test.cc71 libaom_test::ACMRandom rng_; member in pickrst_test_lowbd::PixelProjErrorTest
79 int h_end = run_times != 1 ? 128 : (rng_.Rand16() % MAX_DATA_BLOCK) + 1; in RunPixelProjErrorTest()
80 int v_end = run_times != 1 ? 128 : (rng_.Rand16() % MAX_DATA_BLOCK) + 1; in RunPixelProjErrorTest()
91 dgd_[i] = rng_.Rand8(); in RunPixelProjErrorTest()
92 src_[i] = rng_.Rand8(); in RunPixelProjErrorTest()
93 flt0_[i] = rng_.Rand15Signed(); in RunPixelProjErrorTest()
94 flt1_[i] = rng_.Rand15Signed(); in RunPixelProjErrorTest()
96 xq[0] = rng_.Rand8() % (1 << SGRPROJ_PRJ_BITS); in RunPixelProjErrorTest()
97 xq[1] = rng_.Rand8() % (1 << SGRPROJ_PRJ_BITS); in RunPixelProjErrorTest()
98 params.r[0] = run_times == 1 ? (rng_.Rand8() % MAX_RADIUS) : (iter % 2); in RunPixelProjErrorTest()
[all …]
Dblend_a64_mask_1d_test.cc49 w_ = 2 << this->rng_(MAX_SB_SIZE_LOG2); in Common()
50 h_ = 2 << this->rng_(MAX_SB_SIZE_LOG2); in Common()
52 dst_offset_ = this->rng_(33); in Common()
53 dst_stride_ = this->rng_(kMaxWidth + 1 - w_) + w_; in Common()
55 src0_offset_ = this->rng_(33); in Common()
56 src0_stride_ = this->rng_(kMaxWidth + 1 - w_) + w_; in Common()
58 src1_offset_ = this->rng_(33); in Common()
59 src1_stride_ = this->rng_(kMaxWidth + 1 - w_) + w_; in Common()
64 switch (this->rng_(3)) { in Common()
137 dst_ref_[i] = rng_.Rand8(); in TEST_P()
[all …]
Dav1_wedge_utils_test.cc42 WedgeUtilsSSEFuncTest() : rng_(ACMRandom::DeterministicSeed()) {} in WedgeUtilsSSEFuncTest()
46 ACMRandom rng_; member in __anon7135b31c0111::WedgeUtilsSSEFuncTest
89 s[i] = rng_.Rand8(); in TEST_F()
90 m[i] = rng_(MAX_MASK_VALUE + 1); in TEST_F()
93 const int w = 1 << (rng_(MAX_SB_SIZE_LOG2 + 1 - 3) + 3); in TEST_F()
94 const int h = 1 << (rng_(MAX_SB_SIZE_LOG2 + 1 - 3) + 3); in TEST_F()
98 p0[j] = clamp(s[j] + rng_(33) - 16, 0, UINT8_MAX); in TEST_F()
99 p1[j] = clamp(s[j] + rng_(33) - 16, 0, UINT8_MAX); in TEST_F()
139 r1[i] = rng_(2 * INT8_MAX - 2 * INT8_MIN + 1) + 2 * INT8_MIN; in TEST_F()
140 d[i] = rng_(2 * INT8_MAX - 2 * INT8_MIN + 1) + 2 * INT8_MIN; in TEST_F()
[all …]
Dintra_edge_test.cc82 size_ = 4 * (this->rng_(4) + 1); in TEST_P()
86 pix = rng_.Rand8(); in TEST_P()
127 switch (rng_(3)) { in TEST_P()
134 size_ = 4 * (this->rng_(4) + 1); in TEST_P()
138 pix = rng_(hi); in TEST_P()
211 strength_ = this->rng_(4); in TEST_P()
212 size_ = 4 * (this->rng_(128 / 4) + 1) + 1; in TEST_P()
216 pix = rng_.Rand8(); in TEST_P()
250 switch (rng_(3)) { in TEST_P()
256 strength_ = this->rng_(4); in TEST_P()
[all …]
Dobmc_variance_test.cc50 const int pre_stride = this->rng_(MAX_SB_SIZE + 1); in TEST_P()
53 pre[i] = this->rng_.Rand8(); in TEST_P()
54 wsrc[i] = this->rng_.Rand8() * this->rng_(kMaskMax * kMaskMax + 1); in TEST_P()
55 mask[i] = this->rng_(kMaskMax * kMaskMax + 1); in TEST_P()
101 const int pre_stride = this->rng_(MAX_SB_SIZE + 1); in TEST_P()
104 pre[i] = this->rng_.Rand8(); in TEST_P()
105 wsrc[i] = this->rng_.Rand8() * this->rng_(kMaskMax * kMaskMax + 1); in TEST_P()
106 mask[i] = this->rng_(kMaskMax * kMaskMax + 1); in TEST_P()
193 const int pre_stride = this->rng_(MAX_SB_SIZE + 1); in TEST_P()
196 pre[i] = this->rng_(1 << params_.bit_depth); in TEST_P()
[all …]
Dobmc_sad_test.cc47 const int pre_stride = rng_(MAX_SB_SIZE + 1); in TEST_P()
50 pre[i] = rng_.Rand8(); in TEST_P()
51 wsrc[i] = rng_.Rand8() * rng_(kMaskMax * kMaskMax + 1); in TEST_P()
52 mask[i] = rng_(kMaskMax * kMaskMax + 1); in TEST_P()
146 const int pre_stride = rng_(MAX_SB_SIZE + 1); in TEST_P()
149 pre[i] = rng_(1 << 12); in TEST_P()
150 wsrc[i] = rng_(1 << 12) * rng_(kMaskMax * kMaskMax + 1); in TEST_P()
151 mask[i] = rng_(kMaskMax * kMaskMax + 1); in TEST_P()
Dav1_nn_predict_test.cc66 libaom_test::ACMRandom rng_; member in __anonb536a0660111::NnPredictTest
97 inputs[node] = ((float)rng_.Rand31() - (1 << 30)) / (1u << 31); in RunNnPredictTest()
101 bias[layer][node] = ((float)rng_.Rand31() - (1 << 30)) / (1u << 31); in RunNnPredictTest()
105 weights[layer][node] = ((float)rng_.Rand31() - (1 << 30)) / (1u << 31); in RunNnPredictTest()
111 bias[layer][node] = ((float)rng_.Rand31() - (1 << 30)) / (1u << 31); in RunNnPredictTest()
115 weights[layer][node] = ((float)rng_.Rand31() - (1 << 30)) / (1u << 31); in RunNnPredictTest()
Dwiener_test.cc138 libaom_test::ACMRandom rng_; member in wiener_lowbd::WienerTest
150 const int h_start = ((rng_.Rand16() % (MAX_WIENER_BLOCK / 2)) & (~7)); in RunWienerTest()
152 run_times != 1 ? 256 : ((rng_.Rand16() % MAX_WIENER_BLOCK) & (~7)) + 8; in RunWienerTest()
153 const int v_start = ((rng_.Rand16() % (MAX_WIENER_BLOCK / 2)) & (~7)); in RunWienerTest()
155 run_times != 1 ? 256 : ((rng_.Rand16() % MAX_WIENER_BLOCK) & (~7)) + 8; in RunWienerTest()
161 dgd_buf[i] = rng_.Rand8(); in RunWienerTest()
162 src_buf[i] = rng_.Rand8(); in RunWienerTest()
414 libaom_test::ACMRandom rng_; member in wiener_highbd::WienerTestHighbd
428 const int h_start = ((rng_.Rand16() % (MAX_WIENER_BLOCK / 2)) & (~7)); in RunWienerTest()
430 run_times != 1 ? 256 : ((rng_.Rand16() % MAX_WIENER_BLOCK) & (~7)) + 8; in RunWienerTest()
[all …]
Dsum_squares_test.cc192 src[i] = rng_(kInt13Max * 2 + 1) - kInt13Max; in TEST_P()
194 const int N = rng_(2) ? rng_(kMaxSize * kMaxSize + 1 - kMaxSize) + kMaxSize in TEST_P()
195 : rng_(kMaxSize) + 1; in TEST_P()
209 if (rng_(2)) { in TEST_P()
215 const int N = rng_(2) ? rng_(kMaxSize * kMaxSize + 1 - kMaxSize) + kMaxSize in TEST_P()
216 : rng_(kMaxSize) + 1; in TEST_P()
Dhorver_correlation_test.cc46 ACMRandom rng_; member in __anone90112b00111::HorverTest
59 data_buf_[i] = (rng_.Rand16() % (1 << 12)) - (1 << 11); in RunHorverTest()
78 data_buf_[i] = rng_.Rand16() % (1 << 12); in RunHorverSpeedTest()
Dfunction_equivalence_test.h55 FunctionEquivalenceTest() : rng_(ACMRandom::DeterministicSeed()) {} in FunctionEquivalenceTest()
64 ACMRandom rng_;
Ddr_prediction_test.cc169 above_data_[i] = rng_.Rand8(); in DrPredTest()
170 left_data_[i] = rng_.Rand8(); in DrPredTest()
300 ACMRandom rng_; member in __anond4b7649f0111::DrPredTest
/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_rng.cc64 GpuRng::GpuRng(GpuExecutor* parent) : parent_(parent), rng_(nullptr) {} in GpuRng()
67 if (rng_ != nullptr) { in ~GpuRng()
69 curandDestroyGenerator(rng_); in ~GpuRng()
75 CHECK(rng_ == nullptr); in Init()
78 curandStatus_t ret = curandCreateGenerator(&rng_, CURAND_RNG_PSEUDO_DEFAULT); in Init()
84 CHECK(rng_ != nullptr); in Init()
90 curandStatus_t ret = curandSetStream(rng_, AsGpuStreamValue(stream)); in SetStream()
129 rng_, reinterpret_cast<float*>(GpuMemoryMutable(v)), element_count); in DoPopulateRandUniformInternal()
132 rng_, reinterpret_cast<double*>(GpuMemoryMutable(v)), element_count); in DoPopulateRandUniformInternal()
176 func(rng_, GpuMemoryMutable(v), element_count, mean, stddev); in DoPopulateRandGaussianInternal()
[all …]
/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_rng.cc120 GpuRng::GpuRng(GpuExecutor* parent) : parent_(parent), rng_(nullptr) {} in GpuRng()
123 if (rng_ != nullptr) { in ~GpuRng()
124 wrap::hiprandDestroyGenerator(parent_, rng_); in ~GpuRng()
130 CHECK(rng_ == nullptr); in Init()
133 wrap::hiprandCreateGenerator(parent_, &rng_, HIPRAND_RNG_PSEUDO_DEFAULT); in Init()
139 CHECK(rng_ != nullptr); in Init()
145 wrap::hiprandSetStream(parent_, rng_, AsGpuStreamValue(stream)); in SetStream()
183 parent_, rng_, reinterpret_cast<float*>(GpuMemoryMutable(v)), in DoPopulateRandUniformInternal()
187 parent_, rng_, reinterpret_cast<double*>(GpuMemoryMutable(v)), in DoPopulateRandUniformInternal()
231 func(parent_, rng_, GpuMemoryMutable(v), element_count, mean, stddev); in DoPopulateRandGaussianInternal()
[all …]
/external/tensorflow/tensorflow/contrib/boosted_trees/lib/testutil/
Drandom_tree_gen.cc61 : rng_(rng), in RandomTreeGen()
94 0, rng_->RandFloat()); in Generate()
116 left->mutable_leaf()->mutable_sparse_vector()->add_value(rng_->RandFloat()); in Generate()
120 rng_->RandFloat()); in Generate()
190 if (rng_->RandFloat() < denseSplitProb) { in GenerateSplit()
195 if (rng_->RandFloat() < kLeftProb) { in GenerateSplit()
203 split->set_threshold(rng_->RandFloat()); in GenerateSplit()
204 split->set_feature_column(rng_->Uniform(feature_size)); in GenerateSplit()
Drandom_tree_gen.h50 tensorflow::random::SimplePhilox* rng_;
/external/tensorflow/tensorflow/stream_executor/
Dplugin.cc26 : blas_(kDefault), dnn_(kDefault), fft_(kDefault), rng_(kDefault) {} in PluginConfig()
30 rng_ == rhs.rng_; in operator ==()
49 rng_ = rng; in SetRng()
Dplugin.h79 PluginId rng() const { return rng_; } in rng()
82 PluginId blas_, dnn_, fft_, rng_;
Dstream_executor_pimpl.cc471 if (rng_ != nullptr) { in AsRng()
472 return rng_.get(); in AsRng()
475 rng_.reset(implementation_->CreateRng()); in AsRng()
476 return rng_.get(); in AsRng()
/external/tensorflow/tensorflow/contrib/tensor_forest/hybrid/core/ops/
Dstochastic_hard_routing_function_op.cc102 rng_ = std::unique_ptr<random::SimplePhilox>( in StochasticHardRoutingFunction()
163 if (left_prob < rng_->RandFloat()) { in Compute()
187 std::unique_ptr<random::SimplePhilox> rng_; member in tensorflow::StochasticHardRoutingFunction
/external/tensorflow/tensorflow/contrib/tensor_forest/kernels/v4/
Dinput_data.h68 rng_ = std::unique_ptr<random::SimplePhilox>( in TensorDataSet()
123 std::unique_ptr<random::SimplePhilox> rng_; variable
Dinput_data.cc136 rand_feature = rng_->Uniform(num_total_features); in RandomSample()
/external/tensorflow/tensorflow/core/kernels/
Dword2vec_kernels.cc52 : OpKernel(ctx), rng_(&philox_) { in SkipgramOp()
128 random::SimplePhilox rng_ GUARDED_BY(mu_);
157 if (rng_.RandFloat() > keep_prob) { in NextExample()
165 const int32 skip = 1 + rng_.Uniform(window_size_); in NextExample()
/external/tensorflow/tensorflow/stream_executor/gpu/
Dgpu_rng.h94 GpuRngHandle rng_ GUARDED_BY(mu_);

12