Searched refs:CreateOpShardingFromStringRef (Results 1 – 1 of 1) sorted by relevance
457 static std::optional<xla::OpSharding> CreateOpShardingFromStringRef( in CreateOpShardingFromStringRef() function471 return CreateOpShardingFromStringRef(sharding.getValue()); in CreateOpShardingFromAttribute()545 (*arg_shardings)[i] = CreateOpShardingFromStringRef(sharding.getValue()); in ExtractShardingsFromFunction()552 (*ret_shardings)[i] = CreateOpShardingFromStringRef(sharding.getValue()); in ExtractShardingsFromFunction()875 auto entry = CreateOpShardingFromStringRef(op.entry_metadata()); in ExportXlaOp()877 auto exit = CreateOpShardingFromStringRef(op.exit_metadata()); in ExportXlaOp()