Searched refs:gradient_pos (Results 1 – 1 of 1) sorted by relevance
125 std::map<string, int> gradient_pos = {{"ApplyGradientDescent", 2}, 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()