Home
last modified time | relevance | path

Searched defs:if_region (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
Dfunctional_control_flow_to_regions.cc99 auto if_region = builder.create<TF::IfRegionOp>( in ConvertIfOp() local
Dregion_control_flow_to_functional.cc296 LogicalResult RegionControlFlowToFunctional::ConvertIfOp(IfRegionOp if_region) { in ConvertIfOp()
467 if (auto if_region = llvm::dyn_cast<IfRegionOp>(op)) { in runOnOperation() local
Dtpu_extract_outside_compilation.cc199 TF::IfRegionOp CloneEmptyIfWithPredicate(TF::IfRegionOp if_region, in CloneEmptyIfWithPredicate()
Dshape_inference.cc1281 if (auto if_region = dyn_cast<IfRegionOp>(op)) in InferShapeForSingleOperation() local