Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/python/distribute/
H A Dcollective_all_reduce_strategy.py392 scoped_allocator_enabled_ops=("CollectiveReduce",))
482 scoped_allocator_enabled_ops=("CollectiveReduce",),
/aosp_15_r20/external/tensorflow/tensorflow/python/eager/
H A Dcontext.py831 scoped_allocator_enabled_ops=("CollectiveReduce",), argument
856 scoped_allocator_enabled_ops or
867 self._collective_scoped_allocator_enabled_ops = scoped_allocator_enabled_ops
/aosp_15_r20/external/tensorflow/tensorflow/python/framework/
H A Dconfig_test.py840 scoped_allocator_enabled_ops=('CollectiveReduce',),