def merged_elementwise_kernel(op: FusedElementwise, memory_layout: MemoryLayout) -> List[Kernel]: ops = traverse.listup_operators(op.sub_graph) command_buffer, buffer_injector = generate_elementwise_command_buffer(ops, [_registered_items[op.__class__] for op in ops], memory_layout, dummy2real=op.dummy2real) return elementwise_kernel_base(op, command_buffer, buffer_injector)
def elementwise_kernel(op: Elementwise, memory_layout: MemoryLayout) -> List[Kernel]: builder, buffer_injector = generate_elementwise_command_buffer( [op], [_registered_items[op.__class__]], memory_layout) return elementwise_kernel_base(op, builder, buffer_injector)