Searched refs:math_type (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cuda_blas.cc | 534 cublasMath_t math_type, Args... args) { in DoBlasInternalImpl() argument
545 if (math_type == CUBLAS_TF32_TENSOR_OP_MATH && in DoBlasInternalImpl()
548 if (math_type == CUBLAS_TENSOR_OP_MATH) { in DoBlasInternalImpl()
550 if (!math_mode.Init(math_type)) { in DoBlasInternalImpl()
1775 cublasMath_t math_type = CUBLAS_TENSOR_OP_MATH; in DoBlasInternalImpl() local
1777 cublasMath_t math_type = CUBLAS_DEFAULT_MATH; in DoBlasInternalImpl() local
1782 true /* = err_on_failure= */, math_type, CUDABlasTranspose(transa), in DoBlasInternalImpl()
1829 cublasMath_t math_type = CUBLAS_DEFAULT_MATH; in DoBlasInternalImpl() local
1831 cublasMath_t math_type = CUBLAS_TF32_TENSOR_OP_MATH; in DoBlasInternalImpl() local
1845 true /* = err_on_failure */, math_type, CUDABlasTranspose(transa), in DoBlasInternalImpl()
[all …]
|
| D | cuda_blas.h | 92 cublasMath_t math_type, Args... args);
|
| D | cuda_dnn.cc | 726 cudnnMathType_t math_type = in set_use_tensor_op_math() local
732 CHECK_CUDNN_OK(cudnnSetConvolutionMathType(handle_.get(), math_type)); in set_use_tensor_op_math()
746 cudnnMathType_t math_type; in IsTensorMathOpSet() local
747 CHECK_CUDNN_OK(cudnnGetConvolutionMathType(conv.handle(), &math_type)); in IsTensorMathOpSet()
749 return math_type != CUDNN_FMA_MATH; in IsTensorMathOpSet()
751 return math_type == CUDNN_TENSOR_OP_MATH; in IsTensorMathOpSet()
1127 cudnnMathType_t math_type = in Create() local
1130 cudnnMathType_t math_type = in Create() local
1143 /*mathType=*/math_type, in Create()
1156 CHECK_CUDNN_OK(cudnnSetRNNMatrixMathType(rnn_desc.get(), math_type)); in Create()
|
| /external/tensorflow/tensorflow/stream_executor/ |
| D | dnn.proto | 100 MathType math_type = 2; field
|
| D | dnn.h | 752 return proto_.math_type() == AlgorithmProto::TENSOR_OP_MATH; in tensor_ops_enabled()
|
| /external/lua/src/ |
| D | lmathlib.c | 233 static int math_type (lua_State *L) { in math_type() function
725 {"type", math_type},
|