Searched refs:NumericalCheck (Results 1 – 2 of 2) sorted by relevance
| /external/pytorch/aten/src/ATen/cuda/tunable/ |
| D | GemmCommon.h | 46 static bool NumericalCheck(ScalarType dtype, void* c, void* other_c, int64_t size) { in NumericalCheck() function
136 TuningStatus NumericalCheck(GemmParams<T> *other) { in NumericalCheck() function
138 return detail::NumericalCheck(c_dtype, c, other->c, ldc*n) ? OK : FAIL; in NumericalCheck()
217 TuningStatus NumericalCheck(GemmStridedBatchedParams<T> *other) { in NumericalCheck() function
219 return detail::NumericalCheck(c_dtype, c, other->c, batch*stride_c) ? OK : FAIL; in NumericalCheck()
302 TuningStatus NumericalCheck(ScaledGemmParams<T> *other) { in NumericalCheck() function
303 return detail::NumericalCheck(c_dtype, c, other->c, ldc*n) ? OK : FAIL; in NumericalCheck()
|
| D | TunableOp.h | 170 status = reference_params->NumericalCheck(numerical_params); in FindFastest()
|