Searched defs:weight_cpu (Results 1 – 2 of 2) sorted by relevance
913 const auto weight_cpu = at::rand({4}, at::device(at::kCPU).dtype(at::kFloat)); in TEST_F() local942 const auto weight_cpu = at::rand({8}, at::device(at::kCPU).dtype(at::kFloat)); in TEST_F() local972 const auto weight_cpu = at::rand({52}, at::device(at::kCPU).dtype(at::kFloat)); in TEST_F() local2741 const auto weight_cpu = in test_packed_layer_norm() local2849 const auto weight_cpu = in test_layer_norm() local2901 const auto weight_cpu = in test_native_layer_norm() local
2517 at::Tensor weight_cpu; // weight cpu tensor in test_quantized_conv2d() local