Searched refs:k_block (Results 1 – 3 of 3) sorted by relevance
| /external/XNNPACK/tools/ |
| D | generate-spmm-test.py | 379 def generate_test_cases(ukernel, mr, nr, k_block, is_pipelined, isa): argument
410 "KBLOCK": k_block,
411 "ADJKBLOCK": 2 * k_block if is_pipelined else k_block,
448 k_block = int(ukernel_spec["k-block"])
455 test_case = generate_test_cases(name, mr, nr, k_block, pipelined, isa)
|
| D | generate-gemm-test.py | 762 k_block, is_pipelined, isa): argument
800 "KBLOCK": k_block,
801 "ADJKBLOCK": 2 * k_block if is_pipelined else k_block,
843 k_block = int(ukernel_spec["k-block"])
852 name, mr, nr, kr, sr, k_block, pipelined, isa)
|
| D | generate-pack-test.py | 155 def generate_test_cases(ukernel, mr, k_block, isa): argument
178 "KBLOCK": k_block,
214 k_block = int(ukernel_spec["k-block"])
220 test_case = generate_test_cases(name, mr, k_block, isa)
|