Searched defs:m1_cpu (Results 1 – 1 of 1) sorted by relevance
711 const auto m1_cpu = at::rand({179, 67}, at::device(at::kCPU).dtype(at::kFloat)); in TEST_F() local731 const auto m1_cpu = at::rand({1, 1280}, at::device(at::kCPU).dtype(at::kFloat)); in TEST_F() local751 const auto m1_cpu = at::rand({17, 6}, at::device(at::kCPU).dtype(at::kFloat)); in TEST_F() local772 const auto m1_cpu = at::rand({17, 6}, at::device(at::kCPU).dtype(at::kFloat)); in TEST_F() local997 at::Tensor m1_cpu, in test_baddbmm()1025 const auto m1_cpu = in TEST_F() local1049 const auto m1_cpu = in TEST_F() local1063 const auto m1_cpu = in TEST_F() local1078 const auto m1_cpu = in TEST_F() local1092 const auto m1_cpu = in TEST_F() local[all …]