Searched refs:GetOpGradientCreator (Results 1 – 5 of 5) sorted by relevance
338 Status s = gradient::GetOpGradientCreator(func, &creator); in IsPrimitiveOpWithNoGrad()
698 TF_RETURN_IF_ERROR(gradient::GetOpGradientCreator(func.name(), &creator)); in InstantiateSymbolicGradient()
1153 Status GetOpGradientCreator(const std::string& op, Creator* creator);
2032 Status GetOpGradientCreator(const string& op, Creator* creator) { in GetOpGradientCreator() function
1051 TF_RETURN_IF_ERROR(gradient::GetOpGradientCreator(func.name(), &creator)); in MakeFunctionBodyForInlining()