Searched refs:Uint32sToUint64 (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | random_ops_util.h | 34 PHILOX_DEVICE_INLINE uint64 Uint32sToUint64(uint32 input1, uint32 input2) { in Uint32sToUint64() function
50 ptr[0] = Uint32sToUint64(counter[0], counter[1]); in WriteCounterToMem()
51 ptr[1] = Uint32sToUint64(counter[2], counter[3]); in WriteCounterToMem()
62 *ptr = Uint32sToUint64(key[0], key[1]); in WriteKeyToMem()
|
| /external/tensorflow/tensorflow/compiler/xla/client/lib/ |
| D | prng.cc | 131 XlaOp Uint32sToUint64(std::array<XlaOp, 2> u32s) { in Uint32sToUint64() function
260 XlaOp result = Uint32sToUint64(outputs); in ThreeFryRngBit64()
353 return {Uint32sToUint64({u32s[0], u32s[1]}), in Uint32sToUint128()
354 Uint32sToUint64({u32s[2], u32s[3]})}; in Uint32sToUint128()
446 bits64[0] = Uint32sToUint64({bits32[0], bits32[1]}); in PhiloxRngBit64()
447 bits64[1] = Uint32sToUint64({bits32[2], bits32[3]}); in PhiloxRngBit64()
578 Uint32sToUint64(state_key.second)); in ScramblePhiloxKey()
|