| /external/tensorflow/tensorflow/core/kernels/ |
| D | stateful_random_ops_cpu_gpu.h | 31 PHILOX_DEVICE_INLINE void WritePhiloxRandomToMem(PhiloxRandom const& philox, in WritePhiloxRandomToMem() argument
34 WriteCounterToMem(philox.counter(), ptr_); in WritePhiloxRandomToMem()
35 WriteKeyToMem(philox.key(), ptr_ + 2); in WritePhiloxRandomToMem()
38 PHILOX_DEVICE_INLINE PhiloxRandom SkipPhiloxRandom(PhiloxRandom const& philox, in SkipPhiloxRandom() argument
40 auto new_philox = philox; in SkipPhiloxRandom()
48 PHILOX_DEVICE_INLINE void UpdateMemWithPhiloxRandom(PhiloxRandom const& philox, in UpdateMemWithPhiloxRandom() argument
51 auto new_philox = SkipPhiloxRandom(philox, output_size); in UpdateMemWithPhiloxRandom()
58 auto philox = PhiloxRandom(counter, PhiloxRandom::Key() /*dummy*/); in UpdateCounterMemWithPhiloxRandom() local
59 auto new_philox = SkipPhiloxRandom(philox, output_size); in UpdateCounterMemWithPhiloxRandom()
|
| D | stateful_random_ops_gpu.cu.cc | 46 auto philox = reinterpret_cast<PhiloxRandom*>(philox_raw); in FillKernel() local
48 *philox = GetPhiloxRandomFromMem(state_data); in FillKernel()
53 .Run(/*key=*/nullptr, /*counter=*/nullptr, *philox, output_data, in FillKernel()
59 UpdateMemWithPhiloxRandom(*philox, output_size, state_data); in FillKernel()
|
| D | range_sampler_test.cc | 46 random::PhiloxRandom philox(123, 17); in CheckHistogram() local
47 random::SimplePhilox rnd(&philox); in CheckHistogram()
270 random::PhiloxRandom philox(123, 17); in TEST_F() local
271 random::SimplePhilox rnd(&philox); in TEST_F()
313 random::PhiloxRandom philox(123, 17); in TEST_F() local
314 random::SimplePhilox rnd(&philox); in TEST_F()
|
| D | gather_nd_op_test.cc | 119 random::PhiloxRandom philox(301, 17); in GatherNd() local
120 random::SimplePhilox rnd(&philox); in GatherNd()
|
| D | random_binomial_op.cc | 424 auto philox = GetPhiloxRandomFromMem(var_data); in Compute() local
426 philox, num_batches * 2 * 100 * (samples_per_batch + 3) / 4, var_data); in Compute()
431 counts_tensor.flat<T>(), probs_tensor.flat<T>(), philox, in Compute()
513 auto philox = random::PhiloxRandom(counter, key); in Compute() local
517 counts_tensor.flat<T>(), probs_tensor.flat<T>(), philox, in Compute()
|
| D | dynamic_partition_op_test.cc | 180 random::PhiloxRandom philox(301, 17); in DynamicPartition() local
181 random::SimplePhilox rnd(&philox); in DynamicPartition()
|
| D | gather_op_test.cc | 204 random::PhiloxRandom philox(301, 17); in Gather() local
205 random::SimplePhilox rnd(&philox); in Gather()
|
| D | scatter_nd_op_test.cc | 266 random::PhiloxRandom philox(301, 17); in BM_ScatterNdHelper() local
267 random::SimplePhilox rnd(&philox); in BM_ScatterNdHelper()
|
| D | stateful_random_ops.cc | 44 auto philox = GetPhiloxRandomFromMem(state_data + alg_tag_skip); in operator ()() local
45 UpdateMemWithPhiloxRandom(philox, output_size, state_data + alg_tag_skip); in operator ()()
49 ctx, device, /*key=*/nullptr, /*counter=*/nullptr, philox, output_data, in operator ()()
|
| /external/tensorflow/tensorflow/core/lib/random/ |
| D | simple_philox_test.cc | 30 PhiloxRandom philox(7, 7); in TEST() local
31 SimplePhilox gen(&philox); in TEST()
96 PhiloxRandom philox(17, 17); in TEST() local
97 SimplePhilox gen(&philox); in TEST()
115 PhiloxRandom philox(17, 17); in TEST() local
116 SimplePhilox gen(&philox); in TEST()
|
| D | weighted_picker_test.cc | 37 PhiloxRandom philox(testing::RandomSeed(), 17); in TEST() local
38 SimplePhilox rnd(&philox); in TEST()
93 PhiloxRandom philox(testing::RandomSeed() + 1, 17); in TEST() local
94 SimplePhilox rnd(&philox); in TEST()
108 PhiloxRandom philox(testing::RandomSeed() + 10, 17); in TEST() local
109 SimplePhilox rnd(&philox); in TEST()
260 PhiloxRandom philox(301, 17); in BM_Pick() local
261 SimplePhilox rnd(&philox); in BM_Pick()
|
| D | distribution_sampler_test.cc | 39 PhiloxRandom philox(testing::RandomSeed(), 17); in TestWeights() local
40 SimplePhilox random(&philox); in TestWeights()
87 PhiloxRandom philox(173, 371); in BM_DistributionSampler() local
88 SimplePhilox rand(&philox); in BM_DistributionSampler()
|
| D | BUILD | 30 name = "philox",
93 ":philox",
|
| /external/tensorflow/tensorflow/core/lib/io/ |
| D | table_test.cc | 435 random::PhiloxRandom philox(testing::RandomSeed() + 1, 17); in TEST_F() local
436 random::SimplePhilox rnd(&philox); in TEST_F()
464 random::PhiloxRandom philox(testing::RandomSeed() + 1, 17); in TEST_F() local
465 random::SimplePhilox rnd(&philox); in TEST_F()
474 random::PhiloxRandom philox(testing::RandomSeed() + 2, 17); in TEST_F() local
475 random::SimplePhilox rnd(&philox); in TEST_F()
484 random::PhiloxRandom philox(testing::RandomSeed() + 3, 17); in TEST_F() local
485 random::SimplePhilox rnd(&philox); in TEST_F()
496 random::PhiloxRandom philox(testing::RandomSeed() + 3, 17); in TEST_F() local
497 random::SimplePhilox rnd(&philox); in TEST_F()
[all …]
|
| D | recordio_test.cc | 235 random::PhiloxRandom philox(301, 17); in TEST_F() local
236 random::SimplePhilox rnd(&philox); in TEST_F()
242 random::PhiloxRandom philox(301, 17); in TEST_F() local
243 random::SimplePhilox rnd(&philox); in TEST_F()
|
| /external/tensorflow/tensorflow/core/lib/strings/ |
| D | ordered_code_test.cc | 148 random::PhiloxRandom philox(301, 17); in TestNumbers() local
149 random::SimplePhilox rnd(&philox); in TestNumbers()
218 random::PhiloxRandom philox(301, 17); in TEST() local
219 random::SimplePhilox rnd(&philox); in TEST()
330 random::PhiloxRandom philox(301, 17); in TEST() local
331 random::SimplePhilox rnd(&philox); in TEST()
382 random::PhiloxRandom philox(301, 17); in BM_WriteNum() local
383 random::SimplePhilox rnd(&philox); in BM_WriteNum()
399 random::PhiloxRandom philox(301, 17); in BM_ReadNum() local
400 random::SimplePhilox rnd(&philox); in BM_ReadNum()
[all …]
|
| /external/tensorflow/tensorflow/core/graph/ |
| D | benchmark_testlib.h | 89 random::PhiloxRandom philox(301, 17); in CreateGraphDef()
90 random::SimplePhilox rnd(&philox); in CreateGraphDef()
115 random::PhiloxRandom philox(0x12345); in CreateRandomGraph()
116 random::SimplePhilox rnd(&philox); in CreateRandomGraph()
|
| D | tensor_id_test.cc | 44 random::PhiloxRandom philox(301, 17); in BM_ParseTensorName() local
45 random::SimplePhilox rnd(&philox); in BM_ParseTensorName()
|
| D | optimizer_cse_test.cc | 356 random::PhiloxRandom philox(301, 17); in BM_CSE() local
357 random::SimplePhilox rnd(&philox); in BM_CSE()
|
| /external/tensorflow/tensorflow/core/kernels/boosted_trees/quantiles/ |
| D | weighted_quantiles_stream_test.cc | 81 random::PhiloxRandom philox(13 + worker_id); in GenerateRandUniformFixedWeightsSummary() local
82 random::SimplePhilox rand(&philox); in GenerateRandUniformFixedWeightsSummary()
95 random::PhiloxRandom philox(13 + worker_id); in GenerateRandUniformRandWeightsSummary() local
96 random::SimplePhilox rand(&philox); in GenerateRandUniformRandWeightsSummary()
|
| D | weighted_quantiles_summary_test.cc | 138 random::PhiloxRandom philox(13); in TEST_F() local
139 random::SimplePhilox rand(&philox); in TEST_F()
|
| /external/tensorflow/tensorflow/core/lib/core/ |
| D | bitmap_test.cc | 98 random::PhiloxRandom philox(301, 17); in TEST() local
99 random::SimplePhilox rnd(&philox); in TEST()
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | mkl_tfconversion_pass_test.cc | 277 random::PhiloxRandom philox(301, 17); in BM_RunMklToTfConversionPass() local
278 random::SimplePhilox rnd(&philox); in BM_RunMklToTfConversionPass()
|
| /external/tensorflow/tensorflow/core/kernels/data/ |
| D | single_threaded_executor_test.cc | 213 random::PhiloxRandom philox(0, 17); in BuildTree() local
214 random::SimplePhilox rnd(&philox); in BuildTree()
281 random::PhiloxRandom philox(1729, 17); in BM_executor() local
282 random::SimplePhilox rand(&philox); in BM_executor()
|
| /external/tensorflow/tensorflow/core/lib/gtl/ |
| D | top_n_test.cc | 89 PhiloxRandom philox(1, 1); in TEST() local
90 SimplePhilox random(&philox); in TEST()
|