Searched defs:parallel_execute (Results 1 – 3 of 3) sorted by relevance
52 tf_device::ParallelExecuteOp parallel_execute, Value result) { in GetSingleUseResourceWrite()77 tf_device::ParallelExecuteOp parallel_execute) { in SinkResourceWritesIntoParallelExecute()148 getOperation().walk([&](tf_device::ParallelExecuteOp parallel_execute) { in runOnOperation()152 for (tf_device::ParallelExecuteOp parallel_execute : parallel_executes) in runOnOperation() 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
615 tf_device::ParallelExecuteOp* parallel_execute) { in BuildSingletonParallelExecuteOp()633 tf_device::ParallelExecuteOp parallel_execute, OpBuilder* builder) { in RemoveSingletonParallelExecuteOp()661 tf_device::ParallelExecuteOp parallel_execute) { in CheckTPUPartitionedInputAndOutputAreValid()