Searched defs:new_parallel_execute (Results 1 – 4 of 4) sorted by relevance
393 int cluster_idx, mlir::tf_device::ParallelExecuteOp new_parallel_execute, in GetTileShardedOutputsToMerge()421 int cluster_idx, mlir::tf_device::ParallelExecuteOp new_parallel_execute, in HandleTileShardedOutputs()558 mlir::tf_device::ParallelExecuteOp new_parallel_execute, in RemapOutputsFromLogicalDevices()
132 auto new_parallel_execute = builder.create<tf_device::ParallelExecuteOp>( in SinkResourceWritesIntoParallelExecute() local
468 tf_device::ParallelExecuteOp* new_parallel_execute) { in MovePreservedParallelExecuteChildren()512 tf_device::ParallelExecuteOp* new_parallel_execute, int* cluster_idx) { in AddToParallelExecuteOp()827 tf_device::ParallelExecuteOp new_parallel_execute; in Rewrite() local
384 auto new_parallel_execute = builder->create<tf_device::ParallelExecuteOp>( in ReplaceParallelExecute() local