Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/aten/src/ATen/native/nested/cuda/
H A DNestedTensorTransformerFunctions.cu1223 int used_shared_bytes = used_shared_kb << 10; in jagged_dense_dense_elementwise_jagged_output_matches_opt() local
1287 int used_shared_bytes = used_shared_kb << 10; in calc_used_shared_bytes() local
1292 inline void set_max_dynamic_shared_mem_size_for_opt_search_kernel(const int used_shared_bytes) { in set_max_dynamic_shared_mem_size_for_opt_search_kernel()
1355 int used_shared_bytes = calc_used_shared_bytes(y_reshaped.get_device()); in jagged_dense_elementwise_jagged_output_opt_() local