Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/python/distribute/
H A Dstrategy_combinations.py78 enable_packed_variable=False, argument
125 if enable_packed_variable and enable_spmd_xla_paritioning:
127 …strategy._enable_packed_variable_in_eager_mode = enable_packed_variable # pylint: disable=protect…
359 _get_tpu_strategy_creator(steps_per_run=2, enable_packed_variable=True),