Searched defs:dtype1 (Results 1 – 5 of 5) sorted by relevance
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/ |
| H A D | ReduceOpsUtils.h | 227 at::OptionalIntArrayRef dim_opt, bool keepdim, ScalarType dtype1, in make_reduction()
421 ScalarType dtype1, in make_reduction() argument
|
| /aosp_15_r20/external/pytorch/test/torch_np/numpy_tests/lib/ |
| H A D | test_arraysetops.py | 474 def test_in1d_mixed_dtype(self, dtype1, dtype2, kind): argument
|
| /aosp_15_r20/external/pytorch/test/ |
| H A D | test_decomp.py | 173 def _getDefaultRtolAndAtol(dtype0, dtype1): argument
|
| H A D | test_type_promotion.py | 816 def _test_sparse_op(self, op_name, inplace, dtype1, dtype2, device, coalesced): argument
|
| /aosp_15_r20/external/pytorch/benchmarks/static_runtime/ |
| H A D | test_generated_ops.cc | 1112 auto dtype1 = at::ScalarType::Float; in TEST() local
2958 auto dtype1 = at::ScalarType::Float; in TEST() local
2994 auto dtype1 = at::ScalarType::Float; in TEST() local
|