Searched defs:cluster_op (Results 1 – 5 of 5) sorted by relevance
103 void OutlineCluster(tf_device::ClusterOp cluster_op, SymbolTable* symbol_table, in OutlineCluster()
569 auto cluster_op = in CreateClusterOp() local
1162 if (auto cluster_op = dyn_cast<tf_device::ClusterOp>(op)) { in InferShapeForNonTFDialectOperation() local
117 mlir::Operation *cluster_op = in ReplaceHoistedValues() local
128 auto cluster_op = rewriter.create<chlo::RankSpecializationClusterOp>( in matchAndRewrite() local