Searched refs:numel_for_tensor (Results 1 – 6 of 6) sorted by relevance
44 int64_t numel_for_tensor[depth_to_max_tensors[n - 1]]; member53 int64_t numel_for_tensor[depth_to_max_tensors_scalarlist[n - 1]]; member66 numel_for_tensor[depth_to_max_tensors_scalarlist_of_complex_double[0]]; member78 numel_for_tensor[depth_to_max_tensors_scalarlist_of_complex_double[1]]; member91 int64_t numel_for_tensor[depth_to_max_tensors[n - 1]]; member146 tensorListMeta.numel_for_tensor[loc_tensor_info] = in multi_tensor_apply()190 tensorListMeta.numel_for_tensor[0] = in multi_tensor_apply()191 tensorListMeta.numel_for_tensor[loc_tensor_info - 1]; in multi_tensor_apply()236 tensorListMeta.numel_for_tensor[loc_tensor_info] = in multi_tensor_apply()273 tensorListMeta.numel_for_tensor[0] = in multi_tensor_apply()[all …]
217 auto n = tl.numel_for_tensor[tensor_loc]; in operator ()()240 auto n = tl.numel_for_tensor[tensor_loc]; in operator ()()265 auto n = tl.numel_for_tensor[tensor_loc]; in operator ()()318 auto n = tl.numel_for_tensor[tensor_loc]; in operator ()()371 auto n = tl.numel_for_tensor[tensor_loc]; in operator ()()414 auto n = tl.numel_for_tensor[tensor_loc]; in operator ()()467 auto n = tl.numel_for_tensor[tensor_loc]; in operator ()()490 auto n = tl.numel_for_tensor[tensor_loc]; in operator ()()514 auto n = tl.numel_for_tensor[tensor_loc]; in operator ()()568 auto n = tl.numel_for_tensor[tensor_loc]; in operator ()()[all …]
142 const auto n = tl.numel_for_tensor[tensor_loc] - chunk_idx * chunk_size; in operator ()()
62 auto n = tl.numel_for_tensor[tensor_loc]; in operator ()()252 auto n = tl.numel_for_tensor[tensor_loc]; in operator ()()
87 const auto n = tl.numel_for_tensor[tensor_loc] - chunk_idx * chunk_size; in operator ()()
344 auto n = tl.numel_for_tensor[tensor_loc]; in operator ()()