Searched defs:if_region (Results 1 – 5 of 5) sorted by relevance
101 auto if_region = builder.create<TF::IfRegionOp>( in ConvertIfOp() local
298 LogicalResult RegionControlFlowToFunctional::ConvertIfOp(IfRegionOp if_region) { in ConvertIfOp()469 if (auto if_region = llvm::dyn_cast<IfRegionOp>(op)) { in runOnOperation() local
199 TF::IfRegionOp CloneEmptyIfWithPredicate(TF::IfRegionOp if_region, in CloneEmptyIfWithPredicate()
2077 if (auto if_region = dyn_cast<IfRegionOp>(op)) in InferShapeForSingleOperation() local
265 void CloneEmptyIfWithPredicate(mlir::TF::IfRegionOp if_region, const Mesh& mesh, in CloneEmptyIfWithPredicate()