Home
last modified time | relevance | path

Searched defs:TensorListMetadata (Results 1 – 1 of 1) sorted by relevance

/aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/
H A DMultiTensorApply.cuh42 struct TensorListMetadata { struct
43 const void* addresses[n][depth_to_max_tensors[n - 1]];
44 int64_t numel_for_tensor[depth_to_max_tensors[n - 1]];
45 unsigned char block_to_tensor[depth_to_max_blocks[n - 1]];
46 int block_to_chunk[depth_to_max_blocks[n - 1]];
47 int start_tensor_this_launch;