Home
last modified time | relevance | path

Searched refs:PHILOX_DEVICE_INLINE (Results 1 – 8 of 8) sorted by relevance

/external/tensorflow/tensorflow/core/lib/random/
Dphilox_random.h37 #define PHILOX_DEVICE_INLINE PHILOX_DEVICE_FUNC PHILOX_INLINE macro
53 PHILOX_DEVICE_INLINE Array() { in Array()
59 PHILOX_DEVICE_INLINE const T& operator[](int index) const {
63 PHILOX_DEVICE_INLINE T& operator[](int index) { return data_[index]; }
115 PHILOX_DEVICE_INLINE
118 PHILOX_DEVICE_INLINE
124 PHILOX_DEVICE_INLINE
132 PHILOX_DEVICE_INLINE
135 PHILOX_DEVICE_INLINE
138 PHILOX_DEVICE_INLINE
[all …]
Drandom_distributions.h33 PHILOX_DEVICE_INLINE Eigen::half Uint16ToHalf(uint16 x);
35 PHILOX_DEVICE_INLINE bfloat16 Uint16ToGfloat16(uint16 x);
37 PHILOX_DEVICE_INLINE float Uint32ToFloat(uint32 x);
39 PHILOX_DEVICE_INLINE double Uint64ToDouble(uint32 x0, uint32 x1);
47 PHILOX_DEVICE_INLINE Int SignedAdd(Int a, in SignedAdd()
82 PHILOX_DEVICE_INLINE
106 PHILOX_DEVICE_INLINE
130 PHILOX_DEVICE_INLINE
154 PHILOX_DEVICE_INLINE
182 PHILOX_DEVICE_INLINE
[all …]
Dsimple_philox.h32 PHILOX_DEVICE_INLINE
36 PHILOX_DEVICE_INLINE uint32 Rand32() { return single_(); } in Rand32()
39 PHILOX_DEVICE_INLINE uint64 Rand64() { in Rand64()
45 PHILOX_DEVICE_INLINE float RandFloat() { return Uint32ToFloat(single_()); } in RandFloat()
48 PHILOX_DEVICE_INLINE double RandDouble() { in RandDouble()
Dphilox_random_test.cc42 PHILOX_DEVICE_INLINE
/external/tensorflow/tensorflow/core/kernels/
Drandom_ops_util.h28 PHILOX_DEVICE_INLINE void Uint64ToUint32s(uint64 input, uint32* output1, in Uint64ToUint32s()
34 PHILOX_DEVICE_INLINE uint64 Uint32sToUint64(uint32 input1, uint32 input2) { in Uint32sToUint64()
40 PHILOX_DEVICE_INLINE PhiloxRandom::ResultType GetCounterFromMem( in GetCounterFromMem()
48 PHILOX_DEVICE_INLINE void WriteCounterToMem( in WriteCounterToMem()
54 PHILOX_DEVICE_INLINE PhiloxRandom::Key GetKeyFromMem(uint64 const* ptr) { in GetKeyFromMem()
60 PHILOX_DEVICE_INLINE void WriteKeyToMem(PhiloxRandom::Key const& key, in WriteKeyToMem()
65 PHILOX_DEVICE_INLINE PhiloxRandom GetPhiloxRandomFromCounterKeyMem( in GetPhiloxRandomFromCounterKeyMem()
Dstateful_random_ops_cpu_gpu.h25 PHILOX_DEVICE_INLINE PhiloxRandom
31 PHILOX_DEVICE_INLINE void WritePhiloxRandomToMem(PhiloxRandom const& philox, in WritePhiloxRandomToMem()
38 PHILOX_DEVICE_INLINE PhiloxRandom SkipPhiloxRandom(PhiloxRandom const& philox, in SkipPhiloxRandom()
48 PHILOX_DEVICE_INLINE void UpdateMemWithPhiloxRandom(PhiloxRandom const& philox, in UpdateMemWithPhiloxRandom()
55 PHILOX_DEVICE_INLINE void UpdateCounterMemWithPhiloxRandom( in UpdateCounterMemWithPhiloxRandom()
Dstateless_random_gamma_op.h56 PHILOX_DEVICE_INLINE
60 PHILOX_DEVICE_INLINE
73 PHILOX_DEVICE_INLINE
Drandom_op_gpu.h37 PHILOX_DEVICE_INLINE void Run(const uint64* key, const uint64* counter,
45 PHILOX_DEVICE_INLINE void Run(const uint64* key, const uint64* counter,
141 PHILOX_DEVICE_INLINE void FillPhiloxRandomKernel<Distribution, false>::Run(
176 PHILOX_DEVICE_INLINE void FillPhiloxRandomKernel<Distribution, true>::Run(