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