Searched defs:grads_image (Results 1 - 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
H A Dcrop_and_resize_op.cc404 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 Dcrop_and_resize_op_gpu.cu.cc359 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