Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/pjrt/
Dcompile_options.proto38 bool use_spmd_partitioning = 6; field
/external/tensorflow/tensorflow/compiler/xla/client/
Dexecutable_build_options.cc75 bool use_spmd_partitioning) { in set_use_spmd_partitioning()
Dexecutable_build_options.h91 bool use_spmd_partitioning() const { return use_spmd_partitioning_; } in use_spmd_partitioning() function
/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_module_config.h153 void set_use_spmd_partitioning(bool use_spmd_partitioning) { in set_use_spmd_partitioning()
156 bool use_spmd_partitioning() const { return use_spmd_partitioning_; } in use_spmd_partitioning() function
Dcompiler.h92 virtual bool use_spmd_partitioning() const { return false; } in use_spmd_partitioning() function
/external/tensorflow/tensorflow/compiler/xla/
Dxla.proto491 bool use_spmd_partitioning = 11; field
/external/tensorflow/tensorflow/compiler/xla/stream_executor/tpu/
Dc_api_decl.h308 bool use_spmd_partitioning; member
/external/tensorflow/tensorflow/compiler/xla/pjrt/c/
Dpjrt_c_api.h326 bool use_spmd_partitioning; member