[tests/onert_train] Change the type number of loss and optimizer

[tests/onert_train] Change the type number of loss and optimizer #10802