Searched refs:use_tensor_ops (Results 1 – 2 of 2) sorted by relevance
1219 bool use_tensor_ops = in Create() local1223 if (use_tensor_ops && !allow_tensor_ops) { in Create()1230 use_tensor_ops ? CUDNN_TENSOR_OP_MATH : CUDNN_FMA_MATH; in Create()1233 use_tensor_ops ? CUDNN_TENSOR_OP_MATH : CUDNN_DEFAULT_MATH; in Create()2963 bool use_tensor_ops; in UseTensorOps() local2965 use_tensor_ops = desc->tensor_ops_enabled(); in UseTensorOps()2966 if (use_tensor_ops && !IsTensorMathEnabled(stream, type)) { in UseTensorOps()2971 use_tensor_ops = IsTensorMathEnabled(stream, type); in UseTensorOps()2973 return use_tensor_ops; in UseTensorOps()2992 bool use_tensor_ops; in GetCudnnConvolutionForwardAlgorithm() local[all …]
787 AlgorithmDesc(Index a, bool use_tensor_ops) { in AlgorithmDesc() argument789 proto_.set_math_type(use_tensor_ops ? AlgorithmProto::TENSOR_OP_MATH in AlgorithmDesc()