Searched refs:compilation_key_functions (Results 1 – 1 of 1) sorted by relevance
120 llvm::SmallVector<mlir::func::FuncOp, 4> compilation_key_functions; in CreateSendRecvOpsToTransferProgramKey() local121 compilation_key_functions.reserve(num_tpu_devices); in CreateSendRecvOpsToTransferProgramKey()152 compilation_key_functions.emplace_back(recv_select_fn); in CreateSendRecvOpsToTransferProgramKey()162 for (auto& func : compilation_key_functions) in CreateSendRecvOpsToTransferProgramKey()