Searched refs:visit_tensor_exprs (Results 1 – 1 of 1) sorted by relevance
31 def visit_tensor_exprs(self, callback: Callable[["TensorExpression"], None]): member in TensorExpression50 self.visit_tensor_exprs(visit_affine_exprs)59 self.visit_tensor_exprs(visit_tensor_use)68 self.visit_tensor_exprs(visit_index)77 self.visit_tensor_exprs(visit_scalar_def)157 def visit_tensor_exprs(self, callback: Callable[["TensorExpression"], None]): member in TensorFn158 super().visit_tensor_exprs(callback)160 arg.visit_tensor_exprs(callback)195 def visit_tensor_exprs(self, callback: Callable[["TensorExpression"], None]): member in TensorReduceFn197 arg.visit_tensor_exprs(callback)