Searched refs:activations_op (Results 1 – 1 of 1) sorted by relevance
164 def _embedding_activations_grad(activations_op, grad_wrt_activations): argument167 table_id = activations_op.get_attr("table_id")168 lookup_id = activations_op.get_attr("lookup_id")190 for arg in activations_op.inputs