Searched refs:tensor_args_indices (Results 1 – 2 of 2) sorted by relevance
217 std::vector<int> tensor_args_indices; in ts_eager_fallback() local228 tensor_args_indices.push_back(idx); in ts_eager_fallback()249 for (const auto i : c10::irange(tensor_args_indices.size())) { in ts_eager_fallback()250 auto idx = tensor_args_indices[i]; in ts_eager_fallback()260 for (const auto i : c10::irange(tensor_args_indices.size())) { in ts_eager_fallback()261 auto tensor_idx = tensor_args_indices[i]; in ts_eager_fallback()302 for (const auto i : c10::irange(tensor_args_indices.size())) { in ts_eager_fallback()303 auto input_tensor_idx = tensor_args_indices[i]; in ts_eager_fallback()
101 std::vector<int> tensor_args_indices; in cpu_fallback() local120 tensor_args_indices.push_back(idx); in cpu_fallback()144 for (const auto i : c10::irange(tensor_args_indices.size())) { in cpu_fallback()145 auto idx = tensor_args_indices[i]; in cpu_fallback()155 for (const auto i : c10::irange(tensor_args_indices.size())) { in cpu_fallback()156 auto tensor_idx = tensor_args_indices[i]; in cpu_fallback()231 for (const auto i : c10::irange(tensor_args_indices.size())) { in cpu_fallback()232 auto input_tensor_idx = tensor_args_indices[i]; in cpu_fallback()