Searched refs:set_use_tensor_op_math (Results 1 – 1 of 1) sorted by relevance
883 void set_use_tensor_op_math(bool use_tensor_op_math) { in set_use_tensor_op_math() function in stream_executor::gpu::__anon96353b0e0311::CudnnConvolutionDescriptor3068 conv.set_use_tensor_op_math(use_tensor_ops); in GetCudnnConvolutionForwardAlgorithm()3108 conv.set_use_tensor_op_math(use_tensor_ops); in GetCudnnConvolutionForwardAlgorithm()3130 conv.set_use_tensor_op_math(use_tensor_ops); in GetCudnnConvolutionBackwardDataAlgorithm()3169 conv.set_use_tensor_op_math(use_tensor_ops); in GetCudnnConvolutionBackwardDataAlgorithm()3191 conv.set_use_tensor_op_math(use_tensor_ops); in GetCudnnConvolutionBackwardFilterAlgorithm()3233 conv.set_use_tensor_op_math(use_tensor_ops); in GetCudnnConvolutionBackwardFilterAlgorithm()4281 conv.set_use_tensor_op_math(use_tensor_ops); in DoConvolve()4827 conv.set_use_tensor_op_math(algorithm_desc.tensor_ops_enabled()); in ConvolveRunnerFromDesc()5088 conv.set_use_tensor_op_math(algorithm_desc.tensor_ops_enabled()); in FusedConvolveRunnerFromDesc()[all …]