Home
last modified time | relevance | path

Searched defs:use_auto_spmd_partitioning (Results 1 – 9 of 9) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/
H A Dhlo_module_config.h158 void set_use_auto_spmd_partitioning(bool use_auto_spmd_partitioning) { in set_use_auto_spmd_partitioning()
171 bool use_auto_spmd_partitioning() const { in use_auto_spmd_partitioning() function
H A Dcompiler.h93 virtual bool use_auto_spmd_partitioning() const { return false; } in use_auto_spmd_partitioning() function
H A Dhlo_module.h160 bool use_auto_spmd_partitioning() const { in use_auto_spmd_partitioning() function
H A Dhlo.proto495 bool use_auto_spmd_partitioning = 16; field
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/pjrt/
H A Dcompile_options.proto41 bool use_auto_spmd_partitioning = 7; field
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/client/
H A Dexecutable_build_options.cc81 bool use_auto_spmd_partitioning) { in set_use_auto_spmd_partitioning()
H A Dexecutable_build_options.h95 bool use_auto_spmd_partitioning() const { in use_auto_spmd_partitioning() function
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/
H A Dxla.proto494 bool use_auto_spmd_partitioning = 15; field
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/tpu/
H A Dc_api_decl.h309 bool use_auto_spmd_partitioning; member