Searched defs:output_grad (Results 1 – 11 of 11) sorted by relevance
| /external/tensorflow/tensorflow/cc/gradients/ |
| D | linalg_grad.cc | 184 Output EinsumGradReducedHelper(const Scope& scope, const Output& output_grad, in EinsumGradReducedHelper()
281 Output EinsumGradWrt(const Scope& scope, Output output_grad, in EinsumGradWrt()
|
| /external/tensorflow/tensorflow/python/ops/ragged/ |
| D | ragged_to_tensor_op_test.py | 615 output_grad, argument
664 def _test_gradient_helper(self, rt_val, default_val, shape_val, output_grad, argument
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | linalg_grad.py | 143 def _GetGradReduced(output_grad, output_subs, input_subs, input_shape, argument
220 def _GetGradWrt(output_grad, other_operand, input_shape, input_subs, argument
|
| /external/tensorflow/tensorflow/core/kernels/image/ |
| D | resize_bilinear_op_gpu.cu.cc | 172 T* __restrict__ output_grad) { in ResizeBilinearGradKernel()
236 float offset, T* __restrict__ output_grad) { in ResizeBilinearDeterministicGradKernel()
338 T* __restrict__ output_grad) { in LegacyResizeBilinearGradKernel()
|
| D | resize_bilinear_op.cc | 356 typename TTypes<T, 4>::Tensor output_grad(st.output->tensor<T, 4>()); in Compute() local
366 Tensor output_grad; in Compute() local
|
| D | resize_bicubic_op.cc | 476 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
|
| D | scale_and_translate_op.cc | 439 typename TTypes<T, 4>::Tensor output_grad(output->tensor<T, 4>()); in Compute() local
|
| /external/pytorch/aten/src/ATen/native/nested/ |
| D | NestedTensorBackward.cpp | 147 scalar_t output_grad = output_grad_data[out_idx]; in _nested_sum_backward_cpu() local
|
| /external/pytorch/torch/csrc/jit/runtime/ |
| D | autodiff.cpp | 412 Value* output_grad = reverse_block->addInput()->setType(output->type()); in addReverseInline() local
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | remapper_test.cc | 252 auto output_grad = in TEST_F() local
505 auto output_grad = in TEST_F() local
|
| D | generic_layout_optimizer_transposer_test.cc | 161 auto output_grad = in CreateSimpleMaxPoolGrad() local
|