Searched defs:parallel_op (Results 1 – 2 of 2) sorted by relevance
121 fusion.walk([&](scf::ParallelOp parallel_op) { in matchAndRewrite()127 scf::ParallelOp parallel_op = parallel_ops.front(); in matchAndRewrite() local
627 if (auto parallel_op = llvm::dyn_cast<tf_device::ParallelExecuteOp>( in Rewrite() local