Home
last modified time | relevance | path

Searched defs:input_grad (Results 1 – 6 of 6) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/image/
H A Dresize_bilinear_op_gpu.cu.cc167 const float* __restrict__ input_grad, in ResizeBilinearGradKernel()
232 const int32 nthreads, const float* __restrict__ input_grad, in ResizeBilinearDeterministicGradKernel()
335 const int32 nthreads, const float* __restrict__ input_grad, in LegacyResizeBilinearGradKernel()
H A Dresize_bicubic_op.cc473 inline void ResizeBicubicGrad(typename TTypes<float, 4>::ConstTensor input_grad, in ResizeBicubicGrad()
596 TTypes<float, 4>::ConstTensor input_grad = in Compute() local
H A Dresize_bilinear_op.cc351 TTypes<float, 4>::ConstTensor input_grad = in Compute() local
H A Dscale_and_translate_op.cc438 TTypes<float, 4>::ConstTensor input_grad = input.tensor<float, 4>(); in Compute() local
/aosp_15_r20/external/tensorflow/tensorflow/cc/gradients/
H A Darray_grad.cc617 auto input_grad = ReduceSum( in TileGrad() local
/aosp_15_r20/external/pytorch/torch/csrc/autograd/
H A DFunctionsManual.cpp5354 Tensor input_grad, tau_grad; in householder_product_backward() local