Searched defs:parallel_execute (Results 1 – 3 of 3) sorted by relevance
49 tf_device::ParallelExecuteOp parallel_execute, Value result) { in GetSingleUseResourceWrite()74 tf_device::ParallelExecuteOp parallel_execute) { in SinkResourceWritesIntoParallelExecute()145 getFunction().walk([&](tf_device::ParallelExecuteOp parallel_execute) { in runOnFunction()149 for (tf_device::ParallelExecuteOp parallel_execute : parallel_executes) in runOnFunction() local
140 auto parallel_execute = llvm::dyn_cast<tf_device::ParallelExecuteOp>( in BuildVariableAccessInfo() local317 tf_device::ParallelExecuteOp parallel_execute, int start, in AppendTypes()331 void ReplaceParallelExecute(tf_device::ParallelExecuteOp parallel_execute, in ReplaceParallelExecute()564 if (auto parallel_execute = llvm::dyn_cast<tf_device::ParallelExecuteOp>( in MergeForOneTPUExecute() local588 auto parallel_execute = in ParentParallelExecuteWrapsSingleOp() local
393 mlir::tf_device::ParallelExecuteOp parallel_execute) { in GetTileShardedOutputsToMerge()413 mlir::tf_device::ParallelExecuteOp parallel_execute, in HandleTileShardedOutputs()540 mlir::tf_device::ParallelExecuteOp parallel_execute, in RemapOutputsFromLogicalDevices()