Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/
H A 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
H A Dcompiler.h92 virtual bool use_spmd_partitioning() const { return false; } in use_spmd_partitioning() function
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/pjrt/
H A Dcompile_options.proto38 bool use_spmd_partitioning = 6; field
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/client/
H A Dexecutable_build_options.cc75 bool use_spmd_partitioning) { in set_use_spmd_partitioning()
H A Dexecutable_build_options.h91 bool use_spmd_partitioning() const { return use_spmd_partitioning_; } in use_spmd_partitioning() function
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/
H A Dxla.proto491 bool use_spmd_partitioning = 11; field
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/tpu/
H A Dc_api_decl.h308 bool use_spmd_partitioning; member
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/pjrt/c/
H A Dpjrt_c_api.h326 bool use_spmd_partitioning; member