Home
last modified time | relevance | path

Searched refs:SetTPUInitMode (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/core/tpu/
Dtpu_init_mode.h33 Status SetTPUInitMode(TPUInitMode mode);
Dtpu_init_mode.cc41 Status SetTPUInitMode(const TPUInitMode mode) { in SetTPUInitMode() function
/external/tensorflow/tensorflow/core/tpu/graph_rewrite/
Ddistributed_tpu_configuration_rewrite_pass.cc281 TF_RETURN_IF_ERROR(SetTPUInitMode( in Run()