Home
last modified time | relevance | path

Searched refs:GetPhiloxRandomFromMem (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dstateful_random_ops_gpu.cu.cc46 *philox = GetPhiloxRandomFromMem(state_data); in FillKernel()
101 auto philox = GetPhiloxRandomFromMem(state_data); in SkipKernel()
Dstateful_random_ops_cpu_gpu.h44 GetPhiloxRandomFromMem(StateElementType const* ptr) { in GetPhiloxRandomFromMem() function
Dstateful_random_ops.cc40 auto philox = GetPhiloxRandomFromMem(state_data + alg_tag_skip); in operator ()()
234 auto philox = GetPhiloxRandomFromMem(state_data); in operator ()()
Drandom_binomial_op.cc422 auto philox = GetPhiloxRandomFromMem(var_data); in Compute()