Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/
H A DMultiTensorApply.cuh51 struct TensorListScalarListMetadata { struct
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]];
55 unsigned char block_to_tensor[depth_to_max_blocks[n - 1]];
56 int block_to_chunk[depth_to_max_blocks[n - 1]];
62 struct TensorListScalarListMetadata<c10::complex<double>, 1> { struct
63 const void* addresses[1]
66 numel_for_tensor[depth_to_max_tensors_scalarlist_of_complex_double[0]];
68 scalar_vals[depth_to_max_tensors_scalarlist_of_complex_double[0]];
[all …]