Home
last modified time | relevance | path

Searched refs:buffer_reuse_key (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/pytorch/torch/_inductor/codegen/
H A Dwrapper.py73 def buffer_reuse_key(node: ir.Buffer) -> ReuseKey: function
386 key = buffer_reuse_key(self.node)
421 state.push(buffer_reuse_key(self.node), self)
1916 assert buffer_reuse_key(input_buffer) == buffer_reuse_key(output_buffer)
/aosp_15_r20/external/pytorch/torch/_inductor/
H A Dscheduler.py377 from .codegen.wrapper import buffer_reuse_key
439 and buffer_reuse_key(input_buf.node)
440 == buffer_reuse_key(buf.node)