Searched refs:depth_to_max_tensors (Results 1 – 1 of 1) sorted by relevance
19 static constexpr int depth_to_max_tensors[5] = {110, 64, 48, 36, 30}; variable43 const void* addresses[n][depth_to_max_tensors[n - 1]];44 int64_t numel_for_tensor[depth_to_max_tensors[n - 1]];90 const void* addresses[n][depth_to_max_tensors[n - 1]];91 int64_t numel_for_tensor[depth_to_max_tensors[n - 1]];253 (loc_tensor_info == depth_to_max_tensors[depth - 1] && in multi_tensor_apply()336 (loc_tensor_info == depth_to_max_tensors[depth - 1] && in multi_tensor_apply_for_fused_optimizer()