Searched refs:multi_stride (Results 1 – 2 of 2) sorted by relevance
223 …e_B_array_generic(void *out, const void *in, const int row_stride, const int multi_stride) override in pretranspose_B_array_generic() argument225 pretranspose_B_array(out, static_cast<const To *>(in), row_stride, multi_stride); in pretranspose_B_array_generic()
256 const size_t multi_stride = multi_size / sizeof(TypeInput); in prepare_indirect_buffer() local279 …_indirect_buf.get()[m * multi_stride + b * batch_stride + kernel_xy * output_hw + output_xy] = _in… in prepare_indirect_buffer()283 … _indirect_buf.get()[m * multi_stride + b * batch_stride + kernel_xy * output_hw + output_xy] = in prepare_indirect_buffer()333 const size_t multi_stride = multi_size / sizeof(TypeInputPtr); in configure_indirect() local347 …(_indirect_arg.get())[pos++] = _indirect_buf.get() + m * multi_stride + b * batch_stride + kernel_… in configure_indirect()