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

Method dst_offset

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

Source from the content-addressed store, hash-verified

800}
801
802size_t ConvBiasImpl::NCBKernParam::dst_offset(
803 size_t batch_id, size_t group_pack_id, size_t channel_pack_id,
804 size_t group_pack_size, size_t channel_pack_size) const {
805 size_t batch_offset = batch_id * out_bs * dst_type.size();
806 size_t group_offset = group_pack_size * group_pack_id * filter_meta.ocpg * osz[0] *
807 osz[1] * dst_type.size();
808 size_t channel_offset =
809 channel_pack_size * channel_pack_id * osz[0] * osz[1] * dst_type.size();
810 return (batch_offset + group_offset + channel_offset);
811}
812
813template <typename T>
814T* ConvBiasImpl::NCBKernParam::dst(

Callers 2

dispatch_kernsMethod · 0.80
dispatch_kernsMethod · 0.80

Calls 1

sizeMethod · 0.45

Tested by

no test coverage detected