Add Support For Choosing Input Type

[ctodTT]

Currently, in the decorator defined in python/test_infra/test_utils.py does not have a mechanism for deciding what type of tensors should be passed into the test functions, and defaults to f32. This needs to be changed to support ops that require non-f32 inputs. (e.g. bitwise ops and where)