Home
last modified time | relevance | path

Searched defs:dtype0 (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/
H A DJitLoops.cuh107 const auto dtype0 = iter.dtype(0); in jitted_gpu_kernel() local
/aosp_15_r20/external/pytorch/test/
H A Dtest_decomp.py173 def _getDefaultRtolAndAtol(dtype0, dtype1): argument
/aosp_15_r20/external/pytorch/benchmarks/static_runtime/
H A Dtest_generated_ops.cc1100 auto dtype0 = at::ScalarType::Float; in TEST() local
2947 auto dtype0 = at::ScalarType::Float; in TEST() local
2981 auto dtype0 = at::ScalarType::Float; in TEST() local