Searched defs:_in_graph_mode (Results 1 – 3 of 3) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/python/distribute/ | ||
| H A D | tpu_replicated_variable.py | 136 def _in_graph_mode(self): member in TPUReplicatedVariable |
| H A D | values.py | 714 def _in_graph_mode(self): member in DistributedVariable |
| /aosp_15_r20/external/tensorflow/tensorflow/python/tpu/ | ||
| H A D | tpu_embedding_v2.py | 64 def _in_graph_mode(self): member in TPUEmbeddingVariable |