functional_hlo_runner: read entry_computation_layout from HloModule
functional_hlo_runner: read entry_computation_layout from HloModule

Instead of reading it from CompilationOptions; the compiled HloModule
already contains this information.

While at it, simplify the argument_memory_space lambda -- it is
reasonable to expect parameter tuples to be flattened.
