Searched refs:supported_dtypes (Results 1 – 1 of 1) sorted by relevance
862 def check_type_support(dtype, device, supported_dtypes): argument863 return dtype in supported_dtypes or not context.get_context('device_target') == device