Searched refs:compute_target_device (Results 1 – 2 of 2) sorted by relevance
115 std::optional<c10::Device> compute_target_device( in compute_target_device() function326 std::optional<c10::Device> tgt_device = compute_target_device( in ts_eager_fallback()
61 static std::optional<c10::Device> compute_target_device(std::vector<at::Tensor>& t_args, const std:… in compute_target_device() function218 tgt_device = compute_target_device(tensor_args, tensorlist_args); in cpu_fallback()