Home
last modified time | relevance | path

Searched defs:output_grad (Results 1 – 11 of 11) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/image/
H A Dresize_bilinear_op_gpu.cu.cc172 T* __restrict__ output_grad) { in ResizeBilinearGradKernel()
236 float offset, T* __restrict__ output_grad) { in ResizeBilinearDeterministicGradKernel()
338 T* __restrict__ output_grad) { in LegacyResizeBilinearGradKernel()
H A Dresize_bilinear_op.cc356 typename TTypes<T, 4>::Tensor output_grad(st.output->tensor<T, 4>()); in Compute() local
366 Tensor output_grad; in Compute() local
H A Dresize_bicubic_op.cc476 typename TTypes<T, 4>::Tensor output_grad) { in ResizeBicubicGrad()
599 typename TTypes<T, 4>::Tensor output_grad(st.output->tensor<T, 4>()); in Compute() local
H A Dscale_and_translate_op.cc439 typename TTypes<T, 4>::Tensor output_grad(output->tensor<T, 4>()); in Compute() local
/aosp_15_r20/external/tensorflow/tensorflow/cc/gradients/
H A Dlinalg_grad.cc184 Output EinsumGradReducedHelper(const Scope& scope, const Output& output_grad, in EinsumGradReducedHelper()
281 Output EinsumGradWrt(const Scope& scope, Output output_grad, in EinsumGradWrt()
/aosp_15_r20/external/tensorflow/tensorflow/python/ops/ragged/
H A Dragged_to_tensor_op_test.py615 output_grad, argument
664 def _test_gradient_helper(self, rt_val, default_val, shape_val, output_grad, argument
/aosp_15_r20/external/tensorflow/tensorflow/python/ops/
H A Dlinalg_grad.py143 def _GetGradReduced(output_grad, output_subs, input_subs, input_shape, argument
220 def _GetGradWrt(output_grad, other_operand, input_shape, input_subs, argument
/aosp_15_r20/external/pytorch/aten/src/ATen/native/nested/
H A DNestedTensorBackward.cpp147 scalar_t output_grad = output_grad_data[out_idx]; in _nested_sum_backward_cpu() local
/aosp_15_r20/external/pytorch/torch/csrc/jit/runtime/
H A Dautodiff.cpp412 Value* output_grad = reverse_block->addInput()->setType(output->type()); in addReverseInline() local
/aosp_15_r20/external/tensorflow/tensorflow/core/grappler/optimizers/
H A Dremapper_test.cc252 auto output_grad = in TEST_F() local
505 auto output_grad = in TEST_F() local
H A Dgeneric_layout_optimizer_transposer_test.cc161 auto output_grad = in CreateSimpleMaxPoolGrad() local