Searched refs:enable_packed_variable (Results 1 – 1 of 1) sorted by relevance
78 enable_packed_variable=False, argument125 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),