Searched defs:output_grad (Results 1 – 9 of 9) 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/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
|