Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/benchmarks/sparse/
H A Dtriton_ops.py200 tile_n_list = args.tile_n or [None] variable
240 tile_n_list = [meta["TILE_N"] // 2, meta["TILE_N"], meta["TILE_N"] * 2][ variable
244 tile_n_list = [meta["TILE_N"]] variable