Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/cuda/
H A 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
H A 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
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/
H A Ddnn.proto115 MathType math_type = 2; field
/aosp_15_r20/external/pytorch/aten/src/ATen/cudnn/
H A DDescriptors.h315 auto math_type = CUDNN_DEFAULT_MATH; in set() local
/aosp_15_r20/external/lua/src/
H A Dlmathlib.c233 static int math_type (lua_State *L) { in math_type() function
/aosp_15_r20/external/pytorch/aten/src/ATen/native/cudnn/
H A DConv_v7.cpp685 #define ASSERT_CORRECT_PRECISION(math_type) \ argument
/aosp_15_r20/out/soong/.intermediates/external/tensorflow/tensorflow_core_proto_cpp_lite/android_arm64_armv8-2a_cortex-a55_static_apex33/gen/proto/tensorflow/compiler/xla/stream_executor/
Ddnn.pb.h1374 inline ::stream_executor::dnn::AlgorithmProto_MathType AlgorithmProto::math_type() const { in math_type() function