Searched refs:TPUInitMode (Results 1 – 3 of 3) sorted by relevance
28 TPUInitMode init_mode TF_GUARDED_BY(init_mode_mutex);34 void ForceSetTPUInitMode(const TPUInitMode mode) { in ForceSetTPUInitMode()41 Status SetTPUInitMode(const TPUInitMode mode) { in SetTPUInitMode()42 if (mode == TPUInitMode::kNone) { in SetTPUInitMode()48 if (init_mode != TPUInitMode::kNone && mode != init_mode) { in SetTPUInitMode()61 TPUInitMode GetTPUInitMode() { in GetTPUInitMode()
23 enum class TPUInitMode : int { kNone, kGlobal, kRegular }; enum33 Status SetTPUInitMode(TPUInitMode mode);36 TPUInitMode GetTPUInitMode();41 void ForceSetTPUInitMode(TPUInitMode mode);
282 is_global_init ? TPUInitMode::kGlobal : TPUInitMode::kRegular)); in Run()