Home
last modified time | relevance | path

Searched refs:auto_spmd_partitioning_mesh_shape_ (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/
H A Dhlo_module_config.h176 auto_spmd_partitioning_mesh_shape_ = mesh_shape; in set_auto_spmd_partitioning_mesh_shape()
179 return auto_spmd_partitioning_mesh_shape_; in auto_spmd_partitioning_mesh_shape()
377 std::vector<int64_t> auto_spmd_partitioning_mesh_shape_; variable
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/client/
H A Dexecutable_build_options.h102 return auto_spmd_partitioning_mesh_shape_; in auto_spmd_partitioning_mesh_shape()
185 std::vector<int64_t> auto_spmd_partitioning_mesh_shape_; variable
H A Dexecutable_build_options.cc89 auto_spmd_partitioning_mesh_shape_ = mesh_shape; in set_auto_spmd_partitioning_mesh_shape()