Searched refs:TensorOpMathAvailable (Results 1 – 1 of 1) sorted by relevance
3038 inline bool TensorOpMathAvailable(int cc_major) { in TensorOpMathAvailable() function3045 bool tensor_op_math_available = TensorOpMathAvailable(cc_major); in GetConvolveAlgorithms()3106 bool tensor_op_math_available = TensorOpMathAvailable(cc_major); in GetConvolveBackwardDataAlgorithms()3141 bool tensor_op_math_available = TensorOpMathAvailable(cc_major); in GetConvolveBackwardFilterAlgorithms()