Searched refs:offset_first_element (Results 1 – 5 of 5) sorted by relevance
96 unsigned int offset_first_element = info->offset_first_element_in_bytes(); in add_tensor_argument() local100 offset_first_element += (window.is_broadcasted(n) ? 0 : window[n].start()) * strides[n]; in add_tensor_argument()112 _kernel.setArg<cl_uint>(idx++, offset_first_element); in add_tensor_argument()140 unsigned int offset_first_element = info->offset_first_element_in_bytes(); in add_3d_tensor_nhw_argument() local141 _kernel.setArg<cl_uint>(idx++, offset_first_element); in add_3d_tensor_nhw_argument()167 unsigned int offset_first_element = info->offset_first_element_in_bytes(); in add_4d_tensor_nhwc_argument() local168 _kernel.setArg<cl_uint>(idx++, offset_first_element); in add_4d_tensor_nhwc_argument()
501 unsigned int offset_first_element = 0; in add_array_argument() local505 offset_first_element += window[n].start() * strides[n]; in add_array_argument()517 _kernel.setArg<cl_uint>(idx++, offset_first_element); in add_array_argument()
94 const size_t offset_first_element = _info->offset_first_element_in_bytes(); in update_window_if_needed() local101 const int front_pad_y_available = -static_cast<int>(offset_first_element / strides[1]); in update_window_if_needed()128 …const int front_pad_x_available = -std::min<int>(static_cast<int>(offset_first_element), stride_y … in update_window_if_needed()
99 const size_t offset_first_element = _info->offset_first_element_in_bytes(); in update_window_if_needed() local113 const int front_pad_y_available = -offset_first_element / strides[1]; in update_window_if_needed()156 …const int front_pad_x_available = -std::min<int>(static_cast<int>(offset_first_element) - front_pa… in update_window_if_needed()
117 const size_t offset_first_element = _info->offset_first_element_in_bytes(); in update_window_if_needed() local130 const int front_pad_y_available = -static_cast<int>(offset_first_element / strides[1]); in update_window_if_needed()175 …const int front_pad_x_available = -std::min<int>(static_cast<int>(offset_first_element) - front_pa… in update_window_if_needed()