models/official/vision/modeling/heads
..
__init__.py
dense_prediction_heads.py
dense_prediction_heads_test.py
instance_heads.py
instance_heads_test.py
segmentation_heads.py
segmentation_heads_test.py