HomeSort by relevance Sort by last modified time
    Searched defs:philox (Results 1 - 3 of 3) sorted by null

  /external/tensorflow/tensorflow/core/kernels/
stateful_random_ops_gpu.cu.cc 35 // Threads in this block share `philox`. Thread 0 is responsible for
38 auto philox = reinterpret_cast<PhiloxRandom*>(philox_raw); local
40 *philox = GetPhiloxRandomFromMem(state_data);
45 .Run(*philox, output_data, output_size, dist);
50 UpdateMemWithPhiloxRandom(*philox, output_size, state_data);
stateful_random_ops.cc 34 auto philox = GetPhiloxRandomFromMem(state_data + alg_tag_skip); local
35 UpdateMemWithPhiloxRandom(philox, output_size, state_data + alg_tag_skip);
39 ctx, device, philox, output_data, output_size, Distribution());
84 "For the Philox algorithm, the size of state"
185 // philox, so 'NormalDistribution<PhiloxRandom, ...>' is fine for now.
sparse_matmul_op_test.cc 28 random::PhiloxRandom philox(1, 1);
29 random::SimplePhilox rnd(&philox); member in namespace:tensorflow

Completed in 1145 milliseconds