Searched defs:bw_compiler (Results 1 – 6 of 6) sorted by relevance
13 bw_compiler = tvm_compile(target="llvm", tuning_logfile="bw_keops") variable15 bw_compiler = nop variable
17 bw_compiler = tvm_compile(target="llvm", tuning_logfile="bw_keops") variable
72 bw_compiler = functools.partial(extract_graph, graph_cell=bw_graph_cell) variable
1452 def bw_compiler(model: torch.fx.GraphModule, example_inputs: List[torch.Tensor]): function
560 def _test(context_fn, bw_compiler): argument
180 def bw_compiler(gm, _): function