Searched refs:test_random_from_to (Results 1 – 3 of 3) sorted by relevance
| /aosp_15_r20/external/pytorch/aten/src/ATen/test/ |
| H A D | cpu_rng_test.cpp | 170 test_random_from_to<TestCPUGenerator, torch::kBool, bool>(device); in TEST_F()
171 test_random_from_to<TestCPUGenerator, torch::kUInt8, uint8_t>(device); in TEST_F()
172 test_random_from_to<TestCPUGenerator, torch::kInt8, int8_t>(device); in TEST_F()
173 test_random_from_to<TestCPUGenerator, torch::kInt16, int16_t>(device); in TEST_F()
174 test_random_from_to<TestCPUGenerator, torch::kInt32, int32_t>(device); in TEST_F()
175 test_random_from_to<TestCPUGenerator, torch::kInt64, int64_t>(device); in TEST_F()
176 test_random_from_to<TestCPUGenerator, torch::kFloat32, float>(device); in TEST_F()
177 test_random_from_to<TestCPUGenerator, torch::kFloat64, double>(device); in TEST_F()
|
| H A D | rng_test.h | 63 void test_random_from_to(const at::Device& device) { in test_random_from_to() function
|
| /aosp_15_r20/external/pytorch/test/ |
| H A D | test_tensor_creation_ops.py | 1670 def test_random_from_to(self, device, dtype): member in TestTensorCreation
|