Skip to content

Instantly share code, notes, and snippets.

@trevor-m
Created January 7, 2021 22:52
Show Gist options
  • Star 0 You must be signed in to star a gist
  • Fork 0 You must be signed in to fork a gist
  • Save trevor-m/a32ac7010a148045bb26b5ce22566220 to your computer and use it in GitHub Desktop.
Save trevor-m/a32ac7010a148045bb26b5ce22566220 to your computer and use it in GitHub Desktop.
Slow Conv2ds
%1310 = /* ty=Tensor[(100, 2048, 33, 33), float32] */;
%1311 = nn.conv2d(%1310, meta[relay.Constant][139] /* ty=Tensor[(256, 2048, 3, 3), float32] */, padding=[1, 1, 1, 1], channels=256, kernel_size=[3, 3]) /* ty=Tensor[(100, 256, 33, 33), float32] */;
%1312 = add(%1311, meta[relay.Constant][140] /* ty=Tensor[(1, 256, 1, 1), float32] */) /* ty=Tensor[(100, 256, 33, 33), float32] */;
%1313 = nn.relu(%1312) /* ty=Tensor[(100, 256, 33, 33), float32] */;
%1314 = nn.conv2d(%1313, meta[relay.Constant][141] /* ty=Tensor[(256, 256, 3, 3), float32] */, padding=[1, 1, 1, 1], channels=256, kernel_size=[3, 3]) /* ty=Tensor[(100, 256, 33, 33), float32] */;
%1315 = add(%1314, meta[relay.Constant][142] /* ty=Tensor[(1, 256, 1, 1), float32] */) /* ty=Tensor[(100, 256, 33, 33), float32] */;
%1316 = nn.relu(%1315) /* ty=Tensor[(100, 256, 33, 33), float32] */;
%1317 = nn.conv2d(%1316, meta[relay.Constant][143] /* ty=Tensor[(256, 256, 3, 3), float32] */, padding=[1, 1, 1, 1], channels=256, kernel_size=[3, 3]) /* ty=Tensor[(100, 256, 33, 33), float32] */;
%1318 = add(%1317, meta[relay.Constant][144] /* ty=Tensor[(1, 256, 1, 1), float32] */) /* ty=Tensor[(100, 256, 33, 33), float32] */;
%1319 = nn.relu(%1318) /* ty=Tensor[(100, 256, 33, 33), float32] */;
%1320 = nn.conv2d(%1319, meta[relay.Constant][145] /* ty=Tensor[(90, 256, 3, 3), float32] */, padding=[1, 1, 1, 1], channels=90, kernel_size=[3, 3]) /* ty=Tensor[(100, 90, 33, 33), float32] */;
%1321 = add(%1320, meta[relay.Constant][146] /* ty=Tensor[(1, 90, 1, 1), float32] */) /* ty=Tensor[(100, 90, 33, 33), float32] */;
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment