Searched refs:scoped_allocator_enabled_ops (Results 1 – 3 of 3) sorted by relevance
392 scoped_allocator_enabled_ops=("CollectiveReduce",))482 scoped_allocator_enabled_ops=("CollectiveReduce",),
831 scoped_allocator_enabled_ops=("CollectiveReduce",), argument856 scoped_allocator_enabled_ops or867 self._collective_scoped_allocator_enabled_ops = scoped_allocator_enabled_ops
840 scoped_allocator_enabled_ops=('CollectiveReduce',),