Searched defs:sharding_spec_strs (Results 1 – 2 of 2) sorted by relevance
170 const std::vector<std::string>& sharding_spec_strs = in SliceSpecOnDevice() local
662 const std::vector<std::string>& sharding_spec_strs, const Mesh& mesh) { in GetLayout()1063 std::vector<std::string> sharding_spec_strs = in FromString() local1075 std::vector<std::string> Layout::sharding_spec_strs() const { in sharding_spec_strs() function in tensorflow::dtensor::Layout1076 std::vector<std::string> sharding_spec_strs(sharding_specs().size()); in sharding_spec_strs() local