Home
last modified time | relevance | path

Searched defs:ShardingSpec (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/dtensor/proto/
Dlayout.proto23 message ShardingSpec { message
/external/tensorflow/tensorflow/compiler/xla/python/
Dsharded_device_array.h175 ShardingSpec(std::vector<AvalDimSharding> sharding, in ShardingSpec() function
179 ShardingSpec(pybind11::iterable py_sharding, in ShardingSpec() function
/external/tensorflow/tensorflow/dtensor/mlir/expansions/
Deinsum_spmd_expander.cc208 std::pair<std::vector<ShardingSpec>, absl::flat_hash_map<std::string, int>> in VerifyOrFixLayout()