Searched refs:merged_execute_launch (Results 1 – 1 of 1) sorted by relevance
333 tf_device::LaunchOp merged_execute_launch, in ReplaceParallelExecute() argument343 Region* execute_region = merged_execute_launch->getParentRegion(); in ReplaceParallelExecute()348 output_types.append(merged_execute_launch.getResultTypes().begin(), in ReplaceParallelExecute()349 merged_execute_launch.getResultTypes().end()); in ReplaceParallelExecute()390 merged_execute_launch.getResults()); in ReplaceParallelExecute()408 tf_device::LaunchOp merged_execute_launch, in ReplaceExecute() argument414 merged_execute_launch.getResult(infos.old_to_new_output_mapping[i])); in ReplaceExecute()552 auto merged_execute_launch = builder->create<tf_device::LaunchOp>( in MergeForOneTPUExecute() local555 merged_execute_launch.body().push_back(new Block); in MergeForOneTPUExecute()557 builder->setInsertionPointToEnd(&merged_execute_launch.GetBody()); in MergeForOneTPUExecute()[all …]