Home
last modified time | relevance | path

Searched refs:RemoveShapeDimensions (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_sharding_util.h195 HloSharding RemoveShapeDimensions(const HloSharding& sharding,
Dsharding_propagation.cc729 get_maybe_tuple_sharding(hlo_sharding_util::RemoveShapeDimensions( in InferShardingFromOperands()
1154 return hlo_sharding_util::RemoveShapeDimensions( in GetShardingFromUser()
Dhlo_sharding_util.cc1184 HloSharding RemoveShapeDimensions(const HloSharding& sharding, in RemoveShapeDimensions() function
/external/tensorflow/tensorflow/compiler/xla/service/spmd/
Dspmd_partitioner.cc2251 auto desired_input_sharding = hlo_sharding_util::RemoveShapeDimensions( in HandleBroadcast()
2800 auto sharding = hlo_sharding_util::RemoveShapeDimensions( in HandleReduce()