Home
last modified time | relevance | path

Searched refs:compute_target_device (Results 1 – 2 of 2) sorted by relevance

/external/pytorch/torch/csrc/lazy/ts_backend/
Dts_eager_fallback.cpp115 std::optional<c10::Device> compute_target_device( in compute_target_device() function
326 std::optional<c10::Device> tgt_device = compute_target_device( in ts_eager_fallback()
/external/pytorch/aten/src/ATen/native/
DCPUFallback.cpp61 static std::optional<c10::Device> compute_target_device(std::vector<at::Tensor>& t_args, const std:… in compute_target_device() function
218 tgt_device = compute_target_device(tensor_args, tensorlist_args); in cpu_fallback()