Searched refs:set_use_tensor_op_math (Results 1 – 1 of 1) sorted by relevance
683 this->set_use_tensor_op_math(true); in CudnnConvolutionDescriptor()696 void set_use_tensor_op_math(bool use_tensor_op_math) const { in set_use_tensor_op_math() function in stream_executor::gpu::__anonf41f84630311::CudnnConvolutionDescriptor2219 conv.set_use_tensor_op_math(algorithm_desc.tensor_ops_enabled()); in AllocateCudnnConvolutionForwardWorkspace()2262 conv.set_use_tensor_op_math(algorithm_desc.tensor_ops_enabled()); in AllocateCudnnConvolutionBackwardDataWorkspace()2307 conv.set_use_tensor_op_math(algorithm_desc.tensor_ops_enabled()); in AllocateCudnnConvolutionBackwardFilterWorkspace()