Searched refs:valid_tensors (Results 1 – 2 of 2) sorted by relevance
49 std::vector<at::Tensor> valid_tensors; in to_eager() local58 valid_tensors.push_back(tensor); in to_eager()63 auto eager_valid_tensors = _to_eager(valid_tensors, device_type); in to_eager()78 std::vector<at::Tensor> valid_tensors; in to_eager() local87 valid_tensors.push_back(*tensor); in to_eager()92 auto eager_valid_tensors = _to_eager(valid_tensors, device_type); in to_eager()
30 std::vector<at::Tensor> valid_tensors; in to_cpu() local39 valid_tensors.push_back(tensors[i].value()); in to_cpu()46 valid_tensors.push_back(tensors[i]); in to_cpu()52 auto cpu_valid_tensors = at::_to_cpu(valid_tensors); in to_cpu()