Searched defs:np_dtype (Results 1 – 9 of 9) sorted by relevance
| /external/pytorch/test/onnx/exporter/ |
| D | test_core.py | 39 def test_numpy_returns_correct_dtype(self, dtype: torch.dtype, np_dtype): argument
|
| /external/pytorch/test/torch_np/ |
| D | test_dtype.py | 51 def test_convert_np_dtypes(self, name, np_dtype): argument
|
| /external/tensorflow/tensorflow/python/kernel_tests/sparse_ops/ |
| D | sparse_add_op_test.py | 49 def _randomTensor(self, size, np_dtype, sparse=True): argument
|
| D | sparse_tensor_dense_matmul_op_test.py | 214 def _testLarge(self, np_dtype): argument
|
| /external/tensorflow/tensorflow/python/kernel_tests/math_ops/ |
| D | confusion_matrix_test.py | 76 def _testConfMatrixOnTensors(self, tf_dtype, np_dtype): argument
|
| /external/tensorflow/tensorflow/python/kernel_tests/array_ops/ |
| D | array_ops_test.py | 375 def _reverse1DimAuto(self, np_dtype): argument
386 def _reverse2DimAuto(self, np_dtype): argument
532 def _compareDiffType(self, n, np_dtype, use_gpu): argument
|
| /external/pytorch/torch/_dynamo/variables/ |
| D | misc.py | 1474 dtype as np_dtype, unknown in ConstantLikeVariable
1480 np_dtype = type("invalid_type", (), {}) variable in ConstantLikeVariable
|
| /external/pytorch/torch/testing/_internal/ |
| D | common_utils.py | 1688 def numpy_to_torch_dtype(np_dtype): argument
1695 def has_corresponding_torch_dtype(np_dtype): argument
|
| /external/pytorch/test/quantization/core/ |
| D | test_quantized_op.py | 43 np_dtype = { variable
|