Home
last modified time | relevance | path

Searched refs:depth_to_max_tensors_scalarlist (Results 1 – 1 of 1) sorted by relevance

/aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/
H A DMultiTensorApply.cuh21 static constexpr int depth_to_max_tensors_scalarlist[5] = {96, 64, 48, 36, 30}; variable
52 const void* addresses[n][depth_to_max_tensors_scalarlist[n - 1]];
53 int64_t numel_for_tensor[depth_to_max_tensors_scalarlist[n - 1]];
54 scalar_vals_t scalar_vals[depth_to_max_tensors_scalarlist[n - 1]];
92 const void* state_steps_addresses[depth_to_max_tensors_scalarlist[n - 1]];
170 (loc_tensor_info == depth_to_max_tensors_scalarlist[depth - 1] && in multi_tensor_apply()