This will prevent unneeded include of header files, which may increase
compilation time.
Signed-off-by: NDotan Barak <dbarak@habana.ai>
Signed-off-by: NOded Gabbay <oded.gabbay@gmail.com>