Home
last modified time | relevance | path

Searched refs:GetOpGradientCreator (Results 1 – 5 of 5) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/
H A Dgradients.cc338 Status s = gradient::GetOpGradientCreator(func, &creator); in IsPrimitiveOpWithNoGrad()
H A Dfunction.cc698 TF_RETURN_IF_ERROR(gradient::GetOpGradientCreator(func.name(), &creator)); in InstantiateSymbolicGradient()
/aosp_15_r20/external/tensorflow/tensorflow/core/framework/
H A Dfunction.h1153 Status GetOpGradientCreator(const std::string& op, Creator* creator);
H A Dfunction.cc2032 Status GetOpGradientCreator(const string& op, Creator* creator) { in GetOpGradientCreator() function
/aosp_15_r20/external/tensorflow/tensorflow/core/grappler/optimizers/
H A Dfunction_optimizer.cc1051 TF_RETURN_IF_ERROR(gradient::GetOpGradientCreator(func.name(), &creator)); in MakeFunctionBodyForInlining()