Searched defs:cluster_op (Results 1 – 6 of 6) sorted by relevance
34 void InlineDeviceOp(tf_device::ClusterOp cluster_op) { in InlineDeviceOp()63 [&](tf_device::ClusterOp cluster_op) { ops.push_back(cluster_op); }); in runOnOperation()65 for (auto cluster_op : ops) { in runOnOperation() local
104 void OutlineCluster(tf_device::ClusterOp cluster_op, SymbolTable* symbol_table, in OutlineCluster()
570 auto cluster_op = in CreateClusterOp() local
1947 if (auto cluster_op = dyn_cast<tf_device::ClusterOp>(op)) { in InferShapeForNonTFDialectOperation() local
746 if (auto cluster_op = mlir::dyn_cast<mlir::tf_device::ClusterOp>(op)) in next() local
113 mlir::Operation *cluster_op = in ReplaceHoistedValues() local