Searched defs:is_spmd (Results 1 – 2 of 2) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/ |
| H A D | sharding_propagation.cc | 323 const ShardingPropagation::ComputationMap& computation_map, bool is_spmd, in SupportSpatialPartitioning()
720 int64_t aggressiveness, bool is_spmd, in InferShardingFromUsers()
1101 int64_t aggressiveness, bool is_spmd, in InferUnspecifiedDimsFromOneUser()
1153 int64_t aggressiveness, bool is_spmd, in InferUnspecifiedDimsFromUsers()
1214 const HloInstruction& instruction, bool is_spmd) { in InferBroadcastOperandSharding()
1319 int64_t aggressiveness, bool is_spmd) { in GetShardingFromUser()
|
| H A D | sharding_propagation.h | 59 : is_spmd_(is_spmd), in is_spmd_() argument
|