Searched defs:out_batch_stride (Results 1 – 3 of 3) sorted by relevance
152 int out_batch_stride, in nchw_to_nhwc()272 int out_batch_stride, in nchw_to_nhwc()297 int out_batch_stride, in nchw_to_nhwc()325 int out_batch_stride, in nchw_to_nhwc()382 int out_batch_stride, in nhwc_to_nchw()512 int out_batch_stride, in nhwc_to_nchw()538 int out_batch_stride, in nhwc_to_nchw()567 int out_batch_stride, in nhwc_to_nchw()
197 const int out_batch_stride = dst->info()->strides_in_bytes()[3] / sizeof(T); in run_permute() local214 const int out_batch_stride = dst->info()->strides_in_bytes()[3] / sizeof(T); in run_permute() local
90 const size_t out_batch_stride = dst_strides[batch_idx] / element_size_in_bytes; in run_op() local