Searched refs:use_tensor_ops (Results 1 – 2 of 2) sorted by relevance
1596 bool use_tensor_ops = false; in DoBlasGemm() local1604 use_tensor_ops = true; in DoBlasGemm()1610 true /* = err_on_failure= */, use_tensor_ops, CUDABlasTranspose(transa), in DoBlasGemm()2217 bool use_tensor_ops = TensorOpMathEnabled() && data_type == CUDA_R_16F; in DoBlasGemmBatchedInternal() local2219 (use_tensor_ops ? CUBLAS_GEMM_DFALT_TENSOR_OP : CUBLAS_GEMM_DFALT); in DoBlasGemmBatchedInternal()2231 true /* = err_on_failure */, use_tensor_ops, CUDABlasTranspose(transa), in DoBlasGemmBatchedInternal()2364 bool use_tensor_ops = false; in DoBlasGemmStridedBatched() local2371 use_tensor_ops = true; in DoBlasGemmStridedBatched()2376 (use_tensor_ops ? CUBLAS_GEMM_DFALT_TENSOR_OP : CUBLAS_GEMM_DFALT); in DoBlasGemmStridedBatched()2379 true /* = err_on_failure */, use_tensor_ops, in DoBlasGemmStridedBatched()[all …]
742 AlgorithmDesc(Index a, bool use_tensor_ops) {744 proto_.set_math_type(use_tensor_ops ? AlgorithmProto::TENSOR_OP_MATH