Searched defs:output_grad (Results 1 - 3 of 3) sorted by relevance
/external/tensorflow/tensorflow/core/kernels/ |
H A D | resize_bilinear_op_gpu.cu.cc | 90 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 D | resize_bilinear_op.cc | 261 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 D | resize_bicubic_op.cc | 401 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