Home
last modified time | relevance | path

Searched defs:sharding_tree (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/
H A Dhlo_sharding_metadata.cc256 ShapeTree<HloSharding> sharding_tree( in ApplyShardingFromUsers() local
H A Dhlo_sharding_test.cc431 ShapeTree<HloSharding> sharding_tree(tuple_shape, HloSharding::Replicate()); in TEST_P() local