CollectiveAllReduceExtended (definition) in projects: aosp_15_r20

Searched defs:CollectiveAllReduceExtended (Results 1 – 4 of 4) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/python/distribute/
H A D	multi_worker_continuous_run_test.py	`43 CollectiveAllReduceExtended = ( variable`
H A D	strategy_combinations.py	`49 CollectiveAllReduceExtended = ( variable`
H A D	collective_all_reduce_strategy.py	`306 class CollectiveAllReduceExtended(mirrored_strategy.MirroredExtended): class`
H A D	collective_all_reduce_strategy_test.py	`63 CollectiveAllReduceExtended = ( variable`