Searched refs:func_grad_ (Results 1 – 2 of 2) sorted by relevance
915 : default_registry_(other.default_registry_), func_grad_(other.func_grad_) { in FunctionLibraryDefinition()932 func_grad_[grad.function_name()] = grad.gradient_func(); in FunctionLibraryDefinition()986 string* entry = &func_grad_[grad.function_name()]; in AddGradientDefHelper()1020 for (auto iter : other.func_grad_) { in AddLibrary()1078 const auto& i = func_grad_.find(func); in RemoveGradient()1079 if (i == func_grad_.end()) { in RemoveGradient()1083 func_grad_.erase(i); in RemoveGradient()1101 return gtl::FindWithDefault(func_grad_, func, ""); in FindGradient()1144 for (const auto& g : func_grad_) { in ToProto()
390 gtl::FlatMap<string, string> func_grad_; variable