Searched defs:send_gradient_op_map (Results 1 – 1 of 1) sorted by relevance
68 llvm::StringMap<Operation*>* send_gradient_op_map) { in FindTPUEmbeddingOps()97 const llvm::StringMap<Operation*>& send_gradient_op_map, in UpdateEmbeddingEnqueueOpInput()153 llvm::StringMap<Operation*> send_gradient_op_map; in runOnOperation() local