Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/aten/src/ATen/test/
H A Dvulkan_api_test.cpp710 const auto bias_cpu = at::rand({179, 163}, at::device(at::kCPU).dtype(at::kFloat)); in TEST_F() local
730 const auto bias_cpu = at::rand({1000}, at::device(at::kCPU).dtype(at::kFloat)); in TEST_F() local
750 const auto bias_cpu = at::rand({9}, at::device(at::kCPU).dtype(at::kFloat)); in TEST_F() local
771 const auto bias_cpu = at::rand({5, 5}, at::device(at::kCPU).dtype(at::kFloat)); in TEST_F() local
916 const auto bias_cpu = at::rand({4}, at::device(at::kCPU).dtype(at::kFloat)); in TEST_F() local
945 const auto bias_cpu = at::rand({8}, at::device(at::kCPU).dtype(at::kFloat)); in TEST_F() local
975 const auto bias_cpu = at::rand({52}, at::device(at::kCPU).dtype(at::kFloat)); in TEST_F() local
996 at::Tensor bias_cpu, in test_baddbmm()
1023 const auto bias_cpu = in TEST_F() local
1047 const auto bias_cpu = at::cat({bias_cpu_0, bias_cpu_1, bias_cpu_2}, 0); in TEST_F() local
[all …]
H A Dxnnpack_test.cpp141 const auto bias_cpu = at::randn( in TEST() local
232 auto bias_cpu = at::randn( in TEST() local
H A Dvulkan_quantized_api_test.cpp1503 const auto bias_cpu = (r1 - r2) * in test_conv2d() local
1611 const auto bias_cpu = (r1 - r2) * in TEST_F() local
1714 const auto bias_cpu = (r1 - r2) * in TEST_F() local
2522 at::Tensor bias_cpu; // bias cpu tensor in test_quantized_conv2d() local