Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/
H A Dhlo_module_util.cc107 std::vector<int64_t> mesh_ids; in CreateModuleConfig() local
H A Dhlo_module_config.h182 void set_auto_spmd_partitioning_mesh_ids(std::vector<int64_t> mesh_ids) { in set_auto_spmd_partitioning_mesh_ids()
H A Dhlo_module.cc545 std::vector<int64_t> mesh_ids; in CreateModuleConfigFromShape() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/client/
H A Dexecutable_build_options.cc95 std::vector<int64_t> mesh_ids) { in set_auto_spmd_partitioning_mesh_ids()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/tpu/
H A Dc_api_conversions.cc515 std::vector<int64_t> mesh_ids(mesh_ids_span.begin(), mesh_ids_span.end()); in FromC() local