neuralpredictors/layers/readouts/attention.py
neuralpredictors/layers/readouts/base.py
test/layers/cores/test_conv2d.py
