Searched refs:use_tensor_ops (Results 1 – 2 of 2) sorted by relevance
1292 bool use_tensor_ops = in Create() local1296 if (use_tensor_ops && !allow_tensor_ops) { in Create()1303 use_tensor_ops ? CUDNN_TENSOR_OP_MATH : CUDNN_FMA_MATH; in Create()1306 use_tensor_ops ? CUDNN_TENSOR_OP_MATH : CUDNN_DEFAULT_MATH; in Create()3036 bool use_tensor_ops; in UseTensorOps() local3038 use_tensor_ops = desc->tensor_ops_enabled(); in UseTensorOps()3039 if (use_tensor_ops && !IsTensorMathEnabled(stream, type)) { in UseTensorOps()3044 use_tensor_ops = IsTensorMathEnabled(stream, type); in UseTensorOps()3046 return use_tensor_ops; in UseTensorOps()3065 bool use_tensor_ops; in GetCudnnConvolutionForwardAlgorithm() local[all …]
787 AlgorithmDesc(Index algo_id, bool use_tensor_ops) in AlgorithmDesc() argument788 : AlgorithmDesc(algo_id, use_tensor_ops, std::nullopt) {} in AlgorithmDesc()789 AlgorithmDesc(Index algo_id, bool use_tensor_ops, in AlgorithmDesc() argument793 proto_.set_math_type(use_tensor_ops ? AlgorithmProto::TENSOR_OP_MATH in AlgorithmDesc()