Searched refs:parallel_cat (Results 1 – 1 of 1) sorted by relevance
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/ |
| H A D | Shape.cu | 288 void parallel_cat(const Tensor &out, const MaterializedITensorListRef& inputs, int64_t dimension, in parallel_cat() function
500 … parallel_cat<dtype, CAT_ARRAY_BATCH_SIZE, 1>(result, materialized, dim, nDims, memory_format); in TORCH_IMPL_FUNC()
505 … parallel_cat<dtype, CAT_ARRAY_BATCH_SIZE, 1>(result, materialized, dim, nDims, memory_format); in TORCH_IMPL_FUNC()
518 …parallel_cat<dtype, CAT_ARRAY_BATCH_SIZE/2, CAT_ARRAY_BATCH_SIZE/2>(result, materialized, dim, nDi… in TORCH_IMPL_FUNC()
523 …parallel_cat<dtype, CAT_ARRAY_BATCH_SIZE/2, CAT_ARRAY_BATCH_SIZE/2>(result, materialized, dim, nDi… in TORCH_IMPL_FUNC()
|