Searched defs:cpu_tensors (Results 1 – 3 of 3) sorted by relevance
29 std::vector<T> cpu_tensors(num); in to_cpu() local142 auto cpu_tensors = to_cpu(tensor_args); in cpu_fallback() local170 const auto& cpu_tensors = tensorlist_cpu_args[i].toTensorVector(); in cpu_fallback() local184 const auto& cpu_tensors = optional_tensorlist_cpu_args[i].toOptionalTensorList(); in cpu_fallback() local328 const auto& cpu_tensors = returns[idx].toTensorList().vec(); in cpu_fallback() local
534 std::vector<Tensor> cpu_tensors; in _to_cpu() local
112 std::set<string> cpu_tensors; in TEST_F() local