Home
last modified time | relevance | path

Searched defs:weights_cpu (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/pytorch/aten/src/ATen/test/
H A Dxnnpack_test.cpp139 const auto weights_cpu = in TEST() local
230 auto weights_cpu = in TEST() local
H A Dvulkan_api_test.cpp1390 …const auto weights_cpu = at::ones({channels, 1, kernel_size}, at::device(at::kCPU).dtype(at::kFloa… in TEST_F() local
1432 const auto weights_cpu = at::rand({out_channels, in_group_size, kernel_size}, at::kFloat); in test_conv1d() local
1638 const auto weights_cpu = at::randn(weights.size(), at::device(at::kCPU).dtype(at::kFloat)); in TEST_F() local
1729 const auto weights_cpu = in TEST_F() local
1794 const auto weights_cpu = in TEST_F() local
1858 const auto weights_cpu = at::rand(weights.size(), at::device(at::kCPU).dtype(at::kFloat)); in TEST_F() local
1948 const auto weights_cpu = at::randn(weights.size(), at::device(at::kCPU).dtype(at::kFloat)); in TEST_F() local
2075 const auto weights_cpu = at::randn(weights.size(), at::device(at::kCPU).dtype(at::kFloat)); in TEST_F() local
H A Dvulkan_quantized_api_test.cpp1500 const auto weights_cpu = (r1 - r2) * in test_conv2d() local
1608 const auto weights_cpu = (r1 - r2) * in TEST_F() local
1711 const auto weights_cpu = (r1 - r2) * in TEST_F() local