Searched refs:collect_tensor_uses (Results 1 – 2 of 2) sorted by relevance
140 read_use.collect_tensor_uses(collected_tensor_uses)232 read_expr.collect_tensor_uses(read_uses)
52 def collect_tensor_uses(self, uses: Set["TensorUse"]): member in TensorExpression