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