Searched refs:include_complex (Results 1 – 5 of 5) sorted by relevance
| /aosp_15_r20/external/pytorch/test/ |
| H A D | test_scatter_gather_ops.py | 224 …@dtypesIfCUDA(*get_all_dtypes(include_half=True, include_bfloat16=True, include_complex=False, inc…
232 …@dtypesIfCUDA(*get_all_dtypes(include_half=True, include_bfloat16=True, include_complex=False, inc…
241 @dtypes(*get_all_dtypes(include_half=True, include_bfloat16=True, include_complex=False))
242 …@dtypesIfCUDA(*get_all_dtypes(include_half=True, include_bfloat16=True, include_complex=False, inc…
260 @dtypes(*get_all_dtypes(include_half=True, include_bfloat16=True, include_complex=False))
261 …@dtypesIfCUDA(*get_all_dtypes(include_half=True, include_bfloat16=True, include_complex=False, inc…
|
| /aosp_15_r20/external/pytorch/torch/testing/_internal/ |
| H A D | common_dtype.py | 136 include_complex=True, argument
145 if include_complex:
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/kernel_tests/math_ops/ |
| H A D | cwise_ops_binary_test.py | 924 def _testBCastByFunc(self, np_func, tf_func, include_complex=False): argument
945 if include_complex:
965 self._testBCastByFunc(np.equal, math_ops.equal, include_complex=True)
969 np.not_equal, math_ops.not_equal, include_complex=True)
|
| H A D | cwise_ops_test.py | 157 def _testBCastByFunc(self, np_func, tf_func, include_complex=False): argument
178 if include_complex:
199 self._testBCastByFunc(np.equal, math_ops.equal, include_complex=True)
203 np.not_equal, math_ops.not_equal, include_complex=True)
|
| /aosp_15_r20/external/pytorch/test/inductor/ |
| H A D | test_torchinductor.py | 1634 include_complex=False,
|