Searched refs:allow_spmd_sharding_propagation_to_output_ (Results 1 – 5 of 5) sorted by relevance
147 return allow_spmd_sharding_propagation_to_output_; in allow_spmd_sharding_propagation_to_output()158 allow_spmd_sharding_propagation_to_output_ = in set_allow_spmd_sharding_propagation_to_output()192 bool allow_spmd_sharding_propagation_to_output_ = false; variable
61 allow_spmd_sharding_propagation_to_output_( in is_spmd_()95 bool allow_spmd_sharding_propagation_to_output_; variable
314 allow_spmd_sharding_propagation_to_output_ = in set_allow_spmd_sharding_propagation_to_output()318 return allow_spmd_sharding_propagation_to_output_; in allow_spmd_sharding_propagation_to_output()446 bool allow_spmd_sharding_propagation_to_output_ = false; variable
80 allow_spmd_sharding_propagation_to_output_); in compilation_cache_key()
1739 allow_spmd_sharding_propagation_to_output_, in InferShardingFromOperands()2439 if (!allow_spmd_sharding_propagation_to_output_) { in Run()