Searched refs:is_global_init (Results 1 – 10 of 10) sorted by relevance
22 name: "is_global_init"51 name: "is_global_init"87 name: "is_global_init"
22 name: "is_global_init"51 name: "is_global_init"
25 name: "is_global_init"
274 bool is_global_init = false; in Run() local277 "is_global_init", &is_global_init)); in Run()282 is_global_init ? TPUInitMode::kGlobal : TPUInitMode::kRegular)); in Run()
392 key: "is_global_init"
837 …argspec: "args=[\'embedding_config\', \'tpu_embedding_config\', \'is_global_init\', \'enable_whole…
8385 name: "is_global_init"
2096 DefaultValuedAttr<BoolAttr, "false">:$is_global_init,
13810 name: "is_global_init"