Home
last modified time | relevance | path

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

/external/pytorch/torch/_inductor/codegen/
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)
/external/pytorch/torch/_inductor/
Dscheduler.py377 from .codegen.wrapper import buffer_reuse_key
439 and buffer_reuse_key(input_buf.node)
440 == buffer_reuse_key(buf.node)