Searched refs:parallel_execute_op (Results 1 – 3 of 3) sorted by relevance
94 tf_device::ParallelExecuteOp parallel_execute_op, OpBuilder* builder, in ExpandParallelExecuteToIslands() argument96 const int num_regions = parallel_execute_op.getOperation()->getNumRegions(); in ExpandParallelExecuteToIslands()100 Block& execute_block = parallel_execute_op.GetRegionBlockWithIndex(i); in ExpandParallelExecuteToIslands()124 tf_device::ParallelExecuteOp parallel_execute_op) { in CreateIslandsFromParallelExecute() argument129 ExpandParallelExecuteToIslands(island_op, parallel_execute_op, &builder, in CreateIslandsFromParallelExecute()136 parallel_execute_op.getOperation()->getNumResults()); in CreateIslandsFromParallelExecute()197 auto parallel_execute_op = in runOnOperation() local199 CreateIslandsFromParallelExecute(island_op, parallel_execute_op); in runOnOperation()
365 Operation* parallel_execute_op = parallel_execute.getOperation(); in ReplaceParallelExecute() local370 const int parallel_execute_num_results = parallel_execute_op->getNumResults(); in ReplaceParallelExecute()379 const int num_regions = parallel_execute_op->getNumRegions(); in ReplaceParallelExecute()391 parallel_execute_op->getResult(i).replaceAllUsesWith( in ReplaceParallelExecute()401 parallel_execute_op->getResult(parallel_execute_num_results - i - 1) in ReplaceParallelExecute()429 parallel_execute_op->getRegions())) in ReplaceParallelExecute()433 parallel_execute_op->dropAllUses(); in ReplaceParallelExecute()
728 auto parallel_execute_op = builder.create<tf_device::ParallelExecuteOp>( in CreateParallelExecuteForOutsideCompilation() local731 parallel_execute_op.GetRegionBlockWithIndex(0); in CreateParallelExecuteForOutsideCompilation()773 parallel_execute_op.GetRegionBlockWithIndex(1); in CreateParallelExecuteForOutsideCompilation()783 llvm::zip(tpu_cluster.getResults(), parallel_execute_op.getResults())) { in CreateParallelExecuteForOutsideCompilation()787 if (!parallel_execute_op.getOperation()->isProperAncestor(use.getOwner())) in CreateParallelExecuteForOutsideCompilation()