Home
last modified time | relevance | path

Searched refs:PHILOX_DEVICE_INLINE (Results 1 – 5 of 5) 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.h36 PHILOX_DEVICE_INLINE Eigen::half Uint16ToHalf(uint16 x);
38 PHILOX_DEVICE_INLINE bfloat16 Uint16ToGfloat16(uint16 x);
40 PHILOX_DEVICE_INLINE float Uint32ToFloat(uint32 x);
42 PHILOX_DEVICE_INLINE double Uint64ToDouble(uint32 x0, uint32 x1);
50 PHILOX_DEVICE_INLINE Int SignedAdd(Int a, in SignedAdd()
85 PHILOX_DEVICE_INLINE
109 PHILOX_DEVICE_INLINE
133 PHILOX_DEVICE_INLINE
157 PHILOX_DEVICE_INLINE
185 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/
Dstateful_random_ops_cpu_gpu.h30 PHILOX_DEVICE_INLINE void Int64ToUint32s(int64 input, uint32* output1, in Int64ToUint32s()
37 PHILOX_DEVICE_INLINE int64 Uint32sToInt64(uint32 input1, uint32 input2) { in Uint32sToInt64()
43 PHILOX_DEVICE_INLINE PhiloxRandom
53 PHILOX_DEVICE_INLINE void WritePhiloxRandomToMem(PhiloxRandom const& philox, in WritePhiloxRandomToMem()
62 PHILOX_DEVICE_INLINE void UpdateMemWithPhiloxRandom(PhiloxRandom const& philox, in UpdateMemWithPhiloxRandom()