Searched refs:IsTensorMathEnabled (Results 1 – 1 of 1) sorted by relevance
916 static bool IsTensorMathEnabled(Stream* stream, dnn::DataType input_type) { in IsTensorMathEnabled() function3039 if (use_tensor_ops && !IsTensorMathEnabled(stream, type)) { in UseTensorOps()3044 use_tensor_ops = IsTensorMathEnabled(stream, type); in UseTensorOps()4603 /*disable_tensor_core*/ !IsTensorMathEnabled(stream, input_type)); in CreateOpRunners()