Searched refs:IsTensorMathOpSet (Results 1 – 1 of 1) sorted by relevance
745 static bool IsTensorMathOpSet(const CudnnConvolutionDescriptor& conv) { in IsTensorMathOpSet() function2541 if (IsTensorMathOpSet(conv) != algorithm_desc.tensor_ops_enabled()) { in AllocateCudnnConvolutionForwardWorkspace()2585 if (IsTensorMathOpSet(conv) != algorithm_desc.tensor_ops_enabled()) { in AllocateCudnnConvolutionBackwardDataWorkspace()2631 if (IsTensorMathOpSet(conv) != algorithm_desc.tensor_ops_enabled()) { in AllocateCudnnConvolutionBackwardFilterWorkspace()3262 if (IsTensorMathOpSet(conv) != algo_desc.tensor_ops_enabled()) { in DoFusedConvolveImpl()