Home
last modified time | relevance | path

Searched refs:UniformDistribution (Results 1 – 23 of 23) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Drandom_op_gpu.cu.cc42 GPUDevice, random::UniformDistribution<random::PhiloxRandom, Eigen::half> >;
44 GPUDevice, random::UniformDistribution<random::PhiloxRandom, float> >;
46 GPUDevice, random::UniformDistribution<random::PhiloxRandom, double> >;
48 GPUDevice, random::UniformDistribution<random::PhiloxRandom, int32> >;
50 GPUDevice, random::UniformDistribution<random::PhiloxRandom, int64> >;
Dstateful_random_ops_gpu.cu.cc140 GPUDevice, random::UniformDistribution<random::PhiloxRandom, Eigen::half> >;
142 GPUDevice, random::UniformDistribution<random::PhiloxRandom, float> >;
144 GPUDevice, random::UniformDistribution<random::PhiloxRandom, double> >;
146 GPUDevice, random::UniformDistribution<random::PhiloxRandom, int32> >;
148 GPUDevice, random::UniformDistribution<random::PhiloxRandom, int64> >;
Drandom_op.cc120 typedef random::UniformDistribution<random::PhiloxRandom, IntType> in Compute()
179 typedef random::UniformDistribution<PhiloxRandom, double> Uniform; in Compute()
331 CPUDevice, random::UniformDistribution<random::PhiloxRandom, TYPE>>; \
343 PhiloxRandomOp<CPUDevice, random::UniformDistribution< \
373 CPUDevice, random::UniformDistribution<random::PhiloxRandom, IntType>>; \
404 PhiloxRandomOp<GPUDevice, random::UniformDistribution< \
433 GPUDevice, random::UniformDistribution<random::PhiloxRandom, IntType>>; \
Dstateless_random_ops.cc143 typedef random::UniformDistribution<random::PhiloxRandom, IntType> in Fill()
211 StatelessRandomOp<DEVICE##Device, random::UniformDistribution< \
Dparameterized_truncated_normal_op.cc84 using Uniform = random::UniformDistribution<random::PhiloxRandom, T>; in operator ()()
294 random::UniformDistribution<random::PhiloxRandom, T>::kElementCost; in operator ()()
342 using Uniform = random::UniformDistribution<random::PhiloxRandom, T>; in operator ()()
585 random::UniformDistribution<random::PhiloxRandom, T>::kElementCost; in operator ()()
Dstateless_random_gamma_op_gpu.cu.cc55 typedef random::UniformDistribution<random::PhiloxRandom, double> Uniform; in FillKernel()
Dstateless_random_ops_v2.cc153 typedef random::UniformDistribution<random::PhiloxRandom, IntType> in Fill()
272 StatelessRandomOp<DEVICE##Device, random::UniformDistribution< \
Dmultinomial_op_gpu.cu.cc68 typedef random::UniformDistribution<random::PhiloxRandom, float> Dist; in operator ()()
Dstateful_random_ops.cc219 typedef random::UniformDistribution<random::PhiloxRandom, IntType> in Compute()
371 random::UniformDistribution<PhiloxRandom, TYPE> >); \
Dstateless_random_gamma_op.cc63 typedef random::UniformDistribution<random::PhiloxRandom, double> Uniform; in Fill()
Drandom_poisson_op.cc97 typedef random::UniformDistribution<random::PhiloxRandom, CT> Uniform; in operator ()()
Dparameterized_truncated_normal_op_gpu.cu.cc65 typedef random::UniformDistribution<random::PhiloxRandom, T> Uniform; in TruncatedNormalKernel()
Drandom_binomial_op.cc62 typedef random::UniformDistribution<random::PhiloxRandom, double> Uniform;
/external/webrtc/third_party/abseil-cpp/absl/random/internal/
Duniform_helper.h153 using UniformDistribution = variable
159 struct UniformDistributionWrapper : public UniformDistribution<NumType> {
162 : UniformDistribution<NumType>( in UniformDistributionWrapper()
167 : UniformDistribution<NumType>( in UniformDistributionWrapper()
172 : UniformDistribution<NumType>(std::numeric_limits<NumType>::lowest(), in UniformDistributionWrapper()
/external/abseil-cpp/absl/random/internal/
Duniform_helper.h153 using UniformDistribution = variable
159 struct UniformDistributionWrapper : public UniformDistribution<NumType> {
162 : UniformDistribution<NumType>( in UniformDistributionWrapper()
167 : UniformDistribution<NumType>( in UniformDistributionWrapper()
172 : UniformDistribution<NumType>(std::numeric_limits<NumType>::lowest(), in UniformDistributionWrapper()
/external/openscreen/third_party/abseil/src/absl/random/internal/
Duniform_helper.h212 using UniformDistribution = variable
223 struct UniformDistributionWrapper : public UniformDistribution<NumType> {
226 : UniformDistribution<NumType>( in UniformDistributionWrapper()
231 : UniformDistribution<NumType>( in UniformDistributionWrapper()
236 : UniformDistribution<NumType>(std::numeric_limits<NumType>::lowest(), in UniformDistributionWrapper()
/external/rust/crates/grpcio-sys/grpc/third_party/abseil-cpp/absl/random/internal/
Duniform_helper.h212 using UniformDistribution = variable
223 struct UniformDistributionWrapper : public UniformDistribution<NumType> {
226 : UniformDistribution<NumType>( in UniformDistributionWrapper()
231 : UniformDistribution<NumType>( in UniformDistributionWrapper()
236 : UniformDistribution<NumType>(std::numeric_limits<NumType>::lowest(), in UniformDistributionWrapper()
/external/libtextclassifier/abseil-cpp/absl/random/internal/
Duniform_helper.h212 using UniformDistribution = variable
223 struct UniformDistributionWrapper : public UniformDistribution<NumType> {
226 : UniformDistribution<NumType>( in UniformDistributionWrapper()
231 : UniformDistribution<NumType>( in UniformDistributionWrapper()
236 : UniformDistribution<NumType>(std::numeric_limits<NumType>::lowest(), in UniformDistributionWrapper()
/external/tensorflow/tensorflow/core/lib/random/
Drandom_distributions.h67 class UniformDistribution; variable
70 class UniformDistribution<Generator, Eigen::half> {
94 class UniformDistribution<Generator, bfloat16> {
118 class UniformDistribution<Generator, float> {
142 class UniformDistribution<Generator, double> {
166 class UniformDistribution<Generator, int32> {
179 UniformDistribution(int32 lo, int32 hi) in UniformDistribution() function
201 class UniformDistribution<Generator, int64> {
214 UniformDistribution(int64 lo, int64 hi) in UniformDistribution() function
Drandom_distributions_test.cc157 FillRandoms<UniformDistribution<PhiloxRandom, T> >(gen, &v1[0], v1.size()); in UniformMomentsTest()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dstream_executor_util.cc239 T>::type static UniformDistribution(T lhs, T rhs,
245 T>::type static UniformDistribution(T lhs, T rhs, in UniformDistribution() function
273 auto rand_val = UniformDistribution(RandomType(0), upper_bound, &gen); in InitializeTypedBuffer()
/external/tensorflow/tensorflow/lite/toco/graph_transformations/
Dresolve_constant_random_uniform.cc30 typedef tensorflow::random::UniformDistribution< in ComputeRandomUniformArray()
/external/tensorflow/tensorflow/compiler/mlir/lite/transforms/
Dlegalize_tf.cc166 typedef tensorflow::random::UniformDistribution< in matchAndRewrite()