Searched refs:parallel_op (Results 1 – 1 of 1) sorted by relevance
640 if (auto parallel_op = llvm::dyn_cast<tf_device::ParallelExecuteOp>( in Rewrite() local642 parallel_op.walk([&](TF::_TPUCompileMlirPlaceholderProgramKeyOp key_op) { in Rewrite()