Searched defs:at_tensor_b (Results 1 – 2 of 2) sorted by relevance
147 auto at_tensor_b = at::ones({5, 5}, at::dtype(at::kFloat)); in TEST() local187 auto at_tensor_b = at_tensor_a.view({5, 5}); in TEST() local215 auto at_tensor_b = at_tensor_a.view({25}); in TEST() local
83 auto at_tensor_b = at::ones({5, 5}, at::dtype(at::kFloat).device(at::kCUDA)); in TEST() local124 auto at_tensor_b = at_tensor_a.view({25}); in TEST() local