Searched defs:input_cpu (Results 1 – 4 of 4) sorted by relevance
| /aosp_15_r20/external/pytorch/aten/src/ATen/test/ |
| H A D | xnnpack_test.cpp | 67 const auto input_cpu = in TEST() local
137 const auto input_cpu = in TEST() local
228 const auto input_cpu = in TEST() local
|
| H A D | vulkan_quantized_api_test.cpp | 1497 const auto input_cpu = (r1 - r2) * in test_conv2d() local
1605 const auto input_cpu = (r1 - r2) * in TEST_F() local
1708 const auto input_cpu = (r1 - r2) * in TEST_F() local
2504 at::Tensor input_cpu; // input cpu tensor in test_quantized_conv2d() local
3302 const at::Tensor& input_cpu, in _test_quantized_linear()
3433 const at::Tensor& input_cpu, in test_quantized_linear_for_dtypes()
3478 const auto input_cpu = produce_random_tensor(input_shape); in test_quantized_linear() local
|
| H A D | vulkan_api_test.cpp | 910 const auto input_cpu = at::rand({1, 4, 1, 1}, at::device(at::kCPU).dtype(at::kFloat)); in TEST_F() local
939 const auto input_cpu = at::rand({3, 8, 5, 7}, at::device(at::kCPU).dtype(at::kFloat)); in TEST_F() local
969 const auto input_cpu = at::rand({11, 52, 139, 109}, at::device(at::kCPU).dtype(at::kFloat)); in TEST_F() local
1389 const auto input_cpu = at::arange(lengths * channels, at::kFloat).reshape({1, channels, lengths}); in TEST_F() local
1431 const auto input_cpu = at::rand({batch_size, in_channels, lengths}, at::kFloat); in test_conv1d() local
1637 const auto input_cpu = at::randn(input.size(), at::device(at::kCPU).dtype(at::kFloat)); in TEST_F() local
1727 const auto input_cpu = in TEST_F() local
1792 const auto input_cpu = in TEST_F() local
1857 const auto input_cpu = at::rand(input.size(), at::device(at::kCPU).dtype(at::kFloat)); in TEST_F() local
1947 const auto input_cpu = at::randn(input.size(), at::device(at::kCPU).dtype(at::kFloat)); in TEST_F() local
[all …]
|
| /aosp_15_r20/external/pytorch/test/cpp/api/ |
| H A D | rnn.cpp | 579 auto input_cpu = in TEST_F() local
633 auto input_cpu = in TEST_F() local
690 auto input_cpu = in TEST_F() local
|