Searched defs:sharding (Results 1 - 7 of 7) sorted by relevance
/external/tensorflow/tensorflow/compiler/tf2xla/ |
H A D | sharding_util.cc | 35 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 D | hlo_sharding_test.cc | 44 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 D | user_computation_test.cc | 230 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 D | hlo_instruction.h | 919 // 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 D | user_computation.cc | 1491 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 D | computation_builder.h | 77 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 D | hlo_parser.cc | 170 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