Searched refs:cluster_result (Results 1 – 2 of 2) sorted by relevance
88 %cluster_result = "tf_device.cluster"() ({92 func.return %cluster_result : tensor<2xi32>
662 for (auto cluster_result : parallel_execute.execute_outputs()) { in CheckTPUPartitionedInputAndOutputAreValid() local664 llvm::make_early_inc_range(cluster_result.getUsers())) { in CheckTPUPartitionedInputAndOutputAreValid()