Most of the layers in Tensorflow are the reverse of the Pytorch layers, except the depthwise layers. Those are (2,3,0,1). For example:
Tensorflow: expanded_conv_depthwise/depthwise_kernel:0: (3, 3, 32, 1)
Pytorch: features.1.conv.0.0.weight torch.Size([32,1,3,3])