Searched refs:while_region (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | region_control_flow_to_functional.cc | 59 LogicalResult ConvertWhileOp(WhileRegionOp while_region);
358 WhileRegionOp while_region) { in ConvertWhileOp() argument
380 IsSingleCallRegion(while_region.cond(), /*allow_to_bool=*/true), in ConvertWhileOp()
381 IsSingleCallRegion(while_region.body()), while_arg_matcher); in ConvertWhileOp()
384 auto new_inputs = llvm::to_vector<4>(while_region.getOperands()); in ConvertWhileOp()
387 auto new_result_types = llvm::to_vector<4>(while_region.getResultTypes()); in ConvertWhileOp()
403 CollectExternValues(while_region.cond(), while_region.body()); in ConvertWhileOp()
407 cond_name = GetName(while_region, "_cond"); in ConvertWhileOp()
408 ExtractSingleBlockRegion(while_region.cond(), cond_name, extern_values, in ConvertWhileOp()
411 body_name = GetName(while_region, "_body"); in ConvertWhileOp()
[all …]
|
| D | functional_control_flow_to_regions.cc | 117 auto while_region = OpBuilder(while_op).create<TF::WhileRegionOp>( in ConvertWhileOp() local
121 CopyDeviceAndUnderscoredAttributes(while_op, while_region); in ConvertWhileOp()
125 /*caller_region=*/while_region.cond(), while_op.input(), in ConvertWhileOp()
132 /*caller_region=*/while_region.body(), while_op.input(), in ConvertWhileOp()
134 while_op.replaceAllUsesWith(while_region.getResults()); in ConvertWhileOp()
|
| D | resource_device_inference.cc | 198 } else if (auto while_region = dyn_cast<WhileRegionOp>(op)) { in ComputeResourceDevicesInComputation() local
207 for (auto output : filter_resources(while_region.getResults())) { in ComputeResourceDevicesInComputation()
216 for (Region* region : while_region.getRegions()) { in ComputeResourceDevicesInComputation()
223 while_region, result))) in ComputeResourceDevicesInComputation()
|
| D | resource_op_lifting_cleanup.cc | 433 } else if (auto while_region = dyn_cast<TF::WhileRegionOp>(op)) { in CleanupAndCanonicalize() local
434 if (while_region.cond().walk(check_while_cond).wasInterrupted()) in CleanupAndCanonicalize()
437 (void)CanonicalizeWhileRegion(while_region); in CleanupAndCanonicalize()
|
| D | shape_inference.cc | 242 if (auto while_region = llvm::dyn_cast<WhileRegionOp>(use.getOwner())) { in CanInferTensorListElementType() local
243 for (auto branch : while_region.getRegions()) { in CanInferTensorListElementType()
1077 if (auto while_region = dyn_cast<WhileRegionOp>(op)) in InferShapeForSingleOperation() local
1079 while_region, in InferShapeForSingleOperation()
1080 while_region.body().front().getTerminator()->getOperandTypes()); in InferShapeForSingleOperation()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/analysis/ |
| D | resource_alias_analysis.cc | 319 } else if (auto while_region = dyn_cast<WhileRegionOp>(op)) { in ResourceAliasAnalysisInfo() local
320 AnalyzeWhileLoop(while_region, backtrack_analysis.GetAnalysisForRegion( in ResourceAliasAnalysisInfo()
321 while_region.body())); in ResourceAliasAnalysisInfo()
|