Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
Dfunctional_control_flow_to_regions.cc117 auto while_region = OpBuilder(while_op).create<TF::WhileRegionOp>( in ConvertWhileOp() local
Dregion_control_flow_to_functional.cc358 WhileRegionOp while_region) { in ConvertWhileOp()
454 } else if (auto while_region = llvm::dyn_cast<WhileRegionOp>(op)) { in runOnOperation() local
Dresource_device_inference.cc198 } else if (auto while_region = dyn_cast<WhileRegionOp>(op)) { in ComputeResourceDevicesInComputation() local
Dresource_op_lifting_cleanup.cc433 } else if (auto while_region = dyn_cast<TF::WhileRegionOp>(op)) { in CleanupAndCanonicalize() local
Dshape_inference.cc242 if (auto while_region = llvm::dyn_cast<WhileRegionOp>(use.getOwner())) { in CanInferTensorListElementType() local
1077 if (auto while_region = dyn_cast<WhileRegionOp>(op)) in InferShapeForSingleOperation() local
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/analysis/
Dresource_alias_analysis.cc319 } else if (auto while_region = dyn_cast<WhileRegionOp>(op)) { in ResourceAliasAnalysisInfo() local