Searched refs:tensorlist_args_indices (Results 1 – 1 of 1) sorted by relevance
104 std::vector<int> tensorlist_args_indices; in cpu_fallback() local126 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()