Lines Matching refs:src_height
191 const unsigned int src_height = src.shape().y(); in compute_reduction_operation() local
217 const int in_offset = du * src_height * src_width + x; in compute_reduction_operation()
234 for(unsigned int y = 0; y < src_height; ++y) in compute_reduction_operation()
236 … const int in_offset = du * src_depth * src_height * src_width + y * src_width + x; in compute_reduction_operation()
237 const int out_offset = du * src_width * src_height + y * src_width + x; in compute_reduction_operation()
240 … reduce_operation_arg_min_max<T, OT>(src_row_ptr, reduce_elems, op, src_width * src_height) : in compute_reduction_operation()
241 … reduce_operation<T, OT>(src_row_ptr, reduce_elems, op, src_width * src_height, policy); in compute_reduction_operation()
254 for(unsigned int y = 0; y < src_height; ++y) in compute_reduction_operation()
258 …nst int in_offset = du * src_batch * src_depth * src_height * src_width + z * src_width * src_he… in compute_reduction_operation()
259 …const int out_offset = du * src_depth * src_height * src_width + z * src_width * src_height + y *… in compute_reduction_operation()
262 …e_operation_arg_min_max<T, OT>(src_row_ptr, reduce_elems, op, src_width * src_height * src_depth) : in compute_reduction_operation()
263 …reduce_operation<T, OT>(src_row_ptr, reduce_elems, op, src_width * src_height * src_depth, policy); in compute_reduction_operation()