Searched defs:running_mean_cpu (Results 1 – 1 of 1) sorted by relevance
919 const auto running_mean_cpu = at::rand({4}, at::device(at::kCPU).dtype(at::kFloat)); in TEST_F() local948 const auto running_mean_cpu = at::rand({8}, at::device(at::kCPU).dtype(at::kFloat)); in TEST_F() local978 const auto running_mean_cpu = at::rand({52}, at::device(at::kCPU).dtype(at::kFloat)); in TEST_F() local