Searched refs:apply_gradients_op (Results 1 – 1 of 1) sorted by relevance
137 auto apply_gradients_op = all_nodes_[apply_gradient_node_name]->op(); in Initialize() local142 apply_gradients_node->input(gradient_pos[apply_gradients_op]), in Initialize()145 *apply_gradients_node->mutable_input(gradient_pos[apply_gradients_op]) = in Initialize()