Searched refs:activations_op (Results 1 – 1 of 1) sorted by relevance
166 def _embedding_activations_grad(activations_op, grad_wrt_activations): argument169 table_id = activations_op.get_attr("table_id")170 lookup_id = activations_op.get_attr("lookup_id")192 for arg in activations_op.inputs