Searched refs:Uint64ToUint32s (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | random_ops_util.h | 28 PHILOX_DEVICE_INLINE void Uint64ToUint32s(uint64 input, uint32* output1, in Uint64ToUint32s() function
43 Uint64ToUint32s(ptr[0], &counter[0], &counter[1]); in GetCounterFromMem()
44 Uint64ToUint32s(ptr[1], &counter[2], &counter[3]); in GetCounterFromMem()
56 Uint64ToUint32s(ptr[0], &key[0], &key[1]); in GetKeyFromMem()
|
| /external/tensorflow/tensorflow/compiler/xla/client/lib/ |
| D | prng.cc | 122 std::array<XlaOp, 2> Uint64ToUint32s(XlaOp u64) { in Uint64ToUint32s() function
158 return std::make_pair(Uint64ToUint32s(input_u64), new_state); in GetThreeFryInputsAndUpdatedState()
248 ThreeFry2x32State outputs = ThreeFry2x32(inputs, Uint64ToUint32s(key)); in ThreeFryRngBit32()
259 ThreeFry2x32State outputs = ThreeFry2x32(inputs, Uint64ToUint32s(key)); in ThreeFryRngBit64()
358 std::array<XlaOp, 2> u128_low_32s = Uint64ToUint32s(u128[0]); in Uint128ToUint32s()
359 std::array<XlaOp, 2> u128_high_32s = Uint64ToUint32s(u128[1]); in Uint128ToUint32s()
412 Philox4x32Key key = Uint64ToUint32s(op_key); in PhiloxRngBit32()
439 Philox4x32Key key = Uint64ToUint32s(op_key); in PhiloxRngBit64()
575 Philox4x32Key pkey = Uint64ToUint32s(key); in ScramblePhiloxKey()
|