Home
last modified time | relevance | path

Searched defs:sharding_spec_strs (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/dtensor/cc/
Dsave_restore_util.cc170 const std::vector<std::string>& sharding_spec_strs = in SliceSpecOnDevice() local
Dtensor_layout.cc662 const std::vector<std::string>& sharding_spec_strs, const Mesh& mesh) { in GetLayout()
1063 std::vector<std::string> sharding_spec_strs = in FromString() local
1075 std::vector<std::string> Layout::sharding_spec_strs() const { in sharding_spec_strs() function in tensorflow::dtensor::Layout
1076 std::vector<std::string> sharding_spec_strs(sharding_specs().size()); in sharding_spec_strs() local