Searched defs:parallel_execute (Results 1 – 3 of 3) sorted by relevance
58 tf_device::ParallelExecuteOp parallel_execute, Value result) { in GetSingleUseResourceWrite()83 tf_device::ParallelExecuteOp parallel_execute) { in SinkResourceWritesIntoParallelExecute()154 getFunction().walk([&](tf_device::ParallelExecuteOp parallel_execute) { in runOnFunction()158 for (tf_device::ParallelExecuteOp parallel_execute : parallel_executes) in runOnFunction() local
167 auto parallel_execute = llvm::dyn_cast<tf_device::ParallelExecuteOp>( in BuildVariableAccessInfo() local346 tf_device::ParallelExecuteOp parallel_execute, int start, in AppendTypes()361 tf_device::ParallelExecuteOp parallel_execute, in ReplaceParallelExecute()525 if (auto parallel_execute = llvm::dyn_cast<tf_device::ParallelExecuteOp>( in MergeForOneTPUExecute() local549 auto parallel_execute = in ParentParallelExecuteWrapsSingleOp() local
392 mlir::tf_device::ParallelExecuteOp parallel_execute) { in GetTileShardedOutputsToMerge()415 mlir::tf_device::ParallelExecuteOp parallel_execute, in HandleTileShardedOutputs()540 mlir::tf_device::ParallelExecuteOp parallel_execute, in RemapOutputsFromLogicalDevices()