Searched refs:set_use_tensor_op_math (Results 1 – 1 of 1) sorted by relevance
725 void set_use_tensor_op_math(bool use_tensor_op_math) { in set_use_tensor_op_math() function in stream_executor::gpu::__anon040dea3a0311::CudnnConvolutionDescriptor2703 conv.set_use_tensor_op_math(use_tensor_ops); in GetCudnnConvolutionForwardAlgorithm()2743 conv.set_use_tensor_op_math(use_tensor_ops); in GetCudnnConvolutionForwardAlgorithm()2765 conv.set_use_tensor_op_math(use_tensor_ops); in GetCudnnConvolutionBackwardDataAlgorithm()2804 conv.set_use_tensor_op_math(use_tensor_ops); in GetCudnnConvolutionBackwardDataAlgorithm()2826 conv.set_use_tensor_op_math(use_tensor_ops); in GetCudnnConvolutionBackwardFilterAlgorithm()2865 conv.set_use_tensor_op_math(use_tensor_ops); in GetCudnnConvolutionBackwardFilterAlgorithm()3058 conv.set_use_tensor_op_math(use_tensor_ops); in DoConvolve()3223 conv.set_use_tensor_op_math(algo_desc.tensor_ops_enabled()); in DoFusedConvolveImpl()