MCPcopy Create free account
hub / github.com/MegEngine/MegEngine / src_offset

Method src_offset

dnn/src/fallback/conv_bias/opr_impl.cpp:683–692  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

681namespace fallback {
682
683size_t ConvBiasImpl::NCBKernParam::src_offset(
684 size_t batch_id, size_t group_pack_id, size_t channel_pack_id,
685 size_t group_pack_size, size_t channel_pack_size) const {
686 size_t batch_offset = batch_id * inp_bs * src_type.size();
687 size_t group_offset = group_pack_size * group_pack_id * filter_meta.icpg * isz[0] *
688 isz[1] * src_type.size();
689 size_t channel_offset =
690 channel_pack_size * channel_pack_id * isz[0] * isz[1] * src_type.size();
691 return (batch_offset + group_offset + channel_offset);
692}
693
694template <typename T>
695const T* ConvBiasImpl::NCBKernParam::src(

Callers 2

dispatch_kernsMethod · 0.80
dispatch_kernsMethod · 0.80

Calls 1

sizeMethod · 0.45

Tested by

no test coverage detected