Searched refs:parallel_execute_op (Results 1 – 4 of 4) sorted by relevance
92 tf_device::ParallelExecuteOp parallel_execute_op, OpBuilder* builder, in ExpandParallelExecuteToIslands() argument94 const int num_regions = parallel_execute_op.getOperation()->getNumRegions(); in ExpandParallelExecuteToIslands()98 Block& execute_block = parallel_execute_op.GetRegionBlockWithIndex(i); in ExpandParallelExecuteToIslands()122 tf_device::ParallelExecuteOp parallel_execute_op) { in CreateIslandsFromParallelExecute() argument127 ExpandParallelExecuteToIslands(island_op, parallel_execute_op, &builder, in CreateIslandsFromParallelExecute()134 parallel_execute_op.getOperation()->getNumResults()); in CreateIslandsFromParallelExecute()195 auto parallel_execute_op = in runOnFunction() local197 CreateIslandsFromParallelExecute(island_op, parallel_execute_op); in runOnFunction()
336 Operation* parallel_execute_op = parallel_execute.getOperation(); in ReplaceParallelExecute() local341 const int parallel_execute_num_results = parallel_execute_op->getNumResults(); in ReplaceParallelExecute()350 const int num_regions = parallel_execute_op->getNumRegions(); in ReplaceParallelExecute()362 parallel_execute_op->getResult(i).replaceAllUsesWith( in ReplaceParallelExecute()372 parallel_execute_op->getResult(parallel_execute_num_results - i - 1) in ReplaceParallelExecute()398 parallel_execute_op->getRegions())) in ReplaceParallelExecute()402 parallel_execute_op->dropAllUses(); in ReplaceParallelExecute()
429 auto parallel_execute_op = builder.create<tf_device::ParallelExecuteOp>( in CreateParallelExecuteForOutsideCompilation() local432 parallel_execute_op.GetRegionBlockWithIndex(0); in CreateParallelExecuteForOutsideCompilation()471 parallel_execute_op.GetRegionBlockWithIndex(1); in CreateParallelExecuteForOutsideCompilation()481 llvm::zip(tpu_cluster.getResults(), parallel_execute_op.getResults())) { in CreateParallelExecuteForOutsideCompilation()485 if (!parallel_execute_op.getOperation()->isProperAncestor(use.getOwner())) in CreateParallelExecuteForOutsideCompilation()
473 OpBuilder* builder, tf_device::ParallelExecuteOp* parallel_execute_op) { in BuildParallelExecuteOp() argument494 *parallel_execute_op = builder->create<tf_device::ParallelExecuteOp>( in BuildParallelExecuteOp()501 builder->setInsertionPoint(*parallel_execute_op); in BuildParallelExecuteOp()510 auto& region = parallel_execute_op->GetRegionBlockWithIndex(core); in BuildParallelExecuteOp()