Home
last modified time | relevance | path

Searched refs:AddToAccumGradFunction (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Dtyped_conditional_accumulator_base.h60 AddToAccumGradFunction(ctx, grad); in TryApplyGrad()
78 virtual void AddToAccumGradFunction(OpKernelContext* ctx,
H A Dconditional_accumulator.h95 void AddToAccumGradFunction(OpKernelContext* ctx, in AddToAccumGradFunction() function
H A Dsparse_conditional_accumulator.h174 void AddToAccumGradFunction( in AddToAccumGradFunction() function