Home
last modified time | relevance | path

Searched defs:use_spmd_partitioning (Results 1 – 6 of 6) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_module_config.h143 void set_use_spmd_partitioning(bool use_spmd_partitioning) { in set_use_spmd_partitioning()
146 bool use_spmd_partitioning() const { return use_spmd_partitioning_; } in use_spmd_partitioning() function
Dcompiler.h81 virtual bool use_spmd_partitioning() const { return false; } in use_spmd_partitioning() function
/external/tensorflow/tensorflow/compiler/xla/client/
Dexecutable_build_options.cc75 bool use_spmd_partitioning) { in set_use_spmd_partitioning()
Dexecutable_build_options.h89 bool use_spmd_partitioning() const { return use_spmd_partitioning_; } in use_spmd_partitioning() function
/external/tensorflow/tensorflow/compiler/xla/
Dxla.proto375 bool use_spmd_partitioning = 11; field
/external/tensorflow/tensorflow/stream_executor/tpu/
Dc_api_decl.h281 bool use_spmd_partitioning; member