Searched defs:enqueue_op_map (Results 1 – 1 of 1) sorted by relevance
66 FuncOp func_op, llvm::StringMap<Operation*>* enqueue_op_map, in FindTPUEmbeddingOps()94 const llvm::StringMap<Operation*>& enqueue_op_map, in UpdateEmbeddingEnqueueOpInput()150 llvm::StringMap<Operation*> enqueue_op_map; in runOnFunction() local