Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Dtyped_conditional_accumulator_base.h62 AllocateAndAssignToAccumGradFunction(ctx, grad); in TryApplyGrad()
75 virtual void AllocateAndAssignToAccumGradFunction(
H A Dconditional_accumulator.h87 void AllocateAndAssignToAccumGradFunction(OpKernelContext* ctx, in AllocateAndAssignToAccumGradFunction() function
H A Dsparse_conditional_accumulator.h146 void AllocateAndAssignToAccumGradFunction( in AllocateAndAssignToAccumGradFunction() function