| /external/tensorflow/tensorflow/core/kernels/image/ |
| D | resize_bilinear_op.cc | 236 typename TTypes<T, 4>::Tensor output_grad(st.output->tensor<T, 4>()); in Compute() local
239 st.width_scale, half_pixel_centers_, output_grad); in Compute()
246 Tensor output_grad; in Compute() local
248 DT_FLOAT, st.output->shape(), &output_grad)); in Compute()
251 st.width_scale, half_pixel_centers_, output_grad.tensor<float, 4>()); in Compute()
252 const Tensor& output_grad_const = output_grad; in Compute()
273 typename TTypes<T, 4>::Tensor output_grad) { in ResizeGradCore()
274 const Eigen::Index batch = output_grad.dimension(0); in ResizeGradCore()
275 const Eigen::Index original_height = output_grad.dimension(1); in ResizeGradCore()
276 const Eigen::Index original_width = output_grad.dimension(2); in ResizeGradCore()
[all …]
|
| D | resize_bilinear_op_gpu.cu.cc | 174 T* __restrict__ output_grad) { in ResizeBilinearGradKernel() argument
203 GpuAtomicAdd(output_grad + in ResizeBilinearGradKernel()
209 GpuAtomicAdd(output_grad + in ResizeBilinearGradKernel()
217 GpuAtomicAdd(output_grad + in ResizeBilinearGradKernel()
223 GpuAtomicAdd(output_grad + in ResizeBilinearGradKernel()
238 float offset, T* __restrict__ output_grad) { in ResizeBilinearDeterministicGradKernel() argument
281 output_grad[out_idx] = acc; in ResizeBilinearDeterministicGradKernel()
340 T* __restrict__ output_grad) { in LegacyResizeBilinearGradKernel() argument
366 GpuAtomicAdd(output_grad + in LegacyResizeBilinearGradKernel()
372 GpuAtomicAdd(output_grad + in LegacyResizeBilinearGradKernel()
[all …]
|
| D | resize_bicubic_op.cc | 476 typename TTypes<T, 4>::Tensor output_grad) { in ResizeBicubicGrad() argument
486 output_grad.setZero(); in ResizeBicubicGrad()
505 output_grad(b, y_wai.index_0, x_wai.index_0, c) += in ResizeBicubicGrad()
507 output_grad(b, y_wai.index_0, x_wai.index_1, c) += in ResizeBicubicGrad()
509 output_grad(b, y_wai.index_0, x_wai.index_2, c) += in ResizeBicubicGrad()
511 output_grad(b, y_wai.index_0, x_wai.index_3, c) += in ResizeBicubicGrad()
514 output_grad(b, y_wai.index_1, x_wai.index_0, c) += in ResizeBicubicGrad()
516 output_grad(b, y_wai.index_1, x_wai.index_1, c) += in ResizeBicubicGrad()
518 output_grad(b, y_wai.index_1, x_wai.index_2, c) += in ResizeBicubicGrad()
520 output_grad(b, y_wai.index_1, x_wai.index_3, c) += in ResizeBicubicGrad()
[all …]
|
| D | resize_bilinear_op.h | 40 typename TTypes<T, 4>::Tensor output_grad);
|
| D | resize_nearest_neighbor_op.h | 39 typename TTypes<T, 4>::Tensor output_grad);
|
| D | scale_and_translate_op.cc | 438 typename TTypes<T, 4>::Tensor output_grad(output->tensor<T, 4>()); in Compute() local
476 intermediate_data, output_grad); in Compute()
|
| /external/tensorflow/tensorflow/python/ops/ragged/ |
| D | ragged_to_tensor_op_test.py | 513 output_grad = [[9, 8, 7], [6, 5, 4], [3, 2, 1]]),
523 output_grad = [
534 output_grad = [
544 output_grad = [[[1, 2], [3, 4]], [[5, 6], [7, 8]]]),
554 output_grad = [[[1, 2], [3, 4]], [[5, 6], [7, 8]]]),
564 output_grad = [[[1, 2], [3, 4]], [[5, 6], [7, 8]]]),
574 output_grad = [[[1, 2], [3, 4]], [[5, 6], [7, 8]]]),
586 output_grad = [[[[1], [2]], [[3], [4]], [[5], [6]]],
600 output_grad = [[[[1], [2]], [[3], [4]], [[5], [6]]],
611 output_grad = [])
[all …]
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | linalg_grad.py | 143 def _GetGradReduced(output_grad, output_subs, input_subs, input_shape, argument
189 array_ops.reshape(output_grad, reduced_shape), input_shape)
198 [reduced_dims, array_ops.shape(output_grad)], axis=0)
205 array_ops.shape(output_grad)
211 array_ops.reshape(output_grad, reduced_shape),
220 def _GetGradWrt(output_grad, other_operand, input_shape, input_subs, argument
282 grad_reduced = gen_linalg_ops.einsum([output_grad, other_operand],
|
| /external/tensorflow/tensorflow/python/eager/ |
| D | backprop_test.py | 1511 output_grad = array_ops.ones([])
1512 t.watch(output_grad)
1515 gc = tt.gradient(output, c, output_gradients=output_grad)
1522 transpose = t.gradient(gc, output_grad)
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | array_ops_test.py | 1682 output_grad = gradient_checker_v2.compute_gradient(f, [input_tensor])
1683 self.assertAllClose(output_grad[0], np.zeros([1, 4, 4]))
|
| D | control_flow_ops_py_test.py | 3901 output_grad = control_flow_ops.while_loop(
3903 self.assertAllClose(600.0, self.evaluate(output_grad)[1])
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | generic_layout_optimizer_transposer_test.cc | 161 auto output_grad = in CreateSimpleMaxPoolGrad() local
170 output_data, output_grad, {1, kKernel, kKernel, 1}, in CreateSimpleMaxPoolGrad()
175 output_data, output_grad, {1, kKernel, kKernel, 1}, in CreateSimpleMaxPoolGrad()
|