| /external/tensorflow/tensorflow/stream_executor/ |
| plugin.cc | 27 : blas_(kDefault), dnn_(kDefault), fft_(kDefault), rng_(kDefault) {}
31 rng_ == rhs.rng_;
50 rng_ = rng;
|
| plugin.h | 80 PluginId rng() const { return rng_; }
83 PluginId blas_, dnn_, fft_, rng_; member in class:perftools::gputools::PluginConfig
|
| stream_executor_pimpl.cc | 417 if (rng_ != nullptr) {
418 return rng_.get();
421 rng_.reset(implementation_->CreateRng());
422 return rng_.get();
|
| stream_executor_pimpl.h | 616 std::unique_ptr<rng::RngSupport> rng_ GUARDED_BY(mu_);
[all...] |
| /external/vulkan-validation-layers/demos/smoke/ |
| Simulation.cpp | 60 ColorPicker(unsigned int rng_seed) : rng_(rng_seed), red_(0.0f, 1.0f), green_(0.0f, 1.0f), blue_(0.0f, 1.0f) {}
62 glm::vec3 pick() { return glm::vec3{red_(rng_), green_(rng_), blue_(rng_)}; }
65 std::mt19937 rng_; member in class:__anon43415::ColorPicker
73 Animation::Animation(unsigned int rng_seed, float scale) : rng_(rng_seed), dir_(-1.0f, 1.0f), speed_(0.1f, 1.0f) {
74 float x = dir_(rng_);
75 float y = dir_(rng_);
76 float z = dir_(rng_);
81 current_.speed = speed_(rng_);
140 std::mt19937 rng_; member in class:__anon43416::RandomCurve
[all...] |
| Simulation.h | 43 std::mt19937 rng_; member in class:Animation
70 std::mt19937 rng_; member in class:Path
|
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| cuda_rng.cc | 109 CUDARng::CUDARng(CUDAExecutor *parent) : parent_(parent), rng_(nullptr) {}
112 if (rng_ != nullptr) {
113 wrap::curandDestroyGenerator(parent_, rng_);
119 CHECK(rng_ == nullptr);
122 wrap::curandCreateGenerator(parent_, &rng_, CURAND_RNG_PSEUDO_DEFAULT);
128 CHECK(rng_ != nullptr);
134 wrap::curandSetStream(parent_, rng_, AsCUDAStreamValue(stream));
173 parent_, rng_, reinterpret_cast<float *>(CUDAMemoryMutable(v)),
177 parent_, rng_, reinterpret_cast<double *>(CUDAMemoryMutable(v)),
221 func(parent_, rng_, CUDAMemoryMutable(v), element_count, mean, stddev)
[all...] |
| cuda_rng.h | 95 curandGenerator_t rng_ GUARDED_BY(mu_);
|
| /external/tensorflow/tensorflow/contrib/boosted_trees/lib/testutil/ |
| random_tree_gen.cc | 61 : rng_(rng),
94 0, rng_->RandFloat());
116 left->mutable_leaf()->mutable_sparse_vector()->add_value(rng_->RandFloat());
120 rng_->RandFloat());
190 if (rng_->RandFloat() < denseSplitProb) {
195 if (rng_->RandFloat() < kLeftProb) {
203 split->set_threshold(rng_->RandFloat());
204 split->set_feature_column(rng_->Uniform(feature_size));
|
| random_tree_gen.h | 50 tensorflow::random::SimplePhilox* rng_; member in class:tensorflow::boosted_trees::testutil::RandomTreeGen
|
| /external/tensorflow/tensorflow/contrib/tensor_forest/hybrid/core/ops/ |
| stochastic_hard_routing_function_op.cc | 102 rng_ = std::unique_ptr<random::SimplePhilox>(
163 if (left_prob < rng_->RandFloat()) {
187 std::unique_ptr<random::SimplePhilox> rng_; member in class:tensorflow::StochasticHardRoutingFunction
|
| /external/tensorflow/tensorflow/contrib/tensor_forest/kernels/v4/ |
| input_data.h | 64 rng_ = std::unique_ptr<random::SimplePhilox>(
119 std::unique_ptr<random::SimplePhilox> rng_; member in class:tensorflow::tensorforest::TensorDataSet
|
| input_data.cc | 133 int rand_feature = rng_->Uniform(num_total_features);
|
| grow_stats.cc | 128 rng_ = std::unique_ptr<random::SimplePhilox>(
366 int g1 = BootstrapGini(weight_sum_, 2 * num_outputs_, ds1, rng_.get());
372 int g2 = BootstrapGini(weight_sum_, 2 * num_outputs_, ds2, rng_.get());
[all...] |
| grow_stats.h | 289 std::unique_ptr<random::SimplePhilox> rng_; member in class:tensorflow::tensorforest::ClassificationStats
|
| /external/tensorflow/tensorflow/core/kernels/ |
| word2vec_kernels.cc | 52 : OpKernel(ctx), rng_(&philox_) {
128 random::SimplePhilox rng_ GUARDED_BY(mu_);
157 if (rng_.RandFloat() > keep_prob) {
165 const int32 skip = 1 + rng_.Uniform(window_size_);
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| hlo_module.h | 227 mutable std::mt19937_64 rng_{42};
|
| hlo_module.cc | 595 return rng_();
|
| /external/tensorflow/tensorflow/contrib/tensorboard/db/ |
| summary_db_writer.cc | 692 rng_{std::mt19937_64::default_seed} {
714 i = rng_() % (i + 1);
907 std::mt19937_64 rng_ GUARDED_BY(mu_);
[all...] |
