Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/client/
H A Dexecutable_build_options.h101 std::vector<int64_t> auto_spmd_partitioning_mesh_shape() const { in auto_spmd_partitioning_mesh_shape() function
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/
H A Dcompiler.h94 virtual std::vector<int64_t> auto_spmd_partitioning_mesh_shape() const { in auto_spmd_partitioning_mesh_shape() function
H A Dhlo_module_config.h178 std::vector<int64_t> auto_spmd_partitioning_mesh_shape() const { in auto_spmd_partitioning_mesh_shape() function
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/
H A Dxla.proto498 repeated int64 auto_spmd_partitioning_mesh_shape = 16; field
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/tpu/
H A Dc_api_decl.h310 Int64List auto_spmd_partitioning_mesh_shape; member