MCPcopy Create free account
hub / github.com/apache/tvm / visit_decl_buffer_

Method visit_decl_buffer_

python/tvm/tirx/transform/common.py:131–136  ·  view source on GitHub ↗
(self, op: DeclBuffer)

Source from the content-addressed store, hash-verified

129 return op
130
131 def visit_decl_buffer_(self, op: DeclBuffer):
132 new_buffer = self.mutate_buffer(op.buffer)
133 op = super().visit_decl_buffer_(op)
134 if new_buffer is not None:
135 return DeclBuffer(new_buffer, op.span)
136 return op
137
138 def visit_array_prim_expr_(self, op: list[PrimExpr]):
139 return [self.visit_expr(expr) for expr in op]

Callers

nothing calls this directly

Calls 2

mutate_bufferMethod · 0.95
DeclBufferClass · 0.90

Tested by

no test coverage detected