Searched defs:sharding (Results 1 – 9 of 9) sorted by relevance
44 HloSharding sharding = HloSharding::Replicate(); in TEST_F() local59 HloSharding sharding = HloSharding::AssignDevice(5); in TEST_F() local84 HloSharding sharding = in TEST_F() local93 HloSharding sharding = in TEST_F() local102 HloSharding sharding = in TEST_F() local112 HloSharding sharding = in TEST_F() local121 HloSharding sharding = in TEST_F() local
230 OpSharding sharding; in TEST_F() local
921 const HloSharding& sharding() const { in sharding() function931 void set_sharding(const HloSharding& sharding) { in set_sharding()
1491 const OpSharding& sharding) { in SetOpSharding()
77 void SetSharding(const OpSharding& sharding) { sharding_ = sharding; } in SetSharding()84 const tensorflow::gtl::optional<OpSharding>& sharding() const { in sharding() function1044 tensorflow::gtl::optional<OpSharding> sharding) in ScopedShardingAssignment()1052 void SetSharding(const tensorflow::gtl::optional<OpSharding>& sharding) { in SetSharding()
26 [](tensorflow::gtl::optional<xla::OpSharding> sharding) -> int64 { in TEST()
35 xla::OpSharding sharding; in GetShardingFromNodeDef() local
432 optional<OpSharding> sharding; in ParseInstruction() local1086 bool HloParser::ParseSharding(OpSharding* sharding) { in ParseSharding()1118 bool HloParser::ParseSingleSharding(OpSharding* sharding, in ParseSingleSharding()1891 OpSharding sharding; in ParseAttributeHelper() local
965 OpSharding sharding = 40; field