Home
last modified time | relevance | path

Searched defs:auto_spmd_partitioning_mesh_ids (Results 1 – 5 of 5) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/client/
H A Dexecutable_build_options.h107 std::vector<int64_t> auto_spmd_partitioning_mesh_ids() const { in auto_spmd_partitioning_mesh_ids() function
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/
H A Dcompiler.h97 virtual std::vector<int64_t> auto_spmd_partitioning_mesh_ids() const { in auto_spmd_partitioning_mesh_ids() function
H A Dhlo_module_config.h185 std::vector<int64_t> auto_spmd_partitioning_mesh_ids() const { in auto_spmd_partitioning_mesh_ids() function
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/
H A Dxla.proto502 repeated int64 auto_spmd_partitioning_mesh_ids = 17; field
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/tpu/
H A Dc_api_decl.h311 Int64List auto_spmd_partitioning_mesh_ids; member