Searched defs:value_container (Results 1 – 6 of 6) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/python/distribute/ | ||
| H A D | distribute_utils.py | 199 def value_container(val): function |
| H A D | one_device_strategy.py | 425 def value_container(self, value): member in OneDeviceExtended |
| H A D | parameter_server_strategy.py | 568 def value_container(self, val): member in ParameterServerStrategyExtended |
| H A D | mirrored_strategy.py | 847 def value_container(self, val): member in MirroredExtended |
| H A D | distribute_lib.py | 2662 def value_container(self, value): member in StrategyExtendedV2 |
| H A D | tpu_strategy.py | 1435 def value_container(self, value): member in TPUExtended |