Searched defs:sharding (Results 1 - 7 of 7) sorted by relevance

/external/tensorflow/tensorflow/compiler/tf2xla/
H A Dsharding_util.cc35 xla::OpSharding sharding; local
37 if (!sharding.ParseFromString(value)) {
42 return tensorflow::gtl::optional<xla::OpSharding>(sharding);
84 TF_ASSIGN_OR_RETURN(tensorflow::gtl::optional<xla::OpSharding> sharding,
86 return ParseShardingFromDevice(device_name, num_cores_per_replica, sharding);
95 TF_ASSIGN_OR_RETURN(tensorflow::gtl::optional<xla::OpSharding> sharding,
97 return ParseShardingFromDevice(device_name, num_cores_per_replica, sharding);
/external/tensorflow/tensorflow/compiler/xla/service/
H A Dhlo_sharding_test.cc44 HloSharding sharding = HloSharding::Replicate(); local
45 EXPECT_TRUE(sharding.IsReplicated());
46 EXPECT_TRUE(sharding.IsTileMaximal());
47 EXPECT_TRUE(sharding.UsesDevice(0));
48 EXPECT_TRUE(sharding.UsesDevice(65535));
51 EXPECT_EQ(other, sharding);
53 EXPECT_IS_OK(sharding.Validate(ShapeUtil::MakeShape(U32, {4}),
55 EXPECT_IS_NOT_OK(sharding.UniqueDevice());
59 HloSharding sharding = HloSharding::AssignDevice(5); local
60 EXPECT_FALSE(sharding
84 HloSharding sharding = local
93 HloSharding sharding = local
102 HloSharding sharding = local
112 HloSharding sharding = local
121 HloSharding sharding = local
[all...]
H A Duser_computation_test.cc230 OpSharding sharding; local
231 sharding.set_type(OpSharding::Type::OpSharding_Type_MAXIMAL);
232 sharding.add_tile_assignment_dimensions(1);
233 sharding.add_tile_assignment_devices(kDevice);
235 TF_EXPECT_OK(computation.SetOpSharding(b_handle, sharding));
H A Dhlo_instruction.h919 // Returns the sharding applied to this operator.
921 const HloSharding& sharding() const { function in class:xla::HloInstruction
925 // Returns the sharding applied to this operator, or default_ if none exists.
929 // Sets the sharding of this operator. Should only be called by HloModule or
931 void set_sharding(const HloSharding& sharding) { argument
932 sharding_ = MakeUnique<HloSharding>(sharding);
934 // Remove any sharding from this operator.
936 // Return true if this operator has a sharding assigned.
1423 // The sharding, if one exists.
H A Duser_computation.cc1491 const OpSharding& sharding) {
1502 ->mutable_sharding() = sharding;
2873 OpSharding op_sharding = request.request().sharding();
1490 SetOpSharding(const ComputationDataHandle& handle, const OpSharding& sharding) argument
/external/tensorflow/tensorflow/compiler/xla/client/
H A Dcomputation_builder.h77 void SetSharding(const OpSharding& sharding) { sharding_ = sharding; } argument
79 // Clears the sharding. Ops will be sharded according to the default placement
84 const tensorflow::gtl::optional<OpSharding>& sharding() const { function in class:xla::ComputationBuilder
1039 // RAII-style object: sets the current sharding assignment in builder on
1044 tensorflow::gtl::optional<OpSharding> sharding)
1045 : builder_(builder), prev_sharding_(builder->sharding()) {
1046 SetSharding(sharding);
1052 void SetSharding(const tensorflow::gtl::optional<OpSharding>& sharding) { argument
1053 if (sharding
1043 ScopedShardingAssignment(xla::ComputationBuilder* builder, tensorflow::gtl::optional<OpSharding> sharding) argument
[all...]
/external/tensorflow/tensorflow/compiler/xla/tools/parser/
H A Dhlo_parser.cc170 bool ParseSharding(OpSharding* sharding);
171 bool ParseSingleSharding(OpSharding* sharding, bool lbrace_pre_lexed);
432 optional<OpSharding> sharding; local
433 attrs["sharding"] = {/*required=*/false, AttrTy::kSharding, &sharding};
1062 // Add common attrs (sharding, control predecessors) to the instruction, if
1064 if (sharding) {
1066 HloSharding::FromProto(sharding.value()).ValueOrDie());
1086 bool HloParser::ParseSharding(OpSharding* sharding) { argument
1087 // A single sharding start
1118 ParseSingleSharding(OpSharding* sharding, bool lbrace_pre_lexed) argument
[all...]

Completed in 673 milliseconds