Searched defs:m2_cpu (Results 1 – 1 of 1) sorted by relevance
712 const auto m2_cpu = at::rand({67, 163}, at::device(at::kCPU).dtype(at::kFloat)); in TEST_F() local732 const auto m2_cpu = at::rand({1280, 1000}, at::device(at::kCPU).dtype(at::kFloat)); in TEST_F() local752 const auto m2_cpu = at::rand({6, 9}, at::device(at::kCPU).dtype(at::kFloat)); in TEST_F() local773 const auto m2_cpu = at::rand({6, 9}, at::device(at::kCPU).dtype(at::kFloat)); in TEST_F() local998 at::Tensor m2_cpu, in test_baddbmm()1027 const auto m2_cpu = in TEST_F() local1051 const auto m2_cpu = in TEST_F() local1065 const auto m2_cpu = in TEST_F() local1080 const auto m2_cpu = in TEST_F() local1094 const auto m2_cpu = in TEST_F() local[all …]