Searched defs:output_grad (Results 1 - 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
H A Dresize_bilinear_op_gpu.cu.cc90 int channels, int resized_height, int resized_width, T* output_grad) {
116 CudaAtomicAdd(output_grad +
122 CudaAtomicAdd(output_grad +
130 CudaAtomicAdd(output_grad +
136 CudaAtomicAdd(output_grad +
181 typename TTypes<T, 4>::Tensor output_grad) {
182 const int batch = output_grad.dimension(0);
183 const int original_height = output_grad.dimension(1);
184 const int original_width = output_grad.dimension(2);
185 const int channels = output_grad
87 ResizeBilinearGradKernel( const int32 nthreads, const float* input_grad, float height_scale, float width_scale, int batch, int original_height, int original_width, int channels, int resized_height, int resized_width, T* output_grad) argument
178 operator ()(const GPUDevice& d, typename TTypes<float, 4>::ConstTensor input_grad, const float height_scale, const float width_scale, typename TTypes<T, 4>::Tensor output_grad) argument
[all...]
H A Dresize_bilinear_op.cc261 typename TTypes<T, 4>::Tensor output_grad(st.output->tensor<T, 4>());
265 st.width_scale, output_grad);
279 typename TTypes<T, 4>::Tensor output_grad) {
280 const int batch = output_grad.dimension(0);
281 const int64 original_height = output_grad.dimension(1);
282 const int64 original_width = output_grad.dimension(2);
283 const int channels = output_grad.dimension(3);
288 output_grad.setZero();
313 output_grad(b, top_y_index, left_x_index, c) +=
315 output_grad(
276 operator ()(const CPUDevice& d, typename TTypes<float, 4>::ConstTensor input_grad, const float height_scale, const float width_scale, typename TTypes<T, 4>::Tensor output_grad) argument
[all...]
H A Dresize_bicubic_op.cc401 typename TTypes<T, 4>::Tensor output_grad) {
411 output_grad.setZero();
424 output_grad(b, y_wai.index_0, x_wai.index_0, c) +=
426 output_grad(b, y_wai.index_0, x_wai.index_1, c) +=
428 output_grad(b, y_wai.index_0, x_wai.index_2, c) +=
430 output_grad(b, y_wai.index_0, x_wai.index_3, c) +=
433 output_grad(b, y_wai.index_1, x_wai.index_0, c) +=
435 output_grad(b, y_wai.index_1, x_wai.index_1, c) +=
437 output_grad(b, y_wai.index_1, x_wai.index_2, c) +=
439 output_grad(
399 ResizeBicubicGrad(typename TTypes<float, 4>::ConstTensor input_grad, const ImageResizerGradientState& resizer_state, typename TTypes<T, 4>::Tensor output_grad) argument
[all...]

Completed in 729 milliseconds