Home
last modified time | relevance | path

Searched defs:while_region (Results 1 – 6 of 6) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
H A Dregion_control_flow_to_functional.cc378 WhileRegionOp while_region) { in ConvertWhileOp()
474 } else if (auto while_region = llvm::dyn_cast<WhileRegionOp>(op)) { in runOnOperation() local
H A Dfunctional_control_flow_to_regions.cc119 auto while_region = OpBuilder(while_op).create<TF::WhileRegionOp>( in ConvertWhileOp() local
H A Dresource_device_inference.cc194 } else if (auto while_region = dyn_cast<WhileRegionOp>(op)) { in ComputeResourceDevicesInComputation() local
H A Dresource_op_lifting_cleanup.cc438 } else if (auto while_region = dyn_cast<TF::WhileRegionOp>(op)) { in CleanupAndCanonicalize() local
H A Dshape_inference.cc350 if (auto while_region = llvm::dyn_cast<WhileRegionOp>(use.getOwner())) { in CanInferTensorListElementType() local
2084 if (auto while_region = dyn_cast<WhileRegionOp>(op)) in InferShapeForSingleOperation() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/analysis/
H A Dresource_alias_analysis.cc385 } else if (auto while_region = dyn_cast<WhileRegionOp>(op)) { in ResourceAliasAnalysisInfo() local