| 153 | } |
| 154 | |
| 155 | std::vector<CalculationsPrecision> Environment::GetSupportedPrecisions() const { |
| 156 | std::vector<CalculationsPrecision> precisions; |
| 157 | for (CalculationsPrecision precision : |
| 158 | {CalculationsPrecision::F32, CalculationsPrecision::F32_F16, |
| 159 | CalculationsPrecision::F16}) { |
| 160 | if (IsSupported(precision)) { |
| 161 | precisions.push_back(precision); |
| 162 | } |
| 163 | } |
| 164 | return precisions; |
| 165 | } |
| 166 | |
| 167 | bool Environment::IsSupported(CalculationsPrecision precision) const { |
| 168 | switch (precision) { |