Searched refs:IsTensorMathEnabled (Results 1 – 1 of 1) sorted by relevance
757 static bool IsTensorMathEnabled(Stream* stream, dnn::DataType input_type) { in IsTensorMathEnabled() function2674 if (use_tensor_ops && !IsTensorMathEnabled(stream, type)) { in UseTensorOps()2679 use_tensor_ops = IsTensorMathEnabled(stream, type); in UseTensorOps()