Home
last modified time | relevance | path

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

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