Searched refs:IsVoltaOrLater (Results 1 – 8 of 8) sorted by relevance
38 bool IsVoltaOrLater(const se::StreamExecutor& stream_exec);
120 pipeline.AddPass<CudnnPadForConvolutions>(IsVoltaOrLater(*stream_exec)); in OptimizeHloConvolutionCanonicalization()167 if (IsVoltaOrLater(*stream_exec)) { in OptimizeHloPostLayoutAssignment()
83 !IsVoltaOrLater(*stream_executor) || in HeuristicLayoutAssignment()
41 bool IsVoltaOrLater(const se::StreamExecutor& stream_executor) { in IsVoltaOrLater() function
34 inline bool IsVoltaOrLater(const se::StreamExecutor& stream_exec) { in IsVoltaOrLater() function
223 DataTypeToEnum<T>::value == DT_HALF && IsVoltaOrLater(*stream->parent()); in operator ()()
828 DataTypeToEnum<T>::value == DT_HALF && IsVoltaOrLater(*stream->parent()); in operator ()()
740 DataTypeToEnum<T>::value == DT_HALF && IsVoltaOrLater(*stream->parent()); in operator ()()