Searched refs:nested_tensorlist (Results 1 – 2 of 2) sorted by relevance
286 const nested_optional_tensorvec_t& nested_tensorlist, in _group_tensors_by_first_tensors_device_and_dtype() argument290 TORCH_CHECK(!nested_tensorlist.empty()); in _group_tensors_by_first_tensors_device_and_dtype()291 TORCH_CHECK(!nested_tensorlist[0].empty()); in _group_tensors_by_first_tensors_device_and_dtype()292 const auto num_lists = nested_tensorlist.size(); in _group_tensors_by_first_tensors_device_and_dtype()293 const auto num_tensors = nested_tensorlist[0].size(); in _group_tensors_by_first_tensors_device_and_dtype()296 nested_tensorlist.cbegin(), in _group_tensors_by_first_tensors_device_and_dtype()297 nested_tensorlist.cend(), in _group_tensors_by_first_tensors_device_and_dtype()307 const auto t = nested_tensorlist[0][tensor_index]; in _group_tensors_by_first_tensors_device_and_dtype()318 nested_tensorlist.cbegin(), in _group_tensors_by_first_tensors_device_and_dtype()319 nested_tensorlist.cend(), in _group_tensors_by_first_tensors_device_and_dtype()[all …]
2272 nested_tensorlist, in initModule()2275 nested_tensorlist, with_indices); in initModule()