Searched defs:grads_image (Results 1 - 2 of 2) sorted by relevance
/external/tensorflow/tensorflow/core/kernels/ |
H A D | crop_and_resize_op.cc | 404 typename TTypes<T, 4>::Tensor grads_image) { 405 const int batch_size = grads_image.dimension(0); 406 const int image_height = grads_image.dimension(1); 407 const int image_width = grads_image.dimension(2); 414 grads_image.setZero(); 458 grads_image(b_in, top_y_index, left_x_index, d) += 460 grads_image(b_in, top_y_index, right_x_index, d) += 463 grads_image(b_in, bottom_y_index, left_x_index, d) += 465 grads_image(b_in, bottom_y_index, right_x_index, d) += 400 operator ()(const CPUDevice& d, typename TTypes<float, 4>::ConstTensor grads, typename TTypes<float, 2>::ConstTensor boxes, typename TTypes<int32, 1>::ConstTensor box_index, typename TTypes<T, 4>::Tensor grads_image) argument
|
H A D | crop_and_resize_op_gpu.cu.cc | 359 typename TTypes<T, 4>::Tensor grads_image) { 360 const int batch = grads_image.dimension(0); 361 const int image_height = grads_image.dimension(1); 362 const int image_width = grads_image.dimension(2); 372 // Initialize grads_image with all zeros. 377 config.virtual_thread_count, grads_image.data()); 388 crop_height, crop_width, depth, grads_image.data()); 355 operator ()(const GPUDevice& d, typename TTypes<float, 4>::ConstTensor grads, typename TTypes<float, 2>::ConstTensor boxes, typename TTypes<int32, 1>::ConstTensor box_ind, typename TTypes<T, 4>::Tensor grads_image) argument
|
Completed in 370 milliseconds