Home
last modified time | relevance | path

Searched defs:math_type (Results 1 – 6 of 6) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/stream_executor/
Ddnn.proto115 MathType math_type = 2; field
/external/pytorch/aten/src/ATen/cudnn/
DDescriptors.h315 auto math_type = CUDNN_DEFAULT_MATH; in set() local
/external/tensorflow/tensorflow/compiler/xla/stream_executor/cuda/
Dcuda_blas.cc378 cublasMath_t math_type, in DoBlasInternalImpl()
615 cublasMath_t math_type = CUBLAS_DEFAULT_MATH; in DoBlasInternalImpl() local
935 cublasMath_t math_type = CUBLAS_DEFAULT_MATH; in DoBlasInternalImpl() local
1257 cublasMath_t math_type; in DoBlasInternalImpl() local
1413 cublasMath_t math_type = CUBLAS_DEFAULT_MATH; in DoBlasInternalImpl() local
Dcuda_dnn.cc884 cudnnMathType_t math_type = in set_use_tensor_op_math() local
902 cudnnMathType_t math_type; in IsTensorMathOpSet() local
1302 cudnnMathType_t math_type = in Create() local
1305 cudnnMathType_t math_type = in Create() local
/external/lua/src/
Dlmathlib.c233 static int math_type (lua_State *L) { in math_type() function
/external/pytorch/aten/src/ATen/native/cudnn/
DConv_v7.cpp685 #define ASSERT_CORRECT_PRECISION(math_type) \ argument