Searched refs:IsTensorMathOpSet (Results 1 – 1 of 1) sorted by relevance
901 static bool IsTensorMathOpSet(const CudnnConvolutionDescriptor& conv) { in IsTensorMathOpSet() function2893 if (IsTensorMathOpSet(conv) != algorithm_desc.tensor_ops_enabled()) { in AllocateCudnnConvolutionForwardWorkspace()2942 if (IsTensorMathOpSet(conv) != algorithm_desc.tensor_ops_enabled()) { in AllocateCudnnConvolutionBackwardDataWorkspace()2992 if (IsTensorMathOpSet(conv) != algorithm_desc.tensor_ops_enabled()) { in AllocateCudnnConvolutionBackwardFilterWorkspace()4967 if (IsTensorMathOpSet(conv_) != tensor_ops_enabled_) { in operator ()()