| /external/tensorflow/tensorflow/core/kernels/ |
| D | stateful_random_ops_cpu_gpu.h | 30 PHILOX_DEVICE_INLINE void WritePhiloxRandomToMem(PhiloxRandom const& philox, in WritePhiloxRandomToMem()
37 PHILOX_DEVICE_INLINE PhiloxRandom SkipPhiloxRandom(PhiloxRandom const& philox, in SkipPhiloxRandom()
47 PHILOX_DEVICE_INLINE void UpdateMemWithPhiloxRandom(PhiloxRandom const& philox, in UpdateMemWithPhiloxRandom()
57 auto philox = PhiloxRandom(counter, PhiloxRandom::Key() /*dummy*/); in UpdateCounterMemWithPhiloxRandom() local
|
| D | range_sampler_test.cc | 46 random::PhiloxRandom philox(123, 17); in CheckHistogram() local
270 random::PhiloxRandom philox(123, 17); in TEST_F() local
313 random::PhiloxRandom philox(123, 17); in TEST_F() local
|
| D | gather_nd_op_test.cc | 119 random::PhiloxRandom philox(301, 17); in GatherNd() local
|
| D | random_binomial_op.cc | 424 auto philox = GetPhiloxRandomFromMem(var_data); in Compute() local
513 auto philox = random::PhiloxRandom(counter, key); in Compute() local
|
| D | dynamic_partition_op_test.cc | 180 random::PhiloxRandom philox(301, 17); in DynamicPartition() local
|
| D | gather_op_test.cc | 204 random::PhiloxRandom philox(301, 17); in Gather() local
|
| D | stateful_random_ops_gpu.cu.cc | 46 auto philox = reinterpret_cast<PhiloxRandom*>(philox_raw); in FillKernel() local
|
| D | scatter_nd_op_test.cc | 266 random::PhiloxRandom philox(301, 17); in BM_ScatterNdHelper() local
|
| /external/tensorflow/tensorflow/core/lib/io/ |
| D | table_test.cc | 435 random::PhiloxRandom philox(testing::RandomSeed() + 1, 17); in TEST_F() local
464 random::PhiloxRandom philox(testing::RandomSeed() + 1, 17); in TEST_F() local
474 random::PhiloxRandom philox(testing::RandomSeed() + 2, 17); in TEST_F() local
484 random::PhiloxRandom philox(testing::RandomSeed() + 3, 17); in TEST_F() local
496 random::PhiloxRandom philox(testing::RandomSeed() + 3, 17); in TEST_F() local
509 random::PhiloxRandom philox(testing::RandomSeed() + 4, 17); in TEST_F() local
519 random::PhiloxRandom philox(testing::RandomSeed() + 5, 17); in TEST_F() local
591 random::PhiloxRandom philox(301, 17); in TEST() local
614 random::PhiloxRandom philox(301, 17); in TEST() local
|
| D | recordio_test.cc | 235 random::PhiloxRandom philox(301, 17); in TEST_F() local
242 random::PhiloxRandom philox(301, 17); in TEST_F() local
|
| /external/tensorflow/tensorflow/core/lib/random/ |
| D | weighted_picker_test.cc | 37 PhiloxRandom philox(testing::RandomSeed(), 17); in TEST() local
93 PhiloxRandom philox(testing::RandomSeed() + 1, 17); in TEST() local
108 PhiloxRandom philox(testing::RandomSeed() + 10, 17); in TEST() local
260 PhiloxRandom philox(301, 17); in BM_Pick() local
|
| D | simple_philox_test.cc | 30 PhiloxRandom philox(7, 7); in TEST() local
96 PhiloxRandom philox(17, 17); in TEST() local
115 PhiloxRandom philox(17, 17); in TEST() local
|
| D | distribution_sampler_test.cc | 39 PhiloxRandom philox(testing::RandomSeed(), 17); in TestWeights() local
87 PhiloxRandom philox(173, 371); in BM_DistributionSampler() local
|
| /external/tensorflow/tensorflow/core/lib/strings/ |
| D | ordered_code_test.cc | 148 random::PhiloxRandom philox(301, 17); in TestNumbers() local
218 random::PhiloxRandom philox(301, 17); in TEST() local
330 random::PhiloxRandom philox(301, 17); in TEST() local
382 random::PhiloxRandom philox(301, 17); in BM_WriteNum() local
399 random::PhiloxRandom philox(301, 17); in BM_ReadNum() local
436 random::PhiloxRandom philox(301, 17); in TEST() local
1217 random::PhiloxRandom philox(301, 17); in BM_WriteString() local
1233 random::PhiloxRandom philox(301, 17); in BM_ReadString() local
|
| /external/pytorch/aten/src/ATen/cuda/detail/ |
| D | UnpackRaw.cuh | 5 namespace at::cuda::philox { namespace
|
| /external/pytorch/aten/src/ATen/native/transformers/cuda/flash_attn/ |
| D | philox.cuh | 33 __forceinline__ __device__ uint4 philox(unsigned long long seed, in philox() function
|
| /external/tensorflow/tensorflow/core/graph/ |
| D | tensor_id_test.cc | 45 random::PhiloxRandom philox(301, 17); in BM_ParseTensorName() local
|
| D | optimizer_cse_test.cc | 356 random::PhiloxRandom philox(301, 17); in BM_CSE() local
|
| /external/tensorflow/tensorflow/core/lib/core/ |
| D | bitmap_test.cc | 98 random::PhiloxRandom philox(301, 17); in TEST() local
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | mkl_tfconversion_pass_test.cc | 277 random::PhiloxRandom philox(301, 17); in BM_RunMklToTfConversionPass() local
|
| D | single_threaded_executor_test.cc | 296 random::PhiloxRandom philox(0, 17); in BuildTree() local
364 random::PhiloxRandom philox(1729, 17); in BM_executor() local
|
| D | executor_test.cc | 231 random::PhiloxRandom philox(testing::RandomSeed(), 17); in BuildTree() local
446 random::PhiloxRandom philox(1729, 17); in BM_executor() local
|
| /external/tensorflow/tensorflow/core/lib/gtl/ |
| D | top_n_test.cc | 89 PhiloxRandom philox(1, 1); in TEST() local
|
| /external/tensorflow/tensorflow/core/common_runtime/gpu/ |
| D | gpu_bfc_allocator_test.cc | 137 random::PhiloxRandom philox(123, 17); in TEST_P() local
266 random::PhiloxRandom philox(123, 17); in TEST_P() local
|
| /external/tensorflow/tensorflow/core/util/sparse/ |
| D | sparse_tensor_test.cc | 753 random::PhiloxRandom philox(301, 17); in BM_SparseReorderFloat() local
791 random::PhiloxRandom philox(301, 17); in BM_SparseReorderString() local
|