| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_sharding_test.cc | 42 HloSharding sharding = HloSharding::Replicate(); in TEST_F() local
57 HloSharding sharding = HloSharding::AssignDevice(5); in TEST_F() local
81 HloSharding sharding = HloSharding::Tile(MakeArray({2, 2}, {0, 0, 2, 3})); in TEST_F() local
88 HloSharding sharding = HloSharding::Tile(MakeArray({2, 2}, {0, 1, 2, 3})); in TEST_F() local
96 HloSharding sharding = HloSharding::Tile(MakeArray({2, 2}, {0, 3, 2, 1})); in TEST_F() local
120 HloSharding sharding = HloSharding::SingleTuple(ShapeUtil::MakeTupleShape({}), in TEST_F() local
236 HloSharding sharding = HloSharding::Replicate(); in TEST_F() local
241 HloSharding sharding = HloSharding::AssignDevice(7); in TEST_F() local
246 HloSharding sharding = in TEST_F() local
252 HloSharding sharding = HloSharding::Tuple( in TEST_F() local
[all …]
|
| D | hlo_sharding_metadata.h | 30 explicit ShardingMetadata(std::shared_ptr<const HloSharding> sharding) in ShardingMetadata()
43 const HloSharding* sharding() const { return sharding_.get(); } in sharding() function
77 std::shared_ptr<const HloSharding> sharding; member
|
| D | hlo_sharding_metadata.cc | 53 const HloSharding& sharding) { in SetSingleSharding()
120 const HloSharding& sharding) { in FixupPassThroughDomainLinks()
143 std::shared_ptr<const HloSharding> sharding) { in CloneShardingForDomain()
152 const HloSharding& sharding) { in ApplyDomainSingleSharding()
341 const HloSharding& sharding) { in ApplyDomainSharding()
381 std::shared_ptr<const HloSharding> sharding; in ExtractOriginalCommonSharding() local
403 std::unique_ptr<HloSharding> sharding; in Clone() local
451 const HloSharding* sharding = sharding_metadata->sharding(); in NormalizeShardingDomain() local
|
| D | hlo_matchers.h | 138 explicit HloShardingMatcher(const absl::optional<HloSharding>& sharding) in HloShardingMatcher()
373 const HloSharding& sharding) { in Sharding()
379 absl::string_view sharding) { in Sharding()
|
| D | hlo_sharding.cc | 62 for (auto& sharding : shardings) { in Tuple() local
73 const HloSharding& sharding) { in SingleTuple()
83 const HloSharding& sharding) { in Single()
501 std::ostream& operator<<(std::ostream& out, const HloSharding& sharding) { in operator <<()
|
| D | hlo_instruction.h | 1139 const HloSharding& sharding() const { in sharding() function
1158 void set_sharding(const HloSharding& sharding) { in set_sharding()
1161 void set_sharding(std::shared_ptr<const HloSharding> sharding) { in set_sharding()
|
| D | hlo_matchers_test.cc | 159 auto sharding = HloSharding::Tuple( in TEST() local
|
| D | batchnorm_expander.cc | 305 const HloSharding& sharding = batch_norm->sharding(); in HandleBatchNormTraining() local
411 const HloSharding& sharding = batch_norm->sharding(); in HandleBatchNormInference() local
584 const HloSharding& sharding = batch_norm->sharding(); in HandleBatchNormGrad() local
|
| D | hlo.proto | 167 xla.OpSharding sharding = 40; field
|
| D | hlo_parser.cc | 668 optional<OpSharding> sharding; in ParseInstructionRhs() local
1758 bool HloParser::ParseSharding(OpSharding* sharding) { in ParseSharding()
1790 bool HloParser::ParseSingleSharding(OpSharding* sharding, in ParseSingleSharding()
2766 OpSharding sharding; in ParseAttributeHelper() local
|
| D | layout_assignment.cc | 861 HloSharding sharding = in SetupCopiedInstruction() local
|
| D | hlo_instruction.cc | 1214 void HloInstruction::set_single_sharding(const HloSharding& sharding) { in set_single_sharding()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | sharding_util_test.cc | 26 [](absl::optional<xla::OpSharding> sharding) -> int64 { in TEST()
|
| D | sharding_util.cc | 35 xla::OpSharding sharding; in GetShardingFromNodeDef() local
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | xla_builder.h | 159 void SetSharding(const OpSharding& sharding) { sharding_ = sharding; } in SetSharding()
166 const absl::optional<OpSharding>& sharding() const { return sharding_; } in sharding() function
1014 absl::optional<OpSharding> sharding) in XlaScopedShardingAssignment()
1026 void SetSharding(const absl::optional<OpSharding>& sharding) { in SetSharding()
|
| D | xla_builder.cc | 1315 OpSharding sharding = sharding_builder::AssignDevice(0); in Infeed() local
|