Searched defs:c0_cpu (Results 1 – 1 of 1) sorted by relevance
7929 …const auto c0_cpu = at::rand({num_layers, N, hidden_size}, at::device(at::kCPU).dtype(at::kFloat)); in TEST_F() local8009 …const auto c0_cpu = at::rand({num_layers, N, hidden_size}, at::device(at::kCPU).dtype(at::kFloat)); in TEST_F() local8085 …const auto c0_cpu = at::rand({num_layers, N, hidden_size}, at::device(at::kCPU).dtype(at::kFloat)); in TEST_F() local