Searched refs:in_row_size (Results 1 – 4 of 4) sorted by relevance
118 const int64 in_row_size = in_width * channels; in resize_image() local119 const int64 in_batch_num_values = in_height * in_row_size; in resize_image()129 const T* ys_input_lower_ptr = input_b_ptr + ys[y].lower * in_row_size; in resize_image()130 const T* ys_input_upper_ptr = input_b_ptr + ys[y].upper * in_row_size; in resize_image()174 const T* ys_input_lower_ptr = input_b_ptr + ys[y].lower * in_row_size; in resize_image()175 const T* ys_input_upper_ptr = input_b_ptr + ys[y].upper * in_row_size; in resize_image()
475 const int64 in_row_size = in_width * channels; in ResizeImageReference() local476 const int64 in_batch_num_values = in_height * in_row_size; in ResizeImageReference()484 const T* ys_input_lower_ptr = input_b_ptr + ys.lower[y] * in_row_size; in ResizeImageReference()485 const T* ys_input_upper_ptr = input_b_ptr + ys.upper[y] * in_row_size; in ResizeImageReference()540 const int64 in_row_size = in_width * channels; in ResizeImage() local541 const int64 in_batch_num_values = in_height * in_row_size; in ResizeImage()551 input_b_ptr + ys.lower[y] * in_row_size; in ResizeImage()553 input_b_ptr + ys.upper[y] * in_row_size; in ResizeImage()601 const int64 in_row_size = in_width * channels; in ResizeImage() local602 const int64 in_batch_num_values = in_height * in_row_size; in ResizeImage()[all …]
120 const int64 in_row_size = in_width * channels; in CalcReferenceResizedVal() local121 const int64 in_batch_num_values = in_height * in_row_size; in CalcReferenceResizedVal()124 b * in_batch_num_values + std::get<0>(y_weight) * in_row_size; in CalcReferenceResizedVal()126 b * in_batch_num_values + std::get<1>(y_weight) * in_row_size; in CalcReferenceResizedVal()
196 const int in_row_size = in_width * in_depth; in DepthwiseConv2dGPUKernelNHWCSmall() local197 const int in_size = in_height * in_row_size; in DepthwiseConv2dGPUKernelNHWCSmall()210 kKnownEvenHeight ? in_size / 2 : block_height * in_row_size; in DepthwiseConv2dGPUKernelNHWCSmall()1143 const int in_row_size = in_width * in_depth; in DepthwiseConv2dBackpropFilterGPUKernelNHWCSmall() local1144 const int in_size = in_height * in_row_size; in DepthwiseConv2dBackpropFilterGPUKernelNHWCSmall()1155 const int tensor_offset = block_height * in_row_size; in DepthwiseConv2dBackpropFilterGPUKernelNHWCSmall()