variables_layout: MemoryLayout,
metabuffer_injector: MetaBufferInjector = None) -> List[Kernel]:
x = variables_layout[util.get_input(op, "x")]
b = constants_layout[util.get_input(op, "b")]
y = variables_layout[util.get_output(op, "y")]
if metabuffer_injector is None:
metabuffer_injector = MetaBufferInjector()
After Change
variables_layout: MemoryLayout,
metabuffer_injector: MetaBufferInjector = None) -> List[Kernel]:
x = variables_layout[op.inputs["x"].name]
b = constants_layout[op.inputs["b"].name]
y = variables_layout[op.outputs["y"].name]
if metabuffer_injector is None:
metabuffer_injector = MetaBufferInjector()