Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Dtyped_conditional_accumulator_base.h57 bool is_valid = GetAndValidateTensorInputForApplyGrad(ctx, &grad); in TryApplyGrad()
84 virtual bool GetAndValidateTensorInputForApplyGrad(
H A Dconditional_accumulator.h115 bool GetAndValidateTensorInputForApplyGrad(OpKernelContext* ctx, in GetAndValidateTensorInputForApplyGrad() function
H A Dsparse_conditional_accumulator.h331 bool GetAndValidateTensorInputForApplyGrad( in GetAndValidateTensorInputForApplyGrad() function