Home
last modified time | relevance | path

Searched refs:GetSharding (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/python/
Dpmap_lib.h123 const std::vector<AvalDimSharding>& GetSharding() const { return sharding_; } in GetSharding() function
Dpmap_lib.cc406 .def_property_readonly("sharding", &ShardingSpec::GetSharding) in BuildPmapSubmodule()