Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/aten/src/ATen/native/nested/cuda/
H A DNestedTensorTransformerFunctions.cu1204 int max_shared_bytes; in jagged_dense_dense_elementwise_jagged_output_matches_opt() local
1207 &max_shared_bytes, in jagged_dense_dense_elementwise_jagged_output_matches_opt()
1212 max_shared_bytes = 64 << 10; in jagged_dense_dense_elementwise_jagged_output_matches_opt()
1214 int shared_kb = max_shared_bytes >> 10; in jagged_dense_dense_elementwise_jagged_output_matches_opt()
1268 int max_shared_bytes; in calc_used_shared_bytes() local
1271 &max_shared_bytes, in calc_used_shared_bytes()
1276 max_shared_bytes = 64 << 10; in calc_used_shared_bytes()
1278 int shared_kb = max_shared_bytes >> 10; in calc_used_shared_bytes()