Searched refs:enclosing_cluster (Results 1 – 3 of 3) sorted by relevance
137 auto enclosing_cluster = op->getParentOfType<mlir::tf_device::ClusterOp>(); in ExtractDeviceMeshEnclosingCluster() local138 if (!enclosing_cluster) in ExtractDeviceMeshEnclosingCluster()141 TF_ASSIGN_OR_RETURN(auto mesh, ExtractDeviceMeshFromOp(enclosing_cluster)); in ExtractDeviceMeshEnclosingCluster()
349 auto enclosing_cluster = in TraceUseToNextTFOp() local352 enclosing_cluster.getResult(operand->getOperandNumber())); in TraceUseToNextTFOp()
1165 auto enclosing_cluster = op->getParentOfType<mlir::tf_device::ClusterOp>(); in ExpandOp() local1166 if (!enclosing_cluster) in ExpandOp()1207 if (use.getOwner() == enclosing_cluster.GetBody().getTerminator()) { in ExpandOp()1219 enclosing_cluster->setAttr(kShapeOpInputLayoutIndices, in ExpandOp()1221 enclosing_cluster->setAttr( in ExpandOp()