Searched refs:state_variables (Results 1 – 5 of 5) sorted by relevance
207 std::vector<StateVariableSpecification>* state_variables) { in GetOptimizationAlgorithmStateVariables() argument213 state_variables->push_back( in GetOptimizationAlgorithmStateVariables()324 state_variables->push_back(std::move(gradient_acc)); in GetOptimizationAlgorithmStateVariables()326 if (state_variables->size() > kMaxAuxiliaryParameterCount + 1) { in GetOptimizationAlgorithmStateVariables()
64 std::vector<StateVariableSpecification> *state_variables);
287 self.state_variables = collections.OrderedDict()357 self.state_variables[name] = weight363 for name, var in self.state_variables.items():397 self.state_variables[var_name].assign(value)
64 for name, var in self.state_variables.items():85 state_ops.assign(self.state_variables[var_name], value))
1056 state_variables = {1077 layer._set_state_variables(state_variables)1128 state_variables = {1142 layer._set_state_variables(state_variables)