message: [XLA] Use entry_computation's shape in input_output_aliasing_config.FromProto
  input_output_aliasing_config is conmputed using entry_computation's shape, which is on device shape.
  the result shape is host shape, which could be different from the on device shape.
diff: @@ -330,9 +330,10 @@ StatusOr<std::unique_ptr<HloModule>> HloModule::CreateFromProto(
  }
  TF_RET_CHECK(module->entry_computation_ != nullptr);
  - TF_ASSIGN_OR_RETURN(module->input_output_alias_config_,
  + TF_ASSIGN_OR_RETURN(
  + module->input_output_alias_config_,
  HloInputOutputAliasConfig::CreateFromProto(
  - result_shape, proto.input_output_alias()));
  + entry->ComputeProgramShape().result(), proto.input_output_alias()));
  // Because we didn't uniquify the names or the ids, double-check that the
  // instruction and computation names and ids are unique from the proto.
  