Searched defs:out_row (Results 1 – 4 of 4) sorted by relevance
96 auto terminate_slice = [&](uint32_t out_row, int64_t end_ts) { in ComputeFlatSliceTable()104 uint32_t out_row = std::numeric_limits<uint32_t>::max(); in ComputeFlatSliceTable() member
180 T* const out_row = out_batch + i*out_row_stride; in nchw_to_nhwc() local351 T* const out_row = out_batch + i*out_row_stride; in nchw_to_nhwc() local659 T* out_row = out + i * out_row_stride; in ofm_ifm_h_w_to_h_w_ifm_ofm() local726 T* const out_row = out + i * out_row_stride; in h_w_ifm_ofm_to_ofm_ifm_h_w() local
16 auto* out_row = out + offsets[0]; in cross_kernel() local
56 const int out_row = (index / oW) % oH; in conv_depthwise3d_cuda_kernel() local152 const int out_row = out_row_raw / strideH; in conv_depthwise3d_cuda_backward_input_kernel() local