Home
last modified time | relevance | path

Searched refs:allow_spmd_sharding_propagation_to_output_ (Results 1 – 5 of 5) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/client/
H A Dexecutable_build_options.h147 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
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/
H A Dhlo_module_config.h314 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
H A Dsharding_propagation.h61 allow_spmd_sharding_propagation_to_output_( in is_spmd_()
95 bool allow_spmd_sharding_propagation_to_output_; variable
H A Dhlo_module_config.cc80 allow_spmd_sharding_propagation_to_output_); in compilation_cache_key()
H A Dsharding_propagation.cc1739 allow_spmd_sharding_propagation_to_output_, in InferShardingFromOperands()
2439 if (!allow_spmd_sharding_propagation_to_output_) { in Run()