14424 lines
532 KiB
C++
14424 lines
532 KiB
C++
// Generated by the protocol buffer compiler. DO NOT EDIT!
|
|
// source: xla/xla_data.proto
|
|
|
|
#ifndef GOOGLE_PROTOBUF_INCLUDED_xla_2fxla_5fdata_2eproto
|
|
#define GOOGLE_PROTOBUF_INCLUDED_xla_2fxla_5fdata_2eproto
|
|
|
|
#include <limits>
|
|
#include <string>
|
|
|
|
#include <google/protobuf/port_def.inc>
|
|
#if PROTOBUF_VERSION < 3021000
|
|
#error This file was generated by a newer version of protoc which is
|
|
#error incompatible with your Protocol Buffer headers. Please update
|
|
#error your headers.
|
|
#endif
|
|
#if 3021009 < PROTOBUF_MIN_PROTOC_VERSION
|
|
#error This file was generated by an older version of protoc which is
|
|
#error incompatible with your Protocol Buffer headers. Please
|
|
#error regenerate this file with a newer version of protoc.
|
|
#endif
|
|
|
|
#include <google/protobuf/port_undef.inc>
|
|
#include <google/protobuf/io/coded_stream.h>
|
|
#include <google/protobuf/arena.h>
|
|
#include <google/protobuf/arenastring.h>
|
|
#include <google/protobuf/generated_message_util.h>
|
|
#include <google/protobuf/metadata_lite.h>
|
|
#include <google/protobuf/generated_message_reflection.h>
|
|
#include <google/protobuf/message.h>
|
|
#include <google/protobuf/repeated_field.h> // IWYU pragma: export
|
|
#include <google/protobuf/extension_set.h> // IWYU pragma: export
|
|
#include <google/protobuf/map.h> // IWYU pragma: export
|
|
#include <google/protobuf/map_entry.h>
|
|
#include <google/protobuf/map_field_inl.h>
|
|
#include <google/protobuf/generated_enum_reflection.h>
|
|
#include <google/protobuf/unknown_field_set.h>
|
|
// @@protoc_insertion_point(includes)
|
|
#include <google/protobuf/port_def.inc>
|
|
#define PROTOBUF_INTERNAL_EXPORT_xla_2fxla_5fdata_2eproto
|
|
PROTOBUF_NAMESPACE_OPEN
|
|
namespace internal {
|
|
class AnyMetadata;
|
|
} // namespace internal
|
|
PROTOBUF_NAMESPACE_CLOSE
|
|
|
|
// Internal implementation detail -- do not use these members.
|
|
struct TableStruct_xla_2fxla_5fdata_2eproto {
|
|
static const uint32_t offsets[];
|
|
};
|
|
extern const ::PROTOBUF_NAMESPACE_ID::internal::DescriptorTable descriptor_table_xla_2fxla_5fdata_2eproto;
|
|
namespace xla {
|
|
class ChannelHandle;
|
|
struct ChannelHandleDefaultTypeInternal;
|
|
extern ChannelHandleDefaultTypeInternal _ChannelHandle_default_instance_;
|
|
class CholeskyOptions;
|
|
struct CholeskyOptionsDefaultTypeInternal;
|
|
extern CholeskyOptionsDefaultTypeInternal _CholeskyOptions_default_instance_;
|
|
class ComputationStats;
|
|
struct ComputationStatsDefaultTypeInternal;
|
|
extern ComputationStatsDefaultTypeInternal _ComputationStats_default_instance_;
|
|
class ConvolutionDimensionNumbers;
|
|
struct ConvolutionDimensionNumbersDefaultTypeInternal;
|
|
extern ConvolutionDimensionNumbersDefaultTypeInternal _ConvolutionDimensionNumbers_default_instance_;
|
|
class DeviceAssignmentProto;
|
|
struct DeviceAssignmentProtoDefaultTypeInternal;
|
|
extern DeviceAssignmentProtoDefaultTypeInternal _DeviceAssignmentProto_default_instance_;
|
|
class DeviceAssignmentProto_ComputationDevice;
|
|
struct DeviceAssignmentProto_ComputationDeviceDefaultTypeInternal;
|
|
extern DeviceAssignmentProto_ComputationDeviceDefaultTypeInternal _DeviceAssignmentProto_ComputationDevice_default_instance_;
|
|
class DeviceHandle;
|
|
struct DeviceHandleDefaultTypeInternal;
|
|
extern DeviceHandleDefaultTypeInternal _DeviceHandle_default_instance_;
|
|
class DotDimensionNumbers;
|
|
struct DotDimensionNumbersDefaultTypeInternal;
|
|
extern DotDimensionNumbersDefaultTypeInternal _DotDimensionNumbers_default_instance_;
|
|
class ExecutionHandle;
|
|
struct ExecutionHandleDefaultTypeInternal;
|
|
extern ExecutionHandleDefaultTypeInternal _ExecutionHandle_default_instance_;
|
|
class ExecutionProfile;
|
|
struct ExecutionProfileDefaultTypeInternal;
|
|
extern ExecutionProfileDefaultTypeInternal _ExecutionProfile_default_instance_;
|
|
class FrontendAttributes;
|
|
struct FrontendAttributesDefaultTypeInternal;
|
|
extern FrontendAttributesDefaultTypeInternal _FrontendAttributes_default_instance_;
|
|
class FrontendAttributes_MapEntry_DoNotUse;
|
|
struct FrontendAttributes_MapEntry_DoNotUseDefaultTypeInternal;
|
|
extern FrontendAttributes_MapEntry_DoNotUseDefaultTypeInternal _FrontendAttributes_MapEntry_DoNotUse_default_instance_;
|
|
class GatherDimensionNumbers;
|
|
struct GatherDimensionNumbersDefaultTypeInternal;
|
|
extern GatherDimensionNumbersDefaultTypeInternal _GatherDimensionNumbers_default_instance_;
|
|
class GlobalDataHandle;
|
|
struct GlobalDataHandleDefaultTypeInternal;
|
|
extern GlobalDataHandleDefaultTypeInternal _GlobalDataHandle_default_instance_;
|
|
class LayoutProto;
|
|
struct LayoutProtoDefaultTypeInternal;
|
|
extern LayoutProtoDefaultTypeInternal _LayoutProto_default_instance_;
|
|
class LiteralProto;
|
|
struct LiteralProtoDefaultTypeInternal;
|
|
extern LiteralProtoDefaultTypeInternal _LiteralProto_default_instance_;
|
|
class OpMetadata;
|
|
struct OpMetadataDefaultTypeInternal;
|
|
extern OpMetadataDefaultTypeInternal _OpMetadata_default_instance_;
|
|
class OpMetadata_ProfileInfo;
|
|
struct OpMetadata_ProfileInfoDefaultTypeInternal;
|
|
extern OpMetadata_ProfileInfoDefaultTypeInternal _OpMetadata_ProfileInfo_default_instance_;
|
|
class OpSharding;
|
|
struct OpShardingDefaultTypeInternal;
|
|
extern OpShardingDefaultTypeInternal _OpSharding_default_instance_;
|
|
class OutputOperandAliasing;
|
|
struct OutputOperandAliasingDefaultTypeInternal;
|
|
extern OutputOperandAliasingDefaultTypeInternal _OutputOperandAliasing_default_instance_;
|
|
class PaddingConfig;
|
|
struct PaddingConfigDefaultTypeInternal;
|
|
extern PaddingConfigDefaultTypeInternal _PaddingConfig_default_instance_;
|
|
class PaddingConfig_PaddingConfigDimension;
|
|
struct PaddingConfig_PaddingConfigDimensionDefaultTypeInternal;
|
|
extern PaddingConfig_PaddingConfigDimensionDefaultTypeInternal _PaddingConfig_PaddingConfigDimension_default_instance_;
|
|
class ParameterReplication;
|
|
struct ParameterReplicationDefaultTypeInternal;
|
|
extern ParameterReplicationDefaultTypeInternal _ParameterReplication_default_instance_;
|
|
class PrecisionConfig;
|
|
struct PrecisionConfigDefaultTypeInternal;
|
|
extern PrecisionConfigDefaultTypeInternal _PrecisionConfig_default_instance_;
|
|
class ProgramShapeProto;
|
|
struct ProgramShapeProtoDefaultTypeInternal;
|
|
extern ProgramShapeProtoDefaultTypeInternal _ProgramShapeProto_default_instance_;
|
|
class ReplicaGroup;
|
|
struct ReplicaGroupDefaultTypeInternal;
|
|
extern ReplicaGroupDefaultTypeInternal _ReplicaGroup_default_instance_;
|
|
class ScatterDimensionNumbers;
|
|
struct ScatterDimensionNumbersDefaultTypeInternal;
|
|
extern ScatterDimensionNumbersDefaultTypeInternal _ScatterDimensionNumbers_default_instance_;
|
|
class ShapeProto;
|
|
struct ShapeProtoDefaultTypeInternal;
|
|
extern ShapeProtoDefaultTypeInternal _ShapeProto_default_instance_;
|
|
class SortOptions;
|
|
struct SortOptionsDefaultTypeInternal;
|
|
extern SortOptionsDefaultTypeInternal _SortOptions_default_instance_;
|
|
class SourceTarget;
|
|
struct SourceTargetDefaultTypeInternal;
|
|
extern SourceTargetDefaultTypeInternal _SourceTarget_default_instance_;
|
|
class SparsityDescriptor;
|
|
struct SparsityDescriptorDefaultTypeInternal;
|
|
extern SparsityDescriptorDefaultTypeInternal _SparsityDescriptor_default_instance_;
|
|
class Statistic;
|
|
struct StatisticDefaultTypeInternal;
|
|
extern StatisticDefaultTypeInternal _Statistic_default_instance_;
|
|
class StatisticsViz;
|
|
struct StatisticsVizDefaultTypeInternal;
|
|
extern StatisticsVizDefaultTypeInternal _StatisticsViz_default_instance_;
|
|
class TileProto;
|
|
struct TileProtoDefaultTypeInternal;
|
|
extern TileProtoDefaultTypeInternal _TileProto_default_instance_;
|
|
class TriangularSolveOptions;
|
|
struct TriangularSolveOptionsDefaultTypeInternal;
|
|
extern TriangularSolveOptionsDefaultTypeInternal _TriangularSolveOptions_default_instance_;
|
|
class WhileLoopBackendConfig;
|
|
struct WhileLoopBackendConfigDefaultTypeInternal;
|
|
extern WhileLoopBackendConfigDefaultTypeInternal _WhileLoopBackendConfig_default_instance_;
|
|
class WhileLoopBackendConfig_KnownTripCount;
|
|
struct WhileLoopBackendConfig_KnownTripCountDefaultTypeInternal;
|
|
extern WhileLoopBackendConfig_KnownTripCountDefaultTypeInternal _WhileLoopBackendConfig_KnownTripCount_default_instance_;
|
|
class Window;
|
|
struct WindowDefaultTypeInternal;
|
|
extern WindowDefaultTypeInternal _Window_default_instance_;
|
|
class WindowDimension;
|
|
struct WindowDimensionDefaultTypeInternal;
|
|
extern WindowDimensionDefaultTypeInternal _WindowDimension_default_instance_;
|
|
} // namespace xla
|
|
PROTOBUF_NAMESPACE_OPEN
|
|
template<> ::xla::ChannelHandle* Arena::CreateMaybeMessage<::xla::ChannelHandle>(Arena*);
|
|
template<> ::xla::CholeskyOptions* Arena::CreateMaybeMessage<::xla::CholeskyOptions>(Arena*);
|
|
template<> ::xla::ComputationStats* Arena::CreateMaybeMessage<::xla::ComputationStats>(Arena*);
|
|
template<> ::xla::ConvolutionDimensionNumbers* Arena::CreateMaybeMessage<::xla::ConvolutionDimensionNumbers>(Arena*);
|
|
template<> ::xla::DeviceAssignmentProto* Arena::CreateMaybeMessage<::xla::DeviceAssignmentProto>(Arena*);
|
|
template<> ::xla::DeviceAssignmentProto_ComputationDevice* Arena::CreateMaybeMessage<::xla::DeviceAssignmentProto_ComputationDevice>(Arena*);
|
|
template<> ::xla::DeviceHandle* Arena::CreateMaybeMessage<::xla::DeviceHandle>(Arena*);
|
|
template<> ::xla::DotDimensionNumbers* Arena::CreateMaybeMessage<::xla::DotDimensionNumbers>(Arena*);
|
|
template<> ::xla::ExecutionHandle* Arena::CreateMaybeMessage<::xla::ExecutionHandle>(Arena*);
|
|
template<> ::xla::ExecutionProfile* Arena::CreateMaybeMessage<::xla::ExecutionProfile>(Arena*);
|
|
template<> ::xla::FrontendAttributes* Arena::CreateMaybeMessage<::xla::FrontendAttributes>(Arena*);
|
|
template<> ::xla::FrontendAttributes_MapEntry_DoNotUse* Arena::CreateMaybeMessage<::xla::FrontendAttributes_MapEntry_DoNotUse>(Arena*);
|
|
template<> ::xla::GatherDimensionNumbers* Arena::CreateMaybeMessage<::xla::GatherDimensionNumbers>(Arena*);
|
|
template<> ::xla::GlobalDataHandle* Arena::CreateMaybeMessage<::xla::GlobalDataHandle>(Arena*);
|
|
template<> ::xla::LayoutProto* Arena::CreateMaybeMessage<::xla::LayoutProto>(Arena*);
|
|
template<> ::xla::LiteralProto* Arena::CreateMaybeMessage<::xla::LiteralProto>(Arena*);
|
|
template<> ::xla::OpMetadata* Arena::CreateMaybeMessage<::xla::OpMetadata>(Arena*);
|
|
template<> ::xla::OpMetadata_ProfileInfo* Arena::CreateMaybeMessage<::xla::OpMetadata_ProfileInfo>(Arena*);
|
|
template<> ::xla::OpSharding* Arena::CreateMaybeMessage<::xla::OpSharding>(Arena*);
|
|
template<> ::xla::OutputOperandAliasing* Arena::CreateMaybeMessage<::xla::OutputOperandAliasing>(Arena*);
|
|
template<> ::xla::PaddingConfig* Arena::CreateMaybeMessage<::xla::PaddingConfig>(Arena*);
|
|
template<> ::xla::PaddingConfig_PaddingConfigDimension* Arena::CreateMaybeMessage<::xla::PaddingConfig_PaddingConfigDimension>(Arena*);
|
|
template<> ::xla::ParameterReplication* Arena::CreateMaybeMessage<::xla::ParameterReplication>(Arena*);
|
|
template<> ::xla::PrecisionConfig* Arena::CreateMaybeMessage<::xla::PrecisionConfig>(Arena*);
|
|
template<> ::xla::ProgramShapeProto* Arena::CreateMaybeMessage<::xla::ProgramShapeProto>(Arena*);
|
|
template<> ::xla::ReplicaGroup* Arena::CreateMaybeMessage<::xla::ReplicaGroup>(Arena*);
|
|
template<> ::xla::ScatterDimensionNumbers* Arena::CreateMaybeMessage<::xla::ScatterDimensionNumbers>(Arena*);
|
|
template<> ::xla::ShapeProto* Arena::CreateMaybeMessage<::xla::ShapeProto>(Arena*);
|
|
template<> ::xla::SortOptions* Arena::CreateMaybeMessage<::xla::SortOptions>(Arena*);
|
|
template<> ::xla::SourceTarget* Arena::CreateMaybeMessage<::xla::SourceTarget>(Arena*);
|
|
template<> ::xla::SparsityDescriptor* Arena::CreateMaybeMessage<::xla::SparsityDescriptor>(Arena*);
|
|
template<> ::xla::Statistic* Arena::CreateMaybeMessage<::xla::Statistic>(Arena*);
|
|
template<> ::xla::StatisticsViz* Arena::CreateMaybeMessage<::xla::StatisticsViz>(Arena*);
|
|
template<> ::xla::TileProto* Arena::CreateMaybeMessage<::xla::TileProto>(Arena*);
|
|
template<> ::xla::TriangularSolveOptions* Arena::CreateMaybeMessage<::xla::TriangularSolveOptions>(Arena*);
|
|
template<> ::xla::WhileLoopBackendConfig* Arena::CreateMaybeMessage<::xla::WhileLoopBackendConfig>(Arena*);
|
|
template<> ::xla::WhileLoopBackendConfig_KnownTripCount* Arena::CreateMaybeMessage<::xla::WhileLoopBackendConfig_KnownTripCount>(Arena*);
|
|
template<> ::xla::Window* Arena::CreateMaybeMessage<::xla::Window>(Arena*);
|
|
template<> ::xla::WindowDimension* Arena::CreateMaybeMessage<::xla::WindowDimension>(Arena*);
|
|
PROTOBUF_NAMESPACE_CLOSE
|
|
namespace xla {
|
|
|
|
enum ChannelHandle_ChannelType : int {
|
|
ChannelHandle_ChannelType_CHANNEL_TYPE_INVALID = 0,
|
|
ChannelHandle_ChannelType_DEVICE_TO_DEVICE = 1,
|
|
ChannelHandle_ChannelType_DEVICE_TO_HOST = 2,
|
|
ChannelHandle_ChannelType_HOST_TO_DEVICE = 3,
|
|
ChannelHandle_ChannelType_ChannelHandle_ChannelType_INT_MIN_SENTINEL_DO_NOT_USE_ = std::numeric_limits<int32_t>::min(),
|
|
ChannelHandle_ChannelType_ChannelHandle_ChannelType_INT_MAX_SENTINEL_DO_NOT_USE_ = std::numeric_limits<int32_t>::max()
|
|
};
|
|
bool ChannelHandle_ChannelType_IsValid(int value);
|
|
constexpr ChannelHandle_ChannelType ChannelHandle_ChannelType_ChannelType_MIN = ChannelHandle_ChannelType_CHANNEL_TYPE_INVALID;
|
|
constexpr ChannelHandle_ChannelType ChannelHandle_ChannelType_ChannelType_MAX = ChannelHandle_ChannelType_HOST_TO_DEVICE;
|
|
constexpr int ChannelHandle_ChannelType_ChannelType_ARRAYSIZE = ChannelHandle_ChannelType_ChannelType_MAX + 1;
|
|
|
|
const ::PROTOBUF_NAMESPACE_ID::EnumDescriptor* ChannelHandle_ChannelType_descriptor();
|
|
template<typename T>
|
|
inline const std::string& ChannelHandle_ChannelType_Name(T enum_t_value) {
|
|
static_assert(::std::is_same<T, ChannelHandle_ChannelType>::value ||
|
|
::std::is_integral<T>::value,
|
|
"Incorrect type passed to function ChannelHandle_ChannelType_Name.");
|
|
return ::PROTOBUF_NAMESPACE_ID::internal::NameOfEnum(
|
|
ChannelHandle_ChannelType_descriptor(), enum_t_value);
|
|
}
|
|
inline bool ChannelHandle_ChannelType_Parse(
|
|
::PROTOBUF_NAMESPACE_ID::ConstStringParam name, ChannelHandle_ChannelType* value) {
|
|
return ::PROTOBUF_NAMESPACE_ID::internal::ParseNamedEnum<ChannelHandle_ChannelType>(
|
|
ChannelHandle_ChannelType_descriptor(), name, value);
|
|
}
|
|
enum TriangularSolveOptions_Transpose : int {
|
|
TriangularSolveOptions_Transpose_TRANSPOSE_INVALID = 0,
|
|
TriangularSolveOptions_Transpose_NO_TRANSPOSE = 1,
|
|
TriangularSolveOptions_Transpose_TRANSPOSE = 2,
|
|
TriangularSolveOptions_Transpose_ADJOINT = 3,
|
|
TriangularSolveOptions_Transpose_TriangularSolveOptions_Transpose_INT_MIN_SENTINEL_DO_NOT_USE_ = std::numeric_limits<int32_t>::min(),
|
|
TriangularSolveOptions_Transpose_TriangularSolveOptions_Transpose_INT_MAX_SENTINEL_DO_NOT_USE_ = std::numeric_limits<int32_t>::max()
|
|
};
|
|
bool TriangularSolveOptions_Transpose_IsValid(int value);
|
|
constexpr TriangularSolveOptions_Transpose TriangularSolveOptions_Transpose_Transpose_MIN = TriangularSolveOptions_Transpose_TRANSPOSE_INVALID;
|
|
constexpr TriangularSolveOptions_Transpose TriangularSolveOptions_Transpose_Transpose_MAX = TriangularSolveOptions_Transpose_ADJOINT;
|
|
constexpr int TriangularSolveOptions_Transpose_Transpose_ARRAYSIZE = TriangularSolveOptions_Transpose_Transpose_MAX + 1;
|
|
|
|
const ::PROTOBUF_NAMESPACE_ID::EnumDescriptor* TriangularSolveOptions_Transpose_descriptor();
|
|
template<typename T>
|
|
inline const std::string& TriangularSolveOptions_Transpose_Name(T enum_t_value) {
|
|
static_assert(::std::is_same<T, TriangularSolveOptions_Transpose>::value ||
|
|
::std::is_integral<T>::value,
|
|
"Incorrect type passed to function TriangularSolveOptions_Transpose_Name.");
|
|
return ::PROTOBUF_NAMESPACE_ID::internal::NameOfEnum(
|
|
TriangularSolveOptions_Transpose_descriptor(), enum_t_value);
|
|
}
|
|
inline bool TriangularSolveOptions_Transpose_Parse(
|
|
::PROTOBUF_NAMESPACE_ID::ConstStringParam name, TriangularSolveOptions_Transpose* value) {
|
|
return ::PROTOBUF_NAMESPACE_ID::internal::ParseNamedEnum<TriangularSolveOptions_Transpose>(
|
|
TriangularSolveOptions_Transpose_descriptor(), name, value);
|
|
}
|
|
enum OpSharding_Type : int {
|
|
OpSharding_Type_REPLICATED = 0,
|
|
OpSharding_Type_MAXIMAL = 1,
|
|
OpSharding_Type_TUPLE = 2,
|
|
OpSharding_Type_OTHER = 3,
|
|
OpSharding_Type_MANUAL = 4,
|
|
OpSharding_Type_UNKNOWN = 5,
|
|
OpSharding_Type_OpSharding_Type_INT_MIN_SENTINEL_DO_NOT_USE_ = std::numeric_limits<int32_t>::min(),
|
|
OpSharding_Type_OpSharding_Type_INT_MAX_SENTINEL_DO_NOT_USE_ = std::numeric_limits<int32_t>::max()
|
|
};
|
|
bool OpSharding_Type_IsValid(int value);
|
|
constexpr OpSharding_Type OpSharding_Type_Type_MIN = OpSharding_Type_REPLICATED;
|
|
constexpr OpSharding_Type OpSharding_Type_Type_MAX = OpSharding_Type_UNKNOWN;
|
|
constexpr int OpSharding_Type_Type_ARRAYSIZE = OpSharding_Type_Type_MAX + 1;
|
|
|
|
const ::PROTOBUF_NAMESPACE_ID::EnumDescriptor* OpSharding_Type_descriptor();
|
|
template<typename T>
|
|
inline const std::string& OpSharding_Type_Name(T enum_t_value) {
|
|
static_assert(::std::is_same<T, OpSharding_Type>::value ||
|
|
::std::is_integral<T>::value,
|
|
"Incorrect type passed to function OpSharding_Type_Name.");
|
|
return ::PROTOBUF_NAMESPACE_ID::internal::NameOfEnum(
|
|
OpSharding_Type_descriptor(), enum_t_value);
|
|
}
|
|
inline bool OpSharding_Type_Parse(
|
|
::PROTOBUF_NAMESPACE_ID::ConstStringParam name, OpSharding_Type* value) {
|
|
return ::PROTOBUF_NAMESPACE_ID::internal::ParseNamedEnum<OpSharding_Type>(
|
|
OpSharding_Type_descriptor(), name, value);
|
|
}
|
|
enum OpSharding_ShardGroupType : int {
|
|
OpSharding_ShardGroupType_AS = 0,
|
|
OpSharding_ShardGroupType_LIKE = 1,
|
|
OpSharding_ShardGroupType_OpSharding_ShardGroupType_INT_MIN_SENTINEL_DO_NOT_USE_ = std::numeric_limits<int32_t>::min(),
|
|
OpSharding_ShardGroupType_OpSharding_ShardGroupType_INT_MAX_SENTINEL_DO_NOT_USE_ = std::numeric_limits<int32_t>::max()
|
|
};
|
|
bool OpSharding_ShardGroupType_IsValid(int value);
|
|
constexpr OpSharding_ShardGroupType OpSharding_ShardGroupType_ShardGroupType_MIN = OpSharding_ShardGroupType_AS;
|
|
constexpr OpSharding_ShardGroupType OpSharding_ShardGroupType_ShardGroupType_MAX = OpSharding_ShardGroupType_LIKE;
|
|
constexpr int OpSharding_ShardGroupType_ShardGroupType_ARRAYSIZE = OpSharding_ShardGroupType_ShardGroupType_MAX + 1;
|
|
|
|
const ::PROTOBUF_NAMESPACE_ID::EnumDescriptor* OpSharding_ShardGroupType_descriptor();
|
|
template<typename T>
|
|
inline const std::string& OpSharding_ShardGroupType_Name(T enum_t_value) {
|
|
static_assert(::std::is_same<T, OpSharding_ShardGroupType>::value ||
|
|
::std::is_integral<T>::value,
|
|
"Incorrect type passed to function OpSharding_ShardGroupType_Name.");
|
|
return ::PROTOBUF_NAMESPACE_ID::internal::NameOfEnum(
|
|
OpSharding_ShardGroupType_descriptor(), enum_t_value);
|
|
}
|
|
inline bool OpSharding_ShardGroupType_Parse(
|
|
::PROTOBUF_NAMESPACE_ID::ConstStringParam name, OpSharding_ShardGroupType* value) {
|
|
return ::PROTOBUF_NAMESPACE_ID::internal::ParseNamedEnum<OpSharding_ShardGroupType>(
|
|
OpSharding_ShardGroupType_descriptor(), name, value);
|
|
}
|
|
enum PrecisionConfig_Precision : int {
|
|
PrecisionConfig_Precision_DEFAULT = 0,
|
|
PrecisionConfig_Precision_HIGH = 1,
|
|
PrecisionConfig_Precision_HIGHEST = 2,
|
|
PrecisionConfig_Precision_PACKED_NIBBLE = 3,
|
|
PrecisionConfig_Precision_PrecisionConfig_Precision_INT_MIN_SENTINEL_DO_NOT_USE_ = std::numeric_limits<int32_t>::min(),
|
|
PrecisionConfig_Precision_PrecisionConfig_Precision_INT_MAX_SENTINEL_DO_NOT_USE_ = std::numeric_limits<int32_t>::max()
|
|
};
|
|
bool PrecisionConfig_Precision_IsValid(int value);
|
|
constexpr PrecisionConfig_Precision PrecisionConfig_Precision_Precision_MIN = PrecisionConfig_Precision_DEFAULT;
|
|
constexpr PrecisionConfig_Precision PrecisionConfig_Precision_Precision_MAX = PrecisionConfig_Precision_PACKED_NIBBLE;
|
|
constexpr int PrecisionConfig_Precision_Precision_ARRAYSIZE = PrecisionConfig_Precision_Precision_MAX + 1;
|
|
|
|
const ::PROTOBUF_NAMESPACE_ID::EnumDescriptor* PrecisionConfig_Precision_descriptor();
|
|
template<typename T>
|
|
inline const std::string& PrecisionConfig_Precision_Name(T enum_t_value) {
|
|
static_assert(::std::is_same<T, PrecisionConfig_Precision>::value ||
|
|
::std::is_integral<T>::value,
|
|
"Incorrect type passed to function PrecisionConfig_Precision_Name.");
|
|
return ::PROTOBUF_NAMESPACE_ID::internal::NameOfEnum(
|
|
PrecisionConfig_Precision_descriptor(), enum_t_value);
|
|
}
|
|
inline bool PrecisionConfig_Precision_Parse(
|
|
::PROTOBUF_NAMESPACE_ID::ConstStringParam name, PrecisionConfig_Precision* value) {
|
|
return ::PROTOBUF_NAMESPACE_ID::internal::ParseNamedEnum<PrecisionConfig_Precision>(
|
|
PrecisionConfig_Precision_descriptor(), name, value);
|
|
}
|
|
enum PrimitiveType : int {
|
|
PRIMITIVE_TYPE_INVALID = 0,
|
|
PRED = 1,
|
|
S4 = 21,
|
|
S8 = 2,
|
|
S16 = 3,
|
|
S32 = 4,
|
|
S64 = 5,
|
|
U4 = 22,
|
|
U8 = 6,
|
|
U16 = 7,
|
|
U32 = 8,
|
|
U64 = 9,
|
|
F16 = 10,
|
|
F32 = 11,
|
|
BF16 = 16,
|
|
F64 = 12,
|
|
F8E5M2 = 19,
|
|
F8E4M3FN = 20,
|
|
F8E4M3B11FNUZ = 23,
|
|
F8E5M2FNUZ = 24,
|
|
F8E4M3FNUZ = 25,
|
|
C64 = 15,
|
|
C128 = 18,
|
|
TUPLE = 13,
|
|
OPAQUE_TYPE = 14,
|
|
TOKEN = 17,
|
|
PrimitiveType_INT_MIN_SENTINEL_DO_NOT_USE_ = std::numeric_limits<int32_t>::min(),
|
|
PrimitiveType_INT_MAX_SENTINEL_DO_NOT_USE_ = std::numeric_limits<int32_t>::max()
|
|
};
|
|
bool PrimitiveType_IsValid(int value);
|
|
constexpr PrimitiveType PrimitiveType_MIN = PRIMITIVE_TYPE_INVALID;
|
|
constexpr PrimitiveType PrimitiveType_MAX = F8E4M3FNUZ;
|
|
constexpr int PrimitiveType_ARRAYSIZE = PrimitiveType_MAX + 1;
|
|
|
|
const ::PROTOBUF_NAMESPACE_ID::EnumDescriptor* PrimitiveType_descriptor();
|
|
template<typename T>
|
|
inline const std::string& PrimitiveType_Name(T enum_t_value) {
|
|
static_assert(::std::is_same<T, PrimitiveType>::value ||
|
|
::std::is_integral<T>::value,
|
|
"Incorrect type passed to function PrimitiveType_Name.");
|
|
return ::PROTOBUF_NAMESPACE_ID::internal::NameOfEnum(
|
|
PrimitiveType_descriptor(), enum_t_value);
|
|
}
|
|
inline bool PrimitiveType_Parse(
|
|
::PROTOBUF_NAMESPACE_ID::ConstStringParam name, PrimitiveType* value) {
|
|
return ::PROTOBUF_NAMESPACE_ID::internal::ParseNamedEnum<PrimitiveType>(
|
|
PrimitiveType_descriptor(), name, value);
|
|
}
|
|
enum DimLevelType : int {
|
|
DIM_DENSE = 0,
|
|
DIM_COMPRESSED = 1,
|
|
DIM_SINGLETON = 2,
|
|
DIM_LOOSE_COMPRESSED = 3,
|
|
DimLevelType_INT_MIN_SENTINEL_DO_NOT_USE_ = std::numeric_limits<int32_t>::min(),
|
|
DimLevelType_INT_MAX_SENTINEL_DO_NOT_USE_ = std::numeric_limits<int32_t>::max()
|
|
};
|
|
bool DimLevelType_IsValid(int value);
|
|
constexpr DimLevelType DimLevelType_MIN = DIM_DENSE;
|
|
constexpr DimLevelType DimLevelType_MAX = DIM_LOOSE_COMPRESSED;
|
|
constexpr int DimLevelType_ARRAYSIZE = DimLevelType_MAX + 1;
|
|
|
|
const ::PROTOBUF_NAMESPACE_ID::EnumDescriptor* DimLevelType_descriptor();
|
|
template<typename T>
|
|
inline const std::string& DimLevelType_Name(T enum_t_value) {
|
|
static_assert(::std::is_same<T, DimLevelType>::value ||
|
|
::std::is_integral<T>::value,
|
|
"Incorrect type passed to function DimLevelType_Name.");
|
|
return ::PROTOBUF_NAMESPACE_ID::internal::NameOfEnum(
|
|
DimLevelType_descriptor(), enum_t_value);
|
|
}
|
|
inline bool DimLevelType_Parse(
|
|
::PROTOBUF_NAMESPACE_ID::ConstStringParam name, DimLevelType* value) {
|
|
return ::PROTOBUF_NAMESPACE_ID::internal::ParseNamedEnum<DimLevelType>(
|
|
DimLevelType_descriptor(), name, value);
|
|
}
|
|
enum ProfileType : int {
|
|
INVALID = 0,
|
|
WINDOW = 1,
|
|
FLAG = 2,
|
|
INTEGER = 3,
|
|
ProfileType_INT_MIN_SENTINEL_DO_NOT_USE_ = std::numeric_limits<int32_t>::min(),
|
|
ProfileType_INT_MAX_SENTINEL_DO_NOT_USE_ = std::numeric_limits<int32_t>::max()
|
|
};
|
|
bool ProfileType_IsValid(int value);
|
|
constexpr ProfileType ProfileType_MIN = INVALID;
|
|
constexpr ProfileType ProfileType_MAX = INTEGER;
|
|
constexpr int ProfileType_ARRAYSIZE = ProfileType_MAX + 1;
|
|
|
|
const ::PROTOBUF_NAMESPACE_ID::EnumDescriptor* ProfileType_descriptor();
|
|
template<typename T>
|
|
inline const std::string& ProfileType_Name(T enum_t_value) {
|
|
static_assert(::std::is_same<T, ProfileType>::value ||
|
|
::std::is_integral<T>::value,
|
|
"Incorrect type passed to function ProfileType_Name.");
|
|
return ::PROTOBUF_NAMESPACE_ID::internal::NameOfEnum(
|
|
ProfileType_descriptor(), enum_t_value);
|
|
}
|
|
inline bool ProfileType_Parse(
|
|
::PROTOBUF_NAMESPACE_ID::ConstStringParam name, ProfileType* value) {
|
|
return ::PROTOBUF_NAMESPACE_ID::internal::ParseNamedEnum<ProfileType>(
|
|
ProfileType_descriptor(), name, value);
|
|
}
|
|
enum ProfileSource : int {
|
|
PROFILE_SOURCE_UNKNOWN_SOURCE = 0,
|
|
PROFILE_SOURCE_EMBEDDED = 1,
|
|
PROFILE_SOURCE_REMOTE = 2,
|
|
ProfileSource_INT_MIN_SENTINEL_DO_NOT_USE_ = std::numeric_limits<int32_t>::min(),
|
|
ProfileSource_INT_MAX_SENTINEL_DO_NOT_USE_ = std::numeric_limits<int32_t>::max()
|
|
};
|
|
bool ProfileSource_IsValid(int value);
|
|
constexpr ProfileSource ProfileSource_MIN = PROFILE_SOURCE_UNKNOWN_SOURCE;
|
|
constexpr ProfileSource ProfileSource_MAX = PROFILE_SOURCE_REMOTE;
|
|
constexpr int ProfileSource_ARRAYSIZE = ProfileSource_MAX + 1;
|
|
|
|
const ::PROTOBUF_NAMESPACE_ID::EnumDescriptor* ProfileSource_descriptor();
|
|
template<typename T>
|
|
inline const std::string& ProfileSource_Name(T enum_t_value) {
|
|
static_assert(::std::is_same<T, ProfileSource>::value ||
|
|
::std::is_integral<T>::value,
|
|
"Incorrect type passed to function ProfileSource_Name.");
|
|
return ::PROTOBUF_NAMESPACE_ID::internal::NameOfEnum(
|
|
ProfileSource_descriptor(), enum_t_value);
|
|
}
|
|
inline bool ProfileSource_Parse(
|
|
::PROTOBUF_NAMESPACE_ID::ConstStringParam name, ProfileSource* value) {
|
|
return ::PROTOBUF_NAMESPACE_ID::internal::ParseNamedEnum<ProfileSource>(
|
|
ProfileSource_descriptor(), name, value);
|
|
}
|
|
enum CompilationEvent : int {
|
|
COMPILATION_EVENT_UNKNOWN_EVENT = 0,
|
|
COMPILATION_EVENT_FIRST_COMPILATION = 1,
|
|
COMPILATION_EVENT_RECOMPILATION = 2,
|
|
CompilationEvent_INT_MIN_SENTINEL_DO_NOT_USE_ = std::numeric_limits<int32_t>::min(),
|
|
CompilationEvent_INT_MAX_SENTINEL_DO_NOT_USE_ = std::numeric_limits<int32_t>::max()
|
|
};
|
|
bool CompilationEvent_IsValid(int value);
|
|
constexpr CompilationEvent CompilationEvent_MIN = COMPILATION_EVENT_UNKNOWN_EVENT;
|
|
constexpr CompilationEvent CompilationEvent_MAX = COMPILATION_EVENT_RECOMPILATION;
|
|
constexpr int CompilationEvent_ARRAYSIZE = CompilationEvent_MAX + 1;
|
|
|
|
const ::PROTOBUF_NAMESPACE_ID::EnumDescriptor* CompilationEvent_descriptor();
|
|
template<typename T>
|
|
inline const std::string& CompilationEvent_Name(T enum_t_value) {
|
|
static_assert(::std::is_same<T, CompilationEvent>::value ||
|
|
::std::is_integral<T>::value,
|
|
"Incorrect type passed to function CompilationEvent_Name.");
|
|
return ::PROTOBUF_NAMESPACE_ID::internal::NameOfEnum(
|
|
CompilationEvent_descriptor(), enum_t_value);
|
|
}
|
|
inline bool CompilationEvent_Parse(
|
|
::PROTOBUF_NAMESPACE_ID::ConstStringParam name, CompilationEvent* value) {
|
|
return ::PROTOBUF_NAMESPACE_ID::internal::ParseNamedEnum<CompilationEvent>(
|
|
CompilationEvent_descriptor(), name, value);
|
|
}
|
|
enum PaddingType : int {
|
|
PADDING_INVALID = 0,
|
|
PADDING_VALID = 1,
|
|
PADDING_SAME = 2,
|
|
PaddingType_INT_MIN_SENTINEL_DO_NOT_USE_ = std::numeric_limits<int32_t>::min(),
|
|
PaddingType_INT_MAX_SENTINEL_DO_NOT_USE_ = std::numeric_limits<int32_t>::max()
|
|
};
|
|
bool PaddingType_IsValid(int value);
|
|
constexpr PaddingType PaddingType_MIN = PADDING_INVALID;
|
|
constexpr PaddingType PaddingType_MAX = PADDING_SAME;
|
|
constexpr int PaddingType_ARRAYSIZE = PaddingType_MAX + 1;
|
|
|
|
const ::PROTOBUF_NAMESPACE_ID::EnumDescriptor* PaddingType_descriptor();
|
|
template<typename T>
|
|
inline const std::string& PaddingType_Name(T enum_t_value) {
|
|
static_assert(::std::is_same<T, PaddingType>::value ||
|
|
::std::is_integral<T>::value,
|
|
"Incorrect type passed to function PaddingType_Name.");
|
|
return ::PROTOBUF_NAMESPACE_ID::internal::NameOfEnum(
|
|
PaddingType_descriptor(), enum_t_value);
|
|
}
|
|
inline bool PaddingType_Parse(
|
|
::PROTOBUF_NAMESPACE_ID::ConstStringParam name, PaddingType* value) {
|
|
return ::PROTOBUF_NAMESPACE_ID::internal::ParseNamedEnum<PaddingType>(
|
|
PaddingType_descriptor(), name, value);
|
|
}
|
|
enum FftType : int {
|
|
FFT = 0,
|
|
IFFT = 1,
|
|
RFFT = 2,
|
|
IRFFT = 3,
|
|
FftType_INT_MIN_SENTINEL_DO_NOT_USE_ = std::numeric_limits<int32_t>::min(),
|
|
FftType_INT_MAX_SENTINEL_DO_NOT_USE_ = std::numeric_limits<int32_t>::max()
|
|
};
|
|
bool FftType_IsValid(int value);
|
|
constexpr FftType FftType_MIN = FFT;
|
|
constexpr FftType FftType_MAX = IRFFT;
|
|
constexpr int FftType_ARRAYSIZE = FftType_MAX + 1;
|
|
|
|
const ::PROTOBUF_NAMESPACE_ID::EnumDescriptor* FftType_descriptor();
|
|
template<typename T>
|
|
inline const std::string& FftType_Name(T enum_t_value) {
|
|
static_assert(::std::is_same<T, FftType>::value ||
|
|
::std::is_integral<T>::value,
|
|
"Incorrect type passed to function FftType_Name.");
|
|
return ::PROTOBUF_NAMESPACE_ID::internal::NameOfEnum(
|
|
FftType_descriptor(), enum_t_value);
|
|
}
|
|
inline bool FftType_Parse(
|
|
::PROTOBUF_NAMESPACE_ID::ConstStringParam name, FftType* value) {
|
|
return ::PROTOBUF_NAMESPACE_ID::internal::ParseNamedEnum<FftType>(
|
|
FftType_descriptor(), name, value);
|
|
}
|
|
enum SparsityType : int {
|
|
SPARSITY_INVALID = 0,
|
|
SPARSITY_STRUCTURED_N_M = 1,
|
|
SparsityType_INT_MIN_SENTINEL_DO_NOT_USE_ = std::numeric_limits<int32_t>::min(),
|
|
SparsityType_INT_MAX_SENTINEL_DO_NOT_USE_ = std::numeric_limits<int32_t>::max()
|
|
};
|
|
bool SparsityType_IsValid(int value);
|
|
constexpr SparsityType SparsityType_MIN = SPARSITY_INVALID;
|
|
constexpr SparsityType SparsityType_MAX = SPARSITY_STRUCTURED_N_M;
|
|
constexpr int SparsityType_ARRAYSIZE = SparsityType_MAX + 1;
|
|
|
|
const ::PROTOBUF_NAMESPACE_ID::EnumDescriptor* SparsityType_descriptor();
|
|
template<typename T>
|
|
inline const std::string& SparsityType_Name(T enum_t_value) {
|
|
static_assert(::std::is_same<T, SparsityType>::value ||
|
|
::std::is_integral<T>::value,
|
|
"Incorrect type passed to function SparsityType_Name.");
|
|
return ::PROTOBUF_NAMESPACE_ID::internal::NameOfEnum(
|
|
SparsityType_descriptor(), enum_t_value);
|
|
}
|
|
inline bool SparsityType_Parse(
|
|
::PROTOBUF_NAMESPACE_ID::ConstStringParam name, SparsityType* value) {
|
|
return ::PROTOBUF_NAMESPACE_ID::internal::ParseNamedEnum<SparsityType>(
|
|
SparsityType_descriptor(), name, value);
|
|
}
|
|
enum RandomDistribution : int {
|
|
RNG_INVALID = 0,
|
|
RNG_UNIFORM = 1,
|
|
RNG_NORMAL = 2,
|
|
RandomDistribution_INT_MIN_SENTINEL_DO_NOT_USE_ = std::numeric_limits<int32_t>::min(),
|
|
RandomDistribution_INT_MAX_SENTINEL_DO_NOT_USE_ = std::numeric_limits<int32_t>::max()
|
|
};
|
|
bool RandomDistribution_IsValid(int value);
|
|
constexpr RandomDistribution RandomDistribution_MIN = RNG_INVALID;
|
|
constexpr RandomDistribution RandomDistribution_MAX = RNG_NORMAL;
|
|
constexpr int RandomDistribution_ARRAYSIZE = RandomDistribution_MAX + 1;
|
|
|
|
const ::PROTOBUF_NAMESPACE_ID::EnumDescriptor* RandomDistribution_descriptor();
|
|
template<typename T>
|
|
inline const std::string& RandomDistribution_Name(T enum_t_value) {
|
|
static_assert(::std::is_same<T, RandomDistribution>::value ||
|
|
::std::is_integral<T>::value,
|
|
"Incorrect type passed to function RandomDistribution_Name.");
|
|
return ::PROTOBUF_NAMESPACE_ID::internal::NameOfEnum(
|
|
RandomDistribution_descriptor(), enum_t_value);
|
|
}
|
|
inline bool RandomDistribution_Parse(
|
|
::PROTOBUF_NAMESPACE_ID::ConstStringParam name, RandomDistribution* value) {
|
|
return ::PROTOBUF_NAMESPACE_ID::internal::ParseNamedEnum<RandomDistribution>(
|
|
RandomDistribution_descriptor(), name, value);
|
|
}
|
|
enum RandomAlgorithm : int {
|
|
RNG_DEFAULT = 0,
|
|
RNG_THREE_FRY = 1,
|
|
RNG_PHILOX = 2,
|
|
RandomAlgorithm_INT_MIN_SENTINEL_DO_NOT_USE_ = std::numeric_limits<int32_t>::min(),
|
|
RandomAlgorithm_INT_MAX_SENTINEL_DO_NOT_USE_ = std::numeric_limits<int32_t>::max()
|
|
};
|
|
bool RandomAlgorithm_IsValid(int value);
|
|
constexpr RandomAlgorithm RandomAlgorithm_MIN = RNG_DEFAULT;
|
|
constexpr RandomAlgorithm RandomAlgorithm_MAX = RNG_PHILOX;
|
|
constexpr int RandomAlgorithm_ARRAYSIZE = RandomAlgorithm_MAX + 1;
|
|
|
|
const ::PROTOBUF_NAMESPACE_ID::EnumDescriptor* RandomAlgorithm_descriptor();
|
|
template<typename T>
|
|
inline const std::string& RandomAlgorithm_Name(T enum_t_value) {
|
|
static_assert(::std::is_same<T, RandomAlgorithm>::value ||
|
|
::std::is_integral<T>::value,
|
|
"Incorrect type passed to function RandomAlgorithm_Name.");
|
|
return ::PROTOBUF_NAMESPACE_ID::internal::NameOfEnum(
|
|
RandomAlgorithm_descriptor(), enum_t_value);
|
|
}
|
|
inline bool RandomAlgorithm_Parse(
|
|
::PROTOBUF_NAMESPACE_ID::ConstStringParam name, RandomAlgorithm* value) {
|
|
return ::PROTOBUF_NAMESPACE_ID::internal::ParseNamedEnum<RandomAlgorithm>(
|
|
RandomAlgorithm_descriptor(), name, value);
|
|
}
|
|
// ===================================================================
|
|
|
|
class PaddingConfig_PaddingConfigDimension final :
|
|
public ::PROTOBUF_NAMESPACE_ID::Message /* @@protoc_insertion_point(class_definition:xla.PaddingConfig.PaddingConfigDimension) */ {
|
|
public:
|
|
inline PaddingConfig_PaddingConfigDimension() : PaddingConfig_PaddingConfigDimension(nullptr) {}
|
|
~PaddingConfig_PaddingConfigDimension() override;
|
|
explicit PROTOBUF_CONSTEXPR PaddingConfig_PaddingConfigDimension(::PROTOBUF_NAMESPACE_ID::internal::ConstantInitialized);
|
|
|
|
PaddingConfig_PaddingConfigDimension(const PaddingConfig_PaddingConfigDimension& from);
|
|
PaddingConfig_PaddingConfigDimension(PaddingConfig_PaddingConfigDimension&& from) noexcept
|
|
: PaddingConfig_PaddingConfigDimension() {
|
|
*this = ::std::move(from);
|
|
}
|
|
|
|
inline PaddingConfig_PaddingConfigDimension& operator=(const PaddingConfig_PaddingConfigDimension& from) {
|
|
CopyFrom(from);
|
|
return *this;
|
|
}
|
|
inline PaddingConfig_PaddingConfigDimension& operator=(PaddingConfig_PaddingConfigDimension&& from) noexcept {
|
|
if (this == &from) return *this;
|
|
if (GetOwningArena() == from.GetOwningArena()
|
|
#ifdef PROTOBUF_FORCE_COPY_IN_MOVE
|
|
&& GetOwningArena() != nullptr
|
|
#endif // !PROTOBUF_FORCE_COPY_IN_MOVE
|
|
) {
|
|
InternalSwap(&from);
|
|
} else {
|
|
CopyFrom(from);
|
|
}
|
|
return *this;
|
|
}
|
|
|
|
static const ::PROTOBUF_NAMESPACE_ID::Descriptor* descriptor() {
|
|
return GetDescriptor();
|
|
}
|
|
static const ::PROTOBUF_NAMESPACE_ID::Descriptor* GetDescriptor() {
|
|
return default_instance().GetMetadata().descriptor;
|
|
}
|
|
static const ::PROTOBUF_NAMESPACE_ID::Reflection* GetReflection() {
|
|
return default_instance().GetMetadata().reflection;
|
|
}
|
|
static const PaddingConfig_PaddingConfigDimension& default_instance() {
|
|
return *internal_default_instance();
|
|
}
|
|
static inline const PaddingConfig_PaddingConfigDimension* internal_default_instance() {
|
|
return reinterpret_cast<const PaddingConfig_PaddingConfigDimension*>(
|
|
&_PaddingConfig_PaddingConfigDimension_default_instance_);
|
|
}
|
|
static constexpr int kIndexInFileMessages =
|
|
0;
|
|
|
|
friend void swap(PaddingConfig_PaddingConfigDimension& a, PaddingConfig_PaddingConfigDimension& b) {
|
|
a.Swap(&b);
|
|
}
|
|
inline void Swap(PaddingConfig_PaddingConfigDimension* other) {
|
|
if (other == this) return;
|
|
#ifdef PROTOBUF_FORCE_COPY_IN_SWAP
|
|
if (GetOwningArena() != nullptr &&
|
|
GetOwningArena() == other->GetOwningArena()) {
|
|
#else // PROTOBUF_FORCE_COPY_IN_SWAP
|
|
if (GetOwningArena() == other->GetOwningArena()) {
|
|
#endif // !PROTOBUF_FORCE_COPY_IN_SWAP
|
|
InternalSwap(other);
|
|
} else {
|
|
::PROTOBUF_NAMESPACE_ID::internal::GenericSwap(this, other);
|
|
}
|
|
}
|
|
void UnsafeArenaSwap(PaddingConfig_PaddingConfigDimension* other) {
|
|
if (other == this) return;
|
|
GOOGLE_DCHECK(GetOwningArena() == other->GetOwningArena());
|
|
InternalSwap(other);
|
|
}
|
|
|
|
// implements Message ----------------------------------------------
|
|
|
|
PaddingConfig_PaddingConfigDimension* New(::PROTOBUF_NAMESPACE_ID::Arena* arena = nullptr) const final {
|
|
return CreateMaybeMessage<PaddingConfig_PaddingConfigDimension>(arena);
|
|
}
|
|
using ::PROTOBUF_NAMESPACE_ID::Message::CopyFrom;
|
|
void CopyFrom(const PaddingConfig_PaddingConfigDimension& from);
|
|
using ::PROTOBUF_NAMESPACE_ID::Message::MergeFrom;
|
|
void MergeFrom( const PaddingConfig_PaddingConfigDimension& from) {
|
|
PaddingConfig_PaddingConfigDimension::MergeImpl(*this, from);
|
|
}
|
|
private:
|
|
static void MergeImpl(::PROTOBUF_NAMESPACE_ID::Message& to_msg, const ::PROTOBUF_NAMESPACE_ID::Message& from_msg);
|
|
public:
|
|
PROTOBUF_ATTRIBUTE_REINITIALIZES void Clear() final;
|
|
bool IsInitialized() const final;
|
|
|
|
size_t ByteSizeLong() const final;
|
|
const char* _InternalParse(const char* ptr, ::PROTOBUF_NAMESPACE_ID::internal::ParseContext* ctx) final;
|
|
uint8_t* _InternalSerialize(
|
|
uint8_t* target, ::PROTOBUF_NAMESPACE_ID::io::EpsCopyOutputStream* stream) const final;
|
|
int GetCachedSize() const final { return _impl_._cached_size_.Get(); }
|
|
|
|
private:
|
|
void SharedCtor(::PROTOBUF_NAMESPACE_ID::Arena* arena, bool is_message_owned);
|
|
void SharedDtor();
|
|
void SetCachedSize(int size) const final;
|
|
void InternalSwap(PaddingConfig_PaddingConfigDimension* other);
|
|
|
|
private:
|
|
friend class ::PROTOBUF_NAMESPACE_ID::internal::AnyMetadata;
|
|
static ::PROTOBUF_NAMESPACE_ID::StringPiece FullMessageName() {
|
|
return "xla.PaddingConfig.PaddingConfigDimension";
|
|
}
|
|
protected:
|
|
explicit PaddingConfig_PaddingConfigDimension(::PROTOBUF_NAMESPACE_ID::Arena* arena,
|
|
bool is_message_owned = false);
|
|
public:
|
|
|
|
static const ClassData _class_data_;
|
|
const ::PROTOBUF_NAMESPACE_ID::Message::ClassData*GetClassData() const final;
|
|
|
|
::PROTOBUF_NAMESPACE_ID::Metadata GetMetadata() const final;
|
|
|
|
// nested types ----------------------------------------------------
|
|
|
|
// accessors -------------------------------------------------------
|
|
|
|
enum : int {
|
|
kEdgePaddingLowFieldNumber = 1,
|
|
kEdgePaddingHighFieldNumber = 2,
|
|
kInteriorPaddingFieldNumber = 3,
|
|
};
|
|
// int64 edge_padding_low = 1;
|
|
void clear_edge_padding_low();
|
|
int64_t edge_padding_low() const;
|
|
void set_edge_padding_low(int64_t value);
|
|
private:
|
|
int64_t _internal_edge_padding_low() const;
|
|
void _internal_set_edge_padding_low(int64_t value);
|
|
public:
|
|
|
|
// int64 edge_padding_high = 2;
|
|
void clear_edge_padding_high();
|
|
int64_t edge_padding_high() const;
|
|
void set_edge_padding_high(int64_t value);
|
|
private:
|
|
int64_t _internal_edge_padding_high() const;
|
|
void _internal_set_edge_padding_high(int64_t value);
|
|
public:
|
|
|
|
// int64 interior_padding = 3;
|
|
void clear_interior_padding();
|
|
int64_t interior_padding() const;
|
|
void set_interior_padding(int64_t value);
|
|
private:
|
|
int64_t _internal_interior_padding() const;
|
|
void _internal_set_interior_padding(int64_t value);
|
|
public:
|
|
|
|
// @@protoc_insertion_point(class_scope:xla.PaddingConfig.PaddingConfigDimension)
|
|
private:
|
|
class _Internal;
|
|
|
|
template <typename T> friend class ::PROTOBUF_NAMESPACE_ID::Arena::InternalHelper;
|
|
typedef void InternalArenaConstructable_;
|
|
typedef void DestructorSkippable_;
|
|
struct Impl_ {
|
|
int64_t edge_padding_low_;
|
|
int64_t edge_padding_high_;
|
|
int64_t interior_padding_;
|
|
mutable ::PROTOBUF_NAMESPACE_ID::internal::CachedSize _cached_size_;
|
|
};
|
|
union { Impl_ _impl_; };
|
|
friend struct ::TableStruct_xla_2fxla_5fdata_2eproto;
|
|
};
|
|
// -------------------------------------------------------------------
|
|
|
|
class PaddingConfig final :
|
|
public ::PROTOBUF_NAMESPACE_ID::Message /* @@protoc_insertion_point(class_definition:xla.PaddingConfig) */ {
|
|
public:
|
|
inline PaddingConfig() : PaddingConfig(nullptr) {}
|
|
~PaddingConfig() override;
|
|
explicit PROTOBUF_CONSTEXPR PaddingConfig(::PROTOBUF_NAMESPACE_ID::internal::ConstantInitialized);
|
|
|
|
PaddingConfig(const PaddingConfig& from);
|
|
PaddingConfig(PaddingConfig&& from) noexcept
|
|
: PaddingConfig() {
|
|
*this = ::std::move(from);
|
|
}
|
|
|
|
inline PaddingConfig& operator=(const PaddingConfig& from) {
|
|
CopyFrom(from);
|
|
return *this;
|
|
}
|
|
inline PaddingConfig& operator=(PaddingConfig&& from) noexcept {
|
|
if (this == &from) return *this;
|
|
if (GetOwningArena() == from.GetOwningArena()
|
|
#ifdef PROTOBUF_FORCE_COPY_IN_MOVE
|
|
&& GetOwningArena() != nullptr
|
|
#endif // !PROTOBUF_FORCE_COPY_IN_MOVE
|
|
) {
|
|
InternalSwap(&from);
|
|
} else {
|
|
CopyFrom(from);
|
|
}
|
|
return *this;
|
|
}
|
|
|
|
static const ::PROTOBUF_NAMESPACE_ID::Descriptor* descriptor() {
|
|
return GetDescriptor();
|
|
}
|
|
static const ::PROTOBUF_NAMESPACE_ID::Descriptor* GetDescriptor() {
|
|
return default_instance().GetMetadata().descriptor;
|
|
}
|
|
static const ::PROTOBUF_NAMESPACE_ID::Reflection* GetReflection() {
|
|
return default_instance().GetMetadata().reflection;
|
|
}
|
|
static const PaddingConfig& default_instance() {
|
|
return *internal_default_instance();
|
|
}
|
|
static inline const PaddingConfig* internal_default_instance() {
|
|
return reinterpret_cast<const PaddingConfig*>(
|
|
&_PaddingConfig_default_instance_);
|
|
}
|
|
static constexpr int kIndexInFileMessages =
|
|
1;
|
|
|
|
friend void swap(PaddingConfig& a, PaddingConfig& b) {
|
|
a.Swap(&b);
|
|
}
|
|
inline void Swap(PaddingConfig* other) {
|
|
if (other == this) return;
|
|
#ifdef PROTOBUF_FORCE_COPY_IN_SWAP
|
|
if (GetOwningArena() != nullptr &&
|
|
GetOwningArena() == other->GetOwningArena()) {
|
|
#else // PROTOBUF_FORCE_COPY_IN_SWAP
|
|
if (GetOwningArena() == other->GetOwningArena()) {
|
|
#endif // !PROTOBUF_FORCE_COPY_IN_SWAP
|
|
InternalSwap(other);
|
|
} else {
|
|
::PROTOBUF_NAMESPACE_ID::internal::GenericSwap(this, other);
|
|
}
|
|
}
|
|
void UnsafeArenaSwap(PaddingConfig* other) {
|
|
if (other == this) return;
|
|
GOOGLE_DCHECK(GetOwningArena() == other->GetOwningArena());
|
|
InternalSwap(other);
|
|
}
|
|
|
|
// implements Message ----------------------------------------------
|
|
|
|
PaddingConfig* New(::PROTOBUF_NAMESPACE_ID::Arena* arena = nullptr) const final {
|
|
return CreateMaybeMessage<PaddingConfig>(arena);
|
|
}
|
|
using ::PROTOBUF_NAMESPACE_ID::Message::CopyFrom;
|
|
void CopyFrom(const PaddingConfig& from);
|
|
using ::PROTOBUF_NAMESPACE_ID::Message::MergeFrom;
|
|
void MergeFrom( const PaddingConfig& from) {
|
|
PaddingConfig::MergeImpl(*this, from);
|
|
}
|
|
private:
|
|
static void MergeImpl(::PROTOBUF_NAMESPACE_ID::Message& to_msg, const ::PROTOBUF_NAMESPACE_ID::Message& from_msg);
|
|
public:
|
|
PROTOBUF_ATTRIBUTE_REINITIALIZES void Clear() final;
|
|
bool IsInitialized() const final;
|
|
|
|
size_t ByteSizeLong() const final;
|
|
const char* _InternalParse(const char* ptr, ::PROTOBUF_NAMESPACE_ID::internal::ParseContext* ctx) final;
|
|
uint8_t* _InternalSerialize(
|
|
uint8_t* target, ::PROTOBUF_NAMESPACE_ID::io::EpsCopyOutputStream* stream) const final;
|
|
int GetCachedSize() const final { return _impl_._cached_size_.Get(); }
|
|
|
|
private:
|
|
void SharedCtor(::PROTOBUF_NAMESPACE_ID::Arena* arena, bool is_message_owned);
|
|
void SharedDtor();
|
|
void SetCachedSize(int size) const final;
|
|
void InternalSwap(PaddingConfig* other);
|
|
|
|
private:
|
|
friend class ::PROTOBUF_NAMESPACE_ID::internal::AnyMetadata;
|
|
static ::PROTOBUF_NAMESPACE_ID::StringPiece FullMessageName() {
|
|
return "xla.PaddingConfig";
|
|
}
|
|
protected:
|
|
explicit PaddingConfig(::PROTOBUF_NAMESPACE_ID::Arena* arena,
|
|
bool is_message_owned = false);
|
|
public:
|
|
|
|
static const ClassData _class_data_;
|
|
const ::PROTOBUF_NAMESPACE_ID::Message::ClassData*GetClassData() const final;
|
|
|
|
::PROTOBUF_NAMESPACE_ID::Metadata GetMetadata() const final;
|
|
|
|
// nested types ----------------------------------------------------
|
|
|
|
typedef PaddingConfig_PaddingConfigDimension PaddingConfigDimension;
|
|
|
|
// accessors -------------------------------------------------------
|
|
|
|
enum : int {
|
|
kDimensionsFieldNumber = 1,
|
|
};
|
|
// repeated .xla.PaddingConfig.PaddingConfigDimension dimensions = 1;
|
|
int dimensions_size() const;
|
|
private:
|
|
int _internal_dimensions_size() const;
|
|
public:
|
|
void clear_dimensions();
|
|
::xla::PaddingConfig_PaddingConfigDimension* mutable_dimensions(int index);
|
|
::PROTOBUF_NAMESPACE_ID::RepeatedPtrField< ::xla::PaddingConfig_PaddingConfigDimension >*
|
|
mutable_dimensions();
|
|
private:
|
|
const ::xla::PaddingConfig_PaddingConfigDimension& _internal_dimensions(int index) const;
|
|
::xla::PaddingConfig_PaddingConfigDimension* _internal_add_dimensions();
|
|
public:
|
|
const ::xla::PaddingConfig_PaddingConfigDimension& dimensions(int index) const;
|
|
::xla::PaddingConfig_PaddingConfigDimension* add_dimensions();
|
|
const ::PROTOBUF_NAMESPACE_ID::RepeatedPtrField< ::xla::PaddingConfig_PaddingConfigDimension >&
|
|
dimensions() const;
|
|
|
|
// @@protoc_insertion_point(class_scope:xla.PaddingConfig)
|
|
private:
|
|
class _Internal;
|
|
|
|
template <typename T> friend class ::PROTOBUF_NAMESPACE_ID::Arena::InternalHelper;
|
|
typedef void InternalArenaConstructable_;
|
|
typedef void DestructorSkippable_;
|
|
struct Impl_ {
|
|
::PROTOBUF_NAMESPACE_ID::RepeatedPtrField< ::xla::PaddingConfig_PaddingConfigDimension > dimensions_;
|
|
mutable ::PROTOBUF_NAMESPACE_ID::internal::CachedSize _cached_size_;
|
|
};
|
|
union { Impl_ _impl_; };
|
|
friend struct ::TableStruct_xla_2fxla_5fdata_2eproto;
|
|
};
|
|
// -------------------------------------------------------------------
|
|
|
|
class TileProto final :
|
|
public ::PROTOBUF_NAMESPACE_ID::Message /* @@protoc_insertion_point(class_definition:xla.TileProto) */ {
|
|
public:
|
|
inline TileProto() : TileProto(nullptr) {}
|
|
~TileProto() override;
|
|
explicit PROTOBUF_CONSTEXPR TileProto(::PROTOBUF_NAMESPACE_ID::internal::ConstantInitialized);
|
|
|
|
TileProto(const TileProto& from);
|
|
TileProto(TileProto&& from) noexcept
|
|
: TileProto() {
|
|
*this = ::std::move(from);
|
|
}
|
|
|
|
inline TileProto& operator=(const TileProto& from) {
|
|
CopyFrom(from);
|
|
return *this;
|
|
}
|
|
inline TileProto& operator=(TileProto&& from) noexcept {
|
|
if (this == &from) return *this;
|
|
if (GetOwningArena() == from.GetOwningArena()
|
|
#ifdef PROTOBUF_FORCE_COPY_IN_MOVE
|
|
&& GetOwningArena() != nullptr
|
|
#endif // !PROTOBUF_FORCE_COPY_IN_MOVE
|
|
) {
|
|
InternalSwap(&from);
|
|
} else {
|
|
CopyFrom(from);
|
|
}
|
|
return *this;
|
|
}
|
|
|
|
static const ::PROTOBUF_NAMESPACE_ID::Descriptor* descriptor() {
|
|
return GetDescriptor();
|
|
}
|
|
static const ::PROTOBUF_NAMESPACE_ID::Descriptor* GetDescriptor() {
|
|
return default_instance().GetMetadata().descriptor;
|
|
}
|
|
static const ::PROTOBUF_NAMESPACE_ID::Reflection* GetReflection() {
|
|
return default_instance().GetMetadata().reflection;
|
|
}
|
|
static const TileProto& default_instance() {
|
|
return *internal_default_instance();
|
|
}
|
|
static inline const TileProto* internal_default_instance() {
|
|
return reinterpret_cast<const TileProto*>(
|
|
&_TileProto_default_instance_);
|
|
}
|
|
static constexpr int kIndexInFileMessages =
|
|
2;
|
|
|
|
friend void swap(TileProto& a, TileProto& b) {
|
|
a.Swap(&b);
|
|
}
|
|
inline void Swap(TileProto* other) {
|
|
if (other == this) return;
|
|
#ifdef PROTOBUF_FORCE_COPY_IN_SWAP
|
|
if (GetOwningArena() != nullptr &&
|
|
GetOwningArena() == other->GetOwningArena()) {
|
|
#else // PROTOBUF_FORCE_COPY_IN_SWAP
|
|
if (GetOwningArena() == other->GetOwningArena()) {
|
|
#endif // !PROTOBUF_FORCE_COPY_IN_SWAP
|
|
InternalSwap(other);
|
|
} else {
|
|
::PROTOBUF_NAMESPACE_ID::internal::GenericSwap(this, other);
|
|
}
|
|
}
|
|
void UnsafeArenaSwap(TileProto* other) {
|
|
if (other == this) return;
|
|
GOOGLE_DCHECK(GetOwningArena() == other->GetOwningArena());
|
|
InternalSwap(other);
|
|
}
|
|
|
|
// implements Message ----------------------------------------------
|
|
|
|
TileProto* New(::PROTOBUF_NAMESPACE_ID::Arena* arena = nullptr) const final {
|
|
return CreateMaybeMessage<TileProto>(arena);
|
|
}
|
|
using ::PROTOBUF_NAMESPACE_ID::Message::CopyFrom;
|
|
void CopyFrom(const TileProto& from);
|
|
using ::PROTOBUF_NAMESPACE_ID::Message::MergeFrom;
|
|
void MergeFrom( const TileProto& from) {
|
|
TileProto::MergeImpl(*this, from);
|
|
}
|
|
private:
|
|
static void MergeImpl(::PROTOBUF_NAMESPACE_ID::Message& to_msg, const ::PROTOBUF_NAMESPACE_ID::Message& from_msg);
|
|
public:
|
|
PROTOBUF_ATTRIBUTE_REINITIALIZES void Clear() final;
|
|
bool IsInitialized() const final;
|
|
|
|
size_t ByteSizeLong() const final;
|
|
const char* _InternalParse(const char* ptr, ::PROTOBUF_NAMESPACE_ID::internal::ParseContext* ctx) final;
|
|
uint8_t* _InternalSerialize(
|
|
uint8_t* target, ::PROTOBUF_NAMESPACE_ID::io::EpsCopyOutputStream* stream) const final;
|
|
int GetCachedSize() const final { return _impl_._cached_size_.Get(); }
|
|
|
|
private:
|
|
void SharedCtor(::PROTOBUF_NAMESPACE_ID::Arena* arena, bool is_message_owned);
|
|
void SharedDtor();
|
|
void SetCachedSize(int size) const final;
|
|
void InternalSwap(TileProto* other);
|
|
|
|
private:
|
|
friend class ::PROTOBUF_NAMESPACE_ID::internal::AnyMetadata;
|
|
static ::PROTOBUF_NAMESPACE_ID::StringPiece FullMessageName() {
|
|
return "xla.TileProto";
|
|
}
|
|
protected:
|
|
explicit TileProto(::PROTOBUF_NAMESPACE_ID::Arena* arena,
|
|
bool is_message_owned = false);
|
|
public:
|
|
|
|
static const ClassData _class_data_;
|
|
const ::PROTOBUF_NAMESPACE_ID::Message::ClassData*GetClassData() const final;
|
|
|
|
::PROTOBUF_NAMESPACE_ID::Metadata GetMetadata() const final;
|
|
|
|
// nested types ----------------------------------------------------
|
|
|
|
// accessors -------------------------------------------------------
|
|
|
|
enum : int {
|
|
kDimensionsFieldNumber = 1,
|
|
};
|
|
// repeated int64 dimensions = 1;
|
|
int dimensions_size() const;
|
|
private:
|
|
int _internal_dimensions_size() const;
|
|
public:
|
|
void clear_dimensions();
|
|
private:
|
|
int64_t _internal_dimensions(int index) const;
|
|
const ::PROTOBUF_NAMESPACE_ID::RepeatedField< int64_t >&
|
|
_internal_dimensions() const;
|
|
void _internal_add_dimensions(int64_t value);
|
|
::PROTOBUF_NAMESPACE_ID::RepeatedField< int64_t >*
|
|
_internal_mutable_dimensions();
|
|
public:
|
|
int64_t dimensions(int index) const;
|
|
void set_dimensions(int index, int64_t value);
|
|
void add_dimensions(int64_t value);
|
|
const ::PROTOBUF_NAMESPACE_ID::RepeatedField< int64_t >&
|
|
dimensions() const;
|
|
::PROTOBUF_NAMESPACE_ID::RepeatedField< int64_t >*
|
|
mutable_dimensions();
|
|
|
|
// @@protoc_insertion_point(class_scope:xla.TileProto)
|
|
private:
|
|
class _Internal;
|
|
|
|
template <typename T> friend class ::PROTOBUF_NAMESPACE_ID::Arena::InternalHelper;
|
|
typedef void InternalArenaConstructable_;
|
|
typedef void DestructorSkippable_;
|
|
struct Impl_ {
|
|
::PROTOBUF_NAMESPACE_ID::RepeatedField< int64_t > dimensions_;
|
|
mutable std::atomic<int> _dimensions_cached_byte_size_;
|
|
mutable ::PROTOBUF_NAMESPACE_ID::internal::CachedSize _cached_size_;
|
|
};
|
|
union { Impl_ _impl_; };
|
|
friend struct ::TableStruct_xla_2fxla_5fdata_2eproto;
|
|
};
|
|
// -------------------------------------------------------------------
|
|
|
|
class LayoutProto final :
|
|
public ::PROTOBUF_NAMESPACE_ID::Message /* @@protoc_insertion_point(class_definition:xla.LayoutProto) */ {
|
|
public:
|
|
inline LayoutProto() : LayoutProto(nullptr) {}
|
|
~LayoutProto() override;
|
|
explicit PROTOBUF_CONSTEXPR LayoutProto(::PROTOBUF_NAMESPACE_ID::internal::ConstantInitialized);
|
|
|
|
LayoutProto(const LayoutProto& from);
|
|
LayoutProto(LayoutProto&& from) noexcept
|
|
: LayoutProto() {
|
|
*this = ::std::move(from);
|
|
}
|
|
|
|
inline LayoutProto& operator=(const LayoutProto& from) {
|
|
CopyFrom(from);
|
|
return *this;
|
|
}
|
|
inline LayoutProto& operator=(LayoutProto&& from) noexcept {
|
|
if (this == &from) return *this;
|
|
if (GetOwningArena() == from.GetOwningArena()
|
|
#ifdef PROTOBUF_FORCE_COPY_IN_MOVE
|
|
&& GetOwningArena() != nullptr
|
|
#endif // !PROTOBUF_FORCE_COPY_IN_MOVE
|
|
) {
|
|
InternalSwap(&from);
|
|
} else {
|
|
CopyFrom(from);
|
|
}
|
|
return *this;
|
|
}
|
|
|
|
static const ::PROTOBUF_NAMESPACE_ID::Descriptor* descriptor() {
|
|
return GetDescriptor();
|
|
}
|
|
static const ::PROTOBUF_NAMESPACE_ID::Descriptor* GetDescriptor() {
|
|
return default_instance().GetMetadata().descriptor;
|
|
}
|
|
static const ::PROTOBUF_NAMESPACE_ID::Reflection* GetReflection() {
|
|
return default_instance().GetMetadata().reflection;
|
|
}
|
|
static const LayoutProto& default_instance() {
|
|
return *internal_default_instance();
|
|
}
|
|
static inline const LayoutProto* internal_default_instance() {
|
|
return reinterpret_cast<const LayoutProto*>(
|
|
&_LayoutProto_default_instance_);
|
|
}
|
|
static constexpr int kIndexInFileMessages =
|
|
3;
|
|
|
|
friend void swap(LayoutProto& a, LayoutProto& b) {
|
|
a.Swap(&b);
|
|
}
|
|
inline void Swap(LayoutProto* other) {
|
|
if (other == this) return;
|
|
#ifdef PROTOBUF_FORCE_COPY_IN_SWAP
|
|
if (GetOwningArena() != nullptr &&
|
|
GetOwningArena() == other->GetOwningArena()) {
|
|
#else // PROTOBUF_FORCE_COPY_IN_SWAP
|
|
if (GetOwningArena() == other->GetOwningArena()) {
|
|
#endif // !PROTOBUF_FORCE_COPY_IN_SWAP
|
|
InternalSwap(other);
|
|
} else {
|
|
::PROTOBUF_NAMESPACE_ID::internal::GenericSwap(this, other);
|
|
}
|
|
}
|
|
void UnsafeArenaSwap(LayoutProto* other) {
|
|
if (other == this) return;
|
|
GOOGLE_DCHECK(GetOwningArena() == other->GetOwningArena());
|
|
InternalSwap(other);
|
|
}
|
|
|
|
// implements Message ----------------------------------------------
|
|
|
|
LayoutProto* New(::PROTOBUF_NAMESPACE_ID::Arena* arena = nullptr) const final {
|
|
return CreateMaybeMessage<LayoutProto>(arena);
|
|
}
|
|
using ::PROTOBUF_NAMESPACE_ID::Message::CopyFrom;
|
|
void CopyFrom(const LayoutProto& from);
|
|
using ::PROTOBUF_NAMESPACE_ID::Message::MergeFrom;
|
|
void MergeFrom( const LayoutProto& from) {
|
|
LayoutProto::MergeImpl(*this, from);
|
|
}
|
|
private:
|
|
static void MergeImpl(::PROTOBUF_NAMESPACE_ID::Message& to_msg, const ::PROTOBUF_NAMESPACE_ID::Message& from_msg);
|
|
public:
|
|
PROTOBUF_ATTRIBUTE_REINITIALIZES void Clear() final;
|
|
bool IsInitialized() const final;
|
|
|
|
size_t ByteSizeLong() const final;
|
|
const char* _InternalParse(const char* ptr, ::PROTOBUF_NAMESPACE_ID::internal::ParseContext* ctx) final;
|
|
uint8_t* _InternalSerialize(
|
|
uint8_t* target, ::PROTOBUF_NAMESPACE_ID::io::EpsCopyOutputStream* stream) const final;
|
|
int GetCachedSize() const final { return _impl_._cached_size_.Get(); }
|
|
|
|
private:
|
|
void SharedCtor(::PROTOBUF_NAMESPACE_ID::Arena* arena, bool is_message_owned);
|
|
void SharedDtor();
|
|
void SetCachedSize(int size) const final;
|
|
void InternalSwap(LayoutProto* other);
|
|
|
|
private:
|
|
friend class ::PROTOBUF_NAMESPACE_ID::internal::AnyMetadata;
|
|
static ::PROTOBUF_NAMESPACE_ID::StringPiece FullMessageName() {
|
|
return "xla.LayoutProto";
|
|
}
|
|
protected:
|
|
explicit LayoutProto(::PROTOBUF_NAMESPACE_ID::Arena* arena,
|
|
bool is_message_owned = false);
|
|
public:
|
|
|
|
static const ClassData _class_data_;
|
|
const ::PROTOBUF_NAMESPACE_ID::Message::ClassData*GetClassData() const final;
|
|
|
|
::PROTOBUF_NAMESPACE_ID::Metadata GetMetadata() const final;
|
|
|
|
// nested types ----------------------------------------------------
|
|
|
|
// accessors -------------------------------------------------------
|
|
|
|
enum : int {
|
|
kMinorToMajorFieldNumber = 1,
|
|
kTilesFieldNumber = 6,
|
|
kDimLevelTypesFieldNumber = 9,
|
|
kDimUniqueFieldNumber = 13,
|
|
kDimOrderedFieldNumber = 14,
|
|
kPhysicalShapeFieldNumber = 10,
|
|
kElementSizeInBitsFieldNumber = 7,
|
|
kMemorySpaceFieldNumber = 8,
|
|
kIndexPrimitiveTypeFieldNumber = 11,
|
|
kPointerPrimitiveTypeFieldNumber = 12,
|
|
kDynamicShapeMetadataPrefixBytesFieldNumber = 15,
|
|
kTailPaddingAlignmentInElementsFieldNumber = 16,
|
|
};
|
|
// repeated int64 minor_to_major = 1;
|
|
int minor_to_major_size() const;
|
|
private:
|
|
int _internal_minor_to_major_size() const;
|
|
public:
|
|
void clear_minor_to_major();
|
|
private:
|
|
int64_t _internal_minor_to_major(int index) const;
|
|
const ::PROTOBUF_NAMESPACE_ID::RepeatedField< int64_t >&
|
|
_internal_minor_to_major() const;
|
|
void _internal_add_minor_to_major(int64_t value);
|
|
::PROTOBUF_NAMESPACE_ID::RepeatedField< int64_t >*
|
|
_internal_mutable_minor_to_major();
|
|
public:
|
|
int64_t minor_to_major(int index) const;
|
|
void set_minor_to_major(int index, int64_t value);
|
|
void add_minor_to_major(int64_t value);
|
|
const ::PROTOBUF_NAMESPACE_ID::RepeatedField< int64_t >&
|
|
minor_to_major() const;
|
|
::PROTOBUF_NAMESPACE_ID::RepeatedField< int64_t >*
|
|
mutable_minor_to_major();
|
|
|
|
// repeated .xla.TileProto tiles = 6;
|
|
int tiles_size() const;
|
|
private:
|
|
int _internal_tiles_size() const;
|
|
public:
|
|
void clear_tiles();
|
|
::xla::TileProto* mutable_tiles(int index);
|
|
::PROTOBUF_NAMESPACE_ID::RepeatedPtrField< ::xla::TileProto >*
|
|
mutable_tiles();
|
|
private:
|
|
const ::xla::TileProto& _internal_tiles(int index) const;
|
|
::xla::TileProto* _internal_add_tiles();
|
|
public:
|
|
const ::xla::TileProto& tiles(int index) const;
|
|
::xla::TileProto* add_tiles();
|
|
const ::PROTOBUF_NAMESPACE_ID::RepeatedPtrField< ::xla::TileProto >&
|
|
tiles() const;
|
|
|
|
// repeated .xla.DimLevelType dim_level_types = 9;
|
|
int dim_level_types_size() const;
|
|
private:
|
|
int _internal_dim_level_types_size() const;
|
|
public:
|
|
void clear_dim_level_types();
|
|
private:
|
|
::xla::DimLevelType _internal_dim_level_types(int index) const;
|
|
void _internal_add_dim_level_types(::xla::DimLevelType value);
|
|
::PROTOBUF_NAMESPACE_ID::RepeatedField<int>* _internal_mutable_dim_level_types();
|
|
public:
|
|
::xla::DimLevelType dim_level_types(int index) const;
|
|
void set_dim_level_types(int index, ::xla::DimLevelType value);
|
|
void add_dim_level_types(::xla::DimLevelType value);
|
|
const ::PROTOBUF_NAMESPACE_ID::RepeatedField<int>& dim_level_types() const;
|
|
::PROTOBUF_NAMESPACE_ID::RepeatedField<int>* mutable_dim_level_types();
|
|
|
|
// repeated bool dim_unique = 13;
|
|
int dim_unique_size() const;
|
|
private:
|
|
int _internal_dim_unique_size() const;
|
|
public:
|
|
void clear_dim_unique();
|
|
private:
|
|
bool _internal_dim_unique(int index) const;
|
|
const ::PROTOBUF_NAMESPACE_ID::RepeatedField< bool >&
|
|
_internal_dim_unique() const;
|
|
void _internal_add_dim_unique(bool value);
|
|
::PROTOBUF_NAMESPACE_ID::RepeatedField< bool >*
|
|
_internal_mutable_dim_unique();
|
|
public:
|
|
bool dim_unique(int index) const;
|
|
void set_dim_unique(int index, bool value);
|
|
void add_dim_unique(bool value);
|
|
const ::PROTOBUF_NAMESPACE_ID::RepeatedField< bool >&
|
|
dim_unique() const;
|
|
::PROTOBUF_NAMESPACE_ID::RepeatedField< bool >*
|
|
mutable_dim_unique();
|
|
|
|
// repeated bool dim_ordered = 14;
|
|
int dim_ordered_size() const;
|
|
private:
|
|
int _internal_dim_ordered_size() const;
|
|
public:
|
|
void clear_dim_ordered();
|
|
private:
|
|
bool _internal_dim_ordered(int index) const;
|
|
const ::PROTOBUF_NAMESPACE_ID::RepeatedField< bool >&
|
|
_internal_dim_ordered() const;
|
|
void _internal_add_dim_ordered(bool value);
|
|
::PROTOBUF_NAMESPACE_ID::RepeatedField< bool >*
|
|
_internal_mutable_dim_ordered();
|
|
public:
|
|
bool dim_ordered(int index) const;
|
|
void set_dim_ordered(int index, bool value);
|
|
void add_dim_ordered(bool value);
|
|
const ::PROTOBUF_NAMESPACE_ID::RepeatedField< bool >&
|
|
dim_ordered() const;
|
|
::PROTOBUF_NAMESPACE_ID::RepeatedField< bool >*
|
|
mutable_dim_ordered();
|
|
|
|
// .xla.ShapeProto physical_shape = 10;
|
|
bool has_physical_shape() const;
|
|
private:
|
|
bool _internal_has_physical_shape() const;
|
|
public:
|
|
void clear_physical_shape();
|
|
const ::xla::ShapeProto& physical_shape() const;
|
|
PROTOBUF_NODISCARD ::xla::ShapeProto* release_physical_shape();
|
|
::xla::ShapeProto* mutable_physical_shape();
|
|
void set_allocated_physical_shape(::xla::ShapeProto* physical_shape);
|
|
private:
|
|
const ::xla::ShapeProto& _internal_physical_shape() const;
|
|
::xla::ShapeProto* _internal_mutable_physical_shape();
|
|
public:
|
|
void unsafe_arena_set_allocated_physical_shape(
|
|
::xla::ShapeProto* physical_shape);
|
|
::xla::ShapeProto* unsafe_arena_release_physical_shape();
|
|
|
|
// int64 element_size_in_bits = 7;
|
|
void clear_element_size_in_bits();
|
|
int64_t element_size_in_bits() const;
|
|
void set_element_size_in_bits(int64_t value);
|
|
private:
|
|
int64_t _internal_element_size_in_bits() const;
|
|
void _internal_set_element_size_in_bits(int64_t value);
|
|
public:
|
|
|
|
// int64 memory_space = 8;
|
|
void clear_memory_space();
|
|
int64_t memory_space() const;
|
|
void set_memory_space(int64_t value);
|
|
private:
|
|
int64_t _internal_memory_space() const;
|
|
void _internal_set_memory_space(int64_t value);
|
|
public:
|
|
|
|
// .xla.PrimitiveType index_primitive_type = 11;
|
|
void clear_index_primitive_type();
|
|
::xla::PrimitiveType index_primitive_type() const;
|
|
void set_index_primitive_type(::xla::PrimitiveType value);
|
|
private:
|
|
::xla::PrimitiveType _internal_index_primitive_type() const;
|
|
void _internal_set_index_primitive_type(::xla::PrimitiveType value);
|
|
public:
|
|
|
|
// .xla.PrimitiveType pointer_primitive_type = 12;
|
|
void clear_pointer_primitive_type();
|
|
::xla::PrimitiveType pointer_primitive_type() const;
|
|
void set_pointer_primitive_type(::xla::PrimitiveType value);
|
|
private:
|
|
::xla::PrimitiveType _internal_pointer_primitive_type() const;
|
|
void _internal_set_pointer_primitive_type(::xla::PrimitiveType value);
|
|
public:
|
|
|
|
// int64 dynamic_shape_metadata_prefix_bytes = 15;
|
|
void clear_dynamic_shape_metadata_prefix_bytes();
|
|
int64_t dynamic_shape_metadata_prefix_bytes() const;
|
|
void set_dynamic_shape_metadata_prefix_bytes(int64_t value);
|
|
private:
|
|
int64_t _internal_dynamic_shape_metadata_prefix_bytes() const;
|
|
void _internal_set_dynamic_shape_metadata_prefix_bytes(int64_t value);
|
|
public:
|
|
|
|
// int64 tail_padding_alignment_in_elements = 16;
|
|
void clear_tail_padding_alignment_in_elements();
|
|
int64_t tail_padding_alignment_in_elements() const;
|
|
void set_tail_padding_alignment_in_elements(int64_t value);
|
|
private:
|
|
int64_t _internal_tail_padding_alignment_in_elements() const;
|
|
void _internal_set_tail_padding_alignment_in_elements(int64_t value);
|
|
public:
|
|
|
|
// @@protoc_insertion_point(class_scope:xla.LayoutProto)
|
|
private:
|
|
class _Internal;
|
|
|
|
template <typename T> friend class ::PROTOBUF_NAMESPACE_ID::Arena::InternalHelper;
|
|
typedef void InternalArenaConstructable_;
|
|
typedef void DestructorSkippable_;
|
|
struct Impl_ {
|
|
::PROTOBUF_NAMESPACE_ID::RepeatedField< int64_t > minor_to_major_;
|
|
mutable std::atomic<int> _minor_to_major_cached_byte_size_;
|
|
::PROTOBUF_NAMESPACE_ID::RepeatedPtrField< ::xla::TileProto > tiles_;
|
|
::PROTOBUF_NAMESPACE_ID::RepeatedField<int> dim_level_types_;
|
|
mutable std::atomic<int> _dim_level_types_cached_byte_size_;
|
|
::PROTOBUF_NAMESPACE_ID::RepeatedField< bool > dim_unique_;
|
|
::PROTOBUF_NAMESPACE_ID::RepeatedField< bool > dim_ordered_;
|
|
::xla::ShapeProto* physical_shape_;
|
|
int64_t element_size_in_bits_;
|
|
int64_t memory_space_;
|
|
int index_primitive_type_;
|
|
int pointer_primitive_type_;
|
|
int64_t dynamic_shape_metadata_prefix_bytes_;
|
|
int64_t tail_padding_alignment_in_elements_;
|
|
mutable ::PROTOBUF_NAMESPACE_ID::internal::CachedSize _cached_size_;
|
|
};
|
|
union { Impl_ _impl_; };
|
|
friend struct ::TableStruct_xla_2fxla_5fdata_2eproto;
|
|
};
|
|
// -------------------------------------------------------------------
|
|
|
|
class ShapeProto final :
|
|
public ::PROTOBUF_NAMESPACE_ID::Message /* @@protoc_insertion_point(class_definition:xla.ShapeProto) */ {
|
|
public:
|
|
inline ShapeProto() : ShapeProto(nullptr) {}
|
|
~ShapeProto() override;
|
|
explicit PROTOBUF_CONSTEXPR ShapeProto(::PROTOBUF_NAMESPACE_ID::internal::ConstantInitialized);
|
|
|
|
ShapeProto(const ShapeProto& from);
|
|
ShapeProto(ShapeProto&& from) noexcept
|
|
: ShapeProto() {
|
|
*this = ::std::move(from);
|
|
}
|
|
|
|
inline ShapeProto& operator=(const ShapeProto& from) {
|
|
CopyFrom(from);
|
|
return *this;
|
|
}
|
|
inline ShapeProto& operator=(ShapeProto&& from) noexcept {
|
|
if (this == &from) return *this;
|
|
if (GetOwningArena() == from.GetOwningArena()
|
|
#ifdef PROTOBUF_FORCE_COPY_IN_MOVE
|
|
&& GetOwningArena() != nullptr
|
|
#endif // !PROTOBUF_FORCE_COPY_IN_MOVE
|
|
) {
|
|
InternalSwap(&from);
|
|
} else {
|
|
CopyFrom(from);
|
|
}
|
|
return *this;
|
|
}
|
|
|
|
static const ::PROTOBUF_NAMESPACE_ID::Descriptor* descriptor() {
|
|
return GetDescriptor();
|
|
}
|
|
static const ::PROTOBUF_NAMESPACE_ID::Descriptor* GetDescriptor() {
|
|
return default_instance().GetMetadata().descriptor;
|
|
}
|
|
static const ::PROTOBUF_NAMESPACE_ID::Reflection* GetReflection() {
|
|
return default_instance().GetMetadata().reflection;
|
|
}
|
|
static const ShapeProto& default_instance() {
|
|
return *internal_default_instance();
|
|
}
|
|
static inline const ShapeProto* internal_default_instance() {
|
|
return reinterpret_cast<const ShapeProto*>(
|
|
&_ShapeProto_default_instance_);
|
|
}
|
|
static constexpr int kIndexInFileMessages =
|
|
4;
|
|
|
|
friend void swap(ShapeProto& a, ShapeProto& b) {
|
|
a.Swap(&b);
|
|
}
|
|
inline void Swap(ShapeProto* other) {
|
|
if (other == this) return;
|
|
#ifdef PROTOBUF_FORCE_COPY_IN_SWAP
|
|
if (GetOwningArena() != nullptr &&
|
|
GetOwningArena() == other->GetOwningArena()) {
|
|
#else // PROTOBUF_FORCE_COPY_IN_SWAP
|
|
if (GetOwningArena() == other->GetOwningArena()) {
|
|
#endif // !PROTOBUF_FORCE_COPY_IN_SWAP
|
|
InternalSwap(other);
|
|
} else {
|
|
::PROTOBUF_NAMESPACE_ID::internal::GenericSwap(this, other);
|
|
}
|
|
}
|
|
void UnsafeArenaSwap(ShapeProto* other) {
|
|
if (other == this) return;
|
|
GOOGLE_DCHECK(GetOwningArena() == other->GetOwningArena());
|
|
InternalSwap(other);
|
|
}
|
|
|
|
// implements Message ----------------------------------------------
|
|
|
|
ShapeProto* New(::PROTOBUF_NAMESPACE_ID::Arena* arena = nullptr) const final {
|
|
return CreateMaybeMessage<ShapeProto>(arena);
|
|
}
|
|
using ::PROTOBUF_NAMESPACE_ID::Message::CopyFrom;
|
|
void CopyFrom(const ShapeProto& from);
|
|
using ::PROTOBUF_NAMESPACE_ID::Message::MergeFrom;
|
|
void MergeFrom( const ShapeProto& from) {
|
|
ShapeProto::MergeImpl(*this, from);
|
|
}
|
|
private:
|
|
static void MergeImpl(::PROTOBUF_NAMESPACE_ID::Message& to_msg, const ::PROTOBUF_NAMESPACE_ID::Message& from_msg);
|
|
public:
|
|
PROTOBUF_ATTRIBUTE_REINITIALIZES void Clear() final;
|
|
bool IsInitialized() const final;
|
|
|
|
size_t ByteSizeLong() const final;
|
|
const char* _InternalParse(const char* ptr, ::PROTOBUF_NAMESPACE_ID::internal::ParseContext* ctx) final;
|
|
uint8_t* _InternalSerialize(
|
|
uint8_t* target, ::PROTOBUF_NAMESPACE_ID::io::EpsCopyOutputStream* stream) const final;
|
|
int GetCachedSize() const final { return _impl_._cached_size_.Get(); }
|
|
|
|
private:
|
|
void SharedCtor(::PROTOBUF_NAMESPACE_ID::Arena* arena, bool is_message_owned);
|
|
void SharedDtor();
|
|
void SetCachedSize(int size) const final;
|
|
void InternalSwap(ShapeProto* other);
|
|
|
|
private:
|
|
friend class ::PROTOBUF_NAMESPACE_ID::internal::AnyMetadata;
|
|
static ::PROTOBUF_NAMESPACE_ID::StringPiece FullMessageName() {
|
|
return "xla.ShapeProto";
|
|
}
|
|
protected:
|
|
explicit ShapeProto(::PROTOBUF_NAMESPACE_ID::Arena* arena,
|
|
bool is_message_owned = false);
|
|
public:
|
|
|
|
static const ClassData _class_data_;
|
|
const ::PROTOBUF_NAMESPACE_ID::Message::ClassData*GetClassData() const final;
|
|
|
|
::PROTOBUF_NAMESPACE_ID::Metadata GetMetadata() const final;
|
|
|
|
// nested types ----------------------------------------------------
|
|
|
|
// accessors -------------------------------------------------------
|
|
|
|
enum : int {
|
|
kDimensionsFieldNumber = 3,
|
|
kTupleShapesFieldNumber = 4,
|
|
kIsDynamicDimensionFieldNumber = 6,
|
|
kLayoutFieldNumber = 5,
|
|
kElementTypeFieldNumber = 2,
|
|
};
|
|
// repeated int64 dimensions = 3;
|
|
int dimensions_size() const;
|
|
private:
|
|
int _internal_dimensions_size() const;
|
|
public:
|
|
void clear_dimensions();
|
|
private:
|
|
int64_t _internal_dimensions(int index) const;
|
|
const ::PROTOBUF_NAMESPACE_ID::RepeatedField< int64_t >&
|
|
_internal_dimensions() const;
|
|
void _internal_add_dimensions(int64_t value);
|
|
::PROTOBUF_NAMESPACE_ID::RepeatedField< int64_t >*
|
|
_internal_mutable_dimensions();
|
|
public:
|
|
int64_t dimensions(int index) const;
|
|
void set_dimensions(int index, int64_t value);
|
|
void add_dimensions(int64_t value);
|
|
const ::PROTOBUF_NAMESPACE_ID::RepeatedField< int64_t >&
|
|
dimensions() const;
|
|
::PROTOBUF_NAMESPACE_ID::RepeatedField< int64_t >*
|
|
mutable_dimensions();
|
|
|
|
// repeated .xla.ShapeProto tuple_shapes = 4;
|
|
int tuple_shapes_size() const;
|
|
private:
|
|
int _internal_tuple_shapes_size() const;
|
|
public:
|
|
void clear_tuple_shapes();
|
|
::xla::ShapeProto* mutable_tuple_shapes(int index);
|
|
::PROTOBUF_NAMESPACE_ID::RepeatedPtrField< ::xla::ShapeProto >*
|
|
mutable_tuple_shapes();
|
|
private:
|
|
const ::xla::ShapeProto& _internal_tuple_shapes(int index) const;
|
|
::xla::ShapeProto* _internal_add_tuple_shapes();
|
|
public:
|
|
const ::xla::ShapeProto& tuple_shapes(int index) const;
|
|
::xla::ShapeProto* add_tuple_shapes();
|
|
const ::PROTOBUF_NAMESPACE_ID::RepeatedPtrField< ::xla::ShapeProto >&
|
|
tuple_shapes() const;
|
|
|
|
// repeated bool is_dynamic_dimension = 6;
|
|
int is_dynamic_dimension_size() const;
|
|
private:
|
|
int _internal_is_dynamic_dimension_size() const;
|
|
public:
|
|
void clear_is_dynamic_dimension();
|
|
private:
|
|
bool _internal_is_dynamic_dimension(int index) const;
|
|
const ::PROTOBUF_NAMESPACE_ID::RepeatedField< bool >&
|
|
_internal_is_dynamic_dimension() const;
|
|
void _internal_add_is_dynamic_dimension(bool value);
|
|
::PROTOBUF_NAMESPACE_ID::RepeatedField< bool >*
|
|
_internal_mutable_is_dynamic_dimension();
|
|
public:
|
|
bool is_dynamic_dimension(int index) const;
|
|
void set_is_dynamic_dimension(int index, bool value);
|
|
void add_is_dynamic_dimension(bool value);
|
|
const ::PROTOBUF_NAMESPACE_ID::RepeatedField< bool >&
|
|
is_dynamic_dimension() const;
|
|
::PROTOBUF_NAMESPACE_ID::RepeatedField< bool >*
|
|
mutable_is_dynamic_dimension();
|
|
|
|
// .xla.LayoutProto layout = 5;
|
|
bool has_layout() const;
|
|
private:
|
|
bool _internal_has_layout() const;
|
|
public:
|
|
void clear_layout();
|
|
const ::xla::LayoutProto& layout() const;
|
|
PROTOBUF_NODISCARD ::xla::LayoutProto* release_layout();
|
|
::xla::LayoutProto* mutable_layout();
|
|
void set_allocated_layout(::xla::LayoutProto* layout);
|
|
private:
|
|
const ::xla::LayoutProto& _internal_layout() const;
|
|
::xla::LayoutProto* _internal_mutable_layout();
|
|
public:
|
|
void unsafe_arena_set_allocated_layout(
|
|
::xla::LayoutProto* layout);
|
|
::xla::LayoutProto* unsafe_arena_release_layout();
|
|
|
|
// .xla.PrimitiveType element_type = 2;
|
|
void clear_element_type();
|
|
::xla::PrimitiveType element_type() const;
|
|
void set_element_type(::xla::PrimitiveType value);
|
|
private:
|
|
::xla::PrimitiveType _internal_element_type() const;
|
|
void _internal_set_element_type(::xla::PrimitiveType value);
|
|
public:
|
|
|
|
// @@protoc_insertion_point(class_scope:xla.ShapeProto)
|
|
private:
|
|
class _Internal;
|
|
|
|
template <typename T> friend class ::PROTOBUF_NAMESPACE_ID::Arena::InternalHelper;
|
|
typedef void InternalArenaConstructable_;
|
|
typedef void DestructorSkippable_;
|
|
struct Impl_ {
|
|
::PROTOBUF_NAMESPACE_ID::RepeatedField< int64_t > dimensions_;
|
|
mutable std::atomic<int> _dimensions_cached_byte_size_;
|
|
::PROTOBUF_NAMESPACE_ID::RepeatedPtrField< ::xla::ShapeProto > tuple_shapes_;
|
|
::PROTOBUF_NAMESPACE_ID::RepeatedField< bool > is_dynamic_dimension_;
|
|
::xla::LayoutProto* layout_;
|
|
int element_type_;
|
|
mutable ::PROTOBUF_NAMESPACE_ID::internal::CachedSize _cached_size_;
|
|
};
|
|
union { Impl_ _impl_; };
|
|
friend struct ::TableStruct_xla_2fxla_5fdata_2eproto;
|
|
};
|
|
// -------------------------------------------------------------------
|
|
|
|
class ProgramShapeProto final :
|
|
public ::PROTOBUF_NAMESPACE_ID::Message /* @@protoc_insertion_point(class_definition:xla.ProgramShapeProto) */ {
|
|
public:
|
|
inline ProgramShapeProto() : ProgramShapeProto(nullptr) {}
|
|
~ProgramShapeProto() override;
|
|
explicit PROTOBUF_CONSTEXPR ProgramShapeProto(::PROTOBUF_NAMESPACE_ID::internal::ConstantInitialized);
|
|
|
|
ProgramShapeProto(const ProgramShapeProto& from);
|
|
ProgramShapeProto(ProgramShapeProto&& from) noexcept
|
|
: ProgramShapeProto() {
|
|
*this = ::std::move(from);
|
|
}
|
|
|
|
inline ProgramShapeProto& operator=(const ProgramShapeProto& from) {
|
|
CopyFrom(from);
|
|
return *this;
|
|
}
|
|
inline ProgramShapeProto& operator=(ProgramShapeProto&& from) noexcept {
|
|
if (this == &from) return *this;
|
|
if (GetOwningArena() == from.GetOwningArena()
|
|
#ifdef PROTOBUF_FORCE_COPY_IN_MOVE
|
|
&& GetOwningArena() != nullptr
|
|
#endif // !PROTOBUF_FORCE_COPY_IN_MOVE
|
|
) {
|
|
InternalSwap(&from);
|
|
} else {
|
|
CopyFrom(from);
|
|
}
|
|
return *this;
|
|
}
|
|
|
|
static const ::PROTOBUF_NAMESPACE_ID::Descriptor* descriptor() {
|
|
return GetDescriptor();
|
|
}
|
|
static const ::PROTOBUF_NAMESPACE_ID::Descriptor* GetDescriptor() {
|
|
return default_instance().GetMetadata().descriptor;
|
|
}
|
|
static const ::PROTOBUF_NAMESPACE_ID::Reflection* GetReflection() {
|
|
return default_instance().GetMetadata().reflection;
|
|
}
|
|
static const ProgramShapeProto& default_instance() {
|
|
return *internal_default_instance();
|
|
}
|
|
static inline const ProgramShapeProto* internal_default_instance() {
|
|
return reinterpret_cast<const ProgramShapeProto*>(
|
|
&_ProgramShapeProto_default_instance_);
|
|
}
|
|
static constexpr int kIndexInFileMessages =
|
|
5;
|
|
|
|
friend void swap(ProgramShapeProto& a, ProgramShapeProto& b) {
|
|
a.Swap(&b);
|
|
}
|
|
inline void Swap(ProgramShapeProto* other) {
|
|
if (other == this) return;
|
|
#ifdef PROTOBUF_FORCE_COPY_IN_SWAP
|
|
if (GetOwningArena() != nullptr &&
|
|
GetOwningArena() == other->GetOwningArena()) {
|
|
#else // PROTOBUF_FORCE_COPY_IN_SWAP
|
|
if (GetOwningArena() == other->GetOwningArena()) {
|
|
#endif // !PROTOBUF_FORCE_COPY_IN_SWAP
|
|
InternalSwap(other);
|
|
} else {
|
|
::PROTOBUF_NAMESPACE_ID::internal::GenericSwap(this, other);
|
|
}
|
|
}
|
|
void UnsafeArenaSwap(ProgramShapeProto* other) {
|
|
if (other == this) return;
|
|
GOOGLE_DCHECK(GetOwningArena() == other->GetOwningArena());
|
|
InternalSwap(other);
|
|
}
|
|
|
|
// implements Message ----------------------------------------------
|
|
|
|
ProgramShapeProto* New(::PROTOBUF_NAMESPACE_ID::Arena* arena = nullptr) const final {
|
|
return CreateMaybeMessage<ProgramShapeProto>(arena);
|
|
}
|
|
using ::PROTOBUF_NAMESPACE_ID::Message::CopyFrom;
|
|
void CopyFrom(const ProgramShapeProto& from);
|
|
using ::PROTOBUF_NAMESPACE_ID::Message::MergeFrom;
|
|
void MergeFrom( const ProgramShapeProto& from) {
|
|
ProgramShapeProto::MergeImpl(*this, from);
|
|
}
|
|
private:
|
|
static void MergeImpl(::PROTOBUF_NAMESPACE_ID::Message& to_msg, const ::PROTOBUF_NAMESPACE_ID::Message& from_msg);
|
|
public:
|
|
PROTOBUF_ATTRIBUTE_REINITIALIZES void Clear() final;
|
|
bool IsInitialized() const final;
|
|
|
|
size_t ByteSizeLong() const final;
|
|
const char* _InternalParse(const char* ptr, ::PROTOBUF_NAMESPACE_ID::internal::ParseContext* ctx) final;
|
|
uint8_t* _InternalSerialize(
|
|
uint8_t* target, ::PROTOBUF_NAMESPACE_ID::io::EpsCopyOutputStream* stream) const final;
|
|
int GetCachedSize() const final { return _impl_._cached_size_.Get(); }
|
|
|
|
private:
|
|
void SharedCtor(::PROTOBUF_NAMESPACE_ID::Arena* arena, bool is_message_owned);
|
|
void SharedDtor();
|
|
void SetCachedSize(int size) const final;
|
|
void InternalSwap(ProgramShapeProto* other);
|
|
|
|
private:
|
|
friend class ::PROTOBUF_NAMESPACE_ID::internal::AnyMetadata;
|
|
static ::PROTOBUF_NAMESPACE_ID::StringPiece FullMessageName() {
|
|
return "xla.ProgramShapeProto";
|
|
}
|
|
protected:
|
|
explicit ProgramShapeProto(::PROTOBUF_NAMESPACE_ID::Arena* arena,
|
|
bool is_message_owned = false);
|
|
public:
|
|
|
|
static const ClassData _class_data_;
|
|
const ::PROTOBUF_NAMESPACE_ID::Message::ClassData*GetClassData() const final;
|
|
|
|
::PROTOBUF_NAMESPACE_ID::Metadata GetMetadata() const final;
|
|
|
|
// nested types ----------------------------------------------------
|
|
|
|
// accessors -------------------------------------------------------
|
|
|
|
enum : int {
|
|
kParametersFieldNumber = 1,
|
|
kParameterNamesFieldNumber = 3,
|
|
kResultFieldNumber = 2,
|
|
};
|
|
// repeated .xla.ShapeProto parameters = 1;
|
|
int parameters_size() const;
|
|
private:
|
|
int _internal_parameters_size() const;
|
|
public:
|
|
void clear_parameters();
|
|
::xla::ShapeProto* mutable_parameters(int index);
|
|
::PROTOBUF_NAMESPACE_ID::RepeatedPtrField< ::xla::ShapeProto >*
|
|
mutable_parameters();
|
|
private:
|
|
const ::xla::ShapeProto& _internal_parameters(int index) const;
|
|
::xla::ShapeProto* _internal_add_parameters();
|
|
public:
|
|
const ::xla::ShapeProto& parameters(int index) const;
|
|
::xla::ShapeProto* add_parameters();
|
|
const ::PROTOBUF_NAMESPACE_ID::RepeatedPtrField< ::xla::ShapeProto >&
|
|
parameters() const;
|
|
|
|
// repeated string parameter_names = 3;
|
|
int parameter_names_size() const;
|
|
private:
|
|
int _internal_parameter_names_size() const;
|
|
public:
|
|
void clear_parameter_names();
|
|
const std::string& parameter_names(int index) const;
|
|
std::string* mutable_parameter_names(int index);
|
|
void set_parameter_names(int index, const std::string& value);
|
|
void set_parameter_names(int index, std::string&& value);
|
|
void set_parameter_names(int index, const char* value);
|
|
void set_parameter_names(int index, const char* value, size_t size);
|
|
std::string* add_parameter_names();
|
|
void add_parameter_names(const std::string& value);
|
|
void add_parameter_names(std::string&& value);
|
|
void add_parameter_names(const char* value);
|
|
void add_parameter_names(const char* value, size_t size);
|
|
const ::PROTOBUF_NAMESPACE_ID::RepeatedPtrField<std::string>& parameter_names() const;
|
|
::PROTOBUF_NAMESPACE_ID::RepeatedPtrField<std::string>* mutable_parameter_names();
|
|
private:
|
|
const std::string& _internal_parameter_names(int index) const;
|
|
std::string* _internal_add_parameter_names();
|
|
public:
|
|
|
|
// .xla.ShapeProto result = 2;
|
|
bool has_result() const;
|
|
private:
|
|
bool _internal_has_result() const;
|
|
public:
|
|
void clear_result();
|
|
const ::xla::ShapeProto& result() const;
|
|
PROTOBUF_NODISCARD ::xla::ShapeProto* release_result();
|
|
::xla::ShapeProto* mutable_result();
|
|
void set_allocated_result(::xla::ShapeProto* result);
|
|
private:
|
|
const ::xla::ShapeProto& _internal_result() const;
|
|
::xla::ShapeProto* _internal_mutable_result();
|
|
public:
|
|
void unsafe_arena_set_allocated_result(
|
|
::xla::ShapeProto* result);
|
|
::xla::ShapeProto* unsafe_arena_release_result();
|
|
|
|
// @@protoc_insertion_point(class_scope:xla.ProgramShapeProto)
|
|
private:
|
|
class _Internal;
|
|
|
|
template <typename T> friend class ::PROTOBUF_NAMESPACE_ID::Arena::InternalHelper;
|
|
typedef void InternalArenaConstructable_;
|
|
typedef void DestructorSkippable_;
|
|
struct Impl_ {
|
|
::PROTOBUF_NAMESPACE_ID::RepeatedPtrField< ::xla::ShapeProto > parameters_;
|
|
::PROTOBUF_NAMESPACE_ID::RepeatedPtrField<std::string> parameter_names_;
|
|
::xla::ShapeProto* result_;
|
|
mutable ::PROTOBUF_NAMESPACE_ID::internal::CachedSize _cached_size_;
|
|
};
|
|
union { Impl_ _impl_; };
|
|
friend struct ::TableStruct_xla_2fxla_5fdata_2eproto;
|
|
};
|
|
// -------------------------------------------------------------------
|
|
|
|
class ComputationStats final :
|
|
public ::PROTOBUF_NAMESPACE_ID::Message /* @@protoc_insertion_point(class_definition:xla.ComputationStats) */ {
|
|
public:
|
|
inline ComputationStats() : ComputationStats(nullptr) {}
|
|
~ComputationStats() override;
|
|
explicit PROTOBUF_CONSTEXPR ComputationStats(::PROTOBUF_NAMESPACE_ID::internal::ConstantInitialized);
|
|
|
|
ComputationStats(const ComputationStats& from);
|
|
ComputationStats(ComputationStats&& from) noexcept
|
|
: ComputationStats() {
|
|
*this = ::std::move(from);
|
|
}
|
|
|
|
inline ComputationStats& operator=(const ComputationStats& from) {
|
|
CopyFrom(from);
|
|
return *this;
|
|
}
|
|
inline ComputationStats& operator=(ComputationStats&& from) noexcept {
|
|
if (this == &from) return *this;
|
|
if (GetOwningArena() == from.GetOwningArena()
|
|
#ifdef PROTOBUF_FORCE_COPY_IN_MOVE
|
|
&& GetOwningArena() != nullptr
|
|
#endif // !PROTOBUF_FORCE_COPY_IN_MOVE
|
|
) {
|
|
InternalSwap(&from);
|
|
} else {
|
|
CopyFrom(from);
|
|
}
|
|
return *this;
|
|
}
|
|
|
|
static const ::PROTOBUF_NAMESPACE_ID::Descriptor* descriptor() {
|
|
return GetDescriptor();
|
|
}
|
|
static const ::PROTOBUF_NAMESPACE_ID::Descriptor* GetDescriptor() {
|
|
return default_instance().GetMetadata().descriptor;
|
|
}
|
|
static const ::PROTOBUF_NAMESPACE_ID::Reflection* GetReflection() {
|
|
return default_instance().GetMetadata().reflection;
|
|
}
|
|
static const ComputationStats& default_instance() {
|
|
return *internal_default_instance();
|
|
}
|
|
static inline const ComputationStats* internal_default_instance() {
|
|
return reinterpret_cast<const ComputationStats*>(
|
|
&_ComputationStats_default_instance_);
|
|
}
|
|
static constexpr int kIndexInFileMessages =
|
|
6;
|
|
|
|
friend void swap(ComputationStats& a, ComputationStats& b) {
|
|
a.Swap(&b);
|
|
}
|
|
inline void Swap(ComputationStats* other) {
|
|
if (other == this) return;
|
|
#ifdef PROTOBUF_FORCE_COPY_IN_SWAP
|
|
if (GetOwningArena() != nullptr &&
|
|
GetOwningArena() == other->GetOwningArena()) {
|
|
#else // PROTOBUF_FORCE_COPY_IN_SWAP
|
|
if (GetOwningArena() == other->GetOwningArena()) {
|
|
#endif // !PROTOBUF_FORCE_COPY_IN_SWAP
|
|
InternalSwap(other);
|
|
} else {
|
|
::PROTOBUF_NAMESPACE_ID::internal::GenericSwap(this, other);
|
|
}
|
|
}
|
|
void UnsafeArenaSwap(ComputationStats* other) {
|
|
if (other == this) return;
|
|
GOOGLE_DCHECK(GetOwningArena() == other->GetOwningArena());
|
|
InternalSwap(other);
|
|
}
|
|
|
|
// implements Message ----------------------------------------------
|
|
|
|
ComputationStats* New(::PROTOBUF_NAMESPACE_ID::Arena* arena = nullptr) const final {
|
|
return CreateMaybeMessage<ComputationStats>(arena);
|
|
}
|
|
using ::PROTOBUF_NAMESPACE_ID::Message::CopyFrom;
|
|
void CopyFrom(const ComputationStats& from);
|
|
using ::PROTOBUF_NAMESPACE_ID::Message::MergeFrom;
|
|
void MergeFrom( const ComputationStats& from) {
|
|
ComputationStats::MergeImpl(*this, from);
|
|
}
|
|
private:
|
|
static void MergeImpl(::PROTOBUF_NAMESPACE_ID::Message& to_msg, const ::PROTOBUF_NAMESPACE_ID::Message& from_msg);
|
|
public:
|
|
PROTOBUF_ATTRIBUTE_REINITIALIZES void Clear() final;
|
|
bool IsInitialized() const final;
|
|
|
|
size_t ByteSizeLong() const final;
|
|
const char* _InternalParse(const char* ptr, ::PROTOBUF_NAMESPACE_ID::internal::ParseContext* ctx) final;
|
|
uint8_t* _InternalSerialize(
|
|
uint8_t* target, ::PROTOBUF_NAMESPACE_ID::io::EpsCopyOutputStream* stream) const final;
|
|
int GetCachedSize() const final { return _impl_._cached_size_.Get(); }
|
|
|
|
private:
|
|
void SharedCtor(::PROTOBUF_NAMESPACE_ID::Arena* arena, bool is_message_owned);
|
|
void SharedDtor();
|
|
void SetCachedSize(int size) const final;
|
|
void InternalSwap(ComputationStats* other);
|
|
|
|
private:
|
|
friend class ::PROTOBUF_NAMESPACE_ID::internal::AnyMetadata;
|
|
static ::PROTOBUF_NAMESPACE_ID::StringPiece FullMessageName() {
|
|
return "xla.ComputationStats";
|
|
}
|
|
protected:
|
|
explicit ComputationStats(::PROTOBUF_NAMESPACE_ID::Arena* arena,
|
|
bool is_message_owned = false);
|
|
public:
|
|
|
|
static const ClassData _class_data_;
|
|
const ::PROTOBUF_NAMESPACE_ID::Message::ClassData*GetClassData() const final;
|
|
|
|
::PROTOBUF_NAMESPACE_ID::Metadata GetMetadata() const final;
|
|
|
|
// nested types ----------------------------------------------------
|
|
|
|
// accessors -------------------------------------------------------
|
|
|
|
enum : int {
|
|
kFlopCountFieldNumber = 1,
|
|
kTranscendentalCountFieldNumber = 2,
|
|
};
|
|
// double flop_count = 1;
|
|
void clear_flop_count();
|
|
double flop_count() const;
|
|
void set_flop_count(double value);
|
|
private:
|
|
double _internal_flop_count() const;
|
|
void _internal_set_flop_count(double value);
|
|
public:
|
|
|
|
// double transcendental_count = 2;
|
|
void clear_transcendental_count();
|
|
double transcendental_count() const;
|
|
void set_transcendental_count(double value);
|
|
private:
|
|
double _internal_transcendental_count() const;
|
|
void _internal_set_transcendental_count(double value);
|
|
public:
|
|
|
|
// @@protoc_insertion_point(class_scope:xla.ComputationStats)
|
|
private:
|
|
class _Internal;
|
|
|
|
template <typename T> friend class ::PROTOBUF_NAMESPACE_ID::Arena::InternalHelper;
|
|
typedef void InternalArenaConstructable_;
|
|
typedef void DestructorSkippable_;
|
|
struct Impl_ {
|
|
double flop_count_;
|
|
double transcendental_count_;
|
|
mutable ::PROTOBUF_NAMESPACE_ID::internal::CachedSize _cached_size_;
|
|
};
|
|
union { Impl_ _impl_; };
|
|
friend struct ::TableStruct_xla_2fxla_5fdata_2eproto;
|
|
};
|
|
// -------------------------------------------------------------------
|
|
|
|
class OpMetadata_ProfileInfo final :
|
|
public ::PROTOBUF_NAMESPACE_ID::Message /* @@protoc_insertion_point(class_definition:xla.OpMetadata.ProfileInfo) */ {
|
|
public:
|
|
inline OpMetadata_ProfileInfo() : OpMetadata_ProfileInfo(nullptr) {}
|
|
~OpMetadata_ProfileInfo() override;
|
|
explicit PROTOBUF_CONSTEXPR OpMetadata_ProfileInfo(::PROTOBUF_NAMESPACE_ID::internal::ConstantInitialized);
|
|
|
|
OpMetadata_ProfileInfo(const OpMetadata_ProfileInfo& from);
|
|
OpMetadata_ProfileInfo(OpMetadata_ProfileInfo&& from) noexcept
|
|
: OpMetadata_ProfileInfo() {
|
|
*this = ::std::move(from);
|
|
}
|
|
|
|
inline OpMetadata_ProfileInfo& operator=(const OpMetadata_ProfileInfo& from) {
|
|
CopyFrom(from);
|
|
return *this;
|
|
}
|
|
inline OpMetadata_ProfileInfo& operator=(OpMetadata_ProfileInfo&& from) noexcept {
|
|
if (this == &from) return *this;
|
|
if (GetOwningArena() == from.GetOwningArena()
|
|
#ifdef PROTOBUF_FORCE_COPY_IN_MOVE
|
|
&& GetOwningArena() != nullptr
|
|
#endif // !PROTOBUF_FORCE_COPY_IN_MOVE
|
|
) {
|
|
InternalSwap(&from);
|
|
} else {
|
|
CopyFrom(from);
|
|
}
|
|
return *this;
|
|
}
|
|
|
|
static const ::PROTOBUF_NAMESPACE_ID::Descriptor* descriptor() {
|
|
return GetDescriptor();
|
|
}
|
|
static const ::PROTOBUF_NAMESPACE_ID::Descriptor* GetDescriptor() {
|
|
return default_instance().GetMetadata().descriptor;
|
|
}
|
|
static const ::PROTOBUF_NAMESPACE_ID::Reflection* GetReflection() {
|
|
return default_instance().GetMetadata().reflection;
|
|
}
|
|
static const OpMetadata_ProfileInfo& default_instance() {
|
|
return *internal_default_instance();
|
|
}
|
|
static inline const OpMetadata_ProfileInfo* internal_default_instance() {
|
|
return reinterpret_cast<const OpMetadata_ProfileInfo*>(
|
|
&_OpMetadata_ProfileInfo_default_instance_);
|
|
}
|
|
static constexpr int kIndexInFileMessages =
|
|
7;
|
|
|
|
friend void swap(OpMetadata_ProfileInfo& a, OpMetadata_ProfileInfo& b) {
|
|
a.Swap(&b);
|
|
}
|
|
inline void Swap(OpMetadata_ProfileInfo* other) {
|
|
if (other == this) return;
|
|
#ifdef PROTOBUF_FORCE_COPY_IN_SWAP
|
|
if (GetOwningArena() != nullptr &&
|
|
GetOwningArena() == other->GetOwningArena()) {
|
|
#else // PROTOBUF_FORCE_COPY_IN_SWAP
|
|
if (GetOwningArena() == other->GetOwningArena()) {
|
|
#endif // !PROTOBUF_FORCE_COPY_IN_SWAP
|
|
InternalSwap(other);
|
|
} else {
|
|
::PROTOBUF_NAMESPACE_ID::internal::GenericSwap(this, other);
|
|
}
|
|
}
|
|
void UnsafeArenaSwap(OpMetadata_ProfileInfo* other) {
|
|
if (other == this) return;
|
|
GOOGLE_DCHECK(GetOwningArena() == other->GetOwningArena());
|
|
InternalSwap(other);
|
|
}
|
|
|
|
// implements Message ----------------------------------------------
|
|
|
|
OpMetadata_ProfileInfo* New(::PROTOBUF_NAMESPACE_ID::Arena* arena = nullptr) const final {
|
|
return CreateMaybeMessage<OpMetadata_ProfileInfo>(arena);
|
|
}
|
|
using ::PROTOBUF_NAMESPACE_ID::Message::CopyFrom;
|
|
void CopyFrom(const OpMetadata_ProfileInfo& from);
|
|
using ::PROTOBUF_NAMESPACE_ID::Message::MergeFrom;
|
|
void MergeFrom( const OpMetadata_ProfileInfo& from) {
|
|
OpMetadata_ProfileInfo::MergeImpl(*this, from);
|
|
}
|
|
private:
|
|
static void MergeImpl(::PROTOBUF_NAMESPACE_ID::Message& to_msg, const ::PROTOBUF_NAMESPACE_ID::Message& from_msg);
|
|
public:
|
|
PROTOBUF_ATTRIBUTE_REINITIALIZES void Clear() final;
|
|
bool IsInitialized() const final;
|
|
|
|
size_t ByteSizeLong() const final;
|
|
const char* _InternalParse(const char* ptr, ::PROTOBUF_NAMESPACE_ID::internal::ParseContext* ctx) final;
|
|
uint8_t* _InternalSerialize(
|
|
uint8_t* target, ::PROTOBUF_NAMESPACE_ID::io::EpsCopyOutputStream* stream) const final;
|
|
int GetCachedSize() const final { return _impl_._cached_size_.Get(); }
|
|
|
|
private:
|
|
void SharedCtor(::PROTOBUF_NAMESPACE_ID::Arena* arena, bool is_message_owned);
|
|
void SharedDtor();
|
|
void SetCachedSize(int size) const final;
|
|
void InternalSwap(OpMetadata_ProfileInfo* other);
|
|
|
|
private:
|
|
friend class ::PROTOBUF_NAMESPACE_ID::internal::AnyMetadata;
|
|
static ::PROTOBUF_NAMESPACE_ID::StringPiece FullMessageName() {
|
|
return "xla.OpMetadata.ProfileInfo";
|
|
}
|
|
protected:
|
|
explicit OpMetadata_ProfileInfo(::PROTOBUF_NAMESPACE_ID::Arena* arena,
|
|
bool is_message_owned = false);
|
|
public:
|
|
|
|
static const ClassData _class_data_;
|
|
const ::PROTOBUF_NAMESPACE_ID::Message::ClassData*GetClassData() const final;
|
|
|
|
::PROTOBUF_NAMESPACE_ID::Metadata GetMetadata() const final;
|
|
|
|
// nested types ----------------------------------------------------
|
|
|
|
// accessors -------------------------------------------------------
|
|
|
|
enum : int {
|
|
kProfileTypeFieldNumber = 1,
|
|
kRelativeSpeedupFieldNumber = 2,
|
|
kProfileSourceFieldNumber = 3,
|
|
kCompilationEventFieldNumber = 4,
|
|
};
|
|
// repeated .xla.ProfileType profile_type = 1;
|
|
int profile_type_size() const;
|
|
private:
|
|
int _internal_profile_type_size() const;
|
|
public:
|
|
void clear_profile_type();
|
|
private:
|
|
::xla::ProfileType _internal_profile_type(int index) const;
|
|
void _internal_add_profile_type(::xla::ProfileType value);
|
|
::PROTOBUF_NAMESPACE_ID::RepeatedField<int>* _internal_mutable_profile_type();
|
|
public:
|
|
::xla::ProfileType profile_type(int index) const;
|
|
void set_profile_type(int index, ::xla::ProfileType value);
|
|
void add_profile_type(::xla::ProfileType value);
|
|
const ::PROTOBUF_NAMESPACE_ID::RepeatedField<int>& profile_type() const;
|
|
::PROTOBUF_NAMESPACE_ID::RepeatedField<int>* mutable_profile_type();
|
|
|
|
// double relative_speedup = 2;
|
|
void clear_relative_speedup();
|
|
double relative_speedup() const;
|
|
void set_relative_speedup(double value);
|
|
private:
|
|
double _internal_relative_speedup() const;
|
|
void _internal_set_relative_speedup(double value);
|
|
public:
|
|
|
|
// .xla.ProfileSource profile_source = 3;
|
|
void clear_profile_source();
|
|
::xla::ProfileSource profile_source() const;
|
|
void set_profile_source(::xla::ProfileSource value);
|
|
private:
|
|
::xla::ProfileSource _internal_profile_source() const;
|
|
void _internal_set_profile_source(::xla::ProfileSource value);
|
|
public:
|
|
|
|
// .xla.CompilationEvent compilation_event = 4;
|
|
void clear_compilation_event();
|
|
::xla::CompilationEvent compilation_event() const;
|
|
void set_compilation_event(::xla::CompilationEvent value);
|
|
private:
|
|
::xla::CompilationEvent _internal_compilation_event() const;
|
|
void _internal_set_compilation_event(::xla::CompilationEvent value);
|
|
public:
|
|
|
|
// @@protoc_insertion_point(class_scope:xla.OpMetadata.ProfileInfo)
|
|
private:
|
|
class _Internal;
|
|
|
|
template <typename T> friend class ::PROTOBUF_NAMESPACE_ID::Arena::InternalHelper;
|
|
typedef void InternalArenaConstructable_;
|
|
typedef void DestructorSkippable_;
|
|
struct Impl_ {
|
|
::PROTOBUF_NAMESPACE_ID::RepeatedField<int> profile_type_;
|
|
mutable std::atomic<int> _profile_type_cached_byte_size_;
|
|
double relative_speedup_;
|
|
int profile_source_;
|
|
int compilation_event_;
|
|
mutable ::PROTOBUF_NAMESPACE_ID::internal::CachedSize _cached_size_;
|
|
};
|
|
union { Impl_ _impl_; };
|
|
friend struct ::TableStruct_xla_2fxla_5fdata_2eproto;
|
|
};
|
|
// -------------------------------------------------------------------
|
|
|
|
class OpMetadata final :
|
|
public ::PROTOBUF_NAMESPACE_ID::Message /* @@protoc_insertion_point(class_definition:xla.OpMetadata) */ {
|
|
public:
|
|
inline OpMetadata() : OpMetadata(nullptr) {}
|
|
~OpMetadata() override;
|
|
explicit PROTOBUF_CONSTEXPR OpMetadata(::PROTOBUF_NAMESPACE_ID::internal::ConstantInitialized);
|
|
|
|
OpMetadata(const OpMetadata& from);
|
|
OpMetadata(OpMetadata&& from) noexcept
|
|
: OpMetadata() {
|
|
*this = ::std::move(from);
|
|
}
|
|
|
|
inline OpMetadata& operator=(const OpMetadata& from) {
|
|
CopyFrom(from);
|
|
return *this;
|
|
}
|
|
inline OpMetadata& operator=(OpMetadata&& from) noexcept {
|
|
if (this == &from) return *this;
|
|
if (GetOwningArena() == from.GetOwningArena()
|
|
#ifdef PROTOBUF_FORCE_COPY_IN_MOVE
|
|
&& GetOwningArena() != nullptr
|
|
#endif // !PROTOBUF_FORCE_COPY_IN_MOVE
|
|
) {
|
|
InternalSwap(&from);
|
|
} else {
|
|
CopyFrom(from);
|
|
}
|
|
return *this;
|
|
}
|
|
|
|
static const ::PROTOBUF_NAMESPACE_ID::Descriptor* descriptor() {
|
|
return GetDescriptor();
|
|
}
|
|
static const ::PROTOBUF_NAMESPACE_ID::Descriptor* GetDescriptor() {
|
|
return default_instance().GetMetadata().descriptor;
|
|
}
|
|
static const ::PROTOBUF_NAMESPACE_ID::Reflection* GetReflection() {
|
|
return default_instance().GetMetadata().reflection;
|
|
}
|
|
static const OpMetadata& default_instance() {
|
|
return *internal_default_instance();
|
|
}
|
|
static inline const OpMetadata* internal_default_instance() {
|
|
return reinterpret_cast<const OpMetadata*>(
|
|
&_OpMetadata_default_instance_);
|
|
}
|
|
static constexpr int kIndexInFileMessages =
|
|
8;
|
|
|
|
friend void swap(OpMetadata& a, OpMetadata& b) {
|
|
a.Swap(&b);
|
|
}
|
|
inline void Swap(OpMetadata* other) {
|
|
if (other == this) return;
|
|
#ifdef PROTOBUF_FORCE_COPY_IN_SWAP
|
|
if (GetOwningArena() != nullptr &&
|
|
GetOwningArena() == other->GetOwningArena()) {
|
|
#else // PROTOBUF_FORCE_COPY_IN_SWAP
|
|
if (GetOwningArena() == other->GetOwningArena()) {
|
|
#endif // !PROTOBUF_FORCE_COPY_IN_SWAP
|
|
InternalSwap(other);
|
|
} else {
|
|
::PROTOBUF_NAMESPACE_ID::internal::GenericSwap(this, other);
|
|
}
|
|
}
|
|
void UnsafeArenaSwap(OpMetadata* other) {
|
|
if (other == this) return;
|
|
GOOGLE_DCHECK(GetOwningArena() == other->GetOwningArena());
|
|
InternalSwap(other);
|
|
}
|
|
|
|
// implements Message ----------------------------------------------
|
|
|
|
OpMetadata* New(::PROTOBUF_NAMESPACE_ID::Arena* arena = nullptr) const final {
|
|
return CreateMaybeMessage<OpMetadata>(arena);
|
|
}
|
|
using ::PROTOBUF_NAMESPACE_ID::Message::CopyFrom;
|
|
void CopyFrom(const OpMetadata& from);
|
|
using ::PROTOBUF_NAMESPACE_ID::Message::MergeFrom;
|
|
void MergeFrom( const OpMetadata& from) {
|
|
OpMetadata::MergeImpl(*this, from);
|
|
}
|
|
private:
|
|
static void MergeImpl(::PROTOBUF_NAMESPACE_ID::Message& to_msg, const ::PROTOBUF_NAMESPACE_ID::Message& from_msg);
|
|
public:
|
|
PROTOBUF_ATTRIBUTE_REINITIALIZES void Clear() final;
|
|
bool IsInitialized() const final;
|
|
|
|
size_t ByteSizeLong() const final;
|
|
const char* _InternalParse(const char* ptr, ::PROTOBUF_NAMESPACE_ID::internal::ParseContext* ctx) final;
|
|
uint8_t* _InternalSerialize(
|
|
uint8_t* target, ::PROTOBUF_NAMESPACE_ID::io::EpsCopyOutputStream* stream) const final;
|
|
int GetCachedSize() const final { return _impl_._cached_size_.Get(); }
|
|
|
|
private:
|
|
void SharedCtor(::PROTOBUF_NAMESPACE_ID::Arena* arena, bool is_message_owned);
|
|
void SharedDtor();
|
|
void SetCachedSize(int size) const final;
|
|
void InternalSwap(OpMetadata* other);
|
|
|
|
private:
|
|
friend class ::PROTOBUF_NAMESPACE_ID::internal::AnyMetadata;
|
|
static ::PROTOBUF_NAMESPACE_ID::StringPiece FullMessageName() {
|
|
return "xla.OpMetadata";
|
|
}
|
|
protected:
|
|
explicit OpMetadata(::PROTOBUF_NAMESPACE_ID::Arena* arena,
|
|
bool is_message_owned = false);
|
|
public:
|
|
|
|
static const ClassData _class_data_;
|
|
const ::PROTOBUF_NAMESPACE_ID::Message::ClassData*GetClassData() const final;
|
|
|
|
::PROTOBUF_NAMESPACE_ID::Metadata GetMetadata() const final;
|
|
|
|
// nested types ----------------------------------------------------
|
|
|
|
typedef OpMetadata_ProfileInfo ProfileInfo;
|
|
|
|
// accessors -------------------------------------------------------
|
|
|
|
enum : int {
|
|
kProfileTypeFieldNumber = 5,
|
|
kOpTypeFieldNumber = 1,
|
|
kOpNameFieldNumber = 2,
|
|
kSourceFileFieldNumber = 3,
|
|
kDeduplicatedNameFieldNumber = 12,
|
|
kProfileInfoFieldNumber = 10,
|
|
kCreationPassIdFieldNumber = 6,
|
|
kLogicalCreationPassIdFieldNumber = 7,
|
|
kSizeOfGeneratedCodeInBytesFieldNumber = 8,
|
|
kSourceLineFieldNumber = 4,
|
|
kPreserveLayoutFieldNumber = 13,
|
|
kSizeOfMemoryWorkingSetInBytesFieldNumber = 9,
|
|
kStackFrameIdFieldNumber = 15,
|
|
};
|
|
// repeated .xla.ProfileType profile_type = 5 [deprecated = true];
|
|
PROTOBUF_DEPRECATED int profile_type_size() const;
|
|
private:
|
|
int _internal_profile_type_size() const;
|
|
public:
|
|
PROTOBUF_DEPRECATED void clear_profile_type();
|
|
private:
|
|
::xla::ProfileType _internal_profile_type(int index) const;
|
|
void _internal_add_profile_type(::xla::ProfileType value);
|
|
::PROTOBUF_NAMESPACE_ID::RepeatedField<int>* _internal_mutable_profile_type();
|
|
public:
|
|
PROTOBUF_DEPRECATED ::xla::ProfileType profile_type(int index) const;
|
|
PROTOBUF_DEPRECATED void set_profile_type(int index, ::xla::ProfileType value);
|
|
PROTOBUF_DEPRECATED void add_profile_type(::xla::ProfileType value);
|
|
PROTOBUF_DEPRECATED const ::PROTOBUF_NAMESPACE_ID::RepeatedField<int>& profile_type() const;
|
|
PROTOBUF_DEPRECATED ::PROTOBUF_NAMESPACE_ID::RepeatedField<int>* mutable_profile_type();
|
|
|
|
// string op_type = 1;
|
|
void clear_op_type();
|
|
const std::string& op_type() const;
|
|
template <typename ArgT0 = const std::string&, typename... ArgT>
|
|
void set_op_type(ArgT0&& arg0, ArgT... args);
|
|
std::string* mutable_op_type();
|
|
PROTOBUF_NODISCARD std::string* release_op_type();
|
|
void set_allocated_op_type(std::string* op_type);
|
|
private:
|
|
const std::string& _internal_op_type() const;
|
|
inline PROTOBUF_ALWAYS_INLINE void _internal_set_op_type(const std::string& value);
|
|
std::string* _internal_mutable_op_type();
|
|
public:
|
|
|
|
// string op_name = 2;
|
|
void clear_op_name();
|
|
const std::string& op_name() const;
|
|
template <typename ArgT0 = const std::string&, typename... ArgT>
|
|
void set_op_name(ArgT0&& arg0, ArgT... args);
|
|
std::string* mutable_op_name();
|
|
PROTOBUF_NODISCARD std::string* release_op_name();
|
|
void set_allocated_op_name(std::string* op_name);
|
|
private:
|
|
const std::string& _internal_op_name() const;
|
|
inline PROTOBUF_ALWAYS_INLINE void _internal_set_op_name(const std::string& value);
|
|
std::string* _internal_mutable_op_name();
|
|
public:
|
|
|
|
// string source_file = 3;
|
|
void clear_source_file();
|
|
const std::string& source_file() const;
|
|
template <typename ArgT0 = const std::string&, typename... ArgT>
|
|
void set_source_file(ArgT0&& arg0, ArgT... args);
|
|
std::string* mutable_source_file();
|
|
PROTOBUF_NODISCARD std::string* release_source_file();
|
|
void set_allocated_source_file(std::string* source_file);
|
|
private:
|
|
const std::string& _internal_source_file() const;
|
|
inline PROTOBUF_ALWAYS_INLINE void _internal_set_source_file(const std::string& value);
|
|
std::string* _internal_mutable_source_file();
|
|
public:
|
|
|
|
// string deduplicated_name = 12;
|
|
void clear_deduplicated_name();
|
|
const std::string& deduplicated_name() const;
|
|
template <typename ArgT0 = const std::string&, typename... ArgT>
|
|
void set_deduplicated_name(ArgT0&& arg0, ArgT... args);
|
|
std::string* mutable_deduplicated_name();
|
|
PROTOBUF_NODISCARD std::string* release_deduplicated_name();
|
|
void set_allocated_deduplicated_name(std::string* deduplicated_name);
|
|
private:
|
|
const std::string& _internal_deduplicated_name() const;
|
|
inline PROTOBUF_ALWAYS_INLINE void _internal_set_deduplicated_name(const std::string& value);
|
|
std::string* _internal_mutable_deduplicated_name();
|
|
public:
|
|
|
|
// .xla.OpMetadata.ProfileInfo profile_info = 10;
|
|
bool has_profile_info() const;
|
|
private:
|
|
bool _internal_has_profile_info() const;
|
|
public:
|
|
void clear_profile_info();
|
|
const ::xla::OpMetadata_ProfileInfo& profile_info() const;
|
|
PROTOBUF_NODISCARD ::xla::OpMetadata_ProfileInfo* release_profile_info();
|
|
::xla::OpMetadata_ProfileInfo* mutable_profile_info();
|
|
void set_allocated_profile_info(::xla::OpMetadata_ProfileInfo* profile_info);
|
|
private:
|
|
const ::xla::OpMetadata_ProfileInfo& _internal_profile_info() const;
|
|
::xla::OpMetadata_ProfileInfo* _internal_mutable_profile_info();
|
|
public:
|
|
void unsafe_arena_set_allocated_profile_info(
|
|
::xla::OpMetadata_ProfileInfo* profile_info);
|
|
::xla::OpMetadata_ProfileInfo* unsafe_arena_release_profile_info();
|
|
|
|
// int64 creation_pass_id = 6;
|
|
void clear_creation_pass_id();
|
|
int64_t creation_pass_id() const;
|
|
void set_creation_pass_id(int64_t value);
|
|
private:
|
|
int64_t _internal_creation_pass_id() const;
|
|
void _internal_set_creation_pass_id(int64_t value);
|
|
public:
|
|
|
|
// int64 logical_creation_pass_id = 7;
|
|
void clear_logical_creation_pass_id();
|
|
int64_t logical_creation_pass_id() const;
|
|
void set_logical_creation_pass_id(int64_t value);
|
|
private:
|
|
int64_t _internal_logical_creation_pass_id() const;
|
|
void _internal_set_logical_creation_pass_id(int64_t value);
|
|
public:
|
|
|
|
// int64 size_of_generated_code_in_bytes = 8;
|
|
void clear_size_of_generated_code_in_bytes();
|
|
int64_t size_of_generated_code_in_bytes() const;
|
|
void set_size_of_generated_code_in_bytes(int64_t value);
|
|
private:
|
|
int64_t _internal_size_of_generated_code_in_bytes() const;
|
|
void _internal_set_size_of_generated_code_in_bytes(int64_t value);
|
|
public:
|
|
|
|
// int32 source_line = 4;
|
|
void clear_source_line();
|
|
int32_t source_line() const;
|
|
void set_source_line(int32_t value);
|
|
private:
|
|
int32_t _internal_source_line() const;
|
|
void _internal_set_source_line(int32_t value);
|
|
public:
|
|
|
|
// bool preserve_layout = 13;
|
|
void clear_preserve_layout();
|
|
bool preserve_layout() const;
|
|
void set_preserve_layout(bool value);
|
|
private:
|
|
bool _internal_preserve_layout() const;
|
|
void _internal_set_preserve_layout(bool value);
|
|
public:
|
|
|
|
// int64 size_of_memory_working_set_in_bytes = 9;
|
|
void clear_size_of_memory_working_set_in_bytes();
|
|
int64_t size_of_memory_working_set_in_bytes() const;
|
|
void set_size_of_memory_working_set_in_bytes(int64_t value);
|
|
private:
|
|
int64_t _internal_size_of_memory_working_set_in_bytes() const;
|
|
void _internal_set_size_of_memory_working_set_in_bytes(int64_t value);
|
|
public:
|
|
|
|
// int32 stack_frame_id = 15;
|
|
void clear_stack_frame_id();
|
|
int32_t stack_frame_id() const;
|
|
void set_stack_frame_id(int32_t value);
|
|
private:
|
|
int32_t _internal_stack_frame_id() const;
|
|
void _internal_set_stack_frame_id(int32_t value);
|
|
public:
|
|
|
|
// @@protoc_insertion_point(class_scope:xla.OpMetadata)
|
|
private:
|
|
class _Internal;
|
|
|
|
template <typename T> friend class ::PROTOBUF_NAMESPACE_ID::Arena::InternalHelper;
|
|
typedef void InternalArenaConstructable_;
|
|
typedef void DestructorSkippable_;
|
|
struct Impl_ {
|
|
::PROTOBUF_NAMESPACE_ID::RepeatedField<int> profile_type_;
|
|
mutable std::atomic<int> _profile_type_cached_byte_size_;
|
|
::PROTOBUF_NAMESPACE_ID::internal::ArenaStringPtr op_type_;
|
|
::PROTOBUF_NAMESPACE_ID::internal::ArenaStringPtr op_name_;
|
|
::PROTOBUF_NAMESPACE_ID::internal::ArenaStringPtr source_file_;
|
|
::PROTOBUF_NAMESPACE_ID::internal::ArenaStringPtr deduplicated_name_;
|
|
::xla::OpMetadata_ProfileInfo* profile_info_;
|
|
int64_t creation_pass_id_;
|
|
int64_t logical_creation_pass_id_;
|
|
int64_t size_of_generated_code_in_bytes_;
|
|
int32_t source_line_;
|
|
bool preserve_layout_;
|
|
int64_t size_of_memory_working_set_in_bytes_;
|
|
int32_t stack_frame_id_;
|
|
mutable ::PROTOBUF_NAMESPACE_ID::internal::CachedSize _cached_size_;
|
|
};
|
|
union { Impl_ _impl_; };
|
|
friend struct ::TableStruct_xla_2fxla_5fdata_2eproto;
|
|
};
|
|
// -------------------------------------------------------------------
|
|
|
|
class ExecutionProfile final :
|
|
public ::PROTOBUF_NAMESPACE_ID::Message /* @@protoc_insertion_point(class_definition:xla.ExecutionProfile) */ {
|
|
public:
|
|
inline ExecutionProfile() : ExecutionProfile(nullptr) {}
|
|
~ExecutionProfile() override;
|
|
explicit PROTOBUF_CONSTEXPR ExecutionProfile(::PROTOBUF_NAMESPACE_ID::internal::ConstantInitialized);
|
|
|
|
ExecutionProfile(const ExecutionProfile& from);
|
|
ExecutionProfile(ExecutionProfile&& from) noexcept
|
|
: ExecutionProfile() {
|
|
*this = ::std::move(from);
|
|
}
|
|
|
|
inline ExecutionProfile& operator=(const ExecutionProfile& from) {
|
|
CopyFrom(from);
|
|
return *this;
|
|
}
|
|
inline ExecutionProfile& operator=(ExecutionProfile&& from) noexcept {
|
|
if (this == &from) return *this;
|
|
if (GetOwningArena() == from.GetOwningArena()
|
|
#ifdef PROTOBUF_FORCE_COPY_IN_MOVE
|
|
&& GetOwningArena() != nullptr
|
|
#endif // !PROTOBUF_FORCE_COPY_IN_MOVE
|
|
) {
|
|
InternalSwap(&from);
|
|
} else {
|
|
CopyFrom(from);
|
|
}
|
|
return *this;
|
|
}
|
|
|
|
static const ::PROTOBUF_NAMESPACE_ID::Descriptor* descriptor() {
|
|
return GetDescriptor();
|
|
}
|
|
static const ::PROTOBUF_NAMESPACE_ID::Descriptor* GetDescriptor() {
|
|
return default_instance().GetMetadata().descriptor;
|
|
}
|
|
static const ::PROTOBUF_NAMESPACE_ID::Reflection* GetReflection() {
|
|
return default_instance().GetMetadata().reflection;
|
|
}
|
|
static const ExecutionProfile& default_instance() {
|
|
return *internal_default_instance();
|
|
}
|
|
static inline const ExecutionProfile* internal_default_instance() {
|
|
return reinterpret_cast<const ExecutionProfile*>(
|
|
&_ExecutionProfile_default_instance_);
|
|
}
|
|
static constexpr int kIndexInFileMessages =
|
|
9;
|
|
|
|
friend void swap(ExecutionProfile& a, ExecutionProfile& b) {
|
|
a.Swap(&b);
|
|
}
|
|
inline void Swap(ExecutionProfile* other) {
|
|
if (other == this) return;
|
|
#ifdef PROTOBUF_FORCE_COPY_IN_SWAP
|
|
if (GetOwningArena() != nullptr &&
|
|
GetOwningArena() == other->GetOwningArena()) {
|
|
#else // PROTOBUF_FORCE_COPY_IN_SWAP
|
|
if (GetOwningArena() == other->GetOwningArena()) {
|
|
#endif // !PROTOBUF_FORCE_COPY_IN_SWAP
|
|
InternalSwap(other);
|
|
} else {
|
|
::PROTOBUF_NAMESPACE_ID::internal::GenericSwap(this, other);
|
|
}
|
|
}
|
|
void UnsafeArenaSwap(ExecutionProfile* other) {
|
|
if (other == this) return;
|
|
GOOGLE_DCHECK(GetOwningArena() == other->GetOwningArena());
|
|
InternalSwap(other);
|
|
}
|
|
|
|
// implements Message ----------------------------------------------
|
|
|
|
ExecutionProfile* New(::PROTOBUF_NAMESPACE_ID::Arena* arena = nullptr) const final {
|
|
return CreateMaybeMessage<ExecutionProfile>(arena);
|
|
}
|
|
using ::PROTOBUF_NAMESPACE_ID::Message::CopyFrom;
|
|
void CopyFrom(const ExecutionProfile& from);
|
|
using ::PROTOBUF_NAMESPACE_ID::Message::MergeFrom;
|
|
void MergeFrom( const ExecutionProfile& from) {
|
|
ExecutionProfile::MergeImpl(*this, from);
|
|
}
|
|
private:
|
|
static void MergeImpl(::PROTOBUF_NAMESPACE_ID::Message& to_msg, const ::PROTOBUF_NAMESPACE_ID::Message& from_msg);
|
|
public:
|
|
PROTOBUF_ATTRIBUTE_REINITIALIZES void Clear() final;
|
|
bool IsInitialized() const final;
|
|
|
|
size_t ByteSizeLong() const final;
|
|
const char* _InternalParse(const char* ptr, ::PROTOBUF_NAMESPACE_ID::internal::ParseContext* ctx) final;
|
|
uint8_t* _InternalSerialize(
|
|
uint8_t* target, ::PROTOBUF_NAMESPACE_ID::io::EpsCopyOutputStream* stream) const final;
|
|
int GetCachedSize() const final { return _impl_._cached_size_.Get(); }
|
|
|
|
private:
|
|
void SharedCtor(::PROTOBUF_NAMESPACE_ID::Arena* arena, bool is_message_owned);
|
|
void SharedDtor();
|
|
void SetCachedSize(int size) const final;
|
|
void InternalSwap(ExecutionProfile* other);
|
|
|
|
private:
|
|
friend class ::PROTOBUF_NAMESPACE_ID::internal::AnyMetadata;
|
|
static ::PROTOBUF_NAMESPACE_ID::StringPiece FullMessageName() {
|
|
return "xla.ExecutionProfile";
|
|
}
|
|
protected:
|
|
explicit ExecutionProfile(::PROTOBUF_NAMESPACE_ID::Arena* arena,
|
|
bool is_message_owned = false);
|
|
public:
|
|
|
|
static const ClassData _class_data_;
|
|
const ::PROTOBUF_NAMESPACE_ID::Message::ClassData*GetClassData() const final;
|
|
|
|
::PROTOBUF_NAMESPACE_ID::Metadata GetMetadata() const final;
|
|
|
|
// nested types ----------------------------------------------------
|
|
|
|
// accessors -------------------------------------------------------
|
|
|
|
enum : int {
|
|
kCompileTimeMsFieldNumber = 2,
|
|
kComputeCycleCountFieldNumber = 3,
|
|
kComputeTimeNsFieldNumber = 4,
|
|
kComputeAndTransferTimeNsFieldNumber = 5,
|
|
kExecutableSizeInBytesFieldNumber = 6,
|
|
kCompilationCacheHitFieldNumber = 1,
|
|
kProfileCacheHitFieldNumber = 7,
|
|
};
|
|
// int64 compile_time_ms = 2;
|
|
void clear_compile_time_ms();
|
|
int64_t compile_time_ms() const;
|
|
void set_compile_time_ms(int64_t value);
|
|
private:
|
|
int64_t _internal_compile_time_ms() const;
|
|
void _internal_set_compile_time_ms(int64_t value);
|
|
public:
|
|
|
|
// int64 compute_cycle_count = 3;
|
|
void clear_compute_cycle_count();
|
|
int64_t compute_cycle_count() const;
|
|
void set_compute_cycle_count(int64_t value);
|
|
private:
|
|
int64_t _internal_compute_cycle_count() const;
|
|
void _internal_set_compute_cycle_count(int64_t value);
|
|
public:
|
|
|
|
// int64 compute_time_ns = 4;
|
|
void clear_compute_time_ns();
|
|
int64_t compute_time_ns() const;
|
|
void set_compute_time_ns(int64_t value);
|
|
private:
|
|
int64_t _internal_compute_time_ns() const;
|
|
void _internal_set_compute_time_ns(int64_t value);
|
|
public:
|
|
|
|
// int64 compute_and_transfer_time_ns = 5;
|
|
void clear_compute_and_transfer_time_ns();
|
|
int64_t compute_and_transfer_time_ns() const;
|
|
void set_compute_and_transfer_time_ns(int64_t value);
|
|
private:
|
|
int64_t _internal_compute_and_transfer_time_ns() const;
|
|
void _internal_set_compute_and_transfer_time_ns(int64_t value);
|
|
public:
|
|
|
|
// int64 executable_size_in_bytes = 6;
|
|
void clear_executable_size_in_bytes();
|
|
int64_t executable_size_in_bytes() const;
|
|
void set_executable_size_in_bytes(int64_t value);
|
|
private:
|
|
int64_t _internal_executable_size_in_bytes() const;
|
|
void _internal_set_executable_size_in_bytes(int64_t value);
|
|
public:
|
|
|
|
// bool compilation_cache_hit = 1;
|
|
void clear_compilation_cache_hit();
|
|
bool compilation_cache_hit() const;
|
|
void set_compilation_cache_hit(bool value);
|
|
private:
|
|
bool _internal_compilation_cache_hit() const;
|
|
void _internal_set_compilation_cache_hit(bool value);
|
|
public:
|
|
|
|
// bool profile_cache_hit = 7;
|
|
void clear_profile_cache_hit();
|
|
bool profile_cache_hit() const;
|
|
void set_profile_cache_hit(bool value);
|
|
private:
|
|
bool _internal_profile_cache_hit() const;
|
|
void _internal_set_profile_cache_hit(bool value);
|
|
public:
|
|
|
|
// @@protoc_insertion_point(class_scope:xla.ExecutionProfile)
|
|
private:
|
|
class _Internal;
|
|
|
|
template <typename T> friend class ::PROTOBUF_NAMESPACE_ID::Arena::InternalHelper;
|
|
typedef void InternalArenaConstructable_;
|
|
typedef void DestructorSkippable_;
|
|
struct Impl_ {
|
|
int64_t compile_time_ms_;
|
|
int64_t compute_cycle_count_;
|
|
int64_t compute_time_ns_;
|
|
int64_t compute_and_transfer_time_ns_;
|
|
int64_t executable_size_in_bytes_;
|
|
bool compilation_cache_hit_;
|
|
bool profile_cache_hit_;
|
|
mutable ::PROTOBUF_NAMESPACE_ID::internal::CachedSize _cached_size_;
|
|
};
|
|
union { Impl_ _impl_; };
|
|
friend struct ::TableStruct_xla_2fxla_5fdata_2eproto;
|
|
};
|
|
// -------------------------------------------------------------------
|
|
|
|
class ExecutionHandle final :
|
|
public ::PROTOBUF_NAMESPACE_ID::Message /* @@protoc_insertion_point(class_definition:xla.ExecutionHandle) */ {
|
|
public:
|
|
inline ExecutionHandle() : ExecutionHandle(nullptr) {}
|
|
~ExecutionHandle() override;
|
|
explicit PROTOBUF_CONSTEXPR ExecutionHandle(::PROTOBUF_NAMESPACE_ID::internal::ConstantInitialized);
|
|
|
|
ExecutionHandle(const ExecutionHandle& from);
|
|
ExecutionHandle(ExecutionHandle&& from) noexcept
|
|
: ExecutionHandle() {
|
|
*this = ::std::move(from);
|
|
}
|
|
|
|
inline ExecutionHandle& operator=(const ExecutionHandle& from) {
|
|
CopyFrom(from);
|
|
return *this;
|
|
}
|
|
inline ExecutionHandle& operator=(ExecutionHandle&& from) noexcept {
|
|
if (this == &from) return *this;
|
|
if (GetOwningArena() == from.GetOwningArena()
|
|
#ifdef PROTOBUF_FORCE_COPY_IN_MOVE
|
|
&& GetOwningArena() != nullptr
|
|
#endif // !PROTOBUF_FORCE_COPY_IN_MOVE
|
|
) {
|
|
InternalSwap(&from);
|
|
} else {
|
|
CopyFrom(from);
|
|
}
|
|
return *this;
|
|
}
|
|
|
|
static const ::PROTOBUF_NAMESPACE_ID::Descriptor* descriptor() {
|
|
return GetDescriptor();
|
|
}
|
|
static const ::PROTOBUF_NAMESPACE_ID::Descriptor* GetDescriptor() {
|
|
return default_instance().GetMetadata().descriptor;
|
|
}
|
|
static const ::PROTOBUF_NAMESPACE_ID::Reflection* GetReflection() {
|
|
return default_instance().GetMetadata().reflection;
|
|
}
|
|
static const ExecutionHandle& default_instance() {
|
|
return *internal_default_instance();
|
|
}
|
|
static inline const ExecutionHandle* internal_default_instance() {
|
|
return reinterpret_cast<const ExecutionHandle*>(
|
|
&_ExecutionHandle_default_instance_);
|
|
}
|
|
static constexpr int kIndexInFileMessages =
|
|
10;
|
|
|
|
friend void swap(ExecutionHandle& a, ExecutionHandle& b) {
|
|
a.Swap(&b);
|
|
}
|
|
inline void Swap(ExecutionHandle* other) {
|
|
if (other == this) return;
|
|
#ifdef PROTOBUF_FORCE_COPY_IN_SWAP
|
|
if (GetOwningArena() != nullptr &&
|
|
GetOwningArena() == other->GetOwningArena()) {
|
|
#else // PROTOBUF_FORCE_COPY_IN_SWAP
|
|
if (GetOwningArena() == other->GetOwningArena()) {
|
|
#endif // !PROTOBUF_FORCE_COPY_IN_SWAP
|
|
InternalSwap(other);
|
|
} else {
|
|
::PROTOBUF_NAMESPACE_ID::internal::GenericSwap(this, other);
|
|
}
|
|
}
|
|
void UnsafeArenaSwap(ExecutionHandle* other) {
|
|
if (other == this) return;
|
|
GOOGLE_DCHECK(GetOwningArena() == other->GetOwningArena());
|
|
InternalSwap(other);
|
|
}
|
|
|
|
// implements Message ----------------------------------------------
|
|
|
|
ExecutionHandle* New(::PROTOBUF_NAMESPACE_ID::Arena* arena = nullptr) const final {
|
|
return CreateMaybeMessage<ExecutionHandle>(arena);
|
|
}
|
|
using ::PROTOBUF_NAMESPACE_ID::Message::CopyFrom;
|
|
void CopyFrom(const ExecutionHandle& from);
|
|
using ::PROTOBUF_NAMESPACE_ID::Message::MergeFrom;
|
|
void MergeFrom( const ExecutionHandle& from) {
|
|
ExecutionHandle::MergeImpl(*this, from);
|
|
}
|
|
private:
|
|
static void MergeImpl(::PROTOBUF_NAMESPACE_ID::Message& to_msg, const ::PROTOBUF_NAMESPACE_ID::Message& from_msg);
|
|
public:
|
|
PROTOBUF_ATTRIBUTE_REINITIALIZES void Clear() final;
|
|
bool IsInitialized() const final;
|
|
|
|
size_t ByteSizeLong() const final;
|
|
const char* _InternalParse(const char* ptr, ::PROTOBUF_NAMESPACE_ID::internal::ParseContext* ctx) final;
|
|
uint8_t* _InternalSerialize(
|
|
uint8_t* target, ::PROTOBUF_NAMESPACE_ID::io::EpsCopyOutputStream* stream) const final;
|
|
int GetCachedSize() const final { return _impl_._cached_size_.Get(); }
|
|
|
|
private:
|
|
void SharedCtor(::PROTOBUF_NAMESPACE_ID::Arena* arena, bool is_message_owned);
|
|
void SharedDtor();
|
|
void SetCachedSize(int size) const final;
|
|
void InternalSwap(ExecutionHandle* other);
|
|
|
|
private:
|
|
friend class ::PROTOBUF_NAMESPACE_ID::internal::AnyMetadata;
|
|
static ::PROTOBUF_NAMESPACE_ID::StringPiece FullMessageName() {
|
|
return "xla.ExecutionHandle";
|
|
}
|
|
protected:
|
|
explicit ExecutionHandle(::PROTOBUF_NAMESPACE_ID::Arena* arena,
|
|
bool is_message_owned = false);
|
|
public:
|
|
|
|
static const ClassData _class_data_;
|
|
const ::PROTOBUF_NAMESPACE_ID::Message::ClassData*GetClassData() const final;
|
|
|
|
::PROTOBUF_NAMESPACE_ID::Metadata GetMetadata() const final;
|
|
|
|
// nested types ----------------------------------------------------
|
|
|
|
// accessors -------------------------------------------------------
|
|
|
|
enum : int {
|
|
kHandleFieldNumber = 1,
|
|
};
|
|
// int64 handle = 1;
|
|
void clear_handle();
|
|
int64_t handle() const;
|
|
void set_handle(int64_t value);
|
|
private:
|
|
int64_t _internal_handle() const;
|
|
void _internal_set_handle(int64_t value);
|
|
public:
|
|
|
|
// @@protoc_insertion_point(class_scope:xla.ExecutionHandle)
|
|
private:
|
|
class _Internal;
|
|
|
|
template <typename T> friend class ::PROTOBUF_NAMESPACE_ID::Arena::InternalHelper;
|
|
typedef void InternalArenaConstructable_;
|
|
typedef void DestructorSkippable_;
|
|
struct Impl_ {
|
|
int64_t handle_;
|
|
mutable ::PROTOBUF_NAMESPACE_ID::internal::CachedSize _cached_size_;
|
|
};
|
|
union { Impl_ _impl_; };
|
|
friend struct ::TableStruct_xla_2fxla_5fdata_2eproto;
|
|
};
|
|
// -------------------------------------------------------------------
|
|
|
|
class GlobalDataHandle final :
|
|
public ::PROTOBUF_NAMESPACE_ID::Message /* @@protoc_insertion_point(class_definition:xla.GlobalDataHandle) */ {
|
|
public:
|
|
inline GlobalDataHandle() : GlobalDataHandle(nullptr) {}
|
|
~GlobalDataHandle() override;
|
|
explicit PROTOBUF_CONSTEXPR GlobalDataHandle(::PROTOBUF_NAMESPACE_ID::internal::ConstantInitialized);
|
|
|
|
GlobalDataHandle(const GlobalDataHandle& from);
|
|
GlobalDataHandle(GlobalDataHandle&& from) noexcept
|
|
: GlobalDataHandle() {
|
|
*this = ::std::move(from);
|
|
}
|
|
|
|
inline GlobalDataHandle& operator=(const GlobalDataHandle& from) {
|
|
CopyFrom(from);
|
|
return *this;
|
|
}
|
|
inline GlobalDataHandle& operator=(GlobalDataHandle&& from) noexcept {
|
|
if (this == &from) return *this;
|
|
if (GetOwningArena() == from.GetOwningArena()
|
|
#ifdef PROTOBUF_FORCE_COPY_IN_MOVE
|
|
&& GetOwningArena() != nullptr
|
|
#endif // !PROTOBUF_FORCE_COPY_IN_MOVE
|
|
) {
|
|
InternalSwap(&from);
|
|
} else {
|
|
CopyFrom(from);
|
|
}
|
|
return *this;
|
|
}
|
|
|
|
static const ::PROTOBUF_NAMESPACE_ID::Descriptor* descriptor() {
|
|
return GetDescriptor();
|
|
}
|
|
static const ::PROTOBUF_NAMESPACE_ID::Descriptor* GetDescriptor() {
|
|
return default_instance().GetMetadata().descriptor;
|
|
}
|
|
static const ::PROTOBUF_NAMESPACE_ID::Reflection* GetReflection() {
|
|
return default_instance().GetMetadata().reflection;
|
|
}
|
|
static const GlobalDataHandle& default_instance() {
|
|
return *internal_default_instance();
|
|
}
|
|
static inline const GlobalDataHandle* internal_default_instance() {
|
|
return reinterpret_cast<const GlobalDataHandle*>(
|
|
&_GlobalDataHandle_default_instance_);
|
|
}
|
|
static constexpr int kIndexInFileMessages =
|
|
11;
|
|
|
|
friend void swap(GlobalDataHandle& a, GlobalDataHandle& b) {
|
|
a.Swap(&b);
|
|
}
|
|
inline void Swap(GlobalDataHandle* other) {
|
|
if (other == this) return;
|
|
#ifdef PROTOBUF_FORCE_COPY_IN_SWAP
|
|
if (GetOwningArena() != nullptr &&
|
|
GetOwningArena() == other->GetOwningArena()) {
|
|
#else // PROTOBUF_FORCE_COPY_IN_SWAP
|
|
if (GetOwningArena() == other->GetOwningArena()) {
|
|
#endif // !PROTOBUF_FORCE_COPY_IN_SWAP
|
|
InternalSwap(other);
|
|
} else {
|
|
::PROTOBUF_NAMESPACE_ID::internal::GenericSwap(this, other);
|
|
}
|
|
}
|
|
void UnsafeArenaSwap(GlobalDataHandle* other) {
|
|
if (other == this) return;
|
|
GOOGLE_DCHECK(GetOwningArena() == other->GetOwningArena());
|
|
InternalSwap(other);
|
|
}
|
|
|
|
// implements Message ----------------------------------------------
|
|
|
|
GlobalDataHandle* New(::PROTOBUF_NAMESPACE_ID::Arena* arena = nullptr) const final {
|
|
return CreateMaybeMessage<GlobalDataHandle>(arena);
|
|
}
|
|
using ::PROTOBUF_NAMESPACE_ID::Message::CopyFrom;
|
|
void CopyFrom(const GlobalDataHandle& from);
|
|
using ::PROTOBUF_NAMESPACE_ID::Message::MergeFrom;
|
|
void MergeFrom( const GlobalDataHandle& from) {
|
|
GlobalDataHandle::MergeImpl(*this, from);
|
|
}
|
|
private:
|
|
static void MergeImpl(::PROTOBUF_NAMESPACE_ID::Message& to_msg, const ::PROTOBUF_NAMESPACE_ID::Message& from_msg);
|
|
public:
|
|
PROTOBUF_ATTRIBUTE_REINITIALIZES void Clear() final;
|
|
bool IsInitialized() const final;
|
|
|
|
size_t ByteSizeLong() const final;
|
|
const char* _InternalParse(const char* ptr, ::PROTOBUF_NAMESPACE_ID::internal::ParseContext* ctx) final;
|
|
uint8_t* _InternalSerialize(
|
|
uint8_t* target, ::PROTOBUF_NAMESPACE_ID::io::EpsCopyOutputStream* stream) const final;
|
|
int GetCachedSize() const final { return _impl_._cached_size_.Get(); }
|
|
|
|
private:
|
|
void SharedCtor(::PROTOBUF_NAMESPACE_ID::Arena* arena, bool is_message_owned);
|
|
void SharedDtor();
|
|
void SetCachedSize(int size) const final;
|
|
void InternalSwap(GlobalDataHandle* other);
|
|
|
|
private:
|
|
friend class ::PROTOBUF_NAMESPACE_ID::internal::AnyMetadata;
|
|
static ::PROTOBUF_NAMESPACE_ID::StringPiece FullMessageName() {
|
|
return "xla.GlobalDataHandle";
|
|
}
|
|
protected:
|
|
explicit GlobalDataHandle(::PROTOBUF_NAMESPACE_ID::Arena* arena,
|
|
bool is_message_owned = false);
|
|
public:
|
|
|
|
static const ClassData _class_data_;
|
|
const ::PROTOBUF_NAMESPACE_ID::Message::ClassData*GetClassData() const final;
|
|
|
|
::PROTOBUF_NAMESPACE_ID::Metadata GetMetadata() const final;
|
|
|
|
// nested types ----------------------------------------------------
|
|
|
|
// accessors -------------------------------------------------------
|
|
|
|
enum : int {
|
|
kHandleFieldNumber = 1,
|
|
};
|
|
// int64 handle = 1;
|
|
void clear_handle();
|
|
int64_t handle() const;
|
|
void set_handle(int64_t value);
|
|
private:
|
|
int64_t _internal_handle() const;
|
|
void _internal_set_handle(int64_t value);
|
|
public:
|
|
|
|
// @@protoc_insertion_point(class_scope:xla.GlobalDataHandle)
|
|
private:
|
|
class _Internal;
|
|
|
|
template <typename T> friend class ::PROTOBUF_NAMESPACE_ID::Arena::InternalHelper;
|
|
typedef void InternalArenaConstructable_;
|
|
typedef void DestructorSkippable_;
|
|
struct Impl_ {
|
|
int64_t handle_;
|
|
mutable ::PROTOBUF_NAMESPACE_ID::internal::CachedSize _cached_size_;
|
|
};
|
|
union { Impl_ _impl_; };
|
|
friend struct ::TableStruct_xla_2fxla_5fdata_2eproto;
|
|
};
|
|
// -------------------------------------------------------------------
|
|
|
|
class DeviceHandle final :
|
|
public ::PROTOBUF_NAMESPACE_ID::Message /* @@protoc_insertion_point(class_definition:xla.DeviceHandle) */ {
|
|
public:
|
|
inline DeviceHandle() : DeviceHandle(nullptr) {}
|
|
~DeviceHandle() override;
|
|
explicit PROTOBUF_CONSTEXPR DeviceHandle(::PROTOBUF_NAMESPACE_ID::internal::ConstantInitialized);
|
|
|
|
DeviceHandle(const DeviceHandle& from);
|
|
DeviceHandle(DeviceHandle&& from) noexcept
|
|
: DeviceHandle() {
|
|
*this = ::std::move(from);
|
|
}
|
|
|
|
inline DeviceHandle& operator=(const DeviceHandle& from) {
|
|
CopyFrom(from);
|
|
return *this;
|
|
}
|
|
inline DeviceHandle& operator=(DeviceHandle&& from) noexcept {
|
|
if (this == &from) return *this;
|
|
if (GetOwningArena() == from.GetOwningArena()
|
|
#ifdef PROTOBUF_FORCE_COPY_IN_MOVE
|
|
&& GetOwningArena() != nullptr
|
|
#endif // !PROTOBUF_FORCE_COPY_IN_MOVE
|
|
) {
|
|
InternalSwap(&from);
|
|
} else {
|
|
CopyFrom(from);
|
|
}
|
|
return *this;
|
|
}
|
|
|
|
static const ::PROTOBUF_NAMESPACE_ID::Descriptor* descriptor() {
|
|
return GetDescriptor();
|
|
}
|
|
static const ::PROTOBUF_NAMESPACE_ID::Descriptor* GetDescriptor() {
|
|
return default_instance().GetMetadata().descriptor;
|
|
}
|
|
static const ::PROTOBUF_NAMESPACE_ID::Reflection* GetReflection() {
|
|
return default_instance().GetMetadata().reflection;
|
|
}
|
|
static const DeviceHandle& default_instance() {
|
|
return *internal_default_instance();
|
|
}
|
|
static inline const DeviceHandle* internal_default_instance() {
|
|
return reinterpret_cast<const DeviceHandle*>(
|
|
&_DeviceHandle_default_instance_);
|
|
}
|
|
static constexpr int kIndexInFileMessages =
|
|
12;
|
|
|
|
friend void swap(DeviceHandle& a, DeviceHandle& b) {
|
|
a.Swap(&b);
|
|
}
|
|
inline void Swap(DeviceHandle* other) {
|
|
if (other == this) return;
|
|
#ifdef PROTOBUF_FORCE_COPY_IN_SWAP
|
|
if (GetOwningArena() != nullptr &&
|
|
GetOwningArena() == other->GetOwningArena()) {
|
|
#else // PROTOBUF_FORCE_COPY_IN_SWAP
|
|
if (GetOwningArena() == other->GetOwningArena()) {
|
|
#endif // !PROTOBUF_FORCE_COPY_IN_SWAP
|
|
InternalSwap(other);
|
|
} else {
|
|
::PROTOBUF_NAMESPACE_ID::internal::GenericSwap(this, other);
|
|
}
|
|
}
|
|
void UnsafeArenaSwap(DeviceHandle* other) {
|
|
if (other == this) return;
|
|
GOOGLE_DCHECK(GetOwningArena() == other->GetOwningArena());
|
|
InternalSwap(other);
|
|
}
|
|
|
|
// implements Message ----------------------------------------------
|
|
|
|
DeviceHandle* New(::PROTOBUF_NAMESPACE_ID::Arena* arena = nullptr) const final {
|
|
return CreateMaybeMessage<DeviceHandle>(arena);
|
|
}
|
|
using ::PROTOBUF_NAMESPACE_ID::Message::CopyFrom;
|
|
void CopyFrom(const DeviceHandle& from);
|
|
using ::PROTOBUF_NAMESPACE_ID::Message::MergeFrom;
|
|
void MergeFrom( const DeviceHandle& from) {
|
|
DeviceHandle::MergeImpl(*this, from);
|
|
}
|
|
private:
|
|
static void MergeImpl(::PROTOBUF_NAMESPACE_ID::Message& to_msg, const ::PROTOBUF_NAMESPACE_ID::Message& from_msg);
|
|
public:
|
|
PROTOBUF_ATTRIBUTE_REINITIALIZES void Clear() final;
|
|
bool IsInitialized() const final;
|
|
|
|
size_t ByteSizeLong() const final;
|
|
const char* _InternalParse(const char* ptr, ::PROTOBUF_NAMESPACE_ID::internal::ParseContext* ctx) final;
|
|
uint8_t* _InternalSerialize(
|
|
uint8_t* target, ::PROTOBUF_NAMESPACE_ID::io::EpsCopyOutputStream* stream) const final;
|
|
int GetCachedSize() const final { return _impl_._cached_size_.Get(); }
|
|
|
|
private:
|
|
void SharedCtor(::PROTOBUF_NAMESPACE_ID::Arena* arena, bool is_message_owned);
|
|
void SharedDtor();
|
|
void SetCachedSize(int size) const final;
|
|
void InternalSwap(DeviceHandle* other);
|
|
|
|
private:
|
|
friend class ::PROTOBUF_NAMESPACE_ID::internal::AnyMetadata;
|
|
static ::PROTOBUF_NAMESPACE_ID::StringPiece FullMessageName() {
|
|
return "xla.DeviceHandle";
|
|
}
|
|
protected:
|
|
explicit DeviceHandle(::PROTOBUF_NAMESPACE_ID::Arena* arena,
|
|
bool is_message_owned = false);
|
|
public:
|
|
|
|
static const ClassData _class_data_;
|
|
const ::PROTOBUF_NAMESPACE_ID::Message::ClassData*GetClassData() const final;
|
|
|
|
::PROTOBUF_NAMESPACE_ID::Metadata GetMetadata() const final;
|
|
|
|
// nested types ----------------------------------------------------
|
|
|
|
// accessors -------------------------------------------------------
|
|
|
|
enum : int {
|
|
kHandleFieldNumber = 1,
|
|
kDeviceCountFieldNumber = 2,
|
|
};
|
|
// int64 handle = 1;
|
|
void clear_handle();
|
|
int64_t handle() const;
|
|
void set_handle(int64_t value);
|
|
private:
|
|
int64_t _internal_handle() const;
|
|
void _internal_set_handle(int64_t value);
|
|
public:
|
|
|
|
// int64 device_count = 2;
|
|
void clear_device_count();
|
|
int64_t device_count() const;
|
|
void set_device_count(int64_t value);
|
|
private:
|
|
int64_t _internal_device_count() const;
|
|
void _internal_set_device_count(int64_t value);
|
|
public:
|
|
|
|
// @@protoc_insertion_point(class_scope:xla.DeviceHandle)
|
|
private:
|
|
class _Internal;
|
|
|
|
template <typename T> friend class ::PROTOBUF_NAMESPACE_ID::Arena::InternalHelper;
|
|
typedef void InternalArenaConstructable_;
|
|
typedef void DestructorSkippable_;
|
|
struct Impl_ {
|
|
int64_t handle_;
|
|
int64_t device_count_;
|
|
mutable ::PROTOBUF_NAMESPACE_ID::internal::CachedSize _cached_size_;
|
|
};
|
|
union { Impl_ _impl_; };
|
|
friend struct ::TableStruct_xla_2fxla_5fdata_2eproto;
|
|
};
|
|
// -------------------------------------------------------------------
|
|
|
|
class ChannelHandle final :
|
|
public ::PROTOBUF_NAMESPACE_ID::Message /* @@protoc_insertion_point(class_definition:xla.ChannelHandle) */ {
|
|
public:
|
|
inline ChannelHandle() : ChannelHandle(nullptr) {}
|
|
~ChannelHandle() override;
|
|
explicit PROTOBUF_CONSTEXPR ChannelHandle(::PROTOBUF_NAMESPACE_ID::internal::ConstantInitialized);
|
|
|
|
ChannelHandle(const ChannelHandle& from);
|
|
ChannelHandle(ChannelHandle&& from) noexcept
|
|
: ChannelHandle() {
|
|
*this = ::std::move(from);
|
|
}
|
|
|
|
inline ChannelHandle& operator=(const ChannelHandle& from) {
|
|
CopyFrom(from);
|
|
return *this;
|
|
}
|
|
inline ChannelHandle& operator=(ChannelHandle&& from) noexcept {
|
|
if (this == &from) return *this;
|
|
if (GetOwningArena() == from.GetOwningArena()
|
|
#ifdef PROTOBUF_FORCE_COPY_IN_MOVE
|
|
&& GetOwningArena() != nullptr
|
|
#endif // !PROTOBUF_FORCE_COPY_IN_MOVE
|
|
) {
|
|
InternalSwap(&from);
|
|
} else {
|
|
CopyFrom(from);
|
|
}
|
|
return *this;
|
|
}
|
|
|
|
static const ::PROTOBUF_NAMESPACE_ID::Descriptor* descriptor() {
|
|
return GetDescriptor();
|
|
}
|
|
static const ::PROTOBUF_NAMESPACE_ID::Descriptor* GetDescriptor() {
|
|
return default_instance().GetMetadata().descriptor;
|
|
}
|
|
static const ::PROTOBUF_NAMESPACE_ID::Reflection* GetReflection() {
|
|
return default_instance().GetMetadata().reflection;
|
|
}
|
|
static const ChannelHandle& default_instance() {
|
|
return *internal_default_instance();
|
|
}
|
|
static inline const ChannelHandle* internal_default_instance() {
|
|
return reinterpret_cast<const ChannelHandle*>(
|
|
&_ChannelHandle_default_instance_);
|
|
}
|
|
static constexpr int kIndexInFileMessages =
|
|
13;
|
|
|
|
friend void swap(ChannelHandle& a, ChannelHandle& b) {
|
|
a.Swap(&b);
|
|
}
|
|
inline void Swap(ChannelHandle* other) {
|
|
if (other == this) return;
|
|
#ifdef PROTOBUF_FORCE_COPY_IN_SWAP
|
|
if (GetOwningArena() != nullptr &&
|
|
GetOwningArena() == other->GetOwningArena()) {
|
|
#else // PROTOBUF_FORCE_COPY_IN_SWAP
|
|
if (GetOwningArena() == other->GetOwningArena()) {
|
|
#endif // !PROTOBUF_FORCE_COPY_IN_SWAP
|
|
InternalSwap(other);
|
|
} else {
|
|
::PROTOBUF_NAMESPACE_ID::internal::GenericSwap(this, other);
|
|
}
|
|
}
|
|
void UnsafeArenaSwap(ChannelHandle* other) {
|
|
if (other == this) return;
|
|
GOOGLE_DCHECK(GetOwningArena() == other->GetOwningArena());
|
|
InternalSwap(other);
|
|
}
|
|
|
|
// implements Message ----------------------------------------------
|
|
|
|
ChannelHandle* New(::PROTOBUF_NAMESPACE_ID::Arena* arena = nullptr) const final {
|
|
return CreateMaybeMessage<ChannelHandle>(arena);
|
|
}
|
|
using ::PROTOBUF_NAMESPACE_ID::Message::CopyFrom;
|
|
void CopyFrom(const ChannelHandle& from);
|
|
using ::PROTOBUF_NAMESPACE_ID::Message::MergeFrom;
|
|
void MergeFrom( const ChannelHandle& from) {
|
|
ChannelHandle::MergeImpl(*this, from);
|
|
}
|
|
private:
|
|
static void MergeImpl(::PROTOBUF_NAMESPACE_ID::Message& to_msg, const ::PROTOBUF_NAMESPACE_ID::Message& from_msg);
|
|
public:
|
|
PROTOBUF_ATTRIBUTE_REINITIALIZES void Clear() final;
|
|
bool IsInitialized() const final;
|
|
|
|
size_t ByteSizeLong() const final;
|
|
const char* _InternalParse(const char* ptr, ::PROTOBUF_NAMESPACE_ID::internal::ParseContext* ctx) final;
|
|
uint8_t* _InternalSerialize(
|
|
uint8_t* target, ::PROTOBUF_NAMESPACE_ID::io::EpsCopyOutputStream* stream) const final;
|
|
int GetCachedSize() const final { return _impl_._cached_size_.Get(); }
|
|
|
|
private:
|
|
void SharedCtor(::PROTOBUF_NAMESPACE_ID::Arena* arena, bool is_message_owned);
|
|
void SharedDtor();
|
|
void SetCachedSize(int size) const final;
|
|
void InternalSwap(ChannelHandle* other);
|
|
|
|
private:
|
|
friend class ::PROTOBUF_NAMESPACE_ID::internal::AnyMetadata;
|
|
static ::PROTOBUF_NAMESPACE_ID::StringPiece FullMessageName() {
|
|
return "xla.ChannelHandle";
|
|
}
|
|
protected:
|
|
explicit ChannelHandle(::PROTOBUF_NAMESPACE_ID::Arena* arena,
|
|
bool is_message_owned = false);
|
|
public:
|
|
|
|
static const ClassData _class_data_;
|
|
const ::PROTOBUF_NAMESPACE_ID::Message::ClassData*GetClassData() const final;
|
|
|
|
::PROTOBUF_NAMESPACE_ID::Metadata GetMetadata() const final;
|
|
|
|
// nested types ----------------------------------------------------
|
|
|
|
typedef ChannelHandle_ChannelType ChannelType;
|
|
static constexpr ChannelType CHANNEL_TYPE_INVALID =
|
|
ChannelHandle_ChannelType_CHANNEL_TYPE_INVALID;
|
|
static constexpr ChannelType DEVICE_TO_DEVICE =
|
|
ChannelHandle_ChannelType_DEVICE_TO_DEVICE;
|
|
static constexpr ChannelType DEVICE_TO_HOST =
|
|
ChannelHandle_ChannelType_DEVICE_TO_HOST;
|
|
static constexpr ChannelType HOST_TO_DEVICE =
|
|
ChannelHandle_ChannelType_HOST_TO_DEVICE;
|
|
static inline bool ChannelType_IsValid(int value) {
|
|
return ChannelHandle_ChannelType_IsValid(value);
|
|
}
|
|
static constexpr ChannelType ChannelType_MIN =
|
|
ChannelHandle_ChannelType_ChannelType_MIN;
|
|
static constexpr ChannelType ChannelType_MAX =
|
|
ChannelHandle_ChannelType_ChannelType_MAX;
|
|
static constexpr int ChannelType_ARRAYSIZE =
|
|
ChannelHandle_ChannelType_ChannelType_ARRAYSIZE;
|
|
static inline const ::PROTOBUF_NAMESPACE_ID::EnumDescriptor*
|
|
ChannelType_descriptor() {
|
|
return ChannelHandle_ChannelType_descriptor();
|
|
}
|
|
template<typename T>
|
|
static inline const std::string& ChannelType_Name(T enum_t_value) {
|
|
static_assert(::std::is_same<T, ChannelType>::value ||
|
|
::std::is_integral<T>::value,
|
|
"Incorrect type passed to function ChannelType_Name.");
|
|
return ChannelHandle_ChannelType_Name(enum_t_value);
|
|
}
|
|
static inline bool ChannelType_Parse(::PROTOBUF_NAMESPACE_ID::ConstStringParam name,
|
|
ChannelType* value) {
|
|
return ChannelHandle_ChannelType_Parse(name, value);
|
|
}
|
|
|
|
// accessors -------------------------------------------------------
|
|
|
|
enum : int {
|
|
kHandleFieldNumber = 1,
|
|
kTypeFieldNumber = 2,
|
|
};
|
|
// int64 handle = 1;
|
|
void clear_handle();
|
|
int64_t handle() const;
|
|
void set_handle(int64_t value);
|
|
private:
|
|
int64_t _internal_handle() const;
|
|
void _internal_set_handle(int64_t value);
|
|
public:
|
|
|
|
// .xla.ChannelHandle.ChannelType type = 2;
|
|
void clear_type();
|
|
::xla::ChannelHandle_ChannelType type() const;
|
|
void set_type(::xla::ChannelHandle_ChannelType value);
|
|
private:
|
|
::xla::ChannelHandle_ChannelType _internal_type() const;
|
|
void _internal_set_type(::xla::ChannelHandle_ChannelType value);
|
|
public:
|
|
|
|
// @@protoc_insertion_point(class_scope:xla.ChannelHandle)
|
|
private:
|
|
class _Internal;
|
|
|
|
template <typename T> friend class ::PROTOBUF_NAMESPACE_ID::Arena::InternalHelper;
|
|
typedef void InternalArenaConstructable_;
|
|
typedef void DestructorSkippable_;
|
|
struct Impl_ {
|
|
int64_t handle_;
|
|
int type_;
|
|
mutable ::PROTOBUF_NAMESPACE_ID::internal::CachedSize _cached_size_;
|
|
};
|
|
union { Impl_ _impl_; };
|
|
friend struct ::TableStruct_xla_2fxla_5fdata_2eproto;
|
|
};
|
|
// -------------------------------------------------------------------
|
|
|
|
class DeviceAssignmentProto_ComputationDevice final :
|
|
public ::PROTOBUF_NAMESPACE_ID::Message /* @@protoc_insertion_point(class_definition:xla.DeviceAssignmentProto.ComputationDevice) */ {
|
|
public:
|
|
inline DeviceAssignmentProto_ComputationDevice() : DeviceAssignmentProto_ComputationDevice(nullptr) {}
|
|
~DeviceAssignmentProto_ComputationDevice() override;
|
|
explicit PROTOBUF_CONSTEXPR DeviceAssignmentProto_ComputationDevice(::PROTOBUF_NAMESPACE_ID::internal::ConstantInitialized);
|
|
|
|
DeviceAssignmentProto_ComputationDevice(const DeviceAssignmentProto_ComputationDevice& from);
|
|
DeviceAssignmentProto_ComputationDevice(DeviceAssignmentProto_ComputationDevice&& from) noexcept
|
|
: DeviceAssignmentProto_ComputationDevice() {
|
|
*this = ::std::move(from);
|
|
}
|
|
|
|
inline DeviceAssignmentProto_ComputationDevice& operator=(const DeviceAssignmentProto_ComputationDevice& from) {
|
|
CopyFrom(from);
|
|
return *this;
|
|
}
|
|
inline DeviceAssignmentProto_ComputationDevice& operator=(DeviceAssignmentProto_ComputationDevice&& from) noexcept {
|
|
if (this == &from) return *this;
|
|
if (GetOwningArena() == from.GetOwningArena()
|
|
#ifdef PROTOBUF_FORCE_COPY_IN_MOVE
|
|
&& GetOwningArena() != nullptr
|
|
#endif // !PROTOBUF_FORCE_COPY_IN_MOVE
|
|
) {
|
|
InternalSwap(&from);
|
|
} else {
|
|
CopyFrom(from);
|
|
}
|
|
return *this;
|
|
}
|
|
|
|
static const ::PROTOBUF_NAMESPACE_ID::Descriptor* descriptor() {
|
|
return GetDescriptor();
|
|
}
|
|
static const ::PROTOBUF_NAMESPACE_ID::Descriptor* GetDescriptor() {
|
|
return default_instance().GetMetadata().descriptor;
|
|
}
|
|
static const ::PROTOBUF_NAMESPACE_ID::Reflection* GetReflection() {
|
|
return default_instance().GetMetadata().reflection;
|
|
}
|
|
static const DeviceAssignmentProto_ComputationDevice& default_instance() {
|
|
return *internal_default_instance();
|
|
}
|
|
static inline const DeviceAssignmentProto_ComputationDevice* internal_default_instance() {
|
|
return reinterpret_cast<const DeviceAssignmentProto_ComputationDevice*>(
|
|
&_DeviceAssignmentProto_ComputationDevice_default_instance_);
|
|
}
|
|
static constexpr int kIndexInFileMessages =
|
|
14;
|
|
|
|
friend void swap(DeviceAssignmentProto_ComputationDevice& a, DeviceAssignmentProto_ComputationDevice& b) {
|
|
a.Swap(&b);
|
|
}
|
|
inline void Swap(DeviceAssignmentProto_ComputationDevice* other) {
|
|
if (other == this) return;
|
|
#ifdef PROTOBUF_FORCE_COPY_IN_SWAP
|
|
if (GetOwningArena() != nullptr &&
|
|
GetOwningArena() == other->GetOwningArena()) {
|
|
#else // PROTOBUF_FORCE_COPY_IN_SWAP
|
|
if (GetOwningArena() == other->GetOwningArena()) {
|
|
#endif // !PROTOBUF_FORCE_COPY_IN_SWAP
|
|
InternalSwap(other);
|
|
} else {
|
|
::PROTOBUF_NAMESPACE_ID::internal::GenericSwap(this, other);
|
|
}
|
|
}
|
|
void UnsafeArenaSwap(DeviceAssignmentProto_ComputationDevice* other) {
|
|
if (other == this) return;
|
|
GOOGLE_DCHECK(GetOwningArena() == other->GetOwningArena());
|
|
InternalSwap(other);
|
|
}
|
|
|
|
// implements Message ----------------------------------------------
|
|
|
|
DeviceAssignmentProto_ComputationDevice* New(::PROTOBUF_NAMESPACE_ID::Arena* arena = nullptr) const final {
|
|
return CreateMaybeMessage<DeviceAssignmentProto_ComputationDevice>(arena);
|
|
}
|
|
using ::PROTOBUF_NAMESPACE_ID::Message::CopyFrom;
|
|
void CopyFrom(const DeviceAssignmentProto_ComputationDevice& from);
|
|
using ::PROTOBUF_NAMESPACE_ID::Message::MergeFrom;
|
|
void MergeFrom( const DeviceAssignmentProto_ComputationDevice& from) {
|
|
DeviceAssignmentProto_ComputationDevice::MergeImpl(*this, from);
|
|
}
|
|
private:
|
|
static void MergeImpl(::PROTOBUF_NAMESPACE_ID::Message& to_msg, const ::PROTOBUF_NAMESPACE_ID::Message& from_msg);
|
|
public:
|
|
PROTOBUF_ATTRIBUTE_REINITIALIZES void Clear() final;
|
|
bool IsInitialized() const final;
|
|
|
|
size_t ByteSizeLong() const final;
|
|
const char* _InternalParse(const char* ptr, ::PROTOBUF_NAMESPACE_ID::internal::ParseContext* ctx) final;
|
|
uint8_t* _InternalSerialize(
|
|
uint8_t* target, ::PROTOBUF_NAMESPACE_ID::io::EpsCopyOutputStream* stream) const final;
|
|
int GetCachedSize() const final { return _impl_._cached_size_.Get(); }
|
|
|
|
private:
|
|
void SharedCtor(::PROTOBUF_NAMESPACE_ID::Arena* arena, bool is_message_owned);
|
|
void SharedDtor();
|
|
void SetCachedSize(int size) const final;
|
|
void InternalSwap(DeviceAssignmentProto_ComputationDevice* other);
|
|
|
|
private:
|
|
friend class ::PROTOBUF_NAMESPACE_ID::internal::AnyMetadata;
|
|
static ::PROTOBUF_NAMESPACE_ID::StringPiece FullMessageName() {
|
|
return "xla.DeviceAssignmentProto.ComputationDevice";
|
|
}
|
|
protected:
|
|
explicit DeviceAssignmentProto_ComputationDevice(::PROTOBUF_NAMESPACE_ID::Arena* arena,
|
|
bool is_message_owned = false);
|
|
public:
|
|
|
|
static const ClassData _class_data_;
|
|
const ::PROTOBUF_NAMESPACE_ID::Message::ClassData*GetClassData() const final;
|
|
|
|
::PROTOBUF_NAMESPACE_ID::Metadata GetMetadata() const final;
|
|
|
|
// nested types ----------------------------------------------------
|
|
|
|
// accessors -------------------------------------------------------
|
|
|
|
enum : int {
|
|
kReplicaDeviceIdsFieldNumber = 1,
|
|
};
|
|
// repeated int32 replica_device_ids = 1;
|
|
int replica_device_ids_size() const;
|
|
private:
|
|
int _internal_replica_device_ids_size() const;
|
|
public:
|
|
void clear_replica_device_ids();
|
|
private:
|
|
int32_t _internal_replica_device_ids(int index) const;
|
|
const ::PROTOBUF_NAMESPACE_ID::RepeatedField< int32_t >&
|
|
_internal_replica_device_ids() const;
|
|
void _internal_add_replica_device_ids(int32_t value);
|
|
::PROTOBUF_NAMESPACE_ID::RepeatedField< int32_t >*
|
|
_internal_mutable_replica_device_ids();
|
|
public:
|
|
int32_t replica_device_ids(int index) const;
|
|
void set_replica_device_ids(int index, int32_t value);
|
|
void add_replica_device_ids(int32_t value);
|
|
const ::PROTOBUF_NAMESPACE_ID::RepeatedField< int32_t >&
|
|
replica_device_ids() const;
|
|
::PROTOBUF_NAMESPACE_ID::RepeatedField< int32_t >*
|
|
mutable_replica_device_ids();
|
|
|
|
// @@protoc_insertion_point(class_scope:xla.DeviceAssignmentProto.ComputationDevice)
|
|
private:
|
|
class _Internal;
|
|
|
|
template <typename T> friend class ::PROTOBUF_NAMESPACE_ID::Arena::InternalHelper;
|
|
typedef void InternalArenaConstructable_;
|
|
typedef void DestructorSkippable_;
|
|
struct Impl_ {
|
|
::PROTOBUF_NAMESPACE_ID::RepeatedField< int32_t > replica_device_ids_;
|
|
mutable std::atomic<int> _replica_device_ids_cached_byte_size_;
|
|
mutable ::PROTOBUF_NAMESPACE_ID::internal::CachedSize _cached_size_;
|
|
};
|
|
union { Impl_ _impl_; };
|
|
friend struct ::TableStruct_xla_2fxla_5fdata_2eproto;
|
|
};
|
|
// -------------------------------------------------------------------
|
|
|
|
class DeviceAssignmentProto final :
|
|
public ::PROTOBUF_NAMESPACE_ID::Message /* @@protoc_insertion_point(class_definition:xla.DeviceAssignmentProto) */ {
|
|
public:
|
|
inline DeviceAssignmentProto() : DeviceAssignmentProto(nullptr) {}
|
|
~DeviceAssignmentProto() override;
|
|
explicit PROTOBUF_CONSTEXPR DeviceAssignmentProto(::PROTOBUF_NAMESPACE_ID::internal::ConstantInitialized);
|
|
|
|
DeviceAssignmentProto(const DeviceAssignmentProto& from);
|
|
DeviceAssignmentProto(DeviceAssignmentProto&& from) noexcept
|
|
: DeviceAssignmentProto() {
|
|
*this = ::std::move(from);
|
|
}
|
|
|
|
inline DeviceAssignmentProto& operator=(const DeviceAssignmentProto& from) {
|
|
CopyFrom(from);
|
|
return *this;
|
|
}
|
|
inline DeviceAssignmentProto& operator=(DeviceAssignmentProto&& from) noexcept {
|
|
if (this == &from) return *this;
|
|
if (GetOwningArena() == from.GetOwningArena()
|
|
#ifdef PROTOBUF_FORCE_COPY_IN_MOVE
|
|
&& GetOwningArena() != nullptr
|
|
#endif // !PROTOBUF_FORCE_COPY_IN_MOVE
|
|
) {
|
|
InternalSwap(&from);
|
|
} else {
|
|
CopyFrom(from);
|
|
}
|
|
return *this;
|
|
}
|
|
|
|
static const ::PROTOBUF_NAMESPACE_ID::Descriptor* descriptor() {
|
|
return GetDescriptor();
|
|
}
|
|
static const ::PROTOBUF_NAMESPACE_ID::Descriptor* GetDescriptor() {
|
|
return default_instance().GetMetadata().descriptor;
|
|
}
|
|
static const ::PROTOBUF_NAMESPACE_ID::Reflection* GetReflection() {
|
|
return default_instance().GetMetadata().reflection;
|
|
}
|
|
static const DeviceAssignmentProto& default_instance() {
|
|
return *internal_default_instance();
|
|
}
|
|
static inline const DeviceAssignmentProto* internal_default_instance() {
|
|
return reinterpret_cast<const DeviceAssignmentProto*>(
|
|
&_DeviceAssignmentProto_default_instance_);
|
|
}
|
|
static constexpr int kIndexInFileMessages =
|
|
15;
|
|
|
|
friend void swap(DeviceAssignmentProto& a, DeviceAssignmentProto& b) {
|
|
a.Swap(&b);
|
|
}
|
|
inline void Swap(DeviceAssignmentProto* other) {
|
|
if (other == this) return;
|
|
#ifdef PROTOBUF_FORCE_COPY_IN_SWAP
|
|
if (GetOwningArena() != nullptr &&
|
|
GetOwningArena() == other->GetOwningArena()) {
|
|
#else // PROTOBUF_FORCE_COPY_IN_SWAP
|
|
if (GetOwningArena() == other->GetOwningArena()) {
|
|
#endif // !PROTOBUF_FORCE_COPY_IN_SWAP
|
|
InternalSwap(other);
|
|
} else {
|
|
::PROTOBUF_NAMESPACE_ID::internal::GenericSwap(this, other);
|
|
}
|
|
}
|
|
void UnsafeArenaSwap(DeviceAssignmentProto* other) {
|
|
if (other == this) return;
|
|
GOOGLE_DCHECK(GetOwningArena() == other->GetOwningArena());
|
|
InternalSwap(other);
|
|
}
|
|
|
|
// implements Message ----------------------------------------------
|
|
|
|
DeviceAssignmentProto* New(::PROTOBUF_NAMESPACE_ID::Arena* arena = nullptr) const final {
|
|
return CreateMaybeMessage<DeviceAssignmentProto>(arena);
|
|
}
|
|
using ::PROTOBUF_NAMESPACE_ID::Message::CopyFrom;
|
|
void CopyFrom(const DeviceAssignmentProto& from);
|
|
using ::PROTOBUF_NAMESPACE_ID::Message::MergeFrom;
|
|
void MergeFrom( const DeviceAssignmentProto& from) {
|
|
DeviceAssignmentProto::MergeImpl(*this, from);
|
|
}
|
|
private:
|
|
static void MergeImpl(::PROTOBUF_NAMESPACE_ID::Message& to_msg, const ::PROTOBUF_NAMESPACE_ID::Message& from_msg);
|
|
public:
|
|
PROTOBUF_ATTRIBUTE_REINITIALIZES void Clear() final;
|
|
bool IsInitialized() const final;
|
|
|
|
size_t ByteSizeLong() const final;
|
|
const char* _InternalParse(const char* ptr, ::PROTOBUF_NAMESPACE_ID::internal::ParseContext* ctx) final;
|
|
uint8_t* _InternalSerialize(
|
|
uint8_t* target, ::PROTOBUF_NAMESPACE_ID::io::EpsCopyOutputStream* stream) const final;
|
|
int GetCachedSize() const final { return _impl_._cached_size_.Get(); }
|
|
|
|
private:
|
|
void SharedCtor(::PROTOBUF_NAMESPACE_ID::Arena* arena, bool is_message_owned);
|
|
void SharedDtor();
|
|
void SetCachedSize(int size) const final;
|
|
void InternalSwap(DeviceAssignmentProto* other);
|
|
|
|
private:
|
|
friend class ::PROTOBUF_NAMESPACE_ID::internal::AnyMetadata;
|
|
static ::PROTOBUF_NAMESPACE_ID::StringPiece FullMessageName() {
|
|
return "xla.DeviceAssignmentProto";
|
|
}
|
|
protected:
|
|
explicit DeviceAssignmentProto(::PROTOBUF_NAMESPACE_ID::Arena* arena,
|
|
bool is_message_owned = false);
|
|
public:
|
|
|
|
static const ClassData _class_data_;
|
|
const ::PROTOBUF_NAMESPACE_ID::Message::ClassData*GetClassData() const final;
|
|
|
|
::PROTOBUF_NAMESPACE_ID::Metadata GetMetadata() const final;
|
|
|
|
// nested types ----------------------------------------------------
|
|
|
|
typedef DeviceAssignmentProto_ComputationDevice ComputationDevice;
|
|
|
|
// accessors -------------------------------------------------------
|
|
|
|
enum : int {
|
|
kComputationDevicesFieldNumber = 3,
|
|
kReplicaCountFieldNumber = 1,
|
|
kComputationCountFieldNumber = 2,
|
|
};
|
|
// repeated .xla.DeviceAssignmentProto.ComputationDevice computation_devices = 3;
|
|
int computation_devices_size() const;
|
|
private:
|
|
int _internal_computation_devices_size() const;
|
|
public:
|
|
void clear_computation_devices();
|
|
::xla::DeviceAssignmentProto_ComputationDevice* mutable_computation_devices(int index);
|
|
::PROTOBUF_NAMESPACE_ID::RepeatedPtrField< ::xla::DeviceAssignmentProto_ComputationDevice >*
|
|
mutable_computation_devices();
|
|
private:
|
|
const ::xla::DeviceAssignmentProto_ComputationDevice& _internal_computation_devices(int index) const;
|
|
::xla::DeviceAssignmentProto_ComputationDevice* _internal_add_computation_devices();
|
|
public:
|
|
const ::xla::DeviceAssignmentProto_ComputationDevice& computation_devices(int index) const;
|
|
::xla::DeviceAssignmentProto_ComputationDevice* add_computation_devices();
|
|
const ::PROTOBUF_NAMESPACE_ID::RepeatedPtrField< ::xla::DeviceAssignmentProto_ComputationDevice >&
|
|
computation_devices() const;
|
|
|
|
// int32 replica_count = 1;
|
|
void clear_replica_count();
|
|
int32_t replica_count() const;
|
|
void set_replica_count(int32_t value);
|
|
private:
|
|
int32_t _internal_replica_count() const;
|
|
void _internal_set_replica_count(int32_t value);
|
|
public:
|
|
|
|
// int32 computation_count = 2;
|
|
void clear_computation_count();
|
|
int32_t computation_count() const;
|
|
void set_computation_count(int32_t value);
|
|
private:
|
|
int32_t _internal_computation_count() const;
|
|
void _internal_set_computation_count(int32_t value);
|
|
public:
|
|
|
|
// @@protoc_insertion_point(class_scope:xla.DeviceAssignmentProto)
|
|
private:
|
|
class _Internal;
|
|
|
|
template <typename T> friend class ::PROTOBUF_NAMESPACE_ID::Arena::InternalHelper;
|
|
typedef void InternalArenaConstructable_;
|
|
typedef void DestructorSkippable_;
|
|
struct Impl_ {
|
|
::PROTOBUF_NAMESPACE_ID::RepeatedPtrField< ::xla::DeviceAssignmentProto_ComputationDevice > computation_devices_;
|
|
int32_t replica_count_;
|
|
int32_t computation_count_;
|
|
mutable ::PROTOBUF_NAMESPACE_ID::internal::CachedSize _cached_size_;
|
|
};
|
|
union { Impl_ _impl_; };
|
|
friend struct ::TableStruct_xla_2fxla_5fdata_2eproto;
|
|
};
|
|
// -------------------------------------------------------------------
|
|
|
|
class LiteralProto final :
|
|
public ::PROTOBUF_NAMESPACE_ID::Message /* @@protoc_insertion_point(class_definition:xla.LiteralProto) */ {
|
|
public:
|
|
inline LiteralProto() : LiteralProto(nullptr) {}
|
|
~LiteralProto() override;
|
|
explicit PROTOBUF_CONSTEXPR LiteralProto(::PROTOBUF_NAMESPACE_ID::internal::ConstantInitialized);
|
|
|
|
LiteralProto(const LiteralProto& from);
|
|
LiteralProto(LiteralProto&& from) noexcept
|
|
: LiteralProto() {
|
|
*this = ::std::move(from);
|
|
}
|
|
|
|
inline LiteralProto& operator=(const LiteralProto& from) {
|
|
CopyFrom(from);
|
|
return *this;
|
|
}
|
|
inline LiteralProto& operator=(LiteralProto&& from) noexcept {
|
|
if (this == &from) return *this;
|
|
if (GetOwningArena() == from.GetOwningArena()
|
|
#ifdef PROTOBUF_FORCE_COPY_IN_MOVE
|
|
&& GetOwningArena() != nullptr
|
|
#endif // !PROTOBUF_FORCE_COPY_IN_MOVE
|
|
) {
|
|
InternalSwap(&from);
|
|
} else {
|
|
CopyFrom(from);
|
|
}
|
|
return *this;
|
|
}
|
|
|
|
static const ::PROTOBUF_NAMESPACE_ID::Descriptor* descriptor() {
|
|
return GetDescriptor();
|
|
}
|
|
static const ::PROTOBUF_NAMESPACE_ID::Descriptor* GetDescriptor() {
|
|
return default_instance().GetMetadata().descriptor;
|
|
}
|
|
static const ::PROTOBUF_NAMESPACE_ID::Reflection* GetReflection() {
|
|
return default_instance().GetMetadata().reflection;
|
|
}
|
|
static const LiteralProto& default_instance() {
|
|
return *internal_default_instance();
|
|
}
|
|
static inline const LiteralProto* internal_default_instance() {
|
|
return reinterpret_cast<const LiteralProto*>(
|
|
&_LiteralProto_default_instance_);
|
|
}
|
|
static constexpr int kIndexInFileMessages =
|
|
16;
|
|
|
|
friend void swap(LiteralProto& a, LiteralProto& b) {
|
|
a.Swap(&b);
|
|
}
|
|
inline void Swap(LiteralProto* other) {
|
|
if (other == this) return;
|
|
#ifdef PROTOBUF_FORCE_COPY_IN_SWAP
|
|
if (GetOwningArena() != nullptr &&
|
|
GetOwningArena() == other->GetOwningArena()) {
|
|
#else // PROTOBUF_FORCE_COPY_IN_SWAP
|
|
if (GetOwningArena() == other->GetOwningArena()) {
|
|
#endif // !PROTOBUF_FORCE_COPY_IN_SWAP
|
|
InternalSwap(other);
|
|
} else {
|
|
::PROTOBUF_NAMESPACE_ID::internal::GenericSwap(this, other);
|
|
}
|
|
}
|
|
void UnsafeArenaSwap(LiteralProto* other) {
|
|
if (other == this) return;
|
|
GOOGLE_DCHECK(GetOwningArena() == other->GetOwningArena());
|
|
InternalSwap(other);
|
|
}
|
|
|
|
// implements Message ----------------------------------------------
|
|
|
|
LiteralProto* New(::PROTOBUF_NAMESPACE_ID::Arena* arena = nullptr) const final {
|
|
return CreateMaybeMessage<LiteralProto>(arena);
|
|
}
|
|
using ::PROTOBUF_NAMESPACE_ID::Message::CopyFrom;
|
|
void CopyFrom(const LiteralProto& from);
|
|
using ::PROTOBUF_NAMESPACE_ID::Message::MergeFrom;
|
|
void MergeFrom( const LiteralProto& from) {
|
|
LiteralProto::MergeImpl(*this, from);
|
|
}
|
|
private:
|
|
static void MergeImpl(::PROTOBUF_NAMESPACE_ID::Message& to_msg, const ::PROTOBUF_NAMESPACE_ID::Message& from_msg);
|
|
public:
|
|
PROTOBUF_ATTRIBUTE_REINITIALIZES void Clear() final;
|
|
bool IsInitialized() const final;
|
|
|
|
size_t ByteSizeLong() const final;
|
|
const char* _InternalParse(const char* ptr, ::PROTOBUF_NAMESPACE_ID::internal::ParseContext* ctx) final;
|
|
uint8_t* _InternalSerialize(
|
|
uint8_t* target, ::PROTOBUF_NAMESPACE_ID::io::EpsCopyOutputStream* stream) const final;
|
|
int GetCachedSize() const final { return _impl_._cached_size_.Get(); }
|
|
|
|
private:
|
|
void SharedCtor(::PROTOBUF_NAMESPACE_ID::Arena* arena, bool is_message_owned);
|
|
void SharedDtor();
|
|
void SetCachedSize(int size) const final;
|
|
void InternalSwap(LiteralProto* other);
|
|
|
|
private:
|
|
friend class ::PROTOBUF_NAMESPACE_ID::internal::AnyMetadata;
|
|
static ::PROTOBUF_NAMESPACE_ID::StringPiece FullMessageName() {
|
|
return "xla.LiteralProto";
|
|
}
|
|
protected:
|
|
explicit LiteralProto(::PROTOBUF_NAMESPACE_ID::Arena* arena,
|
|
bool is_message_owned = false);
|
|
public:
|
|
|
|
static const ClassData _class_data_;
|
|
const ::PROTOBUF_NAMESPACE_ID::Message::ClassData*GetClassData() const final;
|
|
|
|
::PROTOBUF_NAMESPACE_ID::Metadata GetMetadata() const final;
|
|
|
|
// nested types ----------------------------------------------------
|
|
|
|
// accessors -------------------------------------------------------
|
|
|
|
enum : int {
|
|
kPredsFieldNumber = 2,
|
|
kS32SFieldNumber = 4,
|
|
kS64SFieldNumber = 5,
|
|
kU32SFieldNumber = 6,
|
|
kU64SFieldNumber = 7,
|
|
kF32SFieldNumber = 8,
|
|
kF64SFieldNumber = 9,
|
|
kTupleLiteralsFieldNumber = 10,
|
|
kC64SFieldNumber = 12,
|
|
kSparseIndicesFieldNumber = 14,
|
|
kC128SFieldNumber = 18,
|
|
kU8SFieldNumber = 3,
|
|
kF16SFieldNumber = 11,
|
|
kBf16SFieldNumber = 13,
|
|
kS8SFieldNumber = 15,
|
|
kU16SFieldNumber = 16,
|
|
kS16SFieldNumber = 17,
|
|
kF8E5M2SFieldNumber = 19,
|
|
kF8E4M3FnsFieldNumber = 20,
|
|
kS4SFieldNumber = 21,
|
|
kU4SFieldNumber = 22,
|
|
kF8E4M3B11FnuzsFieldNumber = 23,
|
|
kF8E5M2FnuzsFieldNumber = 24,
|
|
kF8E4M3FnuzsFieldNumber = 25,
|
|
kShapeFieldNumber = 1,
|
|
};
|
|
// repeated bool preds = 2;
|
|
int preds_size() const;
|
|
private:
|
|
int _internal_preds_size() const;
|
|
public:
|
|
void clear_preds();
|
|
private:
|
|
bool _internal_preds(int index) const;
|
|
const ::PROTOBUF_NAMESPACE_ID::RepeatedField< bool >&
|
|
_internal_preds() const;
|
|
void _internal_add_preds(bool value);
|
|
::PROTOBUF_NAMESPACE_ID::RepeatedField< bool >*
|
|
_internal_mutable_preds();
|
|
public:
|
|
bool preds(int index) const;
|
|
void set_preds(int index, bool value);
|
|
void add_preds(bool value);
|
|
const ::PROTOBUF_NAMESPACE_ID::RepeatedField< bool >&
|
|
preds() const;
|
|
::PROTOBUF_NAMESPACE_ID::RepeatedField< bool >*
|
|
mutable_preds();
|
|
|
|
// repeated int32 s32s = 4;
|
|
int s32s_size() const;
|
|
private:
|
|
int _internal_s32s_size() const;
|
|
public:
|
|
void clear_s32s();
|
|
private:
|
|
int32_t _internal_s32s(int index) const;
|
|
const ::PROTOBUF_NAMESPACE_ID::RepeatedField< int32_t >&
|
|
_internal_s32s() const;
|
|
void _internal_add_s32s(int32_t value);
|
|
::PROTOBUF_NAMESPACE_ID::RepeatedField< int32_t >*
|
|
_internal_mutable_s32s();
|
|
public:
|
|
int32_t s32s(int index) const;
|
|
void set_s32s(int index, int32_t value);
|
|
void add_s32s(int32_t value);
|
|
const ::PROTOBUF_NAMESPACE_ID::RepeatedField< int32_t >&
|
|
s32s() const;
|
|
::PROTOBUF_NAMESPACE_ID::RepeatedField< int32_t >*
|
|
mutable_s32s();
|
|
|
|
// repeated int64 s64s = 5;
|
|
int s64s_size() const;
|
|
private:
|
|
int _internal_s64s_size() const;
|
|
public:
|
|
void clear_s64s();
|
|
private:
|
|
int64_t _internal_s64s(int index) const;
|
|
const ::PROTOBUF_NAMESPACE_ID::RepeatedField< int64_t >&
|
|
_internal_s64s() const;
|
|
void _internal_add_s64s(int64_t value);
|
|
::PROTOBUF_NAMESPACE_ID::RepeatedField< int64_t >*
|
|
_internal_mutable_s64s();
|
|
public:
|
|
int64_t s64s(int index) const;
|
|
void set_s64s(int index, int64_t value);
|
|
void add_s64s(int64_t value);
|
|
const ::PROTOBUF_NAMESPACE_ID::RepeatedField< int64_t >&
|
|
s64s() const;
|
|
::PROTOBUF_NAMESPACE_ID::RepeatedField< int64_t >*
|
|
mutable_s64s();
|
|
|
|
// repeated uint32 u32s = 6;
|
|
int u32s_size() const;
|
|
private:
|
|
int _internal_u32s_size() const;
|
|
public:
|
|
void clear_u32s();
|
|
private:
|
|
uint32_t _internal_u32s(int index) const;
|
|
const ::PROTOBUF_NAMESPACE_ID::RepeatedField< uint32_t >&
|
|
_internal_u32s() const;
|
|
void _internal_add_u32s(uint32_t value);
|
|
::PROTOBUF_NAMESPACE_ID::RepeatedField< uint32_t >*
|
|
_internal_mutable_u32s();
|
|
public:
|
|
uint32_t u32s(int index) const;
|
|
void set_u32s(int index, uint32_t value);
|
|
void add_u32s(uint32_t value);
|
|
const ::PROTOBUF_NAMESPACE_ID::RepeatedField< uint32_t >&
|
|
u32s() const;
|
|
::PROTOBUF_NAMESPACE_ID::RepeatedField< uint32_t >*
|
|
mutable_u32s();
|
|
|
|
// repeated uint64 u64s = 7;
|
|
int u64s_size() const;
|
|
private:
|
|
int _internal_u64s_size() const;
|
|
public:
|
|
void clear_u64s();
|
|
private:
|
|
uint64_t _internal_u64s(int index) const;
|
|
const ::PROTOBUF_NAMESPACE_ID::RepeatedField< uint64_t >&
|
|
_internal_u64s() const;
|
|
void _internal_add_u64s(uint64_t value);
|
|
::PROTOBUF_NAMESPACE_ID::RepeatedField< uint64_t >*
|
|
_internal_mutable_u64s();
|
|
public:
|
|
uint64_t u64s(int index) const;
|
|
void set_u64s(int index, uint64_t value);
|
|
void add_u64s(uint64_t value);
|
|
const ::PROTOBUF_NAMESPACE_ID::RepeatedField< uint64_t >&
|
|
u64s() const;
|
|
::PROTOBUF_NAMESPACE_ID::RepeatedField< uint64_t >*
|
|
mutable_u64s();
|
|
|
|
// repeated float f32s = 8;
|
|
int f32s_size() const;
|
|
private:
|
|
int _internal_f32s_size() const;
|
|
public:
|
|
void clear_f32s();
|
|
private:
|
|
float _internal_f32s(int index) const;
|
|
const ::PROTOBUF_NAMESPACE_ID::RepeatedField< float >&
|
|
_internal_f32s() const;
|
|
void _internal_add_f32s(float value);
|
|
::PROTOBUF_NAMESPACE_ID::RepeatedField< float >*
|
|
_internal_mutable_f32s();
|
|
public:
|
|
float f32s(int index) const;
|
|
void set_f32s(int index, float value);
|
|
void add_f32s(float value);
|
|
const ::PROTOBUF_NAMESPACE_ID::RepeatedField< float >&
|
|
f32s() const;
|
|
::PROTOBUF_NAMESPACE_ID::RepeatedField< float >*
|
|
mutable_f32s();
|
|
|
|
// repeated double f64s = 9;
|
|
int f64s_size() const;
|
|
private:
|
|
int _internal_f64s_size() const;
|
|
public:
|
|
void clear_f64s();
|
|
private:
|
|
double _internal_f64s(int index) const;
|
|
const ::PROTOBUF_NAMESPACE_ID::RepeatedField< double >&
|
|
_internal_f64s() const;
|
|
void _internal_add_f64s(double value);
|
|
::PROTOBUF_NAMESPACE_ID::RepeatedField< double >*
|
|
_internal_mutable_f64s();
|
|
public:
|
|
double f64s(int index) const;
|
|
void set_f64s(int index, double value);
|
|
void add_f64s(double value);
|
|
const ::PROTOBUF_NAMESPACE_ID::RepeatedField< double >&
|
|
f64s() const;
|
|
::PROTOBUF_NAMESPACE_ID::RepeatedField< double >*
|
|
mutable_f64s();
|
|
|
|
// repeated .xla.LiteralProto tuple_literals = 10;
|
|
int tuple_literals_size() const;
|
|
private:
|
|
int _internal_tuple_literals_size() const;
|
|
public:
|
|
void clear_tuple_literals();
|
|
::xla::LiteralProto* mutable_tuple_literals(int index);
|
|
::PROTOBUF_NAMESPACE_ID::RepeatedPtrField< ::xla::LiteralProto >*
|
|
mutable_tuple_literals();
|
|
private:
|
|
const ::xla::LiteralProto& _internal_tuple_literals(int index) const;
|
|
::xla::LiteralProto* _internal_add_tuple_literals();
|
|
public:
|
|
const ::xla::LiteralProto& tuple_literals(int index) const;
|
|
::xla::LiteralProto* add_tuple_literals();
|
|
const ::PROTOBUF_NAMESPACE_ID::RepeatedPtrField< ::xla::LiteralProto >&
|
|
tuple_literals() const;
|
|
|
|
// repeated float c64s = 12;
|
|
int c64s_size() const;
|
|
private:
|
|
int _internal_c64s_size() const;
|
|
public:
|
|
void clear_c64s();
|
|
private:
|
|
float _internal_c64s(int index) const;
|
|
const ::PROTOBUF_NAMESPACE_ID::RepeatedField< float >&
|
|
_internal_c64s() const;
|
|
void _internal_add_c64s(float value);
|
|
::PROTOBUF_NAMESPACE_ID::RepeatedField< float >*
|
|
_internal_mutable_c64s();
|
|
public:
|
|
float c64s(int index) const;
|
|
void set_c64s(int index, float value);
|
|
void add_c64s(float value);
|
|
const ::PROTOBUF_NAMESPACE_ID::RepeatedField< float >&
|
|
c64s() const;
|
|
::PROTOBUF_NAMESPACE_ID::RepeatedField< float >*
|
|
mutable_c64s();
|
|
|
|
// repeated int64 sparse_indices = 14;
|
|
int sparse_indices_size() const;
|
|
private:
|
|
int _internal_sparse_indices_size() const;
|
|
public:
|
|
void clear_sparse_indices();
|
|
private:
|
|
int64_t _internal_sparse_indices(int index) const;
|
|
const ::PROTOBUF_NAMESPACE_ID::RepeatedField< int64_t >&
|
|
_internal_sparse_indices() const;
|
|
void _internal_add_sparse_indices(int64_t value);
|
|
::PROTOBUF_NAMESPACE_ID::RepeatedField< int64_t >*
|
|
_internal_mutable_sparse_indices();
|
|
public:
|
|
int64_t sparse_indices(int index) const;
|
|
void set_sparse_indices(int index, int64_t value);
|
|
void add_sparse_indices(int64_t value);
|
|
const ::PROTOBUF_NAMESPACE_ID::RepeatedField< int64_t >&
|
|
sparse_indices() const;
|
|
::PROTOBUF_NAMESPACE_ID::RepeatedField< int64_t >*
|
|
mutable_sparse_indices();
|
|
|
|
// repeated double c128s = 18;
|
|
int c128s_size() const;
|
|
private:
|
|
int _internal_c128s_size() const;
|
|
public:
|
|
void clear_c128s();
|
|
private:
|
|
double _internal_c128s(int index) const;
|
|
const ::PROTOBUF_NAMESPACE_ID::RepeatedField< double >&
|
|
_internal_c128s() const;
|
|
void _internal_add_c128s(double value);
|
|
::PROTOBUF_NAMESPACE_ID::RepeatedField< double >*
|
|
_internal_mutable_c128s();
|
|
public:
|
|
double c128s(int index) const;
|
|
void set_c128s(int index, double value);
|
|
void add_c128s(double value);
|
|
const ::PROTOBUF_NAMESPACE_ID::RepeatedField< double >&
|
|
c128s() const;
|
|
::PROTOBUF_NAMESPACE_ID::RepeatedField< double >*
|
|
mutable_c128s();
|
|
|
|
// bytes u8s = 3;
|
|
void clear_u8s();
|
|
const std::string& u8s() const;
|
|
template <typename ArgT0 = const std::string&, typename... ArgT>
|
|
void set_u8s(ArgT0&& arg0, ArgT... args);
|
|
std::string* mutable_u8s();
|
|
PROTOBUF_NODISCARD std::string* release_u8s();
|
|
void set_allocated_u8s(std::string* u8s);
|
|
private:
|
|
const std::string& _internal_u8s() const;
|
|
inline PROTOBUF_ALWAYS_INLINE void _internal_set_u8s(const std::string& value);
|
|
std::string* _internal_mutable_u8s();
|
|
public:
|
|
|
|
// bytes f16s = 11;
|
|
void clear_f16s();
|
|
const std::string& f16s() const;
|
|
template <typename ArgT0 = const std::string&, typename... ArgT>
|
|
void set_f16s(ArgT0&& arg0, ArgT... args);
|
|
std::string* mutable_f16s();
|
|
PROTOBUF_NODISCARD std::string* release_f16s();
|
|
void set_allocated_f16s(std::string* f16s);
|
|
private:
|
|
const std::string& _internal_f16s() const;
|
|
inline PROTOBUF_ALWAYS_INLINE void _internal_set_f16s(const std::string& value);
|
|
std::string* _internal_mutable_f16s();
|
|
public:
|
|
|
|
// bytes bf16s = 13;
|
|
void clear_bf16s();
|
|
const std::string& bf16s() const;
|
|
template <typename ArgT0 = const std::string&, typename... ArgT>
|
|
void set_bf16s(ArgT0&& arg0, ArgT... args);
|
|
std::string* mutable_bf16s();
|
|
PROTOBUF_NODISCARD std::string* release_bf16s();
|
|
void set_allocated_bf16s(std::string* bf16s);
|
|
private:
|
|
const std::string& _internal_bf16s() const;
|
|
inline PROTOBUF_ALWAYS_INLINE void _internal_set_bf16s(const std::string& value);
|
|
std::string* _internal_mutable_bf16s();
|
|
public:
|
|
|
|
// bytes s8s = 15;
|
|
void clear_s8s();
|
|
const std::string& s8s() const;
|
|
template <typename ArgT0 = const std::string&, typename... ArgT>
|
|
void set_s8s(ArgT0&& arg0, ArgT... args);
|
|
std::string* mutable_s8s();
|
|
PROTOBUF_NODISCARD std::string* release_s8s();
|
|
void set_allocated_s8s(std::string* s8s);
|
|
private:
|
|
const std::string& _internal_s8s() const;
|
|
inline PROTOBUF_ALWAYS_INLINE void _internal_set_s8s(const std::string& value);
|
|
std::string* _internal_mutable_s8s();
|
|
public:
|
|
|
|
// bytes u16s = 16;
|
|
void clear_u16s();
|
|
const std::string& u16s() const;
|
|
template <typename ArgT0 = const std::string&, typename... ArgT>
|
|
void set_u16s(ArgT0&& arg0, ArgT... args);
|
|
std::string* mutable_u16s();
|
|
PROTOBUF_NODISCARD std::string* release_u16s();
|
|
void set_allocated_u16s(std::string* u16s);
|
|
private:
|
|
const std::string& _internal_u16s() const;
|
|
inline PROTOBUF_ALWAYS_INLINE void _internal_set_u16s(const std::string& value);
|
|
std::string* _internal_mutable_u16s();
|
|
public:
|
|
|
|
// bytes s16s = 17;
|
|
void clear_s16s();
|
|
const std::string& s16s() const;
|
|
template <typename ArgT0 = const std::string&, typename... ArgT>
|
|
void set_s16s(ArgT0&& arg0, ArgT... args);
|
|
std::string* mutable_s16s();
|
|
PROTOBUF_NODISCARD std::string* release_s16s();
|
|
void set_allocated_s16s(std::string* s16s);
|
|
private:
|
|
const std::string& _internal_s16s() const;
|
|
inline PROTOBUF_ALWAYS_INLINE void _internal_set_s16s(const std::string& value);
|
|
std::string* _internal_mutable_s16s();
|
|
public:
|
|
|
|
// bytes f8e5m2s = 19;
|
|
void clear_f8e5m2s();
|
|
const std::string& f8e5m2s() const;
|
|
template <typename ArgT0 = const std::string&, typename... ArgT>
|
|
void set_f8e5m2s(ArgT0&& arg0, ArgT... args);
|
|
std::string* mutable_f8e5m2s();
|
|
PROTOBUF_NODISCARD std::string* release_f8e5m2s();
|
|
void set_allocated_f8e5m2s(std::string* f8e5m2s);
|
|
private:
|
|
const std::string& _internal_f8e5m2s() const;
|
|
inline PROTOBUF_ALWAYS_INLINE void _internal_set_f8e5m2s(const std::string& value);
|
|
std::string* _internal_mutable_f8e5m2s();
|
|
public:
|
|
|
|
// bytes f8e4m3fns = 20;
|
|
void clear_f8e4m3fns();
|
|
const std::string& f8e4m3fns() const;
|
|
template <typename ArgT0 = const std::string&, typename... ArgT>
|
|
void set_f8e4m3fns(ArgT0&& arg0, ArgT... args);
|
|
std::string* mutable_f8e4m3fns();
|
|
PROTOBUF_NODISCARD std::string* release_f8e4m3fns();
|
|
void set_allocated_f8e4m3fns(std::string* f8e4m3fns);
|
|
private:
|
|
const std::string& _internal_f8e4m3fns() const;
|
|
inline PROTOBUF_ALWAYS_INLINE void _internal_set_f8e4m3fns(const std::string& value);
|
|
std::string* _internal_mutable_f8e4m3fns();
|
|
public:
|
|
|
|
// bytes s4s = 21;
|
|
void clear_s4s();
|
|
const std::string& s4s() const;
|
|
template <typename ArgT0 = const std::string&, typename... ArgT>
|
|
void set_s4s(ArgT0&& arg0, ArgT... args);
|
|
std::string* mutable_s4s();
|
|
PROTOBUF_NODISCARD std::string* release_s4s();
|
|
void set_allocated_s4s(std::string* s4s);
|
|
private:
|
|
const std::string& _internal_s4s() const;
|
|
inline PROTOBUF_ALWAYS_INLINE void _internal_set_s4s(const std::string& value);
|
|
std::string* _internal_mutable_s4s();
|
|
public:
|
|
|
|
// bytes u4s = 22;
|
|
void clear_u4s();
|
|
const std::string& u4s() const;
|
|
template <typename ArgT0 = const std::string&, typename... ArgT>
|
|
void set_u4s(ArgT0&& arg0, ArgT... args);
|
|
std::string* mutable_u4s();
|
|
PROTOBUF_NODISCARD std::string* release_u4s();
|
|
void set_allocated_u4s(std::string* u4s);
|
|
private:
|
|
const std::string& _internal_u4s() const;
|
|
inline PROTOBUF_ALWAYS_INLINE void _internal_set_u4s(const std::string& value);
|
|
std::string* _internal_mutable_u4s();
|
|
public:
|
|
|
|
// bytes f8e4m3b11fnuzs = 23;
|
|
void clear_f8e4m3b11fnuzs();
|
|
const std::string& f8e4m3b11fnuzs() const;
|
|
template <typename ArgT0 = const std::string&, typename... ArgT>
|
|
void set_f8e4m3b11fnuzs(ArgT0&& arg0, ArgT... args);
|
|
std::string* mutable_f8e4m3b11fnuzs();
|
|
PROTOBUF_NODISCARD std::string* release_f8e4m3b11fnuzs();
|
|
void set_allocated_f8e4m3b11fnuzs(std::string* f8e4m3b11fnuzs);
|
|
private:
|
|
const std::string& _internal_f8e4m3b11fnuzs() const;
|
|
inline PROTOBUF_ALWAYS_INLINE void _internal_set_f8e4m3b11fnuzs(const std::string& value);
|
|
std::string* _internal_mutable_f8e4m3b11fnuzs();
|
|
public:
|
|
|
|
// bytes f8e5m2fnuzs = 24;
|
|
void clear_f8e5m2fnuzs();
|
|
const std::string& f8e5m2fnuzs() const;
|
|
template <typename ArgT0 = const std::string&, typename... ArgT>
|
|
void set_f8e5m2fnuzs(ArgT0&& arg0, ArgT... args);
|
|
std::string* mutable_f8e5m2fnuzs();
|
|
PROTOBUF_NODISCARD std::string* release_f8e5m2fnuzs();
|
|
void set_allocated_f8e5m2fnuzs(std::string* f8e5m2fnuzs);
|
|
private:
|
|
const std::string& _internal_f8e5m2fnuzs() const;
|
|
inline PROTOBUF_ALWAYS_INLINE void _internal_set_f8e5m2fnuzs(const std::string& value);
|
|
std::string* _internal_mutable_f8e5m2fnuzs();
|
|
public:
|
|
|
|
// bytes f8e4m3fnuzs = 25;
|
|
void clear_f8e4m3fnuzs();
|
|
const std::string& f8e4m3fnuzs() const;
|
|
template <typename ArgT0 = const std::string&, typename... ArgT>
|
|
void set_f8e4m3fnuzs(ArgT0&& arg0, ArgT... args);
|
|
std::string* mutable_f8e4m3fnuzs();
|
|
PROTOBUF_NODISCARD std::string* release_f8e4m3fnuzs();
|
|
void set_allocated_f8e4m3fnuzs(std::string* f8e4m3fnuzs);
|
|
private:
|
|
const std::string& _internal_f8e4m3fnuzs() const;
|
|
inline PROTOBUF_ALWAYS_INLINE void _internal_set_f8e4m3fnuzs(const std::string& value);
|
|
std::string* _internal_mutable_f8e4m3fnuzs();
|
|
public:
|
|
|
|
// .xla.ShapeProto shape = 1;
|
|
bool has_shape() const;
|
|
private:
|
|
bool _internal_has_shape() const;
|
|
public:
|
|
void clear_shape();
|
|
const ::xla::ShapeProto& shape() const;
|
|
PROTOBUF_NODISCARD ::xla::ShapeProto* release_shape();
|
|
::xla::ShapeProto* mutable_shape();
|
|
void set_allocated_shape(::xla::ShapeProto* shape);
|
|
private:
|
|
const ::xla::ShapeProto& _internal_shape() const;
|
|
::xla::ShapeProto* _internal_mutable_shape();
|
|
public:
|
|
void unsafe_arena_set_allocated_shape(
|
|
::xla::ShapeProto* shape);
|
|
::xla::ShapeProto* unsafe_arena_release_shape();
|
|
|
|
// @@protoc_insertion_point(class_scope:xla.LiteralProto)
|
|
private:
|
|
class _Internal;
|
|
|
|
template <typename T> friend class ::PROTOBUF_NAMESPACE_ID::Arena::InternalHelper;
|
|
typedef void InternalArenaConstructable_;
|
|
typedef void DestructorSkippable_;
|
|
struct Impl_ {
|
|
::PROTOBUF_NAMESPACE_ID::RepeatedField< bool > preds_;
|
|
::PROTOBUF_NAMESPACE_ID::RepeatedField< int32_t > s32s_;
|
|
mutable std::atomic<int> _s32s_cached_byte_size_;
|
|
::PROTOBUF_NAMESPACE_ID::RepeatedField< int64_t > s64s_;
|
|
mutable std::atomic<int> _s64s_cached_byte_size_;
|
|
::PROTOBUF_NAMESPACE_ID::RepeatedField< uint32_t > u32s_;
|
|
mutable std::atomic<int> _u32s_cached_byte_size_;
|
|
::PROTOBUF_NAMESPACE_ID::RepeatedField< uint64_t > u64s_;
|
|
mutable std::atomic<int> _u64s_cached_byte_size_;
|
|
::PROTOBUF_NAMESPACE_ID::RepeatedField< float > f32s_;
|
|
::PROTOBUF_NAMESPACE_ID::RepeatedField< double > f64s_;
|
|
::PROTOBUF_NAMESPACE_ID::RepeatedPtrField< ::xla::LiteralProto > tuple_literals_;
|
|
::PROTOBUF_NAMESPACE_ID::RepeatedField< float > c64s_;
|
|
::PROTOBUF_NAMESPACE_ID::RepeatedField< int64_t > sparse_indices_;
|
|
mutable std::atomic<int> _sparse_indices_cached_byte_size_;
|
|
::PROTOBUF_NAMESPACE_ID::RepeatedField< double > c128s_;
|
|
::PROTOBUF_NAMESPACE_ID::internal::ArenaStringPtr u8s_;
|
|
::PROTOBUF_NAMESPACE_ID::internal::ArenaStringPtr f16s_;
|
|
::PROTOBUF_NAMESPACE_ID::internal::ArenaStringPtr bf16s_;
|
|
::PROTOBUF_NAMESPACE_ID::internal::ArenaStringPtr s8s_;
|
|
::PROTOBUF_NAMESPACE_ID::internal::ArenaStringPtr u16s_;
|
|
::PROTOBUF_NAMESPACE_ID::internal::ArenaStringPtr s16s_;
|
|
::PROTOBUF_NAMESPACE_ID::internal::ArenaStringPtr f8e5m2s_;
|
|
::PROTOBUF_NAMESPACE_ID::internal::ArenaStringPtr f8e4m3fns_;
|
|
::PROTOBUF_NAMESPACE_ID::internal::ArenaStringPtr s4s_;
|
|
::PROTOBUF_NAMESPACE_ID::internal::ArenaStringPtr u4s_;
|
|
::PROTOBUF_NAMESPACE_ID::internal::ArenaStringPtr f8e4m3b11fnuzs_;
|
|
::PROTOBUF_NAMESPACE_ID::internal::ArenaStringPtr f8e5m2fnuzs_;
|
|
::PROTOBUF_NAMESPACE_ID::internal::ArenaStringPtr f8e4m3fnuzs_;
|
|
::xla::ShapeProto* shape_;
|
|
mutable ::PROTOBUF_NAMESPACE_ID::internal::CachedSize _cached_size_;
|
|
};
|
|
union { Impl_ _impl_; };
|
|
friend struct ::TableStruct_xla_2fxla_5fdata_2eproto;
|
|
};
|
|
// -------------------------------------------------------------------
|
|
|
|
class WindowDimension final :
|
|
public ::PROTOBUF_NAMESPACE_ID::Message /* @@protoc_insertion_point(class_definition:xla.WindowDimension) */ {
|
|
public:
|
|
inline WindowDimension() : WindowDimension(nullptr) {}
|
|
~WindowDimension() override;
|
|
explicit PROTOBUF_CONSTEXPR WindowDimension(::PROTOBUF_NAMESPACE_ID::internal::ConstantInitialized);
|
|
|
|
WindowDimension(const WindowDimension& from);
|
|
WindowDimension(WindowDimension&& from) noexcept
|
|
: WindowDimension() {
|
|
*this = ::std::move(from);
|
|
}
|
|
|
|
inline WindowDimension& operator=(const WindowDimension& from) {
|
|
CopyFrom(from);
|
|
return *this;
|
|
}
|
|
inline WindowDimension& operator=(WindowDimension&& from) noexcept {
|
|
if (this == &from) return *this;
|
|
if (GetOwningArena() == from.GetOwningArena()
|
|
#ifdef PROTOBUF_FORCE_COPY_IN_MOVE
|
|
&& GetOwningArena() != nullptr
|
|
#endif // !PROTOBUF_FORCE_COPY_IN_MOVE
|
|
) {
|
|
InternalSwap(&from);
|
|
} else {
|
|
CopyFrom(from);
|
|
}
|
|
return *this;
|
|
}
|
|
|
|
static const ::PROTOBUF_NAMESPACE_ID::Descriptor* descriptor() {
|
|
return GetDescriptor();
|
|
}
|
|
static const ::PROTOBUF_NAMESPACE_ID::Descriptor* GetDescriptor() {
|
|
return default_instance().GetMetadata().descriptor;
|
|
}
|
|
static const ::PROTOBUF_NAMESPACE_ID::Reflection* GetReflection() {
|
|
return default_instance().GetMetadata().reflection;
|
|
}
|
|
static const WindowDimension& default_instance() {
|
|
return *internal_default_instance();
|
|
}
|
|
static inline const WindowDimension* internal_default_instance() {
|
|
return reinterpret_cast<const WindowDimension*>(
|
|
&_WindowDimension_default_instance_);
|
|
}
|
|
static constexpr int kIndexInFileMessages =
|
|
17;
|
|
|
|
friend void swap(WindowDimension& a, WindowDimension& b) {
|
|
a.Swap(&b);
|
|
}
|
|
inline void Swap(WindowDimension* other) {
|
|
if (other == this) return;
|
|
#ifdef PROTOBUF_FORCE_COPY_IN_SWAP
|
|
if (GetOwningArena() != nullptr &&
|
|
GetOwningArena() == other->GetOwningArena()) {
|
|
#else // PROTOBUF_FORCE_COPY_IN_SWAP
|
|
if (GetOwningArena() == other->GetOwningArena()) {
|
|
#endif // !PROTOBUF_FORCE_COPY_IN_SWAP
|
|
InternalSwap(other);
|
|
} else {
|
|
::PROTOBUF_NAMESPACE_ID::internal::GenericSwap(this, other);
|
|
}
|
|
}
|
|
void UnsafeArenaSwap(WindowDimension* other) {
|
|
if (other == this) return;
|
|
GOOGLE_DCHECK(GetOwningArena() == other->GetOwningArena());
|
|
InternalSwap(other);
|
|
}
|
|
|
|
// implements Message ----------------------------------------------
|
|
|
|
WindowDimension* New(::PROTOBUF_NAMESPACE_ID::Arena* arena = nullptr) const final {
|
|
return CreateMaybeMessage<WindowDimension>(arena);
|
|
}
|
|
using ::PROTOBUF_NAMESPACE_ID::Message::CopyFrom;
|
|
void CopyFrom(const WindowDimension& from);
|
|
using ::PROTOBUF_NAMESPACE_ID::Message::MergeFrom;
|
|
void MergeFrom( const WindowDimension& from) {
|
|
WindowDimension::MergeImpl(*this, from);
|
|
}
|
|
private:
|
|
static void MergeImpl(::PROTOBUF_NAMESPACE_ID::Message& to_msg, const ::PROTOBUF_NAMESPACE_ID::Message& from_msg);
|
|
public:
|
|
PROTOBUF_ATTRIBUTE_REINITIALIZES void Clear() final;
|
|
bool IsInitialized() const final;
|
|
|
|
size_t ByteSizeLong() const final;
|
|
const char* _InternalParse(const char* ptr, ::PROTOBUF_NAMESPACE_ID::internal::ParseContext* ctx) final;
|
|
uint8_t* _InternalSerialize(
|
|
uint8_t* target, ::PROTOBUF_NAMESPACE_ID::io::EpsCopyOutputStream* stream) const final;
|
|
int GetCachedSize() const final { return _impl_._cached_size_.Get(); }
|
|
|
|
private:
|
|
void SharedCtor(::PROTOBUF_NAMESPACE_ID::Arena* arena, bool is_message_owned);
|
|
void SharedDtor();
|
|
void SetCachedSize(int size) const final;
|
|
void InternalSwap(WindowDimension* other);
|
|
|
|
private:
|
|
friend class ::PROTOBUF_NAMESPACE_ID::internal::AnyMetadata;
|
|
static ::PROTOBUF_NAMESPACE_ID::StringPiece FullMessageName() {
|
|
return "xla.WindowDimension";
|
|
}
|
|
protected:
|
|
explicit WindowDimension(::PROTOBUF_NAMESPACE_ID::Arena* arena,
|
|
bool is_message_owned = false);
|
|
public:
|
|
|
|
static const ClassData _class_data_;
|
|
const ::PROTOBUF_NAMESPACE_ID::Message::ClassData*GetClassData() const final;
|
|
|
|
::PROTOBUF_NAMESPACE_ID::Metadata GetMetadata() const final;
|
|
|
|
// nested types ----------------------------------------------------
|
|
|
|
// accessors -------------------------------------------------------
|
|
|
|
enum : int {
|
|
kSizeFieldNumber = 1,
|
|
kStrideFieldNumber = 2,
|
|
kPaddingLowFieldNumber = 3,
|
|
kPaddingHighFieldNumber = 4,
|
|
kWindowDilationFieldNumber = 5,
|
|
kBaseDilationFieldNumber = 6,
|
|
kWindowReversalFieldNumber = 7,
|
|
};
|
|
// int64 size = 1;
|
|
void clear_size();
|
|
int64_t size() const;
|
|
void set_size(int64_t value);
|
|
private:
|
|
int64_t _internal_size() const;
|
|
void _internal_set_size(int64_t value);
|
|
public:
|
|
|
|
// int64 stride = 2;
|
|
void clear_stride();
|
|
int64_t stride() const;
|
|
void set_stride(int64_t value);
|
|
private:
|
|
int64_t _internal_stride() const;
|
|
void _internal_set_stride(int64_t value);
|
|
public:
|
|
|
|
// int64 padding_low = 3;
|
|
void clear_padding_low();
|
|
int64_t padding_low() const;
|
|
void set_padding_low(int64_t value);
|
|
private:
|
|
int64_t _internal_padding_low() const;
|
|
void _internal_set_padding_low(int64_t value);
|
|
public:
|
|
|
|
// int64 padding_high = 4;
|
|
void clear_padding_high();
|
|
int64_t padding_high() const;
|
|
void set_padding_high(int64_t value);
|
|
private:
|
|
int64_t _internal_padding_high() const;
|
|
void _internal_set_padding_high(int64_t value);
|
|
public:
|
|
|
|
// int64 window_dilation = 5;
|
|
void clear_window_dilation();
|
|
int64_t window_dilation() const;
|
|
void set_window_dilation(int64_t value);
|
|
private:
|
|
int64_t _internal_window_dilation() const;
|
|
void _internal_set_window_dilation(int64_t value);
|
|
public:
|
|
|
|
// int64 base_dilation = 6;
|
|
void clear_base_dilation();
|
|
int64_t base_dilation() const;
|
|
void set_base_dilation(int64_t value);
|
|
private:
|
|
int64_t _internal_base_dilation() const;
|
|
void _internal_set_base_dilation(int64_t value);
|
|
public:
|
|
|
|
// bool window_reversal = 7;
|
|
void clear_window_reversal();
|
|
bool window_reversal() const;
|
|
void set_window_reversal(bool value);
|
|
private:
|
|
bool _internal_window_reversal() const;
|
|
void _internal_set_window_reversal(bool value);
|
|
public:
|
|
|
|
// @@protoc_insertion_point(class_scope:xla.WindowDimension)
|
|
private:
|
|
class _Internal;
|
|
|
|
template <typename T> friend class ::PROTOBUF_NAMESPACE_ID::Arena::InternalHelper;
|
|
typedef void InternalArenaConstructable_;
|
|
typedef void DestructorSkippable_;
|
|
struct Impl_ {
|
|
int64_t size_;
|
|
int64_t stride_;
|
|
int64_t padding_low_;
|
|
int64_t padding_high_;
|
|
int64_t window_dilation_;
|
|
int64_t base_dilation_;
|
|
bool window_reversal_;
|
|
mutable ::PROTOBUF_NAMESPACE_ID::internal::CachedSize _cached_size_;
|
|
};
|
|
union { Impl_ _impl_; };
|
|
friend struct ::TableStruct_xla_2fxla_5fdata_2eproto;
|
|
};
|
|
// -------------------------------------------------------------------
|
|
|
|
class Window final :
|
|
public ::PROTOBUF_NAMESPACE_ID::Message /* @@protoc_insertion_point(class_definition:xla.Window) */ {
|
|
public:
|
|
inline Window() : Window(nullptr) {}
|
|
~Window() override;
|
|
explicit PROTOBUF_CONSTEXPR Window(::PROTOBUF_NAMESPACE_ID::internal::ConstantInitialized);
|
|
|
|
Window(const Window& from);
|
|
Window(Window&& from) noexcept
|
|
: Window() {
|
|
*this = ::std::move(from);
|
|
}
|
|
|
|
inline Window& operator=(const Window& from) {
|
|
CopyFrom(from);
|
|
return *this;
|
|
}
|
|
inline Window& operator=(Window&& from) noexcept {
|
|
if (this == &from) return *this;
|
|
if (GetOwningArena() == from.GetOwningArena()
|
|
#ifdef PROTOBUF_FORCE_COPY_IN_MOVE
|
|
&& GetOwningArena() != nullptr
|
|
#endif // !PROTOBUF_FORCE_COPY_IN_MOVE
|
|
) {
|
|
InternalSwap(&from);
|
|
} else {
|
|
CopyFrom(from);
|
|
}
|
|
return *this;
|
|
}
|
|
|
|
static const ::PROTOBUF_NAMESPACE_ID::Descriptor* descriptor() {
|
|
return GetDescriptor();
|
|
}
|
|
static const ::PROTOBUF_NAMESPACE_ID::Descriptor* GetDescriptor() {
|
|
return default_instance().GetMetadata().descriptor;
|
|
}
|
|
static const ::PROTOBUF_NAMESPACE_ID::Reflection* GetReflection() {
|
|
return default_instance().GetMetadata().reflection;
|
|
}
|
|
static const Window& default_instance() {
|
|
return *internal_default_instance();
|
|
}
|
|
static inline const Window* internal_default_instance() {
|
|
return reinterpret_cast<const Window*>(
|
|
&_Window_default_instance_);
|
|
}
|
|
static constexpr int kIndexInFileMessages =
|
|
18;
|
|
|
|
friend void swap(Window& a, Window& b) {
|
|
a.Swap(&b);
|
|
}
|
|
inline void Swap(Window* other) {
|
|
if (other == this) return;
|
|
#ifdef PROTOBUF_FORCE_COPY_IN_SWAP
|
|
if (GetOwningArena() != nullptr &&
|
|
GetOwningArena() == other->GetOwningArena()) {
|
|
#else // PROTOBUF_FORCE_COPY_IN_SWAP
|
|
if (GetOwningArena() == other->GetOwningArena()) {
|
|
#endif // !PROTOBUF_FORCE_COPY_IN_SWAP
|
|
InternalSwap(other);
|
|
} else {
|
|
::PROTOBUF_NAMESPACE_ID::internal::GenericSwap(this, other);
|
|
}
|
|
}
|
|
void UnsafeArenaSwap(Window* other) {
|
|
if (other == this) return;
|
|
GOOGLE_DCHECK(GetOwningArena() == other->GetOwningArena());
|
|
InternalSwap(other);
|
|
}
|
|
|
|
// implements Message ----------------------------------------------
|
|
|
|
Window* New(::PROTOBUF_NAMESPACE_ID::Arena* arena = nullptr) const final {
|
|
return CreateMaybeMessage<Window>(arena);
|
|
}
|
|
using ::PROTOBUF_NAMESPACE_ID::Message::CopyFrom;
|
|
void CopyFrom(const Window& from);
|
|
using ::PROTOBUF_NAMESPACE_ID::Message::MergeFrom;
|
|
void MergeFrom( const Window& from) {
|
|
Window::MergeImpl(*this, from);
|
|
}
|
|
private:
|
|
static void MergeImpl(::PROTOBUF_NAMESPACE_ID::Message& to_msg, const ::PROTOBUF_NAMESPACE_ID::Message& from_msg);
|
|
public:
|
|
PROTOBUF_ATTRIBUTE_REINITIALIZES void Clear() final;
|
|
bool IsInitialized() const final;
|
|
|
|
size_t ByteSizeLong() const final;
|
|
const char* _InternalParse(const char* ptr, ::PROTOBUF_NAMESPACE_ID::internal::ParseContext* ctx) final;
|
|
uint8_t* _InternalSerialize(
|
|
uint8_t* target, ::PROTOBUF_NAMESPACE_ID::io::EpsCopyOutputStream* stream) const final;
|
|
int GetCachedSize() const final { return _impl_._cached_size_.Get(); }
|
|
|
|
private:
|
|
void SharedCtor(::PROTOBUF_NAMESPACE_ID::Arena* arena, bool is_message_owned);
|
|
void SharedDtor();
|
|
void SetCachedSize(int size) const final;
|
|
void InternalSwap(Window* other);
|
|
|
|
private:
|
|
friend class ::PROTOBUF_NAMESPACE_ID::internal::AnyMetadata;
|
|
static ::PROTOBUF_NAMESPACE_ID::StringPiece FullMessageName() {
|
|
return "xla.Window";
|
|
}
|
|
protected:
|
|
explicit Window(::PROTOBUF_NAMESPACE_ID::Arena* arena,
|
|
bool is_message_owned = false);
|
|
public:
|
|
|
|
static const ClassData _class_data_;
|
|
const ::PROTOBUF_NAMESPACE_ID::Message::ClassData*GetClassData() const final;
|
|
|
|
::PROTOBUF_NAMESPACE_ID::Metadata GetMetadata() const final;
|
|
|
|
// nested types ----------------------------------------------------
|
|
|
|
// accessors -------------------------------------------------------
|
|
|
|
enum : int {
|
|
kDimensionsFieldNumber = 1,
|
|
};
|
|
// repeated .xla.WindowDimension dimensions = 1;
|
|
int dimensions_size() const;
|
|
private:
|
|
int _internal_dimensions_size() const;
|
|
public:
|
|
void clear_dimensions();
|
|
::xla::WindowDimension* mutable_dimensions(int index);
|
|
::PROTOBUF_NAMESPACE_ID::RepeatedPtrField< ::xla::WindowDimension >*
|
|
mutable_dimensions();
|
|
private:
|
|
const ::xla::WindowDimension& _internal_dimensions(int index) const;
|
|
::xla::WindowDimension* _internal_add_dimensions();
|
|
public:
|
|
const ::xla::WindowDimension& dimensions(int index) const;
|
|
::xla::WindowDimension* add_dimensions();
|
|
const ::PROTOBUF_NAMESPACE_ID::RepeatedPtrField< ::xla::WindowDimension >&
|
|
dimensions() const;
|
|
|
|
// @@protoc_insertion_point(class_scope:xla.Window)
|
|
private:
|
|
class _Internal;
|
|
|
|
template <typename T> friend class ::PROTOBUF_NAMESPACE_ID::Arena::InternalHelper;
|
|
typedef void InternalArenaConstructable_;
|
|
typedef void DestructorSkippable_;
|
|
struct Impl_ {
|
|
::PROTOBUF_NAMESPACE_ID::RepeatedPtrField< ::xla::WindowDimension > dimensions_;
|
|
mutable ::PROTOBUF_NAMESPACE_ID::internal::CachedSize _cached_size_;
|
|
};
|
|
union { Impl_ _impl_; };
|
|
friend struct ::TableStruct_xla_2fxla_5fdata_2eproto;
|
|
};
|
|
// -------------------------------------------------------------------
|
|
|
|
class GatherDimensionNumbers final :
|
|
public ::PROTOBUF_NAMESPACE_ID::Message /* @@protoc_insertion_point(class_definition:xla.GatherDimensionNumbers) */ {
|
|
public:
|
|
inline GatherDimensionNumbers() : GatherDimensionNumbers(nullptr) {}
|
|
~GatherDimensionNumbers() override;
|
|
explicit PROTOBUF_CONSTEXPR GatherDimensionNumbers(::PROTOBUF_NAMESPACE_ID::internal::ConstantInitialized);
|
|
|
|
GatherDimensionNumbers(const GatherDimensionNumbers& from);
|
|
GatherDimensionNumbers(GatherDimensionNumbers&& from) noexcept
|
|
: GatherDimensionNumbers() {
|
|
*this = ::std::move(from);
|
|
}
|
|
|
|
inline GatherDimensionNumbers& operator=(const GatherDimensionNumbers& from) {
|
|
CopyFrom(from);
|
|
return *this;
|
|
}
|
|
inline GatherDimensionNumbers& operator=(GatherDimensionNumbers&& from) noexcept {
|
|
if (this == &from) return *this;
|
|
if (GetOwningArena() == from.GetOwningArena()
|
|
#ifdef PROTOBUF_FORCE_COPY_IN_MOVE
|
|
&& GetOwningArena() != nullptr
|
|
#endif // !PROTOBUF_FORCE_COPY_IN_MOVE
|
|
) {
|
|
InternalSwap(&from);
|
|
} else {
|
|
CopyFrom(from);
|
|
}
|
|
return *this;
|
|
}
|
|
|
|
static const ::PROTOBUF_NAMESPACE_ID::Descriptor* descriptor() {
|
|
return GetDescriptor();
|
|
}
|
|
static const ::PROTOBUF_NAMESPACE_ID::Descriptor* GetDescriptor() {
|
|
return default_instance().GetMetadata().descriptor;
|
|
}
|
|
static const ::PROTOBUF_NAMESPACE_ID::Reflection* GetReflection() {
|
|
return default_instance().GetMetadata().reflection;
|
|
}
|
|
static const GatherDimensionNumbers& default_instance() {
|
|
return *internal_default_instance();
|
|
}
|
|
static inline const GatherDimensionNumbers* internal_default_instance() {
|
|
return reinterpret_cast<const GatherDimensionNumbers*>(
|
|
&_GatherDimensionNumbers_default_instance_);
|
|
}
|
|
static constexpr int kIndexInFileMessages =
|
|
19;
|
|
|
|
friend void swap(GatherDimensionNumbers& a, GatherDimensionNumbers& b) {
|
|
a.Swap(&b);
|
|
}
|
|
inline void Swap(GatherDimensionNumbers* other) {
|
|
if (other == this) return;
|
|
#ifdef PROTOBUF_FORCE_COPY_IN_SWAP
|
|
if (GetOwningArena() != nullptr &&
|
|
GetOwningArena() == other->GetOwningArena()) {
|
|
#else // PROTOBUF_FORCE_COPY_IN_SWAP
|
|
if (GetOwningArena() == other->GetOwningArena()) {
|
|
#endif // !PROTOBUF_FORCE_COPY_IN_SWAP
|
|
InternalSwap(other);
|
|
} else {
|
|
::PROTOBUF_NAMESPACE_ID::internal::GenericSwap(this, other);
|
|
}
|
|
}
|
|
void UnsafeArenaSwap(GatherDimensionNumbers* other) {
|
|
if (other == this) return;
|
|
GOOGLE_DCHECK(GetOwningArena() == other->GetOwningArena());
|
|
InternalSwap(other);
|
|
}
|
|
|
|
// implements Message ----------------------------------------------
|
|
|
|
GatherDimensionNumbers* New(::PROTOBUF_NAMESPACE_ID::Arena* arena = nullptr) const final {
|
|
return CreateMaybeMessage<GatherDimensionNumbers>(arena);
|
|
}
|
|
using ::PROTOBUF_NAMESPACE_ID::Message::CopyFrom;
|
|
void CopyFrom(const GatherDimensionNumbers& from);
|
|
using ::PROTOBUF_NAMESPACE_ID::Message::MergeFrom;
|
|
void MergeFrom( const GatherDimensionNumbers& from) {
|
|
GatherDimensionNumbers::MergeImpl(*this, from);
|
|
}
|
|
private:
|
|
static void MergeImpl(::PROTOBUF_NAMESPACE_ID::Message& to_msg, const ::PROTOBUF_NAMESPACE_ID::Message& from_msg);
|
|
public:
|
|
PROTOBUF_ATTRIBUTE_REINITIALIZES void Clear() final;
|
|
bool IsInitialized() const final;
|
|
|
|
size_t ByteSizeLong() const final;
|
|
const char* _InternalParse(const char* ptr, ::PROTOBUF_NAMESPACE_ID::internal::ParseContext* ctx) final;
|
|
uint8_t* _InternalSerialize(
|
|
uint8_t* target, ::PROTOBUF_NAMESPACE_ID::io::EpsCopyOutputStream* stream) const final;
|
|
int GetCachedSize() const final { return _impl_._cached_size_.Get(); }
|
|
|
|
private:
|
|
void SharedCtor(::PROTOBUF_NAMESPACE_ID::Arena* arena, bool is_message_owned);
|
|
void SharedDtor();
|
|
void SetCachedSize(int size) const final;
|
|
void InternalSwap(GatherDimensionNumbers* other);
|
|
|
|
private:
|
|
friend class ::PROTOBUF_NAMESPACE_ID::internal::AnyMetadata;
|
|
static ::PROTOBUF_NAMESPACE_ID::StringPiece FullMessageName() {
|
|
return "xla.GatherDimensionNumbers";
|
|
}
|
|
protected:
|
|
explicit GatherDimensionNumbers(::PROTOBUF_NAMESPACE_ID::Arena* arena,
|
|
bool is_message_owned = false);
|
|
public:
|
|
|
|
static const ClassData _class_data_;
|
|
const ::PROTOBUF_NAMESPACE_ID::Message::ClassData*GetClassData() const final;
|
|
|
|
::PROTOBUF_NAMESPACE_ID::Metadata GetMetadata() const final;
|
|
|
|
// nested types ----------------------------------------------------
|
|
|
|
// accessors -------------------------------------------------------
|
|
|
|
enum : int {
|
|
kOffsetDimsFieldNumber = 1,
|
|
kCollapsedSliceDimsFieldNumber = 2,
|
|
kStartIndexMapFieldNumber = 3,
|
|
kIndexVectorDimFieldNumber = 4,
|
|
};
|
|
// repeated int64 offset_dims = 1;
|
|
int offset_dims_size() const;
|
|
private:
|
|
int _internal_offset_dims_size() const;
|
|
public:
|
|
void clear_offset_dims();
|
|
private:
|
|
int64_t _internal_offset_dims(int index) const;
|
|
const ::PROTOBUF_NAMESPACE_ID::RepeatedField< int64_t >&
|
|
_internal_offset_dims() const;
|
|
void _internal_add_offset_dims(int64_t value);
|
|
::PROTOBUF_NAMESPACE_ID::RepeatedField< int64_t >*
|
|
_internal_mutable_offset_dims();
|
|
public:
|
|
int64_t offset_dims(int index) const;
|
|
void set_offset_dims(int index, int64_t value);
|
|
void add_offset_dims(int64_t value);
|
|
const ::PROTOBUF_NAMESPACE_ID::RepeatedField< int64_t >&
|
|
offset_dims() const;
|
|
::PROTOBUF_NAMESPACE_ID::RepeatedField< int64_t >*
|
|
mutable_offset_dims();
|
|
|
|
// repeated int64 collapsed_slice_dims = 2;
|
|
int collapsed_slice_dims_size() const;
|
|
private:
|
|
int _internal_collapsed_slice_dims_size() const;
|
|
public:
|
|
void clear_collapsed_slice_dims();
|
|
private:
|
|
int64_t _internal_collapsed_slice_dims(int index) const;
|
|
const ::PROTOBUF_NAMESPACE_ID::RepeatedField< int64_t >&
|
|
_internal_collapsed_slice_dims() const;
|
|
void _internal_add_collapsed_slice_dims(int64_t value);
|
|
::PROTOBUF_NAMESPACE_ID::RepeatedField< int64_t >*
|
|
_internal_mutable_collapsed_slice_dims();
|
|
public:
|
|
int64_t collapsed_slice_dims(int index) const;
|
|
void set_collapsed_slice_dims(int index, int64_t value);
|
|
void add_collapsed_slice_dims(int64_t value);
|
|
const ::PROTOBUF_NAMESPACE_ID::RepeatedField< int64_t >&
|
|
collapsed_slice_dims() const;
|
|
::PROTOBUF_NAMESPACE_ID::RepeatedField< int64_t >*
|
|
mutable_collapsed_slice_dims();
|
|
|
|
// repeated int64 start_index_map = 3;
|
|
int start_index_map_size() const;
|
|
private:
|
|
int _internal_start_index_map_size() const;
|
|
public:
|
|
void clear_start_index_map();
|
|
private:
|
|
int64_t _internal_start_index_map(int index) const;
|
|
const ::PROTOBUF_NAMESPACE_ID::RepeatedField< int64_t >&
|
|
_internal_start_index_map() const;
|
|
void _internal_add_start_index_map(int64_t value);
|
|
::PROTOBUF_NAMESPACE_ID::RepeatedField< int64_t >*
|
|
_internal_mutable_start_index_map();
|
|
public:
|
|
int64_t start_index_map(int index) const;
|
|
void set_start_index_map(int index, int64_t value);
|
|
void add_start_index_map(int64_t value);
|
|
const ::PROTOBUF_NAMESPACE_ID::RepeatedField< int64_t >&
|
|
start_index_map() const;
|
|
::PROTOBUF_NAMESPACE_ID::RepeatedField< int64_t >*
|
|
mutable_start_index_map();
|
|
|
|
// int64 index_vector_dim = 4;
|
|
void clear_index_vector_dim();
|
|
int64_t index_vector_dim() const;
|
|
void set_index_vector_dim(int64_t value);
|
|
private:
|
|
int64_t _internal_index_vector_dim() const;
|
|
void _internal_set_index_vector_dim(int64_t value);
|
|
public:
|
|
|
|
// @@protoc_insertion_point(class_scope:xla.GatherDimensionNumbers)
|
|
private:
|
|
class _Internal;
|
|
|
|
template <typename T> friend class ::PROTOBUF_NAMESPACE_ID::Arena::InternalHelper;
|
|
typedef void InternalArenaConstructable_;
|
|
typedef void DestructorSkippable_;
|
|
struct Impl_ {
|
|
::PROTOBUF_NAMESPACE_ID::RepeatedField< int64_t > offset_dims_;
|
|
mutable std::atomic<int> _offset_dims_cached_byte_size_;
|
|
::PROTOBUF_NAMESPACE_ID::RepeatedField< int64_t > collapsed_slice_dims_;
|
|
mutable std::atomic<int> _collapsed_slice_dims_cached_byte_size_;
|
|
::PROTOBUF_NAMESPACE_ID::RepeatedField< int64_t > start_index_map_;
|
|
mutable std::atomic<int> _start_index_map_cached_byte_size_;
|
|
int64_t index_vector_dim_;
|
|
mutable ::PROTOBUF_NAMESPACE_ID::internal::CachedSize _cached_size_;
|
|
};
|
|
union { Impl_ _impl_; };
|
|
friend struct ::TableStruct_xla_2fxla_5fdata_2eproto;
|
|
};
|
|
// -------------------------------------------------------------------
|
|
|
|
class ScatterDimensionNumbers final :
|
|
public ::PROTOBUF_NAMESPACE_ID::Message /* @@protoc_insertion_point(class_definition:xla.ScatterDimensionNumbers) */ {
|
|
public:
|
|
inline ScatterDimensionNumbers() : ScatterDimensionNumbers(nullptr) {}
|
|
~ScatterDimensionNumbers() override;
|
|
explicit PROTOBUF_CONSTEXPR ScatterDimensionNumbers(::PROTOBUF_NAMESPACE_ID::internal::ConstantInitialized);
|
|
|
|
ScatterDimensionNumbers(const ScatterDimensionNumbers& from);
|
|
ScatterDimensionNumbers(ScatterDimensionNumbers&& from) noexcept
|
|
: ScatterDimensionNumbers() {
|
|
*this = ::std::move(from);
|
|
}
|
|
|
|
inline ScatterDimensionNumbers& operator=(const ScatterDimensionNumbers& from) {
|
|
CopyFrom(from);
|
|
return *this;
|
|
}
|
|
inline ScatterDimensionNumbers& operator=(ScatterDimensionNumbers&& from) noexcept {
|
|
if (this == &from) return *this;
|
|
if (GetOwningArena() == from.GetOwningArena()
|
|
#ifdef PROTOBUF_FORCE_COPY_IN_MOVE
|
|
&& GetOwningArena() != nullptr
|
|
#endif // !PROTOBUF_FORCE_COPY_IN_MOVE
|
|
) {
|
|
InternalSwap(&from);
|
|
} else {
|
|
CopyFrom(from);
|
|
}
|
|
return *this;
|
|
}
|
|
|
|
static const ::PROTOBUF_NAMESPACE_ID::Descriptor* descriptor() {
|
|
return GetDescriptor();
|
|
}
|
|
static const ::PROTOBUF_NAMESPACE_ID::Descriptor* GetDescriptor() {
|
|
return default_instance().GetMetadata().descriptor;
|
|
}
|
|
static const ::PROTOBUF_NAMESPACE_ID::Reflection* GetReflection() {
|
|
return default_instance().GetMetadata().reflection;
|
|
}
|
|
static const ScatterDimensionNumbers& default_instance() {
|
|
return *internal_default_instance();
|
|
}
|
|
static inline const ScatterDimensionNumbers* internal_default_instance() {
|
|
return reinterpret_cast<const ScatterDimensionNumbers*>(
|
|
&_ScatterDimensionNumbers_default_instance_);
|
|
}
|
|
static constexpr int kIndexInFileMessages =
|
|
20;
|
|
|
|
friend void swap(ScatterDimensionNumbers& a, ScatterDimensionNumbers& b) {
|
|
a.Swap(&b);
|
|
}
|
|
inline void Swap(ScatterDimensionNumbers* other) {
|
|
if (other == this) return;
|
|
#ifdef PROTOBUF_FORCE_COPY_IN_SWAP
|
|
if (GetOwningArena() != nullptr &&
|
|
GetOwningArena() == other->GetOwningArena()) {
|
|
#else // PROTOBUF_FORCE_COPY_IN_SWAP
|
|
if (GetOwningArena() == other->GetOwningArena()) {
|
|
#endif // !PROTOBUF_FORCE_COPY_IN_SWAP
|
|
InternalSwap(other);
|
|
} else {
|
|
::PROTOBUF_NAMESPACE_ID::internal::GenericSwap(this, other);
|
|
}
|
|
}
|
|
void UnsafeArenaSwap(ScatterDimensionNumbers* other) {
|
|
if (other == this) return;
|
|
GOOGLE_DCHECK(GetOwningArena() == other->GetOwningArena());
|
|
InternalSwap(other);
|
|
}
|
|
|
|
// implements Message ----------------------------------------------
|
|
|
|
ScatterDimensionNumbers* New(::PROTOBUF_NAMESPACE_ID::Arena* arena = nullptr) const final {
|
|
return CreateMaybeMessage<ScatterDimensionNumbers>(arena);
|
|
}
|
|
using ::PROTOBUF_NAMESPACE_ID::Message::CopyFrom;
|
|
void CopyFrom(const ScatterDimensionNumbers& from);
|
|
using ::PROTOBUF_NAMESPACE_ID::Message::MergeFrom;
|
|
void MergeFrom( const ScatterDimensionNumbers& from) {
|
|
ScatterDimensionNumbers::MergeImpl(*this, from);
|
|
}
|
|
private:
|
|
static void MergeImpl(::PROTOBUF_NAMESPACE_ID::Message& to_msg, const ::PROTOBUF_NAMESPACE_ID::Message& from_msg);
|
|
public:
|
|
PROTOBUF_ATTRIBUTE_REINITIALIZES void Clear() final;
|
|
bool IsInitialized() const final;
|
|
|
|
size_t ByteSizeLong() const final;
|
|
const char* _InternalParse(const char* ptr, ::PROTOBUF_NAMESPACE_ID::internal::ParseContext* ctx) final;
|
|
uint8_t* _InternalSerialize(
|
|
uint8_t* target, ::PROTOBUF_NAMESPACE_ID::io::EpsCopyOutputStream* stream) const final;
|
|
int GetCachedSize() const final { return _impl_._cached_size_.Get(); }
|
|
|
|
private:
|
|
void SharedCtor(::PROTOBUF_NAMESPACE_ID::Arena* arena, bool is_message_owned);
|
|
void SharedDtor();
|
|
void SetCachedSize(int size) const final;
|
|
void InternalSwap(ScatterDimensionNumbers* other);
|
|
|
|
private:
|
|
friend class ::PROTOBUF_NAMESPACE_ID::internal::AnyMetadata;
|
|
static ::PROTOBUF_NAMESPACE_ID::StringPiece FullMessageName() {
|
|
return "xla.ScatterDimensionNumbers";
|
|
}
|
|
protected:
|
|
explicit ScatterDimensionNumbers(::PROTOBUF_NAMESPACE_ID::Arena* arena,
|
|
bool is_message_owned = false);
|
|
public:
|
|
|
|
static const ClassData _class_data_;
|
|
const ::PROTOBUF_NAMESPACE_ID::Message::ClassData*GetClassData() const final;
|
|
|
|
::PROTOBUF_NAMESPACE_ID::Metadata GetMetadata() const final;
|
|
|
|
// nested types ----------------------------------------------------
|
|
|
|
// accessors -------------------------------------------------------
|
|
|
|
enum : int {
|
|
kUpdateWindowDimsFieldNumber = 1,
|
|
kInsertedWindowDimsFieldNumber = 2,
|
|
kScatterDimsToOperandDimsFieldNumber = 3,
|
|
kIndexVectorDimFieldNumber = 4,
|
|
};
|
|
// repeated int64 update_window_dims = 1;
|
|
int update_window_dims_size() const;
|
|
private:
|
|
int _internal_update_window_dims_size() const;
|
|
public:
|
|
void clear_update_window_dims();
|
|
private:
|
|
int64_t _internal_update_window_dims(int index) const;
|
|
const ::PROTOBUF_NAMESPACE_ID::RepeatedField< int64_t >&
|
|
_internal_update_window_dims() const;
|
|
void _internal_add_update_window_dims(int64_t value);
|
|
::PROTOBUF_NAMESPACE_ID::RepeatedField< int64_t >*
|
|
_internal_mutable_update_window_dims();
|
|
public:
|
|
int64_t update_window_dims(int index) const;
|
|
void set_update_window_dims(int index, int64_t value);
|
|
void add_update_window_dims(int64_t value);
|
|
const ::PROTOBUF_NAMESPACE_ID::RepeatedField< int64_t >&
|
|
update_window_dims() const;
|
|
::PROTOBUF_NAMESPACE_ID::RepeatedField< int64_t >*
|
|
mutable_update_window_dims();
|
|
|
|
// repeated int64 inserted_window_dims = 2;
|
|
int inserted_window_dims_size() const;
|
|
private:
|
|
int _internal_inserted_window_dims_size() const;
|
|
public:
|
|
void clear_inserted_window_dims();
|
|
private:
|
|
int64_t _internal_inserted_window_dims(int index) const;
|
|
const ::PROTOBUF_NAMESPACE_ID::RepeatedField< int64_t >&
|
|
_internal_inserted_window_dims() const;
|
|
void _internal_add_inserted_window_dims(int64_t value);
|
|
::PROTOBUF_NAMESPACE_ID::RepeatedField< int64_t >*
|
|
_internal_mutable_inserted_window_dims();
|
|
public:
|
|
int64_t inserted_window_dims(int index) const;
|
|
void set_inserted_window_dims(int index, int64_t value);
|
|
void add_inserted_window_dims(int64_t value);
|
|
const ::PROTOBUF_NAMESPACE_ID::RepeatedField< int64_t >&
|
|
inserted_window_dims() const;
|
|
::PROTOBUF_NAMESPACE_ID::RepeatedField< int64_t >*
|
|
mutable_inserted_window_dims();
|
|
|
|
// repeated int64 scatter_dims_to_operand_dims = 3;
|
|
int scatter_dims_to_operand_dims_size() const;
|
|
private:
|
|
int _internal_scatter_dims_to_operand_dims_size() const;
|
|
public:
|
|
void clear_scatter_dims_to_operand_dims();
|
|
private:
|
|
int64_t _internal_scatter_dims_to_operand_dims(int index) const;
|
|
const ::PROTOBUF_NAMESPACE_ID::RepeatedField< int64_t >&
|
|
_internal_scatter_dims_to_operand_dims() const;
|
|
void _internal_add_scatter_dims_to_operand_dims(int64_t value);
|
|
::PROTOBUF_NAMESPACE_ID::RepeatedField< int64_t >*
|
|
_internal_mutable_scatter_dims_to_operand_dims();
|
|
public:
|
|
int64_t scatter_dims_to_operand_dims(int index) const;
|
|
void set_scatter_dims_to_operand_dims(int index, int64_t value);
|
|
void add_scatter_dims_to_operand_dims(int64_t value);
|
|
const ::PROTOBUF_NAMESPACE_ID::RepeatedField< int64_t >&
|
|
scatter_dims_to_operand_dims() const;
|
|
::PROTOBUF_NAMESPACE_ID::RepeatedField< int64_t >*
|
|
mutable_scatter_dims_to_operand_dims();
|
|
|
|
// int64 index_vector_dim = 4;
|
|
void clear_index_vector_dim();
|
|
int64_t index_vector_dim() const;
|
|
void set_index_vector_dim(int64_t value);
|
|
private:
|
|
int64_t _internal_index_vector_dim() const;
|
|
void _internal_set_index_vector_dim(int64_t value);
|
|
public:
|
|
|
|
// @@protoc_insertion_point(class_scope:xla.ScatterDimensionNumbers)
|
|
private:
|
|
class _Internal;
|
|
|
|
template <typename T> friend class ::PROTOBUF_NAMESPACE_ID::Arena::InternalHelper;
|
|
typedef void InternalArenaConstructable_;
|
|
typedef void DestructorSkippable_;
|
|
struct Impl_ {
|
|
::PROTOBUF_NAMESPACE_ID::RepeatedField< int64_t > update_window_dims_;
|
|
mutable std::atomic<int> _update_window_dims_cached_byte_size_;
|
|
::PROTOBUF_NAMESPACE_ID::RepeatedField< int64_t > inserted_window_dims_;
|
|
mutable std::atomic<int> _inserted_window_dims_cached_byte_size_;
|
|
::PROTOBUF_NAMESPACE_ID::RepeatedField< int64_t > scatter_dims_to_operand_dims_;
|
|
mutable std::atomic<int> _scatter_dims_to_operand_dims_cached_byte_size_;
|
|
int64_t index_vector_dim_;
|
|
mutable ::PROTOBUF_NAMESPACE_ID::internal::CachedSize _cached_size_;
|
|
};
|
|
union { Impl_ _impl_; };
|
|
friend struct ::TableStruct_xla_2fxla_5fdata_2eproto;
|
|
};
|
|
// -------------------------------------------------------------------
|
|
|
|
class ConvolutionDimensionNumbers final :
|
|
public ::PROTOBUF_NAMESPACE_ID::Message /* @@protoc_insertion_point(class_definition:xla.ConvolutionDimensionNumbers) */ {
|
|
public:
|
|
inline ConvolutionDimensionNumbers() : ConvolutionDimensionNumbers(nullptr) {}
|
|
~ConvolutionDimensionNumbers() override;
|
|
explicit PROTOBUF_CONSTEXPR ConvolutionDimensionNumbers(::PROTOBUF_NAMESPACE_ID::internal::ConstantInitialized);
|
|
|
|
ConvolutionDimensionNumbers(const ConvolutionDimensionNumbers& from);
|
|
ConvolutionDimensionNumbers(ConvolutionDimensionNumbers&& from) noexcept
|
|
: ConvolutionDimensionNumbers() {
|
|
*this = ::std::move(from);
|
|
}
|
|
|
|
inline ConvolutionDimensionNumbers& operator=(const ConvolutionDimensionNumbers& from) {
|
|
CopyFrom(from);
|
|
return *this;
|
|
}
|
|
inline ConvolutionDimensionNumbers& operator=(ConvolutionDimensionNumbers&& from) noexcept {
|
|
if (this == &from) return *this;
|
|
if (GetOwningArena() == from.GetOwningArena()
|
|
#ifdef PROTOBUF_FORCE_COPY_IN_MOVE
|
|
&& GetOwningArena() != nullptr
|
|
#endif // !PROTOBUF_FORCE_COPY_IN_MOVE
|
|
) {
|
|
InternalSwap(&from);
|
|
} else {
|
|
CopyFrom(from);
|
|
}
|
|
return *this;
|
|
}
|
|
|
|
static const ::PROTOBUF_NAMESPACE_ID::Descriptor* descriptor() {
|
|
return GetDescriptor();
|
|
}
|
|
static const ::PROTOBUF_NAMESPACE_ID::Descriptor* GetDescriptor() {
|
|
return default_instance().GetMetadata().descriptor;
|
|
}
|
|
static const ::PROTOBUF_NAMESPACE_ID::Reflection* GetReflection() {
|
|
return default_instance().GetMetadata().reflection;
|
|
}
|
|
static const ConvolutionDimensionNumbers& default_instance() {
|
|
return *internal_default_instance();
|
|
}
|
|
static inline const ConvolutionDimensionNumbers* internal_default_instance() {
|
|
return reinterpret_cast<const ConvolutionDimensionNumbers*>(
|
|
&_ConvolutionDimensionNumbers_default_instance_);
|
|
}
|
|
static constexpr int kIndexInFileMessages =
|
|
21;
|
|
|
|
friend void swap(ConvolutionDimensionNumbers& a, ConvolutionDimensionNumbers& b) {
|
|
a.Swap(&b);
|
|
}
|
|
inline void Swap(ConvolutionDimensionNumbers* other) {
|
|
if (other == this) return;
|
|
#ifdef PROTOBUF_FORCE_COPY_IN_SWAP
|
|
if (GetOwningArena() != nullptr &&
|
|
GetOwningArena() == other->GetOwningArena()) {
|
|
#else // PROTOBUF_FORCE_COPY_IN_SWAP
|
|
if (GetOwningArena() == other->GetOwningArena()) {
|
|
#endif // !PROTOBUF_FORCE_COPY_IN_SWAP
|
|
InternalSwap(other);
|
|
} else {
|
|
::PROTOBUF_NAMESPACE_ID::internal::GenericSwap(this, other);
|
|
}
|
|
}
|
|
void UnsafeArenaSwap(ConvolutionDimensionNumbers* other) {
|
|
if (other == this) return;
|
|
GOOGLE_DCHECK(GetOwningArena() == other->GetOwningArena());
|
|
InternalSwap(other);
|
|
}
|
|
|
|
// implements Message ----------------------------------------------
|
|
|
|
ConvolutionDimensionNumbers* New(::PROTOBUF_NAMESPACE_ID::Arena* arena = nullptr) const final {
|
|
return CreateMaybeMessage<ConvolutionDimensionNumbers>(arena);
|
|
}
|
|
using ::PROTOBUF_NAMESPACE_ID::Message::CopyFrom;
|
|
void CopyFrom(const ConvolutionDimensionNumbers& from);
|
|
using ::PROTOBUF_NAMESPACE_ID::Message::MergeFrom;
|
|
void MergeFrom( const ConvolutionDimensionNumbers& from) {
|
|
ConvolutionDimensionNumbers::MergeImpl(*this, from);
|
|
}
|
|
private:
|
|
static void MergeImpl(::PROTOBUF_NAMESPACE_ID::Message& to_msg, const ::PROTOBUF_NAMESPACE_ID::Message& from_msg);
|
|
public:
|
|
PROTOBUF_ATTRIBUTE_REINITIALIZES void Clear() final;
|
|
bool IsInitialized() const final;
|
|
|
|
size_t ByteSizeLong() const final;
|
|
const char* _InternalParse(const char* ptr, ::PROTOBUF_NAMESPACE_ID::internal::ParseContext* ctx) final;
|
|
uint8_t* _InternalSerialize(
|
|
uint8_t* target, ::PROTOBUF_NAMESPACE_ID::io::EpsCopyOutputStream* stream) const final;
|
|
int GetCachedSize() const final { return _impl_._cached_size_.Get(); }
|
|
|
|
private:
|
|
void SharedCtor(::PROTOBUF_NAMESPACE_ID::Arena* arena, bool is_message_owned);
|
|
void SharedDtor();
|
|
void SetCachedSize(int size) const final;
|
|
void InternalSwap(ConvolutionDimensionNumbers* other);
|
|
|
|
private:
|
|
friend class ::PROTOBUF_NAMESPACE_ID::internal::AnyMetadata;
|
|
static ::PROTOBUF_NAMESPACE_ID::StringPiece FullMessageName() {
|
|
return "xla.ConvolutionDimensionNumbers";
|
|
}
|
|
protected:
|
|
explicit ConvolutionDimensionNumbers(::PROTOBUF_NAMESPACE_ID::Arena* arena,
|
|
bool is_message_owned = false);
|
|
public:
|
|
|
|
static const ClassData _class_data_;
|
|
const ::PROTOBUF_NAMESPACE_ID::Message::ClassData*GetClassData() const final;
|
|
|
|
::PROTOBUF_NAMESPACE_ID::Metadata GetMetadata() const final;
|
|
|
|
// nested types ----------------------------------------------------
|
|
|
|
// accessors -------------------------------------------------------
|
|
|
|
enum : int {
|
|
kKernelSpatialDimensionsFieldNumber = 6,
|
|
kInputSpatialDimensionsFieldNumber = 11,
|
|
kOutputSpatialDimensionsFieldNumber = 12,
|
|
kKernelInputFeatureDimensionFieldNumber = 3,
|
|
kKernelOutputFeatureDimensionFieldNumber = 4,
|
|
kInputBatchDimensionFieldNumber = 7,
|
|
kInputFeatureDimensionFieldNumber = 8,
|
|
kOutputBatchDimensionFieldNumber = 9,
|
|
kOutputFeatureDimensionFieldNumber = 10,
|
|
};
|
|
// repeated int64 kernel_spatial_dimensions = 6;
|
|
int kernel_spatial_dimensions_size() const;
|
|
private:
|
|
int _internal_kernel_spatial_dimensions_size() const;
|
|
public:
|
|
void clear_kernel_spatial_dimensions();
|
|
private:
|
|
int64_t _internal_kernel_spatial_dimensions(int index) const;
|
|
const ::PROTOBUF_NAMESPACE_ID::RepeatedField< int64_t >&
|
|
_internal_kernel_spatial_dimensions() const;
|
|
void _internal_add_kernel_spatial_dimensions(int64_t value);
|
|
::PROTOBUF_NAMESPACE_ID::RepeatedField< int64_t >*
|
|
_internal_mutable_kernel_spatial_dimensions();
|
|
public:
|
|
int64_t kernel_spatial_dimensions(int index) const;
|
|
void set_kernel_spatial_dimensions(int index, int64_t value);
|
|
void add_kernel_spatial_dimensions(int64_t value);
|
|
const ::PROTOBUF_NAMESPACE_ID::RepeatedField< int64_t >&
|
|
kernel_spatial_dimensions() const;
|
|
::PROTOBUF_NAMESPACE_ID::RepeatedField< int64_t >*
|
|
mutable_kernel_spatial_dimensions();
|
|
|
|
// repeated int64 input_spatial_dimensions = 11;
|
|
int input_spatial_dimensions_size() const;
|
|
private:
|
|
int _internal_input_spatial_dimensions_size() const;
|
|
public:
|
|
void clear_input_spatial_dimensions();
|
|
private:
|
|
int64_t _internal_input_spatial_dimensions(int index) const;
|
|
const ::PROTOBUF_NAMESPACE_ID::RepeatedField< int64_t >&
|
|
_internal_input_spatial_dimensions() const;
|
|
void _internal_add_input_spatial_dimensions(int64_t value);
|
|
::PROTOBUF_NAMESPACE_ID::RepeatedField< int64_t >*
|
|
_internal_mutable_input_spatial_dimensions();
|
|
public:
|
|
int64_t input_spatial_dimensions(int index) const;
|
|
void set_input_spatial_dimensions(int index, int64_t value);
|
|
void add_input_spatial_dimensions(int64_t value);
|
|
const ::PROTOBUF_NAMESPACE_ID::RepeatedField< int64_t >&
|
|
input_spatial_dimensions() const;
|
|
::PROTOBUF_NAMESPACE_ID::RepeatedField< int64_t >*
|
|
mutable_input_spatial_dimensions();
|
|
|
|
// repeated int64 output_spatial_dimensions = 12;
|
|
int output_spatial_dimensions_size() const;
|
|
private:
|
|
int _internal_output_spatial_dimensions_size() const;
|
|
public:
|
|
void clear_output_spatial_dimensions();
|
|
private:
|
|
int64_t _internal_output_spatial_dimensions(int index) const;
|
|
const ::PROTOBUF_NAMESPACE_ID::RepeatedField< int64_t >&
|
|
_internal_output_spatial_dimensions() const;
|
|
void _internal_add_output_spatial_dimensions(int64_t value);
|
|
::PROTOBUF_NAMESPACE_ID::RepeatedField< int64_t >*
|
|
_internal_mutable_output_spatial_dimensions();
|
|
public:
|
|
int64_t output_spatial_dimensions(int index) const;
|
|
void set_output_spatial_dimensions(int index, int64_t value);
|
|
void add_output_spatial_dimensions(int64_t value);
|
|
const ::PROTOBUF_NAMESPACE_ID::RepeatedField< int64_t >&
|
|
output_spatial_dimensions() const;
|
|
::PROTOBUF_NAMESPACE_ID::RepeatedField< int64_t >*
|
|
mutable_output_spatial_dimensions();
|
|
|
|
// int64 kernel_input_feature_dimension = 3;
|
|
void clear_kernel_input_feature_dimension();
|
|
int64_t kernel_input_feature_dimension() const;
|
|
void set_kernel_input_feature_dimension(int64_t value);
|
|
private:
|
|
int64_t _internal_kernel_input_feature_dimension() const;
|
|
void _internal_set_kernel_input_feature_dimension(int64_t value);
|
|
public:
|
|
|
|
// int64 kernel_output_feature_dimension = 4;
|
|
void clear_kernel_output_feature_dimension();
|
|
int64_t kernel_output_feature_dimension() const;
|
|
void set_kernel_output_feature_dimension(int64_t value);
|
|
private:
|
|
int64_t _internal_kernel_output_feature_dimension() const;
|
|
void _internal_set_kernel_output_feature_dimension(int64_t value);
|
|
public:
|
|
|
|
// int64 input_batch_dimension = 7;
|
|
void clear_input_batch_dimension();
|
|
int64_t input_batch_dimension() const;
|
|
void set_input_batch_dimension(int64_t value);
|
|
private:
|
|
int64_t _internal_input_batch_dimension() const;
|
|
void _internal_set_input_batch_dimension(int64_t value);
|
|
public:
|
|
|
|
// int64 input_feature_dimension = 8;
|
|
void clear_input_feature_dimension();
|
|
int64_t input_feature_dimension() const;
|
|
void set_input_feature_dimension(int64_t value);
|
|
private:
|
|
int64_t _internal_input_feature_dimension() const;
|
|
void _internal_set_input_feature_dimension(int64_t value);
|
|
public:
|
|
|
|
// int64 output_batch_dimension = 9;
|
|
void clear_output_batch_dimension();
|
|
int64_t output_batch_dimension() const;
|
|
void set_output_batch_dimension(int64_t value);
|
|
private:
|
|
int64_t _internal_output_batch_dimension() const;
|
|
void _internal_set_output_batch_dimension(int64_t value);
|
|
public:
|
|
|
|
// int64 output_feature_dimension = 10;
|
|
void clear_output_feature_dimension();
|
|
int64_t output_feature_dimension() const;
|
|
void set_output_feature_dimension(int64_t value);
|
|
private:
|
|
int64_t _internal_output_feature_dimension() const;
|
|
void _internal_set_output_feature_dimension(int64_t value);
|
|
public:
|
|
|
|
// @@protoc_insertion_point(class_scope:xla.ConvolutionDimensionNumbers)
|
|
private:
|
|
class _Internal;
|
|
|
|
template <typename T> friend class ::PROTOBUF_NAMESPACE_ID::Arena::InternalHelper;
|
|
typedef void InternalArenaConstructable_;
|
|
typedef void DestructorSkippable_;
|
|
struct Impl_ {
|
|
::PROTOBUF_NAMESPACE_ID::RepeatedField< int64_t > kernel_spatial_dimensions_;
|
|
mutable std::atomic<int> _kernel_spatial_dimensions_cached_byte_size_;
|
|
::PROTOBUF_NAMESPACE_ID::RepeatedField< int64_t > input_spatial_dimensions_;
|
|
mutable std::atomic<int> _input_spatial_dimensions_cached_byte_size_;
|
|
::PROTOBUF_NAMESPACE_ID::RepeatedField< int64_t > output_spatial_dimensions_;
|
|
mutable std::atomic<int> _output_spatial_dimensions_cached_byte_size_;
|
|
int64_t kernel_input_feature_dimension_;
|
|
int64_t kernel_output_feature_dimension_;
|
|
int64_t input_batch_dimension_;
|
|
int64_t input_feature_dimension_;
|
|
int64_t output_batch_dimension_;
|
|
int64_t output_feature_dimension_;
|
|
mutable ::PROTOBUF_NAMESPACE_ID::internal::CachedSize _cached_size_;
|
|
};
|
|
union { Impl_ _impl_; };
|
|
friend struct ::TableStruct_xla_2fxla_5fdata_2eproto;
|
|
};
|
|
// -------------------------------------------------------------------
|
|
|
|
class DotDimensionNumbers final :
|
|
public ::PROTOBUF_NAMESPACE_ID::Message /* @@protoc_insertion_point(class_definition:xla.DotDimensionNumbers) */ {
|
|
public:
|
|
inline DotDimensionNumbers() : DotDimensionNumbers(nullptr) {}
|
|
~DotDimensionNumbers() override;
|
|
explicit PROTOBUF_CONSTEXPR DotDimensionNumbers(::PROTOBUF_NAMESPACE_ID::internal::ConstantInitialized);
|
|
|
|
DotDimensionNumbers(const DotDimensionNumbers& from);
|
|
DotDimensionNumbers(DotDimensionNumbers&& from) noexcept
|
|
: DotDimensionNumbers() {
|
|
*this = ::std::move(from);
|
|
}
|
|
|
|
inline DotDimensionNumbers& operator=(const DotDimensionNumbers& from) {
|
|
CopyFrom(from);
|
|
return *this;
|
|
}
|
|
inline DotDimensionNumbers& operator=(DotDimensionNumbers&& from) noexcept {
|
|
if (this == &from) return *this;
|
|
if (GetOwningArena() == from.GetOwningArena()
|
|
#ifdef PROTOBUF_FORCE_COPY_IN_MOVE
|
|
&& GetOwningArena() != nullptr
|
|
#endif // !PROTOBUF_FORCE_COPY_IN_MOVE
|
|
) {
|
|
InternalSwap(&from);
|
|
} else {
|
|
CopyFrom(from);
|
|
}
|
|
return *this;
|
|
}
|
|
|
|
static const ::PROTOBUF_NAMESPACE_ID::Descriptor* descriptor() {
|
|
return GetDescriptor();
|
|
}
|
|
static const ::PROTOBUF_NAMESPACE_ID::Descriptor* GetDescriptor() {
|
|
return default_instance().GetMetadata().descriptor;
|
|
}
|
|
static const ::PROTOBUF_NAMESPACE_ID::Reflection* GetReflection() {
|
|
return default_instance().GetMetadata().reflection;
|
|
}
|
|
static const DotDimensionNumbers& default_instance() {
|
|
return *internal_default_instance();
|
|
}
|
|
static inline const DotDimensionNumbers* internal_default_instance() {
|
|
return reinterpret_cast<const DotDimensionNumbers*>(
|
|
&_DotDimensionNumbers_default_instance_);
|
|
}
|
|
static constexpr int kIndexInFileMessages =
|
|
22;
|
|
|
|
friend void swap(DotDimensionNumbers& a, DotDimensionNumbers& b) {
|
|
a.Swap(&b);
|
|
}
|
|
inline void Swap(DotDimensionNumbers* other) {
|
|
if (other == this) return;
|
|
#ifdef PROTOBUF_FORCE_COPY_IN_SWAP
|
|
if (GetOwningArena() != nullptr &&
|
|
GetOwningArena() == other->GetOwningArena()) {
|
|
#else // PROTOBUF_FORCE_COPY_IN_SWAP
|
|
if (GetOwningArena() == other->GetOwningArena()) {
|
|
#endif // !PROTOBUF_FORCE_COPY_IN_SWAP
|
|
InternalSwap(other);
|
|
} else {
|
|
::PROTOBUF_NAMESPACE_ID::internal::GenericSwap(this, other);
|
|
}
|
|
}
|
|
void UnsafeArenaSwap(DotDimensionNumbers* other) {
|
|
if (other == this) return;
|
|
GOOGLE_DCHECK(GetOwningArena() == other->GetOwningArena());
|
|
InternalSwap(other);
|
|
}
|
|
|
|
// implements Message ----------------------------------------------
|
|
|
|
DotDimensionNumbers* New(::PROTOBUF_NAMESPACE_ID::Arena* arena = nullptr) const final {
|
|
return CreateMaybeMessage<DotDimensionNumbers>(arena);
|
|
}
|
|
using ::PROTOBUF_NAMESPACE_ID::Message::CopyFrom;
|
|
void CopyFrom(const DotDimensionNumbers& from);
|
|
using ::PROTOBUF_NAMESPACE_ID::Message::MergeFrom;
|
|
void MergeFrom( const DotDimensionNumbers& from) {
|
|
DotDimensionNumbers::MergeImpl(*this, from);
|
|
}
|
|
private:
|
|
static void MergeImpl(::PROTOBUF_NAMESPACE_ID::Message& to_msg, const ::PROTOBUF_NAMESPACE_ID::Message& from_msg);
|
|
public:
|
|
PROTOBUF_ATTRIBUTE_REINITIALIZES void Clear() final;
|
|
bool IsInitialized() const final;
|
|
|
|
size_t ByteSizeLong() const final;
|
|
const char* _InternalParse(const char* ptr, ::PROTOBUF_NAMESPACE_ID::internal::ParseContext* ctx) final;
|
|
uint8_t* _InternalSerialize(
|
|
uint8_t* target, ::PROTOBUF_NAMESPACE_ID::io::EpsCopyOutputStream* stream) const final;
|
|
int GetCachedSize() const final { return _impl_._cached_size_.Get(); }
|
|
|
|
private:
|
|
void SharedCtor(::PROTOBUF_NAMESPACE_ID::Arena* arena, bool is_message_owned);
|
|
void SharedDtor();
|
|
void SetCachedSize(int size) const final;
|
|
void InternalSwap(DotDimensionNumbers* other);
|
|
|
|
private:
|
|
friend class ::PROTOBUF_NAMESPACE_ID::internal::AnyMetadata;
|
|
static ::PROTOBUF_NAMESPACE_ID::StringPiece FullMessageName() {
|
|
return "xla.DotDimensionNumbers";
|
|
}
|
|
protected:
|
|
explicit DotDimensionNumbers(::PROTOBUF_NAMESPACE_ID::Arena* arena,
|
|
bool is_message_owned = false);
|
|
public:
|
|
|
|
static const ClassData _class_data_;
|
|
const ::PROTOBUF_NAMESPACE_ID::Message::ClassData*GetClassData() const final;
|
|
|
|
::PROTOBUF_NAMESPACE_ID::Metadata GetMetadata() const final;
|
|
|
|
// nested types ----------------------------------------------------
|
|
|
|
// accessors -------------------------------------------------------
|
|
|
|
enum : int {
|
|
kLhsContractingDimensionsFieldNumber = 1,
|
|
kRhsContractingDimensionsFieldNumber = 2,
|
|
kLhsBatchDimensionsFieldNumber = 3,
|
|
kRhsBatchDimensionsFieldNumber = 4,
|
|
};
|
|
// repeated int64 lhs_contracting_dimensions = 1;
|
|
int lhs_contracting_dimensions_size() const;
|
|
private:
|
|
int _internal_lhs_contracting_dimensions_size() const;
|
|
public:
|
|
void clear_lhs_contracting_dimensions();
|
|
private:
|
|
int64_t _internal_lhs_contracting_dimensions(int index) const;
|
|
const ::PROTOBUF_NAMESPACE_ID::RepeatedField< int64_t >&
|
|
_internal_lhs_contracting_dimensions() const;
|
|
void _internal_add_lhs_contracting_dimensions(int64_t value);
|
|
::PROTOBUF_NAMESPACE_ID::RepeatedField< int64_t >*
|
|
_internal_mutable_lhs_contracting_dimensions();
|
|
public:
|
|
int64_t lhs_contracting_dimensions(int index) const;
|
|
void set_lhs_contracting_dimensions(int index, int64_t value);
|
|
void add_lhs_contracting_dimensions(int64_t value);
|
|
const ::PROTOBUF_NAMESPACE_ID::RepeatedField< int64_t >&
|
|
lhs_contracting_dimensions() const;
|
|
::PROTOBUF_NAMESPACE_ID::RepeatedField< int64_t >*
|
|
mutable_lhs_contracting_dimensions();
|
|
|
|
// repeated int64 rhs_contracting_dimensions = 2;
|
|
int rhs_contracting_dimensions_size() const;
|
|
private:
|
|
int _internal_rhs_contracting_dimensions_size() const;
|
|
public:
|
|
void clear_rhs_contracting_dimensions();
|
|
private:
|
|
int64_t _internal_rhs_contracting_dimensions(int index) const;
|
|
const ::PROTOBUF_NAMESPACE_ID::RepeatedField< int64_t >&
|
|
_internal_rhs_contracting_dimensions() const;
|
|
void _internal_add_rhs_contracting_dimensions(int64_t value);
|
|
::PROTOBUF_NAMESPACE_ID::RepeatedField< int64_t >*
|
|
_internal_mutable_rhs_contracting_dimensions();
|
|
public:
|
|
int64_t rhs_contracting_dimensions(int index) const;
|
|
void set_rhs_contracting_dimensions(int index, int64_t value);
|
|
void add_rhs_contracting_dimensions(int64_t value);
|
|
const ::PROTOBUF_NAMESPACE_ID::RepeatedField< int64_t >&
|
|
rhs_contracting_dimensions() const;
|
|
::PROTOBUF_NAMESPACE_ID::RepeatedField< int64_t >*
|
|
mutable_rhs_contracting_dimensions();
|
|
|
|
// repeated int64 lhs_batch_dimensions = 3;
|
|
int lhs_batch_dimensions_size() const;
|
|
private:
|
|
int _internal_lhs_batch_dimensions_size() const;
|
|
public:
|
|
void clear_lhs_batch_dimensions();
|
|
private:
|
|
int64_t _internal_lhs_batch_dimensions(int index) const;
|
|
const ::PROTOBUF_NAMESPACE_ID::RepeatedField< int64_t >&
|
|
_internal_lhs_batch_dimensions() const;
|
|
void _internal_add_lhs_batch_dimensions(int64_t value);
|
|
::PROTOBUF_NAMESPACE_ID::RepeatedField< int64_t >*
|
|
_internal_mutable_lhs_batch_dimensions();
|
|
public:
|
|
int64_t lhs_batch_dimensions(int index) const;
|
|
void set_lhs_batch_dimensions(int index, int64_t value);
|
|
void add_lhs_batch_dimensions(int64_t value);
|
|
const ::PROTOBUF_NAMESPACE_ID::RepeatedField< int64_t >&
|
|
lhs_batch_dimensions() const;
|
|
::PROTOBUF_NAMESPACE_ID::RepeatedField< int64_t >*
|
|
mutable_lhs_batch_dimensions();
|
|
|
|
// repeated int64 rhs_batch_dimensions = 4;
|
|
int rhs_batch_dimensions_size() const;
|
|
private:
|
|
int _internal_rhs_batch_dimensions_size() const;
|
|
public:
|
|
void clear_rhs_batch_dimensions();
|
|
private:
|
|
int64_t _internal_rhs_batch_dimensions(int index) const;
|
|
const ::PROTOBUF_NAMESPACE_ID::RepeatedField< int64_t >&
|
|
_internal_rhs_batch_dimensions() const;
|
|
void _internal_add_rhs_batch_dimensions(int64_t value);
|
|
::PROTOBUF_NAMESPACE_ID::RepeatedField< int64_t >*
|
|
_internal_mutable_rhs_batch_dimensions();
|
|
public:
|
|
int64_t rhs_batch_dimensions(int index) const;
|
|
void set_rhs_batch_dimensions(int index, int64_t value);
|
|
void add_rhs_batch_dimensions(int64_t value);
|
|
const ::PROTOBUF_NAMESPACE_ID::RepeatedField< int64_t >&
|
|
rhs_batch_dimensions() const;
|
|
::PROTOBUF_NAMESPACE_ID::RepeatedField< int64_t >*
|
|
mutable_rhs_batch_dimensions();
|
|
|
|
// @@protoc_insertion_point(class_scope:xla.DotDimensionNumbers)
|
|
private:
|
|
class _Internal;
|
|
|
|
template <typename T> friend class ::PROTOBUF_NAMESPACE_ID::Arena::InternalHelper;
|
|
typedef void InternalArenaConstructable_;
|
|
typedef void DestructorSkippable_;
|
|
struct Impl_ {
|
|
::PROTOBUF_NAMESPACE_ID::RepeatedField< int64_t > lhs_contracting_dimensions_;
|
|
mutable std::atomic<int> _lhs_contracting_dimensions_cached_byte_size_;
|
|
::PROTOBUF_NAMESPACE_ID::RepeatedField< int64_t > rhs_contracting_dimensions_;
|
|
mutable std::atomic<int> _rhs_contracting_dimensions_cached_byte_size_;
|
|
::PROTOBUF_NAMESPACE_ID::RepeatedField< int64_t > lhs_batch_dimensions_;
|
|
mutable std::atomic<int> _lhs_batch_dimensions_cached_byte_size_;
|
|
::PROTOBUF_NAMESPACE_ID::RepeatedField< int64_t > rhs_batch_dimensions_;
|
|
mutable std::atomic<int> _rhs_batch_dimensions_cached_byte_size_;
|
|
mutable ::PROTOBUF_NAMESPACE_ID::internal::CachedSize _cached_size_;
|
|
};
|
|
union { Impl_ _impl_; };
|
|
friend struct ::TableStruct_xla_2fxla_5fdata_2eproto;
|
|
};
|
|
// -------------------------------------------------------------------
|
|
|
|
class SparsityDescriptor final :
|
|
public ::PROTOBUF_NAMESPACE_ID::Message /* @@protoc_insertion_point(class_definition:xla.SparsityDescriptor) */ {
|
|
public:
|
|
inline SparsityDescriptor() : SparsityDescriptor(nullptr) {}
|
|
~SparsityDescriptor() override;
|
|
explicit PROTOBUF_CONSTEXPR SparsityDescriptor(::PROTOBUF_NAMESPACE_ID::internal::ConstantInitialized);
|
|
|
|
SparsityDescriptor(const SparsityDescriptor& from);
|
|
SparsityDescriptor(SparsityDescriptor&& from) noexcept
|
|
: SparsityDescriptor() {
|
|
*this = ::std::move(from);
|
|
}
|
|
|
|
inline SparsityDescriptor& operator=(const SparsityDescriptor& from) {
|
|
CopyFrom(from);
|
|
return *this;
|
|
}
|
|
inline SparsityDescriptor& operator=(SparsityDescriptor&& from) noexcept {
|
|
if (this == &from) return *this;
|
|
if (GetOwningArena() == from.GetOwningArena()
|
|
#ifdef PROTOBUF_FORCE_COPY_IN_MOVE
|
|
&& GetOwningArena() != nullptr
|
|
#endif // !PROTOBUF_FORCE_COPY_IN_MOVE
|
|
) {
|
|
InternalSwap(&from);
|
|
} else {
|
|
CopyFrom(from);
|
|
}
|
|
return *this;
|
|
}
|
|
|
|
static const ::PROTOBUF_NAMESPACE_ID::Descriptor* descriptor() {
|
|
return GetDescriptor();
|
|
}
|
|
static const ::PROTOBUF_NAMESPACE_ID::Descriptor* GetDescriptor() {
|
|
return default_instance().GetMetadata().descriptor;
|
|
}
|
|
static const ::PROTOBUF_NAMESPACE_ID::Reflection* GetReflection() {
|
|
return default_instance().GetMetadata().reflection;
|
|
}
|
|
static const SparsityDescriptor& default_instance() {
|
|
return *internal_default_instance();
|
|
}
|
|
static inline const SparsityDescriptor* internal_default_instance() {
|
|
return reinterpret_cast<const SparsityDescriptor*>(
|
|
&_SparsityDescriptor_default_instance_);
|
|
}
|
|
static constexpr int kIndexInFileMessages =
|
|
23;
|
|
|
|
friend void swap(SparsityDescriptor& a, SparsityDescriptor& b) {
|
|
a.Swap(&b);
|
|
}
|
|
inline void Swap(SparsityDescriptor* other) {
|
|
if (other == this) return;
|
|
#ifdef PROTOBUF_FORCE_COPY_IN_SWAP
|
|
if (GetOwningArena() != nullptr &&
|
|
GetOwningArena() == other->GetOwningArena()) {
|
|
#else // PROTOBUF_FORCE_COPY_IN_SWAP
|
|
if (GetOwningArena() == other->GetOwningArena()) {
|
|
#endif // !PROTOBUF_FORCE_COPY_IN_SWAP
|
|
InternalSwap(other);
|
|
} else {
|
|
::PROTOBUF_NAMESPACE_ID::internal::GenericSwap(this, other);
|
|
}
|
|
}
|
|
void UnsafeArenaSwap(SparsityDescriptor* other) {
|
|
if (other == this) return;
|
|
GOOGLE_DCHECK(GetOwningArena() == other->GetOwningArena());
|
|
InternalSwap(other);
|
|
}
|
|
|
|
// implements Message ----------------------------------------------
|
|
|
|
SparsityDescriptor* New(::PROTOBUF_NAMESPACE_ID::Arena* arena = nullptr) const final {
|
|
return CreateMaybeMessage<SparsityDescriptor>(arena);
|
|
}
|
|
using ::PROTOBUF_NAMESPACE_ID::Message::CopyFrom;
|
|
void CopyFrom(const SparsityDescriptor& from);
|
|
using ::PROTOBUF_NAMESPACE_ID::Message::MergeFrom;
|
|
void MergeFrom( const SparsityDescriptor& from) {
|
|
SparsityDescriptor::MergeImpl(*this, from);
|
|
}
|
|
private:
|
|
static void MergeImpl(::PROTOBUF_NAMESPACE_ID::Message& to_msg, const ::PROTOBUF_NAMESPACE_ID::Message& from_msg);
|
|
public:
|
|
PROTOBUF_ATTRIBUTE_REINITIALIZES void Clear() final;
|
|
bool IsInitialized() const final;
|
|
|
|
size_t ByteSizeLong() const final;
|
|
const char* _InternalParse(const char* ptr, ::PROTOBUF_NAMESPACE_ID::internal::ParseContext* ctx) final;
|
|
uint8_t* _InternalSerialize(
|
|
uint8_t* target, ::PROTOBUF_NAMESPACE_ID::io::EpsCopyOutputStream* stream) const final;
|
|
int GetCachedSize() const final { return _impl_._cached_size_.Get(); }
|
|
|
|
private:
|
|
void SharedCtor(::PROTOBUF_NAMESPACE_ID::Arena* arena, bool is_message_owned);
|
|
void SharedDtor();
|
|
void SetCachedSize(int size) const final;
|
|
void InternalSwap(SparsityDescriptor* other);
|
|
|
|
private:
|
|
friend class ::PROTOBUF_NAMESPACE_ID::internal::AnyMetadata;
|
|
static ::PROTOBUF_NAMESPACE_ID::StringPiece FullMessageName() {
|
|
return "xla.SparsityDescriptor";
|
|
}
|
|
protected:
|
|
explicit SparsityDescriptor(::PROTOBUF_NAMESPACE_ID::Arena* arena,
|
|
bool is_message_owned = false);
|
|
public:
|
|
|
|
static const ClassData _class_data_;
|
|
const ::PROTOBUF_NAMESPACE_ID::Message::ClassData*GetClassData() const final;
|
|
|
|
::PROTOBUF_NAMESPACE_ID::Metadata GetMetadata() const final;
|
|
|
|
// nested types ----------------------------------------------------
|
|
|
|
// accessors -------------------------------------------------------
|
|
|
|
enum : int {
|
|
kTypeFieldNumber = 1,
|
|
kIndexFieldNumber = 2,
|
|
kDimensionFieldNumber = 3,
|
|
kNFieldNumber = 4,
|
|
kMFieldNumber = 5,
|
|
};
|
|
// .xla.SparsityType type = 1;
|
|
void clear_type();
|
|
::xla::SparsityType type() const;
|
|
void set_type(::xla::SparsityType value);
|
|
private:
|
|
::xla::SparsityType _internal_type() const;
|
|
void _internal_set_type(::xla::SparsityType value);
|
|
public:
|
|
|
|
// int32 index = 2;
|
|
void clear_index();
|
|
int32_t index() const;
|
|
void set_index(int32_t value);
|
|
private:
|
|
int32_t _internal_index() const;
|
|
void _internal_set_index(int32_t value);
|
|
public:
|
|
|
|
// int32 dimension = 3;
|
|
void clear_dimension();
|
|
int32_t dimension() const;
|
|
void set_dimension(int32_t value);
|
|
private:
|
|
int32_t _internal_dimension() const;
|
|
void _internal_set_dimension(int32_t value);
|
|
public:
|
|
|
|
// int32 n = 4;
|
|
void clear_n();
|
|
int32_t n() const;
|
|
void set_n(int32_t value);
|
|
private:
|
|
int32_t _internal_n() const;
|
|
void _internal_set_n(int32_t value);
|
|
public:
|
|
|
|
// int32 m = 5;
|
|
void clear_m();
|
|
int32_t m() const;
|
|
void set_m(int32_t value);
|
|
private:
|
|
int32_t _internal_m() const;
|
|
void _internal_set_m(int32_t value);
|
|
public:
|
|
|
|
// @@protoc_insertion_point(class_scope:xla.SparsityDescriptor)
|
|
private:
|
|
class _Internal;
|
|
|
|
template <typename T> friend class ::PROTOBUF_NAMESPACE_ID::Arena::InternalHelper;
|
|
typedef void InternalArenaConstructable_;
|
|
typedef void DestructorSkippable_;
|
|
struct Impl_ {
|
|
int type_;
|
|
int32_t index_;
|
|
int32_t dimension_;
|
|
int32_t n_;
|
|
int32_t m_;
|
|
mutable ::PROTOBUF_NAMESPACE_ID::internal::CachedSize _cached_size_;
|
|
};
|
|
union { Impl_ _impl_; };
|
|
friend struct ::TableStruct_xla_2fxla_5fdata_2eproto;
|
|
};
|
|
// -------------------------------------------------------------------
|
|
|
|
class TriangularSolveOptions final :
|
|
public ::PROTOBUF_NAMESPACE_ID::Message /* @@protoc_insertion_point(class_definition:xla.TriangularSolveOptions) */ {
|
|
public:
|
|
inline TriangularSolveOptions() : TriangularSolveOptions(nullptr) {}
|
|
~TriangularSolveOptions() override;
|
|
explicit PROTOBUF_CONSTEXPR TriangularSolveOptions(::PROTOBUF_NAMESPACE_ID::internal::ConstantInitialized);
|
|
|
|
TriangularSolveOptions(const TriangularSolveOptions& from);
|
|
TriangularSolveOptions(TriangularSolveOptions&& from) noexcept
|
|
: TriangularSolveOptions() {
|
|
*this = ::std::move(from);
|
|
}
|
|
|
|
inline TriangularSolveOptions& operator=(const TriangularSolveOptions& from) {
|
|
CopyFrom(from);
|
|
return *this;
|
|
}
|
|
inline TriangularSolveOptions& operator=(TriangularSolveOptions&& from) noexcept {
|
|
if (this == &from) return *this;
|
|
if (GetOwningArena() == from.GetOwningArena()
|
|
#ifdef PROTOBUF_FORCE_COPY_IN_MOVE
|
|
&& GetOwningArena() != nullptr
|
|
#endif // !PROTOBUF_FORCE_COPY_IN_MOVE
|
|
) {
|
|
InternalSwap(&from);
|
|
} else {
|
|
CopyFrom(from);
|
|
}
|
|
return *this;
|
|
}
|
|
|
|
static const ::PROTOBUF_NAMESPACE_ID::Descriptor* descriptor() {
|
|
return GetDescriptor();
|
|
}
|
|
static const ::PROTOBUF_NAMESPACE_ID::Descriptor* GetDescriptor() {
|
|
return default_instance().GetMetadata().descriptor;
|
|
}
|
|
static const ::PROTOBUF_NAMESPACE_ID::Reflection* GetReflection() {
|
|
return default_instance().GetMetadata().reflection;
|
|
}
|
|
static const TriangularSolveOptions& default_instance() {
|
|
return *internal_default_instance();
|
|
}
|
|
static inline const TriangularSolveOptions* internal_default_instance() {
|
|
return reinterpret_cast<const TriangularSolveOptions*>(
|
|
&_TriangularSolveOptions_default_instance_);
|
|
}
|
|
static constexpr int kIndexInFileMessages =
|
|
24;
|
|
|
|
friend void swap(TriangularSolveOptions& a, TriangularSolveOptions& b) {
|
|
a.Swap(&b);
|
|
}
|
|
inline void Swap(TriangularSolveOptions* other) {
|
|
if (other == this) return;
|
|
#ifdef PROTOBUF_FORCE_COPY_IN_SWAP
|
|
if (GetOwningArena() != nullptr &&
|
|
GetOwningArena() == other->GetOwningArena()) {
|
|
#else // PROTOBUF_FORCE_COPY_IN_SWAP
|
|
if (GetOwningArena() == other->GetOwningArena()) {
|
|
#endif // !PROTOBUF_FORCE_COPY_IN_SWAP
|
|
InternalSwap(other);
|
|
} else {
|
|
::PROTOBUF_NAMESPACE_ID::internal::GenericSwap(this, other);
|
|
}
|
|
}
|
|
void UnsafeArenaSwap(TriangularSolveOptions* other) {
|
|
if (other == this) return;
|
|
GOOGLE_DCHECK(GetOwningArena() == other->GetOwningArena());
|
|
InternalSwap(other);
|
|
}
|
|
|
|
// implements Message ----------------------------------------------
|
|
|
|
TriangularSolveOptions* New(::PROTOBUF_NAMESPACE_ID::Arena* arena = nullptr) const final {
|
|
return CreateMaybeMessage<TriangularSolveOptions>(arena);
|
|
}
|
|
using ::PROTOBUF_NAMESPACE_ID::Message::CopyFrom;
|
|
void CopyFrom(const TriangularSolveOptions& from);
|
|
using ::PROTOBUF_NAMESPACE_ID::Message::MergeFrom;
|
|
void MergeFrom( const TriangularSolveOptions& from) {
|
|
TriangularSolveOptions::MergeImpl(*this, from);
|
|
}
|
|
private:
|
|
static void MergeImpl(::PROTOBUF_NAMESPACE_ID::Message& to_msg, const ::PROTOBUF_NAMESPACE_ID::Message& from_msg);
|
|
public:
|
|
PROTOBUF_ATTRIBUTE_REINITIALIZES void Clear() final;
|
|
bool IsInitialized() const final;
|
|
|
|
size_t ByteSizeLong() const final;
|
|
const char* _InternalParse(const char* ptr, ::PROTOBUF_NAMESPACE_ID::internal::ParseContext* ctx) final;
|
|
uint8_t* _InternalSerialize(
|
|
uint8_t* target, ::PROTOBUF_NAMESPACE_ID::io::EpsCopyOutputStream* stream) const final;
|
|
int GetCachedSize() const final { return _impl_._cached_size_.Get(); }
|
|
|
|
private:
|
|
void SharedCtor(::PROTOBUF_NAMESPACE_ID::Arena* arena, bool is_message_owned);
|
|
void SharedDtor();
|
|
void SetCachedSize(int size) const final;
|
|
void InternalSwap(TriangularSolveOptions* other);
|
|
|
|
private:
|
|
friend class ::PROTOBUF_NAMESPACE_ID::internal::AnyMetadata;
|
|
static ::PROTOBUF_NAMESPACE_ID::StringPiece FullMessageName() {
|
|
return "xla.TriangularSolveOptions";
|
|
}
|
|
protected:
|
|
explicit TriangularSolveOptions(::PROTOBUF_NAMESPACE_ID::Arena* arena,
|
|
bool is_message_owned = false);
|
|
public:
|
|
|
|
static const ClassData _class_data_;
|
|
const ::PROTOBUF_NAMESPACE_ID::Message::ClassData*GetClassData() const final;
|
|
|
|
::PROTOBUF_NAMESPACE_ID::Metadata GetMetadata() const final;
|
|
|
|
// nested types ----------------------------------------------------
|
|
|
|
typedef TriangularSolveOptions_Transpose Transpose;
|
|
static constexpr Transpose TRANSPOSE_INVALID =
|
|
TriangularSolveOptions_Transpose_TRANSPOSE_INVALID;
|
|
static constexpr Transpose NO_TRANSPOSE =
|
|
TriangularSolveOptions_Transpose_NO_TRANSPOSE;
|
|
static constexpr Transpose TRANSPOSE =
|
|
TriangularSolveOptions_Transpose_TRANSPOSE;
|
|
static constexpr Transpose ADJOINT =
|
|
TriangularSolveOptions_Transpose_ADJOINT;
|
|
static inline bool Transpose_IsValid(int value) {
|
|
return TriangularSolveOptions_Transpose_IsValid(value);
|
|
}
|
|
static constexpr Transpose Transpose_MIN =
|
|
TriangularSolveOptions_Transpose_Transpose_MIN;
|
|
static constexpr Transpose Transpose_MAX =
|
|
TriangularSolveOptions_Transpose_Transpose_MAX;
|
|
static constexpr int Transpose_ARRAYSIZE =
|
|
TriangularSolveOptions_Transpose_Transpose_ARRAYSIZE;
|
|
static inline const ::PROTOBUF_NAMESPACE_ID::EnumDescriptor*
|
|
Transpose_descriptor() {
|
|
return TriangularSolveOptions_Transpose_descriptor();
|
|
}
|
|
template<typename T>
|
|
static inline const std::string& Transpose_Name(T enum_t_value) {
|
|
static_assert(::std::is_same<T, Transpose>::value ||
|
|
::std::is_integral<T>::value,
|
|
"Incorrect type passed to function Transpose_Name.");
|
|
return TriangularSolveOptions_Transpose_Name(enum_t_value);
|
|
}
|
|
static inline bool Transpose_Parse(::PROTOBUF_NAMESPACE_ID::ConstStringParam name,
|
|
Transpose* value) {
|
|
return TriangularSolveOptions_Transpose_Parse(name, value);
|
|
}
|
|
|
|
// accessors -------------------------------------------------------
|
|
|
|
enum : int {
|
|
kLeftSideFieldNumber = 1,
|
|
kLowerFieldNumber = 2,
|
|
kUnitDiagonalFieldNumber = 3,
|
|
kTransposeAFieldNumber = 4,
|
|
};
|
|
// bool left_side = 1;
|
|
void clear_left_side();
|
|
bool left_side() const;
|
|
void set_left_side(bool value);
|
|
private:
|
|
bool _internal_left_side() const;
|
|
void _internal_set_left_side(bool value);
|
|
public:
|
|
|
|
// bool lower = 2;
|
|
void clear_lower();
|
|
bool lower() const;
|
|
void set_lower(bool value);
|
|
private:
|
|
bool _internal_lower() const;
|
|
void _internal_set_lower(bool value);
|
|
public:
|
|
|
|
// bool unit_diagonal = 3;
|
|
void clear_unit_diagonal();
|
|
bool unit_diagonal() const;
|
|
void set_unit_diagonal(bool value);
|
|
private:
|
|
bool _internal_unit_diagonal() const;
|
|
void _internal_set_unit_diagonal(bool value);
|
|
public:
|
|
|
|
// .xla.TriangularSolveOptions.Transpose transpose_a = 4;
|
|
void clear_transpose_a();
|
|
::xla::TriangularSolveOptions_Transpose transpose_a() const;
|
|
void set_transpose_a(::xla::TriangularSolveOptions_Transpose value);
|
|
private:
|
|
::xla::TriangularSolveOptions_Transpose _internal_transpose_a() const;
|
|
void _internal_set_transpose_a(::xla::TriangularSolveOptions_Transpose value);
|
|
public:
|
|
|
|
// @@protoc_insertion_point(class_scope:xla.TriangularSolveOptions)
|
|
private:
|
|
class _Internal;
|
|
|
|
template <typename T> friend class ::PROTOBUF_NAMESPACE_ID::Arena::InternalHelper;
|
|
typedef void InternalArenaConstructable_;
|
|
typedef void DestructorSkippable_;
|
|
struct Impl_ {
|
|
bool left_side_;
|
|
bool lower_;
|
|
bool unit_diagonal_;
|
|
int transpose_a_;
|
|
mutable ::PROTOBUF_NAMESPACE_ID::internal::CachedSize _cached_size_;
|
|
};
|
|
union { Impl_ _impl_; };
|
|
friend struct ::TableStruct_xla_2fxla_5fdata_2eproto;
|
|
};
|
|
// -------------------------------------------------------------------
|
|
|
|
class CholeskyOptions final :
|
|
public ::PROTOBUF_NAMESPACE_ID::Message /* @@protoc_insertion_point(class_definition:xla.CholeskyOptions) */ {
|
|
public:
|
|
inline CholeskyOptions() : CholeskyOptions(nullptr) {}
|
|
~CholeskyOptions() override;
|
|
explicit PROTOBUF_CONSTEXPR CholeskyOptions(::PROTOBUF_NAMESPACE_ID::internal::ConstantInitialized);
|
|
|
|
CholeskyOptions(const CholeskyOptions& from);
|
|
CholeskyOptions(CholeskyOptions&& from) noexcept
|
|
: CholeskyOptions() {
|
|
*this = ::std::move(from);
|
|
}
|
|
|
|
inline CholeskyOptions& operator=(const CholeskyOptions& from) {
|
|
CopyFrom(from);
|
|
return *this;
|
|
}
|
|
inline CholeskyOptions& operator=(CholeskyOptions&& from) noexcept {
|
|
if (this == &from) return *this;
|
|
if (GetOwningArena() == from.GetOwningArena()
|
|
#ifdef PROTOBUF_FORCE_COPY_IN_MOVE
|
|
&& GetOwningArena() != nullptr
|
|
#endif // !PROTOBUF_FORCE_COPY_IN_MOVE
|
|
) {
|
|
InternalSwap(&from);
|
|
} else {
|
|
CopyFrom(from);
|
|
}
|
|
return *this;
|
|
}
|
|
|
|
static const ::PROTOBUF_NAMESPACE_ID::Descriptor* descriptor() {
|
|
return GetDescriptor();
|
|
}
|
|
static const ::PROTOBUF_NAMESPACE_ID::Descriptor* GetDescriptor() {
|
|
return default_instance().GetMetadata().descriptor;
|
|
}
|
|
static const ::PROTOBUF_NAMESPACE_ID::Reflection* GetReflection() {
|
|
return default_instance().GetMetadata().reflection;
|
|
}
|
|
static const CholeskyOptions& default_instance() {
|
|
return *internal_default_instance();
|
|
}
|
|
static inline const CholeskyOptions* internal_default_instance() {
|
|
return reinterpret_cast<const CholeskyOptions*>(
|
|
&_CholeskyOptions_default_instance_);
|
|
}
|
|
static constexpr int kIndexInFileMessages =
|
|
25;
|
|
|
|
friend void swap(CholeskyOptions& a, CholeskyOptions& b) {
|
|
a.Swap(&b);
|
|
}
|
|
inline void Swap(CholeskyOptions* other) {
|
|
if (other == this) return;
|
|
#ifdef PROTOBUF_FORCE_COPY_IN_SWAP
|
|
if (GetOwningArena() != nullptr &&
|
|
GetOwningArena() == other->GetOwningArena()) {
|
|
#else // PROTOBUF_FORCE_COPY_IN_SWAP
|
|
if (GetOwningArena() == other->GetOwningArena()) {
|
|
#endif // !PROTOBUF_FORCE_COPY_IN_SWAP
|
|
InternalSwap(other);
|
|
} else {
|
|
::PROTOBUF_NAMESPACE_ID::internal::GenericSwap(this, other);
|
|
}
|
|
}
|
|
void UnsafeArenaSwap(CholeskyOptions* other) {
|
|
if (other == this) return;
|
|
GOOGLE_DCHECK(GetOwningArena() == other->GetOwningArena());
|
|
InternalSwap(other);
|
|
}
|
|
|
|
// implements Message ----------------------------------------------
|
|
|
|
CholeskyOptions* New(::PROTOBUF_NAMESPACE_ID::Arena* arena = nullptr) const final {
|
|
return CreateMaybeMessage<CholeskyOptions>(arena);
|
|
}
|
|
using ::PROTOBUF_NAMESPACE_ID::Message::CopyFrom;
|
|
void CopyFrom(const CholeskyOptions& from);
|
|
using ::PROTOBUF_NAMESPACE_ID::Message::MergeFrom;
|
|
void MergeFrom( const CholeskyOptions& from) {
|
|
CholeskyOptions::MergeImpl(*this, from);
|
|
}
|
|
private:
|
|
static void MergeImpl(::PROTOBUF_NAMESPACE_ID::Message& to_msg, const ::PROTOBUF_NAMESPACE_ID::Message& from_msg);
|
|
public:
|
|
PROTOBUF_ATTRIBUTE_REINITIALIZES void Clear() final;
|
|
bool IsInitialized() const final;
|
|
|
|
size_t ByteSizeLong() const final;
|
|
const char* _InternalParse(const char* ptr, ::PROTOBUF_NAMESPACE_ID::internal::ParseContext* ctx) final;
|
|
uint8_t* _InternalSerialize(
|
|
uint8_t* target, ::PROTOBUF_NAMESPACE_ID::io::EpsCopyOutputStream* stream) const final;
|
|
int GetCachedSize() const final { return _impl_._cached_size_.Get(); }
|
|
|
|
private:
|
|
void SharedCtor(::PROTOBUF_NAMESPACE_ID::Arena* arena, bool is_message_owned);
|
|
void SharedDtor();
|
|
void SetCachedSize(int size) const final;
|
|
void InternalSwap(CholeskyOptions* other);
|
|
|
|
private:
|
|
friend class ::PROTOBUF_NAMESPACE_ID::internal::AnyMetadata;
|
|
static ::PROTOBUF_NAMESPACE_ID::StringPiece FullMessageName() {
|
|
return "xla.CholeskyOptions";
|
|
}
|
|
protected:
|
|
explicit CholeskyOptions(::PROTOBUF_NAMESPACE_ID::Arena* arena,
|
|
bool is_message_owned = false);
|
|
public:
|
|
|
|
static const ClassData _class_data_;
|
|
const ::PROTOBUF_NAMESPACE_ID::Message::ClassData*GetClassData() const final;
|
|
|
|
::PROTOBUF_NAMESPACE_ID::Metadata GetMetadata() const final;
|
|
|
|
// nested types ----------------------------------------------------
|
|
|
|
// accessors -------------------------------------------------------
|
|
|
|
enum : int {
|
|
kLowerFieldNumber = 1,
|
|
};
|
|
// bool lower = 1;
|
|
void clear_lower();
|
|
bool lower() const;
|
|
void set_lower(bool value);
|
|
private:
|
|
bool _internal_lower() const;
|
|
void _internal_set_lower(bool value);
|
|
public:
|
|
|
|
// @@protoc_insertion_point(class_scope:xla.CholeskyOptions)
|
|
private:
|
|
class _Internal;
|
|
|
|
template <typename T> friend class ::PROTOBUF_NAMESPACE_ID::Arena::InternalHelper;
|
|
typedef void InternalArenaConstructable_;
|
|
typedef void DestructorSkippable_;
|
|
struct Impl_ {
|
|
bool lower_;
|
|
mutable ::PROTOBUF_NAMESPACE_ID::internal::CachedSize _cached_size_;
|
|
};
|
|
union { Impl_ _impl_; };
|
|
friend struct ::TableStruct_xla_2fxla_5fdata_2eproto;
|
|
};
|
|
// -------------------------------------------------------------------
|
|
|
|
class SortOptions final :
|
|
public ::PROTOBUF_NAMESPACE_ID::Message /* @@protoc_insertion_point(class_definition:xla.SortOptions) */ {
|
|
public:
|
|
inline SortOptions() : SortOptions(nullptr) {}
|
|
~SortOptions() override;
|
|
explicit PROTOBUF_CONSTEXPR SortOptions(::PROTOBUF_NAMESPACE_ID::internal::ConstantInitialized);
|
|
|
|
SortOptions(const SortOptions& from);
|
|
SortOptions(SortOptions&& from) noexcept
|
|
: SortOptions() {
|
|
*this = ::std::move(from);
|
|
}
|
|
|
|
inline SortOptions& operator=(const SortOptions& from) {
|
|
CopyFrom(from);
|
|
return *this;
|
|
}
|
|
inline SortOptions& operator=(SortOptions&& from) noexcept {
|
|
if (this == &from) return *this;
|
|
if (GetOwningArena() == from.GetOwningArena()
|
|
#ifdef PROTOBUF_FORCE_COPY_IN_MOVE
|
|
&& GetOwningArena() != nullptr
|
|
#endif // !PROTOBUF_FORCE_COPY_IN_MOVE
|
|
) {
|
|
InternalSwap(&from);
|
|
} else {
|
|
CopyFrom(from);
|
|
}
|
|
return *this;
|
|
}
|
|
|
|
static const ::PROTOBUF_NAMESPACE_ID::Descriptor* descriptor() {
|
|
return GetDescriptor();
|
|
}
|
|
static const ::PROTOBUF_NAMESPACE_ID::Descriptor* GetDescriptor() {
|
|
return default_instance().GetMetadata().descriptor;
|
|
}
|
|
static const ::PROTOBUF_NAMESPACE_ID::Reflection* GetReflection() {
|
|
return default_instance().GetMetadata().reflection;
|
|
}
|
|
static const SortOptions& default_instance() {
|
|
return *internal_default_instance();
|
|
}
|
|
static inline const SortOptions* internal_default_instance() {
|
|
return reinterpret_cast<const SortOptions*>(
|
|
&_SortOptions_default_instance_);
|
|
}
|
|
static constexpr int kIndexInFileMessages =
|
|
26;
|
|
|
|
friend void swap(SortOptions& a, SortOptions& b) {
|
|
a.Swap(&b);
|
|
}
|
|
inline void Swap(SortOptions* other) {
|
|
if (other == this) return;
|
|
#ifdef PROTOBUF_FORCE_COPY_IN_SWAP
|
|
if (GetOwningArena() != nullptr &&
|
|
GetOwningArena() == other->GetOwningArena()) {
|
|
#else // PROTOBUF_FORCE_COPY_IN_SWAP
|
|
if (GetOwningArena() == other->GetOwningArena()) {
|
|
#endif // !PROTOBUF_FORCE_COPY_IN_SWAP
|
|
InternalSwap(other);
|
|
} else {
|
|
::PROTOBUF_NAMESPACE_ID::internal::GenericSwap(this, other);
|
|
}
|
|
}
|
|
void UnsafeArenaSwap(SortOptions* other) {
|
|
if (other == this) return;
|
|
GOOGLE_DCHECK(GetOwningArena() == other->GetOwningArena());
|
|
InternalSwap(other);
|
|
}
|
|
|
|
// implements Message ----------------------------------------------
|
|
|
|
SortOptions* New(::PROTOBUF_NAMESPACE_ID::Arena* arena = nullptr) const final {
|
|
return CreateMaybeMessage<SortOptions>(arena);
|
|
}
|
|
using ::PROTOBUF_NAMESPACE_ID::Message::CopyFrom;
|
|
void CopyFrom(const SortOptions& from);
|
|
using ::PROTOBUF_NAMESPACE_ID::Message::MergeFrom;
|
|
void MergeFrom( const SortOptions& from) {
|
|
SortOptions::MergeImpl(*this, from);
|
|
}
|
|
private:
|
|
static void MergeImpl(::PROTOBUF_NAMESPACE_ID::Message& to_msg, const ::PROTOBUF_NAMESPACE_ID::Message& from_msg);
|
|
public:
|
|
PROTOBUF_ATTRIBUTE_REINITIALIZES void Clear() final;
|
|
bool IsInitialized() const final;
|
|
|
|
size_t ByteSizeLong() const final;
|
|
const char* _InternalParse(const char* ptr, ::PROTOBUF_NAMESPACE_ID::internal::ParseContext* ctx) final;
|
|
uint8_t* _InternalSerialize(
|
|
uint8_t* target, ::PROTOBUF_NAMESPACE_ID::io::EpsCopyOutputStream* stream) const final;
|
|
int GetCachedSize() const final { return _impl_._cached_size_.Get(); }
|
|
|
|
private:
|
|
void SharedCtor(::PROTOBUF_NAMESPACE_ID::Arena* arena, bool is_message_owned);
|
|
void SharedDtor();
|
|
void SetCachedSize(int size) const final;
|
|
void InternalSwap(SortOptions* other);
|
|
|
|
private:
|
|
friend class ::PROTOBUF_NAMESPACE_ID::internal::AnyMetadata;
|
|
static ::PROTOBUF_NAMESPACE_ID::StringPiece FullMessageName() {
|
|
return "xla.SortOptions";
|
|
}
|
|
protected:
|
|
explicit SortOptions(::PROTOBUF_NAMESPACE_ID::Arena* arena,
|
|
bool is_message_owned = false);
|
|
public:
|
|
|
|
static const ClassData _class_data_;
|
|
const ::PROTOBUF_NAMESPACE_ID::Message::ClassData*GetClassData() const final;
|
|
|
|
::PROTOBUF_NAMESPACE_ID::Metadata GetMetadata() const final;
|
|
|
|
// nested types ----------------------------------------------------
|
|
|
|
// accessors -------------------------------------------------------
|
|
|
|
enum : int {
|
|
kDescendingFieldNumber = 1,
|
|
};
|
|
// bool descending = 1;
|
|
void clear_descending();
|
|
bool descending() const;
|
|
void set_descending(bool value);
|
|
private:
|
|
bool _internal_descending() const;
|
|
void _internal_set_descending(bool value);
|
|
public:
|
|
|
|
// @@protoc_insertion_point(class_scope:xla.SortOptions)
|
|
private:
|
|
class _Internal;
|
|
|
|
template <typename T> friend class ::PROTOBUF_NAMESPACE_ID::Arena::InternalHelper;
|
|
typedef void InternalArenaConstructable_;
|
|
typedef void DestructorSkippable_;
|
|
struct Impl_ {
|
|
bool descending_;
|
|
mutable ::PROTOBUF_NAMESPACE_ID::internal::CachedSize _cached_size_;
|
|
};
|
|
union { Impl_ _impl_; };
|
|
friend struct ::TableStruct_xla_2fxla_5fdata_2eproto;
|
|
};
|
|
// -------------------------------------------------------------------
|
|
|
|
class FrontendAttributes_MapEntry_DoNotUse : public ::PROTOBUF_NAMESPACE_ID::internal::MapEntry<FrontendAttributes_MapEntry_DoNotUse,
|
|
std::string, std::string,
|
|
::PROTOBUF_NAMESPACE_ID::internal::WireFormatLite::TYPE_STRING,
|
|
::PROTOBUF_NAMESPACE_ID::internal::WireFormatLite::TYPE_STRING> {
|
|
public:
|
|
typedef ::PROTOBUF_NAMESPACE_ID::internal::MapEntry<FrontendAttributes_MapEntry_DoNotUse,
|
|
std::string, std::string,
|
|
::PROTOBUF_NAMESPACE_ID::internal::WireFormatLite::TYPE_STRING,
|
|
::PROTOBUF_NAMESPACE_ID::internal::WireFormatLite::TYPE_STRING> SuperType;
|
|
FrontendAttributes_MapEntry_DoNotUse();
|
|
explicit PROTOBUF_CONSTEXPR FrontendAttributes_MapEntry_DoNotUse(
|
|
::PROTOBUF_NAMESPACE_ID::internal::ConstantInitialized);
|
|
explicit FrontendAttributes_MapEntry_DoNotUse(::PROTOBUF_NAMESPACE_ID::Arena* arena);
|
|
void MergeFrom(const FrontendAttributes_MapEntry_DoNotUse& other);
|
|
static const FrontendAttributes_MapEntry_DoNotUse* internal_default_instance() { return reinterpret_cast<const FrontendAttributes_MapEntry_DoNotUse*>(&_FrontendAttributes_MapEntry_DoNotUse_default_instance_); }
|
|
static bool ValidateKey(std::string* s) {
|
|
return ::PROTOBUF_NAMESPACE_ID::internal::WireFormatLite::VerifyUtf8String(s->data(), static_cast<int>(s->size()), ::PROTOBUF_NAMESPACE_ID::internal::WireFormatLite::PARSE, "xla.FrontendAttributes.MapEntry.key");
|
|
}
|
|
static bool ValidateValue(std::string* s) {
|
|
return ::PROTOBUF_NAMESPACE_ID::internal::WireFormatLite::VerifyUtf8String(s->data(), static_cast<int>(s->size()), ::PROTOBUF_NAMESPACE_ID::internal::WireFormatLite::PARSE, "xla.FrontendAttributes.MapEntry.value");
|
|
}
|
|
using ::PROTOBUF_NAMESPACE_ID::Message::MergeFrom;
|
|
::PROTOBUF_NAMESPACE_ID::Metadata GetMetadata() const final;
|
|
friend struct ::TableStruct_xla_2fxla_5fdata_2eproto;
|
|
};
|
|
|
|
// -------------------------------------------------------------------
|
|
|
|
class FrontendAttributes final :
|
|
public ::PROTOBUF_NAMESPACE_ID::Message /* @@protoc_insertion_point(class_definition:xla.FrontendAttributes) */ {
|
|
public:
|
|
inline FrontendAttributes() : FrontendAttributes(nullptr) {}
|
|
~FrontendAttributes() override;
|
|
explicit PROTOBUF_CONSTEXPR FrontendAttributes(::PROTOBUF_NAMESPACE_ID::internal::ConstantInitialized);
|
|
|
|
FrontendAttributes(const FrontendAttributes& from);
|
|
FrontendAttributes(FrontendAttributes&& from) noexcept
|
|
: FrontendAttributes() {
|
|
*this = ::std::move(from);
|
|
}
|
|
|
|
inline FrontendAttributes& operator=(const FrontendAttributes& from) {
|
|
CopyFrom(from);
|
|
return *this;
|
|
}
|
|
inline FrontendAttributes& operator=(FrontendAttributes&& from) noexcept {
|
|
if (this == &from) return *this;
|
|
if (GetOwningArena() == from.GetOwningArena()
|
|
#ifdef PROTOBUF_FORCE_COPY_IN_MOVE
|
|
&& GetOwningArena() != nullptr
|
|
#endif // !PROTOBUF_FORCE_COPY_IN_MOVE
|
|
) {
|
|
InternalSwap(&from);
|
|
} else {
|
|
CopyFrom(from);
|
|
}
|
|
return *this;
|
|
}
|
|
|
|
static const ::PROTOBUF_NAMESPACE_ID::Descriptor* descriptor() {
|
|
return GetDescriptor();
|
|
}
|
|
static const ::PROTOBUF_NAMESPACE_ID::Descriptor* GetDescriptor() {
|
|
return default_instance().GetMetadata().descriptor;
|
|
}
|
|
static const ::PROTOBUF_NAMESPACE_ID::Reflection* GetReflection() {
|
|
return default_instance().GetMetadata().reflection;
|
|
}
|
|
static const FrontendAttributes& default_instance() {
|
|
return *internal_default_instance();
|
|
}
|
|
static inline const FrontendAttributes* internal_default_instance() {
|
|
return reinterpret_cast<const FrontendAttributes*>(
|
|
&_FrontendAttributes_default_instance_);
|
|
}
|
|
static constexpr int kIndexInFileMessages =
|
|
28;
|
|
|
|
friend void swap(FrontendAttributes& a, FrontendAttributes& b) {
|
|
a.Swap(&b);
|
|
}
|
|
inline void Swap(FrontendAttributes* other) {
|
|
if (other == this) return;
|
|
#ifdef PROTOBUF_FORCE_COPY_IN_SWAP
|
|
if (GetOwningArena() != nullptr &&
|
|
GetOwningArena() == other->GetOwningArena()) {
|
|
#else // PROTOBUF_FORCE_COPY_IN_SWAP
|
|
if (GetOwningArena() == other->GetOwningArena()) {
|
|
#endif // !PROTOBUF_FORCE_COPY_IN_SWAP
|
|
InternalSwap(other);
|
|
} else {
|
|
::PROTOBUF_NAMESPACE_ID::internal::GenericSwap(this, other);
|
|
}
|
|
}
|
|
void UnsafeArenaSwap(FrontendAttributes* other) {
|
|
if (other == this) return;
|
|
GOOGLE_DCHECK(GetOwningArena() == other->GetOwningArena());
|
|
InternalSwap(other);
|
|
}
|
|
|
|
// implements Message ----------------------------------------------
|
|
|
|
FrontendAttributes* New(::PROTOBUF_NAMESPACE_ID::Arena* arena = nullptr) const final {
|
|
return CreateMaybeMessage<FrontendAttributes>(arena);
|
|
}
|
|
using ::PROTOBUF_NAMESPACE_ID::Message::CopyFrom;
|
|
void CopyFrom(const FrontendAttributes& from);
|
|
using ::PROTOBUF_NAMESPACE_ID::Message::MergeFrom;
|
|
void MergeFrom( const FrontendAttributes& from) {
|
|
FrontendAttributes::MergeImpl(*this, from);
|
|
}
|
|
private:
|
|
static void MergeImpl(::PROTOBUF_NAMESPACE_ID::Message& to_msg, const ::PROTOBUF_NAMESPACE_ID::Message& from_msg);
|
|
public:
|
|
PROTOBUF_ATTRIBUTE_REINITIALIZES void Clear() final;
|
|
bool IsInitialized() const final;
|
|
|
|
size_t ByteSizeLong() const final;
|
|
const char* _InternalParse(const char* ptr, ::PROTOBUF_NAMESPACE_ID::internal::ParseContext* ctx) final;
|
|
uint8_t* _InternalSerialize(
|
|
uint8_t* target, ::PROTOBUF_NAMESPACE_ID::io::EpsCopyOutputStream* stream) const final;
|
|
int GetCachedSize() const final { return _impl_._cached_size_.Get(); }
|
|
|
|
private:
|
|
void SharedCtor(::PROTOBUF_NAMESPACE_ID::Arena* arena, bool is_message_owned);
|
|
void SharedDtor();
|
|
void SetCachedSize(int size) const final;
|
|
void InternalSwap(FrontendAttributes* other);
|
|
|
|
private:
|
|
friend class ::PROTOBUF_NAMESPACE_ID::internal::AnyMetadata;
|
|
static ::PROTOBUF_NAMESPACE_ID::StringPiece FullMessageName() {
|
|
return "xla.FrontendAttributes";
|
|
}
|
|
protected:
|
|
explicit FrontendAttributes(::PROTOBUF_NAMESPACE_ID::Arena* arena,
|
|
bool is_message_owned = false);
|
|
private:
|
|
static void ArenaDtor(void* object);
|
|
public:
|
|
|
|
static const ClassData _class_data_;
|
|
const ::PROTOBUF_NAMESPACE_ID::Message::ClassData*GetClassData() const final;
|
|
|
|
::PROTOBUF_NAMESPACE_ID::Metadata GetMetadata() const final;
|
|
|
|
// nested types ----------------------------------------------------
|
|
|
|
|
|
// accessors -------------------------------------------------------
|
|
|
|
enum : int {
|
|
kMapFieldNumber = 1,
|
|
};
|
|
// map<string, string> map = 1;
|
|
int map_size() const;
|
|
private:
|
|
int _internal_map_size() const;
|
|
public:
|
|
void clear_map();
|
|
private:
|
|
const ::PROTOBUF_NAMESPACE_ID::Map< std::string, std::string >&
|
|
_internal_map() const;
|
|
::PROTOBUF_NAMESPACE_ID::Map< std::string, std::string >*
|
|
_internal_mutable_map();
|
|
public:
|
|
const ::PROTOBUF_NAMESPACE_ID::Map< std::string, std::string >&
|
|
map() const;
|
|
::PROTOBUF_NAMESPACE_ID::Map< std::string, std::string >*
|
|
mutable_map();
|
|
|
|
// @@protoc_insertion_point(class_scope:xla.FrontendAttributes)
|
|
private:
|
|
class _Internal;
|
|
|
|
template <typename T> friend class ::PROTOBUF_NAMESPACE_ID::Arena::InternalHelper;
|
|
typedef void InternalArenaConstructable_;
|
|
typedef void DestructorSkippable_;
|
|
struct Impl_ {
|
|
::PROTOBUF_NAMESPACE_ID::internal::MapField<
|
|
FrontendAttributes_MapEntry_DoNotUse,
|
|
std::string, std::string,
|
|
::PROTOBUF_NAMESPACE_ID::internal::WireFormatLite::TYPE_STRING,
|
|
::PROTOBUF_NAMESPACE_ID::internal::WireFormatLite::TYPE_STRING> map_;
|
|
mutable ::PROTOBUF_NAMESPACE_ID::internal::CachedSize _cached_size_;
|
|
};
|
|
union { Impl_ _impl_; };
|
|
friend struct ::TableStruct_xla_2fxla_5fdata_2eproto;
|
|
};
|
|
// -------------------------------------------------------------------
|
|
|
|
class Statistic final :
|
|
public ::PROTOBUF_NAMESPACE_ID::Message /* @@protoc_insertion_point(class_definition:xla.Statistic) */ {
|
|
public:
|
|
inline Statistic() : Statistic(nullptr) {}
|
|
~Statistic() override;
|
|
explicit PROTOBUF_CONSTEXPR Statistic(::PROTOBUF_NAMESPACE_ID::internal::ConstantInitialized);
|
|
|
|
Statistic(const Statistic& from);
|
|
Statistic(Statistic&& from) noexcept
|
|
: Statistic() {
|
|
*this = ::std::move(from);
|
|
}
|
|
|
|
inline Statistic& operator=(const Statistic& from) {
|
|
CopyFrom(from);
|
|
return *this;
|
|
}
|
|
inline Statistic& operator=(Statistic&& from) noexcept {
|
|
if (this == &from) return *this;
|
|
if (GetOwningArena() == from.GetOwningArena()
|
|
#ifdef PROTOBUF_FORCE_COPY_IN_MOVE
|
|
&& GetOwningArena() != nullptr
|
|
#endif // !PROTOBUF_FORCE_COPY_IN_MOVE
|
|
) {
|
|
InternalSwap(&from);
|
|
} else {
|
|
CopyFrom(from);
|
|
}
|
|
return *this;
|
|
}
|
|
|
|
static const ::PROTOBUF_NAMESPACE_ID::Descriptor* descriptor() {
|
|
return GetDescriptor();
|
|
}
|
|
static const ::PROTOBUF_NAMESPACE_ID::Descriptor* GetDescriptor() {
|
|
return default_instance().GetMetadata().descriptor;
|
|
}
|
|
static const ::PROTOBUF_NAMESPACE_ID::Reflection* GetReflection() {
|
|
return default_instance().GetMetadata().reflection;
|
|
}
|
|
static const Statistic& default_instance() {
|
|
return *internal_default_instance();
|
|
}
|
|
static inline const Statistic* internal_default_instance() {
|
|
return reinterpret_cast<const Statistic*>(
|
|
&_Statistic_default_instance_);
|
|
}
|
|
static constexpr int kIndexInFileMessages =
|
|
29;
|
|
|
|
friend void swap(Statistic& a, Statistic& b) {
|
|
a.Swap(&b);
|
|
}
|
|
inline void Swap(Statistic* other) {
|
|
if (other == this) return;
|
|
#ifdef PROTOBUF_FORCE_COPY_IN_SWAP
|
|
if (GetOwningArena() != nullptr &&
|
|
GetOwningArena() == other->GetOwningArena()) {
|
|
#else // PROTOBUF_FORCE_COPY_IN_SWAP
|
|
if (GetOwningArena() == other->GetOwningArena()) {
|
|
#endif // !PROTOBUF_FORCE_COPY_IN_SWAP
|
|
InternalSwap(other);
|
|
} else {
|
|
::PROTOBUF_NAMESPACE_ID::internal::GenericSwap(this, other);
|
|
}
|
|
}
|
|
void UnsafeArenaSwap(Statistic* other) {
|
|
if (other == this) return;
|
|
GOOGLE_DCHECK(GetOwningArena() == other->GetOwningArena());
|
|
InternalSwap(other);
|
|
}
|
|
|
|
// implements Message ----------------------------------------------
|
|
|
|
Statistic* New(::PROTOBUF_NAMESPACE_ID::Arena* arena = nullptr) const final {
|
|
return CreateMaybeMessage<Statistic>(arena);
|
|
}
|
|
using ::PROTOBUF_NAMESPACE_ID::Message::CopyFrom;
|
|
void CopyFrom(const Statistic& from);
|
|
using ::PROTOBUF_NAMESPACE_ID::Message::MergeFrom;
|
|
void MergeFrom( const Statistic& from) {
|
|
Statistic::MergeImpl(*this, from);
|
|
}
|
|
private:
|
|
static void MergeImpl(::PROTOBUF_NAMESPACE_ID::Message& to_msg, const ::PROTOBUF_NAMESPACE_ID::Message& from_msg);
|
|
public:
|
|
PROTOBUF_ATTRIBUTE_REINITIALIZES void Clear() final;
|
|
bool IsInitialized() const final;
|
|
|
|
size_t ByteSizeLong() const final;
|
|
const char* _InternalParse(const char* ptr, ::PROTOBUF_NAMESPACE_ID::internal::ParseContext* ctx) final;
|
|
uint8_t* _InternalSerialize(
|
|
uint8_t* target, ::PROTOBUF_NAMESPACE_ID::io::EpsCopyOutputStream* stream) const final;
|
|
int GetCachedSize() const final { return _impl_._cached_size_.Get(); }
|
|
|
|
private:
|
|
void SharedCtor(::PROTOBUF_NAMESPACE_ID::Arena* arena, bool is_message_owned);
|
|
void SharedDtor();
|
|
void SetCachedSize(int size) const final;
|
|
void InternalSwap(Statistic* other);
|
|
|
|
private:
|
|
friend class ::PROTOBUF_NAMESPACE_ID::internal::AnyMetadata;
|
|
static ::PROTOBUF_NAMESPACE_ID::StringPiece FullMessageName() {
|
|
return "xla.Statistic";
|
|
}
|
|
protected:
|
|
explicit Statistic(::PROTOBUF_NAMESPACE_ID::Arena* arena,
|
|
bool is_message_owned = false);
|
|
public:
|
|
|
|
static const ClassData _class_data_;
|
|
const ::PROTOBUF_NAMESPACE_ID::Message::ClassData*GetClassData() const final;
|
|
|
|
::PROTOBUF_NAMESPACE_ID::Metadata GetMetadata() const final;
|
|
|
|
// nested types ----------------------------------------------------
|
|
|
|
// accessors -------------------------------------------------------
|
|
|
|
enum : int {
|
|
kStatNameFieldNumber = 1,
|
|
kStatValFieldNumber = 2,
|
|
};
|
|
// string stat_name = 1;
|
|
void clear_stat_name();
|
|
const std::string& stat_name() const;
|
|
template <typename ArgT0 = const std::string&, typename... ArgT>
|
|
void set_stat_name(ArgT0&& arg0, ArgT... args);
|
|
std::string* mutable_stat_name();
|
|
PROTOBUF_NODISCARD std::string* release_stat_name();
|
|
void set_allocated_stat_name(std::string* stat_name);
|
|
private:
|
|
const std::string& _internal_stat_name() const;
|
|
inline PROTOBUF_ALWAYS_INLINE void _internal_set_stat_name(const std::string& value);
|
|
std::string* _internal_mutable_stat_name();
|
|
public:
|
|
|
|
// double stat_val = 2;
|
|
void clear_stat_val();
|
|
double stat_val() const;
|
|
void set_stat_val(double value);
|
|
private:
|
|
double _internal_stat_val() const;
|
|
void _internal_set_stat_val(double value);
|
|
public:
|
|
|
|
// @@protoc_insertion_point(class_scope:xla.Statistic)
|
|
private:
|
|
class _Internal;
|
|
|
|
template <typename T> friend class ::PROTOBUF_NAMESPACE_ID::Arena::InternalHelper;
|
|
typedef void InternalArenaConstructable_;
|
|
typedef void DestructorSkippable_;
|
|
struct Impl_ {
|
|
::PROTOBUF_NAMESPACE_ID::internal::ArenaStringPtr stat_name_;
|
|
double stat_val_;
|
|
mutable ::PROTOBUF_NAMESPACE_ID::internal::CachedSize _cached_size_;
|
|
};
|
|
union { Impl_ _impl_; };
|
|
friend struct ::TableStruct_xla_2fxla_5fdata_2eproto;
|
|
};
|
|
// -------------------------------------------------------------------
|
|
|
|
class StatisticsViz final :
|
|
public ::PROTOBUF_NAMESPACE_ID::Message /* @@protoc_insertion_point(class_definition:xla.StatisticsViz) */ {
|
|
public:
|
|
inline StatisticsViz() : StatisticsViz(nullptr) {}
|
|
~StatisticsViz() override;
|
|
explicit PROTOBUF_CONSTEXPR StatisticsViz(::PROTOBUF_NAMESPACE_ID::internal::ConstantInitialized);
|
|
|
|
StatisticsViz(const StatisticsViz& from);
|
|
StatisticsViz(StatisticsViz&& from) noexcept
|
|
: StatisticsViz() {
|
|
*this = ::std::move(from);
|
|
}
|
|
|
|
inline StatisticsViz& operator=(const StatisticsViz& from) {
|
|
CopyFrom(from);
|
|
return *this;
|
|
}
|
|
inline StatisticsViz& operator=(StatisticsViz&& from) noexcept {
|
|
if (this == &from) return *this;
|
|
if (GetOwningArena() == from.GetOwningArena()
|
|
#ifdef PROTOBUF_FORCE_COPY_IN_MOVE
|
|
&& GetOwningArena() != nullptr
|
|
#endif // !PROTOBUF_FORCE_COPY_IN_MOVE
|
|
) {
|
|
InternalSwap(&from);
|
|
} else {
|
|
CopyFrom(from);
|
|
}
|
|
return *this;
|
|
}
|
|
|
|
static const ::PROTOBUF_NAMESPACE_ID::Descriptor* descriptor() {
|
|
return GetDescriptor();
|
|
}
|
|
static const ::PROTOBUF_NAMESPACE_ID::Descriptor* GetDescriptor() {
|
|
return default_instance().GetMetadata().descriptor;
|
|
}
|
|
static const ::PROTOBUF_NAMESPACE_ID::Reflection* GetReflection() {
|
|
return default_instance().GetMetadata().reflection;
|
|
}
|
|
static const StatisticsViz& default_instance() {
|
|
return *internal_default_instance();
|
|
}
|
|
static inline const StatisticsViz* internal_default_instance() {
|
|
return reinterpret_cast<const StatisticsViz*>(
|
|
&_StatisticsViz_default_instance_);
|
|
}
|
|
static constexpr int kIndexInFileMessages =
|
|
30;
|
|
|
|
friend void swap(StatisticsViz& a, StatisticsViz& b) {
|
|
a.Swap(&b);
|
|
}
|
|
inline void Swap(StatisticsViz* other) {
|
|
if (other == this) return;
|
|
#ifdef PROTOBUF_FORCE_COPY_IN_SWAP
|
|
if (GetOwningArena() != nullptr &&
|
|
GetOwningArena() == other->GetOwningArena()) {
|
|
#else // PROTOBUF_FORCE_COPY_IN_SWAP
|
|
if (GetOwningArena() == other->GetOwningArena()) {
|
|
#endif // !PROTOBUF_FORCE_COPY_IN_SWAP
|
|
InternalSwap(other);
|
|
} else {
|
|
::PROTOBUF_NAMESPACE_ID::internal::GenericSwap(this, other);
|
|
}
|
|
}
|
|
void UnsafeArenaSwap(StatisticsViz* other) {
|
|
if (other == this) return;
|
|
GOOGLE_DCHECK(GetOwningArena() == other->GetOwningArena());
|
|
InternalSwap(other);
|
|
}
|
|
|
|
// implements Message ----------------------------------------------
|
|
|
|
StatisticsViz* New(::PROTOBUF_NAMESPACE_ID::Arena* arena = nullptr) const final {
|
|
return CreateMaybeMessage<StatisticsViz>(arena);
|
|
}
|
|
using ::PROTOBUF_NAMESPACE_ID::Message::CopyFrom;
|
|
void CopyFrom(const StatisticsViz& from);
|
|
using ::PROTOBUF_NAMESPACE_ID::Message::MergeFrom;
|
|
void MergeFrom( const StatisticsViz& from) {
|
|
StatisticsViz::MergeImpl(*this, from);
|
|
}
|
|
private:
|
|
static void MergeImpl(::PROTOBUF_NAMESPACE_ID::Message& to_msg, const ::PROTOBUF_NAMESPACE_ID::Message& from_msg);
|
|
public:
|
|
PROTOBUF_ATTRIBUTE_REINITIALIZES void Clear() final;
|
|
bool IsInitialized() const final;
|
|
|
|
size_t ByteSizeLong() const final;
|
|
const char* _InternalParse(const char* ptr, ::PROTOBUF_NAMESPACE_ID::internal::ParseContext* ctx) final;
|
|
uint8_t* _InternalSerialize(
|
|
uint8_t* target, ::PROTOBUF_NAMESPACE_ID::io::EpsCopyOutputStream* stream) const final;
|
|
int GetCachedSize() const final { return _impl_._cached_size_.Get(); }
|
|
|
|
private:
|
|
void SharedCtor(::PROTOBUF_NAMESPACE_ID::Arena* arena, bool is_message_owned);
|
|
void SharedDtor();
|
|
void SetCachedSize(int size) const final;
|
|
void InternalSwap(StatisticsViz* other);
|
|
|
|
private:
|
|
friend class ::PROTOBUF_NAMESPACE_ID::internal::AnyMetadata;
|
|
static ::PROTOBUF_NAMESPACE_ID::StringPiece FullMessageName() {
|
|
return "xla.StatisticsViz";
|
|
}
|
|
protected:
|
|
explicit StatisticsViz(::PROTOBUF_NAMESPACE_ID::Arena* arena,
|
|
bool is_message_owned = false);
|
|
public:
|
|
|
|
static const ClassData _class_data_;
|
|
const ::PROTOBUF_NAMESPACE_ID::Message::ClassData*GetClassData() const final;
|
|
|
|
::PROTOBUF_NAMESPACE_ID::Metadata GetMetadata() const final;
|
|
|
|
// nested types ----------------------------------------------------
|
|
|
|
// accessors -------------------------------------------------------
|
|
|
|
enum : int {
|
|
kStatisticsFieldNumber = 2,
|
|
kStatIndexToVisualizeFieldNumber = 1,
|
|
};
|
|
// repeated .xla.Statistic statistics = 2;
|
|
int statistics_size() const;
|
|
private:
|
|
int _internal_statistics_size() const;
|
|
public:
|
|
void clear_statistics();
|
|
::xla::Statistic* mutable_statistics(int index);
|
|
::PROTOBUF_NAMESPACE_ID::RepeatedPtrField< ::xla::Statistic >*
|
|
mutable_statistics();
|
|
private:
|
|
const ::xla::Statistic& _internal_statistics(int index) const;
|
|
::xla::Statistic* _internal_add_statistics();
|
|
public:
|
|
const ::xla::Statistic& statistics(int index) const;
|
|
::xla::Statistic* add_statistics();
|
|
const ::PROTOBUF_NAMESPACE_ID::RepeatedPtrField< ::xla::Statistic >&
|
|
statistics() const;
|
|
|
|
// int64 stat_index_to_visualize = 1;
|
|
void clear_stat_index_to_visualize();
|
|
int64_t stat_index_to_visualize() const;
|
|
void set_stat_index_to_visualize(int64_t value);
|
|
private:
|
|
int64_t _internal_stat_index_to_visualize() const;
|
|
void _internal_set_stat_index_to_visualize(int64_t value);
|
|
public:
|
|
|
|
// @@protoc_insertion_point(class_scope:xla.StatisticsViz)
|
|
private:
|
|
class _Internal;
|
|
|
|
template <typename T> friend class ::PROTOBUF_NAMESPACE_ID::Arena::InternalHelper;
|
|
typedef void InternalArenaConstructable_;
|
|
typedef void DestructorSkippable_;
|
|
struct Impl_ {
|
|
::PROTOBUF_NAMESPACE_ID::RepeatedPtrField< ::xla::Statistic > statistics_;
|
|
int64_t stat_index_to_visualize_;
|
|
mutable ::PROTOBUF_NAMESPACE_ID::internal::CachedSize _cached_size_;
|
|
};
|
|
union { Impl_ _impl_; };
|
|
friend struct ::TableStruct_xla_2fxla_5fdata_2eproto;
|
|
};
|
|
// -------------------------------------------------------------------
|
|
|
|
class OpSharding final :
|
|
public ::PROTOBUF_NAMESPACE_ID::Message /* @@protoc_insertion_point(class_definition:xla.OpSharding) */ {
|
|
public:
|
|
inline OpSharding() : OpSharding(nullptr) {}
|
|
~OpSharding() override;
|
|
explicit PROTOBUF_CONSTEXPR OpSharding(::PROTOBUF_NAMESPACE_ID::internal::ConstantInitialized);
|
|
|
|
OpSharding(const OpSharding& from);
|
|
OpSharding(OpSharding&& from) noexcept
|
|
: OpSharding() {
|
|
*this = ::std::move(from);
|
|
}
|
|
|
|
inline OpSharding& operator=(const OpSharding& from) {
|
|
CopyFrom(from);
|
|
return *this;
|
|
}
|
|
inline OpSharding& operator=(OpSharding&& from) noexcept {
|
|
if (this == &from) return *this;
|
|
if (GetOwningArena() == from.GetOwningArena()
|
|
#ifdef PROTOBUF_FORCE_COPY_IN_MOVE
|
|
&& GetOwningArena() != nullptr
|
|
#endif // !PROTOBUF_FORCE_COPY_IN_MOVE
|
|
) {
|
|
InternalSwap(&from);
|
|
} else {
|
|
CopyFrom(from);
|
|
}
|
|
return *this;
|
|
}
|
|
|
|
static const ::PROTOBUF_NAMESPACE_ID::Descriptor* descriptor() {
|
|
return GetDescriptor();
|
|
}
|
|
static const ::PROTOBUF_NAMESPACE_ID::Descriptor* GetDescriptor() {
|
|
return default_instance().GetMetadata().descriptor;
|
|
}
|
|
static const ::PROTOBUF_NAMESPACE_ID::Reflection* GetReflection() {
|
|
return default_instance().GetMetadata().reflection;
|
|
}
|
|
static const OpSharding& default_instance() {
|
|
return *internal_default_instance();
|
|
}
|
|
static inline const OpSharding* internal_default_instance() {
|
|
return reinterpret_cast<const OpSharding*>(
|
|
&_OpSharding_default_instance_);
|
|
}
|
|
static constexpr int kIndexInFileMessages =
|
|
31;
|
|
|
|
friend void swap(OpSharding& a, OpSharding& b) {
|
|
a.Swap(&b);
|
|
}
|
|
inline void Swap(OpSharding* other) {
|
|
if (other == this) return;
|
|
#ifdef PROTOBUF_FORCE_COPY_IN_SWAP
|
|
if (GetOwningArena() != nullptr &&
|
|
GetOwningArena() == other->GetOwningArena()) {
|
|
#else // PROTOBUF_FORCE_COPY_IN_SWAP
|
|
if (GetOwningArena() == other->GetOwningArena()) {
|
|
#endif // !PROTOBUF_FORCE_COPY_IN_SWAP
|
|
InternalSwap(other);
|
|
} else {
|
|
::PROTOBUF_NAMESPACE_ID::internal::GenericSwap(this, other);
|
|
}
|
|
}
|
|
void UnsafeArenaSwap(OpSharding* other) {
|
|
if (other == this) return;
|
|
GOOGLE_DCHECK(GetOwningArena() == other->GetOwningArena());
|
|
InternalSwap(other);
|
|
}
|
|
|
|
// implements Message ----------------------------------------------
|
|
|
|
OpSharding* New(::PROTOBUF_NAMESPACE_ID::Arena* arena = nullptr) const final {
|
|
return CreateMaybeMessage<OpSharding>(arena);
|
|
}
|
|
using ::PROTOBUF_NAMESPACE_ID::Message::CopyFrom;
|
|
void CopyFrom(const OpSharding& from);
|
|
using ::PROTOBUF_NAMESPACE_ID::Message::MergeFrom;
|
|
void MergeFrom( const OpSharding& from) {
|
|
OpSharding::MergeImpl(*this, from);
|
|
}
|
|
private:
|
|
static void MergeImpl(::PROTOBUF_NAMESPACE_ID::Message& to_msg, const ::PROTOBUF_NAMESPACE_ID::Message& from_msg);
|
|
public:
|
|
PROTOBUF_ATTRIBUTE_REINITIALIZES void Clear() final;
|
|
bool IsInitialized() const final;
|
|
|
|
size_t ByteSizeLong() const final;
|
|
const char* _InternalParse(const char* ptr, ::PROTOBUF_NAMESPACE_ID::internal::ParseContext* ctx) final;
|
|
uint8_t* _InternalSerialize(
|
|
uint8_t* target, ::PROTOBUF_NAMESPACE_ID::io::EpsCopyOutputStream* stream) const final;
|
|
int GetCachedSize() const final { return _impl_._cached_size_.Get(); }
|
|
|
|
private:
|
|
void SharedCtor(::PROTOBUF_NAMESPACE_ID::Arena* arena, bool is_message_owned);
|
|
void SharedDtor();
|
|
void SetCachedSize(int size) const final;
|
|
void InternalSwap(OpSharding* other);
|
|
|
|
private:
|
|
friend class ::PROTOBUF_NAMESPACE_ID::internal::AnyMetadata;
|
|
static ::PROTOBUF_NAMESPACE_ID::StringPiece FullMessageName() {
|
|
return "xla.OpSharding";
|
|
}
|
|
protected:
|
|
explicit OpSharding(::PROTOBUF_NAMESPACE_ID::Arena* arena,
|
|
bool is_message_owned = false);
|
|
public:
|
|
|
|
static const ClassData _class_data_;
|
|
const ::PROTOBUF_NAMESPACE_ID::Message::ClassData*GetClassData() const final;
|
|
|
|
::PROTOBUF_NAMESPACE_ID::Metadata GetMetadata() const final;
|
|
|
|
// nested types ----------------------------------------------------
|
|
|
|
typedef OpSharding_Type Type;
|
|
static constexpr Type REPLICATED =
|
|
OpSharding_Type_REPLICATED;
|
|
static constexpr Type MAXIMAL =
|
|
OpSharding_Type_MAXIMAL;
|
|
static constexpr Type TUPLE =
|
|
OpSharding_Type_TUPLE;
|
|
static constexpr Type OTHER =
|
|
OpSharding_Type_OTHER;
|
|
static constexpr Type MANUAL =
|
|
OpSharding_Type_MANUAL;
|
|
static constexpr Type UNKNOWN =
|
|
OpSharding_Type_UNKNOWN;
|
|
static inline bool Type_IsValid(int value) {
|
|
return OpSharding_Type_IsValid(value);
|
|
}
|
|
static constexpr Type Type_MIN =
|
|
OpSharding_Type_Type_MIN;
|
|
static constexpr Type Type_MAX =
|
|
OpSharding_Type_Type_MAX;
|
|
static constexpr int Type_ARRAYSIZE =
|
|
OpSharding_Type_Type_ARRAYSIZE;
|
|
static inline const ::PROTOBUF_NAMESPACE_ID::EnumDescriptor*
|
|
Type_descriptor() {
|
|
return OpSharding_Type_descriptor();
|
|
}
|
|
template<typename T>
|
|
static inline const std::string& Type_Name(T enum_t_value) {
|
|
static_assert(::std::is_same<T, Type>::value ||
|
|
::std::is_integral<T>::value,
|
|
"Incorrect type passed to function Type_Name.");
|
|
return OpSharding_Type_Name(enum_t_value);
|
|
}
|
|
static inline bool Type_Parse(::PROTOBUF_NAMESPACE_ID::ConstStringParam name,
|
|
Type* value) {
|
|
return OpSharding_Type_Parse(name, value);
|
|
}
|
|
|
|
typedef OpSharding_ShardGroupType ShardGroupType;
|
|
static constexpr ShardGroupType AS =
|
|
OpSharding_ShardGroupType_AS;
|
|
static constexpr ShardGroupType LIKE =
|
|
OpSharding_ShardGroupType_LIKE;
|
|
static inline bool ShardGroupType_IsValid(int value) {
|
|
return OpSharding_ShardGroupType_IsValid(value);
|
|
}
|
|
static constexpr ShardGroupType ShardGroupType_MIN =
|
|
OpSharding_ShardGroupType_ShardGroupType_MIN;
|
|
static constexpr ShardGroupType ShardGroupType_MAX =
|
|
OpSharding_ShardGroupType_ShardGroupType_MAX;
|
|
static constexpr int ShardGroupType_ARRAYSIZE =
|
|
OpSharding_ShardGroupType_ShardGroupType_ARRAYSIZE;
|
|
static inline const ::PROTOBUF_NAMESPACE_ID::EnumDescriptor*
|
|
ShardGroupType_descriptor() {
|
|
return OpSharding_ShardGroupType_descriptor();
|
|
}
|
|
template<typename T>
|
|
static inline const std::string& ShardGroupType_Name(T enum_t_value) {
|
|
static_assert(::std::is_same<T, ShardGroupType>::value ||
|
|
::std::is_integral<T>::value,
|
|
"Incorrect type passed to function ShardGroupType_Name.");
|
|
return OpSharding_ShardGroupType_Name(enum_t_value);
|
|
}
|
|
static inline bool ShardGroupType_Parse(::PROTOBUF_NAMESPACE_ID::ConstStringParam name,
|
|
ShardGroupType* value) {
|
|
return OpSharding_ShardGroupType_Parse(name, value);
|
|
}
|
|
|
|
// accessors -------------------------------------------------------
|
|
|
|
enum : int {
|
|
kTileAssignmentDimensionsFieldNumber = 3,
|
|
kTileAssignmentDevicesFieldNumber = 4,
|
|
kTupleShardingsFieldNumber = 5,
|
|
kMetadataFieldNumber = 7,
|
|
kLastTileDimsFieldNumber = 8,
|
|
kIotaReshapeDimsFieldNumber = 9,
|
|
kIotaTransposePermFieldNumber = 10,
|
|
kTileShapeFieldNumber = 2,
|
|
kTypeFieldNumber = 1,
|
|
kReplicateOnLastTileDimFieldNumber = 6,
|
|
kIsShardGroupFieldNumber = 11,
|
|
kShardGroupIdFieldNumber = 12,
|
|
kShardGroupTypeFieldNumber = 13,
|
|
};
|
|
// repeated int64 tile_assignment_dimensions = 3;
|
|
int tile_assignment_dimensions_size() const;
|
|
private:
|
|
int _internal_tile_assignment_dimensions_size() const;
|
|
public:
|
|
void clear_tile_assignment_dimensions();
|
|
private:
|
|
int64_t _internal_tile_assignment_dimensions(int index) const;
|
|
const ::PROTOBUF_NAMESPACE_ID::RepeatedField< int64_t >&
|
|
_internal_tile_assignment_dimensions() const;
|
|
void _internal_add_tile_assignment_dimensions(int64_t value);
|
|
::PROTOBUF_NAMESPACE_ID::RepeatedField< int64_t >*
|
|
_internal_mutable_tile_assignment_dimensions();
|
|
public:
|
|
int64_t tile_assignment_dimensions(int index) const;
|
|
void set_tile_assignment_dimensions(int index, int64_t value);
|
|
void add_tile_assignment_dimensions(int64_t value);
|
|
const ::PROTOBUF_NAMESPACE_ID::RepeatedField< int64_t >&
|
|
tile_assignment_dimensions() const;
|
|
::PROTOBUF_NAMESPACE_ID::RepeatedField< int64_t >*
|
|
mutable_tile_assignment_dimensions();
|
|
|
|
// repeated int64 tile_assignment_devices = 4;
|
|
int tile_assignment_devices_size() const;
|
|
private:
|
|
int _internal_tile_assignment_devices_size() const;
|
|
public:
|
|
void clear_tile_assignment_devices();
|
|
private:
|
|
int64_t _internal_tile_assignment_devices(int index) const;
|
|
const ::PROTOBUF_NAMESPACE_ID::RepeatedField< int64_t >&
|
|
_internal_tile_assignment_devices() const;
|
|
void _internal_add_tile_assignment_devices(int64_t value);
|
|
::PROTOBUF_NAMESPACE_ID::RepeatedField< int64_t >*
|
|
_internal_mutable_tile_assignment_devices();
|
|
public:
|
|
int64_t tile_assignment_devices(int index) const;
|
|
void set_tile_assignment_devices(int index, int64_t value);
|
|
void add_tile_assignment_devices(int64_t value);
|
|
const ::PROTOBUF_NAMESPACE_ID::RepeatedField< int64_t >&
|
|
tile_assignment_devices() const;
|
|
::PROTOBUF_NAMESPACE_ID::RepeatedField< int64_t >*
|
|
mutable_tile_assignment_devices();
|
|
|
|
// repeated .xla.OpSharding tuple_shardings = 5;
|
|
int tuple_shardings_size() const;
|
|
private:
|
|
int _internal_tuple_shardings_size() const;
|
|
public:
|
|
void clear_tuple_shardings();
|
|
::xla::OpSharding* mutable_tuple_shardings(int index);
|
|
::PROTOBUF_NAMESPACE_ID::RepeatedPtrField< ::xla::OpSharding >*
|
|
mutable_tuple_shardings();
|
|
private:
|
|
const ::xla::OpSharding& _internal_tuple_shardings(int index) const;
|
|
::xla::OpSharding* _internal_add_tuple_shardings();
|
|
public:
|
|
const ::xla::OpSharding& tuple_shardings(int index) const;
|
|
::xla::OpSharding* add_tuple_shardings();
|
|
const ::PROTOBUF_NAMESPACE_ID::RepeatedPtrField< ::xla::OpSharding >&
|
|
tuple_shardings() const;
|
|
|
|
// repeated .xla.OpMetadata metadata = 7;
|
|
int metadata_size() const;
|
|
private:
|
|
int _internal_metadata_size() const;
|
|
public:
|
|
void clear_metadata();
|
|
::xla::OpMetadata* mutable_metadata(int index);
|
|
::PROTOBUF_NAMESPACE_ID::RepeatedPtrField< ::xla::OpMetadata >*
|
|
mutable_metadata();
|
|
private:
|
|
const ::xla::OpMetadata& _internal_metadata(int index) const;
|
|
::xla::OpMetadata* _internal_add_metadata();
|
|
public:
|
|
const ::xla::OpMetadata& metadata(int index) const;
|
|
::xla::OpMetadata* add_metadata();
|
|
const ::PROTOBUF_NAMESPACE_ID::RepeatedPtrField< ::xla::OpMetadata >&
|
|
metadata() const;
|
|
|
|
// repeated .xla.OpSharding.Type last_tile_dims = 8;
|
|
int last_tile_dims_size() const;
|
|
private:
|
|
int _internal_last_tile_dims_size() const;
|
|
public:
|
|
void clear_last_tile_dims();
|
|
private:
|
|
::xla::OpSharding_Type _internal_last_tile_dims(int index) const;
|
|
void _internal_add_last_tile_dims(::xla::OpSharding_Type value);
|
|
::PROTOBUF_NAMESPACE_ID::RepeatedField<int>* _internal_mutable_last_tile_dims();
|
|
public:
|
|
::xla::OpSharding_Type last_tile_dims(int index) const;
|
|
void set_last_tile_dims(int index, ::xla::OpSharding_Type value);
|
|
void add_last_tile_dims(::xla::OpSharding_Type value);
|
|
const ::PROTOBUF_NAMESPACE_ID::RepeatedField<int>& last_tile_dims() const;
|
|
::PROTOBUF_NAMESPACE_ID::RepeatedField<int>* mutable_last_tile_dims();
|
|
|
|
// repeated int64 iota_reshape_dims = 9;
|
|
int iota_reshape_dims_size() const;
|
|
private:
|
|
int _internal_iota_reshape_dims_size() const;
|
|
public:
|
|
void clear_iota_reshape_dims();
|
|
private:
|
|
int64_t _internal_iota_reshape_dims(int index) const;
|
|
const ::PROTOBUF_NAMESPACE_ID::RepeatedField< int64_t >&
|
|
_internal_iota_reshape_dims() const;
|
|
void _internal_add_iota_reshape_dims(int64_t value);
|
|
::PROTOBUF_NAMESPACE_ID::RepeatedField< int64_t >*
|
|
_internal_mutable_iota_reshape_dims();
|
|
public:
|
|
int64_t iota_reshape_dims(int index) const;
|
|
void set_iota_reshape_dims(int index, int64_t value);
|
|
void add_iota_reshape_dims(int64_t value);
|
|
const ::PROTOBUF_NAMESPACE_ID::RepeatedField< int64_t >&
|
|
iota_reshape_dims() const;
|
|
::PROTOBUF_NAMESPACE_ID::RepeatedField< int64_t >*
|
|
mutable_iota_reshape_dims();
|
|
|
|
// repeated int32 iota_transpose_perm = 10;
|
|
int iota_transpose_perm_size() const;
|
|
private:
|
|
int _internal_iota_transpose_perm_size() const;
|
|
public:
|
|
void clear_iota_transpose_perm();
|
|
private:
|
|
int32_t _internal_iota_transpose_perm(int index) const;
|
|
const ::PROTOBUF_NAMESPACE_ID::RepeatedField< int32_t >&
|
|
_internal_iota_transpose_perm() const;
|
|
void _internal_add_iota_transpose_perm(int32_t value);
|
|
::PROTOBUF_NAMESPACE_ID::RepeatedField< int32_t >*
|
|
_internal_mutable_iota_transpose_perm();
|
|
public:
|
|
int32_t iota_transpose_perm(int index) const;
|
|
void set_iota_transpose_perm(int index, int32_t value);
|
|
void add_iota_transpose_perm(int32_t value);
|
|
const ::PROTOBUF_NAMESPACE_ID::RepeatedField< int32_t >&
|
|
iota_transpose_perm() const;
|
|
::PROTOBUF_NAMESPACE_ID::RepeatedField< int32_t >*
|
|
mutable_iota_transpose_perm();
|
|
|
|
// .xla.ShapeProto tile_shape = 2;
|
|
bool has_tile_shape() const;
|
|
private:
|
|
bool _internal_has_tile_shape() const;
|
|
public:
|
|
void clear_tile_shape();
|
|
const ::xla::ShapeProto& tile_shape() const;
|
|
PROTOBUF_NODISCARD ::xla::ShapeProto* release_tile_shape();
|
|
::xla::ShapeProto* mutable_tile_shape();
|
|
void set_allocated_tile_shape(::xla::ShapeProto* tile_shape);
|
|
private:
|
|
const ::xla::ShapeProto& _internal_tile_shape() const;
|
|
::xla::ShapeProto* _internal_mutable_tile_shape();
|
|
public:
|
|
void unsafe_arena_set_allocated_tile_shape(
|
|
::xla::ShapeProto* tile_shape);
|
|
::xla::ShapeProto* unsafe_arena_release_tile_shape();
|
|
|
|
// .xla.OpSharding.Type type = 1;
|
|
void clear_type();
|
|
::xla::OpSharding_Type type() const;
|
|
void set_type(::xla::OpSharding_Type value);
|
|
private:
|
|
::xla::OpSharding_Type _internal_type() const;
|
|
void _internal_set_type(::xla::OpSharding_Type value);
|
|
public:
|
|
|
|
// bool replicate_on_last_tile_dim = 6;
|
|
void clear_replicate_on_last_tile_dim();
|
|
bool replicate_on_last_tile_dim() const;
|
|
void set_replicate_on_last_tile_dim(bool value);
|
|
private:
|
|
bool _internal_replicate_on_last_tile_dim() const;
|
|
void _internal_set_replicate_on_last_tile_dim(bool value);
|
|
public:
|
|
|
|
// bool is_shard_group = 11;
|
|
void clear_is_shard_group();
|
|
bool is_shard_group() const;
|
|
void set_is_shard_group(bool value);
|
|
private:
|
|
bool _internal_is_shard_group() const;
|
|
void _internal_set_is_shard_group(bool value);
|
|
public:
|
|
|
|
// int64 shard_group_id = 12;
|
|
void clear_shard_group_id();
|
|
int64_t shard_group_id() const;
|
|
void set_shard_group_id(int64_t value);
|
|
private:
|
|
int64_t _internal_shard_group_id() const;
|
|
void _internal_set_shard_group_id(int64_t value);
|
|
public:
|
|
|
|
// .xla.OpSharding.ShardGroupType shard_group_type = 13;
|
|
void clear_shard_group_type();
|
|
::xla::OpSharding_ShardGroupType shard_group_type() const;
|
|
void set_shard_group_type(::xla::OpSharding_ShardGroupType value);
|
|
private:
|
|
::xla::OpSharding_ShardGroupType _internal_shard_group_type() const;
|
|
void _internal_set_shard_group_type(::xla::OpSharding_ShardGroupType value);
|
|
public:
|
|
|
|
// @@protoc_insertion_point(class_scope:xla.OpSharding)
|
|
private:
|
|
class _Internal;
|
|
|
|
template <typename T> friend class ::PROTOBUF_NAMESPACE_ID::Arena::InternalHelper;
|
|
typedef void InternalArenaConstructable_;
|
|
typedef void DestructorSkippable_;
|
|
struct Impl_ {
|
|
::PROTOBUF_NAMESPACE_ID::RepeatedField< int64_t > tile_assignment_dimensions_;
|
|
mutable std::atomic<int> _tile_assignment_dimensions_cached_byte_size_;
|
|
::PROTOBUF_NAMESPACE_ID::RepeatedField< int64_t > tile_assignment_devices_;
|
|
mutable std::atomic<int> _tile_assignment_devices_cached_byte_size_;
|
|
::PROTOBUF_NAMESPACE_ID::RepeatedPtrField< ::xla::OpSharding > tuple_shardings_;
|
|
::PROTOBUF_NAMESPACE_ID::RepeatedPtrField< ::xla::OpMetadata > metadata_;
|
|
::PROTOBUF_NAMESPACE_ID::RepeatedField<int> last_tile_dims_;
|
|
mutable std::atomic<int> _last_tile_dims_cached_byte_size_;
|
|
::PROTOBUF_NAMESPACE_ID::RepeatedField< int64_t > iota_reshape_dims_;
|
|
mutable std::atomic<int> _iota_reshape_dims_cached_byte_size_;
|
|
::PROTOBUF_NAMESPACE_ID::RepeatedField< int32_t > iota_transpose_perm_;
|
|
mutable std::atomic<int> _iota_transpose_perm_cached_byte_size_;
|
|
::xla::ShapeProto* tile_shape_;
|
|
int type_;
|
|
bool replicate_on_last_tile_dim_;
|
|
bool is_shard_group_;
|
|
int64_t shard_group_id_;
|
|
int shard_group_type_;
|
|
mutable ::PROTOBUF_NAMESPACE_ID::internal::CachedSize _cached_size_;
|
|
};
|
|
union { Impl_ _impl_; };
|
|
friend struct ::TableStruct_xla_2fxla_5fdata_2eproto;
|
|
};
|
|
// -------------------------------------------------------------------
|
|
|
|
class ReplicaGroup final :
|
|
public ::PROTOBUF_NAMESPACE_ID::Message /* @@protoc_insertion_point(class_definition:xla.ReplicaGroup) */ {
|
|
public:
|
|
inline ReplicaGroup() : ReplicaGroup(nullptr) {}
|
|
~ReplicaGroup() override;
|
|
explicit PROTOBUF_CONSTEXPR ReplicaGroup(::PROTOBUF_NAMESPACE_ID::internal::ConstantInitialized);
|
|
|
|
ReplicaGroup(const ReplicaGroup& from);
|
|
ReplicaGroup(ReplicaGroup&& from) noexcept
|
|
: ReplicaGroup() {
|
|
*this = ::std::move(from);
|
|
}
|
|
|
|
inline ReplicaGroup& operator=(const ReplicaGroup& from) {
|
|
CopyFrom(from);
|
|
return *this;
|
|
}
|
|
inline ReplicaGroup& operator=(ReplicaGroup&& from) noexcept {
|
|
if (this == &from) return *this;
|
|
if (GetOwningArena() == from.GetOwningArena()
|
|
#ifdef PROTOBUF_FORCE_COPY_IN_MOVE
|
|
&& GetOwningArena() != nullptr
|
|
#endif // !PROTOBUF_FORCE_COPY_IN_MOVE
|
|
) {
|
|
InternalSwap(&from);
|
|
} else {
|
|
CopyFrom(from);
|
|
}
|
|
return *this;
|
|
}
|
|
|
|
static const ::PROTOBUF_NAMESPACE_ID::Descriptor* descriptor() {
|
|
return GetDescriptor();
|
|
}
|
|
static const ::PROTOBUF_NAMESPACE_ID::Descriptor* GetDescriptor() {
|
|
return default_instance().GetMetadata().descriptor;
|
|
}
|
|
static const ::PROTOBUF_NAMESPACE_ID::Reflection* GetReflection() {
|
|
return default_instance().GetMetadata().reflection;
|
|
}
|
|
static const ReplicaGroup& default_instance() {
|
|
return *internal_default_instance();
|
|
}
|
|
static inline const ReplicaGroup* internal_default_instance() {
|
|
return reinterpret_cast<const ReplicaGroup*>(
|
|
&_ReplicaGroup_default_instance_);
|
|
}
|
|
static constexpr int kIndexInFileMessages =
|
|
32;
|
|
|
|
friend void swap(ReplicaGroup& a, ReplicaGroup& b) {
|
|
a.Swap(&b);
|
|
}
|
|
inline void Swap(ReplicaGroup* other) {
|
|
if (other == this) return;
|
|
#ifdef PROTOBUF_FORCE_COPY_IN_SWAP
|
|
if (GetOwningArena() != nullptr &&
|
|
GetOwningArena() == other->GetOwningArena()) {
|
|
#else // PROTOBUF_FORCE_COPY_IN_SWAP
|
|
if (GetOwningArena() == other->GetOwningArena()) {
|
|
#endif // !PROTOBUF_FORCE_COPY_IN_SWAP
|
|
InternalSwap(other);
|
|
} else {
|
|
::PROTOBUF_NAMESPACE_ID::internal::GenericSwap(this, other);
|
|
}
|
|
}
|
|
void UnsafeArenaSwap(ReplicaGroup* other) {
|
|
if (other == this) return;
|
|
GOOGLE_DCHECK(GetOwningArena() == other->GetOwningArena());
|
|
InternalSwap(other);
|
|
}
|
|
|
|
// implements Message ----------------------------------------------
|
|
|
|
ReplicaGroup* New(::PROTOBUF_NAMESPACE_ID::Arena* arena = nullptr) const final {
|
|
return CreateMaybeMessage<ReplicaGroup>(arena);
|
|
}
|
|
using ::PROTOBUF_NAMESPACE_ID::Message::CopyFrom;
|
|
void CopyFrom(const ReplicaGroup& from);
|
|
using ::PROTOBUF_NAMESPACE_ID::Message::MergeFrom;
|
|
void MergeFrom( const ReplicaGroup& from) {
|
|
ReplicaGroup::MergeImpl(*this, from);
|
|
}
|
|
private:
|
|
static void MergeImpl(::PROTOBUF_NAMESPACE_ID::Message& to_msg, const ::PROTOBUF_NAMESPACE_ID::Message& from_msg);
|
|
public:
|
|
PROTOBUF_ATTRIBUTE_REINITIALIZES void Clear() final;
|
|
bool IsInitialized() const final;
|
|
|
|
size_t ByteSizeLong() const final;
|
|
const char* _InternalParse(const char* ptr, ::PROTOBUF_NAMESPACE_ID::internal::ParseContext* ctx) final;
|
|
uint8_t* _InternalSerialize(
|
|
uint8_t* target, ::PROTOBUF_NAMESPACE_ID::io::EpsCopyOutputStream* stream) const final;
|
|
int GetCachedSize() const final { return _impl_._cached_size_.Get(); }
|
|
|
|
private:
|
|
void SharedCtor(::PROTOBUF_NAMESPACE_ID::Arena* arena, bool is_message_owned);
|
|
void SharedDtor();
|
|
void SetCachedSize(int size) const final;
|
|
void InternalSwap(ReplicaGroup* other);
|
|
|
|
private:
|
|
friend class ::PROTOBUF_NAMESPACE_ID::internal::AnyMetadata;
|
|
static ::PROTOBUF_NAMESPACE_ID::StringPiece FullMessageName() {
|
|
return "xla.ReplicaGroup";
|
|
}
|
|
protected:
|
|
explicit ReplicaGroup(::PROTOBUF_NAMESPACE_ID::Arena* arena,
|
|
bool is_message_owned = false);
|
|
public:
|
|
|
|
static const ClassData _class_data_;
|
|
const ::PROTOBUF_NAMESPACE_ID::Message::ClassData*GetClassData() const final;
|
|
|
|
::PROTOBUF_NAMESPACE_ID::Metadata GetMetadata() const final;
|
|
|
|
// nested types ----------------------------------------------------
|
|
|
|
// accessors -------------------------------------------------------
|
|
|
|
enum : int {
|
|
kReplicaIdsFieldNumber = 1,
|
|
};
|
|
// repeated int64 replica_ids = 1;
|
|
int replica_ids_size() const;
|
|
private:
|
|
int _internal_replica_ids_size() const;
|
|
public:
|
|
void clear_replica_ids();
|
|
private:
|
|
int64_t _internal_replica_ids(int index) const;
|
|
const ::PROTOBUF_NAMESPACE_ID::RepeatedField< int64_t >&
|
|
_internal_replica_ids() const;
|
|
void _internal_add_replica_ids(int64_t value);
|
|
::PROTOBUF_NAMESPACE_ID::RepeatedField< int64_t >*
|
|
_internal_mutable_replica_ids();
|
|
public:
|
|
int64_t replica_ids(int index) const;
|
|
void set_replica_ids(int index, int64_t value);
|
|
void add_replica_ids(int64_t value);
|
|
const ::PROTOBUF_NAMESPACE_ID::RepeatedField< int64_t >&
|
|
replica_ids() const;
|
|
::PROTOBUF_NAMESPACE_ID::RepeatedField< int64_t >*
|
|
mutable_replica_ids();
|
|
|
|
// @@protoc_insertion_point(class_scope:xla.ReplicaGroup)
|
|
private:
|
|
class _Internal;
|
|
|
|
template <typename T> friend class ::PROTOBUF_NAMESPACE_ID::Arena::InternalHelper;
|
|
typedef void InternalArenaConstructable_;
|
|
typedef void DestructorSkippable_;
|
|
struct Impl_ {
|
|
::PROTOBUF_NAMESPACE_ID::RepeatedField< int64_t > replica_ids_;
|
|
mutable std::atomic<int> _replica_ids_cached_byte_size_;
|
|
mutable ::PROTOBUF_NAMESPACE_ID::internal::CachedSize _cached_size_;
|
|
};
|
|
union { Impl_ _impl_; };
|
|
friend struct ::TableStruct_xla_2fxla_5fdata_2eproto;
|
|
};
|
|
// -------------------------------------------------------------------
|
|
|
|
class SourceTarget final :
|
|
public ::PROTOBUF_NAMESPACE_ID::Message /* @@protoc_insertion_point(class_definition:xla.SourceTarget) */ {
|
|
public:
|
|
inline SourceTarget() : SourceTarget(nullptr) {}
|
|
~SourceTarget() override;
|
|
explicit PROTOBUF_CONSTEXPR SourceTarget(::PROTOBUF_NAMESPACE_ID::internal::ConstantInitialized);
|
|
|
|
SourceTarget(const SourceTarget& from);
|
|
SourceTarget(SourceTarget&& from) noexcept
|
|
: SourceTarget() {
|
|
*this = ::std::move(from);
|
|
}
|
|
|
|
inline SourceTarget& operator=(const SourceTarget& from) {
|
|
CopyFrom(from);
|
|
return *this;
|
|
}
|
|
inline SourceTarget& operator=(SourceTarget&& from) noexcept {
|
|
if (this == &from) return *this;
|
|
if (GetOwningArena() == from.GetOwningArena()
|
|
#ifdef PROTOBUF_FORCE_COPY_IN_MOVE
|
|
&& GetOwningArena() != nullptr
|
|
#endif // !PROTOBUF_FORCE_COPY_IN_MOVE
|
|
) {
|
|
InternalSwap(&from);
|
|
} else {
|
|
CopyFrom(from);
|
|
}
|
|
return *this;
|
|
}
|
|
|
|
static const ::PROTOBUF_NAMESPACE_ID::Descriptor* descriptor() {
|
|
return GetDescriptor();
|
|
}
|
|
static const ::PROTOBUF_NAMESPACE_ID::Descriptor* GetDescriptor() {
|
|
return default_instance().GetMetadata().descriptor;
|
|
}
|
|
static const ::PROTOBUF_NAMESPACE_ID::Reflection* GetReflection() {
|
|
return default_instance().GetMetadata().reflection;
|
|
}
|
|
static const SourceTarget& default_instance() {
|
|
return *internal_default_instance();
|
|
}
|
|
static inline const SourceTarget* internal_default_instance() {
|
|
return reinterpret_cast<const SourceTarget*>(
|
|
&_SourceTarget_default_instance_);
|
|
}
|
|
static constexpr int kIndexInFileMessages =
|
|
33;
|
|
|
|
friend void swap(SourceTarget& a, SourceTarget& b) {
|
|
a.Swap(&b);
|
|
}
|
|
inline void Swap(SourceTarget* other) {
|
|
if (other == this) return;
|
|
#ifdef PROTOBUF_FORCE_COPY_IN_SWAP
|
|
if (GetOwningArena() != nullptr &&
|
|
GetOwningArena() == other->GetOwningArena()) {
|
|
#else // PROTOBUF_FORCE_COPY_IN_SWAP
|
|
if (GetOwningArena() == other->GetOwningArena()) {
|
|
#endif // !PROTOBUF_FORCE_COPY_IN_SWAP
|
|
InternalSwap(other);
|
|
} else {
|
|
::PROTOBUF_NAMESPACE_ID::internal::GenericSwap(this, other);
|
|
}
|
|
}
|
|
void UnsafeArenaSwap(SourceTarget* other) {
|
|
if (other == this) return;
|
|
GOOGLE_DCHECK(GetOwningArena() == other->GetOwningArena());
|
|
InternalSwap(other);
|
|
}
|
|
|
|
// implements Message ----------------------------------------------
|
|
|
|
SourceTarget* New(::PROTOBUF_NAMESPACE_ID::Arena* arena = nullptr) const final {
|
|
return CreateMaybeMessage<SourceTarget>(arena);
|
|
}
|
|
using ::PROTOBUF_NAMESPACE_ID::Message::CopyFrom;
|
|
void CopyFrom(const SourceTarget& from);
|
|
using ::PROTOBUF_NAMESPACE_ID::Message::MergeFrom;
|
|
void MergeFrom( const SourceTarget& from) {
|
|
SourceTarget::MergeImpl(*this, from);
|
|
}
|
|
private:
|
|
static void MergeImpl(::PROTOBUF_NAMESPACE_ID::Message& to_msg, const ::PROTOBUF_NAMESPACE_ID::Message& from_msg);
|
|
public:
|
|
PROTOBUF_ATTRIBUTE_REINITIALIZES void Clear() final;
|
|
bool IsInitialized() const final;
|
|
|
|
size_t ByteSizeLong() const final;
|
|
const char* _InternalParse(const char* ptr, ::PROTOBUF_NAMESPACE_ID::internal::ParseContext* ctx) final;
|
|
uint8_t* _InternalSerialize(
|
|
uint8_t* target, ::PROTOBUF_NAMESPACE_ID::io::EpsCopyOutputStream* stream) const final;
|
|
int GetCachedSize() const final { return _impl_._cached_size_.Get(); }
|
|
|
|
private:
|
|
void SharedCtor(::PROTOBUF_NAMESPACE_ID::Arena* arena, bool is_message_owned);
|
|
void SharedDtor();
|
|
void SetCachedSize(int size) const final;
|
|
void InternalSwap(SourceTarget* other);
|
|
|
|
private:
|
|
friend class ::PROTOBUF_NAMESPACE_ID::internal::AnyMetadata;
|
|
static ::PROTOBUF_NAMESPACE_ID::StringPiece FullMessageName() {
|
|
return "xla.SourceTarget";
|
|
}
|
|
protected:
|
|
explicit SourceTarget(::PROTOBUF_NAMESPACE_ID::Arena* arena,
|
|
bool is_message_owned = false);
|
|
public:
|
|
|
|
static const ClassData _class_data_;
|
|
const ::PROTOBUF_NAMESPACE_ID::Message::ClassData*GetClassData() const final;
|
|
|
|
::PROTOBUF_NAMESPACE_ID::Metadata GetMetadata() const final;
|
|
|
|
// nested types ----------------------------------------------------
|
|
|
|
// accessors -------------------------------------------------------
|
|
|
|
enum : int {
|
|
kSourceFieldNumber = 1,
|
|
kTargetFieldNumber = 2,
|
|
};
|
|
// int64 source = 1;
|
|
void clear_source();
|
|
int64_t source() const;
|
|
void set_source(int64_t value);
|
|
private:
|
|
int64_t _internal_source() const;
|
|
void _internal_set_source(int64_t value);
|
|
public:
|
|
|
|
// int64 target = 2;
|
|
void clear_target();
|
|
int64_t target() const;
|
|
void set_target(int64_t value);
|
|
private:
|
|
int64_t _internal_target() const;
|
|
void _internal_set_target(int64_t value);
|
|
public:
|
|
|
|
// @@protoc_insertion_point(class_scope:xla.SourceTarget)
|
|
private:
|
|
class _Internal;
|
|
|
|
template <typename T> friend class ::PROTOBUF_NAMESPACE_ID::Arena::InternalHelper;
|
|
typedef void InternalArenaConstructable_;
|
|
typedef void DestructorSkippable_;
|
|
struct Impl_ {
|
|
int64_t source_;
|
|
int64_t target_;
|
|
mutable ::PROTOBUF_NAMESPACE_ID::internal::CachedSize _cached_size_;
|
|
};
|
|
union { Impl_ _impl_; };
|
|
friend struct ::TableStruct_xla_2fxla_5fdata_2eproto;
|
|
};
|
|
// -------------------------------------------------------------------
|
|
|
|
class PrecisionConfig final :
|
|
public ::PROTOBUF_NAMESPACE_ID::Message /* @@protoc_insertion_point(class_definition:xla.PrecisionConfig) */ {
|
|
public:
|
|
inline PrecisionConfig() : PrecisionConfig(nullptr) {}
|
|
~PrecisionConfig() override;
|
|
explicit PROTOBUF_CONSTEXPR PrecisionConfig(::PROTOBUF_NAMESPACE_ID::internal::ConstantInitialized);
|
|
|
|
PrecisionConfig(const PrecisionConfig& from);
|
|
PrecisionConfig(PrecisionConfig&& from) noexcept
|
|
: PrecisionConfig() {
|
|
*this = ::std::move(from);
|
|
}
|
|
|
|
inline PrecisionConfig& operator=(const PrecisionConfig& from) {
|
|
CopyFrom(from);
|
|
return *this;
|
|
}
|
|
inline PrecisionConfig& operator=(PrecisionConfig&& from) noexcept {
|
|
if (this == &from) return *this;
|
|
if (GetOwningArena() == from.GetOwningArena()
|
|
#ifdef PROTOBUF_FORCE_COPY_IN_MOVE
|
|
&& GetOwningArena() != nullptr
|
|
#endif // !PROTOBUF_FORCE_COPY_IN_MOVE
|
|
) {
|
|
InternalSwap(&from);
|
|
} else {
|
|
CopyFrom(from);
|
|
}
|
|
return *this;
|
|
}
|
|
|
|
static const ::PROTOBUF_NAMESPACE_ID::Descriptor* descriptor() {
|
|
return GetDescriptor();
|
|
}
|
|
static const ::PROTOBUF_NAMESPACE_ID::Descriptor* GetDescriptor() {
|
|
return default_instance().GetMetadata().descriptor;
|
|
}
|
|
static const ::PROTOBUF_NAMESPACE_ID::Reflection* GetReflection() {
|
|
return default_instance().GetMetadata().reflection;
|
|
}
|
|
static const PrecisionConfig& default_instance() {
|
|
return *internal_default_instance();
|
|
}
|
|
static inline const PrecisionConfig* internal_default_instance() {
|
|
return reinterpret_cast<const PrecisionConfig*>(
|
|
&_PrecisionConfig_default_instance_);
|
|
}
|
|
static constexpr int kIndexInFileMessages =
|
|
34;
|
|
|
|
friend void swap(PrecisionConfig& a, PrecisionConfig& b) {
|
|
a.Swap(&b);
|
|
}
|
|
inline void Swap(PrecisionConfig* other) {
|
|
if (other == this) return;
|
|
#ifdef PROTOBUF_FORCE_COPY_IN_SWAP
|
|
if (GetOwningArena() != nullptr &&
|
|
GetOwningArena() == other->GetOwningArena()) {
|
|
#else // PROTOBUF_FORCE_COPY_IN_SWAP
|
|
if (GetOwningArena() == other->GetOwningArena()) {
|
|
#endif // !PROTOBUF_FORCE_COPY_IN_SWAP
|
|
InternalSwap(other);
|
|
} else {
|
|
::PROTOBUF_NAMESPACE_ID::internal::GenericSwap(this, other);
|
|
}
|
|
}
|
|
void UnsafeArenaSwap(PrecisionConfig* other) {
|
|
if (other == this) return;
|
|
GOOGLE_DCHECK(GetOwningArena() == other->GetOwningArena());
|
|
InternalSwap(other);
|
|
}
|
|
|
|
// implements Message ----------------------------------------------
|
|
|
|
PrecisionConfig* New(::PROTOBUF_NAMESPACE_ID::Arena* arena = nullptr) const final {
|
|
return CreateMaybeMessage<PrecisionConfig>(arena);
|
|
}
|
|
using ::PROTOBUF_NAMESPACE_ID::Message::CopyFrom;
|
|
void CopyFrom(const PrecisionConfig& from);
|
|
using ::PROTOBUF_NAMESPACE_ID::Message::MergeFrom;
|
|
void MergeFrom( const PrecisionConfig& from) {
|
|
PrecisionConfig::MergeImpl(*this, from);
|
|
}
|
|
private:
|
|
static void MergeImpl(::PROTOBUF_NAMESPACE_ID::Message& to_msg, const ::PROTOBUF_NAMESPACE_ID::Message& from_msg);
|
|
public:
|
|
PROTOBUF_ATTRIBUTE_REINITIALIZES void Clear() final;
|
|
bool IsInitialized() const final;
|
|
|
|
size_t ByteSizeLong() const final;
|
|
const char* _InternalParse(const char* ptr, ::PROTOBUF_NAMESPACE_ID::internal::ParseContext* ctx) final;
|
|
uint8_t* _InternalSerialize(
|
|
uint8_t* target, ::PROTOBUF_NAMESPACE_ID::io::EpsCopyOutputStream* stream) const final;
|
|
int GetCachedSize() const final { return _impl_._cached_size_.Get(); }
|
|
|
|
private:
|
|
void SharedCtor(::PROTOBUF_NAMESPACE_ID::Arena* arena, bool is_message_owned);
|
|
void SharedDtor();
|
|
void SetCachedSize(int size) const final;
|
|
void InternalSwap(PrecisionConfig* other);
|
|
|
|
private:
|
|
friend class ::PROTOBUF_NAMESPACE_ID::internal::AnyMetadata;
|
|
static ::PROTOBUF_NAMESPACE_ID::StringPiece FullMessageName() {
|
|
return "xla.PrecisionConfig";
|
|
}
|
|
protected:
|
|
explicit PrecisionConfig(::PROTOBUF_NAMESPACE_ID::Arena* arena,
|
|
bool is_message_owned = false);
|
|
public:
|
|
|
|
static const ClassData _class_data_;
|
|
const ::PROTOBUF_NAMESPACE_ID::Message::ClassData*GetClassData() const final;
|
|
|
|
::PROTOBUF_NAMESPACE_ID::Metadata GetMetadata() const final;
|
|
|
|
// nested types ----------------------------------------------------
|
|
|
|
typedef PrecisionConfig_Precision Precision;
|
|
static constexpr Precision DEFAULT =
|
|
PrecisionConfig_Precision_DEFAULT;
|
|
static constexpr Precision HIGH =
|
|
PrecisionConfig_Precision_HIGH;
|
|
static constexpr Precision HIGHEST =
|
|
PrecisionConfig_Precision_HIGHEST;
|
|
static constexpr Precision PACKED_NIBBLE =
|
|
PrecisionConfig_Precision_PACKED_NIBBLE;
|
|
static inline bool Precision_IsValid(int value) {
|
|
return PrecisionConfig_Precision_IsValid(value);
|
|
}
|
|
static constexpr Precision Precision_MIN =
|
|
PrecisionConfig_Precision_Precision_MIN;
|
|
static constexpr Precision Precision_MAX =
|
|
PrecisionConfig_Precision_Precision_MAX;
|
|
static constexpr int Precision_ARRAYSIZE =
|
|
PrecisionConfig_Precision_Precision_ARRAYSIZE;
|
|
static inline const ::PROTOBUF_NAMESPACE_ID::EnumDescriptor*
|
|
Precision_descriptor() {
|
|
return PrecisionConfig_Precision_descriptor();
|
|
}
|
|
template<typename T>
|
|
static inline const std::string& Precision_Name(T enum_t_value) {
|
|
static_assert(::std::is_same<T, Precision>::value ||
|
|
::std::is_integral<T>::value,
|
|
"Incorrect type passed to function Precision_Name.");
|
|
return PrecisionConfig_Precision_Name(enum_t_value);
|
|
}
|
|
static inline bool Precision_Parse(::PROTOBUF_NAMESPACE_ID::ConstStringParam name,
|
|
Precision* value) {
|
|
return PrecisionConfig_Precision_Parse(name, value);
|
|
}
|
|
|
|
// accessors -------------------------------------------------------
|
|
|
|
enum : int {
|
|
kOperandPrecisionFieldNumber = 1,
|
|
};
|
|
// repeated .xla.PrecisionConfig.Precision operand_precision = 1;
|
|
int operand_precision_size() const;
|
|
private:
|
|
int _internal_operand_precision_size() const;
|
|
public:
|
|
void clear_operand_precision();
|
|
private:
|
|
::xla::PrecisionConfig_Precision _internal_operand_precision(int index) const;
|
|
void _internal_add_operand_precision(::xla::PrecisionConfig_Precision value);
|
|
::PROTOBUF_NAMESPACE_ID::RepeatedField<int>* _internal_mutable_operand_precision();
|
|
public:
|
|
::xla::PrecisionConfig_Precision operand_precision(int index) const;
|
|
void set_operand_precision(int index, ::xla::PrecisionConfig_Precision value);
|
|
void add_operand_precision(::xla::PrecisionConfig_Precision value);
|
|
const ::PROTOBUF_NAMESPACE_ID::RepeatedField<int>& operand_precision() const;
|
|
::PROTOBUF_NAMESPACE_ID::RepeatedField<int>* mutable_operand_precision();
|
|
|
|
// @@protoc_insertion_point(class_scope:xla.PrecisionConfig)
|
|
private:
|
|
class _Internal;
|
|
|
|
template <typename T> friend class ::PROTOBUF_NAMESPACE_ID::Arena::InternalHelper;
|
|
typedef void InternalArenaConstructable_;
|
|
typedef void DestructorSkippable_;
|
|
struct Impl_ {
|
|
::PROTOBUF_NAMESPACE_ID::RepeatedField<int> operand_precision_;
|
|
mutable std::atomic<int> _operand_precision_cached_byte_size_;
|
|
mutable ::PROTOBUF_NAMESPACE_ID::internal::CachedSize _cached_size_;
|
|
};
|
|
union { Impl_ _impl_; };
|
|
friend struct ::TableStruct_xla_2fxla_5fdata_2eproto;
|
|
};
|
|
// -------------------------------------------------------------------
|
|
|
|
class ParameterReplication final :
|
|
public ::PROTOBUF_NAMESPACE_ID::Message /* @@protoc_insertion_point(class_definition:xla.ParameterReplication) */ {
|
|
public:
|
|
inline ParameterReplication() : ParameterReplication(nullptr) {}
|
|
~ParameterReplication() override;
|
|
explicit PROTOBUF_CONSTEXPR ParameterReplication(::PROTOBUF_NAMESPACE_ID::internal::ConstantInitialized);
|
|
|
|
ParameterReplication(const ParameterReplication& from);
|
|
ParameterReplication(ParameterReplication&& from) noexcept
|
|
: ParameterReplication() {
|
|
*this = ::std::move(from);
|
|
}
|
|
|
|
inline ParameterReplication& operator=(const ParameterReplication& from) {
|
|
CopyFrom(from);
|
|
return *this;
|
|
}
|
|
inline ParameterReplication& operator=(ParameterReplication&& from) noexcept {
|
|
if (this == &from) return *this;
|
|
if (GetOwningArena() == from.GetOwningArena()
|
|
#ifdef PROTOBUF_FORCE_COPY_IN_MOVE
|
|
&& GetOwningArena() != nullptr
|
|
#endif // !PROTOBUF_FORCE_COPY_IN_MOVE
|
|
) {
|
|
InternalSwap(&from);
|
|
} else {
|
|
CopyFrom(from);
|
|
}
|
|
return *this;
|
|
}
|
|
|
|
static const ::PROTOBUF_NAMESPACE_ID::Descriptor* descriptor() {
|
|
return GetDescriptor();
|
|
}
|
|
static const ::PROTOBUF_NAMESPACE_ID::Descriptor* GetDescriptor() {
|
|
return default_instance().GetMetadata().descriptor;
|
|
}
|
|
static const ::PROTOBUF_NAMESPACE_ID::Reflection* GetReflection() {
|
|
return default_instance().GetMetadata().reflection;
|
|
}
|
|
static const ParameterReplication& default_instance() {
|
|
return *internal_default_instance();
|
|
}
|
|
static inline const ParameterReplication* internal_default_instance() {
|
|
return reinterpret_cast<const ParameterReplication*>(
|
|
&_ParameterReplication_default_instance_);
|
|
}
|
|
static constexpr int kIndexInFileMessages =
|
|
35;
|
|
|
|
friend void swap(ParameterReplication& a, ParameterReplication& b) {
|
|
a.Swap(&b);
|
|
}
|
|
inline void Swap(ParameterReplication* other) {
|
|
if (other == this) return;
|
|
#ifdef PROTOBUF_FORCE_COPY_IN_SWAP
|
|
if (GetOwningArena() != nullptr &&
|
|
GetOwningArena() == other->GetOwningArena()) {
|
|
#else // PROTOBUF_FORCE_COPY_IN_SWAP
|
|
if (GetOwningArena() == other->GetOwningArena()) {
|
|
#endif // !PROTOBUF_FORCE_COPY_IN_SWAP
|
|
InternalSwap(other);
|
|
} else {
|
|
::PROTOBUF_NAMESPACE_ID::internal::GenericSwap(this, other);
|
|
}
|
|
}
|
|
void UnsafeArenaSwap(ParameterReplication* other) {
|
|
if (other == this) return;
|
|
GOOGLE_DCHECK(GetOwningArena() == other->GetOwningArena());
|
|
InternalSwap(other);
|
|
}
|
|
|
|
// implements Message ----------------------------------------------
|
|
|
|
ParameterReplication* New(::PROTOBUF_NAMESPACE_ID::Arena* arena = nullptr) const final {
|
|
return CreateMaybeMessage<ParameterReplication>(arena);
|
|
}
|
|
using ::PROTOBUF_NAMESPACE_ID::Message::CopyFrom;
|
|
void CopyFrom(const ParameterReplication& from);
|
|
using ::PROTOBUF_NAMESPACE_ID::Message::MergeFrom;
|
|
void MergeFrom( const ParameterReplication& from) {
|
|
ParameterReplication::MergeImpl(*this, from);
|
|
}
|
|
private:
|
|
static void MergeImpl(::PROTOBUF_NAMESPACE_ID::Message& to_msg, const ::PROTOBUF_NAMESPACE_ID::Message& from_msg);
|
|
public:
|
|
PROTOBUF_ATTRIBUTE_REINITIALIZES void Clear() final;
|
|
bool IsInitialized() const final;
|
|
|
|
size_t ByteSizeLong() const final;
|
|
const char* _InternalParse(const char* ptr, ::PROTOBUF_NAMESPACE_ID::internal::ParseContext* ctx) final;
|
|
uint8_t* _InternalSerialize(
|
|
uint8_t* target, ::PROTOBUF_NAMESPACE_ID::io::EpsCopyOutputStream* stream) const final;
|
|
int GetCachedSize() const final { return _impl_._cached_size_.Get(); }
|
|
|
|
private:
|
|
void SharedCtor(::PROTOBUF_NAMESPACE_ID::Arena* arena, bool is_message_owned);
|
|
void SharedDtor();
|
|
void SetCachedSize(int size) const final;
|
|
void InternalSwap(ParameterReplication* other);
|
|
|
|
private:
|
|
friend class ::PROTOBUF_NAMESPACE_ID::internal::AnyMetadata;
|
|
static ::PROTOBUF_NAMESPACE_ID::StringPiece FullMessageName() {
|
|
return "xla.ParameterReplication";
|
|
}
|
|
protected:
|
|
explicit ParameterReplication(::PROTOBUF_NAMESPACE_ID::Arena* arena,
|
|
bool is_message_owned = false);
|
|
public:
|
|
|
|
static const ClassData _class_data_;
|
|
const ::PROTOBUF_NAMESPACE_ID::Message::ClassData*GetClassData() const final;
|
|
|
|
::PROTOBUF_NAMESPACE_ID::Metadata GetMetadata() const final;
|
|
|
|
// nested types ----------------------------------------------------
|
|
|
|
// accessors -------------------------------------------------------
|
|
|
|
enum : int {
|
|
kReplicatedAtLeafBuffersFieldNumber = 1,
|
|
};
|
|
// repeated bool replicated_at_leaf_buffers = 1;
|
|
int replicated_at_leaf_buffers_size() const;
|
|
private:
|
|
int _internal_replicated_at_leaf_buffers_size() const;
|
|
public:
|
|
void clear_replicated_at_leaf_buffers();
|
|
private:
|
|
bool _internal_replicated_at_leaf_buffers(int index) const;
|
|
const ::PROTOBUF_NAMESPACE_ID::RepeatedField< bool >&
|
|
_internal_replicated_at_leaf_buffers() const;
|
|
void _internal_add_replicated_at_leaf_buffers(bool value);
|
|
::PROTOBUF_NAMESPACE_ID::RepeatedField< bool >*
|
|
_internal_mutable_replicated_at_leaf_buffers();
|
|
public:
|
|
bool replicated_at_leaf_buffers(int index) const;
|
|
void set_replicated_at_leaf_buffers(int index, bool value);
|
|
void add_replicated_at_leaf_buffers(bool value);
|
|
const ::PROTOBUF_NAMESPACE_ID::RepeatedField< bool >&
|
|
replicated_at_leaf_buffers() const;
|
|
::PROTOBUF_NAMESPACE_ID::RepeatedField< bool >*
|
|
mutable_replicated_at_leaf_buffers();
|
|
|
|
// @@protoc_insertion_point(class_scope:xla.ParameterReplication)
|
|
private:
|
|
class _Internal;
|
|
|
|
template <typename T> friend class ::PROTOBUF_NAMESPACE_ID::Arena::InternalHelper;
|
|
typedef void InternalArenaConstructable_;
|
|
typedef void DestructorSkippable_;
|
|
struct Impl_ {
|
|
::PROTOBUF_NAMESPACE_ID::RepeatedField< bool > replicated_at_leaf_buffers_;
|
|
mutable ::PROTOBUF_NAMESPACE_ID::internal::CachedSize _cached_size_;
|
|
};
|
|
union { Impl_ _impl_; };
|
|
friend struct ::TableStruct_xla_2fxla_5fdata_2eproto;
|
|
};
|
|
// -------------------------------------------------------------------
|
|
|
|
class WhileLoopBackendConfig_KnownTripCount final :
|
|
public ::PROTOBUF_NAMESPACE_ID::Message /* @@protoc_insertion_point(class_definition:xla.WhileLoopBackendConfig.KnownTripCount) */ {
|
|
public:
|
|
inline WhileLoopBackendConfig_KnownTripCount() : WhileLoopBackendConfig_KnownTripCount(nullptr) {}
|
|
~WhileLoopBackendConfig_KnownTripCount() override;
|
|
explicit PROTOBUF_CONSTEXPR WhileLoopBackendConfig_KnownTripCount(::PROTOBUF_NAMESPACE_ID::internal::ConstantInitialized);
|
|
|
|
WhileLoopBackendConfig_KnownTripCount(const WhileLoopBackendConfig_KnownTripCount& from);
|
|
WhileLoopBackendConfig_KnownTripCount(WhileLoopBackendConfig_KnownTripCount&& from) noexcept
|
|
: WhileLoopBackendConfig_KnownTripCount() {
|
|
*this = ::std::move(from);
|
|
}
|
|
|
|
inline WhileLoopBackendConfig_KnownTripCount& operator=(const WhileLoopBackendConfig_KnownTripCount& from) {
|
|
CopyFrom(from);
|
|
return *this;
|
|
}
|
|
inline WhileLoopBackendConfig_KnownTripCount& operator=(WhileLoopBackendConfig_KnownTripCount&& from) noexcept {
|
|
if (this == &from) return *this;
|
|
if (GetOwningArena() == from.GetOwningArena()
|
|
#ifdef PROTOBUF_FORCE_COPY_IN_MOVE
|
|
&& GetOwningArena() != nullptr
|
|
#endif // !PROTOBUF_FORCE_COPY_IN_MOVE
|
|
) {
|
|
InternalSwap(&from);
|
|
} else {
|
|
CopyFrom(from);
|
|
}
|
|
return *this;
|
|
}
|
|
|
|
static const ::PROTOBUF_NAMESPACE_ID::Descriptor* descriptor() {
|
|
return GetDescriptor();
|
|
}
|
|
static const ::PROTOBUF_NAMESPACE_ID::Descriptor* GetDescriptor() {
|
|
return default_instance().GetMetadata().descriptor;
|
|
}
|
|
static const ::PROTOBUF_NAMESPACE_ID::Reflection* GetReflection() {
|
|
return default_instance().GetMetadata().reflection;
|
|
}
|
|
static const WhileLoopBackendConfig_KnownTripCount& default_instance() {
|
|
return *internal_default_instance();
|
|
}
|
|
static inline const WhileLoopBackendConfig_KnownTripCount* internal_default_instance() {
|
|
return reinterpret_cast<const WhileLoopBackendConfig_KnownTripCount*>(
|
|
&_WhileLoopBackendConfig_KnownTripCount_default_instance_);
|
|
}
|
|
static constexpr int kIndexInFileMessages =
|
|
36;
|
|
|
|
friend void swap(WhileLoopBackendConfig_KnownTripCount& a, WhileLoopBackendConfig_KnownTripCount& b) {
|
|
a.Swap(&b);
|
|
}
|
|
inline void Swap(WhileLoopBackendConfig_KnownTripCount* other) {
|
|
if (other == this) return;
|
|
#ifdef PROTOBUF_FORCE_COPY_IN_SWAP
|
|
if (GetOwningArena() != nullptr &&
|
|
GetOwningArena() == other->GetOwningArena()) {
|
|
#else // PROTOBUF_FORCE_COPY_IN_SWAP
|
|
if (GetOwningArena() == other->GetOwningArena()) {
|
|
#endif // !PROTOBUF_FORCE_COPY_IN_SWAP
|
|
InternalSwap(other);
|
|
} else {
|
|
::PROTOBUF_NAMESPACE_ID::internal::GenericSwap(this, other);
|
|
}
|
|
}
|
|
void UnsafeArenaSwap(WhileLoopBackendConfig_KnownTripCount* other) {
|
|
if (other == this) return;
|
|
GOOGLE_DCHECK(GetOwningArena() == other->GetOwningArena());
|
|
InternalSwap(other);
|
|
}
|
|
|
|
// implements Message ----------------------------------------------
|
|
|
|
WhileLoopBackendConfig_KnownTripCount* New(::PROTOBUF_NAMESPACE_ID::Arena* arena = nullptr) const final {
|
|
return CreateMaybeMessage<WhileLoopBackendConfig_KnownTripCount>(arena);
|
|
}
|
|
using ::PROTOBUF_NAMESPACE_ID::Message::CopyFrom;
|
|
void CopyFrom(const WhileLoopBackendConfig_KnownTripCount& from);
|
|
using ::PROTOBUF_NAMESPACE_ID::Message::MergeFrom;
|
|
void MergeFrom( const WhileLoopBackendConfig_KnownTripCount& from) {
|
|
WhileLoopBackendConfig_KnownTripCount::MergeImpl(*this, from);
|
|
}
|
|
private:
|
|
static void MergeImpl(::PROTOBUF_NAMESPACE_ID::Message& to_msg, const ::PROTOBUF_NAMESPACE_ID::Message& from_msg);
|
|
public:
|
|
PROTOBUF_ATTRIBUTE_REINITIALIZES void Clear() final;
|
|
bool IsInitialized() const final;
|
|
|
|
size_t ByteSizeLong() const final;
|
|
const char* _InternalParse(const char* ptr, ::PROTOBUF_NAMESPACE_ID::internal::ParseContext* ctx) final;
|
|
uint8_t* _InternalSerialize(
|
|
uint8_t* target, ::PROTOBUF_NAMESPACE_ID::io::EpsCopyOutputStream* stream) const final;
|
|
int GetCachedSize() const final { return _impl_._cached_size_.Get(); }
|
|
|
|
private:
|
|
void SharedCtor(::PROTOBUF_NAMESPACE_ID::Arena* arena, bool is_message_owned);
|
|
void SharedDtor();
|
|
void SetCachedSize(int size) const final;
|
|
void InternalSwap(WhileLoopBackendConfig_KnownTripCount* other);
|
|
|
|
private:
|
|
friend class ::PROTOBUF_NAMESPACE_ID::internal::AnyMetadata;
|
|
static ::PROTOBUF_NAMESPACE_ID::StringPiece FullMessageName() {
|
|
return "xla.WhileLoopBackendConfig.KnownTripCount";
|
|
}
|
|
protected:
|
|
explicit WhileLoopBackendConfig_KnownTripCount(::PROTOBUF_NAMESPACE_ID::Arena* arena,
|
|
bool is_message_owned = false);
|
|
public:
|
|
|
|
static const ClassData _class_data_;
|
|
const ::PROTOBUF_NAMESPACE_ID::Message::ClassData*GetClassData() const final;
|
|
|
|
::PROTOBUF_NAMESPACE_ID::Metadata GetMetadata() const final;
|
|
|
|
// nested types ----------------------------------------------------
|
|
|
|
// accessors -------------------------------------------------------
|
|
|
|
enum : int {
|
|
kNFieldNumber = 1,
|
|
};
|
|
// int64 n = 1;
|
|
void clear_n();
|
|
int64_t n() const;
|
|
void set_n(int64_t value);
|
|
private:
|
|
int64_t _internal_n() const;
|
|
void _internal_set_n(int64_t value);
|
|
public:
|
|
|
|
// @@protoc_insertion_point(class_scope:xla.WhileLoopBackendConfig.KnownTripCount)
|
|
private:
|
|
class _Internal;
|
|
|
|
template <typename T> friend class ::PROTOBUF_NAMESPACE_ID::Arena::InternalHelper;
|
|
typedef void InternalArenaConstructable_;
|
|
typedef void DestructorSkippable_;
|
|
struct Impl_ {
|
|
int64_t n_;
|
|
mutable ::PROTOBUF_NAMESPACE_ID::internal::CachedSize _cached_size_;
|
|
};
|
|
union { Impl_ _impl_; };
|
|
friend struct ::TableStruct_xla_2fxla_5fdata_2eproto;
|
|
};
|
|
// -------------------------------------------------------------------
|
|
|
|
class WhileLoopBackendConfig final :
|
|
public ::PROTOBUF_NAMESPACE_ID::Message /* @@protoc_insertion_point(class_definition:xla.WhileLoopBackendConfig) */ {
|
|
public:
|
|
inline WhileLoopBackendConfig() : WhileLoopBackendConfig(nullptr) {}
|
|
~WhileLoopBackendConfig() override;
|
|
explicit PROTOBUF_CONSTEXPR WhileLoopBackendConfig(::PROTOBUF_NAMESPACE_ID::internal::ConstantInitialized);
|
|
|
|
WhileLoopBackendConfig(const WhileLoopBackendConfig& from);
|
|
WhileLoopBackendConfig(WhileLoopBackendConfig&& from) noexcept
|
|
: WhileLoopBackendConfig() {
|
|
*this = ::std::move(from);
|
|
}
|
|
|
|
inline WhileLoopBackendConfig& operator=(const WhileLoopBackendConfig& from) {
|
|
CopyFrom(from);
|
|
return *this;
|
|
}
|
|
inline WhileLoopBackendConfig& operator=(WhileLoopBackendConfig&& from) noexcept {
|
|
if (this == &from) return *this;
|
|
if (GetOwningArena() == from.GetOwningArena()
|
|
#ifdef PROTOBUF_FORCE_COPY_IN_MOVE
|
|
&& GetOwningArena() != nullptr
|
|
#endif // !PROTOBUF_FORCE_COPY_IN_MOVE
|
|
) {
|
|
InternalSwap(&from);
|
|
} else {
|
|
CopyFrom(from);
|
|
}
|
|
return *this;
|
|
}
|
|
|
|
static const ::PROTOBUF_NAMESPACE_ID::Descriptor* descriptor() {
|
|
return GetDescriptor();
|
|
}
|
|
static const ::PROTOBUF_NAMESPACE_ID::Descriptor* GetDescriptor() {
|
|
return default_instance().GetMetadata().descriptor;
|
|
}
|
|
static const ::PROTOBUF_NAMESPACE_ID::Reflection* GetReflection() {
|
|
return default_instance().GetMetadata().reflection;
|
|
}
|
|
static const WhileLoopBackendConfig& default_instance() {
|
|
return *internal_default_instance();
|
|
}
|
|
static inline const WhileLoopBackendConfig* internal_default_instance() {
|
|
return reinterpret_cast<const WhileLoopBackendConfig*>(
|
|
&_WhileLoopBackendConfig_default_instance_);
|
|
}
|
|
static constexpr int kIndexInFileMessages =
|
|
37;
|
|
|
|
friend void swap(WhileLoopBackendConfig& a, WhileLoopBackendConfig& b) {
|
|
a.Swap(&b);
|
|
}
|
|
inline void Swap(WhileLoopBackendConfig* other) {
|
|
if (other == this) return;
|
|
#ifdef PROTOBUF_FORCE_COPY_IN_SWAP
|
|
if (GetOwningArena() != nullptr &&
|
|
GetOwningArena() == other->GetOwningArena()) {
|
|
#else // PROTOBUF_FORCE_COPY_IN_SWAP
|
|
if (GetOwningArena() == other->GetOwningArena()) {
|
|
#endif // !PROTOBUF_FORCE_COPY_IN_SWAP
|
|
InternalSwap(other);
|
|
} else {
|
|
::PROTOBUF_NAMESPACE_ID::internal::GenericSwap(this, other);
|
|
}
|
|
}
|
|
void UnsafeArenaSwap(WhileLoopBackendConfig* other) {
|
|
if (other == this) return;
|
|
GOOGLE_DCHECK(GetOwningArena() == other->GetOwningArena());
|
|
InternalSwap(other);
|
|
}
|
|
|
|
// implements Message ----------------------------------------------
|
|
|
|
WhileLoopBackendConfig* New(::PROTOBUF_NAMESPACE_ID::Arena* arena = nullptr) const final {
|
|
return CreateMaybeMessage<WhileLoopBackendConfig>(arena);
|
|
}
|
|
using ::PROTOBUF_NAMESPACE_ID::Message::CopyFrom;
|
|
void CopyFrom(const WhileLoopBackendConfig& from);
|
|
using ::PROTOBUF_NAMESPACE_ID::Message::MergeFrom;
|
|
void MergeFrom( const WhileLoopBackendConfig& from) {
|
|
WhileLoopBackendConfig::MergeImpl(*this, from);
|
|
}
|
|
private:
|
|
static void MergeImpl(::PROTOBUF_NAMESPACE_ID::Message& to_msg, const ::PROTOBUF_NAMESPACE_ID::Message& from_msg);
|
|
public:
|
|
PROTOBUF_ATTRIBUTE_REINITIALIZES void Clear() final;
|
|
bool IsInitialized() const final;
|
|
|
|
size_t ByteSizeLong() const final;
|
|
const char* _InternalParse(const char* ptr, ::PROTOBUF_NAMESPACE_ID::internal::ParseContext* ctx) final;
|
|
uint8_t* _InternalSerialize(
|
|
uint8_t* target, ::PROTOBUF_NAMESPACE_ID::io::EpsCopyOutputStream* stream) const final;
|
|
int GetCachedSize() const final { return _impl_._cached_size_.Get(); }
|
|
|
|
private:
|
|
void SharedCtor(::PROTOBUF_NAMESPACE_ID::Arena* arena, bool is_message_owned);
|
|
void SharedDtor();
|
|
void SetCachedSize(int size) const final;
|
|
void InternalSwap(WhileLoopBackendConfig* other);
|
|
|
|
private:
|
|
friend class ::PROTOBUF_NAMESPACE_ID::internal::AnyMetadata;
|
|
static ::PROTOBUF_NAMESPACE_ID::StringPiece FullMessageName() {
|
|
return "xla.WhileLoopBackendConfig";
|
|
}
|
|
protected:
|
|
explicit WhileLoopBackendConfig(::PROTOBUF_NAMESPACE_ID::Arena* arena,
|
|
bool is_message_owned = false);
|
|
public:
|
|
|
|
static const ClassData _class_data_;
|
|
const ::PROTOBUF_NAMESPACE_ID::Message::ClassData*GetClassData() const final;
|
|
|
|
::PROTOBUF_NAMESPACE_ID::Metadata GetMetadata() const final;
|
|
|
|
// nested types ----------------------------------------------------
|
|
|
|
typedef WhileLoopBackendConfig_KnownTripCount KnownTripCount;
|
|
|
|
// accessors -------------------------------------------------------
|
|
|
|
enum : int {
|
|
kKnownTripCountFieldNumber = 1,
|
|
};
|
|
// .xla.WhileLoopBackendConfig.KnownTripCount known_trip_count = 1;
|
|
bool has_known_trip_count() const;
|
|
private:
|
|
bool _internal_has_known_trip_count() const;
|
|
public:
|
|
void clear_known_trip_count();
|
|
const ::xla::WhileLoopBackendConfig_KnownTripCount& known_trip_count() const;
|
|
PROTOBUF_NODISCARD ::xla::WhileLoopBackendConfig_KnownTripCount* release_known_trip_count();
|
|
::xla::WhileLoopBackendConfig_KnownTripCount* mutable_known_trip_count();
|
|
void set_allocated_known_trip_count(::xla::WhileLoopBackendConfig_KnownTripCount* known_trip_count);
|
|
private:
|
|
const ::xla::WhileLoopBackendConfig_KnownTripCount& _internal_known_trip_count() const;
|
|
::xla::WhileLoopBackendConfig_KnownTripCount* _internal_mutable_known_trip_count();
|
|
public:
|
|
void unsafe_arena_set_allocated_known_trip_count(
|
|
::xla::WhileLoopBackendConfig_KnownTripCount* known_trip_count);
|
|
::xla::WhileLoopBackendConfig_KnownTripCount* unsafe_arena_release_known_trip_count();
|
|
|
|
// @@protoc_insertion_point(class_scope:xla.WhileLoopBackendConfig)
|
|
private:
|
|
class _Internal;
|
|
|
|
template <typename T> friend class ::PROTOBUF_NAMESPACE_ID::Arena::InternalHelper;
|
|
typedef void InternalArenaConstructable_;
|
|
typedef void DestructorSkippable_;
|
|
struct Impl_ {
|
|
::xla::WhileLoopBackendConfig_KnownTripCount* known_trip_count_;
|
|
mutable ::PROTOBUF_NAMESPACE_ID::internal::CachedSize _cached_size_;
|
|
};
|
|
union { Impl_ _impl_; };
|
|
friend struct ::TableStruct_xla_2fxla_5fdata_2eproto;
|
|
};
|
|
// -------------------------------------------------------------------
|
|
|
|
class OutputOperandAliasing final :
|
|
public ::PROTOBUF_NAMESPACE_ID::Message /* @@protoc_insertion_point(class_definition:xla.OutputOperandAliasing) */ {
|
|
public:
|
|
inline OutputOperandAliasing() : OutputOperandAliasing(nullptr) {}
|
|
~OutputOperandAliasing() override;
|
|
explicit PROTOBUF_CONSTEXPR OutputOperandAliasing(::PROTOBUF_NAMESPACE_ID::internal::ConstantInitialized);
|
|
|
|
OutputOperandAliasing(const OutputOperandAliasing& from);
|
|
OutputOperandAliasing(OutputOperandAliasing&& from) noexcept
|
|
: OutputOperandAliasing() {
|
|
*this = ::std::move(from);
|
|
}
|
|
|
|
inline OutputOperandAliasing& operator=(const OutputOperandAliasing& from) {
|
|
CopyFrom(from);
|
|
return *this;
|
|
}
|
|
inline OutputOperandAliasing& operator=(OutputOperandAliasing&& from) noexcept {
|
|
if (this == &from) return *this;
|
|
if (GetOwningArena() == from.GetOwningArena()
|
|
#ifdef PROTOBUF_FORCE_COPY_IN_MOVE
|
|
&& GetOwningArena() != nullptr
|
|
#endif // !PROTOBUF_FORCE_COPY_IN_MOVE
|
|
) {
|
|
InternalSwap(&from);
|
|
} else {
|
|
CopyFrom(from);
|
|
}
|
|
return *this;
|
|
}
|
|
|
|
static const ::PROTOBUF_NAMESPACE_ID::Descriptor* descriptor() {
|
|
return GetDescriptor();
|
|
}
|
|
static const ::PROTOBUF_NAMESPACE_ID::Descriptor* GetDescriptor() {
|
|
return default_instance().GetMetadata().descriptor;
|
|
}
|
|
static const ::PROTOBUF_NAMESPACE_ID::Reflection* GetReflection() {
|
|
return default_instance().GetMetadata().reflection;
|
|
}
|
|
static const OutputOperandAliasing& default_instance() {
|
|
return *internal_default_instance();
|
|
}
|
|
static inline const OutputOperandAliasing* internal_default_instance() {
|
|
return reinterpret_cast<const OutputOperandAliasing*>(
|
|
&_OutputOperandAliasing_default_instance_);
|
|
}
|
|
static constexpr int kIndexInFileMessages =
|
|
38;
|
|
|
|
friend void swap(OutputOperandAliasing& a, OutputOperandAliasing& b) {
|
|
a.Swap(&b);
|
|
}
|
|
inline void Swap(OutputOperandAliasing* other) {
|
|
if (other == this) return;
|
|
#ifdef PROTOBUF_FORCE_COPY_IN_SWAP
|
|
if (GetOwningArena() != nullptr &&
|
|
GetOwningArena() == other->GetOwningArena()) {
|
|
#else // PROTOBUF_FORCE_COPY_IN_SWAP
|
|
if (GetOwningArena() == other->GetOwningArena()) {
|
|
#endif // !PROTOBUF_FORCE_COPY_IN_SWAP
|
|
InternalSwap(other);
|
|
} else {
|
|
::PROTOBUF_NAMESPACE_ID::internal::GenericSwap(this, other);
|
|
}
|
|
}
|
|
void UnsafeArenaSwap(OutputOperandAliasing* other) {
|
|
if (other == this) return;
|
|
GOOGLE_DCHECK(GetOwningArena() == other->GetOwningArena());
|
|
InternalSwap(other);
|
|
}
|
|
|
|
// implements Message ----------------------------------------------
|
|
|
|
OutputOperandAliasing* New(::PROTOBUF_NAMESPACE_ID::Arena* arena = nullptr) const final {
|
|
return CreateMaybeMessage<OutputOperandAliasing>(arena);
|
|
}
|
|
using ::PROTOBUF_NAMESPACE_ID::Message::CopyFrom;
|
|
void CopyFrom(const OutputOperandAliasing& from);
|
|
using ::PROTOBUF_NAMESPACE_ID::Message::MergeFrom;
|
|
void MergeFrom( const OutputOperandAliasing& from) {
|
|
OutputOperandAliasing::MergeImpl(*this, from);
|
|
}
|
|
private:
|
|
static void MergeImpl(::PROTOBUF_NAMESPACE_ID::Message& to_msg, const ::PROTOBUF_NAMESPACE_ID::Message& from_msg);
|
|
public:
|
|
PROTOBUF_ATTRIBUTE_REINITIALIZES void Clear() final;
|
|
bool IsInitialized() const final;
|
|
|
|
size_t ByteSizeLong() const final;
|
|
const char* _InternalParse(const char* ptr, ::PROTOBUF_NAMESPACE_ID::internal::ParseContext* ctx) final;
|
|
uint8_t* _InternalSerialize(
|
|
uint8_t* target, ::PROTOBUF_NAMESPACE_ID::io::EpsCopyOutputStream* stream) const final;
|
|
int GetCachedSize() const final { return _impl_._cached_size_.Get(); }
|
|
|
|
private:
|
|
void SharedCtor(::PROTOBUF_NAMESPACE_ID::Arena* arena, bool is_message_owned);
|
|
void SharedDtor();
|
|
void SetCachedSize(int size) const final;
|
|
void InternalSwap(OutputOperandAliasing* other);
|
|
|
|
private:
|
|
friend class ::PROTOBUF_NAMESPACE_ID::internal::AnyMetadata;
|
|
static ::PROTOBUF_NAMESPACE_ID::StringPiece FullMessageName() {
|
|
return "xla.OutputOperandAliasing";
|
|
}
|
|
protected:
|
|
explicit OutputOperandAliasing(::PROTOBUF_NAMESPACE_ID::Arena* arena,
|
|
bool is_message_owned = false);
|
|
public:
|
|
|
|
static const ClassData _class_data_;
|
|
const ::PROTOBUF_NAMESPACE_ID::Message::ClassData*GetClassData() const final;
|
|
|
|
::PROTOBUF_NAMESPACE_ID::Metadata GetMetadata() const final;
|
|
|
|
// nested types ----------------------------------------------------
|
|
|
|
// accessors -------------------------------------------------------
|
|
|
|
enum : int {
|
|
kOutputShapeIndexFieldNumber = 1,
|
|
kOperandShapeIndexFieldNumber = 3,
|
|
kOperandIndexFieldNumber = 2,
|
|
};
|
|
// repeated int64 output_shape_index = 1;
|
|
int output_shape_index_size() const;
|
|
private:
|
|
int _internal_output_shape_index_size() const;
|
|
public:
|
|
void clear_output_shape_index();
|
|
private:
|
|
int64_t _internal_output_shape_index(int index) const;
|
|
const ::PROTOBUF_NAMESPACE_ID::RepeatedField< int64_t >&
|
|
_internal_output_shape_index() const;
|
|
void _internal_add_output_shape_index(int64_t value);
|
|
::PROTOBUF_NAMESPACE_ID::RepeatedField< int64_t >*
|
|
_internal_mutable_output_shape_index();
|
|
public:
|
|
int64_t output_shape_index(int index) const;
|
|
void set_output_shape_index(int index, int64_t value);
|
|
void add_output_shape_index(int64_t value);
|
|
const ::PROTOBUF_NAMESPACE_ID::RepeatedField< int64_t >&
|
|
output_shape_index() const;
|
|
::PROTOBUF_NAMESPACE_ID::RepeatedField< int64_t >*
|
|
mutable_output_shape_index();
|
|
|
|
// repeated int64 operand_shape_index = 3;
|
|
int operand_shape_index_size() const;
|
|
private:
|
|
int _internal_operand_shape_index_size() const;
|
|
public:
|
|
void clear_operand_shape_index();
|
|
private:
|
|
int64_t _internal_operand_shape_index(int index) const;
|
|
const ::PROTOBUF_NAMESPACE_ID::RepeatedField< int64_t >&
|
|
_internal_operand_shape_index() const;
|
|
void _internal_add_operand_shape_index(int64_t value);
|
|
::PROTOBUF_NAMESPACE_ID::RepeatedField< int64_t >*
|
|
_internal_mutable_operand_shape_index();
|
|
public:
|
|
int64_t operand_shape_index(int index) const;
|
|
void set_operand_shape_index(int index, int64_t value);
|
|
void add_operand_shape_index(int64_t value);
|
|
const ::PROTOBUF_NAMESPACE_ID::RepeatedField< int64_t >&
|
|
operand_shape_index() const;
|
|
::PROTOBUF_NAMESPACE_ID::RepeatedField< int64_t >*
|
|
mutable_operand_shape_index();
|
|
|
|
// int64 operand_index = 2;
|
|
void clear_operand_index();
|
|
int64_t operand_index() const;
|
|
void set_operand_index(int64_t value);
|
|
private:
|
|
int64_t _internal_operand_index() const;
|
|
void _internal_set_operand_index(int64_t value);
|
|
public:
|
|
|
|
// @@protoc_insertion_point(class_scope:xla.OutputOperandAliasing)
|
|
private:
|
|
class _Internal;
|
|
|
|
template <typename T> friend class ::PROTOBUF_NAMESPACE_ID::Arena::InternalHelper;
|
|
typedef void InternalArenaConstructable_;
|
|
typedef void DestructorSkippable_;
|
|
struct Impl_ {
|
|
::PROTOBUF_NAMESPACE_ID::RepeatedField< int64_t > output_shape_index_;
|
|
mutable std::atomic<int> _output_shape_index_cached_byte_size_;
|
|
::PROTOBUF_NAMESPACE_ID::RepeatedField< int64_t > operand_shape_index_;
|
|
mutable std::atomic<int> _operand_shape_index_cached_byte_size_;
|
|
int64_t operand_index_;
|
|
mutable ::PROTOBUF_NAMESPACE_ID::internal::CachedSize _cached_size_;
|
|
};
|
|
union { Impl_ _impl_; };
|
|
friend struct ::TableStruct_xla_2fxla_5fdata_2eproto;
|
|
};
|
|
// ===================================================================
|
|
|
|
|
|
// ===================================================================
|
|
|
|
#ifdef __GNUC__
|
|
#pragma GCC diagnostic push
|
|
#pragma GCC diagnostic ignored "-Wstrict-aliasing"
|
|
#endif // __GNUC__
|
|
// PaddingConfig_PaddingConfigDimension
|
|
|
|
// int64 edge_padding_low = 1;
|
|
inline void PaddingConfig_PaddingConfigDimension::clear_edge_padding_low() {
|
|
_impl_.edge_padding_low_ = int64_t{0};
|
|
}
|
|
inline int64_t PaddingConfig_PaddingConfigDimension::_internal_edge_padding_low() const {
|
|
return _impl_.edge_padding_low_;
|
|
}
|
|
inline int64_t PaddingConfig_PaddingConfigDimension::edge_padding_low() const {
|
|
// @@protoc_insertion_point(field_get:xla.PaddingConfig.PaddingConfigDimension.edge_padding_low)
|
|
return _internal_edge_padding_low();
|
|
}
|
|
inline void PaddingConfig_PaddingConfigDimension::_internal_set_edge_padding_low(int64_t value) {
|
|
|
|
_impl_.edge_padding_low_ = value;
|
|
}
|
|
inline void PaddingConfig_PaddingConfigDimension::set_edge_padding_low(int64_t value) {
|
|
_internal_set_edge_padding_low(value);
|
|
// @@protoc_insertion_point(field_set:xla.PaddingConfig.PaddingConfigDimension.edge_padding_low)
|
|
}
|
|
|
|
// int64 edge_padding_high = 2;
|
|
inline void PaddingConfig_PaddingConfigDimension::clear_edge_padding_high() {
|
|
_impl_.edge_padding_high_ = int64_t{0};
|
|
}
|
|
inline int64_t PaddingConfig_PaddingConfigDimension::_internal_edge_padding_high() const {
|
|
return _impl_.edge_padding_high_;
|
|
}
|
|
inline int64_t PaddingConfig_PaddingConfigDimension::edge_padding_high() const {
|
|
// @@protoc_insertion_point(field_get:xla.PaddingConfig.PaddingConfigDimension.edge_padding_high)
|
|
return _internal_edge_padding_high();
|
|
}
|
|
inline void PaddingConfig_PaddingConfigDimension::_internal_set_edge_padding_high(int64_t value) {
|
|
|
|
_impl_.edge_padding_high_ = value;
|
|
}
|
|
inline void PaddingConfig_PaddingConfigDimension::set_edge_padding_high(int64_t value) {
|
|
_internal_set_edge_padding_high(value);
|
|
// @@protoc_insertion_point(field_set:xla.PaddingConfig.PaddingConfigDimension.edge_padding_high)
|
|
}
|
|
|
|
// int64 interior_padding = 3;
|
|
inline void PaddingConfig_PaddingConfigDimension::clear_interior_padding() {
|
|
_impl_.interior_padding_ = int64_t{0};
|
|
}
|
|
inline int64_t PaddingConfig_PaddingConfigDimension::_internal_interior_padding() const {
|
|
return _impl_.interior_padding_;
|
|
}
|
|
inline int64_t PaddingConfig_PaddingConfigDimension::interior_padding() const {
|
|
// @@protoc_insertion_point(field_get:xla.PaddingConfig.PaddingConfigDimension.interior_padding)
|
|
return _internal_interior_padding();
|
|
}
|
|
inline void PaddingConfig_PaddingConfigDimension::_internal_set_interior_padding(int64_t value) {
|
|
|
|
_impl_.interior_padding_ = value;
|
|
}
|
|
inline void PaddingConfig_PaddingConfigDimension::set_interior_padding(int64_t value) {
|
|
_internal_set_interior_padding(value);
|
|
// @@protoc_insertion_point(field_set:xla.PaddingConfig.PaddingConfigDimension.interior_padding)
|
|
}
|
|
|
|
// -------------------------------------------------------------------
|
|
|
|
// PaddingConfig
|
|
|
|
// repeated .xla.PaddingConfig.PaddingConfigDimension dimensions = 1;
|
|
inline int PaddingConfig::_internal_dimensions_size() const {
|
|
return _impl_.dimensions_.size();
|
|
}
|
|
inline int PaddingConfig::dimensions_size() const {
|
|
return _internal_dimensions_size();
|
|
}
|
|
inline void PaddingConfig::clear_dimensions() {
|
|
_impl_.dimensions_.Clear();
|
|
}
|
|
inline ::xla::PaddingConfig_PaddingConfigDimension* PaddingConfig::mutable_dimensions(int index) {
|
|
// @@protoc_insertion_point(field_mutable:xla.PaddingConfig.dimensions)
|
|
return _impl_.dimensions_.Mutable(index);
|
|
}
|
|
inline ::PROTOBUF_NAMESPACE_ID::RepeatedPtrField< ::xla::PaddingConfig_PaddingConfigDimension >*
|
|
PaddingConfig::mutable_dimensions() {
|
|
// @@protoc_insertion_point(field_mutable_list:xla.PaddingConfig.dimensions)
|
|
return &_impl_.dimensions_;
|
|
}
|
|
inline const ::xla::PaddingConfig_PaddingConfigDimension& PaddingConfig::_internal_dimensions(int index) const {
|
|
return _impl_.dimensions_.Get(index);
|
|
}
|
|
inline const ::xla::PaddingConfig_PaddingConfigDimension& PaddingConfig::dimensions(int index) const {
|
|
// @@protoc_insertion_point(field_get:xla.PaddingConfig.dimensions)
|
|
return _internal_dimensions(index);
|
|
}
|
|
inline ::xla::PaddingConfig_PaddingConfigDimension* PaddingConfig::_internal_add_dimensions() {
|
|
return _impl_.dimensions_.Add();
|
|
}
|
|
inline ::xla::PaddingConfig_PaddingConfigDimension* PaddingConfig::add_dimensions() {
|
|
::xla::PaddingConfig_PaddingConfigDimension* _add = _internal_add_dimensions();
|
|
// @@protoc_insertion_point(field_add:xla.PaddingConfig.dimensions)
|
|
return _add;
|
|
}
|
|
inline const ::PROTOBUF_NAMESPACE_ID::RepeatedPtrField< ::xla::PaddingConfig_PaddingConfigDimension >&
|
|
PaddingConfig::dimensions() const {
|
|
// @@protoc_insertion_point(field_list:xla.PaddingConfig.dimensions)
|
|
return _impl_.dimensions_;
|
|
}
|
|
|
|
// -------------------------------------------------------------------
|
|
|
|
// TileProto
|
|
|
|
// repeated int64 dimensions = 1;
|
|
inline int TileProto::_internal_dimensions_size() const {
|
|
return _impl_.dimensions_.size();
|
|
}
|
|
inline int TileProto::dimensions_size() const {
|
|
return _internal_dimensions_size();
|
|
}
|
|
inline void TileProto::clear_dimensions() {
|
|
_impl_.dimensions_.Clear();
|
|
}
|
|
inline int64_t TileProto::_internal_dimensions(int index) const {
|
|
return _impl_.dimensions_.Get(index);
|
|
}
|
|
inline int64_t TileProto::dimensions(int index) const {
|
|
// @@protoc_insertion_point(field_get:xla.TileProto.dimensions)
|
|
return _internal_dimensions(index);
|
|
}
|
|
inline void TileProto::set_dimensions(int index, int64_t value) {
|
|
_impl_.dimensions_.Set(index, value);
|
|
// @@protoc_insertion_point(field_set:xla.TileProto.dimensions)
|
|
}
|
|
inline void TileProto::_internal_add_dimensions(int64_t value) {
|
|
_impl_.dimensions_.Add(value);
|
|
}
|
|
inline void TileProto::add_dimensions(int64_t value) {
|
|
_internal_add_dimensions(value);
|
|
// @@protoc_insertion_point(field_add:xla.TileProto.dimensions)
|
|
}
|
|
inline const ::PROTOBUF_NAMESPACE_ID::RepeatedField< int64_t >&
|
|
TileProto::_internal_dimensions() const {
|
|
return _impl_.dimensions_;
|
|
}
|
|
inline const ::PROTOBUF_NAMESPACE_ID::RepeatedField< int64_t >&
|
|
TileProto::dimensions() const {
|
|
// @@protoc_insertion_point(field_list:xla.TileProto.dimensions)
|
|
return _internal_dimensions();
|
|
}
|
|
inline ::PROTOBUF_NAMESPACE_ID::RepeatedField< int64_t >*
|
|
TileProto::_internal_mutable_dimensions() {
|
|
return &_impl_.dimensions_;
|
|
}
|
|
inline ::PROTOBUF_NAMESPACE_ID::RepeatedField< int64_t >*
|
|
TileProto::mutable_dimensions() {
|
|
// @@protoc_insertion_point(field_mutable_list:xla.TileProto.dimensions)
|
|
return _internal_mutable_dimensions();
|
|
}
|
|
|
|
// -------------------------------------------------------------------
|
|
|
|
// LayoutProto
|
|
|
|
// repeated .xla.DimLevelType dim_level_types = 9;
|
|
inline int LayoutProto::_internal_dim_level_types_size() const {
|
|
return _impl_.dim_level_types_.size();
|
|
}
|
|
inline int LayoutProto::dim_level_types_size() const {
|
|
return _internal_dim_level_types_size();
|
|
}
|
|
inline void LayoutProto::clear_dim_level_types() {
|
|
_impl_.dim_level_types_.Clear();
|
|
}
|
|
inline ::xla::DimLevelType LayoutProto::_internal_dim_level_types(int index) const {
|
|
return static_cast< ::xla::DimLevelType >(_impl_.dim_level_types_.Get(index));
|
|
}
|
|
inline ::xla::DimLevelType LayoutProto::dim_level_types(int index) const {
|
|
// @@protoc_insertion_point(field_get:xla.LayoutProto.dim_level_types)
|
|
return _internal_dim_level_types(index);
|
|
}
|
|
inline void LayoutProto::set_dim_level_types(int index, ::xla::DimLevelType value) {
|
|
_impl_.dim_level_types_.Set(index, value);
|
|
// @@protoc_insertion_point(field_set:xla.LayoutProto.dim_level_types)
|
|
}
|
|
inline void LayoutProto::_internal_add_dim_level_types(::xla::DimLevelType value) {
|
|
_impl_.dim_level_types_.Add(value);
|
|
}
|
|
inline void LayoutProto::add_dim_level_types(::xla::DimLevelType value) {
|
|
_internal_add_dim_level_types(value);
|
|
// @@protoc_insertion_point(field_add:xla.LayoutProto.dim_level_types)
|
|
}
|
|
inline const ::PROTOBUF_NAMESPACE_ID::RepeatedField<int>&
|
|
LayoutProto::dim_level_types() const {
|
|
// @@protoc_insertion_point(field_list:xla.LayoutProto.dim_level_types)
|
|
return _impl_.dim_level_types_;
|
|
}
|
|
inline ::PROTOBUF_NAMESPACE_ID::RepeatedField<int>*
|
|
LayoutProto::_internal_mutable_dim_level_types() {
|
|
return &_impl_.dim_level_types_;
|
|
}
|
|
inline ::PROTOBUF_NAMESPACE_ID::RepeatedField<int>*
|
|
LayoutProto::mutable_dim_level_types() {
|
|
// @@protoc_insertion_point(field_mutable_list:xla.LayoutProto.dim_level_types)
|
|
return _internal_mutable_dim_level_types();
|
|
}
|
|
|
|
// repeated bool dim_unique = 13;
|
|
inline int LayoutProto::_internal_dim_unique_size() const {
|
|
return _impl_.dim_unique_.size();
|
|
}
|
|
inline int LayoutProto::dim_unique_size() const {
|
|
return _internal_dim_unique_size();
|
|
}
|
|
inline void LayoutProto::clear_dim_unique() {
|
|
_impl_.dim_unique_.Clear();
|
|
}
|
|
inline bool LayoutProto::_internal_dim_unique(int index) const {
|
|
return _impl_.dim_unique_.Get(index);
|
|
}
|
|
inline bool LayoutProto::dim_unique(int index) const {
|
|
// @@protoc_insertion_point(field_get:xla.LayoutProto.dim_unique)
|
|
return _internal_dim_unique(index);
|
|
}
|
|
inline void LayoutProto::set_dim_unique(int index, bool value) {
|
|
_impl_.dim_unique_.Set(index, value);
|
|
// @@protoc_insertion_point(field_set:xla.LayoutProto.dim_unique)
|
|
}
|
|
inline void LayoutProto::_internal_add_dim_unique(bool value) {
|
|
_impl_.dim_unique_.Add(value);
|
|
}
|
|
inline void LayoutProto::add_dim_unique(bool value) {
|
|
_internal_add_dim_unique(value);
|
|
// @@protoc_insertion_point(field_add:xla.LayoutProto.dim_unique)
|
|
}
|
|
inline const ::PROTOBUF_NAMESPACE_ID::RepeatedField< bool >&
|
|
LayoutProto::_internal_dim_unique() const {
|
|
return _impl_.dim_unique_;
|
|
}
|
|
inline const ::PROTOBUF_NAMESPACE_ID::RepeatedField< bool >&
|
|
LayoutProto::dim_unique() const {
|
|
// @@protoc_insertion_point(field_list:xla.LayoutProto.dim_unique)
|
|
return _internal_dim_unique();
|
|
}
|
|
inline ::PROTOBUF_NAMESPACE_ID::RepeatedField< bool >*
|
|
LayoutProto::_internal_mutable_dim_unique() {
|
|
return &_impl_.dim_unique_;
|
|
}
|
|
inline ::PROTOBUF_NAMESPACE_ID::RepeatedField< bool >*
|
|
LayoutProto::mutable_dim_unique() {
|
|
// @@protoc_insertion_point(field_mutable_list:xla.LayoutProto.dim_unique)
|
|
return _internal_mutable_dim_unique();
|
|
}
|
|
|
|
// repeated bool dim_ordered = 14;
|
|
inline int LayoutProto::_internal_dim_ordered_size() const {
|
|
return _impl_.dim_ordered_.size();
|
|
}
|
|
inline int LayoutProto::dim_ordered_size() const {
|
|
return _internal_dim_ordered_size();
|
|
}
|
|
inline void LayoutProto::clear_dim_ordered() {
|
|
_impl_.dim_ordered_.Clear();
|
|
}
|
|
inline bool LayoutProto::_internal_dim_ordered(int index) const {
|
|
return _impl_.dim_ordered_.Get(index);
|
|
}
|
|
inline bool LayoutProto::dim_ordered(int index) const {
|
|
// @@protoc_insertion_point(field_get:xla.LayoutProto.dim_ordered)
|
|
return _internal_dim_ordered(index);
|
|
}
|
|
inline void LayoutProto::set_dim_ordered(int index, bool value) {
|
|
_impl_.dim_ordered_.Set(index, value);
|
|
// @@protoc_insertion_point(field_set:xla.LayoutProto.dim_ordered)
|
|
}
|
|
inline void LayoutProto::_internal_add_dim_ordered(bool value) {
|
|
_impl_.dim_ordered_.Add(value);
|
|
}
|
|
inline void LayoutProto::add_dim_ordered(bool value) {
|
|
_internal_add_dim_ordered(value);
|
|
// @@protoc_insertion_point(field_add:xla.LayoutProto.dim_ordered)
|
|
}
|
|
inline const ::PROTOBUF_NAMESPACE_ID::RepeatedField< bool >&
|
|
LayoutProto::_internal_dim_ordered() const {
|
|
return _impl_.dim_ordered_;
|
|
}
|
|
inline const ::PROTOBUF_NAMESPACE_ID::RepeatedField< bool >&
|
|
LayoutProto::dim_ordered() const {
|
|
// @@protoc_insertion_point(field_list:xla.LayoutProto.dim_ordered)
|
|
return _internal_dim_ordered();
|
|
}
|
|
inline ::PROTOBUF_NAMESPACE_ID::RepeatedField< bool >*
|
|
LayoutProto::_internal_mutable_dim_ordered() {
|
|
return &_impl_.dim_ordered_;
|
|
}
|
|
inline ::PROTOBUF_NAMESPACE_ID::RepeatedField< bool >*
|
|
LayoutProto::mutable_dim_ordered() {
|
|
// @@protoc_insertion_point(field_mutable_list:xla.LayoutProto.dim_ordered)
|
|
return _internal_mutable_dim_ordered();
|
|
}
|
|
|
|
// repeated int64 minor_to_major = 1;
|
|
inline int LayoutProto::_internal_minor_to_major_size() const {
|
|
return _impl_.minor_to_major_.size();
|
|
}
|
|
inline int LayoutProto::minor_to_major_size() const {
|
|
return _internal_minor_to_major_size();
|
|
}
|
|
inline void LayoutProto::clear_minor_to_major() {
|
|
_impl_.minor_to_major_.Clear();
|
|
}
|
|
inline int64_t LayoutProto::_internal_minor_to_major(int index) const {
|
|
return _impl_.minor_to_major_.Get(index);
|
|
}
|
|
inline int64_t LayoutProto::minor_to_major(int index) const {
|
|
// @@protoc_insertion_point(field_get:xla.LayoutProto.minor_to_major)
|
|
return _internal_minor_to_major(index);
|
|
}
|
|
inline void LayoutProto::set_minor_to_major(int index, int64_t value) {
|
|
_impl_.minor_to_major_.Set(index, value);
|
|
// @@protoc_insertion_point(field_set:xla.LayoutProto.minor_to_major)
|
|
}
|
|
inline void LayoutProto::_internal_add_minor_to_major(int64_t value) {
|
|
_impl_.minor_to_major_.Add(value);
|
|
}
|
|
inline void LayoutProto::add_minor_to_major(int64_t value) {
|
|
_internal_add_minor_to_major(value);
|
|
// @@protoc_insertion_point(field_add:xla.LayoutProto.minor_to_major)
|
|
}
|
|
inline const ::PROTOBUF_NAMESPACE_ID::RepeatedField< int64_t >&
|
|
LayoutProto::_internal_minor_to_major() const {
|
|
return _impl_.minor_to_major_;
|
|
}
|
|
inline const ::PROTOBUF_NAMESPACE_ID::RepeatedField< int64_t >&
|
|
LayoutProto::minor_to_major() const {
|
|
// @@protoc_insertion_point(field_list:xla.LayoutProto.minor_to_major)
|
|
return _internal_minor_to_major();
|
|
}
|
|
inline ::PROTOBUF_NAMESPACE_ID::RepeatedField< int64_t >*
|
|
LayoutProto::_internal_mutable_minor_to_major() {
|
|
return &_impl_.minor_to_major_;
|
|
}
|
|
inline ::PROTOBUF_NAMESPACE_ID::RepeatedField< int64_t >*
|
|
LayoutProto::mutable_minor_to_major() {
|
|
// @@protoc_insertion_point(field_mutable_list:xla.LayoutProto.minor_to_major)
|
|
return _internal_mutable_minor_to_major();
|
|
}
|
|
|
|
// repeated .xla.TileProto tiles = 6;
|
|
inline int LayoutProto::_internal_tiles_size() const {
|
|
return _impl_.tiles_.size();
|
|
}
|
|
inline int LayoutProto::tiles_size() const {
|
|
return _internal_tiles_size();
|
|
}
|
|
inline void LayoutProto::clear_tiles() {
|
|
_impl_.tiles_.Clear();
|
|
}
|
|
inline ::xla::TileProto* LayoutProto::mutable_tiles(int index) {
|
|
// @@protoc_insertion_point(field_mutable:xla.LayoutProto.tiles)
|
|
return _impl_.tiles_.Mutable(index);
|
|
}
|
|
inline ::PROTOBUF_NAMESPACE_ID::RepeatedPtrField< ::xla::TileProto >*
|
|
LayoutProto::mutable_tiles() {
|
|
// @@protoc_insertion_point(field_mutable_list:xla.LayoutProto.tiles)
|
|
return &_impl_.tiles_;
|
|
}
|
|
inline const ::xla::TileProto& LayoutProto::_internal_tiles(int index) const {
|
|
return _impl_.tiles_.Get(index);
|
|
}
|
|
inline const ::xla::TileProto& LayoutProto::tiles(int index) const {
|
|
// @@protoc_insertion_point(field_get:xla.LayoutProto.tiles)
|
|
return _internal_tiles(index);
|
|
}
|
|
inline ::xla::TileProto* LayoutProto::_internal_add_tiles() {
|
|
return _impl_.tiles_.Add();
|
|
}
|
|
inline ::xla::TileProto* LayoutProto::add_tiles() {
|
|
::xla::TileProto* _add = _internal_add_tiles();
|
|
// @@protoc_insertion_point(field_add:xla.LayoutProto.tiles)
|
|
return _add;
|
|
}
|
|
inline const ::PROTOBUF_NAMESPACE_ID::RepeatedPtrField< ::xla::TileProto >&
|
|
LayoutProto::tiles() const {
|
|
// @@protoc_insertion_point(field_list:xla.LayoutProto.tiles)
|
|
return _impl_.tiles_;
|
|
}
|
|
|
|
// int64 tail_padding_alignment_in_elements = 16;
|
|
inline void LayoutProto::clear_tail_padding_alignment_in_elements() {
|
|
_impl_.tail_padding_alignment_in_elements_ = int64_t{0};
|
|
}
|
|
inline int64_t LayoutProto::_internal_tail_padding_alignment_in_elements() const {
|
|
return _impl_.tail_padding_alignment_in_elements_;
|
|
}
|
|
inline int64_t LayoutProto::tail_padding_alignment_in_elements() const {
|
|
// @@protoc_insertion_point(field_get:xla.LayoutProto.tail_padding_alignment_in_elements)
|
|
return _internal_tail_padding_alignment_in_elements();
|
|
}
|
|
inline void LayoutProto::_internal_set_tail_padding_alignment_in_elements(int64_t value) {
|
|
|
|
_impl_.tail_padding_alignment_in_elements_ = value;
|
|
}
|
|
inline void LayoutProto::set_tail_padding_alignment_in_elements(int64_t value) {
|
|
_internal_set_tail_padding_alignment_in_elements(value);
|
|
// @@protoc_insertion_point(field_set:xla.LayoutProto.tail_padding_alignment_in_elements)
|
|
}
|
|
|
|
// int64 element_size_in_bits = 7;
|
|
inline void LayoutProto::clear_element_size_in_bits() {
|
|
_impl_.element_size_in_bits_ = int64_t{0};
|
|
}
|
|
inline int64_t LayoutProto::_internal_element_size_in_bits() const {
|
|
return _impl_.element_size_in_bits_;
|
|
}
|
|
inline int64_t LayoutProto::element_size_in_bits() const {
|
|
// @@protoc_insertion_point(field_get:xla.LayoutProto.element_size_in_bits)
|
|
return _internal_element_size_in_bits();
|
|
}
|
|
inline void LayoutProto::_internal_set_element_size_in_bits(int64_t value) {
|
|
|
|
_impl_.element_size_in_bits_ = value;
|
|
}
|
|
inline void LayoutProto::set_element_size_in_bits(int64_t value) {
|
|
_internal_set_element_size_in_bits(value);
|
|
// @@protoc_insertion_point(field_set:xla.LayoutProto.element_size_in_bits)
|
|
}
|
|
|
|
// int64 memory_space = 8;
|
|
inline void LayoutProto::clear_memory_space() {
|
|
_impl_.memory_space_ = int64_t{0};
|
|
}
|
|
inline int64_t LayoutProto::_internal_memory_space() const {
|
|
return _impl_.memory_space_;
|
|
}
|
|
inline int64_t LayoutProto::memory_space() const {
|
|
// @@protoc_insertion_point(field_get:xla.LayoutProto.memory_space)
|
|
return _internal_memory_space();
|
|
}
|
|
inline void LayoutProto::_internal_set_memory_space(int64_t value) {
|
|
|
|
_impl_.memory_space_ = value;
|
|
}
|
|
inline void LayoutProto::set_memory_space(int64_t value) {
|
|
_internal_set_memory_space(value);
|
|
// @@protoc_insertion_point(field_set:xla.LayoutProto.memory_space)
|
|
}
|
|
|
|
// .xla.PrimitiveType index_primitive_type = 11;
|
|
inline void LayoutProto::clear_index_primitive_type() {
|
|
_impl_.index_primitive_type_ = 0;
|
|
}
|
|
inline ::xla::PrimitiveType LayoutProto::_internal_index_primitive_type() const {
|
|
return static_cast< ::xla::PrimitiveType >(_impl_.index_primitive_type_);
|
|
}
|
|
inline ::xla::PrimitiveType LayoutProto::index_primitive_type() const {
|
|
// @@protoc_insertion_point(field_get:xla.LayoutProto.index_primitive_type)
|
|
return _internal_index_primitive_type();
|
|
}
|
|
inline void LayoutProto::_internal_set_index_primitive_type(::xla::PrimitiveType value) {
|
|
|
|
_impl_.index_primitive_type_ = value;
|
|
}
|
|
inline void LayoutProto::set_index_primitive_type(::xla::PrimitiveType value) {
|
|
_internal_set_index_primitive_type(value);
|
|
// @@protoc_insertion_point(field_set:xla.LayoutProto.index_primitive_type)
|
|
}
|
|
|
|
// .xla.PrimitiveType pointer_primitive_type = 12;
|
|
inline void LayoutProto::clear_pointer_primitive_type() {
|
|
_impl_.pointer_primitive_type_ = 0;
|
|
}
|
|
inline ::xla::PrimitiveType LayoutProto::_internal_pointer_primitive_type() const {
|
|
return static_cast< ::xla::PrimitiveType >(_impl_.pointer_primitive_type_);
|
|
}
|
|
inline ::xla::PrimitiveType LayoutProto::pointer_primitive_type() const {
|
|
// @@protoc_insertion_point(field_get:xla.LayoutProto.pointer_primitive_type)
|
|
return _internal_pointer_primitive_type();
|
|
}
|
|
inline void LayoutProto::_internal_set_pointer_primitive_type(::xla::PrimitiveType value) {
|
|
|
|
_impl_.pointer_primitive_type_ = value;
|
|
}
|
|
inline void LayoutProto::set_pointer_primitive_type(::xla::PrimitiveType value) {
|
|
_internal_set_pointer_primitive_type(value);
|
|
// @@protoc_insertion_point(field_set:xla.LayoutProto.pointer_primitive_type)
|
|
}
|
|
|
|
// .xla.ShapeProto physical_shape = 10;
|
|
inline bool LayoutProto::_internal_has_physical_shape() const {
|
|
return this != internal_default_instance() && _impl_.physical_shape_ != nullptr;
|
|
}
|
|
inline bool LayoutProto::has_physical_shape() const {
|
|
return _internal_has_physical_shape();
|
|
}
|
|
inline void LayoutProto::clear_physical_shape() {
|
|
if (GetArenaForAllocation() == nullptr && _impl_.physical_shape_ != nullptr) {
|
|
delete _impl_.physical_shape_;
|
|
}
|
|
_impl_.physical_shape_ = nullptr;
|
|
}
|
|
inline const ::xla::ShapeProto& LayoutProto::_internal_physical_shape() const {
|
|
const ::xla::ShapeProto* p = _impl_.physical_shape_;
|
|
return p != nullptr ? *p : reinterpret_cast<const ::xla::ShapeProto&>(
|
|
::xla::_ShapeProto_default_instance_);
|
|
}
|
|
inline const ::xla::ShapeProto& LayoutProto::physical_shape() const {
|
|
// @@protoc_insertion_point(field_get:xla.LayoutProto.physical_shape)
|
|
return _internal_physical_shape();
|
|
}
|
|
inline void LayoutProto::unsafe_arena_set_allocated_physical_shape(
|
|
::xla::ShapeProto* physical_shape) {
|
|
if (GetArenaForAllocation() == nullptr) {
|
|
delete reinterpret_cast<::PROTOBUF_NAMESPACE_ID::MessageLite*>(_impl_.physical_shape_);
|
|
}
|
|
_impl_.physical_shape_ = physical_shape;
|
|
if (physical_shape) {
|
|
|
|
} else {
|
|
|
|
}
|
|
// @@protoc_insertion_point(field_unsafe_arena_set_allocated:xla.LayoutProto.physical_shape)
|
|
}
|
|
inline ::xla::ShapeProto* LayoutProto::release_physical_shape() {
|
|
|
|
::xla::ShapeProto* temp = _impl_.physical_shape_;
|
|
_impl_.physical_shape_ = nullptr;
|
|
#ifdef PROTOBUF_FORCE_COPY_IN_RELEASE
|
|
auto* old = reinterpret_cast<::PROTOBUF_NAMESPACE_ID::MessageLite*>(temp);
|
|
temp = ::PROTOBUF_NAMESPACE_ID::internal::DuplicateIfNonNull(temp);
|
|
if (GetArenaForAllocation() == nullptr) { delete old; }
|
|
#else // PROTOBUF_FORCE_COPY_IN_RELEASE
|
|
if (GetArenaForAllocation() != nullptr) {
|
|
temp = ::PROTOBUF_NAMESPACE_ID::internal::DuplicateIfNonNull(temp);
|
|
}
|
|
#endif // !PROTOBUF_FORCE_COPY_IN_RELEASE
|
|
return temp;
|
|
}
|
|
inline ::xla::ShapeProto* LayoutProto::unsafe_arena_release_physical_shape() {
|
|
// @@protoc_insertion_point(field_release:xla.LayoutProto.physical_shape)
|
|
|
|
::xla::ShapeProto* temp = _impl_.physical_shape_;
|
|
_impl_.physical_shape_ = nullptr;
|
|
return temp;
|
|
}
|
|
inline ::xla::ShapeProto* LayoutProto::_internal_mutable_physical_shape() {
|
|
|
|
if (_impl_.physical_shape_ == nullptr) {
|
|
auto* p = CreateMaybeMessage<::xla::ShapeProto>(GetArenaForAllocation());
|
|
_impl_.physical_shape_ = p;
|
|
}
|
|
return _impl_.physical_shape_;
|
|
}
|
|
inline ::xla::ShapeProto* LayoutProto::mutable_physical_shape() {
|
|
::xla::ShapeProto* _msg = _internal_mutable_physical_shape();
|
|
// @@protoc_insertion_point(field_mutable:xla.LayoutProto.physical_shape)
|
|
return _msg;
|
|
}
|
|
inline void LayoutProto::set_allocated_physical_shape(::xla::ShapeProto* physical_shape) {
|
|
::PROTOBUF_NAMESPACE_ID::Arena* message_arena = GetArenaForAllocation();
|
|
if (message_arena == nullptr) {
|
|
delete _impl_.physical_shape_;
|
|
}
|
|
if (physical_shape) {
|
|
::PROTOBUF_NAMESPACE_ID::Arena* submessage_arena =
|
|
::PROTOBUF_NAMESPACE_ID::Arena::InternalGetOwningArena(physical_shape);
|
|
if (message_arena != submessage_arena) {
|
|
physical_shape = ::PROTOBUF_NAMESPACE_ID::internal::GetOwnedMessage(
|
|
message_arena, physical_shape, submessage_arena);
|
|
}
|
|
|
|
} else {
|
|
|
|
}
|
|
_impl_.physical_shape_ = physical_shape;
|
|
// @@protoc_insertion_point(field_set_allocated:xla.LayoutProto.physical_shape)
|
|
}
|
|
|
|
// int64 dynamic_shape_metadata_prefix_bytes = 15;
|
|
inline void LayoutProto::clear_dynamic_shape_metadata_prefix_bytes() {
|
|
_impl_.dynamic_shape_metadata_prefix_bytes_ = int64_t{0};
|
|
}
|
|
inline int64_t LayoutProto::_internal_dynamic_shape_metadata_prefix_bytes() const {
|
|
return _impl_.dynamic_shape_metadata_prefix_bytes_;
|
|
}
|
|
inline int64_t LayoutProto::dynamic_shape_metadata_prefix_bytes() const {
|
|
// @@protoc_insertion_point(field_get:xla.LayoutProto.dynamic_shape_metadata_prefix_bytes)
|
|
return _internal_dynamic_shape_metadata_prefix_bytes();
|
|
}
|
|
inline void LayoutProto::_internal_set_dynamic_shape_metadata_prefix_bytes(int64_t value) {
|
|
|
|
_impl_.dynamic_shape_metadata_prefix_bytes_ = value;
|
|
}
|
|
inline void LayoutProto::set_dynamic_shape_metadata_prefix_bytes(int64_t value) {
|
|
_internal_set_dynamic_shape_metadata_prefix_bytes(value);
|
|
// @@protoc_insertion_point(field_set:xla.LayoutProto.dynamic_shape_metadata_prefix_bytes)
|
|
}
|
|
|
|
// -------------------------------------------------------------------
|
|
|
|
// ShapeProto
|
|
|
|
// .xla.PrimitiveType element_type = 2;
|
|
inline void ShapeProto::clear_element_type() {
|
|
_impl_.element_type_ = 0;
|
|
}
|
|
inline ::xla::PrimitiveType ShapeProto::_internal_element_type() const {
|
|
return static_cast< ::xla::PrimitiveType >(_impl_.element_type_);
|
|
}
|
|
inline ::xla::PrimitiveType ShapeProto::element_type() const {
|
|
// @@protoc_insertion_point(field_get:xla.ShapeProto.element_type)
|
|
return _internal_element_type();
|
|
}
|
|
inline void ShapeProto::_internal_set_element_type(::xla::PrimitiveType value) {
|
|
|
|
_impl_.element_type_ = value;
|
|
}
|
|
inline void ShapeProto::set_element_type(::xla::PrimitiveType value) {
|
|
_internal_set_element_type(value);
|
|
// @@protoc_insertion_point(field_set:xla.ShapeProto.element_type)
|
|
}
|
|
|
|
// repeated int64 dimensions = 3;
|
|
inline int ShapeProto::_internal_dimensions_size() const {
|
|
return _impl_.dimensions_.size();
|
|
}
|
|
inline int ShapeProto::dimensions_size() const {
|
|
return _internal_dimensions_size();
|
|
}
|
|
inline void ShapeProto::clear_dimensions() {
|
|
_impl_.dimensions_.Clear();
|
|
}
|
|
inline int64_t ShapeProto::_internal_dimensions(int index) const {
|
|
return _impl_.dimensions_.Get(index);
|
|
}
|
|
inline int64_t ShapeProto::dimensions(int index) const {
|
|
// @@protoc_insertion_point(field_get:xla.ShapeProto.dimensions)
|
|
return _internal_dimensions(index);
|
|
}
|
|
inline void ShapeProto::set_dimensions(int index, int64_t value) {
|
|
_impl_.dimensions_.Set(index, value);
|
|
// @@protoc_insertion_point(field_set:xla.ShapeProto.dimensions)
|
|
}
|
|
inline void ShapeProto::_internal_add_dimensions(int64_t value) {
|
|
_impl_.dimensions_.Add(value);
|
|
}
|
|
inline void ShapeProto::add_dimensions(int64_t value) {
|
|
_internal_add_dimensions(value);
|
|
// @@protoc_insertion_point(field_add:xla.ShapeProto.dimensions)
|
|
}
|
|
inline const ::PROTOBUF_NAMESPACE_ID::RepeatedField< int64_t >&
|
|
ShapeProto::_internal_dimensions() const {
|
|
return _impl_.dimensions_;
|
|
}
|
|
inline const ::PROTOBUF_NAMESPACE_ID::RepeatedField< int64_t >&
|
|
ShapeProto::dimensions() const {
|
|
// @@protoc_insertion_point(field_list:xla.ShapeProto.dimensions)
|
|
return _internal_dimensions();
|
|
}
|
|
inline ::PROTOBUF_NAMESPACE_ID::RepeatedField< int64_t >*
|
|
ShapeProto::_internal_mutable_dimensions() {
|
|
return &_impl_.dimensions_;
|
|
}
|
|
inline ::PROTOBUF_NAMESPACE_ID::RepeatedField< int64_t >*
|
|
ShapeProto::mutable_dimensions() {
|
|
// @@protoc_insertion_point(field_mutable_list:xla.ShapeProto.dimensions)
|
|
return _internal_mutable_dimensions();
|
|
}
|
|
|
|
// repeated .xla.ShapeProto tuple_shapes = 4;
|
|
inline int ShapeProto::_internal_tuple_shapes_size() const {
|
|
return _impl_.tuple_shapes_.size();
|
|
}
|
|
inline int ShapeProto::tuple_shapes_size() const {
|
|
return _internal_tuple_shapes_size();
|
|
}
|
|
inline void ShapeProto::clear_tuple_shapes() {
|
|
_impl_.tuple_shapes_.Clear();
|
|
}
|
|
inline ::xla::ShapeProto* ShapeProto::mutable_tuple_shapes(int index) {
|
|
// @@protoc_insertion_point(field_mutable:xla.ShapeProto.tuple_shapes)
|
|
return _impl_.tuple_shapes_.Mutable(index);
|
|
}
|
|
inline ::PROTOBUF_NAMESPACE_ID::RepeatedPtrField< ::xla::ShapeProto >*
|
|
ShapeProto::mutable_tuple_shapes() {
|
|
// @@protoc_insertion_point(field_mutable_list:xla.ShapeProto.tuple_shapes)
|
|
return &_impl_.tuple_shapes_;
|
|
}
|
|
inline const ::xla::ShapeProto& ShapeProto::_internal_tuple_shapes(int index) const {
|
|
return _impl_.tuple_shapes_.Get(index);
|
|
}
|
|
inline const ::xla::ShapeProto& ShapeProto::tuple_shapes(int index) const {
|
|
// @@protoc_insertion_point(field_get:xla.ShapeProto.tuple_shapes)
|
|
return _internal_tuple_shapes(index);
|
|
}
|
|
inline ::xla::ShapeProto* ShapeProto::_internal_add_tuple_shapes() {
|
|
return _impl_.tuple_shapes_.Add();
|
|
}
|
|
inline ::xla::ShapeProto* ShapeProto::add_tuple_shapes() {
|
|
::xla::ShapeProto* _add = _internal_add_tuple_shapes();
|
|
// @@protoc_insertion_point(field_add:xla.ShapeProto.tuple_shapes)
|
|
return _add;
|
|
}
|
|
inline const ::PROTOBUF_NAMESPACE_ID::RepeatedPtrField< ::xla::ShapeProto >&
|
|
ShapeProto::tuple_shapes() const {
|
|
// @@protoc_insertion_point(field_list:xla.ShapeProto.tuple_shapes)
|
|
return _impl_.tuple_shapes_;
|
|
}
|
|
|
|
// .xla.LayoutProto layout = 5;
|
|
inline bool ShapeProto::_internal_has_layout() const {
|
|
return this != internal_default_instance() && _impl_.layout_ != nullptr;
|
|
}
|
|
inline bool ShapeProto::has_layout() const {
|
|
return _internal_has_layout();
|
|
}
|
|
inline void ShapeProto::clear_layout() {
|
|
if (GetArenaForAllocation() == nullptr && _impl_.layout_ != nullptr) {
|
|
delete _impl_.layout_;
|
|
}
|
|
_impl_.layout_ = nullptr;
|
|
}
|
|
inline const ::xla::LayoutProto& ShapeProto::_internal_layout() const {
|
|
const ::xla::LayoutProto* p = _impl_.layout_;
|
|
return p != nullptr ? *p : reinterpret_cast<const ::xla::LayoutProto&>(
|
|
::xla::_LayoutProto_default_instance_);
|
|
}
|
|
inline const ::xla::LayoutProto& ShapeProto::layout() const {
|
|
// @@protoc_insertion_point(field_get:xla.ShapeProto.layout)
|
|
return _internal_layout();
|
|
}
|
|
inline void ShapeProto::unsafe_arena_set_allocated_layout(
|
|
::xla::LayoutProto* layout) {
|
|
if (GetArenaForAllocation() == nullptr) {
|
|
delete reinterpret_cast<::PROTOBUF_NAMESPACE_ID::MessageLite*>(_impl_.layout_);
|
|
}
|
|
_impl_.layout_ = layout;
|
|
if (layout) {
|
|
|
|
} else {
|
|
|
|
}
|
|
// @@protoc_insertion_point(field_unsafe_arena_set_allocated:xla.ShapeProto.layout)
|
|
}
|
|
inline ::xla::LayoutProto* ShapeProto::release_layout() {
|
|
|
|
::xla::LayoutProto* temp = _impl_.layout_;
|
|
_impl_.layout_ = nullptr;
|
|
#ifdef PROTOBUF_FORCE_COPY_IN_RELEASE
|
|
auto* old = reinterpret_cast<::PROTOBUF_NAMESPACE_ID::MessageLite*>(temp);
|
|
temp = ::PROTOBUF_NAMESPACE_ID::internal::DuplicateIfNonNull(temp);
|
|
if (GetArenaForAllocation() == nullptr) { delete old; }
|
|
#else // PROTOBUF_FORCE_COPY_IN_RELEASE
|
|
if (GetArenaForAllocation() != nullptr) {
|
|
temp = ::PROTOBUF_NAMESPACE_ID::internal::DuplicateIfNonNull(temp);
|
|
}
|
|
#endif // !PROTOBUF_FORCE_COPY_IN_RELEASE
|
|
return temp;
|
|
}
|
|
inline ::xla::LayoutProto* ShapeProto::unsafe_arena_release_layout() {
|
|
// @@protoc_insertion_point(field_release:xla.ShapeProto.layout)
|
|
|
|
::xla::LayoutProto* temp = _impl_.layout_;
|
|
_impl_.layout_ = nullptr;
|
|
return temp;
|
|
}
|
|
inline ::xla::LayoutProto* ShapeProto::_internal_mutable_layout() {
|
|
|
|
if (_impl_.layout_ == nullptr) {
|
|
auto* p = CreateMaybeMessage<::xla::LayoutProto>(GetArenaForAllocation());
|
|
_impl_.layout_ = p;
|
|
}
|
|
return _impl_.layout_;
|
|
}
|
|
inline ::xla::LayoutProto* ShapeProto::mutable_layout() {
|
|
::xla::LayoutProto* _msg = _internal_mutable_layout();
|
|
// @@protoc_insertion_point(field_mutable:xla.ShapeProto.layout)
|
|
return _msg;
|
|
}
|
|
inline void ShapeProto::set_allocated_layout(::xla::LayoutProto* layout) {
|
|
::PROTOBUF_NAMESPACE_ID::Arena* message_arena = GetArenaForAllocation();
|
|
if (message_arena == nullptr) {
|
|
delete _impl_.layout_;
|
|
}
|
|
if (layout) {
|
|
::PROTOBUF_NAMESPACE_ID::Arena* submessage_arena =
|
|
::PROTOBUF_NAMESPACE_ID::Arena::InternalGetOwningArena(layout);
|
|
if (message_arena != submessage_arena) {
|
|
layout = ::PROTOBUF_NAMESPACE_ID::internal::GetOwnedMessage(
|
|
message_arena, layout, submessage_arena);
|
|
}
|
|
|
|
} else {
|
|
|
|
}
|
|
_impl_.layout_ = layout;
|
|
// @@protoc_insertion_point(field_set_allocated:xla.ShapeProto.layout)
|
|
}
|
|
|
|
// repeated bool is_dynamic_dimension = 6;
|
|
inline int ShapeProto::_internal_is_dynamic_dimension_size() const {
|
|
return _impl_.is_dynamic_dimension_.size();
|
|
}
|
|
inline int ShapeProto::is_dynamic_dimension_size() const {
|
|
return _internal_is_dynamic_dimension_size();
|
|
}
|
|
inline void ShapeProto::clear_is_dynamic_dimension() {
|
|
_impl_.is_dynamic_dimension_.Clear();
|
|
}
|
|
inline bool ShapeProto::_internal_is_dynamic_dimension(int index) const {
|
|
return _impl_.is_dynamic_dimension_.Get(index);
|
|
}
|
|
inline bool ShapeProto::is_dynamic_dimension(int index) const {
|
|
// @@protoc_insertion_point(field_get:xla.ShapeProto.is_dynamic_dimension)
|
|
return _internal_is_dynamic_dimension(index);
|
|
}
|
|
inline void ShapeProto::set_is_dynamic_dimension(int index, bool value) {
|
|
_impl_.is_dynamic_dimension_.Set(index, value);
|
|
// @@protoc_insertion_point(field_set:xla.ShapeProto.is_dynamic_dimension)
|
|
}
|
|
inline void ShapeProto::_internal_add_is_dynamic_dimension(bool value) {
|
|
_impl_.is_dynamic_dimension_.Add(value);
|
|
}
|
|
inline void ShapeProto::add_is_dynamic_dimension(bool value) {
|
|
_internal_add_is_dynamic_dimension(value);
|
|
// @@protoc_insertion_point(field_add:xla.ShapeProto.is_dynamic_dimension)
|
|
}
|
|
inline const ::PROTOBUF_NAMESPACE_ID::RepeatedField< bool >&
|
|
ShapeProto::_internal_is_dynamic_dimension() const {
|
|
return _impl_.is_dynamic_dimension_;
|
|
}
|
|
inline const ::PROTOBUF_NAMESPACE_ID::RepeatedField< bool >&
|
|
ShapeProto::is_dynamic_dimension() const {
|
|
// @@protoc_insertion_point(field_list:xla.ShapeProto.is_dynamic_dimension)
|
|
return _internal_is_dynamic_dimension();
|
|
}
|
|
inline ::PROTOBUF_NAMESPACE_ID::RepeatedField< bool >*
|
|
ShapeProto::_internal_mutable_is_dynamic_dimension() {
|
|
return &_impl_.is_dynamic_dimension_;
|
|
}
|
|
inline ::PROTOBUF_NAMESPACE_ID::RepeatedField< bool >*
|
|
ShapeProto::mutable_is_dynamic_dimension() {
|
|
// @@protoc_insertion_point(field_mutable_list:xla.ShapeProto.is_dynamic_dimension)
|
|
return _internal_mutable_is_dynamic_dimension();
|
|
}
|
|
|
|
// -------------------------------------------------------------------
|
|
|
|
// ProgramShapeProto
|
|
|
|
// repeated .xla.ShapeProto parameters = 1;
|
|
inline int ProgramShapeProto::_internal_parameters_size() const {
|
|
return _impl_.parameters_.size();
|
|
}
|
|
inline int ProgramShapeProto::parameters_size() const {
|
|
return _internal_parameters_size();
|
|
}
|
|
inline void ProgramShapeProto::clear_parameters() {
|
|
_impl_.parameters_.Clear();
|
|
}
|
|
inline ::xla::ShapeProto* ProgramShapeProto::mutable_parameters(int index) {
|
|
// @@protoc_insertion_point(field_mutable:xla.ProgramShapeProto.parameters)
|
|
return _impl_.parameters_.Mutable(index);
|
|
}
|
|
inline ::PROTOBUF_NAMESPACE_ID::RepeatedPtrField< ::xla::ShapeProto >*
|
|
ProgramShapeProto::mutable_parameters() {
|
|
// @@protoc_insertion_point(field_mutable_list:xla.ProgramShapeProto.parameters)
|
|
return &_impl_.parameters_;
|
|
}
|
|
inline const ::xla::ShapeProto& ProgramShapeProto::_internal_parameters(int index) const {
|
|
return _impl_.parameters_.Get(index);
|
|
}
|
|
inline const ::xla::ShapeProto& ProgramShapeProto::parameters(int index) const {
|
|
// @@protoc_insertion_point(field_get:xla.ProgramShapeProto.parameters)
|
|
return _internal_parameters(index);
|
|
}
|
|
inline ::xla::ShapeProto* ProgramShapeProto::_internal_add_parameters() {
|
|
return _impl_.parameters_.Add();
|
|
}
|
|
inline ::xla::ShapeProto* ProgramShapeProto::add_parameters() {
|
|
::xla::ShapeProto* _add = _internal_add_parameters();
|
|
// @@protoc_insertion_point(field_add:xla.ProgramShapeProto.parameters)
|
|
return _add;
|
|
}
|
|
inline const ::PROTOBUF_NAMESPACE_ID::RepeatedPtrField< ::xla::ShapeProto >&
|
|
ProgramShapeProto::parameters() const {
|
|
// @@protoc_insertion_point(field_list:xla.ProgramShapeProto.parameters)
|
|
return _impl_.parameters_;
|
|
}
|
|
|
|
// .xla.ShapeProto result = 2;
|
|
inline bool ProgramShapeProto::_internal_has_result() const {
|
|
return this != internal_default_instance() && _impl_.result_ != nullptr;
|
|
}
|
|
inline bool ProgramShapeProto::has_result() const {
|
|
return _internal_has_result();
|
|
}
|
|
inline void ProgramShapeProto::clear_result() {
|
|
if (GetArenaForAllocation() == nullptr && _impl_.result_ != nullptr) {
|
|
delete _impl_.result_;
|
|
}
|
|
_impl_.result_ = nullptr;
|
|
}
|
|
inline const ::xla::ShapeProto& ProgramShapeProto::_internal_result() const {
|
|
const ::xla::ShapeProto* p = _impl_.result_;
|
|
return p != nullptr ? *p : reinterpret_cast<const ::xla::ShapeProto&>(
|
|
::xla::_ShapeProto_default_instance_);
|
|
}
|
|
inline const ::xla::ShapeProto& ProgramShapeProto::result() const {
|
|
// @@protoc_insertion_point(field_get:xla.ProgramShapeProto.result)
|
|
return _internal_result();
|
|
}
|
|
inline void ProgramShapeProto::unsafe_arena_set_allocated_result(
|
|
::xla::ShapeProto* result) {
|
|
if (GetArenaForAllocation() == nullptr) {
|
|
delete reinterpret_cast<::PROTOBUF_NAMESPACE_ID::MessageLite*>(_impl_.result_);
|
|
}
|
|
_impl_.result_ = result;
|
|
if (result) {
|
|
|
|
} else {
|
|
|
|
}
|
|
// @@protoc_insertion_point(field_unsafe_arena_set_allocated:xla.ProgramShapeProto.result)
|
|
}
|
|
inline ::xla::ShapeProto* ProgramShapeProto::release_result() {
|
|
|
|
::xla::ShapeProto* temp = _impl_.result_;
|
|
_impl_.result_ = nullptr;
|
|
#ifdef PROTOBUF_FORCE_COPY_IN_RELEASE
|
|
auto* old = reinterpret_cast<::PROTOBUF_NAMESPACE_ID::MessageLite*>(temp);
|
|
temp = ::PROTOBUF_NAMESPACE_ID::internal::DuplicateIfNonNull(temp);
|
|
if (GetArenaForAllocation() == nullptr) { delete old; }
|
|
#else // PROTOBUF_FORCE_COPY_IN_RELEASE
|
|
if (GetArenaForAllocation() != nullptr) {
|
|
temp = ::PROTOBUF_NAMESPACE_ID::internal::DuplicateIfNonNull(temp);
|
|
}
|
|
#endif // !PROTOBUF_FORCE_COPY_IN_RELEASE
|
|
return temp;
|
|
}
|
|
inline ::xla::ShapeProto* ProgramShapeProto::unsafe_arena_release_result() {
|
|
// @@protoc_insertion_point(field_release:xla.ProgramShapeProto.result)
|
|
|
|
::xla::ShapeProto* temp = _impl_.result_;
|
|
_impl_.result_ = nullptr;
|
|
return temp;
|
|
}
|
|
inline ::xla::ShapeProto* ProgramShapeProto::_internal_mutable_result() {
|
|
|
|
if (_impl_.result_ == nullptr) {
|
|
auto* p = CreateMaybeMessage<::xla::ShapeProto>(GetArenaForAllocation());
|
|
_impl_.result_ = p;
|
|
}
|
|
return _impl_.result_;
|
|
}
|
|
inline ::xla::ShapeProto* ProgramShapeProto::mutable_result() {
|
|
::xla::ShapeProto* _msg = _internal_mutable_result();
|
|
// @@protoc_insertion_point(field_mutable:xla.ProgramShapeProto.result)
|
|
return _msg;
|
|
}
|
|
inline void ProgramShapeProto::set_allocated_result(::xla::ShapeProto* result) {
|
|
::PROTOBUF_NAMESPACE_ID::Arena* message_arena = GetArenaForAllocation();
|
|
if (message_arena == nullptr) {
|
|
delete _impl_.result_;
|
|
}
|
|
if (result) {
|
|
::PROTOBUF_NAMESPACE_ID::Arena* submessage_arena =
|
|
::PROTOBUF_NAMESPACE_ID::Arena::InternalGetOwningArena(result);
|
|
if (message_arena != submessage_arena) {
|
|
result = ::PROTOBUF_NAMESPACE_ID::internal::GetOwnedMessage(
|
|
message_arena, result, submessage_arena);
|
|
}
|
|
|
|
} else {
|
|
|
|
}
|
|
_impl_.result_ = result;
|
|
// @@protoc_insertion_point(field_set_allocated:xla.ProgramShapeProto.result)
|
|
}
|
|
|
|
// repeated string parameter_names = 3;
|
|
inline int ProgramShapeProto::_internal_parameter_names_size() const {
|
|
return _impl_.parameter_names_.size();
|
|
}
|
|
inline int ProgramShapeProto::parameter_names_size() const {
|
|
return _internal_parameter_names_size();
|
|
}
|
|
inline void ProgramShapeProto::clear_parameter_names() {
|
|
_impl_.parameter_names_.Clear();
|
|
}
|
|
inline std::string* ProgramShapeProto::add_parameter_names() {
|
|
std::string* _s = _internal_add_parameter_names();
|
|
// @@protoc_insertion_point(field_add_mutable:xla.ProgramShapeProto.parameter_names)
|
|
return _s;
|
|
}
|
|
inline const std::string& ProgramShapeProto::_internal_parameter_names(int index) const {
|
|
return _impl_.parameter_names_.Get(index);
|
|
}
|
|
inline const std::string& ProgramShapeProto::parameter_names(int index) const {
|
|
// @@protoc_insertion_point(field_get:xla.ProgramShapeProto.parameter_names)
|
|
return _internal_parameter_names(index);
|
|
}
|
|
inline std::string* ProgramShapeProto::mutable_parameter_names(int index) {
|
|
// @@protoc_insertion_point(field_mutable:xla.ProgramShapeProto.parameter_names)
|
|
return _impl_.parameter_names_.Mutable(index);
|
|
}
|
|
inline void ProgramShapeProto::set_parameter_names(int index, const std::string& value) {
|
|
_impl_.parameter_names_.Mutable(index)->assign(value);
|
|
// @@protoc_insertion_point(field_set:xla.ProgramShapeProto.parameter_names)
|
|
}
|
|
inline void ProgramShapeProto::set_parameter_names(int index, std::string&& value) {
|
|
_impl_.parameter_names_.Mutable(index)->assign(std::move(value));
|
|
// @@protoc_insertion_point(field_set:xla.ProgramShapeProto.parameter_names)
|
|
}
|
|
inline void ProgramShapeProto::set_parameter_names(int index, const char* value) {
|
|
GOOGLE_DCHECK(value != nullptr);
|
|
_impl_.parameter_names_.Mutable(index)->assign(value);
|
|
// @@protoc_insertion_point(field_set_char:xla.ProgramShapeProto.parameter_names)
|
|
}
|
|
inline void ProgramShapeProto::set_parameter_names(int index, const char* value, size_t size) {
|
|
_impl_.parameter_names_.Mutable(index)->assign(
|
|
reinterpret_cast<const char*>(value), size);
|
|
// @@protoc_insertion_point(field_set_pointer:xla.ProgramShapeProto.parameter_names)
|
|
}
|
|
inline std::string* ProgramShapeProto::_internal_add_parameter_names() {
|
|
return _impl_.parameter_names_.Add();
|
|
}
|
|
inline void ProgramShapeProto::add_parameter_names(const std::string& value) {
|
|
_impl_.parameter_names_.Add()->assign(value);
|
|
// @@protoc_insertion_point(field_add:xla.ProgramShapeProto.parameter_names)
|
|
}
|
|
inline void ProgramShapeProto::add_parameter_names(std::string&& value) {
|
|
_impl_.parameter_names_.Add(std::move(value));
|
|
// @@protoc_insertion_point(field_add:xla.ProgramShapeProto.parameter_names)
|
|
}
|
|
inline void ProgramShapeProto::add_parameter_names(const char* value) {
|
|
GOOGLE_DCHECK(value != nullptr);
|
|
_impl_.parameter_names_.Add()->assign(value);
|
|
// @@protoc_insertion_point(field_add_char:xla.ProgramShapeProto.parameter_names)
|
|
}
|
|
inline void ProgramShapeProto::add_parameter_names(const char* value, size_t size) {
|
|
_impl_.parameter_names_.Add()->assign(reinterpret_cast<const char*>(value), size);
|
|
// @@protoc_insertion_point(field_add_pointer:xla.ProgramShapeProto.parameter_names)
|
|
}
|
|
inline const ::PROTOBUF_NAMESPACE_ID::RepeatedPtrField<std::string>&
|
|
ProgramShapeProto::parameter_names() const {
|
|
// @@protoc_insertion_point(field_list:xla.ProgramShapeProto.parameter_names)
|
|
return _impl_.parameter_names_;
|
|
}
|
|
inline ::PROTOBUF_NAMESPACE_ID::RepeatedPtrField<std::string>*
|
|
ProgramShapeProto::mutable_parameter_names() {
|
|
// @@protoc_insertion_point(field_mutable_list:xla.ProgramShapeProto.parameter_names)
|
|
return &_impl_.parameter_names_;
|
|
}
|
|
|
|
// -------------------------------------------------------------------
|
|
|
|
// ComputationStats
|
|
|
|
// double flop_count = 1;
|
|
inline void ComputationStats::clear_flop_count() {
|
|
_impl_.flop_count_ = 0;
|
|
}
|
|
inline double ComputationStats::_internal_flop_count() const {
|
|
return _impl_.flop_count_;
|
|
}
|
|
inline double ComputationStats::flop_count() const {
|
|
// @@protoc_insertion_point(field_get:xla.ComputationStats.flop_count)
|
|
return _internal_flop_count();
|
|
}
|
|
inline void ComputationStats::_internal_set_flop_count(double value) {
|
|
|
|
_impl_.flop_count_ = value;
|
|
}
|
|
inline void ComputationStats::set_flop_count(double value) {
|
|
_internal_set_flop_count(value);
|
|
// @@protoc_insertion_point(field_set:xla.ComputationStats.flop_count)
|
|
}
|
|
|
|
// double transcendental_count = 2;
|
|
inline void ComputationStats::clear_transcendental_count() {
|
|
_impl_.transcendental_count_ = 0;
|
|
}
|
|
inline double ComputationStats::_internal_transcendental_count() const {
|
|
return _impl_.transcendental_count_;
|
|
}
|
|
inline double ComputationStats::transcendental_count() const {
|
|
// @@protoc_insertion_point(field_get:xla.ComputationStats.transcendental_count)
|
|
return _internal_transcendental_count();
|
|
}
|
|
inline void ComputationStats::_internal_set_transcendental_count(double value) {
|
|
|
|
_impl_.transcendental_count_ = value;
|
|
}
|
|
inline void ComputationStats::set_transcendental_count(double value) {
|
|
_internal_set_transcendental_count(value);
|
|
// @@protoc_insertion_point(field_set:xla.ComputationStats.transcendental_count)
|
|
}
|
|
|
|
// -------------------------------------------------------------------
|
|
|
|
// OpMetadata_ProfileInfo
|
|
|
|
// repeated .xla.ProfileType profile_type = 1;
|
|
inline int OpMetadata_ProfileInfo::_internal_profile_type_size() const {
|
|
return _impl_.profile_type_.size();
|
|
}
|
|
inline int OpMetadata_ProfileInfo::profile_type_size() const {
|
|
return _internal_profile_type_size();
|
|
}
|
|
inline void OpMetadata_ProfileInfo::clear_profile_type() {
|
|
_impl_.profile_type_.Clear();
|
|
}
|
|
inline ::xla::ProfileType OpMetadata_ProfileInfo::_internal_profile_type(int index) const {
|
|
return static_cast< ::xla::ProfileType >(_impl_.profile_type_.Get(index));
|
|
}
|
|
inline ::xla::ProfileType OpMetadata_ProfileInfo::profile_type(int index) const {
|
|
// @@protoc_insertion_point(field_get:xla.OpMetadata.ProfileInfo.profile_type)
|
|
return _internal_profile_type(index);
|
|
}
|
|
inline void OpMetadata_ProfileInfo::set_profile_type(int index, ::xla::ProfileType value) {
|
|
_impl_.profile_type_.Set(index, value);
|
|
// @@protoc_insertion_point(field_set:xla.OpMetadata.ProfileInfo.profile_type)
|
|
}
|
|
inline void OpMetadata_ProfileInfo::_internal_add_profile_type(::xla::ProfileType value) {
|
|
_impl_.profile_type_.Add(value);
|
|
}
|
|
inline void OpMetadata_ProfileInfo::add_profile_type(::xla::ProfileType value) {
|
|
_internal_add_profile_type(value);
|
|
// @@protoc_insertion_point(field_add:xla.OpMetadata.ProfileInfo.profile_type)
|
|
}
|
|
inline const ::PROTOBUF_NAMESPACE_ID::RepeatedField<int>&
|
|
OpMetadata_ProfileInfo::profile_type() const {
|
|
// @@protoc_insertion_point(field_list:xla.OpMetadata.ProfileInfo.profile_type)
|
|
return _impl_.profile_type_;
|
|
}
|
|
inline ::PROTOBUF_NAMESPACE_ID::RepeatedField<int>*
|
|
OpMetadata_ProfileInfo::_internal_mutable_profile_type() {
|
|
return &_impl_.profile_type_;
|
|
}
|
|
inline ::PROTOBUF_NAMESPACE_ID::RepeatedField<int>*
|
|
OpMetadata_ProfileInfo::mutable_profile_type() {
|
|
// @@protoc_insertion_point(field_mutable_list:xla.OpMetadata.ProfileInfo.profile_type)
|
|
return _internal_mutable_profile_type();
|
|
}
|
|
|
|
// double relative_speedup = 2;
|
|
inline void OpMetadata_ProfileInfo::clear_relative_speedup() {
|
|
_impl_.relative_speedup_ = 0;
|
|
}
|
|
inline double OpMetadata_ProfileInfo::_internal_relative_speedup() const {
|
|
return _impl_.relative_speedup_;
|
|
}
|
|
inline double OpMetadata_ProfileInfo::relative_speedup() const {
|
|
// @@protoc_insertion_point(field_get:xla.OpMetadata.ProfileInfo.relative_speedup)
|
|
return _internal_relative_speedup();
|
|
}
|
|
inline void OpMetadata_ProfileInfo::_internal_set_relative_speedup(double value) {
|
|
|
|
_impl_.relative_speedup_ = value;
|
|
}
|
|
inline void OpMetadata_ProfileInfo::set_relative_speedup(double value) {
|
|
_internal_set_relative_speedup(value);
|
|
// @@protoc_insertion_point(field_set:xla.OpMetadata.ProfileInfo.relative_speedup)
|
|
}
|
|
|
|
// .xla.ProfileSource profile_source = 3;
|
|
inline void OpMetadata_ProfileInfo::clear_profile_source() {
|
|
_impl_.profile_source_ = 0;
|
|
}
|
|
inline ::xla::ProfileSource OpMetadata_ProfileInfo::_internal_profile_source() const {
|
|
return static_cast< ::xla::ProfileSource >(_impl_.profile_source_);
|
|
}
|
|
inline ::xla::ProfileSource OpMetadata_ProfileInfo::profile_source() const {
|
|
// @@protoc_insertion_point(field_get:xla.OpMetadata.ProfileInfo.profile_source)
|
|
return _internal_profile_source();
|
|
}
|
|
inline void OpMetadata_ProfileInfo::_internal_set_profile_source(::xla::ProfileSource value) {
|
|
|
|
_impl_.profile_source_ = value;
|
|
}
|
|
inline void OpMetadata_ProfileInfo::set_profile_source(::xla::ProfileSource value) {
|
|
_internal_set_profile_source(value);
|
|
// @@protoc_insertion_point(field_set:xla.OpMetadata.ProfileInfo.profile_source)
|
|
}
|
|
|
|
// .xla.CompilationEvent compilation_event = 4;
|
|
inline void OpMetadata_ProfileInfo::clear_compilation_event() {
|
|
_impl_.compilation_event_ = 0;
|
|
}
|
|
inline ::xla::CompilationEvent OpMetadata_ProfileInfo::_internal_compilation_event() const {
|
|
return static_cast< ::xla::CompilationEvent >(_impl_.compilation_event_);
|
|
}
|
|
inline ::xla::CompilationEvent OpMetadata_ProfileInfo::compilation_event() const {
|
|
// @@protoc_insertion_point(field_get:xla.OpMetadata.ProfileInfo.compilation_event)
|
|
return _internal_compilation_event();
|
|
}
|
|
inline void OpMetadata_ProfileInfo::_internal_set_compilation_event(::xla::CompilationEvent value) {
|
|
|
|
_impl_.compilation_event_ = value;
|
|
}
|
|
inline void OpMetadata_ProfileInfo::set_compilation_event(::xla::CompilationEvent value) {
|
|
_internal_set_compilation_event(value);
|
|
// @@protoc_insertion_point(field_set:xla.OpMetadata.ProfileInfo.compilation_event)
|
|
}
|
|
|
|
// -------------------------------------------------------------------
|
|
|
|
// OpMetadata
|
|
|
|
// string op_type = 1;
|
|
inline void OpMetadata::clear_op_type() {
|
|
_impl_.op_type_.ClearToEmpty();
|
|
}
|
|
inline const std::string& OpMetadata::op_type() const {
|
|
// @@protoc_insertion_point(field_get:xla.OpMetadata.op_type)
|
|
return _internal_op_type();
|
|
}
|
|
template <typename ArgT0, typename... ArgT>
|
|
inline PROTOBUF_ALWAYS_INLINE
|
|
void OpMetadata::set_op_type(ArgT0&& arg0, ArgT... args) {
|
|
|
|
_impl_.op_type_.Set(static_cast<ArgT0 &&>(arg0), args..., GetArenaForAllocation());
|
|
// @@protoc_insertion_point(field_set:xla.OpMetadata.op_type)
|
|
}
|
|
inline std::string* OpMetadata::mutable_op_type() {
|
|
std::string* _s = _internal_mutable_op_type();
|
|
// @@protoc_insertion_point(field_mutable:xla.OpMetadata.op_type)
|
|
return _s;
|
|
}
|
|
inline const std::string& OpMetadata::_internal_op_type() const {
|
|
return _impl_.op_type_.Get();
|
|
}
|
|
inline void OpMetadata::_internal_set_op_type(const std::string& value) {
|
|
|
|
_impl_.op_type_.Set(value, GetArenaForAllocation());
|
|
}
|
|
inline std::string* OpMetadata::_internal_mutable_op_type() {
|
|
|
|
return _impl_.op_type_.Mutable(GetArenaForAllocation());
|
|
}
|
|
inline std::string* OpMetadata::release_op_type() {
|
|
// @@protoc_insertion_point(field_release:xla.OpMetadata.op_type)
|
|
return _impl_.op_type_.Release();
|
|
}
|
|
inline void OpMetadata::set_allocated_op_type(std::string* op_type) {
|
|
if (op_type != nullptr) {
|
|
|
|
} else {
|
|
|
|
}
|
|
_impl_.op_type_.SetAllocated(op_type, GetArenaForAllocation());
|
|
#ifdef PROTOBUF_FORCE_COPY_DEFAULT_STRING
|
|
if (_impl_.op_type_.IsDefault()) {
|
|
_impl_.op_type_.Set("", GetArenaForAllocation());
|
|
}
|
|
#endif // PROTOBUF_FORCE_COPY_DEFAULT_STRING
|
|
// @@protoc_insertion_point(field_set_allocated:xla.OpMetadata.op_type)
|
|
}
|
|
|
|
// string op_name = 2;
|
|
inline void OpMetadata::clear_op_name() {
|
|
_impl_.op_name_.ClearToEmpty();
|
|
}
|
|
inline const std::string& OpMetadata::op_name() const {
|
|
// @@protoc_insertion_point(field_get:xla.OpMetadata.op_name)
|
|
return _internal_op_name();
|
|
}
|
|
template <typename ArgT0, typename... ArgT>
|
|
inline PROTOBUF_ALWAYS_INLINE
|
|
void OpMetadata::set_op_name(ArgT0&& arg0, ArgT... args) {
|
|
|
|
_impl_.op_name_.Set(static_cast<ArgT0 &&>(arg0), args..., GetArenaForAllocation());
|
|
// @@protoc_insertion_point(field_set:xla.OpMetadata.op_name)
|
|
}
|
|
inline std::string* OpMetadata::mutable_op_name() {
|
|
std::string* _s = _internal_mutable_op_name();
|
|
// @@protoc_insertion_point(field_mutable:xla.OpMetadata.op_name)
|
|
return _s;
|
|
}
|
|
inline const std::string& OpMetadata::_internal_op_name() const {
|
|
return _impl_.op_name_.Get();
|
|
}
|
|
inline void OpMetadata::_internal_set_op_name(const std::string& value) {
|
|
|
|
_impl_.op_name_.Set(value, GetArenaForAllocation());
|
|
}
|
|
inline std::string* OpMetadata::_internal_mutable_op_name() {
|
|
|
|
return _impl_.op_name_.Mutable(GetArenaForAllocation());
|
|
}
|
|
inline std::string* OpMetadata::release_op_name() {
|
|
// @@protoc_insertion_point(field_release:xla.OpMetadata.op_name)
|
|
return _impl_.op_name_.Release();
|
|
}
|
|
inline void OpMetadata::set_allocated_op_name(std::string* op_name) {
|
|
if (op_name != nullptr) {
|
|
|
|
} else {
|
|
|
|
}
|
|
_impl_.op_name_.SetAllocated(op_name, GetArenaForAllocation());
|
|
#ifdef PROTOBUF_FORCE_COPY_DEFAULT_STRING
|
|
if (_impl_.op_name_.IsDefault()) {
|
|
_impl_.op_name_.Set("", GetArenaForAllocation());
|
|
}
|
|
#endif // PROTOBUF_FORCE_COPY_DEFAULT_STRING
|
|
// @@protoc_insertion_point(field_set_allocated:xla.OpMetadata.op_name)
|
|
}
|
|
|
|
// string source_file = 3;
|
|
inline void OpMetadata::clear_source_file() {
|
|
_impl_.source_file_.ClearToEmpty();
|
|
}
|
|
inline const std::string& OpMetadata::source_file() const {
|
|
// @@protoc_insertion_point(field_get:xla.OpMetadata.source_file)
|
|
return _internal_source_file();
|
|
}
|
|
template <typename ArgT0, typename... ArgT>
|
|
inline PROTOBUF_ALWAYS_INLINE
|
|
void OpMetadata::set_source_file(ArgT0&& arg0, ArgT... args) {
|
|
|
|
_impl_.source_file_.Set(static_cast<ArgT0 &&>(arg0), args..., GetArenaForAllocation());
|
|
// @@protoc_insertion_point(field_set:xla.OpMetadata.source_file)
|
|
}
|
|
inline std::string* OpMetadata::mutable_source_file() {
|
|
std::string* _s = _internal_mutable_source_file();
|
|
// @@protoc_insertion_point(field_mutable:xla.OpMetadata.source_file)
|
|
return _s;
|
|
}
|
|
inline const std::string& OpMetadata::_internal_source_file() const {
|
|
return _impl_.source_file_.Get();
|
|
}
|
|
inline void OpMetadata::_internal_set_source_file(const std::string& value) {
|
|
|
|
_impl_.source_file_.Set(value, GetArenaForAllocation());
|
|
}
|
|
inline std::string* OpMetadata::_internal_mutable_source_file() {
|
|
|
|
return _impl_.source_file_.Mutable(GetArenaForAllocation());
|
|
}
|
|
inline std::string* OpMetadata::release_source_file() {
|
|
// @@protoc_insertion_point(field_release:xla.OpMetadata.source_file)
|
|
return _impl_.source_file_.Release();
|
|
}
|
|
inline void OpMetadata::set_allocated_source_file(std::string* source_file) {
|
|
if (source_file != nullptr) {
|
|
|
|
} else {
|
|
|
|
}
|
|
_impl_.source_file_.SetAllocated(source_file, GetArenaForAllocation());
|
|
#ifdef PROTOBUF_FORCE_COPY_DEFAULT_STRING
|
|
if (_impl_.source_file_.IsDefault()) {
|
|
_impl_.source_file_.Set("", GetArenaForAllocation());
|
|
}
|
|
#endif // PROTOBUF_FORCE_COPY_DEFAULT_STRING
|
|
// @@protoc_insertion_point(field_set_allocated:xla.OpMetadata.source_file)
|
|
}
|
|
|
|
// int32 source_line = 4;
|
|
inline void OpMetadata::clear_source_line() {
|
|
_impl_.source_line_ = 0;
|
|
}
|
|
inline int32_t OpMetadata::_internal_source_line() const {
|
|
return _impl_.source_line_;
|
|
}
|
|
inline int32_t OpMetadata::source_line() const {
|
|
// @@protoc_insertion_point(field_get:xla.OpMetadata.source_line)
|
|
return _internal_source_line();
|
|
}
|
|
inline void OpMetadata::_internal_set_source_line(int32_t value) {
|
|
|
|
_impl_.source_line_ = value;
|
|
}
|
|
inline void OpMetadata::set_source_line(int32_t value) {
|
|
_internal_set_source_line(value);
|
|
// @@protoc_insertion_point(field_set:xla.OpMetadata.source_line)
|
|
}
|
|
|
|
// repeated .xla.ProfileType profile_type = 5 [deprecated = true];
|
|
inline int OpMetadata::_internal_profile_type_size() const {
|
|
return _impl_.profile_type_.size();
|
|
}
|
|
inline int OpMetadata::profile_type_size() const {
|
|
return _internal_profile_type_size();
|
|
}
|
|
inline void OpMetadata::clear_profile_type() {
|
|
_impl_.profile_type_.Clear();
|
|
}
|
|
inline ::xla::ProfileType OpMetadata::_internal_profile_type(int index) const {
|
|
return static_cast< ::xla::ProfileType >(_impl_.profile_type_.Get(index));
|
|
}
|
|
inline ::xla::ProfileType OpMetadata::profile_type(int index) const {
|
|
// @@protoc_insertion_point(field_get:xla.OpMetadata.profile_type)
|
|
return _internal_profile_type(index);
|
|
}
|
|
inline void OpMetadata::set_profile_type(int index, ::xla::ProfileType value) {
|
|
_impl_.profile_type_.Set(index, value);
|
|
// @@protoc_insertion_point(field_set:xla.OpMetadata.profile_type)
|
|
}
|
|
inline void OpMetadata::_internal_add_profile_type(::xla::ProfileType value) {
|
|
_impl_.profile_type_.Add(value);
|
|
}
|
|
inline void OpMetadata::add_profile_type(::xla::ProfileType value) {
|
|
_internal_add_profile_type(value);
|
|
// @@protoc_insertion_point(field_add:xla.OpMetadata.profile_type)
|
|
}
|
|
inline const ::PROTOBUF_NAMESPACE_ID::RepeatedField<int>&
|
|
OpMetadata::profile_type() const {
|
|
// @@protoc_insertion_point(field_list:xla.OpMetadata.profile_type)
|
|
return _impl_.profile_type_;
|
|
}
|
|
inline ::PROTOBUF_NAMESPACE_ID::RepeatedField<int>*
|
|
OpMetadata::_internal_mutable_profile_type() {
|
|
return &_impl_.profile_type_;
|
|
}
|
|
inline ::PROTOBUF_NAMESPACE_ID::RepeatedField<int>*
|
|
OpMetadata::mutable_profile_type() {
|
|
// @@protoc_insertion_point(field_mutable_list:xla.OpMetadata.profile_type)
|
|
return _internal_mutable_profile_type();
|
|
}
|
|
|
|
// int64 creation_pass_id = 6;
|
|
inline void OpMetadata::clear_creation_pass_id() {
|
|
_impl_.creation_pass_id_ = int64_t{0};
|
|
}
|
|
inline int64_t OpMetadata::_internal_creation_pass_id() const {
|
|
return _impl_.creation_pass_id_;
|
|
}
|
|
inline int64_t OpMetadata::creation_pass_id() const {
|
|
// @@protoc_insertion_point(field_get:xla.OpMetadata.creation_pass_id)
|
|
return _internal_creation_pass_id();
|
|
}
|
|
inline void OpMetadata::_internal_set_creation_pass_id(int64_t value) {
|
|
|
|
_impl_.creation_pass_id_ = value;
|
|
}
|
|
inline void OpMetadata::set_creation_pass_id(int64_t value) {
|
|
_internal_set_creation_pass_id(value);
|
|
// @@protoc_insertion_point(field_set:xla.OpMetadata.creation_pass_id)
|
|
}
|
|
|
|
// int64 logical_creation_pass_id = 7;
|
|
inline void OpMetadata::clear_logical_creation_pass_id() {
|
|
_impl_.logical_creation_pass_id_ = int64_t{0};
|
|
}
|
|
inline int64_t OpMetadata::_internal_logical_creation_pass_id() const {
|
|
return _impl_.logical_creation_pass_id_;
|
|
}
|
|
inline int64_t OpMetadata::logical_creation_pass_id() const {
|
|
// @@protoc_insertion_point(field_get:xla.OpMetadata.logical_creation_pass_id)
|
|
return _internal_logical_creation_pass_id();
|
|
}
|
|
inline void OpMetadata::_internal_set_logical_creation_pass_id(int64_t value) {
|
|
|
|
_impl_.logical_creation_pass_id_ = value;
|
|
}
|
|
inline void OpMetadata::set_logical_creation_pass_id(int64_t value) {
|
|
_internal_set_logical_creation_pass_id(value);
|
|
// @@protoc_insertion_point(field_set:xla.OpMetadata.logical_creation_pass_id)
|
|
}
|
|
|
|
// int64 size_of_generated_code_in_bytes = 8;
|
|
inline void OpMetadata::clear_size_of_generated_code_in_bytes() {
|
|
_impl_.size_of_generated_code_in_bytes_ = int64_t{0};
|
|
}
|
|
inline int64_t OpMetadata::_internal_size_of_generated_code_in_bytes() const {
|
|
return _impl_.size_of_generated_code_in_bytes_;
|
|
}
|
|
inline int64_t OpMetadata::size_of_generated_code_in_bytes() const {
|
|
// @@protoc_insertion_point(field_get:xla.OpMetadata.size_of_generated_code_in_bytes)
|
|
return _internal_size_of_generated_code_in_bytes();
|
|
}
|
|
inline void OpMetadata::_internal_set_size_of_generated_code_in_bytes(int64_t value) {
|
|
|
|
_impl_.size_of_generated_code_in_bytes_ = value;
|
|
}
|
|
inline void OpMetadata::set_size_of_generated_code_in_bytes(int64_t value) {
|
|
_internal_set_size_of_generated_code_in_bytes(value);
|
|
// @@protoc_insertion_point(field_set:xla.OpMetadata.size_of_generated_code_in_bytes)
|
|
}
|
|
|
|
// int64 size_of_memory_working_set_in_bytes = 9;
|
|
inline void OpMetadata::clear_size_of_memory_working_set_in_bytes() {
|
|
_impl_.size_of_memory_working_set_in_bytes_ = int64_t{0};
|
|
}
|
|
inline int64_t OpMetadata::_internal_size_of_memory_working_set_in_bytes() const {
|
|
return _impl_.size_of_memory_working_set_in_bytes_;
|
|
}
|
|
inline int64_t OpMetadata::size_of_memory_working_set_in_bytes() const {
|
|
// @@protoc_insertion_point(field_get:xla.OpMetadata.size_of_memory_working_set_in_bytes)
|
|
return _internal_size_of_memory_working_set_in_bytes();
|
|
}
|
|
inline void OpMetadata::_internal_set_size_of_memory_working_set_in_bytes(int64_t value) {
|
|
|
|
_impl_.size_of_memory_working_set_in_bytes_ = value;
|
|
}
|
|
inline void OpMetadata::set_size_of_memory_working_set_in_bytes(int64_t value) {
|
|
_internal_set_size_of_memory_working_set_in_bytes(value);
|
|
// @@protoc_insertion_point(field_set:xla.OpMetadata.size_of_memory_working_set_in_bytes)
|
|
}
|
|
|
|
// .xla.OpMetadata.ProfileInfo profile_info = 10;
|
|
inline bool OpMetadata::_internal_has_profile_info() const {
|
|
return this != internal_default_instance() && _impl_.profile_info_ != nullptr;
|
|
}
|
|
inline bool OpMetadata::has_profile_info() const {
|
|
return _internal_has_profile_info();
|
|
}
|
|
inline void OpMetadata::clear_profile_info() {
|
|
if (GetArenaForAllocation() == nullptr && _impl_.profile_info_ != nullptr) {
|
|
delete _impl_.profile_info_;
|
|
}
|
|
_impl_.profile_info_ = nullptr;
|
|
}
|
|
inline const ::xla::OpMetadata_ProfileInfo& OpMetadata::_internal_profile_info() const {
|
|
const ::xla::OpMetadata_ProfileInfo* p = _impl_.profile_info_;
|
|
return p != nullptr ? *p : reinterpret_cast<const ::xla::OpMetadata_ProfileInfo&>(
|
|
::xla::_OpMetadata_ProfileInfo_default_instance_);
|
|
}
|
|
inline const ::xla::OpMetadata_ProfileInfo& OpMetadata::profile_info() const {
|
|
// @@protoc_insertion_point(field_get:xla.OpMetadata.profile_info)
|
|
return _internal_profile_info();
|
|
}
|
|
inline void OpMetadata::unsafe_arena_set_allocated_profile_info(
|
|
::xla::OpMetadata_ProfileInfo* profile_info) {
|
|
if (GetArenaForAllocation() == nullptr) {
|
|
delete reinterpret_cast<::PROTOBUF_NAMESPACE_ID::MessageLite*>(_impl_.profile_info_);
|
|
}
|
|
_impl_.profile_info_ = profile_info;
|
|
if (profile_info) {
|
|
|
|
} else {
|
|
|
|
}
|
|
// @@protoc_insertion_point(field_unsafe_arena_set_allocated:xla.OpMetadata.profile_info)
|
|
}
|
|
inline ::xla::OpMetadata_ProfileInfo* OpMetadata::release_profile_info() {
|
|
|
|
::xla::OpMetadata_ProfileInfo* temp = _impl_.profile_info_;
|
|
_impl_.profile_info_ = nullptr;
|
|
#ifdef PROTOBUF_FORCE_COPY_IN_RELEASE
|
|
auto* old = reinterpret_cast<::PROTOBUF_NAMESPACE_ID::MessageLite*>(temp);
|
|
temp = ::PROTOBUF_NAMESPACE_ID::internal::DuplicateIfNonNull(temp);
|
|
if (GetArenaForAllocation() == nullptr) { delete old; }
|
|
#else // PROTOBUF_FORCE_COPY_IN_RELEASE
|
|
if (GetArenaForAllocation() != nullptr) {
|
|
temp = ::PROTOBUF_NAMESPACE_ID::internal::DuplicateIfNonNull(temp);
|
|
}
|
|
#endif // !PROTOBUF_FORCE_COPY_IN_RELEASE
|
|
return temp;
|
|
}
|
|
inline ::xla::OpMetadata_ProfileInfo* OpMetadata::unsafe_arena_release_profile_info() {
|
|
// @@protoc_insertion_point(field_release:xla.OpMetadata.profile_info)
|
|
|
|
::xla::OpMetadata_ProfileInfo* temp = _impl_.profile_info_;
|
|
_impl_.profile_info_ = nullptr;
|
|
return temp;
|
|
}
|
|
inline ::xla::OpMetadata_ProfileInfo* OpMetadata::_internal_mutable_profile_info() {
|
|
|
|
if (_impl_.profile_info_ == nullptr) {
|
|
auto* p = CreateMaybeMessage<::xla::OpMetadata_ProfileInfo>(GetArenaForAllocation());
|
|
_impl_.profile_info_ = p;
|
|
}
|
|
return _impl_.profile_info_;
|
|
}
|
|
inline ::xla::OpMetadata_ProfileInfo* OpMetadata::mutable_profile_info() {
|
|
::xla::OpMetadata_ProfileInfo* _msg = _internal_mutable_profile_info();
|
|
// @@protoc_insertion_point(field_mutable:xla.OpMetadata.profile_info)
|
|
return _msg;
|
|
}
|
|
inline void OpMetadata::set_allocated_profile_info(::xla::OpMetadata_ProfileInfo* profile_info) {
|
|
::PROTOBUF_NAMESPACE_ID::Arena* message_arena = GetArenaForAllocation();
|
|
if (message_arena == nullptr) {
|
|
delete _impl_.profile_info_;
|
|
}
|
|
if (profile_info) {
|
|
::PROTOBUF_NAMESPACE_ID::Arena* submessage_arena =
|
|
::PROTOBUF_NAMESPACE_ID::Arena::InternalGetOwningArena(profile_info);
|
|
if (message_arena != submessage_arena) {
|
|
profile_info = ::PROTOBUF_NAMESPACE_ID::internal::GetOwnedMessage(
|
|
message_arena, profile_info, submessage_arena);
|
|
}
|
|
|
|
} else {
|
|
|
|
}
|
|
_impl_.profile_info_ = profile_info;
|
|
// @@protoc_insertion_point(field_set_allocated:xla.OpMetadata.profile_info)
|
|
}
|
|
|
|
// string deduplicated_name = 12;
|
|
inline void OpMetadata::clear_deduplicated_name() {
|
|
_impl_.deduplicated_name_.ClearToEmpty();
|
|
}
|
|
inline const std::string& OpMetadata::deduplicated_name() const {
|
|
// @@protoc_insertion_point(field_get:xla.OpMetadata.deduplicated_name)
|
|
return _internal_deduplicated_name();
|
|
}
|
|
template <typename ArgT0, typename... ArgT>
|
|
inline PROTOBUF_ALWAYS_INLINE
|
|
void OpMetadata::set_deduplicated_name(ArgT0&& arg0, ArgT... args) {
|
|
|
|
_impl_.deduplicated_name_.Set(static_cast<ArgT0 &&>(arg0), args..., GetArenaForAllocation());
|
|
// @@protoc_insertion_point(field_set:xla.OpMetadata.deduplicated_name)
|
|
}
|
|
inline std::string* OpMetadata::mutable_deduplicated_name() {
|
|
std::string* _s = _internal_mutable_deduplicated_name();
|
|
// @@protoc_insertion_point(field_mutable:xla.OpMetadata.deduplicated_name)
|
|
return _s;
|
|
}
|
|
inline const std::string& OpMetadata::_internal_deduplicated_name() const {
|
|
return _impl_.deduplicated_name_.Get();
|
|
}
|
|
inline void OpMetadata::_internal_set_deduplicated_name(const std::string& value) {
|
|
|
|
_impl_.deduplicated_name_.Set(value, GetArenaForAllocation());
|
|
}
|
|
inline std::string* OpMetadata::_internal_mutable_deduplicated_name() {
|
|
|
|
return _impl_.deduplicated_name_.Mutable(GetArenaForAllocation());
|
|
}
|
|
inline std::string* OpMetadata::release_deduplicated_name() {
|
|
// @@protoc_insertion_point(field_release:xla.OpMetadata.deduplicated_name)
|
|
return _impl_.deduplicated_name_.Release();
|
|
}
|
|
inline void OpMetadata::set_allocated_deduplicated_name(std::string* deduplicated_name) {
|
|
if (deduplicated_name != nullptr) {
|
|
|
|
} else {
|
|
|
|
}
|
|
_impl_.deduplicated_name_.SetAllocated(deduplicated_name, GetArenaForAllocation());
|
|
#ifdef PROTOBUF_FORCE_COPY_DEFAULT_STRING
|
|
if (_impl_.deduplicated_name_.IsDefault()) {
|
|
_impl_.deduplicated_name_.Set("", GetArenaForAllocation());
|
|
}
|
|
#endif // PROTOBUF_FORCE_COPY_DEFAULT_STRING
|
|
// @@protoc_insertion_point(field_set_allocated:xla.OpMetadata.deduplicated_name)
|
|
}
|
|
|
|
// bool preserve_layout = 13;
|
|
inline void OpMetadata::clear_preserve_layout() {
|
|
_impl_.preserve_layout_ = false;
|
|
}
|
|
inline bool OpMetadata::_internal_preserve_layout() const {
|
|
return _impl_.preserve_layout_;
|
|
}
|
|
inline bool OpMetadata::preserve_layout() const {
|
|
// @@protoc_insertion_point(field_get:xla.OpMetadata.preserve_layout)
|
|
return _internal_preserve_layout();
|
|
}
|
|
inline void OpMetadata::_internal_set_preserve_layout(bool value) {
|
|
|
|
_impl_.preserve_layout_ = value;
|
|
}
|
|
inline void OpMetadata::set_preserve_layout(bool value) {
|
|
_internal_set_preserve_layout(value);
|
|
// @@protoc_insertion_point(field_set:xla.OpMetadata.preserve_layout)
|
|
}
|
|
|
|
// int32 stack_frame_id = 15;
|
|
inline void OpMetadata::clear_stack_frame_id() {
|
|
_impl_.stack_frame_id_ = 0;
|
|
}
|
|
inline int32_t OpMetadata::_internal_stack_frame_id() const {
|
|
return _impl_.stack_frame_id_;
|
|
}
|
|
inline int32_t OpMetadata::stack_frame_id() const {
|
|
// @@protoc_insertion_point(field_get:xla.OpMetadata.stack_frame_id)
|
|
return _internal_stack_frame_id();
|
|
}
|
|
inline void OpMetadata::_internal_set_stack_frame_id(int32_t value) {
|
|
|
|
_impl_.stack_frame_id_ = value;
|
|
}
|
|
inline void OpMetadata::set_stack_frame_id(int32_t value) {
|
|
_internal_set_stack_frame_id(value);
|
|
// @@protoc_insertion_point(field_set:xla.OpMetadata.stack_frame_id)
|
|
}
|
|
|
|
// -------------------------------------------------------------------
|
|
|
|
// ExecutionProfile
|
|
|
|
// bool compilation_cache_hit = 1;
|
|
inline void ExecutionProfile::clear_compilation_cache_hit() {
|
|
_impl_.compilation_cache_hit_ = false;
|
|
}
|
|
inline bool ExecutionProfile::_internal_compilation_cache_hit() const {
|
|
return _impl_.compilation_cache_hit_;
|
|
}
|
|
inline bool ExecutionProfile::compilation_cache_hit() const {
|
|
// @@protoc_insertion_point(field_get:xla.ExecutionProfile.compilation_cache_hit)
|
|
return _internal_compilation_cache_hit();
|
|
}
|
|
inline void ExecutionProfile::_internal_set_compilation_cache_hit(bool value) {
|
|
|
|
_impl_.compilation_cache_hit_ = value;
|
|
}
|
|
inline void ExecutionProfile::set_compilation_cache_hit(bool value) {
|
|
_internal_set_compilation_cache_hit(value);
|
|
// @@protoc_insertion_point(field_set:xla.ExecutionProfile.compilation_cache_hit)
|
|
}
|
|
|
|
// int64 compile_time_ms = 2;
|
|
inline void ExecutionProfile::clear_compile_time_ms() {
|
|
_impl_.compile_time_ms_ = int64_t{0};
|
|
}
|
|
inline int64_t ExecutionProfile::_internal_compile_time_ms() const {
|
|
return _impl_.compile_time_ms_;
|
|
}
|
|
inline int64_t ExecutionProfile::compile_time_ms() const {
|
|
// @@protoc_insertion_point(field_get:xla.ExecutionProfile.compile_time_ms)
|
|
return _internal_compile_time_ms();
|
|
}
|
|
inline void ExecutionProfile::_internal_set_compile_time_ms(int64_t value) {
|
|
|
|
_impl_.compile_time_ms_ = value;
|
|
}
|
|
inline void ExecutionProfile::set_compile_time_ms(int64_t value) {
|
|
_internal_set_compile_time_ms(value);
|
|
// @@protoc_insertion_point(field_set:xla.ExecutionProfile.compile_time_ms)
|
|
}
|
|
|
|
// int64 compute_cycle_count = 3;
|
|
inline void ExecutionProfile::clear_compute_cycle_count() {
|
|
_impl_.compute_cycle_count_ = int64_t{0};
|
|
}
|
|
inline int64_t ExecutionProfile::_internal_compute_cycle_count() const {
|
|
return _impl_.compute_cycle_count_;
|
|
}
|
|
inline int64_t ExecutionProfile::compute_cycle_count() const {
|
|
// @@protoc_insertion_point(field_get:xla.ExecutionProfile.compute_cycle_count)
|
|
return _internal_compute_cycle_count();
|
|
}
|
|
inline void ExecutionProfile::_internal_set_compute_cycle_count(int64_t value) {
|
|
|
|
_impl_.compute_cycle_count_ = value;
|
|
}
|
|
inline void ExecutionProfile::set_compute_cycle_count(int64_t value) {
|
|
_internal_set_compute_cycle_count(value);
|
|
// @@protoc_insertion_point(field_set:xla.ExecutionProfile.compute_cycle_count)
|
|
}
|
|
|
|
// int64 compute_time_ns = 4;
|
|
inline void ExecutionProfile::clear_compute_time_ns() {
|
|
_impl_.compute_time_ns_ = int64_t{0};
|
|
}
|
|
inline int64_t ExecutionProfile::_internal_compute_time_ns() const {
|
|
return _impl_.compute_time_ns_;
|
|
}
|
|
inline int64_t ExecutionProfile::compute_time_ns() const {
|
|
// @@protoc_insertion_point(field_get:xla.ExecutionProfile.compute_time_ns)
|
|
return _internal_compute_time_ns();
|
|
}
|
|
inline void ExecutionProfile::_internal_set_compute_time_ns(int64_t value) {
|
|
|
|
_impl_.compute_time_ns_ = value;
|
|
}
|
|
inline void ExecutionProfile::set_compute_time_ns(int64_t value) {
|
|
_internal_set_compute_time_ns(value);
|
|
// @@protoc_insertion_point(field_set:xla.ExecutionProfile.compute_time_ns)
|
|
}
|
|
|
|
// int64 compute_and_transfer_time_ns = 5;
|
|
inline void ExecutionProfile::clear_compute_and_transfer_time_ns() {
|
|
_impl_.compute_and_transfer_time_ns_ = int64_t{0};
|
|
}
|
|
inline int64_t ExecutionProfile::_internal_compute_and_transfer_time_ns() const {
|
|
return _impl_.compute_and_transfer_time_ns_;
|
|
}
|
|
inline int64_t ExecutionProfile::compute_and_transfer_time_ns() const {
|
|
// @@protoc_insertion_point(field_get:xla.ExecutionProfile.compute_and_transfer_time_ns)
|
|
return _internal_compute_and_transfer_time_ns();
|
|
}
|
|
inline void ExecutionProfile::_internal_set_compute_and_transfer_time_ns(int64_t value) {
|
|
|
|
_impl_.compute_and_transfer_time_ns_ = value;
|
|
}
|
|
inline void ExecutionProfile::set_compute_and_transfer_time_ns(int64_t value) {
|
|
_internal_set_compute_and_transfer_time_ns(value);
|
|
// @@protoc_insertion_point(field_set:xla.ExecutionProfile.compute_and_transfer_time_ns)
|
|
}
|
|
|
|
// int64 executable_size_in_bytes = 6;
|
|
inline void ExecutionProfile::clear_executable_size_in_bytes() {
|
|
_impl_.executable_size_in_bytes_ = int64_t{0};
|
|
}
|
|
inline int64_t ExecutionProfile::_internal_executable_size_in_bytes() const {
|
|
return _impl_.executable_size_in_bytes_;
|
|
}
|
|
inline int64_t ExecutionProfile::executable_size_in_bytes() const {
|
|
// @@protoc_insertion_point(field_get:xla.ExecutionProfile.executable_size_in_bytes)
|
|
return _internal_executable_size_in_bytes();
|
|
}
|
|
inline void ExecutionProfile::_internal_set_executable_size_in_bytes(int64_t value) {
|
|
|
|
_impl_.executable_size_in_bytes_ = value;
|
|
}
|
|
inline void ExecutionProfile::set_executable_size_in_bytes(int64_t value) {
|
|
_internal_set_executable_size_in_bytes(value);
|
|
// @@protoc_insertion_point(field_set:xla.ExecutionProfile.executable_size_in_bytes)
|
|
}
|
|
|
|
// bool profile_cache_hit = 7;
|
|
inline void ExecutionProfile::clear_profile_cache_hit() {
|
|
_impl_.profile_cache_hit_ = false;
|
|
}
|
|
inline bool ExecutionProfile::_internal_profile_cache_hit() const {
|
|
return _impl_.profile_cache_hit_;
|
|
}
|
|
inline bool ExecutionProfile::profile_cache_hit() const {
|
|
// @@protoc_insertion_point(field_get:xla.ExecutionProfile.profile_cache_hit)
|
|
return _internal_profile_cache_hit();
|
|
}
|
|
inline void ExecutionProfile::_internal_set_profile_cache_hit(bool value) {
|
|
|
|
_impl_.profile_cache_hit_ = value;
|
|
}
|
|
inline void ExecutionProfile::set_profile_cache_hit(bool value) {
|
|
_internal_set_profile_cache_hit(value);
|
|
// @@protoc_insertion_point(field_set:xla.ExecutionProfile.profile_cache_hit)
|
|
}
|
|
|
|
// -------------------------------------------------------------------
|
|
|
|
// ExecutionHandle
|
|
|
|
// int64 handle = 1;
|
|
inline void ExecutionHandle::clear_handle() {
|
|
_impl_.handle_ = int64_t{0};
|
|
}
|
|
inline int64_t ExecutionHandle::_internal_handle() const {
|
|
return _impl_.handle_;
|
|
}
|
|
inline int64_t ExecutionHandle::handle() const {
|
|
// @@protoc_insertion_point(field_get:xla.ExecutionHandle.handle)
|
|
return _internal_handle();
|
|
}
|
|
inline void ExecutionHandle::_internal_set_handle(int64_t value) {
|
|
|
|
_impl_.handle_ = value;
|
|
}
|
|
inline void ExecutionHandle::set_handle(int64_t value) {
|
|
_internal_set_handle(value);
|
|
// @@protoc_insertion_point(field_set:xla.ExecutionHandle.handle)
|
|
}
|
|
|
|
// -------------------------------------------------------------------
|
|
|
|
// GlobalDataHandle
|
|
|
|
// int64 handle = 1;
|
|
inline void GlobalDataHandle::clear_handle() {
|
|
_impl_.handle_ = int64_t{0};
|
|
}
|
|
inline int64_t GlobalDataHandle::_internal_handle() const {
|
|
return _impl_.handle_;
|
|
}
|
|
inline int64_t GlobalDataHandle::handle() const {
|
|
// @@protoc_insertion_point(field_get:xla.GlobalDataHandle.handle)
|
|
return _internal_handle();
|
|
}
|
|
inline void GlobalDataHandle::_internal_set_handle(int64_t value) {
|
|
|
|
_impl_.handle_ = value;
|
|
}
|
|
inline void GlobalDataHandle::set_handle(int64_t value) {
|
|
_internal_set_handle(value);
|
|
// @@protoc_insertion_point(field_set:xla.GlobalDataHandle.handle)
|
|
}
|
|
|
|
// -------------------------------------------------------------------
|
|
|
|
// DeviceHandle
|
|
|
|
// int64 handle = 1;
|
|
inline void DeviceHandle::clear_handle() {
|
|
_impl_.handle_ = int64_t{0};
|
|
}
|
|
inline int64_t DeviceHandle::_internal_handle() const {
|
|
return _impl_.handle_;
|
|
}
|
|
inline int64_t DeviceHandle::handle() const {
|
|
// @@protoc_insertion_point(field_get:xla.DeviceHandle.handle)
|
|
return _internal_handle();
|
|
}
|
|
inline void DeviceHandle::_internal_set_handle(int64_t value) {
|
|
|
|
_impl_.handle_ = value;
|
|
}
|
|
inline void DeviceHandle::set_handle(int64_t value) {
|
|
_internal_set_handle(value);
|
|
// @@protoc_insertion_point(field_set:xla.DeviceHandle.handle)
|
|
}
|
|
|
|
// int64 device_count = 2;
|
|
inline void DeviceHandle::clear_device_count() {
|
|
_impl_.device_count_ = int64_t{0};
|
|
}
|
|
inline int64_t DeviceHandle::_internal_device_count() const {
|
|
return _impl_.device_count_;
|
|
}
|
|
inline int64_t DeviceHandle::device_count() const {
|
|
// @@protoc_insertion_point(field_get:xla.DeviceHandle.device_count)
|
|
return _internal_device_count();
|
|
}
|
|
inline void DeviceHandle::_internal_set_device_count(int64_t value) {
|
|
|
|
_impl_.device_count_ = value;
|
|
}
|
|
inline void DeviceHandle::set_device_count(int64_t value) {
|
|
_internal_set_device_count(value);
|
|
// @@protoc_insertion_point(field_set:xla.DeviceHandle.device_count)
|
|
}
|
|
|
|
// -------------------------------------------------------------------
|
|
|
|
// ChannelHandle
|
|
|
|
// int64 handle = 1;
|
|
inline void ChannelHandle::clear_handle() {
|
|
_impl_.handle_ = int64_t{0};
|
|
}
|
|
inline int64_t ChannelHandle::_internal_handle() const {
|
|
return _impl_.handle_;
|
|
}
|
|
inline int64_t ChannelHandle::handle() const {
|
|
// @@protoc_insertion_point(field_get:xla.ChannelHandle.handle)
|
|
return _internal_handle();
|
|
}
|
|
inline void ChannelHandle::_internal_set_handle(int64_t value) {
|
|
|
|
_impl_.handle_ = value;
|
|
}
|
|
inline void ChannelHandle::set_handle(int64_t value) {
|
|
_internal_set_handle(value);
|
|
// @@protoc_insertion_point(field_set:xla.ChannelHandle.handle)
|
|
}
|
|
|
|
// .xla.ChannelHandle.ChannelType type = 2;
|
|
inline void ChannelHandle::clear_type() {
|
|
_impl_.type_ = 0;
|
|
}
|
|
inline ::xla::ChannelHandle_ChannelType ChannelHandle::_internal_type() const {
|
|
return static_cast< ::xla::ChannelHandle_ChannelType >(_impl_.type_);
|
|
}
|
|
inline ::xla::ChannelHandle_ChannelType ChannelHandle::type() const {
|
|
// @@protoc_insertion_point(field_get:xla.ChannelHandle.type)
|
|
return _internal_type();
|
|
}
|
|
inline void ChannelHandle::_internal_set_type(::xla::ChannelHandle_ChannelType value) {
|
|
|
|
_impl_.type_ = value;
|
|
}
|
|
inline void ChannelHandle::set_type(::xla::ChannelHandle_ChannelType value) {
|
|
_internal_set_type(value);
|
|
// @@protoc_insertion_point(field_set:xla.ChannelHandle.type)
|
|
}
|
|
|
|
// -------------------------------------------------------------------
|
|
|
|
// DeviceAssignmentProto_ComputationDevice
|
|
|
|
// repeated int32 replica_device_ids = 1;
|
|
inline int DeviceAssignmentProto_ComputationDevice::_internal_replica_device_ids_size() const {
|
|
return _impl_.replica_device_ids_.size();
|
|
}
|
|
inline int DeviceAssignmentProto_ComputationDevice::replica_device_ids_size() const {
|
|
return _internal_replica_device_ids_size();
|
|
}
|
|
inline void DeviceAssignmentProto_ComputationDevice::clear_replica_device_ids() {
|
|
_impl_.replica_device_ids_.Clear();
|
|
}
|
|
inline int32_t DeviceAssignmentProto_ComputationDevice::_internal_replica_device_ids(int index) const {
|
|
return _impl_.replica_device_ids_.Get(index);
|
|
}
|
|
inline int32_t DeviceAssignmentProto_ComputationDevice::replica_device_ids(int index) const {
|
|
// @@protoc_insertion_point(field_get:xla.DeviceAssignmentProto.ComputationDevice.replica_device_ids)
|
|
return _internal_replica_device_ids(index);
|
|
}
|
|
inline void DeviceAssignmentProto_ComputationDevice::set_replica_device_ids(int index, int32_t value) {
|
|
_impl_.replica_device_ids_.Set(index, value);
|
|
// @@protoc_insertion_point(field_set:xla.DeviceAssignmentProto.ComputationDevice.replica_device_ids)
|
|
}
|
|
inline void DeviceAssignmentProto_ComputationDevice::_internal_add_replica_device_ids(int32_t value) {
|
|
_impl_.replica_device_ids_.Add(value);
|
|
}
|
|
inline void DeviceAssignmentProto_ComputationDevice::add_replica_device_ids(int32_t value) {
|
|
_internal_add_replica_device_ids(value);
|
|
// @@protoc_insertion_point(field_add:xla.DeviceAssignmentProto.ComputationDevice.replica_device_ids)
|
|
}
|
|
inline const ::PROTOBUF_NAMESPACE_ID::RepeatedField< int32_t >&
|
|
DeviceAssignmentProto_ComputationDevice::_internal_replica_device_ids() const {
|
|
return _impl_.replica_device_ids_;
|
|
}
|
|
inline const ::PROTOBUF_NAMESPACE_ID::RepeatedField< int32_t >&
|
|
DeviceAssignmentProto_ComputationDevice::replica_device_ids() const {
|
|
// @@protoc_insertion_point(field_list:xla.DeviceAssignmentProto.ComputationDevice.replica_device_ids)
|
|
return _internal_replica_device_ids();
|
|
}
|
|
inline ::PROTOBUF_NAMESPACE_ID::RepeatedField< int32_t >*
|
|
DeviceAssignmentProto_ComputationDevice::_internal_mutable_replica_device_ids() {
|
|
return &_impl_.replica_device_ids_;
|
|
}
|
|
inline ::PROTOBUF_NAMESPACE_ID::RepeatedField< int32_t >*
|
|
DeviceAssignmentProto_ComputationDevice::mutable_replica_device_ids() {
|
|
// @@protoc_insertion_point(field_mutable_list:xla.DeviceAssignmentProto.ComputationDevice.replica_device_ids)
|
|
return _internal_mutable_replica_device_ids();
|
|
}
|
|
|
|
// -------------------------------------------------------------------
|
|
|
|
// DeviceAssignmentProto
|
|
|
|
// int32 replica_count = 1;
|
|
inline void DeviceAssignmentProto::clear_replica_count() {
|
|
_impl_.replica_count_ = 0;
|
|
}
|
|
inline int32_t DeviceAssignmentProto::_internal_replica_count() const {
|
|
return _impl_.replica_count_;
|
|
}
|
|
inline int32_t DeviceAssignmentProto::replica_count() const {
|
|
// @@protoc_insertion_point(field_get:xla.DeviceAssignmentProto.replica_count)
|
|
return _internal_replica_count();
|
|
}
|
|
inline void DeviceAssignmentProto::_internal_set_replica_count(int32_t value) {
|
|
|
|
_impl_.replica_count_ = value;
|
|
}
|
|
inline void DeviceAssignmentProto::set_replica_count(int32_t value) {
|
|
_internal_set_replica_count(value);
|
|
// @@protoc_insertion_point(field_set:xla.DeviceAssignmentProto.replica_count)
|
|
}
|
|
|
|
// int32 computation_count = 2;
|
|
inline void DeviceAssignmentProto::clear_computation_count() {
|
|
_impl_.computation_count_ = 0;
|
|
}
|
|
inline int32_t DeviceAssignmentProto::_internal_computation_count() const {
|
|
return _impl_.computation_count_;
|
|
}
|
|
inline int32_t DeviceAssignmentProto::computation_count() const {
|
|
// @@protoc_insertion_point(field_get:xla.DeviceAssignmentProto.computation_count)
|
|
return _internal_computation_count();
|
|
}
|
|
inline void DeviceAssignmentProto::_internal_set_computation_count(int32_t value) {
|
|
|
|
_impl_.computation_count_ = value;
|
|
}
|
|
inline void DeviceAssignmentProto::set_computation_count(int32_t value) {
|
|
_internal_set_computation_count(value);
|
|
// @@protoc_insertion_point(field_set:xla.DeviceAssignmentProto.computation_count)
|
|
}
|
|
|
|
// repeated .xla.DeviceAssignmentProto.ComputationDevice computation_devices = 3;
|
|
inline int DeviceAssignmentProto::_internal_computation_devices_size() const {
|
|
return _impl_.computation_devices_.size();
|
|
}
|
|
inline int DeviceAssignmentProto::computation_devices_size() const {
|
|
return _internal_computation_devices_size();
|
|
}
|
|
inline void DeviceAssignmentProto::clear_computation_devices() {
|
|
_impl_.computation_devices_.Clear();
|
|
}
|
|
inline ::xla::DeviceAssignmentProto_ComputationDevice* DeviceAssignmentProto::mutable_computation_devices(int index) {
|
|
// @@protoc_insertion_point(field_mutable:xla.DeviceAssignmentProto.computation_devices)
|
|
return _impl_.computation_devices_.Mutable(index);
|
|
}
|
|
inline ::PROTOBUF_NAMESPACE_ID::RepeatedPtrField< ::xla::DeviceAssignmentProto_ComputationDevice >*
|
|
DeviceAssignmentProto::mutable_computation_devices() {
|
|
// @@protoc_insertion_point(field_mutable_list:xla.DeviceAssignmentProto.computation_devices)
|
|
return &_impl_.computation_devices_;
|
|
}
|
|
inline const ::xla::DeviceAssignmentProto_ComputationDevice& DeviceAssignmentProto::_internal_computation_devices(int index) const {
|
|
return _impl_.computation_devices_.Get(index);
|
|
}
|
|
inline const ::xla::DeviceAssignmentProto_ComputationDevice& DeviceAssignmentProto::computation_devices(int index) const {
|
|
// @@protoc_insertion_point(field_get:xla.DeviceAssignmentProto.computation_devices)
|
|
return _internal_computation_devices(index);
|
|
}
|
|
inline ::xla::DeviceAssignmentProto_ComputationDevice* DeviceAssignmentProto::_internal_add_computation_devices() {
|
|
return _impl_.computation_devices_.Add();
|
|
}
|
|
inline ::xla::DeviceAssignmentProto_ComputationDevice* DeviceAssignmentProto::add_computation_devices() {
|
|
::xla::DeviceAssignmentProto_ComputationDevice* _add = _internal_add_computation_devices();
|
|
// @@protoc_insertion_point(field_add:xla.DeviceAssignmentProto.computation_devices)
|
|
return _add;
|
|
}
|
|
inline const ::PROTOBUF_NAMESPACE_ID::RepeatedPtrField< ::xla::DeviceAssignmentProto_ComputationDevice >&
|
|
DeviceAssignmentProto::computation_devices() const {
|
|
// @@protoc_insertion_point(field_list:xla.DeviceAssignmentProto.computation_devices)
|
|
return _impl_.computation_devices_;
|
|
}
|
|
|
|
// -------------------------------------------------------------------
|
|
|
|
// LiteralProto
|
|
|
|
// .xla.ShapeProto shape = 1;
|
|
inline bool LiteralProto::_internal_has_shape() const {
|
|
return this != internal_default_instance() && _impl_.shape_ != nullptr;
|
|
}
|
|
inline bool LiteralProto::has_shape() const {
|
|
return _internal_has_shape();
|
|
}
|
|
inline void LiteralProto::clear_shape() {
|
|
if (GetArenaForAllocation() == nullptr && _impl_.shape_ != nullptr) {
|
|
delete _impl_.shape_;
|
|
}
|
|
_impl_.shape_ = nullptr;
|
|
}
|
|
inline const ::xla::ShapeProto& LiteralProto::_internal_shape() const {
|
|
const ::xla::ShapeProto* p = _impl_.shape_;
|
|
return p != nullptr ? *p : reinterpret_cast<const ::xla::ShapeProto&>(
|
|
::xla::_ShapeProto_default_instance_);
|
|
}
|
|
inline const ::xla::ShapeProto& LiteralProto::shape() const {
|
|
// @@protoc_insertion_point(field_get:xla.LiteralProto.shape)
|
|
return _internal_shape();
|
|
}
|
|
inline void LiteralProto::unsafe_arena_set_allocated_shape(
|
|
::xla::ShapeProto* shape) {
|
|
if (GetArenaForAllocation() == nullptr) {
|
|
delete reinterpret_cast<::PROTOBUF_NAMESPACE_ID::MessageLite*>(_impl_.shape_);
|
|
}
|
|
_impl_.shape_ = shape;
|
|
if (shape) {
|
|
|
|
} else {
|
|
|
|
}
|
|
// @@protoc_insertion_point(field_unsafe_arena_set_allocated:xla.LiteralProto.shape)
|
|
}
|
|
inline ::xla::ShapeProto* LiteralProto::release_shape() {
|
|
|
|
::xla::ShapeProto* temp = _impl_.shape_;
|
|
_impl_.shape_ = nullptr;
|
|
#ifdef PROTOBUF_FORCE_COPY_IN_RELEASE
|
|
auto* old = reinterpret_cast<::PROTOBUF_NAMESPACE_ID::MessageLite*>(temp);
|
|
temp = ::PROTOBUF_NAMESPACE_ID::internal::DuplicateIfNonNull(temp);
|
|
if (GetArenaForAllocation() == nullptr) { delete old; }
|
|
#else // PROTOBUF_FORCE_COPY_IN_RELEASE
|
|
if (GetArenaForAllocation() != nullptr) {
|
|
temp = ::PROTOBUF_NAMESPACE_ID::internal::DuplicateIfNonNull(temp);
|
|
}
|
|
#endif // !PROTOBUF_FORCE_COPY_IN_RELEASE
|
|
return temp;
|
|
}
|
|
inline ::xla::ShapeProto* LiteralProto::unsafe_arena_release_shape() {
|
|
// @@protoc_insertion_point(field_release:xla.LiteralProto.shape)
|
|
|
|
::xla::ShapeProto* temp = _impl_.shape_;
|
|
_impl_.shape_ = nullptr;
|
|
return temp;
|
|
}
|
|
inline ::xla::ShapeProto* LiteralProto::_internal_mutable_shape() {
|
|
|
|
if (_impl_.shape_ == nullptr) {
|
|
auto* p = CreateMaybeMessage<::xla::ShapeProto>(GetArenaForAllocation());
|
|
_impl_.shape_ = p;
|
|
}
|
|
return _impl_.shape_;
|
|
}
|
|
inline ::xla::ShapeProto* LiteralProto::mutable_shape() {
|
|
::xla::ShapeProto* _msg = _internal_mutable_shape();
|
|
// @@protoc_insertion_point(field_mutable:xla.LiteralProto.shape)
|
|
return _msg;
|
|
}
|
|
inline void LiteralProto::set_allocated_shape(::xla::ShapeProto* shape) {
|
|
::PROTOBUF_NAMESPACE_ID::Arena* message_arena = GetArenaForAllocation();
|
|
if (message_arena == nullptr) {
|
|
delete _impl_.shape_;
|
|
}
|
|
if (shape) {
|
|
::PROTOBUF_NAMESPACE_ID::Arena* submessage_arena =
|
|
::PROTOBUF_NAMESPACE_ID::Arena::InternalGetOwningArena(shape);
|
|
if (message_arena != submessage_arena) {
|
|
shape = ::PROTOBUF_NAMESPACE_ID::internal::GetOwnedMessage(
|
|
message_arena, shape, submessage_arena);
|
|
}
|
|
|
|
} else {
|
|
|
|
}
|
|
_impl_.shape_ = shape;
|
|
// @@protoc_insertion_point(field_set_allocated:xla.LiteralProto.shape)
|
|
}
|
|
|
|
// repeated bool preds = 2;
|
|
inline int LiteralProto::_internal_preds_size() const {
|
|
return _impl_.preds_.size();
|
|
}
|
|
inline int LiteralProto::preds_size() const {
|
|
return _internal_preds_size();
|
|
}
|
|
inline void LiteralProto::clear_preds() {
|
|
_impl_.preds_.Clear();
|
|
}
|
|
inline bool LiteralProto::_internal_preds(int index) const {
|
|
return _impl_.preds_.Get(index);
|
|
}
|
|
inline bool LiteralProto::preds(int index) const {
|
|
// @@protoc_insertion_point(field_get:xla.LiteralProto.preds)
|
|
return _internal_preds(index);
|
|
}
|
|
inline void LiteralProto::set_preds(int index, bool value) {
|
|
_impl_.preds_.Set(index, value);
|
|
// @@protoc_insertion_point(field_set:xla.LiteralProto.preds)
|
|
}
|
|
inline void LiteralProto::_internal_add_preds(bool value) {
|
|
_impl_.preds_.Add(value);
|
|
}
|
|
inline void LiteralProto::add_preds(bool value) {
|
|
_internal_add_preds(value);
|
|
// @@protoc_insertion_point(field_add:xla.LiteralProto.preds)
|
|
}
|
|
inline const ::PROTOBUF_NAMESPACE_ID::RepeatedField< bool >&
|
|
LiteralProto::_internal_preds() const {
|
|
return _impl_.preds_;
|
|
}
|
|
inline const ::PROTOBUF_NAMESPACE_ID::RepeatedField< bool >&
|
|
LiteralProto::preds() const {
|
|
// @@protoc_insertion_point(field_list:xla.LiteralProto.preds)
|
|
return _internal_preds();
|
|
}
|
|
inline ::PROTOBUF_NAMESPACE_ID::RepeatedField< bool >*
|
|
LiteralProto::_internal_mutable_preds() {
|
|
return &_impl_.preds_;
|
|
}
|
|
inline ::PROTOBUF_NAMESPACE_ID::RepeatedField< bool >*
|
|
LiteralProto::mutable_preds() {
|
|
// @@protoc_insertion_point(field_mutable_list:xla.LiteralProto.preds)
|
|
return _internal_mutable_preds();
|
|
}
|
|
|
|
// bytes s4s = 21;
|
|
inline void LiteralProto::clear_s4s() {
|
|
_impl_.s4s_.ClearToEmpty();
|
|
}
|
|
inline const std::string& LiteralProto::s4s() const {
|
|
// @@protoc_insertion_point(field_get:xla.LiteralProto.s4s)
|
|
return _internal_s4s();
|
|
}
|
|
template <typename ArgT0, typename... ArgT>
|
|
inline PROTOBUF_ALWAYS_INLINE
|
|
void LiteralProto::set_s4s(ArgT0&& arg0, ArgT... args) {
|
|
|
|
_impl_.s4s_.SetBytes(static_cast<ArgT0 &&>(arg0), args..., GetArenaForAllocation());
|
|
// @@protoc_insertion_point(field_set:xla.LiteralProto.s4s)
|
|
}
|
|
inline std::string* LiteralProto::mutable_s4s() {
|
|
std::string* _s = _internal_mutable_s4s();
|
|
// @@protoc_insertion_point(field_mutable:xla.LiteralProto.s4s)
|
|
return _s;
|
|
}
|
|
inline const std::string& LiteralProto::_internal_s4s() const {
|
|
return _impl_.s4s_.Get();
|
|
}
|
|
inline void LiteralProto::_internal_set_s4s(const std::string& value) {
|
|
|
|
_impl_.s4s_.Set(value, GetArenaForAllocation());
|
|
}
|
|
inline std::string* LiteralProto::_internal_mutable_s4s() {
|
|
|
|
return _impl_.s4s_.Mutable(GetArenaForAllocation());
|
|
}
|
|
inline std::string* LiteralProto::release_s4s() {
|
|
// @@protoc_insertion_point(field_release:xla.LiteralProto.s4s)
|
|
return _impl_.s4s_.Release();
|
|
}
|
|
inline void LiteralProto::set_allocated_s4s(std::string* s4s) {
|
|
if (s4s != nullptr) {
|
|
|
|
} else {
|
|
|
|
}
|
|
_impl_.s4s_.SetAllocated(s4s, GetArenaForAllocation());
|
|
#ifdef PROTOBUF_FORCE_COPY_DEFAULT_STRING
|
|
if (_impl_.s4s_.IsDefault()) {
|
|
_impl_.s4s_.Set("", GetArenaForAllocation());
|
|
}
|
|
#endif // PROTOBUF_FORCE_COPY_DEFAULT_STRING
|
|
// @@protoc_insertion_point(field_set_allocated:xla.LiteralProto.s4s)
|
|
}
|
|
|
|
// bytes u4s = 22;
|
|
inline void LiteralProto::clear_u4s() {
|
|
_impl_.u4s_.ClearToEmpty();
|
|
}
|
|
inline const std::string& LiteralProto::u4s() const {
|
|
// @@protoc_insertion_point(field_get:xla.LiteralProto.u4s)
|
|
return _internal_u4s();
|
|
}
|
|
template <typename ArgT0, typename... ArgT>
|
|
inline PROTOBUF_ALWAYS_INLINE
|
|
void LiteralProto::set_u4s(ArgT0&& arg0, ArgT... args) {
|
|
|
|
_impl_.u4s_.SetBytes(static_cast<ArgT0 &&>(arg0), args..., GetArenaForAllocation());
|
|
// @@protoc_insertion_point(field_set:xla.LiteralProto.u4s)
|
|
}
|
|
inline std::string* LiteralProto::mutable_u4s() {
|
|
std::string* _s = _internal_mutable_u4s();
|
|
// @@protoc_insertion_point(field_mutable:xla.LiteralProto.u4s)
|
|
return _s;
|
|
}
|
|
inline const std::string& LiteralProto::_internal_u4s() const {
|
|
return _impl_.u4s_.Get();
|
|
}
|
|
inline void LiteralProto::_internal_set_u4s(const std::string& value) {
|
|
|
|
_impl_.u4s_.Set(value, GetArenaForAllocation());
|
|
}
|
|
inline std::string* LiteralProto::_internal_mutable_u4s() {
|
|
|
|
return _impl_.u4s_.Mutable(GetArenaForAllocation());
|
|
}
|
|
inline std::string* LiteralProto::release_u4s() {
|
|
// @@protoc_insertion_point(field_release:xla.LiteralProto.u4s)
|
|
return _impl_.u4s_.Release();
|
|
}
|
|
inline void LiteralProto::set_allocated_u4s(std::string* u4s) {
|
|
if (u4s != nullptr) {
|
|
|
|
} else {
|
|
|
|
}
|
|
_impl_.u4s_.SetAllocated(u4s, GetArenaForAllocation());
|
|
#ifdef PROTOBUF_FORCE_COPY_DEFAULT_STRING
|
|
if (_impl_.u4s_.IsDefault()) {
|
|
_impl_.u4s_.Set("", GetArenaForAllocation());
|
|
}
|
|
#endif // PROTOBUF_FORCE_COPY_DEFAULT_STRING
|
|
// @@protoc_insertion_point(field_set_allocated:xla.LiteralProto.u4s)
|
|
}
|
|
|
|
// bytes s8s = 15;
|
|
inline void LiteralProto::clear_s8s() {
|
|
_impl_.s8s_.ClearToEmpty();
|
|
}
|
|
inline const std::string& LiteralProto::s8s() const {
|
|
// @@protoc_insertion_point(field_get:xla.LiteralProto.s8s)
|
|
return _internal_s8s();
|
|
}
|
|
template <typename ArgT0, typename... ArgT>
|
|
inline PROTOBUF_ALWAYS_INLINE
|
|
void LiteralProto::set_s8s(ArgT0&& arg0, ArgT... args) {
|
|
|
|
_impl_.s8s_.SetBytes(static_cast<ArgT0 &&>(arg0), args..., GetArenaForAllocation());
|
|
// @@protoc_insertion_point(field_set:xla.LiteralProto.s8s)
|
|
}
|
|
inline std::string* LiteralProto::mutable_s8s() {
|
|
std::string* _s = _internal_mutable_s8s();
|
|
// @@protoc_insertion_point(field_mutable:xla.LiteralProto.s8s)
|
|
return _s;
|
|
}
|
|
inline const std::string& LiteralProto::_internal_s8s() const {
|
|
return _impl_.s8s_.Get();
|
|
}
|
|
inline void LiteralProto::_internal_set_s8s(const std::string& value) {
|
|
|
|
_impl_.s8s_.Set(value, GetArenaForAllocation());
|
|
}
|
|
inline std::string* LiteralProto::_internal_mutable_s8s() {
|
|
|
|
return _impl_.s8s_.Mutable(GetArenaForAllocation());
|
|
}
|
|
inline std::string* LiteralProto::release_s8s() {
|
|
// @@protoc_insertion_point(field_release:xla.LiteralProto.s8s)
|
|
return _impl_.s8s_.Release();
|
|
}
|
|
inline void LiteralProto::set_allocated_s8s(std::string* s8s) {
|
|
if (s8s != nullptr) {
|
|
|
|
} else {
|
|
|
|
}
|
|
_impl_.s8s_.SetAllocated(s8s, GetArenaForAllocation());
|
|
#ifdef PROTOBUF_FORCE_COPY_DEFAULT_STRING
|
|
if (_impl_.s8s_.IsDefault()) {
|
|
_impl_.s8s_.Set("", GetArenaForAllocation());
|
|
}
|
|
#endif // PROTOBUF_FORCE_COPY_DEFAULT_STRING
|
|
// @@protoc_insertion_point(field_set_allocated:xla.LiteralProto.s8s)
|
|
}
|
|
|
|
// bytes u8s = 3;
|
|
inline void LiteralProto::clear_u8s() {
|
|
_impl_.u8s_.ClearToEmpty();
|
|
}
|
|
inline const std::string& LiteralProto::u8s() const {
|
|
// @@protoc_insertion_point(field_get:xla.LiteralProto.u8s)
|
|
return _internal_u8s();
|
|
}
|
|
template <typename ArgT0, typename... ArgT>
|
|
inline PROTOBUF_ALWAYS_INLINE
|
|
void LiteralProto::set_u8s(ArgT0&& arg0, ArgT... args) {
|
|
|
|
_impl_.u8s_.SetBytes(static_cast<ArgT0 &&>(arg0), args..., GetArenaForAllocation());
|
|
// @@protoc_insertion_point(field_set:xla.LiteralProto.u8s)
|
|
}
|
|
inline std::string* LiteralProto::mutable_u8s() {
|
|
std::string* _s = _internal_mutable_u8s();
|
|
// @@protoc_insertion_point(field_mutable:xla.LiteralProto.u8s)
|
|
return _s;
|
|
}
|
|
inline const std::string& LiteralProto::_internal_u8s() const {
|
|
return _impl_.u8s_.Get();
|
|
}
|
|
inline void LiteralProto::_internal_set_u8s(const std::string& value) {
|
|
|
|
_impl_.u8s_.Set(value, GetArenaForAllocation());
|
|
}
|
|
inline std::string* LiteralProto::_internal_mutable_u8s() {
|
|
|
|
return _impl_.u8s_.Mutable(GetArenaForAllocation());
|
|
}
|
|
inline std::string* LiteralProto::release_u8s() {
|
|
// @@protoc_insertion_point(field_release:xla.LiteralProto.u8s)
|
|
return _impl_.u8s_.Release();
|
|
}
|
|
inline void LiteralProto::set_allocated_u8s(std::string* u8s) {
|
|
if (u8s != nullptr) {
|
|
|
|
} else {
|
|
|
|
}
|
|
_impl_.u8s_.SetAllocated(u8s, GetArenaForAllocation());
|
|
#ifdef PROTOBUF_FORCE_COPY_DEFAULT_STRING
|
|
if (_impl_.u8s_.IsDefault()) {
|
|
_impl_.u8s_.Set("", GetArenaForAllocation());
|
|
}
|
|
#endif // PROTOBUF_FORCE_COPY_DEFAULT_STRING
|
|
// @@protoc_insertion_point(field_set_allocated:xla.LiteralProto.u8s)
|
|
}
|
|
|
|
// repeated int32 s32s = 4;
|
|
inline int LiteralProto::_internal_s32s_size() const {
|
|
return _impl_.s32s_.size();
|
|
}
|
|
inline int LiteralProto::s32s_size() const {
|
|
return _internal_s32s_size();
|
|
}
|
|
inline void LiteralProto::clear_s32s() {
|
|
_impl_.s32s_.Clear();
|
|
}
|
|
inline int32_t LiteralProto::_internal_s32s(int index) const {
|
|
return _impl_.s32s_.Get(index);
|
|
}
|
|
inline int32_t LiteralProto::s32s(int index) const {
|
|
// @@protoc_insertion_point(field_get:xla.LiteralProto.s32s)
|
|
return _internal_s32s(index);
|
|
}
|
|
inline void LiteralProto::set_s32s(int index, int32_t value) {
|
|
_impl_.s32s_.Set(index, value);
|
|
// @@protoc_insertion_point(field_set:xla.LiteralProto.s32s)
|
|
}
|
|
inline void LiteralProto::_internal_add_s32s(int32_t value) {
|
|
_impl_.s32s_.Add(value);
|
|
}
|
|
inline void LiteralProto::add_s32s(int32_t value) {
|
|
_internal_add_s32s(value);
|
|
// @@protoc_insertion_point(field_add:xla.LiteralProto.s32s)
|
|
}
|
|
inline const ::PROTOBUF_NAMESPACE_ID::RepeatedField< int32_t >&
|
|
LiteralProto::_internal_s32s() const {
|
|
return _impl_.s32s_;
|
|
}
|
|
inline const ::PROTOBUF_NAMESPACE_ID::RepeatedField< int32_t >&
|
|
LiteralProto::s32s() const {
|
|
// @@protoc_insertion_point(field_list:xla.LiteralProto.s32s)
|
|
return _internal_s32s();
|
|
}
|
|
inline ::PROTOBUF_NAMESPACE_ID::RepeatedField< int32_t >*
|
|
LiteralProto::_internal_mutable_s32s() {
|
|
return &_impl_.s32s_;
|
|
}
|
|
inline ::PROTOBUF_NAMESPACE_ID::RepeatedField< int32_t >*
|
|
LiteralProto::mutable_s32s() {
|
|
// @@protoc_insertion_point(field_mutable_list:xla.LiteralProto.s32s)
|
|
return _internal_mutable_s32s();
|
|
}
|
|
|
|
// repeated int64 s64s = 5;
|
|
inline int LiteralProto::_internal_s64s_size() const {
|
|
return _impl_.s64s_.size();
|
|
}
|
|
inline int LiteralProto::s64s_size() const {
|
|
return _internal_s64s_size();
|
|
}
|
|
inline void LiteralProto::clear_s64s() {
|
|
_impl_.s64s_.Clear();
|
|
}
|
|
inline int64_t LiteralProto::_internal_s64s(int index) const {
|
|
return _impl_.s64s_.Get(index);
|
|
}
|
|
inline int64_t LiteralProto::s64s(int index) const {
|
|
// @@protoc_insertion_point(field_get:xla.LiteralProto.s64s)
|
|
return _internal_s64s(index);
|
|
}
|
|
inline void LiteralProto::set_s64s(int index, int64_t value) {
|
|
_impl_.s64s_.Set(index, value);
|
|
// @@protoc_insertion_point(field_set:xla.LiteralProto.s64s)
|
|
}
|
|
inline void LiteralProto::_internal_add_s64s(int64_t value) {
|
|
_impl_.s64s_.Add(value);
|
|
}
|
|
inline void LiteralProto::add_s64s(int64_t value) {
|
|
_internal_add_s64s(value);
|
|
// @@protoc_insertion_point(field_add:xla.LiteralProto.s64s)
|
|
}
|
|
inline const ::PROTOBUF_NAMESPACE_ID::RepeatedField< int64_t >&
|
|
LiteralProto::_internal_s64s() const {
|
|
return _impl_.s64s_;
|
|
}
|
|
inline const ::PROTOBUF_NAMESPACE_ID::RepeatedField< int64_t >&
|
|
LiteralProto::s64s() const {
|
|
// @@protoc_insertion_point(field_list:xla.LiteralProto.s64s)
|
|
return _internal_s64s();
|
|
}
|
|
inline ::PROTOBUF_NAMESPACE_ID::RepeatedField< int64_t >*
|
|
LiteralProto::_internal_mutable_s64s() {
|
|
return &_impl_.s64s_;
|
|
}
|
|
inline ::PROTOBUF_NAMESPACE_ID::RepeatedField< int64_t >*
|
|
LiteralProto::mutable_s64s() {
|
|
// @@protoc_insertion_point(field_mutable_list:xla.LiteralProto.s64s)
|
|
return _internal_mutable_s64s();
|
|
}
|
|
|
|
// repeated uint32 u32s = 6;
|
|
inline int LiteralProto::_internal_u32s_size() const {
|
|
return _impl_.u32s_.size();
|
|
}
|
|
inline int LiteralProto::u32s_size() const {
|
|
return _internal_u32s_size();
|
|
}
|
|
inline void LiteralProto::clear_u32s() {
|
|
_impl_.u32s_.Clear();
|
|
}
|
|
inline uint32_t LiteralProto::_internal_u32s(int index) const {
|
|
return _impl_.u32s_.Get(index);
|
|
}
|
|
inline uint32_t LiteralProto::u32s(int index) const {
|
|
// @@protoc_insertion_point(field_get:xla.LiteralProto.u32s)
|
|
return _internal_u32s(index);
|
|
}
|
|
inline void LiteralProto::set_u32s(int index, uint32_t value) {
|
|
_impl_.u32s_.Set(index, value);
|
|
// @@protoc_insertion_point(field_set:xla.LiteralProto.u32s)
|
|
}
|
|
inline void LiteralProto::_internal_add_u32s(uint32_t value) {
|
|
_impl_.u32s_.Add(value);
|
|
}
|
|
inline void LiteralProto::add_u32s(uint32_t value) {
|
|
_internal_add_u32s(value);
|
|
// @@protoc_insertion_point(field_add:xla.LiteralProto.u32s)
|
|
}
|
|
inline const ::PROTOBUF_NAMESPACE_ID::RepeatedField< uint32_t >&
|
|
LiteralProto::_internal_u32s() const {
|
|
return _impl_.u32s_;
|
|
}
|
|
inline const ::PROTOBUF_NAMESPACE_ID::RepeatedField< uint32_t >&
|
|
LiteralProto::u32s() const {
|
|
// @@protoc_insertion_point(field_list:xla.LiteralProto.u32s)
|
|
return _internal_u32s();
|
|
}
|
|
inline ::PROTOBUF_NAMESPACE_ID::RepeatedField< uint32_t >*
|
|
LiteralProto::_internal_mutable_u32s() {
|
|
return &_impl_.u32s_;
|
|
}
|
|
inline ::PROTOBUF_NAMESPACE_ID::RepeatedField< uint32_t >*
|
|
LiteralProto::mutable_u32s() {
|
|
// @@protoc_insertion_point(field_mutable_list:xla.LiteralProto.u32s)
|
|
return _internal_mutable_u32s();
|
|
}
|
|
|
|
// repeated uint64 u64s = 7;
|
|
inline int LiteralProto::_internal_u64s_size() const {
|
|
return _impl_.u64s_.size();
|
|
}
|
|
inline int LiteralProto::u64s_size() const {
|
|
return _internal_u64s_size();
|
|
}
|
|
inline void LiteralProto::clear_u64s() {
|
|
_impl_.u64s_.Clear();
|
|
}
|
|
inline uint64_t LiteralProto::_internal_u64s(int index) const {
|
|
return _impl_.u64s_.Get(index);
|
|
}
|
|
inline uint64_t LiteralProto::u64s(int index) const {
|
|
// @@protoc_insertion_point(field_get:xla.LiteralProto.u64s)
|
|
return _internal_u64s(index);
|
|
}
|
|
inline void LiteralProto::set_u64s(int index, uint64_t value) {
|
|
_impl_.u64s_.Set(index, value);
|
|
// @@protoc_insertion_point(field_set:xla.LiteralProto.u64s)
|
|
}
|
|
inline void LiteralProto::_internal_add_u64s(uint64_t value) {
|
|
_impl_.u64s_.Add(value);
|
|
}
|
|
inline void LiteralProto::add_u64s(uint64_t value) {
|
|
_internal_add_u64s(value);
|
|
// @@protoc_insertion_point(field_add:xla.LiteralProto.u64s)
|
|
}
|
|
inline const ::PROTOBUF_NAMESPACE_ID::RepeatedField< uint64_t >&
|
|
LiteralProto::_internal_u64s() const {
|
|
return _impl_.u64s_;
|
|
}
|
|
inline const ::PROTOBUF_NAMESPACE_ID::RepeatedField< uint64_t >&
|
|
LiteralProto::u64s() const {
|
|
// @@protoc_insertion_point(field_list:xla.LiteralProto.u64s)
|
|
return _internal_u64s();
|
|
}
|
|
inline ::PROTOBUF_NAMESPACE_ID::RepeatedField< uint64_t >*
|
|
LiteralProto::_internal_mutable_u64s() {
|
|
return &_impl_.u64s_;
|
|
}
|
|
inline ::PROTOBUF_NAMESPACE_ID::RepeatedField< uint64_t >*
|
|
LiteralProto::mutable_u64s() {
|
|
// @@protoc_insertion_point(field_mutable_list:xla.LiteralProto.u64s)
|
|
return _internal_mutable_u64s();
|
|
}
|
|
|
|
// repeated float f32s = 8;
|
|
inline int LiteralProto::_internal_f32s_size() const {
|
|
return _impl_.f32s_.size();
|
|
}
|
|
inline int LiteralProto::f32s_size() const {
|
|
return _internal_f32s_size();
|
|
}
|
|
inline void LiteralProto::clear_f32s() {
|
|
_impl_.f32s_.Clear();
|
|
}
|
|
inline float LiteralProto::_internal_f32s(int index) const {
|
|
return _impl_.f32s_.Get(index);
|
|
}
|
|
inline float LiteralProto::f32s(int index) const {
|
|
// @@protoc_insertion_point(field_get:xla.LiteralProto.f32s)
|
|
return _internal_f32s(index);
|
|
}
|
|
inline void LiteralProto::set_f32s(int index, float value) {
|
|
_impl_.f32s_.Set(index, value);
|
|
// @@protoc_insertion_point(field_set:xla.LiteralProto.f32s)
|
|
}
|
|
inline void LiteralProto::_internal_add_f32s(float value) {
|
|
_impl_.f32s_.Add(value);
|
|
}
|
|
inline void LiteralProto::add_f32s(float value) {
|
|
_internal_add_f32s(value);
|
|
// @@protoc_insertion_point(field_add:xla.LiteralProto.f32s)
|
|
}
|
|
inline const ::PROTOBUF_NAMESPACE_ID::RepeatedField< float >&
|
|
LiteralProto::_internal_f32s() const {
|
|
return _impl_.f32s_;
|
|
}
|
|
inline const ::PROTOBUF_NAMESPACE_ID::RepeatedField< float >&
|
|
LiteralProto::f32s() const {
|
|
// @@protoc_insertion_point(field_list:xla.LiteralProto.f32s)
|
|
return _internal_f32s();
|
|
}
|
|
inline ::PROTOBUF_NAMESPACE_ID::RepeatedField< float >*
|
|
LiteralProto::_internal_mutable_f32s() {
|
|
return &_impl_.f32s_;
|
|
}
|
|
inline ::PROTOBUF_NAMESPACE_ID::RepeatedField< float >*
|
|
LiteralProto::mutable_f32s() {
|
|
// @@protoc_insertion_point(field_mutable_list:xla.LiteralProto.f32s)
|
|
return _internal_mutable_f32s();
|
|
}
|
|
|
|
// repeated double f64s = 9;
|
|
inline int LiteralProto::_internal_f64s_size() const {
|
|
return _impl_.f64s_.size();
|
|
}
|
|
inline int LiteralProto::f64s_size() const {
|
|
return _internal_f64s_size();
|
|
}
|
|
inline void LiteralProto::clear_f64s() {
|
|
_impl_.f64s_.Clear();
|
|
}
|
|
inline double LiteralProto::_internal_f64s(int index) const {
|
|
return _impl_.f64s_.Get(index);
|
|
}
|
|
inline double LiteralProto::f64s(int index) const {
|
|
// @@protoc_insertion_point(field_get:xla.LiteralProto.f64s)
|
|
return _internal_f64s(index);
|
|
}
|
|
inline void LiteralProto::set_f64s(int index, double value) {
|
|
_impl_.f64s_.Set(index, value);
|
|
// @@protoc_insertion_point(field_set:xla.LiteralProto.f64s)
|
|
}
|
|
inline void LiteralProto::_internal_add_f64s(double value) {
|
|
_impl_.f64s_.Add(value);
|
|
}
|
|
inline void LiteralProto::add_f64s(double value) {
|
|
_internal_add_f64s(value);
|
|
// @@protoc_insertion_point(field_add:xla.LiteralProto.f64s)
|
|
}
|
|
inline const ::PROTOBUF_NAMESPACE_ID::RepeatedField< double >&
|
|
LiteralProto::_internal_f64s() const {
|
|
return _impl_.f64s_;
|
|
}
|
|
inline const ::PROTOBUF_NAMESPACE_ID::RepeatedField< double >&
|
|
LiteralProto::f64s() const {
|
|
// @@protoc_insertion_point(field_list:xla.LiteralProto.f64s)
|
|
return _internal_f64s();
|
|
}
|
|
inline ::PROTOBUF_NAMESPACE_ID::RepeatedField< double >*
|
|
LiteralProto::_internal_mutable_f64s() {
|
|
return &_impl_.f64s_;
|
|
}
|
|
inline ::PROTOBUF_NAMESPACE_ID::RepeatedField< double >*
|
|
LiteralProto::mutable_f64s() {
|
|
// @@protoc_insertion_point(field_mutable_list:xla.LiteralProto.f64s)
|
|
return _internal_mutable_f64s();
|
|
}
|
|
|
|
// repeated float c64s = 12;
|
|
inline int LiteralProto::_internal_c64s_size() const {
|
|
return _impl_.c64s_.size();
|
|
}
|
|
inline int LiteralProto::c64s_size() const {
|
|
return _internal_c64s_size();
|
|
}
|
|
inline void LiteralProto::clear_c64s() {
|
|
_impl_.c64s_.Clear();
|
|
}
|
|
inline float LiteralProto::_internal_c64s(int index) const {
|
|
return _impl_.c64s_.Get(index);
|
|
}
|
|
inline float LiteralProto::c64s(int index) const {
|
|
// @@protoc_insertion_point(field_get:xla.LiteralProto.c64s)
|
|
return _internal_c64s(index);
|
|
}
|
|
inline void LiteralProto::set_c64s(int index, float value) {
|
|
_impl_.c64s_.Set(index, value);
|
|
// @@protoc_insertion_point(field_set:xla.LiteralProto.c64s)
|
|
}
|
|
inline void LiteralProto::_internal_add_c64s(float value) {
|
|
_impl_.c64s_.Add(value);
|
|
}
|
|
inline void LiteralProto::add_c64s(float value) {
|
|
_internal_add_c64s(value);
|
|
// @@protoc_insertion_point(field_add:xla.LiteralProto.c64s)
|
|
}
|
|
inline const ::PROTOBUF_NAMESPACE_ID::RepeatedField< float >&
|
|
LiteralProto::_internal_c64s() const {
|
|
return _impl_.c64s_;
|
|
}
|
|
inline const ::PROTOBUF_NAMESPACE_ID::RepeatedField< float >&
|
|
LiteralProto::c64s() const {
|
|
// @@protoc_insertion_point(field_list:xla.LiteralProto.c64s)
|
|
return _internal_c64s();
|
|
}
|
|
inline ::PROTOBUF_NAMESPACE_ID::RepeatedField< float >*
|
|
LiteralProto::_internal_mutable_c64s() {
|
|
return &_impl_.c64s_;
|
|
}
|
|
inline ::PROTOBUF_NAMESPACE_ID::RepeatedField< float >*
|
|
LiteralProto::mutable_c64s() {
|
|
// @@protoc_insertion_point(field_mutable_list:xla.LiteralProto.c64s)
|
|
return _internal_mutable_c64s();
|
|
}
|
|
|
|
// repeated double c128s = 18;
|
|
inline int LiteralProto::_internal_c128s_size() const {
|
|
return _impl_.c128s_.size();
|
|
}
|
|
inline int LiteralProto::c128s_size() const {
|
|
return _internal_c128s_size();
|
|
}
|
|
inline void LiteralProto::clear_c128s() {
|
|
_impl_.c128s_.Clear();
|
|
}
|
|
inline double LiteralProto::_internal_c128s(int index) const {
|
|
return _impl_.c128s_.Get(index);
|
|
}
|
|
inline double LiteralProto::c128s(int index) const {
|
|
// @@protoc_insertion_point(field_get:xla.LiteralProto.c128s)
|
|
return _internal_c128s(index);
|
|
}
|
|
inline void LiteralProto::set_c128s(int index, double value) {
|
|
_impl_.c128s_.Set(index, value);
|
|
// @@protoc_insertion_point(field_set:xla.LiteralProto.c128s)
|
|
}
|
|
inline void LiteralProto::_internal_add_c128s(double value) {
|
|
_impl_.c128s_.Add(value);
|
|
}
|
|
inline void LiteralProto::add_c128s(double value) {
|
|
_internal_add_c128s(value);
|
|
// @@protoc_insertion_point(field_add:xla.LiteralProto.c128s)
|
|
}
|
|
inline const ::PROTOBUF_NAMESPACE_ID::RepeatedField< double >&
|
|
LiteralProto::_internal_c128s() const {
|
|
return _impl_.c128s_;
|
|
}
|
|
inline const ::PROTOBUF_NAMESPACE_ID::RepeatedField< double >&
|
|
LiteralProto::c128s() const {
|
|
// @@protoc_insertion_point(field_list:xla.LiteralProto.c128s)
|
|
return _internal_c128s();
|
|
}
|
|
inline ::PROTOBUF_NAMESPACE_ID::RepeatedField< double >*
|
|
LiteralProto::_internal_mutable_c128s() {
|
|
return &_impl_.c128s_;
|
|
}
|
|
inline ::PROTOBUF_NAMESPACE_ID::RepeatedField< double >*
|
|
LiteralProto::mutable_c128s() {
|
|
// @@protoc_insertion_point(field_mutable_list:xla.LiteralProto.c128s)
|
|
return _internal_mutable_c128s();
|
|
}
|
|
|
|
// repeated .xla.LiteralProto tuple_literals = 10;
|
|
inline int LiteralProto::_internal_tuple_literals_size() const {
|
|
return _impl_.tuple_literals_.size();
|
|
}
|
|
inline int LiteralProto::tuple_literals_size() const {
|
|
return _internal_tuple_literals_size();
|
|
}
|
|
inline void LiteralProto::clear_tuple_literals() {
|
|
_impl_.tuple_literals_.Clear();
|
|
}
|
|
inline ::xla::LiteralProto* LiteralProto::mutable_tuple_literals(int index) {
|
|
// @@protoc_insertion_point(field_mutable:xla.LiteralProto.tuple_literals)
|
|
return _impl_.tuple_literals_.Mutable(index);
|
|
}
|
|
inline ::PROTOBUF_NAMESPACE_ID::RepeatedPtrField< ::xla::LiteralProto >*
|
|
LiteralProto::mutable_tuple_literals() {
|
|
// @@protoc_insertion_point(field_mutable_list:xla.LiteralProto.tuple_literals)
|
|
return &_impl_.tuple_literals_;
|
|
}
|
|
inline const ::xla::LiteralProto& LiteralProto::_internal_tuple_literals(int index) const {
|
|
return _impl_.tuple_literals_.Get(index);
|
|
}
|
|
inline const ::xla::LiteralProto& LiteralProto::tuple_literals(int index) const {
|
|
// @@protoc_insertion_point(field_get:xla.LiteralProto.tuple_literals)
|
|
return _internal_tuple_literals(index);
|
|
}
|
|
inline ::xla::LiteralProto* LiteralProto::_internal_add_tuple_literals() {
|
|
return _impl_.tuple_literals_.Add();
|
|
}
|
|
inline ::xla::LiteralProto* LiteralProto::add_tuple_literals() {
|
|
::xla::LiteralProto* _add = _internal_add_tuple_literals();
|
|
// @@protoc_insertion_point(field_add:xla.LiteralProto.tuple_literals)
|
|
return _add;
|
|
}
|
|
inline const ::PROTOBUF_NAMESPACE_ID::RepeatedPtrField< ::xla::LiteralProto >&
|
|
LiteralProto::tuple_literals() const {
|
|
// @@protoc_insertion_point(field_list:xla.LiteralProto.tuple_literals)
|
|
return _impl_.tuple_literals_;
|
|
}
|
|
|
|
// bytes f16s = 11;
|
|
inline void LiteralProto::clear_f16s() {
|
|
_impl_.f16s_.ClearToEmpty();
|
|
}
|
|
inline const std::string& LiteralProto::f16s() const {
|
|
// @@protoc_insertion_point(field_get:xla.LiteralProto.f16s)
|
|
return _internal_f16s();
|
|
}
|
|
template <typename ArgT0, typename... ArgT>
|
|
inline PROTOBUF_ALWAYS_INLINE
|
|
void LiteralProto::set_f16s(ArgT0&& arg0, ArgT... args) {
|
|
|
|
_impl_.f16s_.SetBytes(static_cast<ArgT0 &&>(arg0), args..., GetArenaForAllocation());
|
|
// @@protoc_insertion_point(field_set:xla.LiteralProto.f16s)
|
|
}
|
|
inline std::string* LiteralProto::mutable_f16s() {
|
|
std::string* _s = _internal_mutable_f16s();
|
|
// @@protoc_insertion_point(field_mutable:xla.LiteralProto.f16s)
|
|
return _s;
|
|
}
|
|
inline const std::string& LiteralProto::_internal_f16s() const {
|
|
return _impl_.f16s_.Get();
|
|
}
|
|
inline void LiteralProto::_internal_set_f16s(const std::string& value) {
|
|
|
|
_impl_.f16s_.Set(value, GetArenaForAllocation());
|
|
}
|
|
inline std::string* LiteralProto::_internal_mutable_f16s() {
|
|
|
|
return _impl_.f16s_.Mutable(GetArenaForAllocation());
|
|
}
|
|
inline std::string* LiteralProto::release_f16s() {
|
|
// @@protoc_insertion_point(field_release:xla.LiteralProto.f16s)
|
|
return _impl_.f16s_.Release();
|
|
}
|
|
inline void LiteralProto::set_allocated_f16s(std::string* f16s) {
|
|
if (f16s != nullptr) {
|
|
|
|
} else {
|
|
|
|
}
|
|
_impl_.f16s_.SetAllocated(f16s, GetArenaForAllocation());
|
|
#ifdef PROTOBUF_FORCE_COPY_DEFAULT_STRING
|
|
if (_impl_.f16s_.IsDefault()) {
|
|
_impl_.f16s_.Set("", GetArenaForAllocation());
|
|
}
|
|
#endif // PROTOBUF_FORCE_COPY_DEFAULT_STRING
|
|
// @@protoc_insertion_point(field_set_allocated:xla.LiteralProto.f16s)
|
|
}
|
|
|
|
// bytes bf16s = 13;
|
|
inline void LiteralProto::clear_bf16s() {
|
|
_impl_.bf16s_.ClearToEmpty();
|
|
}
|
|
inline const std::string& LiteralProto::bf16s() const {
|
|
// @@protoc_insertion_point(field_get:xla.LiteralProto.bf16s)
|
|
return _internal_bf16s();
|
|
}
|
|
template <typename ArgT0, typename... ArgT>
|
|
inline PROTOBUF_ALWAYS_INLINE
|
|
void LiteralProto::set_bf16s(ArgT0&& arg0, ArgT... args) {
|
|
|
|
_impl_.bf16s_.SetBytes(static_cast<ArgT0 &&>(arg0), args..., GetArenaForAllocation());
|
|
// @@protoc_insertion_point(field_set:xla.LiteralProto.bf16s)
|
|
}
|
|
inline std::string* LiteralProto::mutable_bf16s() {
|
|
std::string* _s = _internal_mutable_bf16s();
|
|
// @@protoc_insertion_point(field_mutable:xla.LiteralProto.bf16s)
|
|
return _s;
|
|
}
|
|
inline const std::string& LiteralProto::_internal_bf16s() const {
|
|
return _impl_.bf16s_.Get();
|
|
}
|
|
inline void LiteralProto::_internal_set_bf16s(const std::string& value) {
|
|
|
|
_impl_.bf16s_.Set(value, GetArenaForAllocation());
|
|
}
|
|
inline std::string* LiteralProto::_internal_mutable_bf16s() {
|
|
|
|
return _impl_.bf16s_.Mutable(GetArenaForAllocation());
|
|
}
|
|
inline std::string* LiteralProto::release_bf16s() {
|
|
// @@protoc_insertion_point(field_release:xla.LiteralProto.bf16s)
|
|
return _impl_.bf16s_.Release();
|
|
}
|
|
inline void LiteralProto::set_allocated_bf16s(std::string* bf16s) {
|
|
if (bf16s != nullptr) {
|
|
|
|
} else {
|
|
|
|
}
|
|
_impl_.bf16s_.SetAllocated(bf16s, GetArenaForAllocation());
|
|
#ifdef PROTOBUF_FORCE_COPY_DEFAULT_STRING
|
|
if (_impl_.bf16s_.IsDefault()) {
|
|
_impl_.bf16s_.Set("", GetArenaForAllocation());
|
|
}
|
|
#endif // PROTOBUF_FORCE_COPY_DEFAULT_STRING
|
|
// @@protoc_insertion_point(field_set_allocated:xla.LiteralProto.bf16s)
|
|
}
|
|
|
|
// bytes u16s = 16;
|
|
inline void LiteralProto::clear_u16s() {
|
|
_impl_.u16s_.ClearToEmpty();
|
|
}
|
|
inline const std::string& LiteralProto::u16s() const {
|
|
// @@protoc_insertion_point(field_get:xla.LiteralProto.u16s)
|
|
return _internal_u16s();
|
|
}
|
|
template <typename ArgT0, typename... ArgT>
|
|
inline PROTOBUF_ALWAYS_INLINE
|
|
void LiteralProto::set_u16s(ArgT0&& arg0, ArgT... args) {
|
|
|
|
_impl_.u16s_.SetBytes(static_cast<ArgT0 &&>(arg0), args..., GetArenaForAllocation());
|
|
// @@protoc_insertion_point(field_set:xla.LiteralProto.u16s)
|
|
}
|
|
inline std::string* LiteralProto::mutable_u16s() {
|
|
std::string* _s = _internal_mutable_u16s();
|
|
// @@protoc_insertion_point(field_mutable:xla.LiteralProto.u16s)
|
|
return _s;
|
|
}
|
|
inline const std::string& LiteralProto::_internal_u16s() const {
|
|
return _impl_.u16s_.Get();
|
|
}
|
|
inline void LiteralProto::_internal_set_u16s(const std::string& value) {
|
|
|
|
_impl_.u16s_.Set(value, GetArenaForAllocation());
|
|
}
|
|
inline std::string* LiteralProto::_internal_mutable_u16s() {
|
|
|
|
return _impl_.u16s_.Mutable(GetArenaForAllocation());
|
|
}
|
|
inline std::string* LiteralProto::release_u16s() {
|
|
// @@protoc_insertion_point(field_release:xla.LiteralProto.u16s)
|
|
return _impl_.u16s_.Release();
|
|
}
|
|
inline void LiteralProto::set_allocated_u16s(std::string* u16s) {
|
|
if (u16s != nullptr) {
|
|
|
|
} else {
|
|
|
|
}
|
|
_impl_.u16s_.SetAllocated(u16s, GetArenaForAllocation());
|
|
#ifdef PROTOBUF_FORCE_COPY_DEFAULT_STRING
|
|
if (_impl_.u16s_.IsDefault()) {
|
|
_impl_.u16s_.Set("", GetArenaForAllocation());
|
|
}
|
|
#endif // PROTOBUF_FORCE_COPY_DEFAULT_STRING
|
|
// @@protoc_insertion_point(field_set_allocated:xla.LiteralProto.u16s)
|
|
}
|
|
|
|
// bytes s16s = 17;
|
|
inline void LiteralProto::clear_s16s() {
|
|
_impl_.s16s_.ClearToEmpty();
|
|
}
|
|
inline const std::string& LiteralProto::s16s() const {
|
|
// @@protoc_insertion_point(field_get:xla.LiteralProto.s16s)
|
|
return _internal_s16s();
|
|
}
|
|
template <typename ArgT0, typename... ArgT>
|
|
inline PROTOBUF_ALWAYS_INLINE
|
|
void LiteralProto::set_s16s(ArgT0&& arg0, ArgT... args) {
|
|
|
|
_impl_.s16s_.SetBytes(static_cast<ArgT0 &&>(arg0), args..., GetArenaForAllocation());
|
|
// @@protoc_insertion_point(field_set:xla.LiteralProto.s16s)
|
|
}
|
|
inline std::string* LiteralProto::mutable_s16s() {
|
|
std::string* _s = _internal_mutable_s16s();
|
|
// @@protoc_insertion_point(field_mutable:xla.LiteralProto.s16s)
|
|
return _s;
|
|
}
|
|
inline const std::string& LiteralProto::_internal_s16s() const {
|
|
return _impl_.s16s_.Get();
|
|
}
|
|
inline void LiteralProto::_internal_set_s16s(const std::string& value) {
|
|
|
|
_impl_.s16s_.Set(value, GetArenaForAllocation());
|
|
}
|
|
inline std::string* LiteralProto::_internal_mutable_s16s() {
|
|
|
|
return _impl_.s16s_.Mutable(GetArenaForAllocation());
|
|
}
|
|
inline std::string* LiteralProto::release_s16s() {
|
|
// @@protoc_insertion_point(field_release:xla.LiteralProto.s16s)
|
|
return _impl_.s16s_.Release();
|
|
}
|
|
inline void LiteralProto::set_allocated_s16s(std::string* s16s) {
|
|
if (s16s != nullptr) {
|
|
|
|
} else {
|
|
|
|
}
|
|
_impl_.s16s_.SetAllocated(s16s, GetArenaForAllocation());
|
|
#ifdef PROTOBUF_FORCE_COPY_DEFAULT_STRING
|
|
if (_impl_.s16s_.IsDefault()) {
|
|
_impl_.s16s_.Set("", GetArenaForAllocation());
|
|
}
|
|
#endif // PROTOBUF_FORCE_COPY_DEFAULT_STRING
|
|
// @@protoc_insertion_point(field_set_allocated:xla.LiteralProto.s16s)
|
|
}
|
|
|
|
// bytes f8e5m2s = 19;
|
|
inline void LiteralProto::clear_f8e5m2s() {
|
|
_impl_.f8e5m2s_.ClearToEmpty();
|
|
}
|
|
inline const std::string& LiteralProto::f8e5m2s() const {
|
|
// @@protoc_insertion_point(field_get:xla.LiteralProto.f8e5m2s)
|
|
return _internal_f8e5m2s();
|
|
}
|
|
template <typename ArgT0, typename... ArgT>
|
|
inline PROTOBUF_ALWAYS_INLINE
|
|
void LiteralProto::set_f8e5m2s(ArgT0&& arg0, ArgT... args) {
|
|
|
|
_impl_.f8e5m2s_.SetBytes(static_cast<ArgT0 &&>(arg0), args..., GetArenaForAllocation());
|
|
// @@protoc_insertion_point(field_set:xla.LiteralProto.f8e5m2s)
|
|
}
|
|
inline std::string* LiteralProto::mutable_f8e5m2s() {
|
|
std::string* _s = _internal_mutable_f8e5m2s();
|
|
// @@protoc_insertion_point(field_mutable:xla.LiteralProto.f8e5m2s)
|
|
return _s;
|
|
}
|
|
inline const std::string& LiteralProto::_internal_f8e5m2s() const {
|
|
return _impl_.f8e5m2s_.Get();
|
|
}
|
|
inline void LiteralProto::_internal_set_f8e5m2s(const std::string& value) {
|
|
|
|
_impl_.f8e5m2s_.Set(value, GetArenaForAllocation());
|
|
}
|
|
inline std::string* LiteralProto::_internal_mutable_f8e5m2s() {
|
|
|
|
return _impl_.f8e5m2s_.Mutable(GetArenaForAllocation());
|
|
}
|
|
inline std::string* LiteralProto::release_f8e5m2s() {
|
|
// @@protoc_insertion_point(field_release:xla.LiteralProto.f8e5m2s)
|
|
return _impl_.f8e5m2s_.Release();
|
|
}
|
|
inline void LiteralProto::set_allocated_f8e5m2s(std::string* f8e5m2s) {
|
|
if (f8e5m2s != nullptr) {
|
|
|
|
} else {
|
|
|
|
}
|
|
_impl_.f8e5m2s_.SetAllocated(f8e5m2s, GetArenaForAllocation());
|
|
#ifdef PROTOBUF_FORCE_COPY_DEFAULT_STRING
|
|
if (_impl_.f8e5m2s_.IsDefault()) {
|
|
_impl_.f8e5m2s_.Set("", GetArenaForAllocation());
|
|
}
|
|
#endif // PROTOBUF_FORCE_COPY_DEFAULT_STRING
|
|
// @@protoc_insertion_point(field_set_allocated:xla.LiteralProto.f8e5m2s)
|
|
}
|
|
|
|
// bytes f8e4m3fns = 20;
|
|
inline void LiteralProto::clear_f8e4m3fns() {
|
|
_impl_.f8e4m3fns_.ClearToEmpty();
|
|
}
|
|
inline const std::string& LiteralProto::f8e4m3fns() const {
|
|
// @@protoc_insertion_point(field_get:xla.LiteralProto.f8e4m3fns)
|
|
return _internal_f8e4m3fns();
|
|
}
|
|
template <typename ArgT0, typename... ArgT>
|
|
inline PROTOBUF_ALWAYS_INLINE
|
|
void LiteralProto::set_f8e4m3fns(ArgT0&& arg0, ArgT... args) {
|
|
|
|
_impl_.f8e4m3fns_.SetBytes(static_cast<ArgT0 &&>(arg0), args..., GetArenaForAllocation());
|
|
// @@protoc_insertion_point(field_set:xla.LiteralProto.f8e4m3fns)
|
|
}
|
|
inline std::string* LiteralProto::mutable_f8e4m3fns() {
|
|
std::string* _s = _internal_mutable_f8e4m3fns();
|
|
// @@protoc_insertion_point(field_mutable:xla.LiteralProto.f8e4m3fns)
|
|
return _s;
|
|
}
|
|
inline const std::string& LiteralProto::_internal_f8e4m3fns() const {
|
|
return _impl_.f8e4m3fns_.Get();
|
|
}
|
|
inline void LiteralProto::_internal_set_f8e4m3fns(const std::string& value) {
|
|
|
|
_impl_.f8e4m3fns_.Set(value, GetArenaForAllocation());
|
|
}
|
|
inline std::string* LiteralProto::_internal_mutable_f8e4m3fns() {
|
|
|
|
return _impl_.f8e4m3fns_.Mutable(GetArenaForAllocation());
|
|
}
|
|
inline std::string* LiteralProto::release_f8e4m3fns() {
|
|
// @@protoc_insertion_point(field_release:xla.LiteralProto.f8e4m3fns)
|
|
return _impl_.f8e4m3fns_.Release();
|
|
}
|
|
inline void LiteralProto::set_allocated_f8e4m3fns(std::string* f8e4m3fns) {
|
|
if (f8e4m3fns != nullptr) {
|
|
|
|
} else {
|
|
|
|
}
|
|
_impl_.f8e4m3fns_.SetAllocated(f8e4m3fns, GetArenaForAllocation());
|
|
#ifdef PROTOBUF_FORCE_COPY_DEFAULT_STRING
|
|
if (_impl_.f8e4m3fns_.IsDefault()) {
|
|
_impl_.f8e4m3fns_.Set("", GetArenaForAllocation());
|
|
}
|
|
#endif // PROTOBUF_FORCE_COPY_DEFAULT_STRING
|
|
// @@protoc_insertion_point(field_set_allocated:xla.LiteralProto.f8e4m3fns)
|
|
}
|
|
|
|
// bytes f8e4m3b11fnuzs = 23;
|
|
inline void LiteralProto::clear_f8e4m3b11fnuzs() {
|
|
_impl_.f8e4m3b11fnuzs_.ClearToEmpty();
|
|
}
|
|
inline const std::string& LiteralProto::f8e4m3b11fnuzs() const {
|
|
// @@protoc_insertion_point(field_get:xla.LiteralProto.f8e4m3b11fnuzs)
|
|
return _internal_f8e4m3b11fnuzs();
|
|
}
|
|
template <typename ArgT0, typename... ArgT>
|
|
inline PROTOBUF_ALWAYS_INLINE
|
|
void LiteralProto::set_f8e4m3b11fnuzs(ArgT0&& arg0, ArgT... args) {
|
|
|
|
_impl_.f8e4m3b11fnuzs_.SetBytes(static_cast<ArgT0 &&>(arg0), args..., GetArenaForAllocation());
|
|
// @@protoc_insertion_point(field_set:xla.LiteralProto.f8e4m3b11fnuzs)
|
|
}
|
|
inline std::string* LiteralProto::mutable_f8e4m3b11fnuzs() {
|
|
std::string* _s = _internal_mutable_f8e4m3b11fnuzs();
|
|
// @@protoc_insertion_point(field_mutable:xla.LiteralProto.f8e4m3b11fnuzs)
|
|
return _s;
|
|
}
|
|
inline const std::string& LiteralProto::_internal_f8e4m3b11fnuzs() const {
|
|
return _impl_.f8e4m3b11fnuzs_.Get();
|
|
}
|
|
inline void LiteralProto::_internal_set_f8e4m3b11fnuzs(const std::string& value) {
|
|
|
|
_impl_.f8e4m3b11fnuzs_.Set(value, GetArenaForAllocation());
|
|
}
|
|
inline std::string* LiteralProto::_internal_mutable_f8e4m3b11fnuzs() {
|
|
|
|
return _impl_.f8e4m3b11fnuzs_.Mutable(GetArenaForAllocation());
|
|
}
|
|
inline std::string* LiteralProto::release_f8e4m3b11fnuzs() {
|
|
// @@protoc_insertion_point(field_release:xla.LiteralProto.f8e4m3b11fnuzs)
|
|
return _impl_.f8e4m3b11fnuzs_.Release();
|
|
}
|
|
inline void LiteralProto::set_allocated_f8e4m3b11fnuzs(std::string* f8e4m3b11fnuzs) {
|
|
if (f8e4m3b11fnuzs != nullptr) {
|
|
|
|
} else {
|
|
|
|
}
|
|
_impl_.f8e4m3b11fnuzs_.SetAllocated(f8e4m3b11fnuzs, GetArenaForAllocation());
|
|
#ifdef PROTOBUF_FORCE_COPY_DEFAULT_STRING
|
|
if (_impl_.f8e4m3b11fnuzs_.IsDefault()) {
|
|
_impl_.f8e4m3b11fnuzs_.Set("", GetArenaForAllocation());
|
|
}
|
|
#endif // PROTOBUF_FORCE_COPY_DEFAULT_STRING
|
|
// @@protoc_insertion_point(field_set_allocated:xla.LiteralProto.f8e4m3b11fnuzs)
|
|
}
|
|
|
|
// bytes f8e5m2fnuzs = 24;
|
|
inline void LiteralProto::clear_f8e5m2fnuzs() {
|
|
_impl_.f8e5m2fnuzs_.ClearToEmpty();
|
|
}
|
|
inline const std::string& LiteralProto::f8e5m2fnuzs() const {
|
|
// @@protoc_insertion_point(field_get:xla.LiteralProto.f8e5m2fnuzs)
|
|
return _internal_f8e5m2fnuzs();
|
|
}
|
|
template <typename ArgT0, typename... ArgT>
|
|
inline PROTOBUF_ALWAYS_INLINE
|
|
void LiteralProto::set_f8e5m2fnuzs(ArgT0&& arg0, ArgT... args) {
|
|
|
|
_impl_.f8e5m2fnuzs_.SetBytes(static_cast<ArgT0 &&>(arg0), args..., GetArenaForAllocation());
|
|
// @@protoc_insertion_point(field_set:xla.LiteralProto.f8e5m2fnuzs)
|
|
}
|
|
inline std::string* LiteralProto::mutable_f8e5m2fnuzs() {
|
|
std::string* _s = _internal_mutable_f8e5m2fnuzs();
|
|
// @@protoc_insertion_point(field_mutable:xla.LiteralProto.f8e5m2fnuzs)
|
|
return _s;
|
|
}
|
|
inline const std::string& LiteralProto::_internal_f8e5m2fnuzs() const {
|
|
return _impl_.f8e5m2fnuzs_.Get();
|
|
}
|
|
inline void LiteralProto::_internal_set_f8e5m2fnuzs(const std::string& value) {
|
|
|
|
_impl_.f8e5m2fnuzs_.Set(value, GetArenaForAllocation());
|
|
}
|
|
inline std::string* LiteralProto::_internal_mutable_f8e5m2fnuzs() {
|
|
|
|
return _impl_.f8e5m2fnuzs_.Mutable(GetArenaForAllocation());
|
|
}
|
|
inline std::string* LiteralProto::release_f8e5m2fnuzs() {
|
|
// @@protoc_insertion_point(field_release:xla.LiteralProto.f8e5m2fnuzs)
|
|
return _impl_.f8e5m2fnuzs_.Release();
|
|
}
|
|
inline void LiteralProto::set_allocated_f8e5m2fnuzs(std::string* f8e5m2fnuzs) {
|
|
if (f8e5m2fnuzs != nullptr) {
|
|
|
|
} else {
|
|
|
|
}
|
|
_impl_.f8e5m2fnuzs_.SetAllocated(f8e5m2fnuzs, GetArenaForAllocation());
|
|
#ifdef PROTOBUF_FORCE_COPY_DEFAULT_STRING
|
|
if (_impl_.f8e5m2fnuzs_.IsDefault()) {
|
|
_impl_.f8e5m2fnuzs_.Set("", GetArenaForAllocation());
|
|
}
|
|
#endif // PROTOBUF_FORCE_COPY_DEFAULT_STRING
|
|
// @@protoc_insertion_point(field_set_allocated:xla.LiteralProto.f8e5m2fnuzs)
|
|
}
|
|
|
|
// bytes f8e4m3fnuzs = 25;
|
|
inline void LiteralProto::clear_f8e4m3fnuzs() {
|
|
_impl_.f8e4m3fnuzs_.ClearToEmpty();
|
|
}
|
|
inline const std::string& LiteralProto::f8e4m3fnuzs() const {
|
|
// @@protoc_insertion_point(field_get:xla.LiteralProto.f8e4m3fnuzs)
|
|
return _internal_f8e4m3fnuzs();
|
|
}
|
|
template <typename ArgT0, typename... ArgT>
|
|
inline PROTOBUF_ALWAYS_INLINE
|
|
void LiteralProto::set_f8e4m3fnuzs(ArgT0&& arg0, ArgT... args) {
|
|
|
|
_impl_.f8e4m3fnuzs_.SetBytes(static_cast<ArgT0 &&>(arg0), args..., GetArenaForAllocation());
|
|
// @@protoc_insertion_point(field_set:xla.LiteralProto.f8e4m3fnuzs)
|
|
}
|
|
inline std::string* LiteralProto::mutable_f8e4m3fnuzs() {
|
|
std::string* _s = _internal_mutable_f8e4m3fnuzs();
|
|
// @@protoc_insertion_point(field_mutable:xla.LiteralProto.f8e4m3fnuzs)
|
|
return _s;
|
|
}
|
|
inline const std::string& LiteralProto::_internal_f8e4m3fnuzs() const {
|
|
return _impl_.f8e4m3fnuzs_.Get();
|
|
}
|
|
inline void LiteralProto::_internal_set_f8e4m3fnuzs(const std::string& value) {
|
|
|
|
_impl_.f8e4m3fnuzs_.Set(value, GetArenaForAllocation());
|
|
}
|
|
inline std::string* LiteralProto::_internal_mutable_f8e4m3fnuzs() {
|
|
|
|
return _impl_.f8e4m3fnuzs_.Mutable(GetArenaForAllocation());
|
|
}
|
|
inline std::string* LiteralProto::release_f8e4m3fnuzs() {
|
|
// @@protoc_insertion_point(field_release:xla.LiteralProto.f8e4m3fnuzs)
|
|
return _impl_.f8e4m3fnuzs_.Release();
|
|
}
|
|
inline void LiteralProto::set_allocated_f8e4m3fnuzs(std::string* f8e4m3fnuzs) {
|
|
if (f8e4m3fnuzs != nullptr) {
|
|
|
|
} else {
|
|
|
|
}
|
|
_impl_.f8e4m3fnuzs_.SetAllocated(f8e4m3fnuzs, GetArenaForAllocation());
|
|
#ifdef PROTOBUF_FORCE_COPY_DEFAULT_STRING
|
|
if (_impl_.f8e4m3fnuzs_.IsDefault()) {
|
|
_impl_.f8e4m3fnuzs_.Set("", GetArenaForAllocation());
|
|
}
|
|
#endif // PROTOBUF_FORCE_COPY_DEFAULT_STRING
|
|
// @@protoc_insertion_point(field_set_allocated:xla.LiteralProto.f8e4m3fnuzs)
|
|
}
|
|
|
|
// repeated int64 sparse_indices = 14;
|
|
inline int LiteralProto::_internal_sparse_indices_size() const {
|
|
return _impl_.sparse_indices_.size();
|
|
}
|
|
inline int LiteralProto::sparse_indices_size() const {
|
|
return _internal_sparse_indices_size();
|
|
}
|
|
inline void LiteralProto::clear_sparse_indices() {
|
|
_impl_.sparse_indices_.Clear();
|
|
}
|
|
inline int64_t LiteralProto::_internal_sparse_indices(int index) const {
|
|
return _impl_.sparse_indices_.Get(index);
|
|
}
|
|
inline int64_t LiteralProto::sparse_indices(int index) const {
|
|
// @@protoc_insertion_point(field_get:xla.LiteralProto.sparse_indices)
|
|
return _internal_sparse_indices(index);
|
|
}
|
|
inline void LiteralProto::set_sparse_indices(int index, int64_t value) {
|
|
_impl_.sparse_indices_.Set(index, value);
|
|
// @@protoc_insertion_point(field_set:xla.LiteralProto.sparse_indices)
|
|
}
|
|
inline void LiteralProto::_internal_add_sparse_indices(int64_t value) {
|
|
_impl_.sparse_indices_.Add(value);
|
|
}
|
|
inline void LiteralProto::add_sparse_indices(int64_t value) {
|
|
_internal_add_sparse_indices(value);
|
|
// @@protoc_insertion_point(field_add:xla.LiteralProto.sparse_indices)
|
|
}
|
|
inline const ::PROTOBUF_NAMESPACE_ID::RepeatedField< int64_t >&
|
|
LiteralProto::_internal_sparse_indices() const {
|
|
return _impl_.sparse_indices_;
|
|
}
|
|
inline const ::PROTOBUF_NAMESPACE_ID::RepeatedField< int64_t >&
|
|
LiteralProto::sparse_indices() const {
|
|
// @@protoc_insertion_point(field_list:xla.LiteralProto.sparse_indices)
|
|
return _internal_sparse_indices();
|
|
}
|
|
inline ::PROTOBUF_NAMESPACE_ID::RepeatedField< int64_t >*
|
|
LiteralProto::_internal_mutable_sparse_indices() {
|
|
return &_impl_.sparse_indices_;
|
|
}
|
|
inline ::PROTOBUF_NAMESPACE_ID::RepeatedField< int64_t >*
|
|
LiteralProto::mutable_sparse_indices() {
|
|
// @@protoc_insertion_point(field_mutable_list:xla.LiteralProto.sparse_indices)
|
|
return _internal_mutable_sparse_indices();
|
|
}
|
|
|
|
// -------------------------------------------------------------------
|
|
|
|
// WindowDimension
|
|
|
|
// int64 size = 1;
|
|
inline void WindowDimension::clear_size() {
|
|
_impl_.size_ = int64_t{0};
|
|
}
|
|
inline int64_t WindowDimension::_internal_size() const {
|
|
return _impl_.size_;
|
|
}
|
|
inline int64_t WindowDimension::size() const {
|
|
// @@protoc_insertion_point(field_get:xla.WindowDimension.size)
|
|
return _internal_size();
|
|
}
|
|
inline void WindowDimension::_internal_set_size(int64_t value) {
|
|
|
|
_impl_.size_ = value;
|
|
}
|
|
inline void WindowDimension::set_size(int64_t value) {
|
|
_internal_set_size(value);
|
|
// @@protoc_insertion_point(field_set:xla.WindowDimension.size)
|
|
}
|
|
|
|
// int64 stride = 2;
|
|
inline void WindowDimension::clear_stride() {
|
|
_impl_.stride_ = int64_t{0};
|
|
}
|
|
inline int64_t WindowDimension::_internal_stride() const {
|
|
return _impl_.stride_;
|
|
}
|
|
inline int64_t WindowDimension::stride() const {
|
|
// @@protoc_insertion_point(field_get:xla.WindowDimension.stride)
|
|
return _internal_stride();
|
|
}
|
|
inline void WindowDimension::_internal_set_stride(int64_t value) {
|
|
|
|
_impl_.stride_ = value;
|
|
}
|
|
inline void WindowDimension::set_stride(int64_t value) {
|
|
_internal_set_stride(value);
|
|
// @@protoc_insertion_point(field_set:xla.WindowDimension.stride)
|
|
}
|
|
|
|
// int64 padding_low = 3;
|
|
inline void WindowDimension::clear_padding_low() {
|
|
_impl_.padding_low_ = int64_t{0};
|
|
}
|
|
inline int64_t WindowDimension::_internal_padding_low() const {
|
|
return _impl_.padding_low_;
|
|
}
|
|
inline int64_t WindowDimension::padding_low() const {
|
|
// @@protoc_insertion_point(field_get:xla.WindowDimension.padding_low)
|
|
return _internal_padding_low();
|
|
}
|
|
inline void WindowDimension::_internal_set_padding_low(int64_t value) {
|
|
|
|
_impl_.padding_low_ = value;
|
|
}
|
|
inline void WindowDimension::set_padding_low(int64_t value) {
|
|
_internal_set_padding_low(value);
|
|
// @@protoc_insertion_point(field_set:xla.WindowDimension.padding_low)
|
|
}
|
|
|
|
// int64 padding_high = 4;
|
|
inline void WindowDimension::clear_padding_high() {
|
|
_impl_.padding_high_ = int64_t{0};
|
|
}
|
|
inline int64_t WindowDimension::_internal_padding_high() const {
|
|
return _impl_.padding_high_;
|
|
}
|
|
inline int64_t WindowDimension::padding_high() const {
|
|
// @@protoc_insertion_point(field_get:xla.WindowDimension.padding_high)
|
|
return _internal_padding_high();
|
|
}
|
|
inline void WindowDimension::_internal_set_padding_high(int64_t value) {
|
|
|
|
_impl_.padding_high_ = value;
|
|
}
|
|
inline void WindowDimension::set_padding_high(int64_t value) {
|
|
_internal_set_padding_high(value);
|
|
// @@protoc_insertion_point(field_set:xla.WindowDimension.padding_high)
|
|
}
|
|
|
|
// int64 window_dilation = 5;
|
|
inline void WindowDimension::clear_window_dilation() {
|
|
_impl_.window_dilation_ = int64_t{0};
|
|
}
|
|
inline int64_t WindowDimension::_internal_window_dilation() const {
|
|
return _impl_.window_dilation_;
|
|
}
|
|
inline int64_t WindowDimension::window_dilation() const {
|
|
// @@protoc_insertion_point(field_get:xla.WindowDimension.window_dilation)
|
|
return _internal_window_dilation();
|
|
}
|
|
inline void WindowDimension::_internal_set_window_dilation(int64_t value) {
|
|
|
|
_impl_.window_dilation_ = value;
|
|
}
|
|
inline void WindowDimension::set_window_dilation(int64_t value) {
|
|
_internal_set_window_dilation(value);
|
|
// @@protoc_insertion_point(field_set:xla.WindowDimension.window_dilation)
|
|
}
|
|
|
|
// int64 base_dilation = 6;
|
|
inline void WindowDimension::clear_base_dilation() {
|
|
_impl_.base_dilation_ = int64_t{0};
|
|
}
|
|
inline int64_t WindowDimension::_internal_base_dilation() const {
|
|
return _impl_.base_dilation_;
|
|
}
|
|
inline int64_t WindowDimension::base_dilation() const {
|
|
// @@protoc_insertion_point(field_get:xla.WindowDimension.base_dilation)
|
|
return _internal_base_dilation();
|
|
}
|
|
inline void WindowDimension::_internal_set_base_dilation(int64_t value) {
|
|
|
|
_impl_.base_dilation_ = value;
|
|
}
|
|
inline void WindowDimension::set_base_dilation(int64_t value) {
|
|
_internal_set_base_dilation(value);
|
|
// @@protoc_insertion_point(field_set:xla.WindowDimension.base_dilation)
|
|
}
|
|
|
|
// bool window_reversal = 7;
|
|
inline void WindowDimension::clear_window_reversal() {
|
|
_impl_.window_reversal_ = false;
|
|
}
|
|
inline bool WindowDimension::_internal_window_reversal() const {
|
|
return _impl_.window_reversal_;
|
|
}
|
|
inline bool WindowDimension::window_reversal() const {
|
|
// @@protoc_insertion_point(field_get:xla.WindowDimension.window_reversal)
|
|
return _internal_window_reversal();
|
|
}
|
|
inline void WindowDimension::_internal_set_window_reversal(bool value) {
|
|
|
|
_impl_.window_reversal_ = value;
|
|
}
|
|
inline void WindowDimension::set_window_reversal(bool value) {
|
|
_internal_set_window_reversal(value);
|
|
// @@protoc_insertion_point(field_set:xla.WindowDimension.window_reversal)
|
|
}
|
|
|
|
// -------------------------------------------------------------------
|
|
|
|
// Window
|
|
|
|
// repeated .xla.WindowDimension dimensions = 1;
|
|
inline int Window::_internal_dimensions_size() const {
|
|
return _impl_.dimensions_.size();
|
|
}
|
|
inline int Window::dimensions_size() const {
|
|
return _internal_dimensions_size();
|
|
}
|
|
inline void Window::clear_dimensions() {
|
|
_impl_.dimensions_.Clear();
|
|
}
|
|
inline ::xla::WindowDimension* Window::mutable_dimensions(int index) {
|
|
// @@protoc_insertion_point(field_mutable:xla.Window.dimensions)
|
|
return _impl_.dimensions_.Mutable(index);
|
|
}
|
|
inline ::PROTOBUF_NAMESPACE_ID::RepeatedPtrField< ::xla::WindowDimension >*
|
|
Window::mutable_dimensions() {
|
|
// @@protoc_insertion_point(field_mutable_list:xla.Window.dimensions)
|
|
return &_impl_.dimensions_;
|
|
}
|
|
inline const ::xla::WindowDimension& Window::_internal_dimensions(int index) const {
|
|
return _impl_.dimensions_.Get(index);
|
|
}
|
|
inline const ::xla::WindowDimension& Window::dimensions(int index) const {
|
|
// @@protoc_insertion_point(field_get:xla.Window.dimensions)
|
|
return _internal_dimensions(index);
|
|
}
|
|
inline ::xla::WindowDimension* Window::_internal_add_dimensions() {
|
|
return _impl_.dimensions_.Add();
|
|
}
|
|
inline ::xla::WindowDimension* Window::add_dimensions() {
|
|
::xla::WindowDimension* _add = _internal_add_dimensions();
|
|
// @@protoc_insertion_point(field_add:xla.Window.dimensions)
|
|
return _add;
|
|
}
|
|
inline const ::PROTOBUF_NAMESPACE_ID::RepeatedPtrField< ::xla::WindowDimension >&
|
|
Window::dimensions() const {
|
|
// @@protoc_insertion_point(field_list:xla.Window.dimensions)
|
|
return _impl_.dimensions_;
|
|
}
|
|
|
|
// -------------------------------------------------------------------
|
|
|
|
// GatherDimensionNumbers
|
|
|
|
// repeated int64 offset_dims = 1;
|
|
inline int GatherDimensionNumbers::_internal_offset_dims_size() const {
|
|
return _impl_.offset_dims_.size();
|
|
}
|
|
inline int GatherDimensionNumbers::offset_dims_size() const {
|
|
return _internal_offset_dims_size();
|
|
}
|
|
inline void GatherDimensionNumbers::clear_offset_dims() {
|
|
_impl_.offset_dims_.Clear();
|
|
}
|
|
inline int64_t GatherDimensionNumbers::_internal_offset_dims(int index) const {
|
|
return _impl_.offset_dims_.Get(index);
|
|
}
|
|
inline int64_t GatherDimensionNumbers::offset_dims(int index) const {
|
|
// @@protoc_insertion_point(field_get:xla.GatherDimensionNumbers.offset_dims)
|
|
return _internal_offset_dims(index);
|
|
}
|
|
inline void GatherDimensionNumbers::set_offset_dims(int index, int64_t value) {
|
|
_impl_.offset_dims_.Set(index, value);
|
|
// @@protoc_insertion_point(field_set:xla.GatherDimensionNumbers.offset_dims)
|
|
}
|
|
inline void GatherDimensionNumbers::_internal_add_offset_dims(int64_t value) {
|
|
_impl_.offset_dims_.Add(value);
|
|
}
|
|
inline void GatherDimensionNumbers::add_offset_dims(int64_t value) {
|
|
_internal_add_offset_dims(value);
|
|
// @@protoc_insertion_point(field_add:xla.GatherDimensionNumbers.offset_dims)
|
|
}
|
|
inline const ::PROTOBUF_NAMESPACE_ID::RepeatedField< int64_t >&
|
|
GatherDimensionNumbers::_internal_offset_dims() const {
|
|
return _impl_.offset_dims_;
|
|
}
|
|
inline const ::PROTOBUF_NAMESPACE_ID::RepeatedField< int64_t >&
|
|
GatherDimensionNumbers::offset_dims() const {
|
|
// @@protoc_insertion_point(field_list:xla.GatherDimensionNumbers.offset_dims)
|
|
return _internal_offset_dims();
|
|
}
|
|
inline ::PROTOBUF_NAMESPACE_ID::RepeatedField< int64_t >*
|
|
GatherDimensionNumbers::_internal_mutable_offset_dims() {
|
|
return &_impl_.offset_dims_;
|
|
}
|
|
inline ::PROTOBUF_NAMESPACE_ID::RepeatedField< int64_t >*
|
|
GatherDimensionNumbers::mutable_offset_dims() {
|
|
// @@protoc_insertion_point(field_mutable_list:xla.GatherDimensionNumbers.offset_dims)
|
|
return _internal_mutable_offset_dims();
|
|
}
|
|
|
|
// repeated int64 collapsed_slice_dims = 2;
|
|
inline int GatherDimensionNumbers::_internal_collapsed_slice_dims_size() const {
|
|
return _impl_.collapsed_slice_dims_.size();
|
|
}
|
|
inline int GatherDimensionNumbers::collapsed_slice_dims_size() const {
|
|
return _internal_collapsed_slice_dims_size();
|
|
}
|
|
inline void GatherDimensionNumbers::clear_collapsed_slice_dims() {
|
|
_impl_.collapsed_slice_dims_.Clear();
|
|
}
|
|
inline int64_t GatherDimensionNumbers::_internal_collapsed_slice_dims(int index) const {
|
|
return _impl_.collapsed_slice_dims_.Get(index);
|
|
}
|
|
inline int64_t GatherDimensionNumbers::collapsed_slice_dims(int index) const {
|
|
// @@protoc_insertion_point(field_get:xla.GatherDimensionNumbers.collapsed_slice_dims)
|
|
return _internal_collapsed_slice_dims(index);
|
|
}
|
|
inline void GatherDimensionNumbers::set_collapsed_slice_dims(int index, int64_t value) {
|
|
_impl_.collapsed_slice_dims_.Set(index, value);
|
|
// @@protoc_insertion_point(field_set:xla.GatherDimensionNumbers.collapsed_slice_dims)
|
|
}
|
|
inline void GatherDimensionNumbers::_internal_add_collapsed_slice_dims(int64_t value) {
|
|
_impl_.collapsed_slice_dims_.Add(value);
|
|
}
|
|
inline void GatherDimensionNumbers::add_collapsed_slice_dims(int64_t value) {
|
|
_internal_add_collapsed_slice_dims(value);
|
|
// @@protoc_insertion_point(field_add:xla.GatherDimensionNumbers.collapsed_slice_dims)
|
|
}
|
|
inline const ::PROTOBUF_NAMESPACE_ID::RepeatedField< int64_t >&
|
|
GatherDimensionNumbers::_internal_collapsed_slice_dims() const {
|
|
return _impl_.collapsed_slice_dims_;
|
|
}
|
|
inline const ::PROTOBUF_NAMESPACE_ID::RepeatedField< int64_t >&
|
|
GatherDimensionNumbers::collapsed_slice_dims() const {
|
|
// @@protoc_insertion_point(field_list:xla.GatherDimensionNumbers.collapsed_slice_dims)
|
|
return _internal_collapsed_slice_dims();
|
|
}
|
|
inline ::PROTOBUF_NAMESPACE_ID::RepeatedField< int64_t >*
|
|
GatherDimensionNumbers::_internal_mutable_collapsed_slice_dims() {
|
|
return &_impl_.collapsed_slice_dims_;
|
|
}
|
|
inline ::PROTOBUF_NAMESPACE_ID::RepeatedField< int64_t >*
|
|
GatherDimensionNumbers::mutable_collapsed_slice_dims() {
|
|
// @@protoc_insertion_point(field_mutable_list:xla.GatherDimensionNumbers.collapsed_slice_dims)
|
|
return _internal_mutable_collapsed_slice_dims();
|
|
}
|
|
|
|
// repeated int64 start_index_map = 3;
|
|
inline int GatherDimensionNumbers::_internal_start_index_map_size() const {
|
|
return _impl_.start_index_map_.size();
|
|
}
|
|
inline int GatherDimensionNumbers::start_index_map_size() const {
|
|
return _internal_start_index_map_size();
|
|
}
|
|
inline void GatherDimensionNumbers::clear_start_index_map() {
|
|
_impl_.start_index_map_.Clear();
|
|
}
|
|
inline int64_t GatherDimensionNumbers::_internal_start_index_map(int index) const {
|
|
return _impl_.start_index_map_.Get(index);
|
|
}
|
|
inline int64_t GatherDimensionNumbers::start_index_map(int index) const {
|
|
// @@protoc_insertion_point(field_get:xla.GatherDimensionNumbers.start_index_map)
|
|
return _internal_start_index_map(index);
|
|
}
|
|
inline void GatherDimensionNumbers::set_start_index_map(int index, int64_t value) {
|
|
_impl_.start_index_map_.Set(index, value);
|
|
// @@protoc_insertion_point(field_set:xla.GatherDimensionNumbers.start_index_map)
|
|
}
|
|
inline void GatherDimensionNumbers::_internal_add_start_index_map(int64_t value) {
|
|
_impl_.start_index_map_.Add(value);
|
|
}
|
|
inline void GatherDimensionNumbers::add_start_index_map(int64_t value) {
|
|
_internal_add_start_index_map(value);
|
|
// @@protoc_insertion_point(field_add:xla.GatherDimensionNumbers.start_index_map)
|
|
}
|
|
inline const ::PROTOBUF_NAMESPACE_ID::RepeatedField< int64_t >&
|
|
GatherDimensionNumbers::_internal_start_index_map() const {
|
|
return _impl_.start_index_map_;
|
|
}
|
|
inline const ::PROTOBUF_NAMESPACE_ID::RepeatedField< int64_t >&
|
|
GatherDimensionNumbers::start_index_map() const {
|
|
// @@protoc_insertion_point(field_list:xla.GatherDimensionNumbers.start_index_map)
|
|
return _internal_start_index_map();
|
|
}
|
|
inline ::PROTOBUF_NAMESPACE_ID::RepeatedField< int64_t >*
|
|
GatherDimensionNumbers::_internal_mutable_start_index_map() {
|
|
return &_impl_.start_index_map_;
|
|
}
|
|
inline ::PROTOBUF_NAMESPACE_ID::RepeatedField< int64_t >*
|
|
GatherDimensionNumbers::mutable_start_index_map() {
|
|
// @@protoc_insertion_point(field_mutable_list:xla.GatherDimensionNumbers.start_index_map)
|
|
return _internal_mutable_start_index_map();
|
|
}
|
|
|
|
// int64 index_vector_dim = 4;
|
|
inline void GatherDimensionNumbers::clear_index_vector_dim() {
|
|
_impl_.index_vector_dim_ = int64_t{0};
|
|
}
|
|
inline int64_t GatherDimensionNumbers::_internal_index_vector_dim() const {
|
|
return _impl_.index_vector_dim_;
|
|
}
|
|
inline int64_t GatherDimensionNumbers::index_vector_dim() const {
|
|
// @@protoc_insertion_point(field_get:xla.GatherDimensionNumbers.index_vector_dim)
|
|
return _internal_index_vector_dim();
|
|
}
|
|
inline void GatherDimensionNumbers::_internal_set_index_vector_dim(int64_t value) {
|
|
|
|
_impl_.index_vector_dim_ = value;
|
|
}
|
|
inline void GatherDimensionNumbers::set_index_vector_dim(int64_t value) {
|
|
_internal_set_index_vector_dim(value);
|
|
// @@protoc_insertion_point(field_set:xla.GatherDimensionNumbers.index_vector_dim)
|
|
}
|
|
|
|
// -------------------------------------------------------------------
|
|
|
|
// ScatterDimensionNumbers
|
|
|
|
// repeated int64 update_window_dims = 1;
|
|
inline int ScatterDimensionNumbers::_internal_update_window_dims_size() const {
|
|
return _impl_.update_window_dims_.size();
|
|
}
|
|
inline int ScatterDimensionNumbers::update_window_dims_size() const {
|
|
return _internal_update_window_dims_size();
|
|
}
|
|
inline void ScatterDimensionNumbers::clear_update_window_dims() {
|
|
_impl_.update_window_dims_.Clear();
|
|
}
|
|
inline int64_t ScatterDimensionNumbers::_internal_update_window_dims(int index) const {
|
|
return _impl_.update_window_dims_.Get(index);
|
|
}
|
|
inline int64_t ScatterDimensionNumbers::update_window_dims(int index) const {
|
|
// @@protoc_insertion_point(field_get:xla.ScatterDimensionNumbers.update_window_dims)
|
|
return _internal_update_window_dims(index);
|
|
}
|
|
inline void ScatterDimensionNumbers::set_update_window_dims(int index, int64_t value) {
|
|
_impl_.update_window_dims_.Set(index, value);
|
|
// @@protoc_insertion_point(field_set:xla.ScatterDimensionNumbers.update_window_dims)
|
|
}
|
|
inline void ScatterDimensionNumbers::_internal_add_update_window_dims(int64_t value) {
|
|
_impl_.update_window_dims_.Add(value);
|
|
}
|
|
inline void ScatterDimensionNumbers::add_update_window_dims(int64_t value) {
|
|
_internal_add_update_window_dims(value);
|
|
// @@protoc_insertion_point(field_add:xla.ScatterDimensionNumbers.update_window_dims)
|
|
}
|
|
inline const ::PROTOBUF_NAMESPACE_ID::RepeatedField< int64_t >&
|
|
ScatterDimensionNumbers::_internal_update_window_dims() const {
|
|
return _impl_.update_window_dims_;
|
|
}
|
|
inline const ::PROTOBUF_NAMESPACE_ID::RepeatedField< int64_t >&
|
|
ScatterDimensionNumbers::update_window_dims() const {
|
|
// @@protoc_insertion_point(field_list:xla.ScatterDimensionNumbers.update_window_dims)
|
|
return _internal_update_window_dims();
|
|
}
|
|
inline ::PROTOBUF_NAMESPACE_ID::RepeatedField< int64_t >*
|
|
ScatterDimensionNumbers::_internal_mutable_update_window_dims() {
|
|
return &_impl_.update_window_dims_;
|
|
}
|
|
inline ::PROTOBUF_NAMESPACE_ID::RepeatedField< int64_t >*
|
|
ScatterDimensionNumbers::mutable_update_window_dims() {
|
|
// @@protoc_insertion_point(field_mutable_list:xla.ScatterDimensionNumbers.update_window_dims)
|
|
return _internal_mutable_update_window_dims();
|
|
}
|
|
|
|
// repeated int64 inserted_window_dims = 2;
|
|
inline int ScatterDimensionNumbers::_internal_inserted_window_dims_size() const {
|
|
return _impl_.inserted_window_dims_.size();
|
|
}
|
|
inline int ScatterDimensionNumbers::inserted_window_dims_size() const {
|
|
return _internal_inserted_window_dims_size();
|
|
}
|
|
inline void ScatterDimensionNumbers::clear_inserted_window_dims() {
|
|
_impl_.inserted_window_dims_.Clear();
|
|
}
|
|
inline int64_t ScatterDimensionNumbers::_internal_inserted_window_dims(int index) const {
|
|
return _impl_.inserted_window_dims_.Get(index);
|
|
}
|
|
inline int64_t ScatterDimensionNumbers::inserted_window_dims(int index) const {
|
|
// @@protoc_insertion_point(field_get:xla.ScatterDimensionNumbers.inserted_window_dims)
|
|
return _internal_inserted_window_dims(index);
|
|
}
|
|
inline void ScatterDimensionNumbers::set_inserted_window_dims(int index, int64_t value) {
|
|
_impl_.inserted_window_dims_.Set(index, value);
|
|
// @@protoc_insertion_point(field_set:xla.ScatterDimensionNumbers.inserted_window_dims)
|
|
}
|
|
inline void ScatterDimensionNumbers::_internal_add_inserted_window_dims(int64_t value) {
|
|
_impl_.inserted_window_dims_.Add(value);
|
|
}
|
|
inline void ScatterDimensionNumbers::add_inserted_window_dims(int64_t value) {
|
|
_internal_add_inserted_window_dims(value);
|
|
// @@protoc_insertion_point(field_add:xla.ScatterDimensionNumbers.inserted_window_dims)
|
|
}
|
|
inline const ::PROTOBUF_NAMESPACE_ID::RepeatedField< int64_t >&
|
|
ScatterDimensionNumbers::_internal_inserted_window_dims() const {
|
|
return _impl_.inserted_window_dims_;
|
|
}
|
|
inline const ::PROTOBUF_NAMESPACE_ID::RepeatedField< int64_t >&
|
|
ScatterDimensionNumbers::inserted_window_dims() const {
|
|
// @@protoc_insertion_point(field_list:xla.ScatterDimensionNumbers.inserted_window_dims)
|
|
return _internal_inserted_window_dims();
|
|
}
|
|
inline ::PROTOBUF_NAMESPACE_ID::RepeatedField< int64_t >*
|
|
ScatterDimensionNumbers::_internal_mutable_inserted_window_dims() {
|
|
return &_impl_.inserted_window_dims_;
|
|
}
|
|
inline ::PROTOBUF_NAMESPACE_ID::RepeatedField< int64_t >*
|
|
ScatterDimensionNumbers::mutable_inserted_window_dims() {
|
|
// @@protoc_insertion_point(field_mutable_list:xla.ScatterDimensionNumbers.inserted_window_dims)
|
|
return _internal_mutable_inserted_window_dims();
|
|
}
|
|
|
|
// repeated int64 scatter_dims_to_operand_dims = 3;
|
|
inline int ScatterDimensionNumbers::_internal_scatter_dims_to_operand_dims_size() const {
|
|
return _impl_.scatter_dims_to_operand_dims_.size();
|
|
}
|
|
inline int ScatterDimensionNumbers::scatter_dims_to_operand_dims_size() const {
|
|
return _internal_scatter_dims_to_operand_dims_size();
|
|
}
|
|
inline void ScatterDimensionNumbers::clear_scatter_dims_to_operand_dims() {
|
|
_impl_.scatter_dims_to_operand_dims_.Clear();
|
|
}
|
|
inline int64_t ScatterDimensionNumbers::_internal_scatter_dims_to_operand_dims(int index) const {
|
|
return _impl_.scatter_dims_to_operand_dims_.Get(index);
|
|
}
|
|
inline int64_t ScatterDimensionNumbers::scatter_dims_to_operand_dims(int index) const {
|
|
// @@protoc_insertion_point(field_get:xla.ScatterDimensionNumbers.scatter_dims_to_operand_dims)
|
|
return _internal_scatter_dims_to_operand_dims(index);
|
|
}
|
|
inline void ScatterDimensionNumbers::set_scatter_dims_to_operand_dims(int index, int64_t value) {
|
|
_impl_.scatter_dims_to_operand_dims_.Set(index, value);
|
|
// @@protoc_insertion_point(field_set:xla.ScatterDimensionNumbers.scatter_dims_to_operand_dims)
|
|
}
|
|
inline void ScatterDimensionNumbers::_internal_add_scatter_dims_to_operand_dims(int64_t value) {
|
|
_impl_.scatter_dims_to_operand_dims_.Add(value);
|
|
}
|
|
inline void ScatterDimensionNumbers::add_scatter_dims_to_operand_dims(int64_t value) {
|
|
_internal_add_scatter_dims_to_operand_dims(value);
|
|
// @@protoc_insertion_point(field_add:xla.ScatterDimensionNumbers.scatter_dims_to_operand_dims)
|
|
}
|
|
inline const ::PROTOBUF_NAMESPACE_ID::RepeatedField< int64_t >&
|
|
ScatterDimensionNumbers::_internal_scatter_dims_to_operand_dims() const {
|
|
return _impl_.scatter_dims_to_operand_dims_;
|
|
}
|
|
inline const ::PROTOBUF_NAMESPACE_ID::RepeatedField< int64_t >&
|
|
ScatterDimensionNumbers::scatter_dims_to_operand_dims() const {
|
|
// @@protoc_insertion_point(field_list:xla.ScatterDimensionNumbers.scatter_dims_to_operand_dims)
|
|
return _internal_scatter_dims_to_operand_dims();
|
|
}
|
|
inline ::PROTOBUF_NAMESPACE_ID::RepeatedField< int64_t >*
|
|
ScatterDimensionNumbers::_internal_mutable_scatter_dims_to_operand_dims() {
|
|
return &_impl_.scatter_dims_to_operand_dims_;
|
|
}
|
|
inline ::PROTOBUF_NAMESPACE_ID::RepeatedField< int64_t >*
|
|
ScatterDimensionNumbers::mutable_scatter_dims_to_operand_dims() {
|
|
// @@protoc_insertion_point(field_mutable_list:xla.ScatterDimensionNumbers.scatter_dims_to_operand_dims)
|
|
return _internal_mutable_scatter_dims_to_operand_dims();
|
|
}
|
|
|
|
// int64 index_vector_dim = 4;
|
|
inline void ScatterDimensionNumbers::clear_index_vector_dim() {
|
|
_impl_.index_vector_dim_ = int64_t{0};
|
|
}
|
|
inline int64_t ScatterDimensionNumbers::_internal_index_vector_dim() const {
|
|
return _impl_.index_vector_dim_;
|
|
}
|
|
inline int64_t ScatterDimensionNumbers::index_vector_dim() const {
|
|
// @@protoc_insertion_point(field_get:xla.ScatterDimensionNumbers.index_vector_dim)
|
|
return _internal_index_vector_dim();
|
|
}
|
|
inline void ScatterDimensionNumbers::_internal_set_index_vector_dim(int64_t value) {
|
|
|
|
_impl_.index_vector_dim_ = value;
|
|
}
|
|
inline void ScatterDimensionNumbers::set_index_vector_dim(int64_t value) {
|
|
_internal_set_index_vector_dim(value);
|
|
// @@protoc_insertion_point(field_set:xla.ScatterDimensionNumbers.index_vector_dim)
|
|
}
|
|
|
|
// -------------------------------------------------------------------
|
|
|
|
// ConvolutionDimensionNumbers
|
|
|
|
// int64 input_batch_dimension = 7;
|
|
inline void ConvolutionDimensionNumbers::clear_input_batch_dimension() {
|
|
_impl_.input_batch_dimension_ = int64_t{0};
|
|
}
|
|
inline int64_t ConvolutionDimensionNumbers::_internal_input_batch_dimension() const {
|
|
return _impl_.input_batch_dimension_;
|
|
}
|
|
inline int64_t ConvolutionDimensionNumbers::input_batch_dimension() const {
|
|
// @@protoc_insertion_point(field_get:xla.ConvolutionDimensionNumbers.input_batch_dimension)
|
|
return _internal_input_batch_dimension();
|
|
}
|
|
inline void ConvolutionDimensionNumbers::_internal_set_input_batch_dimension(int64_t value) {
|
|
|
|
_impl_.input_batch_dimension_ = value;
|
|
}
|
|
inline void ConvolutionDimensionNumbers::set_input_batch_dimension(int64_t value) {
|
|
_internal_set_input_batch_dimension(value);
|
|
// @@protoc_insertion_point(field_set:xla.ConvolutionDimensionNumbers.input_batch_dimension)
|
|
}
|
|
|
|
// int64 input_feature_dimension = 8;
|
|
inline void ConvolutionDimensionNumbers::clear_input_feature_dimension() {
|
|
_impl_.input_feature_dimension_ = int64_t{0};
|
|
}
|
|
inline int64_t ConvolutionDimensionNumbers::_internal_input_feature_dimension() const {
|
|
return _impl_.input_feature_dimension_;
|
|
}
|
|
inline int64_t ConvolutionDimensionNumbers::input_feature_dimension() const {
|
|
// @@protoc_insertion_point(field_get:xla.ConvolutionDimensionNumbers.input_feature_dimension)
|
|
return _internal_input_feature_dimension();
|
|
}
|
|
inline void ConvolutionDimensionNumbers::_internal_set_input_feature_dimension(int64_t value) {
|
|
|
|
_impl_.input_feature_dimension_ = value;
|
|
}
|
|
inline void ConvolutionDimensionNumbers::set_input_feature_dimension(int64_t value) {
|
|
_internal_set_input_feature_dimension(value);
|
|
// @@protoc_insertion_point(field_set:xla.ConvolutionDimensionNumbers.input_feature_dimension)
|
|
}
|
|
|
|
// repeated int64 input_spatial_dimensions = 11;
|
|
inline int ConvolutionDimensionNumbers::_internal_input_spatial_dimensions_size() const {
|
|
return _impl_.input_spatial_dimensions_.size();
|
|
}
|
|
inline int ConvolutionDimensionNumbers::input_spatial_dimensions_size() const {
|
|
return _internal_input_spatial_dimensions_size();
|
|
}
|
|
inline void ConvolutionDimensionNumbers::clear_input_spatial_dimensions() {
|
|
_impl_.input_spatial_dimensions_.Clear();
|
|
}
|
|
inline int64_t ConvolutionDimensionNumbers::_internal_input_spatial_dimensions(int index) const {
|
|
return _impl_.input_spatial_dimensions_.Get(index);
|
|
}
|
|
inline int64_t ConvolutionDimensionNumbers::input_spatial_dimensions(int index) const {
|
|
// @@protoc_insertion_point(field_get:xla.ConvolutionDimensionNumbers.input_spatial_dimensions)
|
|
return _internal_input_spatial_dimensions(index);
|
|
}
|
|
inline void ConvolutionDimensionNumbers::set_input_spatial_dimensions(int index, int64_t value) {
|
|
_impl_.input_spatial_dimensions_.Set(index, value);
|
|
// @@protoc_insertion_point(field_set:xla.ConvolutionDimensionNumbers.input_spatial_dimensions)
|
|
}
|
|
inline void ConvolutionDimensionNumbers::_internal_add_input_spatial_dimensions(int64_t value) {
|
|
_impl_.input_spatial_dimensions_.Add(value);
|
|
}
|
|
inline void ConvolutionDimensionNumbers::add_input_spatial_dimensions(int64_t value) {
|
|
_internal_add_input_spatial_dimensions(value);
|
|
// @@protoc_insertion_point(field_add:xla.ConvolutionDimensionNumbers.input_spatial_dimensions)
|
|
}
|
|
inline const ::PROTOBUF_NAMESPACE_ID::RepeatedField< int64_t >&
|
|
ConvolutionDimensionNumbers::_internal_input_spatial_dimensions() const {
|
|
return _impl_.input_spatial_dimensions_;
|
|
}
|
|
inline const ::PROTOBUF_NAMESPACE_ID::RepeatedField< int64_t >&
|
|
ConvolutionDimensionNumbers::input_spatial_dimensions() const {
|
|
// @@protoc_insertion_point(field_list:xla.ConvolutionDimensionNumbers.input_spatial_dimensions)
|
|
return _internal_input_spatial_dimensions();
|
|
}
|
|
inline ::PROTOBUF_NAMESPACE_ID::RepeatedField< int64_t >*
|
|
ConvolutionDimensionNumbers::_internal_mutable_input_spatial_dimensions() {
|
|
return &_impl_.input_spatial_dimensions_;
|
|
}
|
|
inline ::PROTOBUF_NAMESPACE_ID::RepeatedField< int64_t >*
|
|
ConvolutionDimensionNumbers::mutable_input_spatial_dimensions() {
|
|
// @@protoc_insertion_point(field_mutable_list:xla.ConvolutionDimensionNumbers.input_spatial_dimensions)
|
|
return _internal_mutable_input_spatial_dimensions();
|
|
}
|
|
|
|
// int64 kernel_input_feature_dimension = 3;
|
|
inline void ConvolutionDimensionNumbers::clear_kernel_input_feature_dimension() {
|
|
_impl_.kernel_input_feature_dimension_ = int64_t{0};
|
|
}
|
|
inline int64_t ConvolutionDimensionNumbers::_internal_kernel_input_feature_dimension() const {
|
|
return _impl_.kernel_input_feature_dimension_;
|
|
}
|
|
inline int64_t ConvolutionDimensionNumbers::kernel_input_feature_dimension() const {
|
|
// @@protoc_insertion_point(field_get:xla.ConvolutionDimensionNumbers.kernel_input_feature_dimension)
|
|
return _internal_kernel_input_feature_dimension();
|
|
}
|
|
inline void ConvolutionDimensionNumbers::_internal_set_kernel_input_feature_dimension(int64_t value) {
|
|
|
|
_impl_.kernel_input_feature_dimension_ = value;
|
|
}
|
|
inline void ConvolutionDimensionNumbers::set_kernel_input_feature_dimension(int64_t value) {
|
|
_internal_set_kernel_input_feature_dimension(value);
|
|
// @@protoc_insertion_point(field_set:xla.ConvolutionDimensionNumbers.kernel_input_feature_dimension)
|
|
}
|
|
|
|
// int64 kernel_output_feature_dimension = 4;
|
|
inline void ConvolutionDimensionNumbers::clear_kernel_output_feature_dimension() {
|
|
_impl_.kernel_output_feature_dimension_ = int64_t{0};
|
|
}
|
|
inline int64_t ConvolutionDimensionNumbers::_internal_kernel_output_feature_dimension() const {
|
|
return _impl_.kernel_output_feature_dimension_;
|
|
}
|
|
inline int64_t ConvolutionDimensionNumbers::kernel_output_feature_dimension() const {
|
|
// @@protoc_insertion_point(field_get:xla.ConvolutionDimensionNumbers.kernel_output_feature_dimension)
|
|
return _internal_kernel_output_feature_dimension();
|
|
}
|
|
inline void ConvolutionDimensionNumbers::_internal_set_kernel_output_feature_dimension(int64_t value) {
|
|
|
|
_impl_.kernel_output_feature_dimension_ = value;
|
|
}
|
|
inline void ConvolutionDimensionNumbers::set_kernel_output_feature_dimension(int64_t value) {
|
|
_internal_set_kernel_output_feature_dimension(value);
|
|
// @@protoc_insertion_point(field_set:xla.ConvolutionDimensionNumbers.kernel_output_feature_dimension)
|
|
}
|
|
|
|
// repeated int64 kernel_spatial_dimensions = 6;
|
|
inline int ConvolutionDimensionNumbers::_internal_kernel_spatial_dimensions_size() const {
|
|
return _impl_.kernel_spatial_dimensions_.size();
|
|
}
|
|
inline int ConvolutionDimensionNumbers::kernel_spatial_dimensions_size() const {
|
|
return _internal_kernel_spatial_dimensions_size();
|
|
}
|
|
inline void ConvolutionDimensionNumbers::clear_kernel_spatial_dimensions() {
|
|
_impl_.kernel_spatial_dimensions_.Clear();
|
|
}
|
|
inline int64_t ConvolutionDimensionNumbers::_internal_kernel_spatial_dimensions(int index) const {
|
|
return _impl_.kernel_spatial_dimensions_.Get(index);
|
|
}
|
|
inline int64_t ConvolutionDimensionNumbers::kernel_spatial_dimensions(int index) const {
|
|
// @@protoc_insertion_point(field_get:xla.ConvolutionDimensionNumbers.kernel_spatial_dimensions)
|
|
return _internal_kernel_spatial_dimensions(index);
|
|
}
|
|
inline void ConvolutionDimensionNumbers::set_kernel_spatial_dimensions(int index, int64_t value) {
|
|
_impl_.kernel_spatial_dimensions_.Set(index, value);
|
|
// @@protoc_insertion_point(field_set:xla.ConvolutionDimensionNumbers.kernel_spatial_dimensions)
|
|
}
|
|
inline void ConvolutionDimensionNumbers::_internal_add_kernel_spatial_dimensions(int64_t value) {
|
|
_impl_.kernel_spatial_dimensions_.Add(value);
|
|
}
|
|
inline void ConvolutionDimensionNumbers::add_kernel_spatial_dimensions(int64_t value) {
|
|
_internal_add_kernel_spatial_dimensions(value);
|
|
// @@protoc_insertion_point(field_add:xla.ConvolutionDimensionNumbers.kernel_spatial_dimensions)
|
|
}
|
|
inline const ::PROTOBUF_NAMESPACE_ID::RepeatedField< int64_t >&
|
|
ConvolutionDimensionNumbers::_internal_kernel_spatial_dimensions() const {
|
|
return _impl_.kernel_spatial_dimensions_;
|
|
}
|
|
inline const ::PROTOBUF_NAMESPACE_ID::RepeatedField< int64_t >&
|
|
ConvolutionDimensionNumbers::kernel_spatial_dimensions() const {
|
|
// @@protoc_insertion_point(field_list:xla.ConvolutionDimensionNumbers.kernel_spatial_dimensions)
|
|
return _internal_kernel_spatial_dimensions();
|
|
}
|
|
inline ::PROTOBUF_NAMESPACE_ID::RepeatedField< int64_t >*
|
|
ConvolutionDimensionNumbers::_internal_mutable_kernel_spatial_dimensions() {
|
|
return &_impl_.kernel_spatial_dimensions_;
|
|
}
|
|
inline ::PROTOBUF_NAMESPACE_ID::RepeatedField< int64_t >*
|
|
ConvolutionDimensionNumbers::mutable_kernel_spatial_dimensions() {
|
|
// @@protoc_insertion_point(field_mutable_list:xla.ConvolutionDimensionNumbers.kernel_spatial_dimensions)
|
|
return _internal_mutable_kernel_spatial_dimensions();
|
|
}
|
|
|
|
// int64 output_batch_dimension = 9;
|
|
inline void ConvolutionDimensionNumbers::clear_output_batch_dimension() {
|
|
_impl_.output_batch_dimension_ = int64_t{0};
|
|
}
|
|
inline int64_t ConvolutionDimensionNumbers::_internal_output_batch_dimension() const {
|
|
return _impl_.output_batch_dimension_;
|
|
}
|
|
inline int64_t ConvolutionDimensionNumbers::output_batch_dimension() const {
|
|
// @@protoc_insertion_point(field_get:xla.ConvolutionDimensionNumbers.output_batch_dimension)
|
|
return _internal_output_batch_dimension();
|
|
}
|
|
inline void ConvolutionDimensionNumbers::_internal_set_output_batch_dimension(int64_t value) {
|
|
|
|
_impl_.output_batch_dimension_ = value;
|
|
}
|
|
inline void ConvolutionDimensionNumbers::set_output_batch_dimension(int64_t value) {
|
|
_internal_set_output_batch_dimension(value);
|
|
// @@protoc_insertion_point(field_set:xla.ConvolutionDimensionNumbers.output_batch_dimension)
|
|
}
|
|
|
|
// int64 output_feature_dimension = 10;
|
|
inline void ConvolutionDimensionNumbers::clear_output_feature_dimension() {
|
|
_impl_.output_feature_dimension_ = int64_t{0};
|
|
}
|
|
inline int64_t ConvolutionDimensionNumbers::_internal_output_feature_dimension() const {
|
|
return _impl_.output_feature_dimension_;
|
|
}
|
|
inline int64_t ConvolutionDimensionNumbers::output_feature_dimension() const {
|
|
// @@protoc_insertion_point(field_get:xla.ConvolutionDimensionNumbers.output_feature_dimension)
|
|
return _internal_output_feature_dimension();
|
|
}
|
|
inline void ConvolutionDimensionNumbers::_internal_set_output_feature_dimension(int64_t value) {
|
|
|
|
_impl_.output_feature_dimension_ = value;
|
|
}
|
|
inline void ConvolutionDimensionNumbers::set_output_feature_dimension(int64_t value) {
|
|
_internal_set_output_feature_dimension(value);
|
|
// @@protoc_insertion_point(field_set:xla.ConvolutionDimensionNumbers.output_feature_dimension)
|
|
}
|
|
|
|
// repeated int64 output_spatial_dimensions = 12;
|
|
inline int ConvolutionDimensionNumbers::_internal_output_spatial_dimensions_size() const {
|
|
return _impl_.output_spatial_dimensions_.size();
|
|
}
|
|
inline int ConvolutionDimensionNumbers::output_spatial_dimensions_size() const {
|
|
return _internal_output_spatial_dimensions_size();
|
|
}
|
|
inline void ConvolutionDimensionNumbers::clear_output_spatial_dimensions() {
|
|
_impl_.output_spatial_dimensions_.Clear();
|
|
}
|
|
inline int64_t ConvolutionDimensionNumbers::_internal_output_spatial_dimensions(int index) const {
|
|
return _impl_.output_spatial_dimensions_.Get(index);
|
|
}
|
|
inline int64_t ConvolutionDimensionNumbers::output_spatial_dimensions(int index) const {
|
|
// @@protoc_insertion_point(field_get:xla.ConvolutionDimensionNumbers.output_spatial_dimensions)
|
|
return _internal_output_spatial_dimensions(index);
|
|
}
|
|
inline void ConvolutionDimensionNumbers::set_output_spatial_dimensions(int index, int64_t value) {
|
|
_impl_.output_spatial_dimensions_.Set(index, value);
|
|
// @@protoc_insertion_point(field_set:xla.ConvolutionDimensionNumbers.output_spatial_dimensions)
|
|
}
|
|
inline void ConvolutionDimensionNumbers::_internal_add_output_spatial_dimensions(int64_t value) {
|
|
_impl_.output_spatial_dimensions_.Add(value);
|
|
}
|
|
inline void ConvolutionDimensionNumbers::add_output_spatial_dimensions(int64_t value) {
|
|
_internal_add_output_spatial_dimensions(value);
|
|
// @@protoc_insertion_point(field_add:xla.ConvolutionDimensionNumbers.output_spatial_dimensions)
|
|
}
|
|
inline const ::PROTOBUF_NAMESPACE_ID::RepeatedField< int64_t >&
|
|
ConvolutionDimensionNumbers::_internal_output_spatial_dimensions() const {
|
|
return _impl_.output_spatial_dimensions_;
|
|
}
|
|
inline const ::PROTOBUF_NAMESPACE_ID::RepeatedField< int64_t >&
|
|
ConvolutionDimensionNumbers::output_spatial_dimensions() const {
|
|
// @@protoc_insertion_point(field_list:xla.ConvolutionDimensionNumbers.output_spatial_dimensions)
|
|
return _internal_output_spatial_dimensions();
|
|
}
|
|
inline ::PROTOBUF_NAMESPACE_ID::RepeatedField< int64_t >*
|
|
ConvolutionDimensionNumbers::_internal_mutable_output_spatial_dimensions() {
|
|
return &_impl_.output_spatial_dimensions_;
|
|
}
|
|
inline ::PROTOBUF_NAMESPACE_ID::RepeatedField< int64_t >*
|
|
ConvolutionDimensionNumbers::mutable_output_spatial_dimensions() {
|
|
// @@protoc_insertion_point(field_mutable_list:xla.ConvolutionDimensionNumbers.output_spatial_dimensions)
|
|
return _internal_mutable_output_spatial_dimensions();
|
|
}
|
|
|
|
// -------------------------------------------------------------------
|
|
|
|
// DotDimensionNumbers
|
|
|
|
// repeated int64 lhs_contracting_dimensions = 1;
|
|
inline int DotDimensionNumbers::_internal_lhs_contracting_dimensions_size() const {
|
|
return _impl_.lhs_contracting_dimensions_.size();
|
|
}
|
|
inline int DotDimensionNumbers::lhs_contracting_dimensions_size() const {
|
|
return _internal_lhs_contracting_dimensions_size();
|
|
}
|
|
inline void DotDimensionNumbers::clear_lhs_contracting_dimensions() {
|
|
_impl_.lhs_contracting_dimensions_.Clear();
|
|
}
|
|
inline int64_t DotDimensionNumbers::_internal_lhs_contracting_dimensions(int index) const {
|
|
return _impl_.lhs_contracting_dimensions_.Get(index);
|
|
}
|
|
inline int64_t DotDimensionNumbers::lhs_contracting_dimensions(int index) const {
|
|
// @@protoc_insertion_point(field_get:xla.DotDimensionNumbers.lhs_contracting_dimensions)
|
|
return _internal_lhs_contracting_dimensions(index);
|
|
}
|
|
inline void DotDimensionNumbers::set_lhs_contracting_dimensions(int index, int64_t value) {
|
|
_impl_.lhs_contracting_dimensions_.Set(index, value);
|
|
// @@protoc_insertion_point(field_set:xla.DotDimensionNumbers.lhs_contracting_dimensions)
|
|
}
|
|
inline void DotDimensionNumbers::_internal_add_lhs_contracting_dimensions(int64_t value) {
|
|
_impl_.lhs_contracting_dimensions_.Add(value);
|
|
}
|
|
inline void DotDimensionNumbers::add_lhs_contracting_dimensions(int64_t value) {
|
|
_internal_add_lhs_contracting_dimensions(value);
|
|
// @@protoc_insertion_point(field_add:xla.DotDimensionNumbers.lhs_contracting_dimensions)
|
|
}
|
|
inline const ::PROTOBUF_NAMESPACE_ID::RepeatedField< int64_t >&
|
|
DotDimensionNumbers::_internal_lhs_contracting_dimensions() const {
|
|
return _impl_.lhs_contracting_dimensions_;
|
|
}
|
|
inline const ::PROTOBUF_NAMESPACE_ID::RepeatedField< int64_t >&
|
|
DotDimensionNumbers::lhs_contracting_dimensions() const {
|
|
// @@protoc_insertion_point(field_list:xla.DotDimensionNumbers.lhs_contracting_dimensions)
|
|
return _internal_lhs_contracting_dimensions();
|
|
}
|
|
inline ::PROTOBUF_NAMESPACE_ID::RepeatedField< int64_t >*
|
|
DotDimensionNumbers::_internal_mutable_lhs_contracting_dimensions() {
|
|
return &_impl_.lhs_contracting_dimensions_;
|
|
}
|
|
inline ::PROTOBUF_NAMESPACE_ID::RepeatedField< int64_t >*
|
|
DotDimensionNumbers::mutable_lhs_contracting_dimensions() {
|
|
// @@protoc_insertion_point(field_mutable_list:xla.DotDimensionNumbers.lhs_contracting_dimensions)
|
|
return _internal_mutable_lhs_contracting_dimensions();
|
|
}
|
|
|
|
// repeated int64 rhs_contracting_dimensions = 2;
|
|
inline int DotDimensionNumbers::_internal_rhs_contracting_dimensions_size() const {
|
|
return _impl_.rhs_contracting_dimensions_.size();
|
|
}
|
|
inline int DotDimensionNumbers::rhs_contracting_dimensions_size() const {
|
|
return _internal_rhs_contracting_dimensions_size();
|
|
}
|
|
inline void DotDimensionNumbers::clear_rhs_contracting_dimensions() {
|
|
_impl_.rhs_contracting_dimensions_.Clear();
|
|
}
|
|
inline int64_t DotDimensionNumbers::_internal_rhs_contracting_dimensions(int index) const {
|
|
return _impl_.rhs_contracting_dimensions_.Get(index);
|
|
}
|
|
inline int64_t DotDimensionNumbers::rhs_contracting_dimensions(int index) const {
|
|
// @@protoc_insertion_point(field_get:xla.DotDimensionNumbers.rhs_contracting_dimensions)
|
|
return _internal_rhs_contracting_dimensions(index);
|
|
}
|
|
inline void DotDimensionNumbers::set_rhs_contracting_dimensions(int index, int64_t value) {
|
|
_impl_.rhs_contracting_dimensions_.Set(index, value);
|
|
// @@protoc_insertion_point(field_set:xla.DotDimensionNumbers.rhs_contracting_dimensions)
|
|
}
|
|
inline void DotDimensionNumbers::_internal_add_rhs_contracting_dimensions(int64_t value) {
|
|
_impl_.rhs_contracting_dimensions_.Add(value);
|
|
}
|
|
inline void DotDimensionNumbers::add_rhs_contracting_dimensions(int64_t value) {
|
|
_internal_add_rhs_contracting_dimensions(value);
|
|
// @@protoc_insertion_point(field_add:xla.DotDimensionNumbers.rhs_contracting_dimensions)
|
|
}
|
|
inline const ::PROTOBUF_NAMESPACE_ID::RepeatedField< int64_t >&
|
|
DotDimensionNumbers::_internal_rhs_contracting_dimensions() const {
|
|
return _impl_.rhs_contracting_dimensions_;
|
|
}
|
|
inline const ::PROTOBUF_NAMESPACE_ID::RepeatedField< int64_t >&
|
|
DotDimensionNumbers::rhs_contracting_dimensions() const {
|
|
// @@protoc_insertion_point(field_list:xla.DotDimensionNumbers.rhs_contracting_dimensions)
|
|
return _internal_rhs_contracting_dimensions();
|
|
}
|
|
inline ::PROTOBUF_NAMESPACE_ID::RepeatedField< int64_t >*
|
|
DotDimensionNumbers::_internal_mutable_rhs_contracting_dimensions() {
|
|
return &_impl_.rhs_contracting_dimensions_;
|
|
}
|
|
inline ::PROTOBUF_NAMESPACE_ID::RepeatedField< int64_t >*
|
|
DotDimensionNumbers::mutable_rhs_contracting_dimensions() {
|
|
// @@protoc_insertion_point(field_mutable_list:xla.DotDimensionNumbers.rhs_contracting_dimensions)
|
|
return _internal_mutable_rhs_contracting_dimensions();
|
|
}
|
|
|
|
// repeated int64 lhs_batch_dimensions = 3;
|
|
inline int DotDimensionNumbers::_internal_lhs_batch_dimensions_size() const {
|
|
return _impl_.lhs_batch_dimensions_.size();
|
|
}
|
|
inline int DotDimensionNumbers::lhs_batch_dimensions_size() const {
|
|
return _internal_lhs_batch_dimensions_size();
|
|
}
|
|
inline void DotDimensionNumbers::clear_lhs_batch_dimensions() {
|
|
_impl_.lhs_batch_dimensions_.Clear();
|
|
}
|
|
inline int64_t DotDimensionNumbers::_internal_lhs_batch_dimensions(int index) const {
|
|
return _impl_.lhs_batch_dimensions_.Get(index);
|
|
}
|
|
inline int64_t DotDimensionNumbers::lhs_batch_dimensions(int index) const {
|
|
// @@protoc_insertion_point(field_get:xla.DotDimensionNumbers.lhs_batch_dimensions)
|
|
return _internal_lhs_batch_dimensions(index);
|
|
}
|
|
inline void DotDimensionNumbers::set_lhs_batch_dimensions(int index, int64_t value) {
|
|
_impl_.lhs_batch_dimensions_.Set(index, value);
|
|
// @@protoc_insertion_point(field_set:xla.DotDimensionNumbers.lhs_batch_dimensions)
|
|
}
|
|
inline void DotDimensionNumbers::_internal_add_lhs_batch_dimensions(int64_t value) {
|
|
_impl_.lhs_batch_dimensions_.Add(value);
|
|
}
|
|
inline void DotDimensionNumbers::add_lhs_batch_dimensions(int64_t value) {
|
|
_internal_add_lhs_batch_dimensions(value);
|
|
// @@protoc_insertion_point(field_add:xla.DotDimensionNumbers.lhs_batch_dimensions)
|
|
}
|
|
inline const ::PROTOBUF_NAMESPACE_ID::RepeatedField< int64_t >&
|
|
DotDimensionNumbers::_internal_lhs_batch_dimensions() const {
|
|
return _impl_.lhs_batch_dimensions_;
|
|
}
|
|
inline const ::PROTOBUF_NAMESPACE_ID::RepeatedField< int64_t >&
|
|
DotDimensionNumbers::lhs_batch_dimensions() const {
|
|
// @@protoc_insertion_point(field_list:xla.DotDimensionNumbers.lhs_batch_dimensions)
|
|
return _internal_lhs_batch_dimensions();
|
|
}
|
|
inline ::PROTOBUF_NAMESPACE_ID::RepeatedField< int64_t >*
|
|
DotDimensionNumbers::_internal_mutable_lhs_batch_dimensions() {
|
|
return &_impl_.lhs_batch_dimensions_;
|
|
}
|
|
inline ::PROTOBUF_NAMESPACE_ID::RepeatedField< int64_t >*
|
|
DotDimensionNumbers::mutable_lhs_batch_dimensions() {
|
|
// @@protoc_insertion_point(field_mutable_list:xla.DotDimensionNumbers.lhs_batch_dimensions)
|
|
return _internal_mutable_lhs_batch_dimensions();
|
|
}
|
|
|
|
// repeated int64 rhs_batch_dimensions = 4;
|
|
inline int DotDimensionNumbers::_internal_rhs_batch_dimensions_size() const {
|
|
return _impl_.rhs_batch_dimensions_.size();
|
|
}
|
|
inline int DotDimensionNumbers::rhs_batch_dimensions_size() const {
|
|
return _internal_rhs_batch_dimensions_size();
|
|
}
|
|
inline void DotDimensionNumbers::clear_rhs_batch_dimensions() {
|
|
_impl_.rhs_batch_dimensions_.Clear();
|
|
}
|
|
inline int64_t DotDimensionNumbers::_internal_rhs_batch_dimensions(int index) const {
|
|
return _impl_.rhs_batch_dimensions_.Get(index);
|
|
}
|
|
inline int64_t DotDimensionNumbers::rhs_batch_dimensions(int index) const {
|
|
// @@protoc_insertion_point(field_get:xla.DotDimensionNumbers.rhs_batch_dimensions)
|
|
return _internal_rhs_batch_dimensions(index);
|
|
}
|
|
inline void DotDimensionNumbers::set_rhs_batch_dimensions(int index, int64_t value) {
|
|
_impl_.rhs_batch_dimensions_.Set(index, value);
|
|
// @@protoc_insertion_point(field_set:xla.DotDimensionNumbers.rhs_batch_dimensions)
|
|
}
|
|
inline void DotDimensionNumbers::_internal_add_rhs_batch_dimensions(int64_t value) {
|
|
_impl_.rhs_batch_dimensions_.Add(value);
|
|
}
|
|
inline void DotDimensionNumbers::add_rhs_batch_dimensions(int64_t value) {
|
|
_internal_add_rhs_batch_dimensions(value);
|
|
// @@protoc_insertion_point(field_add:xla.DotDimensionNumbers.rhs_batch_dimensions)
|
|
}
|
|
inline const ::PROTOBUF_NAMESPACE_ID::RepeatedField< int64_t >&
|
|
DotDimensionNumbers::_internal_rhs_batch_dimensions() const {
|
|
return _impl_.rhs_batch_dimensions_;
|
|
}
|
|
inline const ::PROTOBUF_NAMESPACE_ID::RepeatedField< int64_t >&
|
|
DotDimensionNumbers::rhs_batch_dimensions() const {
|
|
// @@protoc_insertion_point(field_list:xla.DotDimensionNumbers.rhs_batch_dimensions)
|
|
return _internal_rhs_batch_dimensions();
|
|
}
|
|
inline ::PROTOBUF_NAMESPACE_ID::RepeatedField< int64_t >*
|
|
DotDimensionNumbers::_internal_mutable_rhs_batch_dimensions() {
|
|
return &_impl_.rhs_batch_dimensions_;
|
|
}
|
|
inline ::PROTOBUF_NAMESPACE_ID::RepeatedField< int64_t >*
|
|
DotDimensionNumbers::mutable_rhs_batch_dimensions() {
|
|
// @@protoc_insertion_point(field_mutable_list:xla.DotDimensionNumbers.rhs_batch_dimensions)
|
|
return _internal_mutable_rhs_batch_dimensions();
|
|
}
|
|
|
|
// -------------------------------------------------------------------
|
|
|
|
// SparsityDescriptor
|
|
|
|
// .xla.SparsityType type = 1;
|
|
inline void SparsityDescriptor::clear_type() {
|
|
_impl_.type_ = 0;
|
|
}
|
|
inline ::xla::SparsityType SparsityDescriptor::_internal_type() const {
|
|
return static_cast< ::xla::SparsityType >(_impl_.type_);
|
|
}
|
|
inline ::xla::SparsityType SparsityDescriptor::type() const {
|
|
// @@protoc_insertion_point(field_get:xla.SparsityDescriptor.type)
|
|
return _internal_type();
|
|
}
|
|
inline void SparsityDescriptor::_internal_set_type(::xla::SparsityType value) {
|
|
|
|
_impl_.type_ = value;
|
|
}
|
|
inline void SparsityDescriptor::set_type(::xla::SparsityType value) {
|
|
_internal_set_type(value);
|
|
// @@protoc_insertion_point(field_set:xla.SparsityDescriptor.type)
|
|
}
|
|
|
|
// int32 index = 2;
|
|
inline void SparsityDescriptor::clear_index() {
|
|
_impl_.index_ = 0;
|
|
}
|
|
inline int32_t SparsityDescriptor::_internal_index() const {
|
|
return _impl_.index_;
|
|
}
|
|
inline int32_t SparsityDescriptor::index() const {
|
|
// @@protoc_insertion_point(field_get:xla.SparsityDescriptor.index)
|
|
return _internal_index();
|
|
}
|
|
inline void SparsityDescriptor::_internal_set_index(int32_t value) {
|
|
|
|
_impl_.index_ = value;
|
|
}
|
|
inline void SparsityDescriptor::set_index(int32_t value) {
|
|
_internal_set_index(value);
|
|
// @@protoc_insertion_point(field_set:xla.SparsityDescriptor.index)
|
|
}
|
|
|
|
// int32 dimension = 3;
|
|
inline void SparsityDescriptor::clear_dimension() {
|
|
_impl_.dimension_ = 0;
|
|
}
|
|
inline int32_t SparsityDescriptor::_internal_dimension() const {
|
|
return _impl_.dimension_;
|
|
}
|
|
inline int32_t SparsityDescriptor::dimension() const {
|
|
// @@protoc_insertion_point(field_get:xla.SparsityDescriptor.dimension)
|
|
return _internal_dimension();
|
|
}
|
|
inline void SparsityDescriptor::_internal_set_dimension(int32_t value) {
|
|
|
|
_impl_.dimension_ = value;
|
|
}
|
|
inline void SparsityDescriptor::set_dimension(int32_t value) {
|
|
_internal_set_dimension(value);
|
|
// @@protoc_insertion_point(field_set:xla.SparsityDescriptor.dimension)
|
|
}
|
|
|
|
// int32 n = 4;
|
|
inline void SparsityDescriptor::clear_n() {
|
|
_impl_.n_ = 0;
|
|
}
|
|
inline int32_t SparsityDescriptor::_internal_n() const {
|
|
return _impl_.n_;
|
|
}
|
|
inline int32_t SparsityDescriptor::n() const {
|
|
// @@protoc_insertion_point(field_get:xla.SparsityDescriptor.n)
|
|
return _internal_n();
|
|
}
|
|
inline void SparsityDescriptor::_internal_set_n(int32_t value) {
|
|
|
|
_impl_.n_ = value;
|
|
}
|
|
inline void SparsityDescriptor::set_n(int32_t value) {
|
|
_internal_set_n(value);
|
|
// @@protoc_insertion_point(field_set:xla.SparsityDescriptor.n)
|
|
}
|
|
|
|
// int32 m = 5;
|
|
inline void SparsityDescriptor::clear_m() {
|
|
_impl_.m_ = 0;
|
|
}
|
|
inline int32_t SparsityDescriptor::_internal_m() const {
|
|
return _impl_.m_;
|
|
}
|
|
inline int32_t SparsityDescriptor::m() const {
|
|
// @@protoc_insertion_point(field_get:xla.SparsityDescriptor.m)
|
|
return _internal_m();
|
|
}
|
|
inline void SparsityDescriptor::_internal_set_m(int32_t value) {
|
|
|
|
_impl_.m_ = value;
|
|
}
|
|
inline void SparsityDescriptor::set_m(int32_t value) {
|
|
_internal_set_m(value);
|
|
// @@protoc_insertion_point(field_set:xla.SparsityDescriptor.m)
|
|
}
|
|
|
|
// -------------------------------------------------------------------
|
|
|
|
// TriangularSolveOptions
|
|
|
|
// bool left_side = 1;
|
|
inline void TriangularSolveOptions::clear_left_side() {
|
|
_impl_.left_side_ = false;
|
|
}
|
|
inline bool TriangularSolveOptions::_internal_left_side() const {
|
|
return _impl_.left_side_;
|
|
}
|
|
inline bool TriangularSolveOptions::left_side() const {
|
|
// @@protoc_insertion_point(field_get:xla.TriangularSolveOptions.left_side)
|
|
return _internal_left_side();
|
|
}
|
|
inline void TriangularSolveOptions::_internal_set_left_side(bool value) {
|
|
|
|
_impl_.left_side_ = value;
|
|
}
|
|
inline void TriangularSolveOptions::set_left_side(bool value) {
|
|
_internal_set_left_side(value);
|
|
// @@protoc_insertion_point(field_set:xla.TriangularSolveOptions.left_side)
|
|
}
|
|
|
|
// bool lower = 2;
|
|
inline void TriangularSolveOptions::clear_lower() {
|
|
_impl_.lower_ = false;
|
|
}
|
|
inline bool TriangularSolveOptions::_internal_lower() const {
|
|
return _impl_.lower_;
|
|
}
|
|
inline bool TriangularSolveOptions::lower() const {
|
|
// @@protoc_insertion_point(field_get:xla.TriangularSolveOptions.lower)
|
|
return _internal_lower();
|
|
}
|
|
inline void TriangularSolveOptions::_internal_set_lower(bool value) {
|
|
|
|
_impl_.lower_ = value;
|
|
}
|
|
inline void TriangularSolveOptions::set_lower(bool value) {
|
|
_internal_set_lower(value);
|
|
// @@protoc_insertion_point(field_set:xla.TriangularSolveOptions.lower)
|
|
}
|
|
|
|
// bool unit_diagonal = 3;
|
|
inline void TriangularSolveOptions::clear_unit_diagonal() {
|
|
_impl_.unit_diagonal_ = false;
|
|
}
|
|
inline bool TriangularSolveOptions::_internal_unit_diagonal() const {
|
|
return _impl_.unit_diagonal_;
|
|
}
|
|
inline bool TriangularSolveOptions::unit_diagonal() const {
|
|
// @@protoc_insertion_point(field_get:xla.TriangularSolveOptions.unit_diagonal)
|
|
return _internal_unit_diagonal();
|
|
}
|
|
inline void TriangularSolveOptions::_internal_set_unit_diagonal(bool value) {
|
|
|
|
_impl_.unit_diagonal_ = value;
|
|
}
|
|
inline void TriangularSolveOptions::set_unit_diagonal(bool value) {
|
|
_internal_set_unit_diagonal(value);
|
|
// @@protoc_insertion_point(field_set:xla.TriangularSolveOptions.unit_diagonal)
|
|
}
|
|
|
|
// .xla.TriangularSolveOptions.Transpose transpose_a = 4;
|
|
inline void TriangularSolveOptions::clear_transpose_a() {
|
|
_impl_.transpose_a_ = 0;
|
|
}
|
|
inline ::xla::TriangularSolveOptions_Transpose TriangularSolveOptions::_internal_transpose_a() const {
|
|
return static_cast< ::xla::TriangularSolveOptions_Transpose >(_impl_.transpose_a_);
|
|
}
|
|
inline ::xla::TriangularSolveOptions_Transpose TriangularSolveOptions::transpose_a() const {
|
|
// @@protoc_insertion_point(field_get:xla.TriangularSolveOptions.transpose_a)
|
|
return _internal_transpose_a();
|
|
}
|
|
inline void TriangularSolveOptions::_internal_set_transpose_a(::xla::TriangularSolveOptions_Transpose value) {
|
|
|
|
_impl_.transpose_a_ = value;
|
|
}
|
|
inline void TriangularSolveOptions::set_transpose_a(::xla::TriangularSolveOptions_Transpose value) {
|
|
_internal_set_transpose_a(value);
|
|
// @@protoc_insertion_point(field_set:xla.TriangularSolveOptions.transpose_a)
|
|
}
|
|
|
|
// -------------------------------------------------------------------
|
|
|
|
// CholeskyOptions
|
|
|
|
// bool lower = 1;
|
|
inline void CholeskyOptions::clear_lower() {
|
|
_impl_.lower_ = false;
|
|
}
|
|
inline bool CholeskyOptions::_internal_lower() const {
|
|
return _impl_.lower_;
|
|
}
|
|
inline bool CholeskyOptions::lower() const {
|
|
// @@protoc_insertion_point(field_get:xla.CholeskyOptions.lower)
|
|
return _internal_lower();
|
|
}
|
|
inline void CholeskyOptions::_internal_set_lower(bool value) {
|
|
|
|
_impl_.lower_ = value;
|
|
}
|
|
inline void CholeskyOptions::set_lower(bool value) {
|
|
_internal_set_lower(value);
|
|
// @@protoc_insertion_point(field_set:xla.CholeskyOptions.lower)
|
|
}
|
|
|
|
// -------------------------------------------------------------------
|
|
|
|
// SortOptions
|
|
|
|
// bool descending = 1;
|
|
inline void SortOptions::clear_descending() {
|
|
_impl_.descending_ = false;
|
|
}
|
|
inline bool SortOptions::_internal_descending() const {
|
|
return _impl_.descending_;
|
|
}
|
|
inline bool SortOptions::descending() const {
|
|
// @@protoc_insertion_point(field_get:xla.SortOptions.descending)
|
|
return _internal_descending();
|
|
}
|
|
inline void SortOptions::_internal_set_descending(bool value) {
|
|
|
|
_impl_.descending_ = value;
|
|
}
|
|
inline void SortOptions::set_descending(bool value) {
|
|
_internal_set_descending(value);
|
|
// @@protoc_insertion_point(field_set:xla.SortOptions.descending)
|
|
}
|
|
|
|
// -------------------------------------------------------------------
|
|
|
|
// -------------------------------------------------------------------
|
|
|
|
// FrontendAttributes
|
|
|
|
// map<string, string> map = 1;
|
|
inline int FrontendAttributes::_internal_map_size() const {
|
|
return _impl_.map_.size();
|
|
}
|
|
inline int FrontendAttributes::map_size() const {
|
|
return _internal_map_size();
|
|
}
|
|
inline void FrontendAttributes::clear_map() {
|
|
_impl_.map_.Clear();
|
|
}
|
|
inline const ::PROTOBUF_NAMESPACE_ID::Map< std::string, std::string >&
|
|
FrontendAttributes::_internal_map() const {
|
|
return _impl_.map_.GetMap();
|
|
}
|
|
inline const ::PROTOBUF_NAMESPACE_ID::Map< std::string, std::string >&
|
|
FrontendAttributes::map() const {
|
|
// @@protoc_insertion_point(field_map:xla.FrontendAttributes.map)
|
|
return _internal_map();
|
|
}
|
|
inline ::PROTOBUF_NAMESPACE_ID::Map< std::string, std::string >*
|
|
FrontendAttributes::_internal_mutable_map() {
|
|
return _impl_.map_.MutableMap();
|
|
}
|
|
inline ::PROTOBUF_NAMESPACE_ID::Map< std::string, std::string >*
|
|
FrontendAttributes::mutable_map() {
|
|
// @@protoc_insertion_point(field_mutable_map:xla.FrontendAttributes.map)
|
|
return _internal_mutable_map();
|
|
}
|
|
|
|
// -------------------------------------------------------------------
|
|
|
|
// Statistic
|
|
|
|
// string stat_name = 1;
|
|
inline void Statistic::clear_stat_name() {
|
|
_impl_.stat_name_.ClearToEmpty();
|
|
}
|
|
inline const std::string& Statistic::stat_name() const {
|
|
// @@protoc_insertion_point(field_get:xla.Statistic.stat_name)
|
|
return _internal_stat_name();
|
|
}
|
|
template <typename ArgT0, typename... ArgT>
|
|
inline PROTOBUF_ALWAYS_INLINE
|
|
void Statistic::set_stat_name(ArgT0&& arg0, ArgT... args) {
|
|
|
|
_impl_.stat_name_.Set(static_cast<ArgT0 &&>(arg0), args..., GetArenaForAllocation());
|
|
// @@protoc_insertion_point(field_set:xla.Statistic.stat_name)
|
|
}
|
|
inline std::string* Statistic::mutable_stat_name() {
|
|
std::string* _s = _internal_mutable_stat_name();
|
|
// @@protoc_insertion_point(field_mutable:xla.Statistic.stat_name)
|
|
return _s;
|
|
}
|
|
inline const std::string& Statistic::_internal_stat_name() const {
|
|
return _impl_.stat_name_.Get();
|
|
}
|
|
inline void Statistic::_internal_set_stat_name(const std::string& value) {
|
|
|
|
_impl_.stat_name_.Set(value, GetArenaForAllocation());
|
|
}
|
|
inline std::string* Statistic::_internal_mutable_stat_name() {
|
|
|
|
return _impl_.stat_name_.Mutable(GetArenaForAllocation());
|
|
}
|
|
inline std::string* Statistic::release_stat_name() {
|
|
// @@protoc_insertion_point(field_release:xla.Statistic.stat_name)
|
|
return _impl_.stat_name_.Release();
|
|
}
|
|
inline void Statistic::set_allocated_stat_name(std::string* stat_name) {
|
|
if (stat_name != nullptr) {
|
|
|
|
} else {
|
|
|
|
}
|
|
_impl_.stat_name_.SetAllocated(stat_name, GetArenaForAllocation());
|
|
#ifdef PROTOBUF_FORCE_COPY_DEFAULT_STRING
|
|
if (_impl_.stat_name_.IsDefault()) {
|
|
_impl_.stat_name_.Set("", GetArenaForAllocation());
|
|
}
|
|
#endif // PROTOBUF_FORCE_COPY_DEFAULT_STRING
|
|
// @@protoc_insertion_point(field_set_allocated:xla.Statistic.stat_name)
|
|
}
|
|
|
|
// double stat_val = 2;
|
|
inline void Statistic::clear_stat_val() {
|
|
_impl_.stat_val_ = 0;
|
|
}
|
|
inline double Statistic::_internal_stat_val() const {
|
|
return _impl_.stat_val_;
|
|
}
|
|
inline double Statistic::stat_val() const {
|
|
// @@protoc_insertion_point(field_get:xla.Statistic.stat_val)
|
|
return _internal_stat_val();
|
|
}
|
|
inline void Statistic::_internal_set_stat_val(double value) {
|
|
|
|
_impl_.stat_val_ = value;
|
|
}
|
|
inline void Statistic::set_stat_val(double value) {
|
|
_internal_set_stat_val(value);
|
|
// @@protoc_insertion_point(field_set:xla.Statistic.stat_val)
|
|
}
|
|
|
|
// -------------------------------------------------------------------
|
|
|
|
// StatisticsViz
|
|
|
|
// int64 stat_index_to_visualize = 1;
|
|
inline void StatisticsViz::clear_stat_index_to_visualize() {
|
|
_impl_.stat_index_to_visualize_ = int64_t{0};
|
|
}
|
|
inline int64_t StatisticsViz::_internal_stat_index_to_visualize() const {
|
|
return _impl_.stat_index_to_visualize_;
|
|
}
|
|
inline int64_t StatisticsViz::stat_index_to_visualize() const {
|
|
// @@protoc_insertion_point(field_get:xla.StatisticsViz.stat_index_to_visualize)
|
|
return _internal_stat_index_to_visualize();
|
|
}
|
|
inline void StatisticsViz::_internal_set_stat_index_to_visualize(int64_t value) {
|
|
|
|
_impl_.stat_index_to_visualize_ = value;
|
|
}
|
|
inline void StatisticsViz::set_stat_index_to_visualize(int64_t value) {
|
|
_internal_set_stat_index_to_visualize(value);
|
|
// @@protoc_insertion_point(field_set:xla.StatisticsViz.stat_index_to_visualize)
|
|
}
|
|
|
|
// repeated .xla.Statistic statistics = 2;
|
|
inline int StatisticsViz::_internal_statistics_size() const {
|
|
return _impl_.statistics_.size();
|
|
}
|
|
inline int StatisticsViz::statistics_size() const {
|
|
return _internal_statistics_size();
|
|
}
|
|
inline void StatisticsViz::clear_statistics() {
|
|
_impl_.statistics_.Clear();
|
|
}
|
|
inline ::xla::Statistic* StatisticsViz::mutable_statistics(int index) {
|
|
// @@protoc_insertion_point(field_mutable:xla.StatisticsViz.statistics)
|
|
return _impl_.statistics_.Mutable(index);
|
|
}
|
|
inline ::PROTOBUF_NAMESPACE_ID::RepeatedPtrField< ::xla::Statistic >*
|
|
StatisticsViz::mutable_statistics() {
|
|
// @@protoc_insertion_point(field_mutable_list:xla.StatisticsViz.statistics)
|
|
return &_impl_.statistics_;
|
|
}
|
|
inline const ::xla::Statistic& StatisticsViz::_internal_statistics(int index) const {
|
|
return _impl_.statistics_.Get(index);
|
|
}
|
|
inline const ::xla::Statistic& StatisticsViz::statistics(int index) const {
|
|
// @@protoc_insertion_point(field_get:xla.StatisticsViz.statistics)
|
|
return _internal_statistics(index);
|
|
}
|
|
inline ::xla::Statistic* StatisticsViz::_internal_add_statistics() {
|
|
return _impl_.statistics_.Add();
|
|
}
|
|
inline ::xla::Statistic* StatisticsViz::add_statistics() {
|
|
::xla::Statistic* _add = _internal_add_statistics();
|
|
// @@protoc_insertion_point(field_add:xla.StatisticsViz.statistics)
|
|
return _add;
|
|
}
|
|
inline const ::PROTOBUF_NAMESPACE_ID::RepeatedPtrField< ::xla::Statistic >&
|
|
StatisticsViz::statistics() const {
|
|
// @@protoc_insertion_point(field_list:xla.StatisticsViz.statistics)
|
|
return _impl_.statistics_;
|
|
}
|
|
|
|
// -------------------------------------------------------------------
|
|
|
|
// OpSharding
|
|
|
|
// .xla.OpSharding.Type type = 1;
|
|
inline void OpSharding::clear_type() {
|
|
_impl_.type_ = 0;
|
|
}
|
|
inline ::xla::OpSharding_Type OpSharding::_internal_type() const {
|
|
return static_cast< ::xla::OpSharding_Type >(_impl_.type_);
|
|
}
|
|
inline ::xla::OpSharding_Type OpSharding::type() const {
|
|
// @@protoc_insertion_point(field_get:xla.OpSharding.type)
|
|
return _internal_type();
|
|
}
|
|
inline void OpSharding::_internal_set_type(::xla::OpSharding_Type value) {
|
|
|
|
_impl_.type_ = value;
|
|
}
|
|
inline void OpSharding::set_type(::xla::OpSharding_Type value) {
|
|
_internal_set_type(value);
|
|
// @@protoc_insertion_point(field_set:xla.OpSharding.type)
|
|
}
|
|
|
|
// .xla.ShapeProto tile_shape = 2;
|
|
inline bool OpSharding::_internal_has_tile_shape() const {
|
|
return this != internal_default_instance() && _impl_.tile_shape_ != nullptr;
|
|
}
|
|
inline bool OpSharding::has_tile_shape() const {
|
|
return _internal_has_tile_shape();
|
|
}
|
|
inline void OpSharding::clear_tile_shape() {
|
|
if (GetArenaForAllocation() == nullptr && _impl_.tile_shape_ != nullptr) {
|
|
delete _impl_.tile_shape_;
|
|
}
|
|
_impl_.tile_shape_ = nullptr;
|
|
}
|
|
inline const ::xla::ShapeProto& OpSharding::_internal_tile_shape() const {
|
|
const ::xla::ShapeProto* p = _impl_.tile_shape_;
|
|
return p != nullptr ? *p : reinterpret_cast<const ::xla::ShapeProto&>(
|
|
::xla::_ShapeProto_default_instance_);
|
|
}
|
|
inline const ::xla::ShapeProto& OpSharding::tile_shape() const {
|
|
// @@protoc_insertion_point(field_get:xla.OpSharding.tile_shape)
|
|
return _internal_tile_shape();
|
|
}
|
|
inline void OpSharding::unsafe_arena_set_allocated_tile_shape(
|
|
::xla::ShapeProto* tile_shape) {
|
|
if (GetArenaForAllocation() == nullptr) {
|
|
delete reinterpret_cast<::PROTOBUF_NAMESPACE_ID::MessageLite*>(_impl_.tile_shape_);
|
|
}
|
|
_impl_.tile_shape_ = tile_shape;
|
|
if (tile_shape) {
|
|
|
|
} else {
|
|
|
|
}
|
|
// @@protoc_insertion_point(field_unsafe_arena_set_allocated:xla.OpSharding.tile_shape)
|
|
}
|
|
inline ::xla::ShapeProto* OpSharding::release_tile_shape() {
|
|
|
|
::xla::ShapeProto* temp = _impl_.tile_shape_;
|
|
_impl_.tile_shape_ = nullptr;
|
|
#ifdef PROTOBUF_FORCE_COPY_IN_RELEASE
|
|
auto* old = reinterpret_cast<::PROTOBUF_NAMESPACE_ID::MessageLite*>(temp);
|
|
temp = ::PROTOBUF_NAMESPACE_ID::internal::DuplicateIfNonNull(temp);
|
|
if (GetArenaForAllocation() == nullptr) { delete old; }
|
|
#else // PROTOBUF_FORCE_COPY_IN_RELEASE
|
|
if (GetArenaForAllocation() != nullptr) {
|
|
temp = ::PROTOBUF_NAMESPACE_ID::internal::DuplicateIfNonNull(temp);
|
|
}
|
|
#endif // !PROTOBUF_FORCE_COPY_IN_RELEASE
|
|
return temp;
|
|
}
|
|
inline ::xla::ShapeProto* OpSharding::unsafe_arena_release_tile_shape() {
|
|
// @@protoc_insertion_point(field_release:xla.OpSharding.tile_shape)
|
|
|
|
::xla::ShapeProto* temp = _impl_.tile_shape_;
|
|
_impl_.tile_shape_ = nullptr;
|
|
return temp;
|
|
}
|
|
inline ::xla::ShapeProto* OpSharding::_internal_mutable_tile_shape() {
|
|
|
|
if (_impl_.tile_shape_ == nullptr) {
|
|
auto* p = CreateMaybeMessage<::xla::ShapeProto>(GetArenaForAllocation());
|
|
_impl_.tile_shape_ = p;
|
|
}
|
|
return _impl_.tile_shape_;
|
|
}
|
|
inline ::xla::ShapeProto* OpSharding::mutable_tile_shape() {
|
|
::xla::ShapeProto* _msg = _internal_mutable_tile_shape();
|
|
// @@protoc_insertion_point(field_mutable:xla.OpSharding.tile_shape)
|
|
return _msg;
|
|
}
|
|
inline void OpSharding::set_allocated_tile_shape(::xla::ShapeProto* tile_shape) {
|
|
::PROTOBUF_NAMESPACE_ID::Arena* message_arena = GetArenaForAllocation();
|
|
if (message_arena == nullptr) {
|
|
delete _impl_.tile_shape_;
|
|
}
|
|
if (tile_shape) {
|
|
::PROTOBUF_NAMESPACE_ID::Arena* submessage_arena =
|
|
::PROTOBUF_NAMESPACE_ID::Arena::InternalGetOwningArena(tile_shape);
|
|
if (message_arena != submessage_arena) {
|
|
tile_shape = ::PROTOBUF_NAMESPACE_ID::internal::GetOwnedMessage(
|
|
message_arena, tile_shape, submessage_arena);
|
|
}
|
|
|
|
} else {
|
|
|
|
}
|
|
_impl_.tile_shape_ = tile_shape;
|
|
// @@protoc_insertion_point(field_set_allocated:xla.OpSharding.tile_shape)
|
|
}
|
|
|
|
// repeated int64 tile_assignment_dimensions = 3;
|
|
inline int OpSharding::_internal_tile_assignment_dimensions_size() const {
|
|
return _impl_.tile_assignment_dimensions_.size();
|
|
}
|
|
inline int OpSharding::tile_assignment_dimensions_size() const {
|
|
return _internal_tile_assignment_dimensions_size();
|
|
}
|
|
inline void OpSharding::clear_tile_assignment_dimensions() {
|
|
_impl_.tile_assignment_dimensions_.Clear();
|
|
}
|
|
inline int64_t OpSharding::_internal_tile_assignment_dimensions(int index) const {
|
|
return _impl_.tile_assignment_dimensions_.Get(index);
|
|
}
|
|
inline int64_t OpSharding::tile_assignment_dimensions(int index) const {
|
|
// @@protoc_insertion_point(field_get:xla.OpSharding.tile_assignment_dimensions)
|
|
return _internal_tile_assignment_dimensions(index);
|
|
}
|
|
inline void OpSharding::set_tile_assignment_dimensions(int index, int64_t value) {
|
|
_impl_.tile_assignment_dimensions_.Set(index, value);
|
|
// @@protoc_insertion_point(field_set:xla.OpSharding.tile_assignment_dimensions)
|
|
}
|
|
inline void OpSharding::_internal_add_tile_assignment_dimensions(int64_t value) {
|
|
_impl_.tile_assignment_dimensions_.Add(value);
|
|
}
|
|
inline void OpSharding::add_tile_assignment_dimensions(int64_t value) {
|
|
_internal_add_tile_assignment_dimensions(value);
|
|
// @@protoc_insertion_point(field_add:xla.OpSharding.tile_assignment_dimensions)
|
|
}
|
|
inline const ::PROTOBUF_NAMESPACE_ID::RepeatedField< int64_t >&
|
|
OpSharding::_internal_tile_assignment_dimensions() const {
|
|
return _impl_.tile_assignment_dimensions_;
|
|
}
|
|
inline const ::PROTOBUF_NAMESPACE_ID::RepeatedField< int64_t >&
|
|
OpSharding::tile_assignment_dimensions() const {
|
|
// @@protoc_insertion_point(field_list:xla.OpSharding.tile_assignment_dimensions)
|
|
return _internal_tile_assignment_dimensions();
|
|
}
|
|
inline ::PROTOBUF_NAMESPACE_ID::RepeatedField< int64_t >*
|
|
OpSharding::_internal_mutable_tile_assignment_dimensions() {
|
|
return &_impl_.tile_assignment_dimensions_;
|
|
}
|
|
inline ::PROTOBUF_NAMESPACE_ID::RepeatedField< int64_t >*
|
|
OpSharding::mutable_tile_assignment_dimensions() {
|
|
// @@protoc_insertion_point(field_mutable_list:xla.OpSharding.tile_assignment_dimensions)
|
|
return _internal_mutable_tile_assignment_dimensions();
|
|
}
|
|
|
|
// repeated int64 tile_assignment_devices = 4;
|
|
inline int OpSharding::_internal_tile_assignment_devices_size() const {
|
|
return _impl_.tile_assignment_devices_.size();
|
|
}
|
|
inline int OpSharding::tile_assignment_devices_size() const {
|
|
return _internal_tile_assignment_devices_size();
|
|
}
|
|
inline void OpSharding::clear_tile_assignment_devices() {
|
|
_impl_.tile_assignment_devices_.Clear();
|
|
}
|
|
inline int64_t OpSharding::_internal_tile_assignment_devices(int index) const {
|
|
return _impl_.tile_assignment_devices_.Get(index);
|
|
}
|
|
inline int64_t OpSharding::tile_assignment_devices(int index) const {
|
|
// @@protoc_insertion_point(field_get:xla.OpSharding.tile_assignment_devices)
|
|
return _internal_tile_assignment_devices(index);
|
|
}
|
|
inline void OpSharding::set_tile_assignment_devices(int index, int64_t value) {
|
|
_impl_.tile_assignment_devices_.Set(index, value);
|
|
// @@protoc_insertion_point(field_set:xla.OpSharding.tile_assignment_devices)
|
|
}
|
|
inline void OpSharding::_internal_add_tile_assignment_devices(int64_t value) {
|
|
_impl_.tile_assignment_devices_.Add(value);
|
|
}
|
|
inline void OpSharding::add_tile_assignment_devices(int64_t value) {
|
|
_internal_add_tile_assignment_devices(value);
|
|
// @@protoc_insertion_point(field_add:xla.OpSharding.tile_assignment_devices)
|
|
}
|
|
inline const ::PROTOBUF_NAMESPACE_ID::RepeatedField< int64_t >&
|
|
OpSharding::_internal_tile_assignment_devices() const {
|
|
return _impl_.tile_assignment_devices_;
|
|
}
|
|
inline const ::PROTOBUF_NAMESPACE_ID::RepeatedField< int64_t >&
|
|
OpSharding::tile_assignment_devices() const {
|
|
// @@protoc_insertion_point(field_list:xla.OpSharding.tile_assignment_devices)
|
|
return _internal_tile_assignment_devices();
|
|
}
|
|
inline ::PROTOBUF_NAMESPACE_ID::RepeatedField< int64_t >*
|
|
OpSharding::_internal_mutable_tile_assignment_devices() {
|
|
return &_impl_.tile_assignment_devices_;
|
|
}
|
|
inline ::PROTOBUF_NAMESPACE_ID::RepeatedField< int64_t >*
|
|
OpSharding::mutable_tile_assignment_devices() {
|
|
// @@protoc_insertion_point(field_mutable_list:xla.OpSharding.tile_assignment_devices)
|
|
return _internal_mutable_tile_assignment_devices();
|
|
}
|
|
|
|
// repeated .xla.OpSharding tuple_shardings = 5;
|
|
inline int OpSharding::_internal_tuple_shardings_size() const {
|
|
return _impl_.tuple_shardings_.size();
|
|
}
|
|
inline int OpSharding::tuple_shardings_size() const {
|
|
return _internal_tuple_shardings_size();
|
|
}
|
|
inline void OpSharding::clear_tuple_shardings() {
|
|
_impl_.tuple_shardings_.Clear();
|
|
}
|
|
inline ::xla::OpSharding* OpSharding::mutable_tuple_shardings(int index) {
|
|
// @@protoc_insertion_point(field_mutable:xla.OpSharding.tuple_shardings)
|
|
return _impl_.tuple_shardings_.Mutable(index);
|
|
}
|
|
inline ::PROTOBUF_NAMESPACE_ID::RepeatedPtrField< ::xla::OpSharding >*
|
|
OpSharding::mutable_tuple_shardings() {
|
|
// @@protoc_insertion_point(field_mutable_list:xla.OpSharding.tuple_shardings)
|
|
return &_impl_.tuple_shardings_;
|
|
}
|
|
inline const ::xla::OpSharding& OpSharding::_internal_tuple_shardings(int index) const {
|
|
return _impl_.tuple_shardings_.Get(index);
|
|
}
|
|
inline const ::xla::OpSharding& OpSharding::tuple_shardings(int index) const {
|
|
// @@protoc_insertion_point(field_get:xla.OpSharding.tuple_shardings)
|
|
return _internal_tuple_shardings(index);
|
|
}
|
|
inline ::xla::OpSharding* OpSharding::_internal_add_tuple_shardings() {
|
|
return _impl_.tuple_shardings_.Add();
|
|
}
|
|
inline ::xla::OpSharding* OpSharding::add_tuple_shardings() {
|
|
::xla::OpSharding* _add = _internal_add_tuple_shardings();
|
|
// @@protoc_insertion_point(field_add:xla.OpSharding.tuple_shardings)
|
|
return _add;
|
|
}
|
|
inline const ::PROTOBUF_NAMESPACE_ID::RepeatedPtrField< ::xla::OpSharding >&
|
|
OpSharding::tuple_shardings() const {
|
|
// @@protoc_insertion_point(field_list:xla.OpSharding.tuple_shardings)
|
|
return _impl_.tuple_shardings_;
|
|
}
|
|
|
|
// bool replicate_on_last_tile_dim = 6;
|
|
inline void OpSharding::clear_replicate_on_last_tile_dim() {
|
|
_impl_.replicate_on_last_tile_dim_ = false;
|
|
}
|
|
inline bool OpSharding::_internal_replicate_on_last_tile_dim() const {
|
|
return _impl_.replicate_on_last_tile_dim_;
|
|
}
|
|
inline bool OpSharding::replicate_on_last_tile_dim() const {
|
|
// @@protoc_insertion_point(field_get:xla.OpSharding.replicate_on_last_tile_dim)
|
|
return _internal_replicate_on_last_tile_dim();
|
|
}
|
|
inline void OpSharding::_internal_set_replicate_on_last_tile_dim(bool value) {
|
|
|
|
_impl_.replicate_on_last_tile_dim_ = value;
|
|
}
|
|
inline void OpSharding::set_replicate_on_last_tile_dim(bool value) {
|
|
_internal_set_replicate_on_last_tile_dim(value);
|
|
// @@protoc_insertion_point(field_set:xla.OpSharding.replicate_on_last_tile_dim)
|
|
}
|
|
|
|
// repeated .xla.OpMetadata metadata = 7;
|
|
inline int OpSharding::_internal_metadata_size() const {
|
|
return _impl_.metadata_.size();
|
|
}
|
|
inline int OpSharding::metadata_size() const {
|
|
return _internal_metadata_size();
|
|
}
|
|
inline void OpSharding::clear_metadata() {
|
|
_impl_.metadata_.Clear();
|
|
}
|
|
inline ::xla::OpMetadata* OpSharding::mutable_metadata(int index) {
|
|
// @@protoc_insertion_point(field_mutable:xla.OpSharding.metadata)
|
|
return _impl_.metadata_.Mutable(index);
|
|
}
|
|
inline ::PROTOBUF_NAMESPACE_ID::RepeatedPtrField< ::xla::OpMetadata >*
|
|
OpSharding::mutable_metadata() {
|
|
// @@protoc_insertion_point(field_mutable_list:xla.OpSharding.metadata)
|
|
return &_impl_.metadata_;
|
|
}
|
|
inline const ::xla::OpMetadata& OpSharding::_internal_metadata(int index) const {
|
|
return _impl_.metadata_.Get(index);
|
|
}
|
|
inline const ::xla::OpMetadata& OpSharding::metadata(int index) const {
|
|
// @@protoc_insertion_point(field_get:xla.OpSharding.metadata)
|
|
return _internal_metadata(index);
|
|
}
|
|
inline ::xla::OpMetadata* OpSharding::_internal_add_metadata() {
|
|
return _impl_.metadata_.Add();
|
|
}
|
|
inline ::xla::OpMetadata* OpSharding::add_metadata() {
|
|
::xla::OpMetadata* _add = _internal_add_metadata();
|
|
// @@protoc_insertion_point(field_add:xla.OpSharding.metadata)
|
|
return _add;
|
|
}
|
|
inline const ::PROTOBUF_NAMESPACE_ID::RepeatedPtrField< ::xla::OpMetadata >&
|
|
OpSharding::metadata() const {
|
|
// @@protoc_insertion_point(field_list:xla.OpSharding.metadata)
|
|
return _impl_.metadata_;
|
|
}
|
|
|
|
// repeated .xla.OpSharding.Type last_tile_dims = 8;
|
|
inline int OpSharding::_internal_last_tile_dims_size() const {
|
|
return _impl_.last_tile_dims_.size();
|
|
}
|
|
inline int OpSharding::last_tile_dims_size() const {
|
|
return _internal_last_tile_dims_size();
|
|
}
|
|
inline void OpSharding::clear_last_tile_dims() {
|
|
_impl_.last_tile_dims_.Clear();
|
|
}
|
|
inline ::xla::OpSharding_Type OpSharding::_internal_last_tile_dims(int index) const {
|
|
return static_cast< ::xla::OpSharding_Type >(_impl_.last_tile_dims_.Get(index));
|
|
}
|
|
inline ::xla::OpSharding_Type OpSharding::last_tile_dims(int index) const {
|
|
// @@protoc_insertion_point(field_get:xla.OpSharding.last_tile_dims)
|
|
return _internal_last_tile_dims(index);
|
|
}
|
|
inline void OpSharding::set_last_tile_dims(int index, ::xla::OpSharding_Type value) {
|
|
_impl_.last_tile_dims_.Set(index, value);
|
|
// @@protoc_insertion_point(field_set:xla.OpSharding.last_tile_dims)
|
|
}
|
|
inline void OpSharding::_internal_add_last_tile_dims(::xla::OpSharding_Type value) {
|
|
_impl_.last_tile_dims_.Add(value);
|
|
}
|
|
inline void OpSharding::add_last_tile_dims(::xla::OpSharding_Type value) {
|
|
_internal_add_last_tile_dims(value);
|
|
// @@protoc_insertion_point(field_add:xla.OpSharding.last_tile_dims)
|
|
}
|
|
inline const ::PROTOBUF_NAMESPACE_ID::RepeatedField<int>&
|
|
OpSharding::last_tile_dims() const {
|
|
// @@protoc_insertion_point(field_list:xla.OpSharding.last_tile_dims)
|
|
return _impl_.last_tile_dims_;
|
|
}
|
|
inline ::PROTOBUF_NAMESPACE_ID::RepeatedField<int>*
|
|
OpSharding::_internal_mutable_last_tile_dims() {
|
|
return &_impl_.last_tile_dims_;
|
|
}
|
|
inline ::PROTOBUF_NAMESPACE_ID::RepeatedField<int>*
|
|
OpSharding::mutable_last_tile_dims() {
|
|
// @@protoc_insertion_point(field_mutable_list:xla.OpSharding.last_tile_dims)
|
|
return _internal_mutable_last_tile_dims();
|
|
}
|
|
|
|
// repeated int64 iota_reshape_dims = 9;
|
|
inline int OpSharding::_internal_iota_reshape_dims_size() const {
|
|
return _impl_.iota_reshape_dims_.size();
|
|
}
|
|
inline int OpSharding::iota_reshape_dims_size() const {
|
|
return _internal_iota_reshape_dims_size();
|
|
}
|
|
inline void OpSharding::clear_iota_reshape_dims() {
|
|
_impl_.iota_reshape_dims_.Clear();
|
|
}
|
|
inline int64_t OpSharding::_internal_iota_reshape_dims(int index) const {
|
|
return _impl_.iota_reshape_dims_.Get(index);
|
|
}
|
|
inline int64_t OpSharding::iota_reshape_dims(int index) const {
|
|
// @@protoc_insertion_point(field_get:xla.OpSharding.iota_reshape_dims)
|
|
return _internal_iota_reshape_dims(index);
|
|
}
|
|
inline void OpSharding::set_iota_reshape_dims(int index, int64_t value) {
|
|
_impl_.iota_reshape_dims_.Set(index, value);
|
|
// @@protoc_insertion_point(field_set:xla.OpSharding.iota_reshape_dims)
|
|
}
|
|
inline void OpSharding::_internal_add_iota_reshape_dims(int64_t value) {
|
|
_impl_.iota_reshape_dims_.Add(value);
|
|
}
|
|
inline void OpSharding::add_iota_reshape_dims(int64_t value) {
|
|
_internal_add_iota_reshape_dims(value);
|
|
// @@protoc_insertion_point(field_add:xla.OpSharding.iota_reshape_dims)
|
|
}
|
|
inline const ::PROTOBUF_NAMESPACE_ID::RepeatedField< int64_t >&
|
|
OpSharding::_internal_iota_reshape_dims() const {
|
|
return _impl_.iota_reshape_dims_;
|
|
}
|
|
inline const ::PROTOBUF_NAMESPACE_ID::RepeatedField< int64_t >&
|
|
OpSharding::iota_reshape_dims() const {
|
|
// @@protoc_insertion_point(field_list:xla.OpSharding.iota_reshape_dims)
|
|
return _internal_iota_reshape_dims();
|
|
}
|
|
inline ::PROTOBUF_NAMESPACE_ID::RepeatedField< int64_t >*
|
|
OpSharding::_internal_mutable_iota_reshape_dims() {
|
|
return &_impl_.iota_reshape_dims_;
|
|
}
|
|
inline ::PROTOBUF_NAMESPACE_ID::RepeatedField< int64_t >*
|
|
OpSharding::mutable_iota_reshape_dims() {
|
|
// @@protoc_insertion_point(field_mutable_list:xla.OpSharding.iota_reshape_dims)
|
|
return _internal_mutable_iota_reshape_dims();
|
|
}
|
|
|
|
// repeated int32 iota_transpose_perm = 10;
|
|
inline int OpSharding::_internal_iota_transpose_perm_size() const {
|
|
return _impl_.iota_transpose_perm_.size();
|
|
}
|
|
inline int OpSharding::iota_transpose_perm_size() const {
|
|
return _internal_iota_transpose_perm_size();
|
|
}
|
|
inline void OpSharding::clear_iota_transpose_perm() {
|
|
_impl_.iota_transpose_perm_.Clear();
|
|
}
|
|
inline int32_t OpSharding::_internal_iota_transpose_perm(int index) const {
|
|
return _impl_.iota_transpose_perm_.Get(index);
|
|
}
|
|
inline int32_t OpSharding::iota_transpose_perm(int index) const {
|
|
// @@protoc_insertion_point(field_get:xla.OpSharding.iota_transpose_perm)
|
|
return _internal_iota_transpose_perm(index);
|
|
}
|
|
inline void OpSharding::set_iota_transpose_perm(int index, int32_t value) {
|
|
_impl_.iota_transpose_perm_.Set(index, value);
|
|
// @@protoc_insertion_point(field_set:xla.OpSharding.iota_transpose_perm)
|
|
}
|
|
inline void OpSharding::_internal_add_iota_transpose_perm(int32_t value) {
|
|
_impl_.iota_transpose_perm_.Add(value);
|
|
}
|
|
inline void OpSharding::add_iota_transpose_perm(int32_t value) {
|
|
_internal_add_iota_transpose_perm(value);
|
|
// @@protoc_insertion_point(field_add:xla.OpSharding.iota_transpose_perm)
|
|
}
|
|
inline const ::PROTOBUF_NAMESPACE_ID::RepeatedField< int32_t >&
|
|
OpSharding::_internal_iota_transpose_perm() const {
|
|
return _impl_.iota_transpose_perm_;
|
|
}
|
|
inline const ::PROTOBUF_NAMESPACE_ID::RepeatedField< int32_t >&
|
|
OpSharding::iota_transpose_perm() const {
|
|
// @@protoc_insertion_point(field_list:xla.OpSharding.iota_transpose_perm)
|
|
return _internal_iota_transpose_perm();
|
|
}
|
|
inline ::PROTOBUF_NAMESPACE_ID::RepeatedField< int32_t >*
|
|
OpSharding::_internal_mutable_iota_transpose_perm() {
|
|
return &_impl_.iota_transpose_perm_;
|
|
}
|
|
inline ::PROTOBUF_NAMESPACE_ID::RepeatedField< int32_t >*
|
|
OpSharding::mutable_iota_transpose_perm() {
|
|
// @@protoc_insertion_point(field_mutable_list:xla.OpSharding.iota_transpose_perm)
|
|
return _internal_mutable_iota_transpose_perm();
|
|
}
|
|
|
|
// bool is_shard_group = 11;
|
|
inline void OpSharding::clear_is_shard_group() {
|
|
_impl_.is_shard_group_ = false;
|
|
}
|
|
inline bool OpSharding::_internal_is_shard_group() const {
|
|
return _impl_.is_shard_group_;
|
|
}
|
|
inline bool OpSharding::is_shard_group() const {
|
|
// @@protoc_insertion_point(field_get:xla.OpSharding.is_shard_group)
|
|
return _internal_is_shard_group();
|
|
}
|
|
inline void OpSharding::_internal_set_is_shard_group(bool value) {
|
|
|
|
_impl_.is_shard_group_ = value;
|
|
}
|
|
inline void OpSharding::set_is_shard_group(bool value) {
|
|
_internal_set_is_shard_group(value);
|
|
// @@protoc_insertion_point(field_set:xla.OpSharding.is_shard_group)
|
|
}
|
|
|
|
// int64 shard_group_id = 12;
|
|
inline void OpSharding::clear_shard_group_id() {
|
|
_impl_.shard_group_id_ = int64_t{0};
|
|
}
|
|
inline int64_t OpSharding::_internal_shard_group_id() const {
|
|
return _impl_.shard_group_id_;
|
|
}
|
|
inline int64_t OpSharding::shard_group_id() const {
|
|
// @@protoc_insertion_point(field_get:xla.OpSharding.shard_group_id)
|
|
return _internal_shard_group_id();
|
|
}
|
|
inline void OpSharding::_internal_set_shard_group_id(int64_t value) {
|
|
|
|
_impl_.shard_group_id_ = value;
|
|
}
|
|
inline void OpSharding::set_shard_group_id(int64_t value) {
|
|
_internal_set_shard_group_id(value);
|
|
// @@protoc_insertion_point(field_set:xla.OpSharding.shard_group_id)
|
|
}
|
|
|
|
// .xla.OpSharding.ShardGroupType shard_group_type = 13;
|
|
inline void OpSharding::clear_shard_group_type() {
|
|
_impl_.shard_group_type_ = 0;
|
|
}
|
|
inline ::xla::OpSharding_ShardGroupType OpSharding::_internal_shard_group_type() const {
|
|
return static_cast< ::xla::OpSharding_ShardGroupType >(_impl_.shard_group_type_);
|
|
}
|
|
inline ::xla::OpSharding_ShardGroupType OpSharding::shard_group_type() const {
|
|
// @@protoc_insertion_point(field_get:xla.OpSharding.shard_group_type)
|
|
return _internal_shard_group_type();
|
|
}
|
|
inline void OpSharding::_internal_set_shard_group_type(::xla::OpSharding_ShardGroupType value) {
|
|
|
|
_impl_.shard_group_type_ = value;
|
|
}
|
|
inline void OpSharding::set_shard_group_type(::xla::OpSharding_ShardGroupType value) {
|
|
_internal_set_shard_group_type(value);
|
|
// @@protoc_insertion_point(field_set:xla.OpSharding.shard_group_type)
|
|
}
|
|
|
|
// -------------------------------------------------------------------
|
|
|
|
// ReplicaGroup
|
|
|
|
// repeated int64 replica_ids = 1;
|
|
inline int ReplicaGroup::_internal_replica_ids_size() const {
|
|
return _impl_.replica_ids_.size();
|
|
}
|
|
inline int ReplicaGroup::replica_ids_size() const {
|
|
return _internal_replica_ids_size();
|
|
}
|
|
inline void ReplicaGroup::clear_replica_ids() {
|
|
_impl_.replica_ids_.Clear();
|
|
}
|
|
inline int64_t ReplicaGroup::_internal_replica_ids(int index) const {
|
|
return _impl_.replica_ids_.Get(index);
|
|
}
|
|
inline int64_t ReplicaGroup::replica_ids(int index) const {
|
|
// @@protoc_insertion_point(field_get:xla.ReplicaGroup.replica_ids)
|
|
return _internal_replica_ids(index);
|
|
}
|
|
inline void ReplicaGroup::set_replica_ids(int index, int64_t value) {
|
|
_impl_.replica_ids_.Set(index, value);
|
|
// @@protoc_insertion_point(field_set:xla.ReplicaGroup.replica_ids)
|
|
}
|
|
inline void ReplicaGroup::_internal_add_replica_ids(int64_t value) {
|
|
_impl_.replica_ids_.Add(value);
|
|
}
|
|
inline void ReplicaGroup::add_replica_ids(int64_t value) {
|
|
_internal_add_replica_ids(value);
|
|
// @@protoc_insertion_point(field_add:xla.ReplicaGroup.replica_ids)
|
|
}
|
|
inline const ::PROTOBUF_NAMESPACE_ID::RepeatedField< int64_t >&
|
|
ReplicaGroup::_internal_replica_ids() const {
|
|
return _impl_.replica_ids_;
|
|
}
|
|
inline const ::PROTOBUF_NAMESPACE_ID::RepeatedField< int64_t >&
|
|
ReplicaGroup::replica_ids() const {
|
|
// @@protoc_insertion_point(field_list:xla.ReplicaGroup.replica_ids)
|
|
return _internal_replica_ids();
|
|
}
|
|
inline ::PROTOBUF_NAMESPACE_ID::RepeatedField< int64_t >*
|
|
ReplicaGroup::_internal_mutable_replica_ids() {
|
|
return &_impl_.replica_ids_;
|
|
}
|
|
inline ::PROTOBUF_NAMESPACE_ID::RepeatedField< int64_t >*
|
|
ReplicaGroup::mutable_replica_ids() {
|
|
// @@protoc_insertion_point(field_mutable_list:xla.ReplicaGroup.replica_ids)
|
|
return _internal_mutable_replica_ids();
|
|
}
|
|
|
|
// -------------------------------------------------------------------
|
|
|
|
// SourceTarget
|
|
|
|
// int64 source = 1;
|
|
inline void SourceTarget::clear_source() {
|
|
_impl_.source_ = int64_t{0};
|
|
}
|
|
inline int64_t SourceTarget::_internal_source() const {
|
|
return _impl_.source_;
|
|
}
|
|
inline int64_t SourceTarget::source() const {
|
|
// @@protoc_insertion_point(field_get:xla.SourceTarget.source)
|
|
return _internal_source();
|
|
}
|
|
inline void SourceTarget::_internal_set_source(int64_t value) {
|
|
|
|
_impl_.source_ = value;
|
|
}
|
|
inline void SourceTarget::set_source(int64_t value) {
|
|
_internal_set_source(value);
|
|
// @@protoc_insertion_point(field_set:xla.SourceTarget.source)
|
|
}
|
|
|
|
// int64 target = 2;
|
|
inline void SourceTarget::clear_target() {
|
|
_impl_.target_ = int64_t{0};
|
|
}
|
|
inline int64_t SourceTarget::_internal_target() const {
|
|
return _impl_.target_;
|
|
}
|
|
inline int64_t SourceTarget::target() const {
|
|
// @@protoc_insertion_point(field_get:xla.SourceTarget.target)
|
|
return _internal_target();
|
|
}
|
|
inline void SourceTarget::_internal_set_target(int64_t value) {
|
|
|
|
_impl_.target_ = value;
|
|
}
|
|
inline void SourceTarget::set_target(int64_t value) {
|
|
_internal_set_target(value);
|
|
// @@protoc_insertion_point(field_set:xla.SourceTarget.target)
|
|
}
|
|
|
|
// -------------------------------------------------------------------
|
|
|
|
// PrecisionConfig
|
|
|
|
// repeated .xla.PrecisionConfig.Precision operand_precision = 1;
|
|
inline int PrecisionConfig::_internal_operand_precision_size() const {
|
|
return _impl_.operand_precision_.size();
|
|
}
|
|
inline int PrecisionConfig::operand_precision_size() const {
|
|
return _internal_operand_precision_size();
|
|
}
|
|
inline void PrecisionConfig::clear_operand_precision() {
|
|
_impl_.operand_precision_.Clear();
|
|
}
|
|
inline ::xla::PrecisionConfig_Precision PrecisionConfig::_internal_operand_precision(int index) const {
|
|
return static_cast< ::xla::PrecisionConfig_Precision >(_impl_.operand_precision_.Get(index));
|
|
}
|
|
inline ::xla::PrecisionConfig_Precision PrecisionConfig::operand_precision(int index) const {
|
|
// @@protoc_insertion_point(field_get:xla.PrecisionConfig.operand_precision)
|
|
return _internal_operand_precision(index);
|
|
}
|
|
inline void PrecisionConfig::set_operand_precision(int index, ::xla::PrecisionConfig_Precision value) {
|
|
_impl_.operand_precision_.Set(index, value);
|
|
// @@protoc_insertion_point(field_set:xla.PrecisionConfig.operand_precision)
|
|
}
|
|
inline void PrecisionConfig::_internal_add_operand_precision(::xla::PrecisionConfig_Precision value) {
|
|
_impl_.operand_precision_.Add(value);
|
|
}
|
|
inline void PrecisionConfig::add_operand_precision(::xla::PrecisionConfig_Precision value) {
|
|
_internal_add_operand_precision(value);
|
|
// @@protoc_insertion_point(field_add:xla.PrecisionConfig.operand_precision)
|
|
}
|
|
inline const ::PROTOBUF_NAMESPACE_ID::RepeatedField<int>&
|
|
PrecisionConfig::operand_precision() const {
|
|
// @@protoc_insertion_point(field_list:xla.PrecisionConfig.operand_precision)
|
|
return _impl_.operand_precision_;
|
|
}
|
|
inline ::PROTOBUF_NAMESPACE_ID::RepeatedField<int>*
|
|
PrecisionConfig::_internal_mutable_operand_precision() {
|
|
return &_impl_.operand_precision_;
|
|
}
|
|
inline ::PROTOBUF_NAMESPACE_ID::RepeatedField<int>*
|
|
PrecisionConfig::mutable_operand_precision() {
|
|
// @@protoc_insertion_point(field_mutable_list:xla.PrecisionConfig.operand_precision)
|
|
return _internal_mutable_operand_precision();
|
|
}
|
|
|
|
// -------------------------------------------------------------------
|
|
|
|
// ParameterReplication
|
|
|
|
// repeated bool replicated_at_leaf_buffers = 1;
|
|
inline int ParameterReplication::_internal_replicated_at_leaf_buffers_size() const {
|
|
return _impl_.replicated_at_leaf_buffers_.size();
|
|
}
|
|
inline int ParameterReplication::replicated_at_leaf_buffers_size() const {
|
|
return _internal_replicated_at_leaf_buffers_size();
|
|
}
|
|
inline void ParameterReplication::clear_replicated_at_leaf_buffers() {
|
|
_impl_.replicated_at_leaf_buffers_.Clear();
|
|
}
|
|
inline bool ParameterReplication::_internal_replicated_at_leaf_buffers(int index) const {
|
|
return _impl_.replicated_at_leaf_buffers_.Get(index);
|
|
}
|
|
inline bool ParameterReplication::replicated_at_leaf_buffers(int index) const {
|
|
// @@protoc_insertion_point(field_get:xla.ParameterReplication.replicated_at_leaf_buffers)
|
|
return _internal_replicated_at_leaf_buffers(index);
|
|
}
|
|
inline void ParameterReplication::set_replicated_at_leaf_buffers(int index, bool value) {
|
|
_impl_.replicated_at_leaf_buffers_.Set(index, value);
|
|
// @@protoc_insertion_point(field_set:xla.ParameterReplication.replicated_at_leaf_buffers)
|
|
}
|
|
inline void ParameterReplication::_internal_add_replicated_at_leaf_buffers(bool value) {
|
|
_impl_.replicated_at_leaf_buffers_.Add(value);
|
|
}
|
|
inline void ParameterReplication::add_replicated_at_leaf_buffers(bool value) {
|
|
_internal_add_replicated_at_leaf_buffers(value);
|
|
// @@protoc_insertion_point(field_add:xla.ParameterReplication.replicated_at_leaf_buffers)
|
|
}
|
|
inline const ::PROTOBUF_NAMESPACE_ID::RepeatedField< bool >&
|
|
ParameterReplication::_internal_replicated_at_leaf_buffers() const {
|
|
return _impl_.replicated_at_leaf_buffers_;
|
|
}
|
|
inline const ::PROTOBUF_NAMESPACE_ID::RepeatedField< bool >&
|
|
ParameterReplication::replicated_at_leaf_buffers() const {
|
|
// @@protoc_insertion_point(field_list:xla.ParameterReplication.replicated_at_leaf_buffers)
|
|
return _internal_replicated_at_leaf_buffers();
|
|
}
|
|
inline ::PROTOBUF_NAMESPACE_ID::RepeatedField< bool >*
|
|
ParameterReplication::_internal_mutable_replicated_at_leaf_buffers() {
|
|
return &_impl_.replicated_at_leaf_buffers_;
|
|
}
|
|
inline ::PROTOBUF_NAMESPACE_ID::RepeatedField< bool >*
|
|
ParameterReplication::mutable_replicated_at_leaf_buffers() {
|
|
// @@protoc_insertion_point(field_mutable_list:xla.ParameterReplication.replicated_at_leaf_buffers)
|
|
return _internal_mutable_replicated_at_leaf_buffers();
|
|
}
|
|
|
|
// -------------------------------------------------------------------
|
|
|
|
// WhileLoopBackendConfig_KnownTripCount
|
|
|
|
// int64 n = 1;
|
|
inline void WhileLoopBackendConfig_KnownTripCount::clear_n() {
|
|
_impl_.n_ = int64_t{0};
|
|
}
|
|
inline int64_t WhileLoopBackendConfig_KnownTripCount::_internal_n() const {
|
|
return _impl_.n_;
|
|
}
|
|
inline int64_t WhileLoopBackendConfig_KnownTripCount::n() const {
|
|
// @@protoc_insertion_point(field_get:xla.WhileLoopBackendConfig.KnownTripCount.n)
|
|
return _internal_n();
|
|
}
|
|
inline void WhileLoopBackendConfig_KnownTripCount::_internal_set_n(int64_t value) {
|
|
|
|
_impl_.n_ = value;
|
|
}
|
|
inline void WhileLoopBackendConfig_KnownTripCount::set_n(int64_t value) {
|
|
_internal_set_n(value);
|
|
// @@protoc_insertion_point(field_set:xla.WhileLoopBackendConfig.KnownTripCount.n)
|
|
}
|
|
|
|
// -------------------------------------------------------------------
|
|
|
|
// WhileLoopBackendConfig
|
|
|
|
// .xla.WhileLoopBackendConfig.KnownTripCount known_trip_count = 1;
|
|
inline bool WhileLoopBackendConfig::_internal_has_known_trip_count() const {
|
|
return this != internal_default_instance() && _impl_.known_trip_count_ != nullptr;
|
|
}
|
|
inline bool WhileLoopBackendConfig::has_known_trip_count() const {
|
|
return _internal_has_known_trip_count();
|
|
}
|
|
inline void WhileLoopBackendConfig::clear_known_trip_count() {
|
|
if (GetArenaForAllocation() == nullptr && _impl_.known_trip_count_ != nullptr) {
|
|
delete _impl_.known_trip_count_;
|
|
}
|
|
_impl_.known_trip_count_ = nullptr;
|
|
}
|
|
inline const ::xla::WhileLoopBackendConfig_KnownTripCount& WhileLoopBackendConfig::_internal_known_trip_count() const {
|
|
const ::xla::WhileLoopBackendConfig_KnownTripCount* p = _impl_.known_trip_count_;
|
|
return p != nullptr ? *p : reinterpret_cast<const ::xla::WhileLoopBackendConfig_KnownTripCount&>(
|
|
::xla::_WhileLoopBackendConfig_KnownTripCount_default_instance_);
|
|
}
|
|
inline const ::xla::WhileLoopBackendConfig_KnownTripCount& WhileLoopBackendConfig::known_trip_count() const {
|
|
// @@protoc_insertion_point(field_get:xla.WhileLoopBackendConfig.known_trip_count)
|
|
return _internal_known_trip_count();
|
|
}
|
|
inline void WhileLoopBackendConfig::unsafe_arena_set_allocated_known_trip_count(
|
|
::xla::WhileLoopBackendConfig_KnownTripCount* known_trip_count) {
|
|
if (GetArenaForAllocation() == nullptr) {
|
|
delete reinterpret_cast<::PROTOBUF_NAMESPACE_ID::MessageLite*>(_impl_.known_trip_count_);
|
|
}
|
|
_impl_.known_trip_count_ = known_trip_count;
|
|
if (known_trip_count) {
|
|
|
|
} else {
|
|
|
|
}
|
|
// @@protoc_insertion_point(field_unsafe_arena_set_allocated:xla.WhileLoopBackendConfig.known_trip_count)
|
|
}
|
|
inline ::xla::WhileLoopBackendConfig_KnownTripCount* WhileLoopBackendConfig::release_known_trip_count() {
|
|
|
|
::xla::WhileLoopBackendConfig_KnownTripCount* temp = _impl_.known_trip_count_;
|
|
_impl_.known_trip_count_ = nullptr;
|
|
#ifdef PROTOBUF_FORCE_COPY_IN_RELEASE
|
|
auto* old = reinterpret_cast<::PROTOBUF_NAMESPACE_ID::MessageLite*>(temp);
|
|
temp = ::PROTOBUF_NAMESPACE_ID::internal::DuplicateIfNonNull(temp);
|
|
if (GetArenaForAllocation() == nullptr) { delete old; }
|
|
#else // PROTOBUF_FORCE_COPY_IN_RELEASE
|
|
if (GetArenaForAllocation() != nullptr) {
|
|
temp = ::PROTOBUF_NAMESPACE_ID::internal::DuplicateIfNonNull(temp);
|
|
}
|
|
#endif // !PROTOBUF_FORCE_COPY_IN_RELEASE
|
|
return temp;
|
|
}
|
|
inline ::xla::WhileLoopBackendConfig_KnownTripCount* WhileLoopBackendConfig::unsafe_arena_release_known_trip_count() {
|
|
// @@protoc_insertion_point(field_release:xla.WhileLoopBackendConfig.known_trip_count)
|
|
|
|
::xla::WhileLoopBackendConfig_KnownTripCount* temp = _impl_.known_trip_count_;
|
|
_impl_.known_trip_count_ = nullptr;
|
|
return temp;
|
|
}
|
|
inline ::xla::WhileLoopBackendConfig_KnownTripCount* WhileLoopBackendConfig::_internal_mutable_known_trip_count() {
|
|
|
|
if (_impl_.known_trip_count_ == nullptr) {
|
|
auto* p = CreateMaybeMessage<::xla::WhileLoopBackendConfig_KnownTripCount>(GetArenaForAllocation());
|
|
_impl_.known_trip_count_ = p;
|
|
}
|
|
return _impl_.known_trip_count_;
|
|
}
|
|
inline ::xla::WhileLoopBackendConfig_KnownTripCount* WhileLoopBackendConfig::mutable_known_trip_count() {
|
|
::xla::WhileLoopBackendConfig_KnownTripCount* _msg = _internal_mutable_known_trip_count();
|
|
// @@protoc_insertion_point(field_mutable:xla.WhileLoopBackendConfig.known_trip_count)
|
|
return _msg;
|
|
}
|
|
inline void WhileLoopBackendConfig::set_allocated_known_trip_count(::xla::WhileLoopBackendConfig_KnownTripCount* known_trip_count) {
|
|
::PROTOBUF_NAMESPACE_ID::Arena* message_arena = GetArenaForAllocation();
|
|
if (message_arena == nullptr) {
|
|
delete _impl_.known_trip_count_;
|
|
}
|
|
if (known_trip_count) {
|
|
::PROTOBUF_NAMESPACE_ID::Arena* submessage_arena =
|
|
::PROTOBUF_NAMESPACE_ID::Arena::InternalGetOwningArena(known_trip_count);
|
|
if (message_arena != submessage_arena) {
|
|
known_trip_count = ::PROTOBUF_NAMESPACE_ID::internal::GetOwnedMessage(
|
|
message_arena, known_trip_count, submessage_arena);
|
|
}
|
|
|
|
} else {
|
|
|
|
}
|
|
_impl_.known_trip_count_ = known_trip_count;
|
|
// @@protoc_insertion_point(field_set_allocated:xla.WhileLoopBackendConfig.known_trip_count)
|
|
}
|
|
|
|
// -------------------------------------------------------------------
|
|
|
|
// OutputOperandAliasing
|
|
|
|
// repeated int64 output_shape_index = 1;
|
|
inline int OutputOperandAliasing::_internal_output_shape_index_size() const {
|
|
return _impl_.output_shape_index_.size();
|
|
}
|
|
inline int OutputOperandAliasing::output_shape_index_size() const {
|
|
return _internal_output_shape_index_size();
|
|
}
|
|
inline void OutputOperandAliasing::clear_output_shape_index() {
|
|
_impl_.output_shape_index_.Clear();
|
|
}
|
|
inline int64_t OutputOperandAliasing::_internal_output_shape_index(int index) const {
|
|
return _impl_.output_shape_index_.Get(index);
|
|
}
|
|
inline int64_t OutputOperandAliasing::output_shape_index(int index) const {
|
|
// @@protoc_insertion_point(field_get:xla.OutputOperandAliasing.output_shape_index)
|
|
return _internal_output_shape_index(index);
|
|
}
|
|
inline void OutputOperandAliasing::set_output_shape_index(int index, int64_t value) {
|
|
_impl_.output_shape_index_.Set(index, value);
|
|
// @@protoc_insertion_point(field_set:xla.OutputOperandAliasing.output_shape_index)
|
|
}
|
|
inline void OutputOperandAliasing::_internal_add_output_shape_index(int64_t value) {
|
|
_impl_.output_shape_index_.Add(value);
|
|
}
|
|
inline void OutputOperandAliasing::add_output_shape_index(int64_t value) {
|
|
_internal_add_output_shape_index(value);
|
|
// @@protoc_insertion_point(field_add:xla.OutputOperandAliasing.output_shape_index)
|
|
}
|
|
inline const ::PROTOBUF_NAMESPACE_ID::RepeatedField< int64_t >&
|
|
OutputOperandAliasing::_internal_output_shape_index() const {
|
|
return _impl_.output_shape_index_;
|
|
}
|
|
inline const ::PROTOBUF_NAMESPACE_ID::RepeatedField< int64_t >&
|
|
OutputOperandAliasing::output_shape_index() const {
|
|
// @@protoc_insertion_point(field_list:xla.OutputOperandAliasing.output_shape_index)
|
|
return _internal_output_shape_index();
|
|
}
|
|
inline ::PROTOBUF_NAMESPACE_ID::RepeatedField< int64_t >*
|
|
OutputOperandAliasing::_internal_mutable_output_shape_index() {
|
|
return &_impl_.output_shape_index_;
|
|
}
|
|
inline ::PROTOBUF_NAMESPACE_ID::RepeatedField< int64_t >*
|
|
OutputOperandAliasing::mutable_output_shape_index() {
|
|
// @@protoc_insertion_point(field_mutable_list:xla.OutputOperandAliasing.output_shape_index)
|
|
return _internal_mutable_output_shape_index();
|
|
}
|
|
|
|
// int64 operand_index = 2;
|
|
inline void OutputOperandAliasing::clear_operand_index() {
|
|
_impl_.operand_index_ = int64_t{0};
|
|
}
|
|
inline int64_t OutputOperandAliasing::_internal_operand_index() const {
|
|
return _impl_.operand_index_;
|
|
}
|
|
inline int64_t OutputOperandAliasing::operand_index() const {
|
|
// @@protoc_insertion_point(field_get:xla.OutputOperandAliasing.operand_index)
|
|
return _internal_operand_index();
|
|
}
|
|
inline void OutputOperandAliasing::_internal_set_operand_index(int64_t value) {
|
|
|
|
_impl_.operand_index_ = value;
|
|
}
|
|
inline void OutputOperandAliasing::set_operand_index(int64_t value) {
|
|
_internal_set_operand_index(value);
|
|
// @@protoc_insertion_point(field_set:xla.OutputOperandAliasing.operand_index)
|
|
}
|
|
|
|
// repeated int64 operand_shape_index = 3;
|
|
inline int OutputOperandAliasing::_internal_operand_shape_index_size() const {
|
|
return _impl_.operand_shape_index_.size();
|
|
}
|
|
inline int OutputOperandAliasing::operand_shape_index_size() const {
|
|
return _internal_operand_shape_index_size();
|
|
}
|
|
inline void OutputOperandAliasing::clear_operand_shape_index() {
|
|
_impl_.operand_shape_index_.Clear();
|
|
}
|
|
inline int64_t OutputOperandAliasing::_internal_operand_shape_index(int index) const {
|
|
return _impl_.operand_shape_index_.Get(index);
|
|
}
|
|
inline int64_t OutputOperandAliasing::operand_shape_index(int index) const {
|
|
// @@protoc_insertion_point(field_get:xla.OutputOperandAliasing.operand_shape_index)
|
|
return _internal_operand_shape_index(index);
|
|
}
|
|
inline void OutputOperandAliasing::set_operand_shape_index(int index, int64_t value) {
|
|
_impl_.operand_shape_index_.Set(index, value);
|
|
// @@protoc_insertion_point(field_set:xla.OutputOperandAliasing.operand_shape_index)
|
|
}
|
|
inline void OutputOperandAliasing::_internal_add_operand_shape_index(int64_t value) {
|
|
_impl_.operand_shape_index_.Add(value);
|
|
}
|
|
inline void OutputOperandAliasing::add_operand_shape_index(int64_t value) {
|
|
_internal_add_operand_shape_index(value);
|
|
// @@protoc_insertion_point(field_add:xla.OutputOperandAliasing.operand_shape_index)
|
|
}
|
|
inline const ::PROTOBUF_NAMESPACE_ID::RepeatedField< int64_t >&
|
|
OutputOperandAliasing::_internal_operand_shape_index() const {
|
|
return _impl_.operand_shape_index_;
|
|
}
|
|
inline const ::PROTOBUF_NAMESPACE_ID::RepeatedField< int64_t >&
|
|
OutputOperandAliasing::operand_shape_index() const {
|
|
// @@protoc_insertion_point(field_list:xla.OutputOperandAliasing.operand_shape_index)
|
|
return _internal_operand_shape_index();
|
|
}
|
|
inline ::PROTOBUF_NAMESPACE_ID::RepeatedField< int64_t >*
|
|
OutputOperandAliasing::_internal_mutable_operand_shape_index() {
|
|
return &_impl_.operand_shape_index_;
|
|
}
|
|
inline ::PROTOBUF_NAMESPACE_ID::RepeatedField< int64_t >*
|
|
OutputOperandAliasing::mutable_operand_shape_index() {
|
|
// @@protoc_insertion_point(field_mutable_list:xla.OutputOperandAliasing.operand_shape_index)
|
|
return _internal_mutable_operand_shape_index();
|
|
}
|
|
|
|
#ifdef __GNUC__
|
|
#pragma GCC diagnostic pop
|
|
#endif // __GNUC__
|
|
// -------------------------------------------------------------------
|
|
|
|
// -------------------------------------------------------------------
|
|
|
|
// -------------------------------------------------------------------
|
|
|
|
// -------------------------------------------------------------------
|
|
|
|
// -------------------------------------------------------------------
|
|
|
|
// -------------------------------------------------------------------
|
|
|
|
// -------------------------------------------------------------------
|
|
|
|
// -------------------------------------------------------------------
|
|
|
|
// -------------------------------------------------------------------
|
|
|
|
// -------------------------------------------------------------------
|
|
|
|
// -------------------------------------------------------------------
|
|
|
|
// -------------------------------------------------------------------
|
|
|
|
// -------------------------------------------------------------------
|
|
|
|
// -------------------------------------------------------------------
|
|
|
|
// -------------------------------------------------------------------
|
|
|
|
// -------------------------------------------------------------------
|
|
|
|
// -------------------------------------------------------------------
|
|
|
|
// -------------------------------------------------------------------
|
|
|
|
// -------------------------------------------------------------------
|
|
|
|
// -------------------------------------------------------------------
|
|
|
|
// -------------------------------------------------------------------
|
|
|
|
// -------------------------------------------------------------------
|
|
|
|
// -------------------------------------------------------------------
|
|
|
|
// -------------------------------------------------------------------
|
|
|
|
// -------------------------------------------------------------------
|
|
|
|
// -------------------------------------------------------------------
|
|
|
|
// -------------------------------------------------------------------
|
|
|
|
// -------------------------------------------------------------------
|
|
|
|
// -------------------------------------------------------------------
|
|
|
|
// -------------------------------------------------------------------
|
|
|
|
// -------------------------------------------------------------------
|
|
|
|
// -------------------------------------------------------------------
|
|
|
|
// -------------------------------------------------------------------
|
|
|
|
// -------------------------------------------------------------------
|
|
|
|
// -------------------------------------------------------------------
|
|
|
|
// -------------------------------------------------------------------
|
|
|
|
// -------------------------------------------------------------------
|
|
|
|
// -------------------------------------------------------------------
|
|
|
|
|
|
// @@protoc_insertion_point(namespace_scope)
|
|
|
|
} // namespace xla
|
|
|
|
PROTOBUF_NAMESPACE_OPEN
|
|
|
|
template <> struct is_proto_enum< ::xla::ChannelHandle_ChannelType> : ::std::true_type {};
|
|
template <>
|
|
inline const EnumDescriptor* GetEnumDescriptor< ::xla::ChannelHandle_ChannelType>() {
|
|
return ::xla::ChannelHandle_ChannelType_descriptor();
|
|
}
|
|
template <> struct is_proto_enum< ::xla::TriangularSolveOptions_Transpose> : ::std::true_type {};
|
|
template <>
|
|
inline const EnumDescriptor* GetEnumDescriptor< ::xla::TriangularSolveOptions_Transpose>() {
|
|
return ::xla::TriangularSolveOptions_Transpose_descriptor();
|
|
}
|
|
template <> struct is_proto_enum< ::xla::OpSharding_Type> : ::std::true_type {};
|
|
template <>
|
|
inline const EnumDescriptor* GetEnumDescriptor< ::xla::OpSharding_Type>() {
|
|
return ::xla::OpSharding_Type_descriptor();
|
|
}
|
|
template <> struct is_proto_enum< ::xla::OpSharding_ShardGroupType> : ::std::true_type {};
|
|
template <>
|
|
inline const EnumDescriptor* GetEnumDescriptor< ::xla::OpSharding_ShardGroupType>() {
|
|
return ::xla::OpSharding_ShardGroupType_descriptor();
|
|
}
|
|
template <> struct is_proto_enum< ::xla::PrecisionConfig_Precision> : ::std::true_type {};
|
|
template <>
|
|
inline const EnumDescriptor* GetEnumDescriptor< ::xla::PrecisionConfig_Precision>() {
|
|
return ::xla::PrecisionConfig_Precision_descriptor();
|
|
}
|
|
template <> struct is_proto_enum< ::xla::PrimitiveType> : ::std::true_type {};
|
|
template <>
|
|
inline const EnumDescriptor* GetEnumDescriptor< ::xla::PrimitiveType>() {
|
|
return ::xla::PrimitiveType_descriptor();
|
|
}
|
|
template <> struct is_proto_enum< ::xla::DimLevelType> : ::std::true_type {};
|
|
template <>
|
|
inline const EnumDescriptor* GetEnumDescriptor< ::xla::DimLevelType>() {
|
|
return ::xla::DimLevelType_descriptor();
|
|
}
|
|
template <> struct is_proto_enum< ::xla::ProfileType> : ::std::true_type {};
|
|
template <>
|
|
inline const EnumDescriptor* GetEnumDescriptor< ::xla::ProfileType>() {
|
|
return ::xla::ProfileType_descriptor();
|
|
}
|
|
template <> struct is_proto_enum< ::xla::ProfileSource> : ::std::true_type {};
|
|
template <>
|
|
inline const EnumDescriptor* GetEnumDescriptor< ::xla::ProfileSource>() {
|
|
return ::xla::ProfileSource_descriptor();
|
|
}
|
|
template <> struct is_proto_enum< ::xla::CompilationEvent> : ::std::true_type {};
|
|
template <>
|
|
inline const EnumDescriptor* GetEnumDescriptor< ::xla::CompilationEvent>() {
|
|
return ::xla::CompilationEvent_descriptor();
|
|
}
|
|
template <> struct is_proto_enum< ::xla::PaddingType> : ::std::true_type {};
|
|
template <>
|
|
inline const EnumDescriptor* GetEnumDescriptor< ::xla::PaddingType>() {
|
|
return ::xla::PaddingType_descriptor();
|
|
}
|
|
template <> struct is_proto_enum< ::xla::FftType> : ::std::true_type {};
|
|
template <>
|
|
inline const EnumDescriptor* GetEnumDescriptor< ::xla::FftType>() {
|
|
return ::xla::FftType_descriptor();
|
|
}
|
|
template <> struct is_proto_enum< ::xla::SparsityType> : ::std::true_type {};
|
|
template <>
|
|
inline const EnumDescriptor* GetEnumDescriptor< ::xla::SparsityType>() {
|
|
return ::xla::SparsityType_descriptor();
|
|
}
|
|
template <> struct is_proto_enum< ::xla::RandomDistribution> : ::std::true_type {};
|
|
template <>
|
|
inline const EnumDescriptor* GetEnumDescriptor< ::xla::RandomDistribution>() {
|
|
return ::xla::RandomDistribution_descriptor();
|
|
}
|
|
template <> struct is_proto_enum< ::xla::RandomAlgorithm> : ::std::true_type {};
|
|
template <>
|
|
inline const EnumDescriptor* GetEnumDescriptor< ::xla::RandomAlgorithm>() {
|
|
return ::xla::RandomAlgorithm_descriptor();
|
|
}
|
|
|
|
PROTOBUF_NAMESPACE_CLOSE
|
|
|
|
// @@protoc_insertion_point(global_scope)
|
|
|
|
#include <google/protobuf/port_undef.inc>
|
|
#endif // GOOGLE_PROTOBUF_INCLUDED_GOOGLE_PROTOBUF_INCLUDED_xla_2fxla_5fdata_2eproto
|