Home
last modified time | relevance | path

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

/external/pytorch/aten/src/ATen/native/
DCPUFallback.cpp104 std::vector<int> tensorlist_args_indices; in cpu_fallback() local
126 tensorlist_args_indices.push_back(idx); in cpu_fallback()
166 for (const auto i : c10::irange(tensorlist_args_indices.size())) { in cpu_fallback()
167 auto tensorlist_idx = tensorlist_args_indices[i]; in cpu_fallback()
254 for (const auto i : c10::irange(tensorlist_args_indices.size())) { in cpu_fallback()
255 auto input_tensor_idx = tensorlist_args_indices[i]; in cpu_fallback()