Home
last modified time | relevance | path

Searched defs:grad_tensor (Results 1 – 5 of 5) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Dconditional_accumulator.h119 const Tensor* grad_tensor; in GetAndValidateTensorInputForApplyGrad() local
/aosp_15_r20/external/tensorflow/tensorflow/c/eager/
H A Dgradient_checker.cc188 TF_Tensor* grad_tensor; in CalcNumericalGrad() local
H A Dtape.h999 Gradient* grad_tensor = grad[grad_index]; in ForwardpropFromTape() local
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/mkl/
H A Dmkl_avgpooling_op.cc191 const Tensor& grad_tensor = in Compute() local
H A Dmkl_maxpooling_op.cc245 const Tensor& grad_tensor = in Compute() local