Searched refs:old_parallel_execute (Results 1 – 3 of 3) sorted by relevance
467 tf_device::ParallelExecuteOp old_parallel_execute, in MovePreservedParallelExecuteChildren() argument470 const int num_moved_children = old_parallel_execute.regions().size() - 1; in MovePreservedParallelExecuteChildren()472 old_parallel_execute->getLoc(), in MovePreservedParallelExecuteChildren()478 for (int child_idx = 0; child_idx < old_parallel_execute.regions().size(); in MovePreservedParallelExecuteChildren()480 auto& block = old_parallel_execute.GetRegionBlockWithIndex(child_idx); in MovePreservedParallelExecuteChildren()494 *old_parallel_execute.getRegions()[old_idx]); in MovePreservedParallelExecuteChildren()511 OpBuilder* builder, tf_device::ParallelExecuteOp old_parallel_execute, in AddToParallelExecuteOp() argument537 old_parallel_execute, new_parallel_execute); in AddToParallelExecuteOp()703 tf_device::ParallelExecuteOp old_parallel_execute = in Rewrite() local705 if (old_parallel_execute && in Rewrite()[all …]
557 mlir::tf_device::ParallelExecuteOp old_parallel_execute, int cluster_idx, in RemapOutputsFromLogicalDevices() argument561 llvm::enumerate(old_parallel_execute.getResults())) { in RemapOutputsFromLogicalDevices()587 return old_parallel_execute.emitOpError() in RemapOutputsFromLogicalDevices()
70 mlir::tf_device::ParallelExecuteOp old_parallel_execute, int cluster_idx,