Home
last modified time | relevance | path

Searched refs:AutoMixedPrecisionMode (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/core/grappler/optimizers/
Dauto_mixed_precision.h25 enum class AutoMixedPrecisionMode { CUDA, MKL }; enum
35 AutoMixedPrecisionMode mode = AutoMixedPrecisionMode::CUDA)
41 return mode_ == AutoMixedPrecisionMode::CUDA ? "auto_mixed_precision_cuda" in name()
54 const AutoMixedPrecisionMode mode_;
Dauto_mixed_precision.cc931 AutoMixedPrecisionMode mode) in AutoMixedPrecisionImpl()
941 target_dtype_(mode_ == AutoMixedPrecisionMode::CUDA ? DT_HALF in AutoMixedPrecisionImpl()
951 case AutoMixedPrecisionMode::CUDA: in get_mixed_precision_lists()
954 case AutoMixedPrecisionMode::MKL: in get_mixed_precision_lists()
1009 AutoMixedPrecisionMode mode_;
1258 if (force_all_fp16_ && mode_ == AutoMixedPrecisionMode::MKL) { in Optimize()
1282 case AutoMixedPrecisionMode::CUDA: in Optimize()
1287 case AutoMixedPrecisionMode::MKL: in Optimize()
1990 if (mode_ == AutoMixedPrecisionMode::MKL) { in Optimize()
2006 if (num_gpus < 1 && mode_ == AutoMixedPrecisionMode::CUDA) { in Optimize()
Dmeta_optimizer.cc200 new AutoMixedPrecision(AutoMixedPrecisionMode::CUDA)); in MakeNewOptimizer()
202 new AutoMixedPrecision(AutoMixedPrecisionMode::MKL)); in MakeNewOptimizer()
265 MakeUnique<AutoMixedPrecision>(AutoMixedPrecisionMode::CUDA)); in InitializeOptimizers()
269 MakeUnique<AutoMixedPrecision>(AutoMixedPrecisionMode::MKL)); in InitializeOptimizers()
Dauto_mixed_precision_test.cc1209 AutoMixedPrecision optimizer{AutoMixedPrecisionMode::MKL}; in TEST_F()
1252 AutoMixedPrecision optimizer{AutoMixedPrecisionMode::MKL}; in TEST_F()
1323 AutoMixedPrecision optimizer{AutoMixedPrecisionMode::MKL}; in TEST_F()