| /external/tensorflow/tensorflow/core/kernels/image/ |
| D | resize_bilinear_op.cc | 232 TTypes<float, 4>::ConstTensor input_grad = input.tensor<float, 4>(); in Compute() local
238 context->eigen_device<Device>(), input_grad, st.height_scale, in Compute()
250 context->eigen_device<Device>(), input_grad, st.height_scale, in Compute()
271 typename TTypes<float, 4>::ConstTensor input_grad, in ResizeGradCore()
279 const Eigen::Index resized_height = input_grad.dimension(1); in ResizeGradCore()
280 const Eigen::Index resized_width = input_grad.dimension(2); in ResizeGradCore()
316 T(input_grad(b, y, x, c) * inverse_y_lerp * inverse_x_lerp); in ResizeGradCore()
318 T(input_grad(b, y, x, c) * inverse_y_lerp * x_lerp); in ResizeGradCore()
320 T(input_grad(b, y, x, c) * y_lerp * inverse_x_lerp); in ResizeGradCore()
322 T(input_grad(b, y, x, c) * y_lerp * x_lerp); in ResizeGradCore()
[all …]
|
| D | resize_bilinear_op_gpu.cu.cc | 169 const float* __restrict__ input_grad, in ResizeBilinearGradKernel() argument
202 const float dtop = (1 - y_lerp) * input_grad[in_idx]; in ResizeBilinearGradKernel()
216 const float dbottom = y_lerp * input_grad[in_idx]; in ResizeBilinearGradKernel()
234 const int32 nthreads, const float* __restrict__ input_grad, in ResizeBilinearDeterministicGradKernel() argument
274 acc += static_cast<T>(input_grad[in_idx] * y_lerp * x_lerp); in ResizeBilinearDeterministicGradKernel()
337 const int32 nthreads, const float* __restrict__ input_grad, in LegacyResizeBilinearGradKernel() argument
365 const float dtop = (1 - y_lerp) * input_grad[in_idx]; in LegacyResizeBilinearGradKernel()
379 const float dbottom = y_lerp * input_grad[in_idx]; in LegacyResizeBilinearGradKernel()
466 typename TTypes<float, 4>::ConstTensor input_grad, in operator ()()
475 const int resized_height = input_grad.dimension(1); in operator ()()
[all …]
|
| D | resize_bilinear_op.h | 37 typename TTypes<float, 4>::ConstTensor input_grad,
|
| D | resize_nearest_neighbor_op.h | 37 typename TTypes<T, 4>::ConstTensor input_grad,
|
| D | resize_bicubic_op.cc | 473 inline void ResizeBicubicGrad(typename TTypes<float, 4>::ConstTensor input_grad, in ResizeBicubicGrad() argument
503 T curr_input_grad = input_grad(b, y, x, c); in ResizeBicubicGrad()
599 TTypes<float, 4>::ConstTensor input_grad = input.tensor<float, 4>(); in Compute() local
602 ResizeBicubicGrad<T>(input_grad, st, half_pixel_centers_, output_grad); in Compute()
|
| D | scale_and_translate_op.cc | 437 TTypes<float, 4>::ConstTensor input_grad = input.tensor<float, 4>(); in Compute() local
440 const int64 forward_output_height = input_grad.dimension(1); in Compute()
441 const int64 forward_output_width = input_grad.dimension(2); in Compute()
475 row_weights, col_spans.span_size, col_starts, col_weights, input_grad, in Compute()
|
| /external/tensorflow/tensorflow/python/kernel_tests/array_ops/ |
| D | scatter_nd_ops_test.py | 646 updates_grad, input_grad = tape.gradient([outputs], [updates, input_],
653 self.assertAllEqual(expected_input_grad, self.evaluate(input_grad))
668 updates_grad, input_grad = tape.gradient([outputs], [updates, input_],
676 self.assertAllEqual(expected_input_grad, self.evaluate(input_grad))
694 updates_grad, input_grad = tape.gradient([outputs], [updates, input_],
702 self.assertAllEqual(expected_input_grad, self.evaluate(input_grad))
722 updates_grad, input_grad = tape.gradient([outputs], [updates, input_],
732 self.assertAllEqual(expected_input_grad, self.evaluate(input_grad))
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | array_grad.py | 847 input_grad = math_ops.reduce_sum(array_ops.reshape(grad, split_shape), axes)
850 input_grad.set_shape(op.inputs[0].get_shape())
851 return [input_grad, None]
|
| /external/tensorflow/tensorflow/compiler/jit/tests/ |
| D | opens2s_gnmt_mixed_precision.pbtxt.gz | |