Searched refs:in_row_size (Results 1 – 5 of 5) sorted by relevance
126 const int64 in_row_size = in_width * channels; in resize_image() local127 const int64 in_batch_num_values = in_height * in_row_size; in resize_image()137 const T* ys_input_lower_ptr = input_b_ptr + ys[y].lower * in_row_size; in resize_image()138 const T* ys_input_upper_ptr = input_b_ptr + ys[y].upper * in_row_size; in resize_image()182 const T* ys_input_lower_ptr = input_b_ptr + ys[y].lower * in_row_size; in resize_image()183 const T* ys_input_upper_ptr = input_b_ptr + ys[y].upper * in_row_size; in resize_image()
482 const int64 in_row_size = in_width * channels; in ResizeImageReference() local483 const int64 in_batch_num_values = in_height * in_row_size; in ResizeImageReference()491 const T* ys_input_lower_ptr = input_b_ptr + ys.lower[y] * in_row_size; in ResizeImageReference()492 const T* ys_input_upper_ptr = input_b_ptr + ys.upper[y] * in_row_size; in ResizeImageReference()550 const int64 in_row_size = in_width * channels; in ResizeImage() local551 const int64 in_batch_num_values = in_height * in_row_size; in ResizeImage()561 input_b_ptr + ys.lower[y] * in_row_size; in ResizeImage()563 input_b_ptr + ys.upper[y] * in_row_size; in ResizeImage()613 const int64 in_row_size = in_width * channels; in ResizeImage() local614 const int64 in_batch_num_values = in_height * in_row_size; in ResizeImage()[all …]
488 const int64 in_row_size = input_width * channels; in GatherColumns() local492 const T* input_row_start = image + in_row_size * y; in GatherColumns()530 const int64 in_row_size = input_width * channels; in GatherRows() local537 const T* in_row_data = image + in_row_size * in_row; in GatherRows()545 AddScaledVector(in_row_data, in_row_size, *weight_it, out_row_data); in GatherRows()546 in_row_data += in_row_size; in GatherRows()
121 const int64 in_row_size = in_width * channels; in CalcReferenceResizedVal() local122 const int64 in_batch_num_values = in_height * in_row_size; in CalcReferenceResizedVal()125 b * in_batch_num_values + std::get<0>(y_weight) * in_row_size; in CalcReferenceResizedVal()127 b * in_batch_num_values + std::get<1>(y_weight) * in_row_size; in CalcReferenceResizedVal()
202 const int in_row_size = in_width * in_depth; in DepthwiseConv2dGPUKernelNHWCSmall() local203 const int in_size = in_height * in_row_size; in DepthwiseConv2dGPUKernelNHWCSmall()216 kKnownEvenHeight ? in_size / 2 : block_height * in_row_size; in DepthwiseConv2dGPUKernelNHWCSmall()1195 const int in_row_size = in_width * in_depth;1196 const int in_size = in_height * in_row_size;1207 const int tensor_offset = block_height * in_row_size;