Searched defs:gpu_tensors (Results 1 – 2 of 2) sorted by relevance
125 std::set<string> gpu_tensors; in TEST_F() local152 std::set<string> gpu_tensors; in TEST_F() local
102 const std::vector<at::Tensor>& gpu_tensors) { in getCpuTensors()