Home
last modified time | relevance | path

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

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