21547 lines
872 KiB
C++
21547 lines
872 KiB
C++
// Generated by the protocol buffer compiler. DO NOT EDIT!
|
|
// source: xla/xla.proto
|
|
|
|
#ifndef GOOGLE_PROTOBUF_INCLUDED_xla_2fxla_2eproto
|
|
#define GOOGLE_PROTOBUF_INCLUDED_xla_2fxla_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_bases.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>
|
|
#include <google/protobuf/any.pb.h>
|
|
#include "xla/service/hlo.pb.h"
|
|
#include "xla/xla_data.pb.h"
|
|
// @@protoc_insertion_point(includes)
|
|
#include <google/protobuf/port_def.inc>
|
|
#define PROTOBUF_INTERNAL_EXPORT_xla_2fxla_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_2eproto {
|
|
static const uint32_t offsets[];
|
|
};
|
|
extern const ::PROTOBUF_NAMESPACE_ID::internal::DescriptorTable descriptor_table_xla_2fxla_2eproto;
|
|
namespace xla {
|
|
class CompilationEnvironmentsProto;
|
|
struct CompilationEnvironmentsProtoDefaultTypeInternal;
|
|
extern CompilationEnvironmentsProtoDefaultTypeInternal _CompilationEnvironmentsProto_default_instance_;
|
|
class CompileRequest;
|
|
struct CompileRequestDefaultTypeInternal;
|
|
extern CompileRequestDefaultTypeInternal _CompileRequest_default_instance_;
|
|
class CompileResponse;
|
|
struct CompileResponseDefaultTypeInternal;
|
|
extern CompileResponseDefaultTypeInternal _CompileResponse_default_instance_;
|
|
class ComputationGraphStatsRequest;
|
|
struct ComputationGraphStatsRequestDefaultTypeInternal;
|
|
extern ComputationGraphStatsRequestDefaultTypeInternal _ComputationGraphStatsRequest_default_instance_;
|
|
class ComputationStatsResponse;
|
|
struct ComputationStatsResponseDefaultTypeInternal;
|
|
extern ComputationStatsResponseDefaultTypeInternal _ComputationStatsResponse_default_instance_;
|
|
class ComputeConstantGraphRequest;
|
|
struct ComputeConstantGraphRequestDefaultTypeInternal;
|
|
extern ComputeConstantGraphRequestDefaultTypeInternal _ComputeConstantGraphRequest_default_instance_;
|
|
class ComputeConstantResponse;
|
|
struct ComputeConstantResponseDefaultTypeInternal;
|
|
extern ComputeConstantResponseDefaultTypeInternal _ComputeConstantResponse_default_instance_;
|
|
class CreateChannelHandleRequest;
|
|
struct CreateChannelHandleRequestDefaultTypeInternal;
|
|
extern CreateChannelHandleRequestDefaultTypeInternal _CreateChannelHandleRequest_default_instance_;
|
|
class CreateChannelHandleResponse;
|
|
struct CreateChannelHandleResponseDefaultTypeInternal;
|
|
extern CreateChannelHandleResponseDefaultTypeInternal _CreateChannelHandleResponse_default_instance_;
|
|
class DebugOptions;
|
|
struct DebugOptionsDefaultTypeInternal;
|
|
extern DebugOptionsDefaultTypeInternal _DebugOptions_default_instance_;
|
|
class DebugOptions_XlaBackendExtraOptionsEntry_DoNotUse;
|
|
struct DebugOptions_XlaBackendExtraOptionsEntry_DoNotUseDefaultTypeInternal;
|
|
extern DebugOptions_XlaBackendExtraOptionsEntry_DoNotUseDefaultTypeInternal _DebugOptions_XlaBackendExtraOptionsEntry_DoNotUse_default_instance_;
|
|
class DeconstructTupleRequest;
|
|
struct DeconstructTupleRequestDefaultTypeInternal;
|
|
extern DeconstructTupleRequestDefaultTypeInternal _DeconstructTupleRequest_default_instance_;
|
|
class DeconstructTupleResponse;
|
|
struct DeconstructTupleResponseDefaultTypeInternal;
|
|
extern DeconstructTupleResponseDefaultTypeInternal _DeconstructTupleResponse_default_instance_;
|
|
class ExecuteGraphParallelRequest;
|
|
struct ExecuteGraphParallelRequestDefaultTypeInternal;
|
|
extern ExecuteGraphParallelRequestDefaultTypeInternal _ExecuteGraphParallelRequest_default_instance_;
|
|
class ExecuteGraphRequest;
|
|
struct ExecuteGraphRequestDefaultTypeInternal;
|
|
extern ExecuteGraphRequestDefaultTypeInternal _ExecuteGraphRequest_default_instance_;
|
|
class ExecuteParallelResponse;
|
|
struct ExecuteParallelResponseDefaultTypeInternal;
|
|
extern ExecuteParallelResponseDefaultTypeInternal _ExecuteParallelResponse_default_instance_;
|
|
class ExecuteRequest;
|
|
struct ExecuteRequestDefaultTypeInternal;
|
|
extern ExecuteRequestDefaultTypeInternal _ExecuteRequest_default_instance_;
|
|
class ExecuteResponse;
|
|
struct ExecuteResponseDefaultTypeInternal;
|
|
extern ExecuteResponseDefaultTypeInternal _ExecuteResponse_default_instance_;
|
|
class ExecutionOptions;
|
|
struct ExecutionOptionsDefaultTypeInternal;
|
|
extern ExecutionOptionsDefaultTypeInternal _ExecutionOptions_default_instance_;
|
|
class GetDeviceHandlesRequest;
|
|
struct GetDeviceHandlesRequestDefaultTypeInternal;
|
|
extern GetDeviceHandlesRequestDefaultTypeInternal _GetDeviceHandlesRequest_default_instance_;
|
|
class GetDeviceHandlesResponse;
|
|
struct GetDeviceHandlesResponseDefaultTypeInternal;
|
|
extern GetDeviceHandlesResponseDefaultTypeInternal _GetDeviceHandlesResponse_default_instance_;
|
|
class GetShapeRequest;
|
|
struct GetShapeRequestDefaultTypeInternal;
|
|
extern GetShapeRequestDefaultTypeInternal _GetShapeRequest_default_instance_;
|
|
class GetShapeResponse;
|
|
struct GetShapeResponseDefaultTypeInternal;
|
|
extern GetShapeResponseDefaultTypeInternal _GetShapeResponse_default_instance_;
|
|
class GpuCompilationEnvironment;
|
|
struct GpuCompilationEnvironmentDefaultTypeInternal;
|
|
extern GpuCompilationEnvironmentDefaultTypeInternal _GpuCompilationEnvironment_default_instance_;
|
|
class HloModuleConfigProto;
|
|
struct HloModuleConfigProtoDefaultTypeInternal;
|
|
extern HloModuleConfigProtoDefaultTypeInternal _HloModuleConfigProto_default_instance_;
|
|
class HloModuleConfigProto_AnalysisAllowanceMapEntry_DoNotUse;
|
|
struct HloModuleConfigProto_AnalysisAllowanceMapEntry_DoNotUseDefaultTypeInternal;
|
|
extern HloModuleConfigProto_AnalysisAllowanceMapEntry_DoNotUseDefaultTypeInternal _HloModuleConfigProto_AnalysisAllowanceMapEntry_DoNotUse_default_instance_;
|
|
class HloModuleConfigProto_BoolList;
|
|
struct HloModuleConfigProto_BoolListDefaultTypeInternal;
|
|
extern HloModuleConfigProto_BoolListDefaultTypeInternal _HloModuleConfigProto_BoolList_default_instance_;
|
|
class HloModuleConfigProto_DotConfigEntry_DoNotUse;
|
|
struct HloModuleConfigProto_DotConfigEntry_DoNotUseDefaultTypeInternal;
|
|
extern HloModuleConfigProto_DotConfigEntry_DoNotUseDefaultTypeInternal _HloModuleConfigProto_DotConfigEntry_DoNotUse_default_instance_;
|
|
class HloModuleConfigProto_Int64List;
|
|
struct HloModuleConfigProto_Int64ListDefaultTypeInternal;
|
|
extern HloModuleConfigProto_Int64ListDefaultTypeInternal _HloModuleConfigProto_Int64List_default_instance_;
|
|
class HloModuleConfigProto_Int64ListList;
|
|
struct HloModuleConfigProto_Int64ListListDefaultTypeInternal;
|
|
extern HloModuleConfigProto_Int64ListListDefaultTypeInternal _HloModuleConfigProto_Int64ListList_default_instance_;
|
|
class HloModuleProtoWithConfig;
|
|
struct HloModuleProtoWithConfigDefaultTypeInternal;
|
|
extern HloModuleProtoWithConfigDefaultTypeInternal _HloModuleProtoWithConfig_default_instance_;
|
|
class LoadDataRequest;
|
|
struct LoadDataRequestDefaultTypeInternal;
|
|
extern LoadDataRequestDefaultTypeInternal _LoadDataRequest_default_instance_;
|
|
class LoadDataResponse;
|
|
struct LoadDataResponseDefaultTypeInternal;
|
|
extern LoadDataResponseDefaultTypeInternal _LoadDataResponse_default_instance_;
|
|
class ResetDeviceRequest;
|
|
struct ResetDeviceRequestDefaultTypeInternal;
|
|
extern ResetDeviceRequestDefaultTypeInternal _ResetDeviceRequest_default_instance_;
|
|
class ResetDeviceResponse;
|
|
struct ResetDeviceResponseDefaultTypeInternal;
|
|
extern ResetDeviceResponseDefaultTypeInternal _ResetDeviceResponse_default_instance_;
|
|
class ScheduleProto;
|
|
struct ScheduleProtoDefaultTypeInternal;
|
|
extern ScheduleProtoDefaultTypeInternal _ScheduleProto_default_instance_;
|
|
class ScheduleProto_Instruction;
|
|
struct ScheduleProto_InstructionDefaultTypeInternal;
|
|
extern ScheduleProto_InstructionDefaultTypeInternal _ScheduleProto_Instruction_default_instance_;
|
|
class ShardableValueUpdatePairProto;
|
|
struct ShardableValueUpdatePairProtoDefaultTypeInternal;
|
|
extern ShardableValueUpdatePairProtoDefaultTypeInternal _ShardableValueUpdatePairProto_default_instance_;
|
|
class TransferFromOutfeedRequest;
|
|
struct TransferFromOutfeedRequestDefaultTypeInternal;
|
|
extern TransferFromOutfeedRequestDefaultTypeInternal _TransferFromOutfeedRequest_default_instance_;
|
|
class TransferFromOutfeedResponse;
|
|
struct TransferFromOutfeedResponseDefaultTypeInternal;
|
|
extern TransferFromOutfeedResponseDefaultTypeInternal _TransferFromOutfeedResponse_default_instance_;
|
|
class TransferToClientRequest;
|
|
struct TransferToClientRequestDefaultTypeInternal;
|
|
extern TransferToClientRequestDefaultTypeInternal _TransferToClientRequest_default_instance_;
|
|
class TransferToClientResponse;
|
|
struct TransferToClientResponseDefaultTypeInternal;
|
|
extern TransferToClientResponseDefaultTypeInternal _TransferToClientResponse_default_instance_;
|
|
class TransferToInfeedRequest;
|
|
struct TransferToInfeedRequestDefaultTypeInternal;
|
|
extern TransferToInfeedRequestDefaultTypeInternal _TransferToInfeedRequest_default_instance_;
|
|
class TransferToInfeedResponse;
|
|
struct TransferToInfeedResponseDefaultTypeInternal;
|
|
extern TransferToInfeedResponseDefaultTypeInternal _TransferToInfeedResponse_default_instance_;
|
|
class TransferToServerRequest;
|
|
struct TransferToServerRequestDefaultTypeInternal;
|
|
extern TransferToServerRequestDefaultTypeInternal _TransferToServerRequest_default_instance_;
|
|
class TransferToServerResponse;
|
|
struct TransferToServerResponseDefaultTypeInternal;
|
|
extern TransferToServerResponseDefaultTypeInternal _TransferToServerResponse_default_instance_;
|
|
class UnpackRequest;
|
|
struct UnpackRequestDefaultTypeInternal;
|
|
extern UnpackRequestDefaultTypeInternal _UnpackRequest_default_instance_;
|
|
class UnpackResponse;
|
|
struct UnpackResponseDefaultTypeInternal;
|
|
extern UnpackResponseDefaultTypeInternal _UnpackResponse_default_instance_;
|
|
class UnregisterRequest;
|
|
struct UnregisterRequestDefaultTypeInternal;
|
|
extern UnregisterRequestDefaultTypeInternal _UnregisterRequest_default_instance_;
|
|
class UnregisterResponse;
|
|
struct UnregisterResponseDefaultTypeInternal;
|
|
extern UnregisterResponseDefaultTypeInternal _UnregisterResponse_default_instance_;
|
|
} // namespace xla
|
|
PROTOBUF_NAMESPACE_OPEN
|
|
template<> ::xla::CompilationEnvironmentsProto* Arena::CreateMaybeMessage<::xla::CompilationEnvironmentsProto>(Arena*);
|
|
template<> ::xla::CompileRequest* Arena::CreateMaybeMessage<::xla::CompileRequest>(Arena*);
|
|
template<> ::xla::CompileResponse* Arena::CreateMaybeMessage<::xla::CompileResponse>(Arena*);
|
|
template<> ::xla::ComputationGraphStatsRequest* Arena::CreateMaybeMessage<::xla::ComputationGraphStatsRequest>(Arena*);
|
|
template<> ::xla::ComputationStatsResponse* Arena::CreateMaybeMessage<::xla::ComputationStatsResponse>(Arena*);
|
|
template<> ::xla::ComputeConstantGraphRequest* Arena::CreateMaybeMessage<::xla::ComputeConstantGraphRequest>(Arena*);
|
|
template<> ::xla::ComputeConstantResponse* Arena::CreateMaybeMessage<::xla::ComputeConstantResponse>(Arena*);
|
|
template<> ::xla::CreateChannelHandleRequest* Arena::CreateMaybeMessage<::xla::CreateChannelHandleRequest>(Arena*);
|
|
template<> ::xla::CreateChannelHandleResponse* Arena::CreateMaybeMessage<::xla::CreateChannelHandleResponse>(Arena*);
|
|
template<> ::xla::DebugOptions* Arena::CreateMaybeMessage<::xla::DebugOptions>(Arena*);
|
|
template<> ::xla::DebugOptions_XlaBackendExtraOptionsEntry_DoNotUse* Arena::CreateMaybeMessage<::xla::DebugOptions_XlaBackendExtraOptionsEntry_DoNotUse>(Arena*);
|
|
template<> ::xla::DeconstructTupleRequest* Arena::CreateMaybeMessage<::xla::DeconstructTupleRequest>(Arena*);
|
|
template<> ::xla::DeconstructTupleResponse* Arena::CreateMaybeMessage<::xla::DeconstructTupleResponse>(Arena*);
|
|
template<> ::xla::ExecuteGraphParallelRequest* Arena::CreateMaybeMessage<::xla::ExecuteGraphParallelRequest>(Arena*);
|
|
template<> ::xla::ExecuteGraphRequest* Arena::CreateMaybeMessage<::xla::ExecuteGraphRequest>(Arena*);
|
|
template<> ::xla::ExecuteParallelResponse* Arena::CreateMaybeMessage<::xla::ExecuteParallelResponse>(Arena*);
|
|
template<> ::xla::ExecuteRequest* Arena::CreateMaybeMessage<::xla::ExecuteRequest>(Arena*);
|
|
template<> ::xla::ExecuteResponse* Arena::CreateMaybeMessage<::xla::ExecuteResponse>(Arena*);
|
|
template<> ::xla::ExecutionOptions* Arena::CreateMaybeMessage<::xla::ExecutionOptions>(Arena*);
|
|
template<> ::xla::GetDeviceHandlesRequest* Arena::CreateMaybeMessage<::xla::GetDeviceHandlesRequest>(Arena*);
|
|
template<> ::xla::GetDeviceHandlesResponse* Arena::CreateMaybeMessage<::xla::GetDeviceHandlesResponse>(Arena*);
|
|
template<> ::xla::GetShapeRequest* Arena::CreateMaybeMessage<::xla::GetShapeRequest>(Arena*);
|
|
template<> ::xla::GetShapeResponse* Arena::CreateMaybeMessage<::xla::GetShapeResponse>(Arena*);
|
|
template<> ::xla::GpuCompilationEnvironment* Arena::CreateMaybeMessage<::xla::GpuCompilationEnvironment>(Arena*);
|
|
template<> ::xla::HloModuleConfigProto* Arena::CreateMaybeMessage<::xla::HloModuleConfigProto>(Arena*);
|
|
template<> ::xla::HloModuleConfigProto_AnalysisAllowanceMapEntry_DoNotUse* Arena::CreateMaybeMessage<::xla::HloModuleConfigProto_AnalysisAllowanceMapEntry_DoNotUse>(Arena*);
|
|
template<> ::xla::HloModuleConfigProto_BoolList* Arena::CreateMaybeMessage<::xla::HloModuleConfigProto_BoolList>(Arena*);
|
|
template<> ::xla::HloModuleConfigProto_DotConfigEntry_DoNotUse* Arena::CreateMaybeMessage<::xla::HloModuleConfigProto_DotConfigEntry_DoNotUse>(Arena*);
|
|
template<> ::xla::HloModuleConfigProto_Int64List* Arena::CreateMaybeMessage<::xla::HloModuleConfigProto_Int64List>(Arena*);
|
|
template<> ::xla::HloModuleConfigProto_Int64ListList* Arena::CreateMaybeMessage<::xla::HloModuleConfigProto_Int64ListList>(Arena*);
|
|
template<> ::xla::HloModuleProtoWithConfig* Arena::CreateMaybeMessage<::xla::HloModuleProtoWithConfig>(Arena*);
|
|
template<> ::xla::LoadDataRequest* Arena::CreateMaybeMessage<::xla::LoadDataRequest>(Arena*);
|
|
template<> ::xla::LoadDataResponse* Arena::CreateMaybeMessage<::xla::LoadDataResponse>(Arena*);
|
|
template<> ::xla::ResetDeviceRequest* Arena::CreateMaybeMessage<::xla::ResetDeviceRequest>(Arena*);
|
|
template<> ::xla::ResetDeviceResponse* Arena::CreateMaybeMessage<::xla::ResetDeviceResponse>(Arena*);
|
|
template<> ::xla::ScheduleProto* Arena::CreateMaybeMessage<::xla::ScheduleProto>(Arena*);
|
|
template<> ::xla::ScheduleProto_Instruction* Arena::CreateMaybeMessage<::xla::ScheduleProto_Instruction>(Arena*);
|
|
template<> ::xla::ShardableValueUpdatePairProto* Arena::CreateMaybeMessage<::xla::ShardableValueUpdatePairProto>(Arena*);
|
|
template<> ::xla::TransferFromOutfeedRequest* Arena::CreateMaybeMessage<::xla::TransferFromOutfeedRequest>(Arena*);
|
|
template<> ::xla::TransferFromOutfeedResponse* Arena::CreateMaybeMessage<::xla::TransferFromOutfeedResponse>(Arena*);
|
|
template<> ::xla::TransferToClientRequest* Arena::CreateMaybeMessage<::xla::TransferToClientRequest>(Arena*);
|
|
template<> ::xla::TransferToClientResponse* Arena::CreateMaybeMessage<::xla::TransferToClientResponse>(Arena*);
|
|
template<> ::xla::TransferToInfeedRequest* Arena::CreateMaybeMessage<::xla::TransferToInfeedRequest>(Arena*);
|
|
template<> ::xla::TransferToInfeedResponse* Arena::CreateMaybeMessage<::xla::TransferToInfeedResponse>(Arena*);
|
|
template<> ::xla::TransferToServerRequest* Arena::CreateMaybeMessage<::xla::TransferToServerRequest>(Arena*);
|
|
template<> ::xla::TransferToServerResponse* Arena::CreateMaybeMessage<::xla::TransferToServerResponse>(Arena*);
|
|
template<> ::xla::UnpackRequest* Arena::CreateMaybeMessage<::xla::UnpackRequest>(Arena*);
|
|
template<> ::xla::UnpackResponse* Arena::CreateMaybeMessage<::xla::UnpackResponse>(Arena*);
|
|
template<> ::xla::UnregisterRequest* Arena::CreateMaybeMessage<::xla::UnregisterRequest>(Arena*);
|
|
template<> ::xla::UnregisterResponse* Arena::CreateMaybeMessage<::xla::UnregisterResponse>(Arena*);
|
|
PROTOBUF_NAMESPACE_CLOSE
|
|
namespace xla {
|
|
|
|
enum DebugOptions_ShapeChecks : int {
|
|
DebugOptions_ShapeChecks_IGNORE = 0,
|
|
DebugOptions_ShapeChecks_RUNTIME = 1,
|
|
DebugOptions_ShapeChecks_COMPILE_TIME = 2,
|
|
DebugOptions_ShapeChecks_DebugOptions_ShapeChecks_INT_MIN_SENTINEL_DO_NOT_USE_ = std::numeric_limits<int32_t>::min(),
|
|
DebugOptions_ShapeChecks_DebugOptions_ShapeChecks_INT_MAX_SENTINEL_DO_NOT_USE_ = std::numeric_limits<int32_t>::max()
|
|
};
|
|
bool DebugOptions_ShapeChecks_IsValid(int value);
|
|
constexpr DebugOptions_ShapeChecks DebugOptions_ShapeChecks_ShapeChecks_MIN = DebugOptions_ShapeChecks_IGNORE;
|
|
constexpr DebugOptions_ShapeChecks DebugOptions_ShapeChecks_ShapeChecks_MAX = DebugOptions_ShapeChecks_COMPILE_TIME;
|
|
constexpr int DebugOptions_ShapeChecks_ShapeChecks_ARRAYSIZE = DebugOptions_ShapeChecks_ShapeChecks_MAX + 1;
|
|
|
|
const ::PROTOBUF_NAMESPACE_ID::EnumDescriptor* DebugOptions_ShapeChecks_descriptor();
|
|
template<typename T>
|
|
inline const std::string& DebugOptions_ShapeChecks_Name(T enum_t_value) {
|
|
static_assert(::std::is_same<T, DebugOptions_ShapeChecks>::value ||
|
|
::std::is_integral<T>::value,
|
|
"Incorrect type passed to function DebugOptions_ShapeChecks_Name.");
|
|
return ::PROTOBUF_NAMESPACE_ID::internal::NameOfEnum(
|
|
DebugOptions_ShapeChecks_descriptor(), enum_t_value);
|
|
}
|
|
inline bool DebugOptions_ShapeChecks_Parse(
|
|
::PROTOBUF_NAMESPACE_ID::ConstStringParam name, DebugOptions_ShapeChecks* value) {
|
|
return ::PROTOBUF_NAMESPACE_ID::internal::ParseNamedEnum<DebugOptions_ShapeChecks>(
|
|
DebugOptions_ShapeChecks_descriptor(), name, value);
|
|
}
|
|
enum DebugOptions_StepMarkerLocation : int {
|
|
DebugOptions_StepMarkerLocation_STEP_MARK_AT_ENTRY = 0,
|
|
DebugOptions_StepMarkerLocation_STEP_MARK_AT_TOP_LEVEL_WHILE_LOOP = 1,
|
|
DebugOptions_StepMarkerLocation_STEP_MARK_AT_SECOND_LEVEL_WHILE_LOOP = 3,
|
|
DebugOptions_StepMarkerLocation_STEP_MARK_NONE = 2,
|
|
DebugOptions_StepMarkerLocation_DebugOptions_StepMarkerLocation_INT_MIN_SENTINEL_DO_NOT_USE_ = std::numeric_limits<int32_t>::min(),
|
|
DebugOptions_StepMarkerLocation_DebugOptions_StepMarkerLocation_INT_MAX_SENTINEL_DO_NOT_USE_ = std::numeric_limits<int32_t>::max()
|
|
};
|
|
bool DebugOptions_StepMarkerLocation_IsValid(int value);
|
|
constexpr DebugOptions_StepMarkerLocation DebugOptions_StepMarkerLocation_StepMarkerLocation_MIN = DebugOptions_StepMarkerLocation_STEP_MARK_AT_ENTRY;
|
|
constexpr DebugOptions_StepMarkerLocation DebugOptions_StepMarkerLocation_StepMarkerLocation_MAX = DebugOptions_StepMarkerLocation_STEP_MARK_AT_SECOND_LEVEL_WHILE_LOOP;
|
|
constexpr int DebugOptions_StepMarkerLocation_StepMarkerLocation_ARRAYSIZE = DebugOptions_StepMarkerLocation_StepMarkerLocation_MAX + 1;
|
|
|
|
const ::PROTOBUF_NAMESPACE_ID::EnumDescriptor* DebugOptions_StepMarkerLocation_descriptor();
|
|
template<typename T>
|
|
inline const std::string& DebugOptions_StepMarkerLocation_Name(T enum_t_value) {
|
|
static_assert(::std::is_same<T, DebugOptions_StepMarkerLocation>::value ||
|
|
::std::is_integral<T>::value,
|
|
"Incorrect type passed to function DebugOptions_StepMarkerLocation_Name.");
|
|
return ::PROTOBUF_NAMESPACE_ID::internal::NameOfEnum(
|
|
DebugOptions_StepMarkerLocation_descriptor(), enum_t_value);
|
|
}
|
|
inline bool DebugOptions_StepMarkerLocation_Parse(
|
|
::PROTOBUF_NAMESPACE_ID::ConstStringParam name, DebugOptions_StepMarkerLocation* value) {
|
|
return ::PROTOBUF_NAMESPACE_ID::internal::ParseNamedEnum<DebugOptions_StepMarkerLocation>(
|
|
DebugOptions_StepMarkerLocation_descriptor(), name, value);
|
|
}
|
|
enum DebugOptions_CommandBufferCmdType : int {
|
|
DebugOptions_CommandBufferCmdType_INVALID = 0,
|
|
DebugOptions_CommandBufferCmdType_FUSION = 1,
|
|
DebugOptions_CommandBufferCmdType_CUBLAS = 2,
|
|
DebugOptions_CommandBufferCmdType_CUDNN = 3,
|
|
DebugOptions_CommandBufferCmdType_COLLECTIVES = 4,
|
|
DebugOptions_CommandBufferCmdType_CONDITIONALS = 5,
|
|
DebugOptions_CommandBufferCmdType_CUSTOM_CALL = 6,
|
|
DebugOptions_CommandBufferCmdType_DebugOptions_CommandBufferCmdType_INT_MIN_SENTINEL_DO_NOT_USE_ = std::numeric_limits<int32_t>::min(),
|
|
DebugOptions_CommandBufferCmdType_DebugOptions_CommandBufferCmdType_INT_MAX_SENTINEL_DO_NOT_USE_ = std::numeric_limits<int32_t>::max()
|
|
};
|
|
bool DebugOptions_CommandBufferCmdType_IsValid(int value);
|
|
constexpr DebugOptions_CommandBufferCmdType DebugOptions_CommandBufferCmdType_CommandBufferCmdType_MIN = DebugOptions_CommandBufferCmdType_INVALID;
|
|
constexpr DebugOptions_CommandBufferCmdType DebugOptions_CommandBufferCmdType_CommandBufferCmdType_MAX = DebugOptions_CommandBufferCmdType_CUSTOM_CALL;
|
|
constexpr int DebugOptions_CommandBufferCmdType_CommandBufferCmdType_ARRAYSIZE = DebugOptions_CommandBufferCmdType_CommandBufferCmdType_MAX + 1;
|
|
|
|
const ::PROTOBUF_NAMESPACE_ID::EnumDescriptor* DebugOptions_CommandBufferCmdType_descriptor();
|
|
template<typename T>
|
|
inline const std::string& DebugOptions_CommandBufferCmdType_Name(T enum_t_value) {
|
|
static_assert(::std::is_same<T, DebugOptions_CommandBufferCmdType>::value ||
|
|
::std::is_integral<T>::value,
|
|
"Incorrect type passed to function DebugOptions_CommandBufferCmdType_Name.");
|
|
return ::PROTOBUF_NAMESPACE_ID::internal::NameOfEnum(
|
|
DebugOptions_CommandBufferCmdType_descriptor(), enum_t_value);
|
|
}
|
|
inline bool DebugOptions_CommandBufferCmdType_Parse(
|
|
::PROTOBUF_NAMESPACE_ID::ConstStringParam name, DebugOptions_CommandBufferCmdType* value) {
|
|
return ::PROTOBUF_NAMESPACE_ID::internal::ParseNamedEnum<DebugOptions_CommandBufferCmdType>(
|
|
DebugOptions_CommandBufferCmdType_descriptor(), name, value);
|
|
}
|
|
enum DebugOptions_PartitioningAlgorithm : int {
|
|
DebugOptions_PartitioningAlgorithm_PARTITIONING_ALGORITHM_NOOP = 0,
|
|
DebugOptions_PartitioningAlgorithm_PARTITIONING_ALGORITHM_EXP0 = 1,
|
|
DebugOptions_PartitioningAlgorithm_PARTITIONING_ALGORITHM_EXP1 = 2,
|
|
DebugOptions_PartitioningAlgorithm_PARTITIONING_ALGORITHM_EXP2 = 3,
|
|
DebugOptions_PartitioningAlgorithm_DebugOptions_PartitioningAlgorithm_INT_MIN_SENTINEL_DO_NOT_USE_ = std::numeric_limits<int32_t>::min(),
|
|
DebugOptions_PartitioningAlgorithm_DebugOptions_PartitioningAlgorithm_INT_MAX_SENTINEL_DO_NOT_USE_ = std::numeric_limits<int32_t>::max()
|
|
};
|
|
bool DebugOptions_PartitioningAlgorithm_IsValid(int value);
|
|
constexpr DebugOptions_PartitioningAlgorithm DebugOptions_PartitioningAlgorithm_PartitioningAlgorithm_MIN = DebugOptions_PartitioningAlgorithm_PARTITIONING_ALGORITHM_NOOP;
|
|
constexpr DebugOptions_PartitioningAlgorithm DebugOptions_PartitioningAlgorithm_PartitioningAlgorithm_MAX = DebugOptions_PartitioningAlgorithm_PARTITIONING_ALGORITHM_EXP2;
|
|
constexpr int DebugOptions_PartitioningAlgorithm_PartitioningAlgorithm_ARRAYSIZE = DebugOptions_PartitioningAlgorithm_PartitioningAlgorithm_MAX + 1;
|
|
|
|
const ::PROTOBUF_NAMESPACE_ID::EnumDescriptor* DebugOptions_PartitioningAlgorithm_descriptor();
|
|
template<typename T>
|
|
inline const std::string& DebugOptions_PartitioningAlgorithm_Name(T enum_t_value) {
|
|
static_assert(::std::is_same<T, DebugOptions_PartitioningAlgorithm>::value ||
|
|
::std::is_integral<T>::value,
|
|
"Incorrect type passed to function DebugOptions_PartitioningAlgorithm_Name.");
|
|
return ::PROTOBUF_NAMESPACE_ID::internal::NameOfEnum(
|
|
DebugOptions_PartitioningAlgorithm_descriptor(), enum_t_value);
|
|
}
|
|
inline bool DebugOptions_PartitioningAlgorithm_Parse(
|
|
::PROTOBUF_NAMESPACE_ID::ConstStringParam name, DebugOptions_PartitioningAlgorithm* value) {
|
|
return ::PROTOBUF_NAMESPACE_ID::internal::ParseNamedEnum<DebugOptions_PartitioningAlgorithm>(
|
|
DebugOptions_PartitioningAlgorithm_descriptor(), name, value);
|
|
}
|
|
enum HloModuleConfigProto_FusionConfigCollection : int {
|
|
HloModuleConfigProto_FusionConfigCollection_OFF = 0,
|
|
HloModuleConfigProto_FusionConfigCollection_PER_EDGE = 1,
|
|
HloModuleConfigProto_FusionConfigCollection_PER_NODE = 2,
|
|
HloModuleConfigProto_FusionConfigCollection_HloModuleConfigProto_FusionConfigCollection_INT_MIN_SENTINEL_DO_NOT_USE_ = std::numeric_limits<int32_t>::min(),
|
|
HloModuleConfigProto_FusionConfigCollection_HloModuleConfigProto_FusionConfigCollection_INT_MAX_SENTINEL_DO_NOT_USE_ = std::numeric_limits<int32_t>::max()
|
|
};
|
|
bool HloModuleConfigProto_FusionConfigCollection_IsValid(int value);
|
|
constexpr HloModuleConfigProto_FusionConfigCollection HloModuleConfigProto_FusionConfigCollection_FusionConfigCollection_MIN = HloModuleConfigProto_FusionConfigCollection_OFF;
|
|
constexpr HloModuleConfigProto_FusionConfigCollection HloModuleConfigProto_FusionConfigCollection_FusionConfigCollection_MAX = HloModuleConfigProto_FusionConfigCollection_PER_NODE;
|
|
constexpr int HloModuleConfigProto_FusionConfigCollection_FusionConfigCollection_ARRAYSIZE = HloModuleConfigProto_FusionConfigCollection_FusionConfigCollection_MAX + 1;
|
|
|
|
const ::PROTOBUF_NAMESPACE_ID::EnumDescriptor* HloModuleConfigProto_FusionConfigCollection_descriptor();
|
|
template<typename T>
|
|
inline const std::string& HloModuleConfigProto_FusionConfigCollection_Name(T enum_t_value) {
|
|
static_assert(::std::is_same<T, HloModuleConfigProto_FusionConfigCollection>::value ||
|
|
::std::is_integral<T>::value,
|
|
"Incorrect type passed to function HloModuleConfigProto_FusionConfigCollection_Name.");
|
|
return ::PROTOBUF_NAMESPACE_ID::internal::NameOfEnum(
|
|
HloModuleConfigProto_FusionConfigCollection_descriptor(), enum_t_value);
|
|
}
|
|
inline bool HloModuleConfigProto_FusionConfigCollection_Parse(
|
|
::PROTOBUF_NAMESPACE_ID::ConstStringParam name, HloModuleConfigProto_FusionConfigCollection* value) {
|
|
return ::PROTOBUF_NAMESPACE_ID::internal::ParseNamedEnum<HloModuleConfigProto_FusionConfigCollection>(
|
|
HloModuleConfigProto_FusionConfigCollection_descriptor(), name, value);
|
|
}
|
|
// ===================================================================
|
|
|
|
class CompilationEnvironmentsProto final :
|
|
public ::PROTOBUF_NAMESPACE_ID::Message /* @@protoc_insertion_point(class_definition:xla.CompilationEnvironmentsProto) */ {
|
|
public:
|
|
inline CompilationEnvironmentsProto() : CompilationEnvironmentsProto(nullptr) {}
|
|
~CompilationEnvironmentsProto() override;
|
|
explicit PROTOBUF_CONSTEXPR CompilationEnvironmentsProto(::PROTOBUF_NAMESPACE_ID::internal::ConstantInitialized);
|
|
|
|
CompilationEnvironmentsProto(const CompilationEnvironmentsProto& from);
|
|
CompilationEnvironmentsProto(CompilationEnvironmentsProto&& from) noexcept
|
|
: CompilationEnvironmentsProto() {
|
|
*this = ::std::move(from);
|
|
}
|
|
|
|
inline CompilationEnvironmentsProto& operator=(const CompilationEnvironmentsProto& from) {
|
|
CopyFrom(from);
|
|
return *this;
|
|
}
|
|
inline CompilationEnvironmentsProto& operator=(CompilationEnvironmentsProto&& 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 CompilationEnvironmentsProto& default_instance() {
|
|
return *internal_default_instance();
|
|
}
|
|
static inline const CompilationEnvironmentsProto* internal_default_instance() {
|
|
return reinterpret_cast<const CompilationEnvironmentsProto*>(
|
|
&_CompilationEnvironmentsProto_default_instance_);
|
|
}
|
|
static constexpr int kIndexInFileMessages =
|
|
0;
|
|
|
|
friend void swap(CompilationEnvironmentsProto& a, CompilationEnvironmentsProto& b) {
|
|
a.Swap(&b);
|
|
}
|
|
inline void Swap(CompilationEnvironmentsProto* 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(CompilationEnvironmentsProto* other) {
|
|
if (other == this) return;
|
|
GOOGLE_DCHECK(GetOwningArena() == other->GetOwningArena());
|
|
InternalSwap(other);
|
|
}
|
|
|
|
// implements Message ----------------------------------------------
|
|
|
|
CompilationEnvironmentsProto* New(::PROTOBUF_NAMESPACE_ID::Arena* arena = nullptr) const final {
|
|
return CreateMaybeMessage<CompilationEnvironmentsProto>(arena);
|
|
}
|
|
using ::PROTOBUF_NAMESPACE_ID::Message::CopyFrom;
|
|
void CopyFrom(const CompilationEnvironmentsProto& from);
|
|
using ::PROTOBUF_NAMESPACE_ID::Message::MergeFrom;
|
|
void MergeFrom( const CompilationEnvironmentsProto& from) {
|
|
CompilationEnvironmentsProto::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(CompilationEnvironmentsProto* other);
|
|
|
|
private:
|
|
friend class ::PROTOBUF_NAMESPACE_ID::internal::AnyMetadata;
|
|
static ::PROTOBUF_NAMESPACE_ID::StringPiece FullMessageName() {
|
|
return "xla.CompilationEnvironmentsProto";
|
|
}
|
|
protected:
|
|
explicit CompilationEnvironmentsProto(::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 {
|
|
kEnvironmentsFieldNumber = 1,
|
|
};
|
|
// repeated .google.protobuf.Any environments = 1;
|
|
int environments_size() const;
|
|
private:
|
|
int _internal_environments_size() const;
|
|
public:
|
|
void clear_environments();
|
|
::PROTOBUF_NAMESPACE_ID::Any* mutable_environments(int index);
|
|
::PROTOBUF_NAMESPACE_ID::RepeatedPtrField< ::PROTOBUF_NAMESPACE_ID::Any >*
|
|
mutable_environments();
|
|
private:
|
|
const ::PROTOBUF_NAMESPACE_ID::Any& _internal_environments(int index) const;
|
|
::PROTOBUF_NAMESPACE_ID::Any* _internal_add_environments();
|
|
public:
|
|
const ::PROTOBUF_NAMESPACE_ID::Any& environments(int index) const;
|
|
::PROTOBUF_NAMESPACE_ID::Any* add_environments();
|
|
const ::PROTOBUF_NAMESPACE_ID::RepeatedPtrField< ::PROTOBUF_NAMESPACE_ID::Any >&
|
|
environments() const;
|
|
|
|
// @@protoc_insertion_point(class_scope:xla.CompilationEnvironmentsProto)
|
|
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< ::PROTOBUF_NAMESPACE_ID::Any > environments_;
|
|
mutable ::PROTOBUF_NAMESPACE_ID::internal::CachedSize _cached_size_;
|
|
};
|
|
union { Impl_ _impl_; };
|
|
friend struct ::TableStruct_xla_2fxla_2eproto;
|
|
};
|
|
// -------------------------------------------------------------------
|
|
|
|
class DebugOptions_XlaBackendExtraOptionsEntry_DoNotUse : public ::PROTOBUF_NAMESPACE_ID::internal::MapEntry<DebugOptions_XlaBackendExtraOptionsEntry_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<DebugOptions_XlaBackendExtraOptionsEntry_DoNotUse,
|
|
std::string, std::string,
|
|
::PROTOBUF_NAMESPACE_ID::internal::WireFormatLite::TYPE_STRING,
|
|
::PROTOBUF_NAMESPACE_ID::internal::WireFormatLite::TYPE_STRING> SuperType;
|
|
DebugOptions_XlaBackendExtraOptionsEntry_DoNotUse();
|
|
explicit PROTOBUF_CONSTEXPR DebugOptions_XlaBackendExtraOptionsEntry_DoNotUse(
|
|
::PROTOBUF_NAMESPACE_ID::internal::ConstantInitialized);
|
|
explicit DebugOptions_XlaBackendExtraOptionsEntry_DoNotUse(::PROTOBUF_NAMESPACE_ID::Arena* arena);
|
|
void MergeFrom(const DebugOptions_XlaBackendExtraOptionsEntry_DoNotUse& other);
|
|
static const DebugOptions_XlaBackendExtraOptionsEntry_DoNotUse* internal_default_instance() { return reinterpret_cast<const DebugOptions_XlaBackendExtraOptionsEntry_DoNotUse*>(&_DebugOptions_XlaBackendExtraOptionsEntry_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.DebugOptions.XlaBackendExtraOptionsEntry.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.DebugOptions.XlaBackendExtraOptionsEntry.value");
|
|
}
|
|
using ::PROTOBUF_NAMESPACE_ID::Message::MergeFrom;
|
|
::PROTOBUF_NAMESPACE_ID::Metadata GetMetadata() const final;
|
|
friend struct ::TableStruct_xla_2fxla_2eproto;
|
|
};
|
|
|
|
// -------------------------------------------------------------------
|
|
|
|
class DebugOptions final :
|
|
public ::PROTOBUF_NAMESPACE_ID::Message /* @@protoc_insertion_point(class_definition:xla.DebugOptions) */ {
|
|
public:
|
|
inline DebugOptions() : DebugOptions(nullptr) {}
|
|
~DebugOptions() override;
|
|
explicit PROTOBUF_CONSTEXPR DebugOptions(::PROTOBUF_NAMESPACE_ID::internal::ConstantInitialized);
|
|
|
|
DebugOptions(const DebugOptions& from);
|
|
DebugOptions(DebugOptions&& from) noexcept
|
|
: DebugOptions() {
|
|
*this = ::std::move(from);
|
|
}
|
|
|
|
inline DebugOptions& operator=(const DebugOptions& from) {
|
|
CopyFrom(from);
|
|
return *this;
|
|
}
|
|
inline DebugOptions& operator=(DebugOptions&& 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 DebugOptions& default_instance() {
|
|
return *internal_default_instance();
|
|
}
|
|
static inline const DebugOptions* internal_default_instance() {
|
|
return reinterpret_cast<const DebugOptions*>(
|
|
&_DebugOptions_default_instance_);
|
|
}
|
|
static constexpr int kIndexInFileMessages =
|
|
2;
|
|
|
|
friend void swap(DebugOptions& a, DebugOptions& b) {
|
|
a.Swap(&b);
|
|
}
|
|
inline void Swap(DebugOptions* 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(DebugOptions* other) {
|
|
if (other == this) return;
|
|
GOOGLE_DCHECK(GetOwningArena() == other->GetOwningArena());
|
|
InternalSwap(other);
|
|
}
|
|
|
|
// implements Message ----------------------------------------------
|
|
|
|
DebugOptions* New(::PROTOBUF_NAMESPACE_ID::Arena* arena = nullptr) const final {
|
|
return CreateMaybeMessage<DebugOptions>(arena);
|
|
}
|
|
using ::PROTOBUF_NAMESPACE_ID::Message::CopyFrom;
|
|
void CopyFrom(const DebugOptions& from);
|
|
using ::PROTOBUF_NAMESPACE_ID::Message::MergeFrom;
|
|
void MergeFrom( const DebugOptions& from) {
|
|
DebugOptions::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(DebugOptions* other);
|
|
|
|
private:
|
|
friend class ::PROTOBUF_NAMESPACE_ID::internal::AnyMetadata;
|
|
static ::PROTOBUF_NAMESPACE_ID::StringPiece FullMessageName() {
|
|
return "xla.DebugOptions";
|
|
}
|
|
protected:
|
|
explicit DebugOptions(::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 ----------------------------------------------------
|
|
|
|
|
|
typedef DebugOptions_ShapeChecks ShapeChecks;
|
|
static constexpr ShapeChecks IGNORE =
|
|
DebugOptions_ShapeChecks_IGNORE;
|
|
static constexpr ShapeChecks RUNTIME =
|
|
DebugOptions_ShapeChecks_RUNTIME;
|
|
static constexpr ShapeChecks COMPILE_TIME =
|
|
DebugOptions_ShapeChecks_COMPILE_TIME;
|
|
static inline bool ShapeChecks_IsValid(int value) {
|
|
return DebugOptions_ShapeChecks_IsValid(value);
|
|
}
|
|
static constexpr ShapeChecks ShapeChecks_MIN =
|
|
DebugOptions_ShapeChecks_ShapeChecks_MIN;
|
|
static constexpr ShapeChecks ShapeChecks_MAX =
|
|
DebugOptions_ShapeChecks_ShapeChecks_MAX;
|
|
static constexpr int ShapeChecks_ARRAYSIZE =
|
|
DebugOptions_ShapeChecks_ShapeChecks_ARRAYSIZE;
|
|
static inline const ::PROTOBUF_NAMESPACE_ID::EnumDescriptor*
|
|
ShapeChecks_descriptor() {
|
|
return DebugOptions_ShapeChecks_descriptor();
|
|
}
|
|
template<typename T>
|
|
static inline const std::string& ShapeChecks_Name(T enum_t_value) {
|
|
static_assert(::std::is_same<T, ShapeChecks>::value ||
|
|
::std::is_integral<T>::value,
|
|
"Incorrect type passed to function ShapeChecks_Name.");
|
|
return DebugOptions_ShapeChecks_Name(enum_t_value);
|
|
}
|
|
static inline bool ShapeChecks_Parse(::PROTOBUF_NAMESPACE_ID::ConstStringParam name,
|
|
ShapeChecks* value) {
|
|
return DebugOptions_ShapeChecks_Parse(name, value);
|
|
}
|
|
|
|
typedef DebugOptions_StepMarkerLocation StepMarkerLocation;
|
|
static constexpr StepMarkerLocation STEP_MARK_AT_ENTRY =
|
|
DebugOptions_StepMarkerLocation_STEP_MARK_AT_ENTRY;
|
|
static constexpr StepMarkerLocation STEP_MARK_AT_TOP_LEVEL_WHILE_LOOP =
|
|
DebugOptions_StepMarkerLocation_STEP_MARK_AT_TOP_LEVEL_WHILE_LOOP;
|
|
static constexpr StepMarkerLocation STEP_MARK_AT_SECOND_LEVEL_WHILE_LOOP =
|
|
DebugOptions_StepMarkerLocation_STEP_MARK_AT_SECOND_LEVEL_WHILE_LOOP;
|
|
static constexpr StepMarkerLocation STEP_MARK_NONE =
|
|
DebugOptions_StepMarkerLocation_STEP_MARK_NONE;
|
|
static inline bool StepMarkerLocation_IsValid(int value) {
|
|
return DebugOptions_StepMarkerLocation_IsValid(value);
|
|
}
|
|
static constexpr StepMarkerLocation StepMarkerLocation_MIN =
|
|
DebugOptions_StepMarkerLocation_StepMarkerLocation_MIN;
|
|
static constexpr StepMarkerLocation StepMarkerLocation_MAX =
|
|
DebugOptions_StepMarkerLocation_StepMarkerLocation_MAX;
|
|
static constexpr int StepMarkerLocation_ARRAYSIZE =
|
|
DebugOptions_StepMarkerLocation_StepMarkerLocation_ARRAYSIZE;
|
|
static inline const ::PROTOBUF_NAMESPACE_ID::EnumDescriptor*
|
|
StepMarkerLocation_descriptor() {
|
|
return DebugOptions_StepMarkerLocation_descriptor();
|
|
}
|
|
template<typename T>
|
|
static inline const std::string& StepMarkerLocation_Name(T enum_t_value) {
|
|
static_assert(::std::is_same<T, StepMarkerLocation>::value ||
|
|
::std::is_integral<T>::value,
|
|
"Incorrect type passed to function StepMarkerLocation_Name.");
|
|
return DebugOptions_StepMarkerLocation_Name(enum_t_value);
|
|
}
|
|
static inline bool StepMarkerLocation_Parse(::PROTOBUF_NAMESPACE_ID::ConstStringParam name,
|
|
StepMarkerLocation* value) {
|
|
return DebugOptions_StepMarkerLocation_Parse(name, value);
|
|
}
|
|
|
|
typedef DebugOptions_CommandBufferCmdType CommandBufferCmdType;
|
|
static constexpr CommandBufferCmdType INVALID =
|
|
DebugOptions_CommandBufferCmdType_INVALID;
|
|
static constexpr CommandBufferCmdType FUSION =
|
|
DebugOptions_CommandBufferCmdType_FUSION;
|
|
static constexpr CommandBufferCmdType CUBLAS =
|
|
DebugOptions_CommandBufferCmdType_CUBLAS;
|
|
static constexpr CommandBufferCmdType CUDNN =
|
|
DebugOptions_CommandBufferCmdType_CUDNN;
|
|
static constexpr CommandBufferCmdType COLLECTIVES =
|
|
DebugOptions_CommandBufferCmdType_COLLECTIVES;
|
|
static constexpr CommandBufferCmdType CONDITIONALS =
|
|
DebugOptions_CommandBufferCmdType_CONDITIONALS;
|
|
static constexpr CommandBufferCmdType CUSTOM_CALL =
|
|
DebugOptions_CommandBufferCmdType_CUSTOM_CALL;
|
|
static inline bool CommandBufferCmdType_IsValid(int value) {
|
|
return DebugOptions_CommandBufferCmdType_IsValid(value);
|
|
}
|
|
static constexpr CommandBufferCmdType CommandBufferCmdType_MIN =
|
|
DebugOptions_CommandBufferCmdType_CommandBufferCmdType_MIN;
|
|
static constexpr CommandBufferCmdType CommandBufferCmdType_MAX =
|
|
DebugOptions_CommandBufferCmdType_CommandBufferCmdType_MAX;
|
|
static constexpr int CommandBufferCmdType_ARRAYSIZE =
|
|
DebugOptions_CommandBufferCmdType_CommandBufferCmdType_ARRAYSIZE;
|
|
static inline const ::PROTOBUF_NAMESPACE_ID::EnumDescriptor*
|
|
CommandBufferCmdType_descriptor() {
|
|
return DebugOptions_CommandBufferCmdType_descriptor();
|
|
}
|
|
template<typename T>
|
|
static inline const std::string& CommandBufferCmdType_Name(T enum_t_value) {
|
|
static_assert(::std::is_same<T, CommandBufferCmdType>::value ||
|
|
::std::is_integral<T>::value,
|
|
"Incorrect type passed to function CommandBufferCmdType_Name.");
|
|
return DebugOptions_CommandBufferCmdType_Name(enum_t_value);
|
|
}
|
|
static inline bool CommandBufferCmdType_Parse(::PROTOBUF_NAMESPACE_ID::ConstStringParam name,
|
|
CommandBufferCmdType* value) {
|
|
return DebugOptions_CommandBufferCmdType_Parse(name, value);
|
|
}
|
|
|
|
typedef DebugOptions_PartitioningAlgorithm PartitioningAlgorithm;
|
|
static constexpr PartitioningAlgorithm PARTITIONING_ALGORITHM_NOOP =
|
|
DebugOptions_PartitioningAlgorithm_PARTITIONING_ALGORITHM_NOOP;
|
|
static constexpr PartitioningAlgorithm PARTITIONING_ALGORITHM_EXP0 =
|
|
DebugOptions_PartitioningAlgorithm_PARTITIONING_ALGORITHM_EXP0;
|
|
static constexpr PartitioningAlgorithm PARTITIONING_ALGORITHM_EXP1 =
|
|
DebugOptions_PartitioningAlgorithm_PARTITIONING_ALGORITHM_EXP1;
|
|
static constexpr PartitioningAlgorithm PARTITIONING_ALGORITHM_EXP2 =
|
|
DebugOptions_PartitioningAlgorithm_PARTITIONING_ALGORITHM_EXP2;
|
|
static inline bool PartitioningAlgorithm_IsValid(int value) {
|
|
return DebugOptions_PartitioningAlgorithm_IsValid(value);
|
|
}
|
|
static constexpr PartitioningAlgorithm PartitioningAlgorithm_MIN =
|
|
DebugOptions_PartitioningAlgorithm_PartitioningAlgorithm_MIN;
|
|
static constexpr PartitioningAlgorithm PartitioningAlgorithm_MAX =
|
|
DebugOptions_PartitioningAlgorithm_PartitioningAlgorithm_MAX;
|
|
static constexpr int PartitioningAlgorithm_ARRAYSIZE =
|
|
DebugOptions_PartitioningAlgorithm_PartitioningAlgorithm_ARRAYSIZE;
|
|
static inline const ::PROTOBUF_NAMESPACE_ID::EnumDescriptor*
|
|
PartitioningAlgorithm_descriptor() {
|
|
return DebugOptions_PartitioningAlgorithm_descriptor();
|
|
}
|
|
template<typename T>
|
|
static inline const std::string& PartitioningAlgorithm_Name(T enum_t_value) {
|
|
static_assert(::std::is_same<T, PartitioningAlgorithm>::value ||
|
|
::std::is_integral<T>::value,
|
|
"Incorrect type passed to function PartitioningAlgorithm_Name.");
|
|
return DebugOptions_PartitioningAlgorithm_Name(enum_t_value);
|
|
}
|
|
static inline bool PartitioningAlgorithm_Parse(::PROTOBUF_NAMESPACE_ID::ConstStringParam name,
|
|
PartitioningAlgorithm* value) {
|
|
return DebugOptions_PartitioningAlgorithm_Parse(name, value);
|
|
}
|
|
|
|
// accessors -------------------------------------------------------
|
|
|
|
enum : int {
|
|
kXlaDisableHloPassesFieldNumber = 30,
|
|
kXlaEnableHloPassesOnlyFieldNumber = 124,
|
|
kXlaGpuPtxFileFieldNumber = 127,
|
|
kXlaGpuLlvmIrFileFieldNumber = 150,
|
|
kXlaGpuEnableCommandBufferFieldNumber = 258,
|
|
kXlaBackendExtraOptionsFieldNumber = 500,
|
|
kXlaGpuCudaDataDirFieldNumber = 61,
|
|
kXlaDumpToFieldNumber = 109,
|
|
kXlaDumpHloModuleReFieldNumber = 110,
|
|
kXlaDumpHloPassReFieldNumber = 111,
|
|
kXlaGpuAlgorithmDenylistPathFieldNumber = 128,
|
|
kXlaGpuAsmExtraFlagsFieldNumber = 141,
|
|
kXlaDumpHloPipelineReFieldNumber = 154,
|
|
kXlaGpuPgleProfileFileOrDirectoryPathFieldNumber = 210,
|
|
kXlaGpuDumpAutotuneResultsToFieldNumber = 222,
|
|
kXlaGpuLoadAutotuneResultsFromFieldNumber = 223,
|
|
kXlaGpuTargetConfigFilenameFieldNumber = 261,
|
|
kXlaGpuEnableCustomFusionsReFieldNumber = 264,
|
|
kXlaBackendOptimizationLevelFieldNumber = 31,
|
|
kXlaHloGraphAddressesFieldNumber = 2,
|
|
kXlaHloProfileFieldNumber = 9,
|
|
kXlaDisableAllHloPassesFieldNumber = 104,
|
|
kXlaEmbedIrInExecutableFieldNumber = 33,
|
|
kXlaEliminateHloImplicitBroadcastFieldNumber = 35,
|
|
kXlaCpuMultiThreadEigenFieldNumber = 60,
|
|
kXlaGpuFtzFieldNumber = 62,
|
|
kXlaLlvmEnableAliasScopeMetadataFieldNumber = 70,
|
|
kXlaLlvmEnableNoaliasMetadataFieldNumber = 71,
|
|
kXlaLlvmEnableInvariantLoadMetadataFieldNumber = 72,
|
|
kXlaLlvmDisableExpensivePassesFieldNumber = 73,
|
|
kXlaTestAllOutputLayoutsFieldNumber = 90,
|
|
kXlaForceHostPlatformDeviceCountFieldNumber = 102,
|
|
kXlaGpuCrashOnVerificationFailuresFieldNumber = 101,
|
|
kXlaGpuDisableGpuasmOptimizationsFieldNumber = 103,
|
|
kXlaHloEvaluatorUseFastPathFieldNumber = 106,
|
|
kXlaAllowScalarIndexDynamicOpsFieldNumber = 107,
|
|
kXlaStepMarkerLocationFieldNumber = 108,
|
|
kXlaDumpHloAsTextFieldNumber = 112,
|
|
kXlaDumpHloAsProtoFieldNumber = 113,
|
|
kXlaDumpHloAsDotFieldNumber = 114,
|
|
kXlaDumpHloAsUrlFieldNumber = 115,
|
|
kXlaTestAllInputLayoutsFieldNumber = 91,
|
|
kXlaHloGraphShardingColorFieldNumber = 92,
|
|
kXlaCpuUseMklDnnFieldNumber = 97,
|
|
kXlaCpuUseXlaRuntimeFieldNumber = 177,
|
|
kXlaCpuEnableFastMathFieldNumber = 99,
|
|
kXlaCpuFastMathHonorNansFieldNumber = 120,
|
|
kXlaCpuFastMathHonorInfsFieldNumber = 121,
|
|
kXlaCpuFastMathHonorDivisionFieldNumber = 126,
|
|
kXlaCpuFastMathHonorFunctionsFieldNumber = 129,
|
|
kXlaCpuEnableFastMinMaxFieldNumber = 140,
|
|
kXlaGpuEnableFastMinMaxFieldNumber = 100,
|
|
kXlaAllowExcessPrecisionFieldNumber = 122,
|
|
kXlaGpuAutotuneLevelFieldNumber = 123,
|
|
kXlaDumpHloAsHtmlFieldNumber = 116,
|
|
kXlaDumpFusionVisualizationFieldNumber = 149,
|
|
kXlaDumpHloSnapshotsFieldNumber = 118,
|
|
kXlaDumpIncludeTimestampFieldNumber = 131,
|
|
kXlaDumpMaxHloModulesFieldNumber = 132,
|
|
kXlaMultiheapSizeConstraintPerHeapFieldNumber = 142,
|
|
kXlaDumpModuleMetadataFieldNumber = 144,
|
|
kXlaDumpCompressProtosFieldNumber = 151,
|
|
kXlaDumpHloAsLongTextFieldNumber = 164,
|
|
kXlaGpuForceConvNchwFieldNumber = 125,
|
|
kXlaGpuForceCompilationParallelismFieldNumber = 147,
|
|
kXlaGpuForceConvNhwcFieldNumber = 146,
|
|
kXlaGpuDumpLlvmirFieldNumber = 155,
|
|
kXlaDumpEnableMlirPrettyFormFieldNumber = 185,
|
|
kXlaTpuDetectNanFieldNumber = 135,
|
|
kXlaGpuAllReduceCombineThresholdBytesFieldNumber = 157,
|
|
kXlaGpuNcclTerminationTimeoutSecondsFieldNumber = 163,
|
|
kXlaGpuAllReduceBlueconnectNumDevicesPerHostFieldNumber = 159,
|
|
kXlaTpuDetectInfFieldNumber = 136,
|
|
kXlaCpuEnableXprofTracemeFieldNumber = 137,
|
|
kXlaGpuUnsafeFallbackToDriverOnPtxasNotFoundFieldNumber = 138,
|
|
kXlaDetailedLoggingFieldNumber = 252,
|
|
kXlaGpuRedzoneScratchMaxMegabytesFieldNumber = 167,
|
|
kXlaGpuShapeChecksFieldNumber = 170,
|
|
kXlaGpuEnableXlaRuntimeExecutableFieldNumber = 169,
|
|
kXlaGpuEnableCustomFusionsFieldNumber = 263,
|
|
kXlaGpuEnableAddressComputationFusionFieldNumber = 105,
|
|
kXlaGpuEnableSharedConstantsFieldNumber = 165,
|
|
kXlaCpuUseAclFieldNumber = 174,
|
|
kXlaCpuStrictDotConvMathFieldNumber = 175,
|
|
kXlaGpuUseRuntimeFusionFieldNumber = 181,
|
|
kXlaDumpLatencyHidingScheduleFieldNumber = 182,
|
|
kXlaGpuEnableCublasltFieldNumber = 166,
|
|
kXlaGpuGraphEnableConcurrentRegionFieldNumber = 215,
|
|
kXlaGpuSimplifyAllFpConversionsFieldNumber = 168,
|
|
kXlaGpuNormalizeLayoutsFieldNumber = 172,
|
|
kXlaGpuEnableAsyncAllReduceFieldNumber = 152,
|
|
kXlaGpuEnableAsyncCollectivePermuteFieldNumber = 183,
|
|
kXlaGpuEnableAsyncAllGatherFieldNumber = 199,
|
|
kXlaGpuEnableAsyncReduceScatterFieldNumber = 200,
|
|
kXlaPartitioningAlgorithmFieldNumber = 187,
|
|
kXlaCpuEnableMlirTilingAndFusionFieldNumber = 184,
|
|
kXlaCpuEnableCustomMatmulTilingFieldNumber = 195,
|
|
kXlaCpuEnableMlirFusionOutliningFieldNumber = 192,
|
|
kXlaCpuEnableExperimentalDeallocationFieldNumber = 191,
|
|
kXlaGpuEnableReassociationForConvertedArFieldNumber = 209,
|
|
kXlaGpuEnableWhileLoopReduceScatterCodeMotionFieldNumber = 203,
|
|
kXlaGpuEnableCudnnFrontendFieldNumber = 160,
|
|
kXlaGpuEnableCudnnFmhaFieldNumber = 218,
|
|
kXlaCpuMatmulTilingMDimFieldNumber = 196,
|
|
kXlaCpuMatmulTilingNDimFieldNumber = 197,
|
|
kXlaCpuMatmulTilingKDimFieldNumber = 198,
|
|
kXlaGpuFusedAttentionUseCudnnRngFieldNumber = 235,
|
|
kXlaGpuEnableCudnnLayerNormFieldNumber = 262,
|
|
kXlaDumpDisableMetadataFieldNumber = 153,
|
|
kXlaGpuStrictConvAlgorithmPickerFieldNumber = 156,
|
|
kXlaGpuGraphNumRunsToInstantiateFieldNumber = 202,
|
|
kXlaGpuEnablePipelinedP2PFieldNumber = 246,
|
|
kXlaGpuEnableTritonGemmFieldNumber = 188,
|
|
kXlaGpuEnableCudnnInt8X32ConvolutionReorderingFieldNumber = 189,
|
|
kXlaGpuTritonGemmAnyFieldNumber = 190,
|
|
kXlaGpuCollectiveInflationFactorFieldNumber = 205,
|
|
kXlaCpuSparseCudaThreadsFieldNumber = 207,
|
|
kXlaGpuGraphMinGraphSizeFieldNumber = 208,
|
|
kXlaGpuAllGatherCombineThresholdBytesFieldNumber = 212,
|
|
kXlaGpuReduceScatterCombineThresholdBytesFieldNumber = 213,
|
|
kXlaGpuEnableLatencyHidingSchedulerFieldNumber = 186,
|
|
kXlaGpuEnableHighestPriorityAsyncStreamFieldNumber = 216,
|
|
kXlaGpuEnableAnalyticalLatencyEstimatorFieldNumber = 255,
|
|
kXlaGpuLhsEnableGpuAsyncTrackerFieldNumber = 204,
|
|
kXlaGpuEnableAsyncAllToAllFieldNumber = 201,
|
|
kXlaGpuEnableAllGatherCombineByDimFieldNumber = 254,
|
|
kXlaGpuEnableReduceScatterCombineByDimFieldNumber = 257,
|
|
kXlaGpuAllReduceContiguousFieldNumber = 158,
|
|
kXlaGpuExhaustiveTilingSearchFieldNumber = 219,
|
|
kXlaGpuEnableTritonSoftmaxFusionFieldNumber = 220,
|
|
kXlaGpuEnablePriorityFusionFieldNumber = 221,
|
|
kXlaGpuTritonGemmDisableReducedPrecisionReductionFieldNumber = 226,
|
|
kXlaGpuAutoSpmdPartitioningMemoryBudgetGbFieldNumber = 224,
|
|
kXlaGpuAutoSpmdPartitioningMemoryBudgetRatioFieldNumber = 225,
|
|
kXlaEnableDumpingFieldNumber = 253,
|
|
kXlaGpuEnableLlvmModuleCompilationParallelismFieldNumber = 268,
|
|
kXlaGpuDeterministicOpsFieldNumber = 148,
|
|
kXlaGpuEnableAsyncCollectivesFieldNumber = 238,
|
|
kXlaGpuRedzonePaddingBytesFieldNumber = 228,
|
|
kXlaGpuEnablePipelinedCollectivesFieldNumber = 239,
|
|
kXlaGpuEnablePipelinedAllReduceFieldNumber = 217,
|
|
kXlaGpuEnablePipelinedAllGatherFieldNumber = 227,
|
|
kXlaGpuEnablePipelinedReduceScatterFieldNumber = 231,
|
|
kXlaGpuTritonFusionLevelFieldNumber = 229,
|
|
kXlaGpuGraphEvictionTimeoutSecondsFieldNumber = 230,
|
|
kXlaGpuDumpAutotunedTritonFusionsFieldNumber = 232,
|
|
kXlaGpuCopyInsertionUseRegionAnalysisFieldNumber = 236,
|
|
kXlaGpuCollectCostModelStatsFieldNumber = 240,
|
|
kXlaGpuEnableSplitKAutotuningFieldNumber = 241,
|
|
kXlaGpuCollectivePermuteDecomposerThresholdFieldNumber = 237,
|
|
kXlaGpuEnableReductionEpilogueFusionFieldNumber = 243,
|
|
kXlaGpuEnableNcclCliqueOptimizationFieldNumber = 244,
|
|
kXlaGpuMockCustomCallsFieldNumber = 245,
|
|
kXlaGpuCublasFallbackFieldNumber = 247,
|
|
kXlaGpuEnableWhileLoopDoubleBufferingFieldNumber = 248,
|
|
kXlaGpuEnsureMinorDotContractionDimsFieldNumber = 249,
|
|
kXlaGpuFilterKernelsSpillingRegistersOnAutotuningFieldNumber = 250,
|
|
kXlaGpuEnableCubRadixSortFieldNumber = 259,
|
|
kXlaDebugBufferAssignmentShowMaxFieldNumber = 251,
|
|
kXlaGpuLlvmVerificationLevelFieldNumber = 256,
|
|
kXlaGpuMemoryLimitSlopFactorFieldNumber = 260,
|
|
kXlaGpuThresholdForWindowedEinsumMibFieldNumber = 265,
|
|
kXlaGpuEnableTritonHopperFieldNumber = 266,
|
|
kXlaGpuEnableNcclUserBuffersFieldNumber = 267,
|
|
kXlaGpuEnableLibnvptxcompilerFieldNumber = 269,
|
|
kXlaGpuEnableDotStrengthReductionFieldNumber = 270,
|
|
};
|
|
// repeated string xla_disable_hlo_passes = 30;
|
|
int xla_disable_hlo_passes_size() const;
|
|
private:
|
|
int _internal_xla_disable_hlo_passes_size() const;
|
|
public:
|
|
void clear_xla_disable_hlo_passes();
|
|
const std::string& xla_disable_hlo_passes(int index) const;
|
|
std::string* mutable_xla_disable_hlo_passes(int index);
|
|
void set_xla_disable_hlo_passes(int index, const std::string& value);
|
|
void set_xla_disable_hlo_passes(int index, std::string&& value);
|
|
void set_xla_disable_hlo_passes(int index, const char* value);
|
|
void set_xla_disable_hlo_passes(int index, const char* value, size_t size);
|
|
std::string* add_xla_disable_hlo_passes();
|
|
void add_xla_disable_hlo_passes(const std::string& value);
|
|
void add_xla_disable_hlo_passes(std::string&& value);
|
|
void add_xla_disable_hlo_passes(const char* value);
|
|
void add_xla_disable_hlo_passes(const char* value, size_t size);
|
|
const ::PROTOBUF_NAMESPACE_ID::RepeatedPtrField<std::string>& xla_disable_hlo_passes() const;
|
|
::PROTOBUF_NAMESPACE_ID::RepeatedPtrField<std::string>* mutable_xla_disable_hlo_passes();
|
|
private:
|
|
const std::string& _internal_xla_disable_hlo_passes(int index) const;
|
|
std::string* _internal_add_xla_disable_hlo_passes();
|
|
public:
|
|
|
|
// repeated string xla_enable_hlo_passes_only = 124;
|
|
int xla_enable_hlo_passes_only_size() const;
|
|
private:
|
|
int _internal_xla_enable_hlo_passes_only_size() const;
|
|
public:
|
|
void clear_xla_enable_hlo_passes_only();
|
|
const std::string& xla_enable_hlo_passes_only(int index) const;
|
|
std::string* mutable_xla_enable_hlo_passes_only(int index);
|
|
void set_xla_enable_hlo_passes_only(int index, const std::string& value);
|
|
void set_xla_enable_hlo_passes_only(int index, std::string&& value);
|
|
void set_xla_enable_hlo_passes_only(int index, const char* value);
|
|
void set_xla_enable_hlo_passes_only(int index, const char* value, size_t size);
|
|
std::string* add_xla_enable_hlo_passes_only();
|
|
void add_xla_enable_hlo_passes_only(const std::string& value);
|
|
void add_xla_enable_hlo_passes_only(std::string&& value);
|
|
void add_xla_enable_hlo_passes_only(const char* value);
|
|
void add_xla_enable_hlo_passes_only(const char* value, size_t size);
|
|
const ::PROTOBUF_NAMESPACE_ID::RepeatedPtrField<std::string>& xla_enable_hlo_passes_only() const;
|
|
::PROTOBUF_NAMESPACE_ID::RepeatedPtrField<std::string>* mutable_xla_enable_hlo_passes_only();
|
|
private:
|
|
const std::string& _internal_xla_enable_hlo_passes_only(int index) const;
|
|
std::string* _internal_add_xla_enable_hlo_passes_only();
|
|
public:
|
|
|
|
// repeated string xla_gpu_ptx_file = 127;
|
|
int xla_gpu_ptx_file_size() const;
|
|
private:
|
|
int _internal_xla_gpu_ptx_file_size() const;
|
|
public:
|
|
void clear_xla_gpu_ptx_file();
|
|
const std::string& xla_gpu_ptx_file(int index) const;
|
|
std::string* mutable_xla_gpu_ptx_file(int index);
|
|
void set_xla_gpu_ptx_file(int index, const std::string& value);
|
|
void set_xla_gpu_ptx_file(int index, std::string&& value);
|
|
void set_xla_gpu_ptx_file(int index, const char* value);
|
|
void set_xla_gpu_ptx_file(int index, const char* value, size_t size);
|
|
std::string* add_xla_gpu_ptx_file();
|
|
void add_xla_gpu_ptx_file(const std::string& value);
|
|
void add_xla_gpu_ptx_file(std::string&& value);
|
|
void add_xla_gpu_ptx_file(const char* value);
|
|
void add_xla_gpu_ptx_file(const char* value, size_t size);
|
|
const ::PROTOBUF_NAMESPACE_ID::RepeatedPtrField<std::string>& xla_gpu_ptx_file() const;
|
|
::PROTOBUF_NAMESPACE_ID::RepeatedPtrField<std::string>* mutable_xla_gpu_ptx_file();
|
|
private:
|
|
const std::string& _internal_xla_gpu_ptx_file(int index) const;
|
|
std::string* _internal_add_xla_gpu_ptx_file();
|
|
public:
|
|
|
|
// repeated string xla_gpu_llvm_ir_file = 150;
|
|
int xla_gpu_llvm_ir_file_size() const;
|
|
private:
|
|
int _internal_xla_gpu_llvm_ir_file_size() const;
|
|
public:
|
|
void clear_xla_gpu_llvm_ir_file();
|
|
const std::string& xla_gpu_llvm_ir_file(int index) const;
|
|
std::string* mutable_xla_gpu_llvm_ir_file(int index);
|
|
void set_xla_gpu_llvm_ir_file(int index, const std::string& value);
|
|
void set_xla_gpu_llvm_ir_file(int index, std::string&& value);
|
|
void set_xla_gpu_llvm_ir_file(int index, const char* value);
|
|
void set_xla_gpu_llvm_ir_file(int index, const char* value, size_t size);
|
|
std::string* add_xla_gpu_llvm_ir_file();
|
|
void add_xla_gpu_llvm_ir_file(const std::string& value);
|
|
void add_xla_gpu_llvm_ir_file(std::string&& value);
|
|
void add_xla_gpu_llvm_ir_file(const char* value);
|
|
void add_xla_gpu_llvm_ir_file(const char* value, size_t size);
|
|
const ::PROTOBUF_NAMESPACE_ID::RepeatedPtrField<std::string>& xla_gpu_llvm_ir_file() const;
|
|
::PROTOBUF_NAMESPACE_ID::RepeatedPtrField<std::string>* mutable_xla_gpu_llvm_ir_file();
|
|
private:
|
|
const std::string& _internal_xla_gpu_llvm_ir_file(int index) const;
|
|
std::string* _internal_add_xla_gpu_llvm_ir_file();
|
|
public:
|
|
|
|
// repeated .xla.DebugOptions.CommandBufferCmdType xla_gpu_enable_command_buffer = 258;
|
|
int xla_gpu_enable_command_buffer_size() const;
|
|
private:
|
|
int _internal_xla_gpu_enable_command_buffer_size() const;
|
|
public:
|
|
void clear_xla_gpu_enable_command_buffer();
|
|
private:
|
|
::xla::DebugOptions_CommandBufferCmdType _internal_xla_gpu_enable_command_buffer(int index) const;
|
|
void _internal_add_xla_gpu_enable_command_buffer(::xla::DebugOptions_CommandBufferCmdType value);
|
|
::PROTOBUF_NAMESPACE_ID::RepeatedField<int>* _internal_mutable_xla_gpu_enable_command_buffer();
|
|
public:
|
|
::xla::DebugOptions_CommandBufferCmdType xla_gpu_enable_command_buffer(int index) const;
|
|
void set_xla_gpu_enable_command_buffer(int index, ::xla::DebugOptions_CommandBufferCmdType value);
|
|
void add_xla_gpu_enable_command_buffer(::xla::DebugOptions_CommandBufferCmdType value);
|
|
const ::PROTOBUF_NAMESPACE_ID::RepeatedField<int>& xla_gpu_enable_command_buffer() const;
|
|
::PROTOBUF_NAMESPACE_ID::RepeatedField<int>* mutable_xla_gpu_enable_command_buffer();
|
|
|
|
// map<string, string> xla_backend_extra_options = 500;
|
|
int xla_backend_extra_options_size() const;
|
|
private:
|
|
int _internal_xla_backend_extra_options_size() const;
|
|
public:
|
|
void clear_xla_backend_extra_options();
|
|
private:
|
|
const ::PROTOBUF_NAMESPACE_ID::Map< std::string, std::string >&
|
|
_internal_xla_backend_extra_options() const;
|
|
::PROTOBUF_NAMESPACE_ID::Map< std::string, std::string >*
|
|
_internal_mutable_xla_backend_extra_options();
|
|
public:
|
|
const ::PROTOBUF_NAMESPACE_ID::Map< std::string, std::string >&
|
|
xla_backend_extra_options() const;
|
|
::PROTOBUF_NAMESPACE_ID::Map< std::string, std::string >*
|
|
mutable_xla_backend_extra_options();
|
|
|
|
// string xla_gpu_cuda_data_dir = 61;
|
|
void clear_xla_gpu_cuda_data_dir();
|
|
const std::string& xla_gpu_cuda_data_dir() const;
|
|
template <typename ArgT0 = const std::string&, typename... ArgT>
|
|
void set_xla_gpu_cuda_data_dir(ArgT0&& arg0, ArgT... args);
|
|
std::string* mutable_xla_gpu_cuda_data_dir();
|
|
PROTOBUF_NODISCARD std::string* release_xla_gpu_cuda_data_dir();
|
|
void set_allocated_xla_gpu_cuda_data_dir(std::string* xla_gpu_cuda_data_dir);
|
|
private:
|
|
const std::string& _internal_xla_gpu_cuda_data_dir() const;
|
|
inline PROTOBUF_ALWAYS_INLINE void _internal_set_xla_gpu_cuda_data_dir(const std::string& value);
|
|
std::string* _internal_mutable_xla_gpu_cuda_data_dir();
|
|
public:
|
|
|
|
// string xla_dump_to = 109;
|
|
void clear_xla_dump_to();
|
|
const std::string& xla_dump_to() const;
|
|
template <typename ArgT0 = const std::string&, typename... ArgT>
|
|
void set_xla_dump_to(ArgT0&& arg0, ArgT... args);
|
|
std::string* mutable_xla_dump_to();
|
|
PROTOBUF_NODISCARD std::string* release_xla_dump_to();
|
|
void set_allocated_xla_dump_to(std::string* xla_dump_to);
|
|
private:
|
|
const std::string& _internal_xla_dump_to() const;
|
|
inline PROTOBUF_ALWAYS_INLINE void _internal_set_xla_dump_to(const std::string& value);
|
|
std::string* _internal_mutable_xla_dump_to();
|
|
public:
|
|
|
|
// string xla_dump_hlo_module_re = 110;
|
|
void clear_xla_dump_hlo_module_re();
|
|
const std::string& xla_dump_hlo_module_re() const;
|
|
template <typename ArgT0 = const std::string&, typename... ArgT>
|
|
void set_xla_dump_hlo_module_re(ArgT0&& arg0, ArgT... args);
|
|
std::string* mutable_xla_dump_hlo_module_re();
|
|
PROTOBUF_NODISCARD std::string* release_xla_dump_hlo_module_re();
|
|
void set_allocated_xla_dump_hlo_module_re(std::string* xla_dump_hlo_module_re);
|
|
private:
|
|
const std::string& _internal_xla_dump_hlo_module_re() const;
|
|
inline PROTOBUF_ALWAYS_INLINE void _internal_set_xla_dump_hlo_module_re(const std::string& value);
|
|
std::string* _internal_mutable_xla_dump_hlo_module_re();
|
|
public:
|
|
|
|
// string xla_dump_hlo_pass_re = 111;
|
|
void clear_xla_dump_hlo_pass_re();
|
|
const std::string& xla_dump_hlo_pass_re() const;
|
|
template <typename ArgT0 = const std::string&, typename... ArgT>
|
|
void set_xla_dump_hlo_pass_re(ArgT0&& arg0, ArgT... args);
|
|
std::string* mutable_xla_dump_hlo_pass_re();
|
|
PROTOBUF_NODISCARD std::string* release_xla_dump_hlo_pass_re();
|
|
void set_allocated_xla_dump_hlo_pass_re(std::string* xla_dump_hlo_pass_re);
|
|
private:
|
|
const std::string& _internal_xla_dump_hlo_pass_re() const;
|
|
inline PROTOBUF_ALWAYS_INLINE void _internal_set_xla_dump_hlo_pass_re(const std::string& value);
|
|
std::string* _internal_mutable_xla_dump_hlo_pass_re();
|
|
public:
|
|
|
|
// string xla_gpu_algorithm_denylist_path = 128;
|
|
void clear_xla_gpu_algorithm_denylist_path();
|
|
const std::string& xla_gpu_algorithm_denylist_path() const;
|
|
template <typename ArgT0 = const std::string&, typename... ArgT>
|
|
void set_xla_gpu_algorithm_denylist_path(ArgT0&& arg0, ArgT... args);
|
|
std::string* mutable_xla_gpu_algorithm_denylist_path();
|
|
PROTOBUF_NODISCARD std::string* release_xla_gpu_algorithm_denylist_path();
|
|
void set_allocated_xla_gpu_algorithm_denylist_path(std::string* xla_gpu_algorithm_denylist_path);
|
|
private:
|
|
const std::string& _internal_xla_gpu_algorithm_denylist_path() const;
|
|
inline PROTOBUF_ALWAYS_INLINE void _internal_set_xla_gpu_algorithm_denylist_path(const std::string& value);
|
|
std::string* _internal_mutable_xla_gpu_algorithm_denylist_path();
|
|
public:
|
|
|
|
// string xla_gpu_asm_extra_flags = 141;
|
|
void clear_xla_gpu_asm_extra_flags();
|
|
const std::string& xla_gpu_asm_extra_flags() const;
|
|
template <typename ArgT0 = const std::string&, typename... ArgT>
|
|
void set_xla_gpu_asm_extra_flags(ArgT0&& arg0, ArgT... args);
|
|
std::string* mutable_xla_gpu_asm_extra_flags();
|
|
PROTOBUF_NODISCARD std::string* release_xla_gpu_asm_extra_flags();
|
|
void set_allocated_xla_gpu_asm_extra_flags(std::string* xla_gpu_asm_extra_flags);
|
|
private:
|
|
const std::string& _internal_xla_gpu_asm_extra_flags() const;
|
|
inline PROTOBUF_ALWAYS_INLINE void _internal_set_xla_gpu_asm_extra_flags(const std::string& value);
|
|
std::string* _internal_mutable_xla_gpu_asm_extra_flags();
|
|
public:
|
|
|
|
// string xla_dump_hlo_pipeline_re = 154;
|
|
void clear_xla_dump_hlo_pipeline_re();
|
|
const std::string& xla_dump_hlo_pipeline_re() const;
|
|
template <typename ArgT0 = const std::string&, typename... ArgT>
|
|
void set_xla_dump_hlo_pipeline_re(ArgT0&& arg0, ArgT... args);
|
|
std::string* mutable_xla_dump_hlo_pipeline_re();
|
|
PROTOBUF_NODISCARD std::string* release_xla_dump_hlo_pipeline_re();
|
|
void set_allocated_xla_dump_hlo_pipeline_re(std::string* xla_dump_hlo_pipeline_re);
|
|
private:
|
|
const std::string& _internal_xla_dump_hlo_pipeline_re() const;
|
|
inline PROTOBUF_ALWAYS_INLINE void _internal_set_xla_dump_hlo_pipeline_re(const std::string& value);
|
|
std::string* _internal_mutable_xla_dump_hlo_pipeline_re();
|
|
public:
|
|
|
|
// string xla_gpu_pgle_profile_file_or_directory_path = 210;
|
|
void clear_xla_gpu_pgle_profile_file_or_directory_path();
|
|
const std::string& xla_gpu_pgle_profile_file_or_directory_path() const;
|
|
template <typename ArgT0 = const std::string&, typename... ArgT>
|
|
void set_xla_gpu_pgle_profile_file_or_directory_path(ArgT0&& arg0, ArgT... args);
|
|
std::string* mutable_xla_gpu_pgle_profile_file_or_directory_path();
|
|
PROTOBUF_NODISCARD std::string* release_xla_gpu_pgle_profile_file_or_directory_path();
|
|
void set_allocated_xla_gpu_pgle_profile_file_or_directory_path(std::string* xla_gpu_pgle_profile_file_or_directory_path);
|
|
private:
|
|
const std::string& _internal_xla_gpu_pgle_profile_file_or_directory_path() const;
|
|
inline PROTOBUF_ALWAYS_INLINE void _internal_set_xla_gpu_pgle_profile_file_or_directory_path(const std::string& value);
|
|
std::string* _internal_mutable_xla_gpu_pgle_profile_file_or_directory_path();
|
|
public:
|
|
|
|
// string xla_gpu_dump_autotune_results_to = 222;
|
|
void clear_xla_gpu_dump_autotune_results_to();
|
|
const std::string& xla_gpu_dump_autotune_results_to() const;
|
|
template <typename ArgT0 = const std::string&, typename... ArgT>
|
|
void set_xla_gpu_dump_autotune_results_to(ArgT0&& arg0, ArgT... args);
|
|
std::string* mutable_xla_gpu_dump_autotune_results_to();
|
|
PROTOBUF_NODISCARD std::string* release_xla_gpu_dump_autotune_results_to();
|
|
void set_allocated_xla_gpu_dump_autotune_results_to(std::string* xla_gpu_dump_autotune_results_to);
|
|
private:
|
|
const std::string& _internal_xla_gpu_dump_autotune_results_to() const;
|
|
inline PROTOBUF_ALWAYS_INLINE void _internal_set_xla_gpu_dump_autotune_results_to(const std::string& value);
|
|
std::string* _internal_mutable_xla_gpu_dump_autotune_results_to();
|
|
public:
|
|
|
|
// string xla_gpu_load_autotune_results_from = 223;
|
|
void clear_xla_gpu_load_autotune_results_from();
|
|
const std::string& xla_gpu_load_autotune_results_from() const;
|
|
template <typename ArgT0 = const std::string&, typename... ArgT>
|
|
void set_xla_gpu_load_autotune_results_from(ArgT0&& arg0, ArgT... args);
|
|
std::string* mutable_xla_gpu_load_autotune_results_from();
|
|
PROTOBUF_NODISCARD std::string* release_xla_gpu_load_autotune_results_from();
|
|
void set_allocated_xla_gpu_load_autotune_results_from(std::string* xla_gpu_load_autotune_results_from);
|
|
private:
|
|
const std::string& _internal_xla_gpu_load_autotune_results_from() const;
|
|
inline PROTOBUF_ALWAYS_INLINE void _internal_set_xla_gpu_load_autotune_results_from(const std::string& value);
|
|
std::string* _internal_mutable_xla_gpu_load_autotune_results_from();
|
|
public:
|
|
|
|
// string xla_gpu_target_config_filename = 261;
|
|
void clear_xla_gpu_target_config_filename();
|
|
const std::string& xla_gpu_target_config_filename() const;
|
|
template <typename ArgT0 = const std::string&, typename... ArgT>
|
|
void set_xla_gpu_target_config_filename(ArgT0&& arg0, ArgT... args);
|
|
std::string* mutable_xla_gpu_target_config_filename();
|
|
PROTOBUF_NODISCARD std::string* release_xla_gpu_target_config_filename();
|
|
void set_allocated_xla_gpu_target_config_filename(std::string* xla_gpu_target_config_filename);
|
|
private:
|
|
const std::string& _internal_xla_gpu_target_config_filename() const;
|
|
inline PROTOBUF_ALWAYS_INLINE void _internal_set_xla_gpu_target_config_filename(const std::string& value);
|
|
std::string* _internal_mutable_xla_gpu_target_config_filename();
|
|
public:
|
|
|
|
// string xla_gpu_enable_custom_fusions_re = 264;
|
|
void clear_xla_gpu_enable_custom_fusions_re();
|
|
const std::string& xla_gpu_enable_custom_fusions_re() const;
|
|
template <typename ArgT0 = const std::string&, typename... ArgT>
|
|
void set_xla_gpu_enable_custom_fusions_re(ArgT0&& arg0, ArgT... args);
|
|
std::string* mutable_xla_gpu_enable_custom_fusions_re();
|
|
PROTOBUF_NODISCARD std::string* release_xla_gpu_enable_custom_fusions_re();
|
|
void set_allocated_xla_gpu_enable_custom_fusions_re(std::string* xla_gpu_enable_custom_fusions_re);
|
|
private:
|
|
const std::string& _internal_xla_gpu_enable_custom_fusions_re() const;
|
|
inline PROTOBUF_ALWAYS_INLINE void _internal_set_xla_gpu_enable_custom_fusions_re(const std::string& value);
|
|
std::string* _internal_mutable_xla_gpu_enable_custom_fusions_re();
|
|
public:
|
|
|
|
// int32 xla_backend_optimization_level = 31;
|
|
void clear_xla_backend_optimization_level();
|
|
int32_t xla_backend_optimization_level() const;
|
|
void set_xla_backend_optimization_level(int32_t value);
|
|
private:
|
|
int32_t _internal_xla_backend_optimization_level() const;
|
|
void _internal_set_xla_backend_optimization_level(int32_t value);
|
|
public:
|
|
|
|
// bool xla_hlo_graph_addresses = 2;
|
|
void clear_xla_hlo_graph_addresses();
|
|
bool xla_hlo_graph_addresses() const;
|
|
void set_xla_hlo_graph_addresses(bool value);
|
|
private:
|
|
bool _internal_xla_hlo_graph_addresses() const;
|
|
void _internal_set_xla_hlo_graph_addresses(bool value);
|
|
public:
|
|
|
|
// bool xla_hlo_profile = 9;
|
|
void clear_xla_hlo_profile();
|
|
bool xla_hlo_profile() const;
|
|
void set_xla_hlo_profile(bool value);
|
|
private:
|
|
bool _internal_xla_hlo_profile() const;
|
|
void _internal_set_xla_hlo_profile(bool value);
|
|
public:
|
|
|
|
// bool xla_disable_all_hlo_passes = 104;
|
|
void clear_xla_disable_all_hlo_passes();
|
|
bool xla_disable_all_hlo_passes() const;
|
|
void set_xla_disable_all_hlo_passes(bool value);
|
|
private:
|
|
bool _internal_xla_disable_all_hlo_passes() const;
|
|
void _internal_set_xla_disable_all_hlo_passes(bool value);
|
|
public:
|
|
|
|
// bool xla_embed_ir_in_executable = 33;
|
|
void clear_xla_embed_ir_in_executable();
|
|
bool xla_embed_ir_in_executable() const;
|
|
void set_xla_embed_ir_in_executable(bool value);
|
|
private:
|
|
bool _internal_xla_embed_ir_in_executable() const;
|
|
void _internal_set_xla_embed_ir_in_executable(bool value);
|
|
public:
|
|
|
|
// bool xla_eliminate_hlo_implicit_broadcast = 35;
|
|
void clear_xla_eliminate_hlo_implicit_broadcast();
|
|
bool xla_eliminate_hlo_implicit_broadcast() const;
|
|
void set_xla_eliminate_hlo_implicit_broadcast(bool value);
|
|
private:
|
|
bool _internal_xla_eliminate_hlo_implicit_broadcast() const;
|
|
void _internal_set_xla_eliminate_hlo_implicit_broadcast(bool value);
|
|
public:
|
|
|
|
// bool xla_cpu_multi_thread_eigen = 60;
|
|
void clear_xla_cpu_multi_thread_eigen();
|
|
bool xla_cpu_multi_thread_eigen() const;
|
|
void set_xla_cpu_multi_thread_eigen(bool value);
|
|
private:
|
|
bool _internal_xla_cpu_multi_thread_eigen() const;
|
|
void _internal_set_xla_cpu_multi_thread_eigen(bool value);
|
|
public:
|
|
|
|
// bool xla_gpu_ftz = 62;
|
|
void clear_xla_gpu_ftz();
|
|
bool xla_gpu_ftz() const;
|
|
void set_xla_gpu_ftz(bool value);
|
|
private:
|
|
bool _internal_xla_gpu_ftz() const;
|
|
void _internal_set_xla_gpu_ftz(bool value);
|
|
public:
|
|
|
|
// bool xla_llvm_enable_alias_scope_metadata = 70;
|
|
void clear_xla_llvm_enable_alias_scope_metadata();
|
|
bool xla_llvm_enable_alias_scope_metadata() const;
|
|
void set_xla_llvm_enable_alias_scope_metadata(bool value);
|
|
private:
|
|
bool _internal_xla_llvm_enable_alias_scope_metadata() const;
|
|
void _internal_set_xla_llvm_enable_alias_scope_metadata(bool value);
|
|
public:
|
|
|
|
// bool xla_llvm_enable_noalias_metadata = 71;
|
|
void clear_xla_llvm_enable_noalias_metadata();
|
|
bool xla_llvm_enable_noalias_metadata() const;
|
|
void set_xla_llvm_enable_noalias_metadata(bool value);
|
|
private:
|
|
bool _internal_xla_llvm_enable_noalias_metadata() const;
|
|
void _internal_set_xla_llvm_enable_noalias_metadata(bool value);
|
|
public:
|
|
|
|
// bool xla_llvm_enable_invariant_load_metadata = 72;
|
|
void clear_xla_llvm_enable_invariant_load_metadata();
|
|
bool xla_llvm_enable_invariant_load_metadata() const;
|
|
void set_xla_llvm_enable_invariant_load_metadata(bool value);
|
|
private:
|
|
bool _internal_xla_llvm_enable_invariant_load_metadata() const;
|
|
void _internal_set_xla_llvm_enable_invariant_load_metadata(bool value);
|
|
public:
|
|
|
|
// bool xla_llvm_disable_expensive_passes = 73;
|
|
void clear_xla_llvm_disable_expensive_passes();
|
|
bool xla_llvm_disable_expensive_passes() const;
|
|
void set_xla_llvm_disable_expensive_passes(bool value);
|
|
private:
|
|
bool _internal_xla_llvm_disable_expensive_passes() const;
|
|
void _internal_set_xla_llvm_disable_expensive_passes(bool value);
|
|
public:
|
|
|
|
// bool xla_test_all_output_layouts = 90;
|
|
void clear_xla_test_all_output_layouts();
|
|
bool xla_test_all_output_layouts() const;
|
|
void set_xla_test_all_output_layouts(bool value);
|
|
private:
|
|
bool _internal_xla_test_all_output_layouts() const;
|
|
void _internal_set_xla_test_all_output_layouts(bool value);
|
|
public:
|
|
|
|
// int32 xla_force_host_platform_device_count = 102;
|
|
void clear_xla_force_host_platform_device_count();
|
|
int32_t xla_force_host_platform_device_count() const;
|
|
void set_xla_force_host_platform_device_count(int32_t value);
|
|
private:
|
|
int32_t _internal_xla_force_host_platform_device_count() const;
|
|
void _internal_set_xla_force_host_platform_device_count(int32_t value);
|
|
public:
|
|
|
|
// bool xla_gpu_crash_on_verification_failures = 101;
|
|
void clear_xla_gpu_crash_on_verification_failures();
|
|
bool xla_gpu_crash_on_verification_failures() const;
|
|
void set_xla_gpu_crash_on_verification_failures(bool value);
|
|
private:
|
|
bool _internal_xla_gpu_crash_on_verification_failures() const;
|
|
void _internal_set_xla_gpu_crash_on_verification_failures(bool value);
|
|
public:
|
|
|
|
// bool xla_gpu_disable_gpuasm_optimizations = 103;
|
|
void clear_xla_gpu_disable_gpuasm_optimizations();
|
|
bool xla_gpu_disable_gpuasm_optimizations() const;
|
|
void set_xla_gpu_disable_gpuasm_optimizations(bool value);
|
|
private:
|
|
bool _internal_xla_gpu_disable_gpuasm_optimizations() const;
|
|
void _internal_set_xla_gpu_disable_gpuasm_optimizations(bool value);
|
|
public:
|
|
|
|
// bool xla_hlo_evaluator_use_fast_path = 106;
|
|
void clear_xla_hlo_evaluator_use_fast_path();
|
|
bool xla_hlo_evaluator_use_fast_path() const;
|
|
void set_xla_hlo_evaluator_use_fast_path(bool value);
|
|
private:
|
|
bool _internal_xla_hlo_evaluator_use_fast_path() const;
|
|
void _internal_set_xla_hlo_evaluator_use_fast_path(bool value);
|
|
public:
|
|
|
|
// bool xla_allow_scalar_index_dynamic_ops = 107;
|
|
void clear_xla_allow_scalar_index_dynamic_ops();
|
|
bool xla_allow_scalar_index_dynamic_ops() const;
|
|
void set_xla_allow_scalar_index_dynamic_ops(bool value);
|
|
private:
|
|
bool _internal_xla_allow_scalar_index_dynamic_ops() const;
|
|
void _internal_set_xla_allow_scalar_index_dynamic_ops(bool value);
|
|
public:
|
|
|
|
// .xla.DebugOptions.StepMarkerLocation xla_step_marker_location = 108;
|
|
void clear_xla_step_marker_location();
|
|
::xla::DebugOptions_StepMarkerLocation xla_step_marker_location() const;
|
|
void set_xla_step_marker_location(::xla::DebugOptions_StepMarkerLocation value);
|
|
private:
|
|
::xla::DebugOptions_StepMarkerLocation _internal_xla_step_marker_location() const;
|
|
void _internal_set_xla_step_marker_location(::xla::DebugOptions_StepMarkerLocation value);
|
|
public:
|
|
|
|
// bool xla_dump_hlo_as_text = 112;
|
|
void clear_xla_dump_hlo_as_text();
|
|
bool xla_dump_hlo_as_text() const;
|
|
void set_xla_dump_hlo_as_text(bool value);
|
|
private:
|
|
bool _internal_xla_dump_hlo_as_text() const;
|
|
void _internal_set_xla_dump_hlo_as_text(bool value);
|
|
public:
|
|
|
|
// bool xla_dump_hlo_as_proto = 113;
|
|
void clear_xla_dump_hlo_as_proto();
|
|
bool xla_dump_hlo_as_proto() const;
|
|
void set_xla_dump_hlo_as_proto(bool value);
|
|
private:
|
|
bool _internal_xla_dump_hlo_as_proto() const;
|
|
void _internal_set_xla_dump_hlo_as_proto(bool value);
|
|
public:
|
|
|
|
// bool xla_dump_hlo_as_dot = 114;
|
|
void clear_xla_dump_hlo_as_dot();
|
|
bool xla_dump_hlo_as_dot() const;
|
|
void set_xla_dump_hlo_as_dot(bool value);
|
|
private:
|
|
bool _internal_xla_dump_hlo_as_dot() const;
|
|
void _internal_set_xla_dump_hlo_as_dot(bool value);
|
|
public:
|
|
|
|
// bool xla_dump_hlo_as_url = 115;
|
|
void clear_xla_dump_hlo_as_url();
|
|
bool xla_dump_hlo_as_url() const;
|
|
void set_xla_dump_hlo_as_url(bool value);
|
|
private:
|
|
bool _internal_xla_dump_hlo_as_url() const;
|
|
void _internal_set_xla_dump_hlo_as_url(bool value);
|
|
public:
|
|
|
|
// bool xla_test_all_input_layouts = 91;
|
|
void clear_xla_test_all_input_layouts();
|
|
bool xla_test_all_input_layouts() const;
|
|
void set_xla_test_all_input_layouts(bool value);
|
|
private:
|
|
bool _internal_xla_test_all_input_layouts() const;
|
|
void _internal_set_xla_test_all_input_layouts(bool value);
|
|
public:
|
|
|
|
// bool xla_hlo_graph_sharding_color = 92;
|
|
void clear_xla_hlo_graph_sharding_color();
|
|
bool xla_hlo_graph_sharding_color() const;
|
|
void set_xla_hlo_graph_sharding_color(bool value);
|
|
private:
|
|
bool _internal_xla_hlo_graph_sharding_color() const;
|
|
void _internal_set_xla_hlo_graph_sharding_color(bool value);
|
|
public:
|
|
|
|
// bool xla_cpu_use_mkl_dnn = 97;
|
|
void clear_xla_cpu_use_mkl_dnn();
|
|
bool xla_cpu_use_mkl_dnn() const;
|
|
void set_xla_cpu_use_mkl_dnn(bool value);
|
|
private:
|
|
bool _internal_xla_cpu_use_mkl_dnn() const;
|
|
void _internal_set_xla_cpu_use_mkl_dnn(bool value);
|
|
public:
|
|
|
|
// bool xla_cpu_use_xla_runtime = 177;
|
|
void clear_xla_cpu_use_xla_runtime();
|
|
bool xla_cpu_use_xla_runtime() const;
|
|
void set_xla_cpu_use_xla_runtime(bool value);
|
|
private:
|
|
bool _internal_xla_cpu_use_xla_runtime() const;
|
|
void _internal_set_xla_cpu_use_xla_runtime(bool value);
|
|
public:
|
|
|
|
// bool xla_cpu_enable_fast_math = 99;
|
|
void clear_xla_cpu_enable_fast_math();
|
|
bool xla_cpu_enable_fast_math() const;
|
|
void set_xla_cpu_enable_fast_math(bool value);
|
|
private:
|
|
bool _internal_xla_cpu_enable_fast_math() const;
|
|
void _internal_set_xla_cpu_enable_fast_math(bool value);
|
|
public:
|
|
|
|
// bool xla_cpu_fast_math_honor_nans = 120;
|
|
void clear_xla_cpu_fast_math_honor_nans();
|
|
bool xla_cpu_fast_math_honor_nans() const;
|
|
void set_xla_cpu_fast_math_honor_nans(bool value);
|
|
private:
|
|
bool _internal_xla_cpu_fast_math_honor_nans() const;
|
|
void _internal_set_xla_cpu_fast_math_honor_nans(bool value);
|
|
public:
|
|
|
|
// bool xla_cpu_fast_math_honor_infs = 121;
|
|
void clear_xla_cpu_fast_math_honor_infs();
|
|
bool xla_cpu_fast_math_honor_infs() const;
|
|
void set_xla_cpu_fast_math_honor_infs(bool value);
|
|
private:
|
|
bool _internal_xla_cpu_fast_math_honor_infs() const;
|
|
void _internal_set_xla_cpu_fast_math_honor_infs(bool value);
|
|
public:
|
|
|
|
// bool xla_cpu_fast_math_honor_division = 126;
|
|
void clear_xla_cpu_fast_math_honor_division();
|
|
bool xla_cpu_fast_math_honor_division() const;
|
|
void set_xla_cpu_fast_math_honor_division(bool value);
|
|
private:
|
|
bool _internal_xla_cpu_fast_math_honor_division() const;
|
|
void _internal_set_xla_cpu_fast_math_honor_division(bool value);
|
|
public:
|
|
|
|
// bool xla_cpu_fast_math_honor_functions = 129;
|
|
void clear_xla_cpu_fast_math_honor_functions();
|
|
bool xla_cpu_fast_math_honor_functions() const;
|
|
void set_xla_cpu_fast_math_honor_functions(bool value);
|
|
private:
|
|
bool _internal_xla_cpu_fast_math_honor_functions() const;
|
|
void _internal_set_xla_cpu_fast_math_honor_functions(bool value);
|
|
public:
|
|
|
|
// bool xla_cpu_enable_fast_min_max = 140;
|
|
void clear_xla_cpu_enable_fast_min_max();
|
|
bool xla_cpu_enable_fast_min_max() const;
|
|
void set_xla_cpu_enable_fast_min_max(bool value);
|
|
private:
|
|
bool _internal_xla_cpu_enable_fast_min_max() const;
|
|
void _internal_set_xla_cpu_enable_fast_min_max(bool value);
|
|
public:
|
|
|
|
// bool xla_gpu_enable_fast_min_max = 100;
|
|
void clear_xla_gpu_enable_fast_min_max();
|
|
bool xla_gpu_enable_fast_min_max() const;
|
|
void set_xla_gpu_enable_fast_min_max(bool value);
|
|
private:
|
|
bool _internal_xla_gpu_enable_fast_min_max() const;
|
|
void _internal_set_xla_gpu_enable_fast_min_max(bool value);
|
|
public:
|
|
|
|
// bool xla_allow_excess_precision = 122;
|
|
void clear_xla_allow_excess_precision();
|
|
bool xla_allow_excess_precision() const;
|
|
void set_xla_allow_excess_precision(bool value);
|
|
private:
|
|
bool _internal_xla_allow_excess_precision() const;
|
|
void _internal_set_xla_allow_excess_precision(bool value);
|
|
public:
|
|
|
|
// int32 xla_gpu_autotune_level = 123;
|
|
void clear_xla_gpu_autotune_level();
|
|
int32_t xla_gpu_autotune_level() const;
|
|
void set_xla_gpu_autotune_level(int32_t value);
|
|
private:
|
|
int32_t _internal_xla_gpu_autotune_level() const;
|
|
void _internal_set_xla_gpu_autotune_level(int32_t value);
|
|
public:
|
|
|
|
// bool xla_dump_hlo_as_html = 116;
|
|
void clear_xla_dump_hlo_as_html();
|
|
bool xla_dump_hlo_as_html() const;
|
|
void set_xla_dump_hlo_as_html(bool value);
|
|
private:
|
|
bool _internal_xla_dump_hlo_as_html() const;
|
|
void _internal_set_xla_dump_hlo_as_html(bool value);
|
|
public:
|
|
|
|
// bool xla_dump_fusion_visualization = 149;
|
|
void clear_xla_dump_fusion_visualization();
|
|
bool xla_dump_fusion_visualization() const;
|
|
void set_xla_dump_fusion_visualization(bool value);
|
|
private:
|
|
bool _internal_xla_dump_fusion_visualization() const;
|
|
void _internal_set_xla_dump_fusion_visualization(bool value);
|
|
public:
|
|
|
|
// bool xla_dump_hlo_snapshots = 118;
|
|
void clear_xla_dump_hlo_snapshots();
|
|
bool xla_dump_hlo_snapshots() const;
|
|
void set_xla_dump_hlo_snapshots(bool value);
|
|
private:
|
|
bool _internal_xla_dump_hlo_snapshots() const;
|
|
void _internal_set_xla_dump_hlo_snapshots(bool value);
|
|
public:
|
|
|
|
// bool xla_dump_include_timestamp = 131;
|
|
void clear_xla_dump_include_timestamp();
|
|
bool xla_dump_include_timestamp() const;
|
|
void set_xla_dump_include_timestamp(bool value);
|
|
private:
|
|
bool _internal_xla_dump_include_timestamp() const;
|
|
void _internal_set_xla_dump_include_timestamp(bool value);
|
|
public:
|
|
|
|
// int32 xla_dump_max_hlo_modules = 132;
|
|
void clear_xla_dump_max_hlo_modules();
|
|
int32_t xla_dump_max_hlo_modules() const;
|
|
void set_xla_dump_max_hlo_modules(int32_t value);
|
|
private:
|
|
int32_t _internal_xla_dump_max_hlo_modules() const;
|
|
void _internal_set_xla_dump_max_hlo_modules(int32_t value);
|
|
public:
|
|
|
|
// int32 xla_multiheap_size_constraint_per_heap = 142;
|
|
void clear_xla_multiheap_size_constraint_per_heap();
|
|
int32_t xla_multiheap_size_constraint_per_heap() const;
|
|
void set_xla_multiheap_size_constraint_per_heap(int32_t value);
|
|
private:
|
|
int32_t _internal_xla_multiheap_size_constraint_per_heap() const;
|
|
void _internal_set_xla_multiheap_size_constraint_per_heap(int32_t value);
|
|
public:
|
|
|
|
// bool xla_dump_module_metadata = 144;
|
|
void clear_xla_dump_module_metadata();
|
|
bool xla_dump_module_metadata() const;
|
|
void set_xla_dump_module_metadata(bool value);
|
|
private:
|
|
bool _internal_xla_dump_module_metadata() const;
|
|
void _internal_set_xla_dump_module_metadata(bool value);
|
|
public:
|
|
|
|
// bool xla_dump_compress_protos = 151;
|
|
void clear_xla_dump_compress_protos();
|
|
bool xla_dump_compress_protos() const;
|
|
void set_xla_dump_compress_protos(bool value);
|
|
private:
|
|
bool _internal_xla_dump_compress_protos() const;
|
|
void _internal_set_xla_dump_compress_protos(bool value);
|
|
public:
|
|
|
|
// bool xla_dump_hlo_as_long_text = 164;
|
|
void clear_xla_dump_hlo_as_long_text();
|
|
bool xla_dump_hlo_as_long_text() const;
|
|
void set_xla_dump_hlo_as_long_text(bool value);
|
|
private:
|
|
bool _internal_xla_dump_hlo_as_long_text() const;
|
|
void _internal_set_xla_dump_hlo_as_long_text(bool value);
|
|
public:
|
|
|
|
// bool xla_gpu_force_conv_nchw = 125;
|
|
void clear_xla_gpu_force_conv_nchw();
|
|
bool xla_gpu_force_conv_nchw() const;
|
|
void set_xla_gpu_force_conv_nchw(bool value);
|
|
private:
|
|
bool _internal_xla_gpu_force_conv_nchw() const;
|
|
void _internal_set_xla_gpu_force_conv_nchw(bool value);
|
|
public:
|
|
|
|
// int32 xla_gpu_force_compilation_parallelism = 147;
|
|
void clear_xla_gpu_force_compilation_parallelism();
|
|
int32_t xla_gpu_force_compilation_parallelism() const;
|
|
void set_xla_gpu_force_compilation_parallelism(int32_t value);
|
|
private:
|
|
int32_t _internal_xla_gpu_force_compilation_parallelism() const;
|
|
void _internal_set_xla_gpu_force_compilation_parallelism(int32_t value);
|
|
public:
|
|
|
|
// bool xla_gpu_force_conv_nhwc = 146;
|
|
void clear_xla_gpu_force_conv_nhwc();
|
|
bool xla_gpu_force_conv_nhwc() const;
|
|
void set_xla_gpu_force_conv_nhwc(bool value);
|
|
private:
|
|
bool _internal_xla_gpu_force_conv_nhwc() const;
|
|
void _internal_set_xla_gpu_force_conv_nhwc(bool value);
|
|
public:
|
|
|
|
// bool xla_gpu_dump_llvmir = 155;
|
|
void clear_xla_gpu_dump_llvmir();
|
|
bool xla_gpu_dump_llvmir() const;
|
|
void set_xla_gpu_dump_llvmir(bool value);
|
|
private:
|
|
bool _internal_xla_gpu_dump_llvmir() const;
|
|
void _internal_set_xla_gpu_dump_llvmir(bool value);
|
|
public:
|
|
|
|
// bool xla_dump_enable_mlir_pretty_form = 185;
|
|
void clear_xla_dump_enable_mlir_pretty_form();
|
|
bool xla_dump_enable_mlir_pretty_form() const;
|
|
void set_xla_dump_enable_mlir_pretty_form(bool value);
|
|
private:
|
|
bool _internal_xla_dump_enable_mlir_pretty_form() const;
|
|
void _internal_set_xla_dump_enable_mlir_pretty_form(bool value);
|
|
public:
|
|
|
|
// bool xla_tpu_detect_nan = 135;
|
|
void clear_xla_tpu_detect_nan();
|
|
bool xla_tpu_detect_nan() const;
|
|
void set_xla_tpu_detect_nan(bool value);
|
|
private:
|
|
bool _internal_xla_tpu_detect_nan() const;
|
|
void _internal_set_xla_tpu_detect_nan(bool value);
|
|
public:
|
|
|
|
// int64 xla_gpu_all_reduce_combine_threshold_bytes = 157;
|
|
void clear_xla_gpu_all_reduce_combine_threshold_bytes();
|
|
int64_t xla_gpu_all_reduce_combine_threshold_bytes() const;
|
|
void set_xla_gpu_all_reduce_combine_threshold_bytes(int64_t value);
|
|
private:
|
|
int64_t _internal_xla_gpu_all_reduce_combine_threshold_bytes() const;
|
|
void _internal_set_xla_gpu_all_reduce_combine_threshold_bytes(int64_t value);
|
|
public:
|
|
|
|
// int64 xla_gpu_nccl_termination_timeout_seconds = 163;
|
|
void clear_xla_gpu_nccl_termination_timeout_seconds();
|
|
int64_t xla_gpu_nccl_termination_timeout_seconds() const;
|
|
void set_xla_gpu_nccl_termination_timeout_seconds(int64_t value);
|
|
private:
|
|
int64_t _internal_xla_gpu_nccl_termination_timeout_seconds() const;
|
|
void _internal_set_xla_gpu_nccl_termination_timeout_seconds(int64_t value);
|
|
public:
|
|
|
|
// int32 xla_gpu_all_reduce_blueconnect_num_devices_per_host = 159;
|
|
void clear_xla_gpu_all_reduce_blueconnect_num_devices_per_host();
|
|
int32_t xla_gpu_all_reduce_blueconnect_num_devices_per_host() const;
|
|
void set_xla_gpu_all_reduce_blueconnect_num_devices_per_host(int32_t value);
|
|
private:
|
|
int32_t _internal_xla_gpu_all_reduce_blueconnect_num_devices_per_host() const;
|
|
void _internal_set_xla_gpu_all_reduce_blueconnect_num_devices_per_host(int32_t value);
|
|
public:
|
|
|
|
// bool xla_tpu_detect_inf = 136;
|
|
void clear_xla_tpu_detect_inf();
|
|
bool xla_tpu_detect_inf() const;
|
|
void set_xla_tpu_detect_inf(bool value);
|
|
private:
|
|
bool _internal_xla_tpu_detect_inf() const;
|
|
void _internal_set_xla_tpu_detect_inf(bool value);
|
|
public:
|
|
|
|
// bool xla_cpu_enable_xprof_traceme = 137;
|
|
void clear_xla_cpu_enable_xprof_traceme();
|
|
bool xla_cpu_enable_xprof_traceme() const;
|
|
void set_xla_cpu_enable_xprof_traceme(bool value);
|
|
private:
|
|
bool _internal_xla_cpu_enable_xprof_traceme() const;
|
|
void _internal_set_xla_cpu_enable_xprof_traceme(bool value);
|
|
public:
|
|
|
|
// bool xla_gpu_unsafe_fallback_to_driver_on_ptxas_not_found = 138;
|
|
void clear_xla_gpu_unsafe_fallback_to_driver_on_ptxas_not_found();
|
|
bool xla_gpu_unsafe_fallback_to_driver_on_ptxas_not_found() const;
|
|
void set_xla_gpu_unsafe_fallback_to_driver_on_ptxas_not_found(bool value);
|
|
private:
|
|
bool _internal_xla_gpu_unsafe_fallback_to_driver_on_ptxas_not_found() const;
|
|
void _internal_set_xla_gpu_unsafe_fallback_to_driver_on_ptxas_not_found(bool value);
|
|
public:
|
|
|
|
// bool xla_detailed_logging = 252;
|
|
void clear_xla_detailed_logging();
|
|
bool xla_detailed_logging() const;
|
|
void set_xla_detailed_logging(bool value);
|
|
private:
|
|
bool _internal_xla_detailed_logging() const;
|
|
void _internal_set_xla_detailed_logging(bool value);
|
|
public:
|
|
|
|
// int64 xla_gpu_redzone_scratch_max_megabytes = 167;
|
|
void clear_xla_gpu_redzone_scratch_max_megabytes();
|
|
int64_t xla_gpu_redzone_scratch_max_megabytes() const;
|
|
void set_xla_gpu_redzone_scratch_max_megabytes(int64_t value);
|
|
private:
|
|
int64_t _internal_xla_gpu_redzone_scratch_max_megabytes() const;
|
|
void _internal_set_xla_gpu_redzone_scratch_max_megabytes(int64_t value);
|
|
public:
|
|
|
|
// .xla.DebugOptions.ShapeChecks xla_gpu_shape_checks = 170;
|
|
void clear_xla_gpu_shape_checks();
|
|
::xla::DebugOptions_ShapeChecks xla_gpu_shape_checks() const;
|
|
void set_xla_gpu_shape_checks(::xla::DebugOptions_ShapeChecks value);
|
|
private:
|
|
::xla::DebugOptions_ShapeChecks _internal_xla_gpu_shape_checks() const;
|
|
void _internal_set_xla_gpu_shape_checks(::xla::DebugOptions_ShapeChecks value);
|
|
public:
|
|
|
|
// bool xla_gpu_enable_xla_runtime_executable = 169;
|
|
void clear_xla_gpu_enable_xla_runtime_executable();
|
|
bool xla_gpu_enable_xla_runtime_executable() const;
|
|
void set_xla_gpu_enable_xla_runtime_executable(bool value);
|
|
private:
|
|
bool _internal_xla_gpu_enable_xla_runtime_executable() const;
|
|
void _internal_set_xla_gpu_enable_xla_runtime_executable(bool value);
|
|
public:
|
|
|
|
// bool xla_gpu_enable_custom_fusions = 263;
|
|
void clear_xla_gpu_enable_custom_fusions();
|
|
bool xla_gpu_enable_custom_fusions() const;
|
|
void set_xla_gpu_enable_custom_fusions(bool value);
|
|
private:
|
|
bool _internal_xla_gpu_enable_custom_fusions() const;
|
|
void _internal_set_xla_gpu_enable_custom_fusions(bool value);
|
|
public:
|
|
|
|
// bool xla_gpu_enable_address_computation_fusion = 105;
|
|
void clear_xla_gpu_enable_address_computation_fusion();
|
|
bool xla_gpu_enable_address_computation_fusion() const;
|
|
void set_xla_gpu_enable_address_computation_fusion(bool value);
|
|
private:
|
|
bool _internal_xla_gpu_enable_address_computation_fusion() const;
|
|
void _internal_set_xla_gpu_enable_address_computation_fusion(bool value);
|
|
public:
|
|
|
|
// bool xla_gpu_enable_shared_constants = 165;
|
|
void clear_xla_gpu_enable_shared_constants();
|
|
bool xla_gpu_enable_shared_constants() const;
|
|
void set_xla_gpu_enable_shared_constants(bool value);
|
|
private:
|
|
bool _internal_xla_gpu_enable_shared_constants() const;
|
|
void _internal_set_xla_gpu_enable_shared_constants(bool value);
|
|
public:
|
|
|
|
// bool xla_cpu_use_acl = 174;
|
|
void clear_xla_cpu_use_acl();
|
|
bool xla_cpu_use_acl() const;
|
|
void set_xla_cpu_use_acl(bool value);
|
|
private:
|
|
bool _internal_xla_cpu_use_acl() const;
|
|
void _internal_set_xla_cpu_use_acl(bool value);
|
|
public:
|
|
|
|
// bool xla_cpu_strict_dot_conv_math = 175;
|
|
void clear_xla_cpu_strict_dot_conv_math();
|
|
bool xla_cpu_strict_dot_conv_math() const;
|
|
void set_xla_cpu_strict_dot_conv_math(bool value);
|
|
private:
|
|
bool _internal_xla_cpu_strict_dot_conv_math() const;
|
|
void _internal_set_xla_cpu_strict_dot_conv_math(bool value);
|
|
public:
|
|
|
|
// bool xla_gpu_use_runtime_fusion = 181;
|
|
void clear_xla_gpu_use_runtime_fusion();
|
|
bool xla_gpu_use_runtime_fusion() const;
|
|
void set_xla_gpu_use_runtime_fusion(bool value);
|
|
private:
|
|
bool _internal_xla_gpu_use_runtime_fusion() const;
|
|
void _internal_set_xla_gpu_use_runtime_fusion(bool value);
|
|
public:
|
|
|
|
// bool xla_dump_latency_hiding_schedule = 182;
|
|
void clear_xla_dump_latency_hiding_schedule();
|
|
bool xla_dump_latency_hiding_schedule() const;
|
|
void set_xla_dump_latency_hiding_schedule(bool value);
|
|
private:
|
|
bool _internal_xla_dump_latency_hiding_schedule() const;
|
|
void _internal_set_xla_dump_latency_hiding_schedule(bool value);
|
|
public:
|
|
|
|
// bool xla_gpu_enable_cublaslt = 166;
|
|
void clear_xla_gpu_enable_cublaslt();
|
|
bool xla_gpu_enable_cublaslt() const;
|
|
void set_xla_gpu_enable_cublaslt(bool value);
|
|
private:
|
|
bool _internal_xla_gpu_enable_cublaslt() const;
|
|
void _internal_set_xla_gpu_enable_cublaslt(bool value);
|
|
public:
|
|
|
|
// bool xla_gpu_graph_enable_concurrent_region = 215;
|
|
void clear_xla_gpu_graph_enable_concurrent_region();
|
|
bool xla_gpu_graph_enable_concurrent_region() const;
|
|
void set_xla_gpu_graph_enable_concurrent_region(bool value);
|
|
private:
|
|
bool _internal_xla_gpu_graph_enable_concurrent_region() const;
|
|
void _internal_set_xla_gpu_graph_enable_concurrent_region(bool value);
|
|
public:
|
|
|
|
// bool xla_gpu_simplify_all_fp_conversions = 168;
|
|
void clear_xla_gpu_simplify_all_fp_conversions();
|
|
bool xla_gpu_simplify_all_fp_conversions() const;
|
|
void set_xla_gpu_simplify_all_fp_conversions(bool value);
|
|
private:
|
|
bool _internal_xla_gpu_simplify_all_fp_conversions() const;
|
|
void _internal_set_xla_gpu_simplify_all_fp_conversions(bool value);
|
|
public:
|
|
|
|
// bool xla_gpu_normalize_layouts = 172;
|
|
void clear_xla_gpu_normalize_layouts();
|
|
bool xla_gpu_normalize_layouts() const;
|
|
void set_xla_gpu_normalize_layouts(bool value);
|
|
private:
|
|
bool _internal_xla_gpu_normalize_layouts() const;
|
|
void _internal_set_xla_gpu_normalize_layouts(bool value);
|
|
public:
|
|
|
|
// bool xla_gpu_enable_async_all_reduce = 152;
|
|
void clear_xla_gpu_enable_async_all_reduce();
|
|
bool xla_gpu_enable_async_all_reduce() const;
|
|
void set_xla_gpu_enable_async_all_reduce(bool value);
|
|
private:
|
|
bool _internal_xla_gpu_enable_async_all_reduce() const;
|
|
void _internal_set_xla_gpu_enable_async_all_reduce(bool value);
|
|
public:
|
|
|
|
// bool xla_gpu_enable_async_collective_permute = 183;
|
|
void clear_xla_gpu_enable_async_collective_permute();
|
|
bool xla_gpu_enable_async_collective_permute() const;
|
|
void set_xla_gpu_enable_async_collective_permute(bool value);
|
|
private:
|
|
bool _internal_xla_gpu_enable_async_collective_permute() const;
|
|
void _internal_set_xla_gpu_enable_async_collective_permute(bool value);
|
|
public:
|
|
|
|
// bool xla_gpu_enable_async_all_gather = 199;
|
|
void clear_xla_gpu_enable_async_all_gather();
|
|
bool xla_gpu_enable_async_all_gather() const;
|
|
void set_xla_gpu_enable_async_all_gather(bool value);
|
|
private:
|
|
bool _internal_xla_gpu_enable_async_all_gather() const;
|
|
void _internal_set_xla_gpu_enable_async_all_gather(bool value);
|
|
public:
|
|
|
|
// bool xla_gpu_enable_async_reduce_scatter = 200;
|
|
void clear_xla_gpu_enable_async_reduce_scatter();
|
|
bool xla_gpu_enable_async_reduce_scatter() const;
|
|
void set_xla_gpu_enable_async_reduce_scatter(bool value);
|
|
private:
|
|
bool _internal_xla_gpu_enable_async_reduce_scatter() const;
|
|
void _internal_set_xla_gpu_enable_async_reduce_scatter(bool value);
|
|
public:
|
|
|
|
// .xla.DebugOptions.PartitioningAlgorithm xla_partitioning_algorithm = 187;
|
|
void clear_xla_partitioning_algorithm();
|
|
::xla::DebugOptions_PartitioningAlgorithm xla_partitioning_algorithm() const;
|
|
void set_xla_partitioning_algorithm(::xla::DebugOptions_PartitioningAlgorithm value);
|
|
private:
|
|
::xla::DebugOptions_PartitioningAlgorithm _internal_xla_partitioning_algorithm() const;
|
|
void _internal_set_xla_partitioning_algorithm(::xla::DebugOptions_PartitioningAlgorithm value);
|
|
public:
|
|
|
|
// bool xla_cpu_enable_mlir_tiling_and_fusion = 184;
|
|
void clear_xla_cpu_enable_mlir_tiling_and_fusion();
|
|
bool xla_cpu_enable_mlir_tiling_and_fusion() const;
|
|
void set_xla_cpu_enable_mlir_tiling_and_fusion(bool value);
|
|
private:
|
|
bool _internal_xla_cpu_enable_mlir_tiling_and_fusion() const;
|
|
void _internal_set_xla_cpu_enable_mlir_tiling_and_fusion(bool value);
|
|
public:
|
|
|
|
// bool xla_cpu_enable_custom_matmul_tiling = 195;
|
|
void clear_xla_cpu_enable_custom_matmul_tiling();
|
|
bool xla_cpu_enable_custom_matmul_tiling() const;
|
|
void set_xla_cpu_enable_custom_matmul_tiling(bool value);
|
|
private:
|
|
bool _internal_xla_cpu_enable_custom_matmul_tiling() const;
|
|
void _internal_set_xla_cpu_enable_custom_matmul_tiling(bool value);
|
|
public:
|
|
|
|
// bool xla_cpu_enable_mlir_fusion_outlining = 192;
|
|
void clear_xla_cpu_enable_mlir_fusion_outlining();
|
|
bool xla_cpu_enable_mlir_fusion_outlining() const;
|
|
void set_xla_cpu_enable_mlir_fusion_outlining(bool value);
|
|
private:
|
|
bool _internal_xla_cpu_enable_mlir_fusion_outlining() const;
|
|
void _internal_set_xla_cpu_enable_mlir_fusion_outlining(bool value);
|
|
public:
|
|
|
|
// bool xla_cpu_enable_experimental_deallocation = 191;
|
|
void clear_xla_cpu_enable_experimental_deallocation();
|
|
bool xla_cpu_enable_experimental_deallocation() const;
|
|
void set_xla_cpu_enable_experimental_deallocation(bool value);
|
|
private:
|
|
bool _internal_xla_cpu_enable_experimental_deallocation() const;
|
|
void _internal_set_xla_cpu_enable_experimental_deallocation(bool value);
|
|
public:
|
|
|
|
// bool xla_gpu_enable_reassociation_for_converted_ar = 209;
|
|
void clear_xla_gpu_enable_reassociation_for_converted_ar();
|
|
bool xla_gpu_enable_reassociation_for_converted_ar() const;
|
|
void set_xla_gpu_enable_reassociation_for_converted_ar(bool value);
|
|
private:
|
|
bool _internal_xla_gpu_enable_reassociation_for_converted_ar() const;
|
|
void _internal_set_xla_gpu_enable_reassociation_for_converted_ar(bool value);
|
|
public:
|
|
|
|
// bool xla_gpu_enable_while_loop_reduce_scatter_code_motion = 203;
|
|
void clear_xla_gpu_enable_while_loop_reduce_scatter_code_motion();
|
|
bool xla_gpu_enable_while_loop_reduce_scatter_code_motion() const;
|
|
void set_xla_gpu_enable_while_loop_reduce_scatter_code_motion(bool value);
|
|
private:
|
|
bool _internal_xla_gpu_enable_while_loop_reduce_scatter_code_motion() const;
|
|
void _internal_set_xla_gpu_enable_while_loop_reduce_scatter_code_motion(bool value);
|
|
public:
|
|
|
|
// bool xla_gpu_enable_cudnn_frontend = 160;
|
|
void clear_xla_gpu_enable_cudnn_frontend();
|
|
bool xla_gpu_enable_cudnn_frontend() const;
|
|
void set_xla_gpu_enable_cudnn_frontend(bool value);
|
|
private:
|
|
bool _internal_xla_gpu_enable_cudnn_frontend() const;
|
|
void _internal_set_xla_gpu_enable_cudnn_frontend(bool value);
|
|
public:
|
|
|
|
// bool xla_gpu_enable_cudnn_fmha = 218;
|
|
void clear_xla_gpu_enable_cudnn_fmha();
|
|
bool xla_gpu_enable_cudnn_fmha() const;
|
|
void set_xla_gpu_enable_cudnn_fmha(bool value);
|
|
private:
|
|
bool _internal_xla_gpu_enable_cudnn_fmha() const;
|
|
void _internal_set_xla_gpu_enable_cudnn_fmha(bool value);
|
|
public:
|
|
|
|
// int64 xla_cpu_matmul_tiling_m_dim = 196;
|
|
void clear_xla_cpu_matmul_tiling_m_dim();
|
|
int64_t xla_cpu_matmul_tiling_m_dim() const;
|
|
void set_xla_cpu_matmul_tiling_m_dim(int64_t value);
|
|
private:
|
|
int64_t _internal_xla_cpu_matmul_tiling_m_dim() const;
|
|
void _internal_set_xla_cpu_matmul_tiling_m_dim(int64_t value);
|
|
public:
|
|
|
|
// int64 xla_cpu_matmul_tiling_n_dim = 197;
|
|
void clear_xla_cpu_matmul_tiling_n_dim();
|
|
int64_t xla_cpu_matmul_tiling_n_dim() const;
|
|
void set_xla_cpu_matmul_tiling_n_dim(int64_t value);
|
|
private:
|
|
int64_t _internal_xla_cpu_matmul_tiling_n_dim() const;
|
|
void _internal_set_xla_cpu_matmul_tiling_n_dim(int64_t value);
|
|
public:
|
|
|
|
// int64 xla_cpu_matmul_tiling_k_dim = 198;
|
|
void clear_xla_cpu_matmul_tiling_k_dim();
|
|
int64_t xla_cpu_matmul_tiling_k_dim() const;
|
|
void set_xla_cpu_matmul_tiling_k_dim(int64_t value);
|
|
private:
|
|
int64_t _internal_xla_cpu_matmul_tiling_k_dim() const;
|
|
void _internal_set_xla_cpu_matmul_tiling_k_dim(int64_t value);
|
|
public:
|
|
|
|
// bool xla_gpu_fused_attention_use_cudnn_rng = 235;
|
|
void clear_xla_gpu_fused_attention_use_cudnn_rng();
|
|
bool xla_gpu_fused_attention_use_cudnn_rng() const;
|
|
void set_xla_gpu_fused_attention_use_cudnn_rng(bool value);
|
|
private:
|
|
bool _internal_xla_gpu_fused_attention_use_cudnn_rng() const;
|
|
void _internal_set_xla_gpu_fused_attention_use_cudnn_rng(bool value);
|
|
public:
|
|
|
|
// bool xla_gpu_enable_cudnn_layer_norm = 262;
|
|
void clear_xla_gpu_enable_cudnn_layer_norm();
|
|
bool xla_gpu_enable_cudnn_layer_norm() const;
|
|
void set_xla_gpu_enable_cudnn_layer_norm(bool value);
|
|
private:
|
|
bool _internal_xla_gpu_enable_cudnn_layer_norm() const;
|
|
void _internal_set_xla_gpu_enable_cudnn_layer_norm(bool value);
|
|
public:
|
|
|
|
// bool xla_dump_disable_metadata = 153;
|
|
void clear_xla_dump_disable_metadata();
|
|
bool xla_dump_disable_metadata() const;
|
|
void set_xla_dump_disable_metadata(bool value);
|
|
private:
|
|
bool _internal_xla_dump_disable_metadata() const;
|
|
void _internal_set_xla_dump_disable_metadata(bool value);
|
|
public:
|
|
|
|
// bool xla_gpu_strict_conv_algorithm_picker = 156;
|
|
void clear_xla_gpu_strict_conv_algorithm_picker();
|
|
bool xla_gpu_strict_conv_algorithm_picker() const;
|
|
void set_xla_gpu_strict_conv_algorithm_picker(bool value);
|
|
private:
|
|
bool _internal_xla_gpu_strict_conv_algorithm_picker() const;
|
|
void _internal_set_xla_gpu_strict_conv_algorithm_picker(bool value);
|
|
public:
|
|
|
|
// int32 xla_gpu_graph_num_runs_to_instantiate = 202;
|
|
void clear_xla_gpu_graph_num_runs_to_instantiate();
|
|
int32_t xla_gpu_graph_num_runs_to_instantiate() const;
|
|
void set_xla_gpu_graph_num_runs_to_instantiate(int32_t value);
|
|
private:
|
|
int32_t _internal_xla_gpu_graph_num_runs_to_instantiate() const;
|
|
void _internal_set_xla_gpu_graph_num_runs_to_instantiate(int32_t value);
|
|
public:
|
|
|
|
// bool xla_gpu_enable_pipelined_p2p = 246;
|
|
void clear_xla_gpu_enable_pipelined_p2p();
|
|
bool xla_gpu_enable_pipelined_p2p() const;
|
|
void set_xla_gpu_enable_pipelined_p2p(bool value);
|
|
private:
|
|
bool _internal_xla_gpu_enable_pipelined_p2p() const;
|
|
void _internal_set_xla_gpu_enable_pipelined_p2p(bool value);
|
|
public:
|
|
|
|
// bool xla_gpu_enable_triton_gemm = 188;
|
|
void clear_xla_gpu_enable_triton_gemm();
|
|
bool xla_gpu_enable_triton_gemm() const;
|
|
void set_xla_gpu_enable_triton_gemm(bool value);
|
|
private:
|
|
bool _internal_xla_gpu_enable_triton_gemm() const;
|
|
void _internal_set_xla_gpu_enable_triton_gemm(bool value);
|
|
public:
|
|
|
|
// bool xla_gpu_enable_cudnn_int8x32_convolution_reordering = 189;
|
|
void clear_xla_gpu_enable_cudnn_int8x32_convolution_reordering();
|
|
bool xla_gpu_enable_cudnn_int8x32_convolution_reordering() const;
|
|
void set_xla_gpu_enable_cudnn_int8x32_convolution_reordering(bool value);
|
|
private:
|
|
bool _internal_xla_gpu_enable_cudnn_int8x32_convolution_reordering() const;
|
|
void _internal_set_xla_gpu_enable_cudnn_int8x32_convolution_reordering(bool value);
|
|
public:
|
|
|
|
// bool xla_gpu_triton_gemm_any = 190;
|
|
void clear_xla_gpu_triton_gemm_any();
|
|
bool xla_gpu_triton_gemm_any() const;
|
|
void set_xla_gpu_triton_gemm_any(bool value);
|
|
private:
|
|
bool _internal_xla_gpu_triton_gemm_any() const;
|
|
void _internal_set_xla_gpu_triton_gemm_any(bool value);
|
|
public:
|
|
|
|
// int32 xla_gpu_collective_inflation_factor = 205;
|
|
void clear_xla_gpu_collective_inflation_factor();
|
|
int32_t xla_gpu_collective_inflation_factor() const;
|
|
void set_xla_gpu_collective_inflation_factor(int32_t value);
|
|
private:
|
|
int32_t _internal_xla_gpu_collective_inflation_factor() const;
|
|
void _internal_set_xla_gpu_collective_inflation_factor(int32_t value);
|
|
public:
|
|
|
|
// int32 xla_cpu_sparse_cuda_threads = 207;
|
|
void clear_xla_cpu_sparse_cuda_threads();
|
|
int32_t xla_cpu_sparse_cuda_threads() const;
|
|
void set_xla_cpu_sparse_cuda_threads(int32_t value);
|
|
private:
|
|
int32_t _internal_xla_cpu_sparse_cuda_threads() const;
|
|
void _internal_set_xla_cpu_sparse_cuda_threads(int32_t value);
|
|
public:
|
|
|
|
// int32 xla_gpu_graph_min_graph_size = 208;
|
|
void clear_xla_gpu_graph_min_graph_size();
|
|
int32_t xla_gpu_graph_min_graph_size() const;
|
|
void set_xla_gpu_graph_min_graph_size(int32_t value);
|
|
private:
|
|
int32_t _internal_xla_gpu_graph_min_graph_size() const;
|
|
void _internal_set_xla_gpu_graph_min_graph_size(int32_t value);
|
|
public:
|
|
|
|
// int64 xla_gpu_all_gather_combine_threshold_bytes = 212;
|
|
void clear_xla_gpu_all_gather_combine_threshold_bytes();
|
|
int64_t xla_gpu_all_gather_combine_threshold_bytes() const;
|
|
void set_xla_gpu_all_gather_combine_threshold_bytes(int64_t value);
|
|
private:
|
|
int64_t _internal_xla_gpu_all_gather_combine_threshold_bytes() const;
|
|
void _internal_set_xla_gpu_all_gather_combine_threshold_bytes(int64_t value);
|
|
public:
|
|
|
|
// int64 xla_gpu_reduce_scatter_combine_threshold_bytes = 213;
|
|
void clear_xla_gpu_reduce_scatter_combine_threshold_bytes();
|
|
int64_t xla_gpu_reduce_scatter_combine_threshold_bytes() const;
|
|
void set_xla_gpu_reduce_scatter_combine_threshold_bytes(int64_t value);
|
|
private:
|
|
int64_t _internal_xla_gpu_reduce_scatter_combine_threshold_bytes() const;
|
|
void _internal_set_xla_gpu_reduce_scatter_combine_threshold_bytes(int64_t value);
|
|
public:
|
|
|
|
// bool xla_gpu_enable_latency_hiding_scheduler = 186;
|
|
void clear_xla_gpu_enable_latency_hiding_scheduler();
|
|
bool xla_gpu_enable_latency_hiding_scheduler() const;
|
|
void set_xla_gpu_enable_latency_hiding_scheduler(bool value);
|
|
private:
|
|
bool _internal_xla_gpu_enable_latency_hiding_scheduler() const;
|
|
void _internal_set_xla_gpu_enable_latency_hiding_scheduler(bool value);
|
|
public:
|
|
|
|
// bool xla_gpu_enable_highest_priority_async_stream = 216;
|
|
void clear_xla_gpu_enable_highest_priority_async_stream();
|
|
bool xla_gpu_enable_highest_priority_async_stream() const;
|
|
void set_xla_gpu_enable_highest_priority_async_stream(bool value);
|
|
private:
|
|
bool _internal_xla_gpu_enable_highest_priority_async_stream() const;
|
|
void _internal_set_xla_gpu_enable_highest_priority_async_stream(bool value);
|
|
public:
|
|
|
|
// bool xla_gpu_enable_analytical_latency_estimator = 255;
|
|
void clear_xla_gpu_enable_analytical_latency_estimator();
|
|
bool xla_gpu_enable_analytical_latency_estimator() const;
|
|
void set_xla_gpu_enable_analytical_latency_estimator(bool value);
|
|
private:
|
|
bool _internal_xla_gpu_enable_analytical_latency_estimator() const;
|
|
void _internal_set_xla_gpu_enable_analytical_latency_estimator(bool value);
|
|
public:
|
|
|
|
// bool xla_gpu_lhs_enable_gpu_async_tracker = 204;
|
|
void clear_xla_gpu_lhs_enable_gpu_async_tracker();
|
|
bool xla_gpu_lhs_enable_gpu_async_tracker() const;
|
|
void set_xla_gpu_lhs_enable_gpu_async_tracker(bool value);
|
|
private:
|
|
bool _internal_xla_gpu_lhs_enable_gpu_async_tracker() const;
|
|
void _internal_set_xla_gpu_lhs_enable_gpu_async_tracker(bool value);
|
|
public:
|
|
|
|
// bool xla_gpu_enable_async_all_to_all = 201;
|
|
void clear_xla_gpu_enable_async_all_to_all();
|
|
bool xla_gpu_enable_async_all_to_all() const;
|
|
void set_xla_gpu_enable_async_all_to_all(bool value);
|
|
private:
|
|
bool _internal_xla_gpu_enable_async_all_to_all() const;
|
|
void _internal_set_xla_gpu_enable_async_all_to_all(bool value);
|
|
public:
|
|
|
|
// bool xla_gpu_enable_all_gather_combine_by_dim = 254;
|
|
void clear_xla_gpu_enable_all_gather_combine_by_dim();
|
|
bool xla_gpu_enable_all_gather_combine_by_dim() const;
|
|
void set_xla_gpu_enable_all_gather_combine_by_dim(bool value);
|
|
private:
|
|
bool _internal_xla_gpu_enable_all_gather_combine_by_dim() const;
|
|
void _internal_set_xla_gpu_enable_all_gather_combine_by_dim(bool value);
|
|
public:
|
|
|
|
// bool xla_gpu_enable_reduce_scatter_combine_by_dim = 257;
|
|
void clear_xla_gpu_enable_reduce_scatter_combine_by_dim();
|
|
bool xla_gpu_enable_reduce_scatter_combine_by_dim() const;
|
|
void set_xla_gpu_enable_reduce_scatter_combine_by_dim(bool value);
|
|
private:
|
|
bool _internal_xla_gpu_enable_reduce_scatter_combine_by_dim() const;
|
|
void _internal_set_xla_gpu_enable_reduce_scatter_combine_by_dim(bool value);
|
|
public:
|
|
|
|
// bool xla_gpu_all_reduce_contiguous = 158;
|
|
void clear_xla_gpu_all_reduce_contiguous();
|
|
bool xla_gpu_all_reduce_contiguous() const;
|
|
void set_xla_gpu_all_reduce_contiguous(bool value);
|
|
private:
|
|
bool _internal_xla_gpu_all_reduce_contiguous() const;
|
|
void _internal_set_xla_gpu_all_reduce_contiguous(bool value);
|
|
public:
|
|
|
|
// bool xla_gpu_exhaustive_tiling_search = 219;
|
|
void clear_xla_gpu_exhaustive_tiling_search();
|
|
bool xla_gpu_exhaustive_tiling_search() const;
|
|
void set_xla_gpu_exhaustive_tiling_search(bool value);
|
|
private:
|
|
bool _internal_xla_gpu_exhaustive_tiling_search() const;
|
|
void _internal_set_xla_gpu_exhaustive_tiling_search(bool value);
|
|
public:
|
|
|
|
// bool xla_gpu_enable_triton_softmax_fusion = 220;
|
|
void clear_xla_gpu_enable_triton_softmax_fusion();
|
|
bool xla_gpu_enable_triton_softmax_fusion() const;
|
|
void set_xla_gpu_enable_triton_softmax_fusion(bool value);
|
|
private:
|
|
bool _internal_xla_gpu_enable_triton_softmax_fusion() const;
|
|
void _internal_set_xla_gpu_enable_triton_softmax_fusion(bool value);
|
|
public:
|
|
|
|
// bool xla_gpu_enable_priority_fusion = 221;
|
|
void clear_xla_gpu_enable_priority_fusion();
|
|
bool xla_gpu_enable_priority_fusion() const;
|
|
void set_xla_gpu_enable_priority_fusion(bool value);
|
|
private:
|
|
bool _internal_xla_gpu_enable_priority_fusion() const;
|
|
void _internal_set_xla_gpu_enable_priority_fusion(bool value);
|
|
public:
|
|
|
|
// bool xla_gpu_triton_gemm_disable_reduced_precision_reduction = 226;
|
|
void clear_xla_gpu_triton_gemm_disable_reduced_precision_reduction();
|
|
bool xla_gpu_triton_gemm_disable_reduced_precision_reduction() const;
|
|
void set_xla_gpu_triton_gemm_disable_reduced_precision_reduction(bool value);
|
|
private:
|
|
bool _internal_xla_gpu_triton_gemm_disable_reduced_precision_reduction() const;
|
|
void _internal_set_xla_gpu_triton_gemm_disable_reduced_precision_reduction(bool value);
|
|
public:
|
|
|
|
// int32 xla_gpu_auto_spmd_partitioning_memory_budget_gb = 224;
|
|
void clear_xla_gpu_auto_spmd_partitioning_memory_budget_gb();
|
|
int32_t xla_gpu_auto_spmd_partitioning_memory_budget_gb() const;
|
|
void set_xla_gpu_auto_spmd_partitioning_memory_budget_gb(int32_t value);
|
|
private:
|
|
int32_t _internal_xla_gpu_auto_spmd_partitioning_memory_budget_gb() const;
|
|
void _internal_set_xla_gpu_auto_spmd_partitioning_memory_budget_gb(int32_t value);
|
|
public:
|
|
|
|
// float xla_gpu_auto_spmd_partitioning_memory_budget_ratio = 225;
|
|
void clear_xla_gpu_auto_spmd_partitioning_memory_budget_ratio();
|
|
float xla_gpu_auto_spmd_partitioning_memory_budget_ratio() const;
|
|
void set_xla_gpu_auto_spmd_partitioning_memory_budget_ratio(float value);
|
|
private:
|
|
float _internal_xla_gpu_auto_spmd_partitioning_memory_budget_ratio() const;
|
|
void _internal_set_xla_gpu_auto_spmd_partitioning_memory_budget_ratio(float value);
|
|
public:
|
|
|
|
// bool xla_enable_dumping = 253;
|
|
void clear_xla_enable_dumping();
|
|
bool xla_enable_dumping() const;
|
|
void set_xla_enable_dumping(bool value);
|
|
private:
|
|
bool _internal_xla_enable_dumping() const;
|
|
void _internal_set_xla_enable_dumping(bool value);
|
|
public:
|
|
|
|
// bool xla_gpu_enable_llvm_module_compilation_parallelism = 268;
|
|
void clear_xla_gpu_enable_llvm_module_compilation_parallelism();
|
|
bool xla_gpu_enable_llvm_module_compilation_parallelism() const;
|
|
void set_xla_gpu_enable_llvm_module_compilation_parallelism(bool value);
|
|
private:
|
|
bool _internal_xla_gpu_enable_llvm_module_compilation_parallelism() const;
|
|
void _internal_set_xla_gpu_enable_llvm_module_compilation_parallelism(bool value);
|
|
public:
|
|
|
|
// bool xla_gpu_deterministic_ops = 148;
|
|
void clear_xla_gpu_deterministic_ops();
|
|
bool xla_gpu_deterministic_ops() const;
|
|
void set_xla_gpu_deterministic_ops(bool value);
|
|
private:
|
|
bool _internal_xla_gpu_deterministic_ops() const;
|
|
void _internal_set_xla_gpu_deterministic_ops(bool value);
|
|
public:
|
|
|
|
// bool xla_gpu_enable_async_collectives = 238;
|
|
void clear_xla_gpu_enable_async_collectives();
|
|
bool xla_gpu_enable_async_collectives() const;
|
|
void set_xla_gpu_enable_async_collectives(bool value);
|
|
private:
|
|
bool _internal_xla_gpu_enable_async_collectives() const;
|
|
void _internal_set_xla_gpu_enable_async_collectives(bool value);
|
|
public:
|
|
|
|
// int64 xla_gpu_redzone_padding_bytes = 228;
|
|
void clear_xla_gpu_redzone_padding_bytes();
|
|
int64_t xla_gpu_redzone_padding_bytes() const;
|
|
void set_xla_gpu_redzone_padding_bytes(int64_t value);
|
|
private:
|
|
int64_t _internal_xla_gpu_redzone_padding_bytes() const;
|
|
void _internal_set_xla_gpu_redzone_padding_bytes(int64_t value);
|
|
public:
|
|
|
|
// bool xla_gpu_enable_pipelined_collectives = 239;
|
|
void clear_xla_gpu_enable_pipelined_collectives();
|
|
bool xla_gpu_enable_pipelined_collectives() const;
|
|
void set_xla_gpu_enable_pipelined_collectives(bool value);
|
|
private:
|
|
bool _internal_xla_gpu_enable_pipelined_collectives() const;
|
|
void _internal_set_xla_gpu_enable_pipelined_collectives(bool value);
|
|
public:
|
|
|
|
// bool xla_gpu_enable_pipelined_all_reduce = 217;
|
|
void clear_xla_gpu_enable_pipelined_all_reduce();
|
|
bool xla_gpu_enable_pipelined_all_reduce() const;
|
|
void set_xla_gpu_enable_pipelined_all_reduce(bool value);
|
|
private:
|
|
bool _internal_xla_gpu_enable_pipelined_all_reduce() const;
|
|
void _internal_set_xla_gpu_enable_pipelined_all_reduce(bool value);
|
|
public:
|
|
|
|
// bool xla_gpu_enable_pipelined_all_gather = 227;
|
|
void clear_xla_gpu_enable_pipelined_all_gather();
|
|
bool xla_gpu_enable_pipelined_all_gather() const;
|
|
void set_xla_gpu_enable_pipelined_all_gather(bool value);
|
|
private:
|
|
bool _internal_xla_gpu_enable_pipelined_all_gather() const;
|
|
void _internal_set_xla_gpu_enable_pipelined_all_gather(bool value);
|
|
public:
|
|
|
|
// bool xla_gpu_enable_pipelined_reduce_scatter = 231;
|
|
void clear_xla_gpu_enable_pipelined_reduce_scatter();
|
|
bool xla_gpu_enable_pipelined_reduce_scatter() const;
|
|
void set_xla_gpu_enable_pipelined_reduce_scatter(bool value);
|
|
private:
|
|
bool _internal_xla_gpu_enable_pipelined_reduce_scatter() const;
|
|
void _internal_set_xla_gpu_enable_pipelined_reduce_scatter(bool value);
|
|
public:
|
|
|
|
// int32 xla_gpu_triton_fusion_level = 229;
|
|
void clear_xla_gpu_triton_fusion_level();
|
|
int32_t xla_gpu_triton_fusion_level() const;
|
|
void set_xla_gpu_triton_fusion_level(int32_t value);
|
|
private:
|
|
int32_t _internal_xla_gpu_triton_fusion_level() const;
|
|
void _internal_set_xla_gpu_triton_fusion_level(int32_t value);
|
|
public:
|
|
|
|
// int32 xla_gpu_graph_eviction_timeout_seconds = 230;
|
|
void clear_xla_gpu_graph_eviction_timeout_seconds();
|
|
int32_t xla_gpu_graph_eviction_timeout_seconds() const;
|
|
void set_xla_gpu_graph_eviction_timeout_seconds(int32_t value);
|
|
private:
|
|
int32_t _internal_xla_gpu_graph_eviction_timeout_seconds() const;
|
|
void _internal_set_xla_gpu_graph_eviction_timeout_seconds(int32_t value);
|
|
public:
|
|
|
|
// bool xla_gpu_dump_autotuned_triton_fusions = 232;
|
|
void clear_xla_gpu_dump_autotuned_triton_fusions();
|
|
bool xla_gpu_dump_autotuned_triton_fusions() const;
|
|
void set_xla_gpu_dump_autotuned_triton_fusions(bool value);
|
|
private:
|
|
bool _internal_xla_gpu_dump_autotuned_triton_fusions() const;
|
|
void _internal_set_xla_gpu_dump_autotuned_triton_fusions(bool value);
|
|
public:
|
|
|
|
// bool xla_gpu_copy_insertion_use_region_analysis = 236;
|
|
void clear_xla_gpu_copy_insertion_use_region_analysis();
|
|
bool xla_gpu_copy_insertion_use_region_analysis() const;
|
|
void set_xla_gpu_copy_insertion_use_region_analysis(bool value);
|
|
private:
|
|
bool _internal_xla_gpu_copy_insertion_use_region_analysis() const;
|
|
void _internal_set_xla_gpu_copy_insertion_use_region_analysis(bool value);
|
|
public:
|
|
|
|
// bool xla_gpu_collect_cost_model_stats = 240;
|
|
void clear_xla_gpu_collect_cost_model_stats();
|
|
bool xla_gpu_collect_cost_model_stats() const;
|
|
void set_xla_gpu_collect_cost_model_stats(bool value);
|
|
private:
|
|
bool _internal_xla_gpu_collect_cost_model_stats() const;
|
|
void _internal_set_xla_gpu_collect_cost_model_stats(bool value);
|
|
public:
|
|
|
|
// bool xla_gpu_enable_split_k_autotuning = 241;
|
|
void clear_xla_gpu_enable_split_k_autotuning();
|
|
bool xla_gpu_enable_split_k_autotuning() const;
|
|
void set_xla_gpu_enable_split_k_autotuning(bool value);
|
|
private:
|
|
bool _internal_xla_gpu_enable_split_k_autotuning() const;
|
|
void _internal_set_xla_gpu_enable_split_k_autotuning(bool value);
|
|
public:
|
|
|
|
// int64 xla_gpu_collective_permute_decomposer_threshold = 237;
|
|
void clear_xla_gpu_collective_permute_decomposer_threshold();
|
|
int64_t xla_gpu_collective_permute_decomposer_threshold() const;
|
|
void set_xla_gpu_collective_permute_decomposer_threshold(int64_t value);
|
|
private:
|
|
int64_t _internal_xla_gpu_collective_permute_decomposer_threshold() const;
|
|
void _internal_set_xla_gpu_collective_permute_decomposer_threshold(int64_t value);
|
|
public:
|
|
|
|
// bool xla_gpu_enable_reduction_epilogue_fusion = 243;
|
|
void clear_xla_gpu_enable_reduction_epilogue_fusion();
|
|
bool xla_gpu_enable_reduction_epilogue_fusion() const;
|
|
void set_xla_gpu_enable_reduction_epilogue_fusion(bool value);
|
|
private:
|
|
bool _internal_xla_gpu_enable_reduction_epilogue_fusion() const;
|
|
void _internal_set_xla_gpu_enable_reduction_epilogue_fusion(bool value);
|
|
public:
|
|
|
|
// bool xla_gpu_enable_nccl_clique_optimization = 244;
|
|
void clear_xla_gpu_enable_nccl_clique_optimization();
|
|
bool xla_gpu_enable_nccl_clique_optimization() const;
|
|
void set_xla_gpu_enable_nccl_clique_optimization(bool value);
|
|
private:
|
|
bool _internal_xla_gpu_enable_nccl_clique_optimization() const;
|
|
void _internal_set_xla_gpu_enable_nccl_clique_optimization(bool value);
|
|
public:
|
|
|
|
// bool xla_gpu_mock_custom_calls = 245;
|
|
void clear_xla_gpu_mock_custom_calls();
|
|
bool xla_gpu_mock_custom_calls() const;
|
|
void set_xla_gpu_mock_custom_calls(bool value);
|
|
private:
|
|
bool _internal_xla_gpu_mock_custom_calls() const;
|
|
void _internal_set_xla_gpu_mock_custom_calls(bool value);
|
|
public:
|
|
|
|
// bool xla_gpu_cublas_fallback = 247;
|
|
void clear_xla_gpu_cublas_fallback();
|
|
bool xla_gpu_cublas_fallback() const;
|
|
void set_xla_gpu_cublas_fallback(bool value);
|
|
private:
|
|
bool _internal_xla_gpu_cublas_fallback() const;
|
|
void _internal_set_xla_gpu_cublas_fallback(bool value);
|
|
public:
|
|
|
|
// bool xla_gpu_enable_while_loop_double_buffering = 248;
|
|
void clear_xla_gpu_enable_while_loop_double_buffering();
|
|
bool xla_gpu_enable_while_loop_double_buffering() const;
|
|
void set_xla_gpu_enable_while_loop_double_buffering(bool value);
|
|
private:
|
|
bool _internal_xla_gpu_enable_while_loop_double_buffering() const;
|
|
void _internal_set_xla_gpu_enable_while_loop_double_buffering(bool value);
|
|
public:
|
|
|
|
// bool xla_gpu_ensure_minor_dot_contraction_dims = 249;
|
|
void clear_xla_gpu_ensure_minor_dot_contraction_dims();
|
|
bool xla_gpu_ensure_minor_dot_contraction_dims() const;
|
|
void set_xla_gpu_ensure_minor_dot_contraction_dims(bool value);
|
|
private:
|
|
bool _internal_xla_gpu_ensure_minor_dot_contraction_dims() const;
|
|
void _internal_set_xla_gpu_ensure_minor_dot_contraction_dims(bool value);
|
|
public:
|
|
|
|
// bool xla_gpu_filter_kernels_spilling_registers_on_autotuning = 250;
|
|
void clear_xla_gpu_filter_kernels_spilling_registers_on_autotuning();
|
|
bool xla_gpu_filter_kernels_spilling_registers_on_autotuning() const;
|
|
void set_xla_gpu_filter_kernels_spilling_registers_on_autotuning(bool value);
|
|
private:
|
|
bool _internal_xla_gpu_filter_kernels_spilling_registers_on_autotuning() const;
|
|
void _internal_set_xla_gpu_filter_kernels_spilling_registers_on_autotuning(bool value);
|
|
public:
|
|
|
|
// bool xla_gpu_enable_cub_radix_sort = 259;
|
|
void clear_xla_gpu_enable_cub_radix_sort();
|
|
bool xla_gpu_enable_cub_radix_sort() const;
|
|
void set_xla_gpu_enable_cub_radix_sort(bool value);
|
|
private:
|
|
bool _internal_xla_gpu_enable_cub_radix_sort() const;
|
|
void _internal_set_xla_gpu_enable_cub_radix_sort(bool value);
|
|
public:
|
|
|
|
// int64 xla_debug_buffer_assignment_show_max = 251;
|
|
void clear_xla_debug_buffer_assignment_show_max();
|
|
int64_t xla_debug_buffer_assignment_show_max() const;
|
|
void set_xla_debug_buffer_assignment_show_max(int64_t value);
|
|
private:
|
|
int64_t _internal_xla_debug_buffer_assignment_show_max() const;
|
|
void _internal_set_xla_debug_buffer_assignment_show_max(int64_t value);
|
|
public:
|
|
|
|
// int32 xla_gpu_llvm_verification_level = 256;
|
|
void clear_xla_gpu_llvm_verification_level();
|
|
int32_t xla_gpu_llvm_verification_level() const;
|
|
void set_xla_gpu_llvm_verification_level(int32_t value);
|
|
private:
|
|
int32_t _internal_xla_gpu_llvm_verification_level() const;
|
|
void _internal_set_xla_gpu_llvm_verification_level(int32_t value);
|
|
public:
|
|
|
|
// int32 xla_gpu_memory_limit_slop_factor = 260;
|
|
void clear_xla_gpu_memory_limit_slop_factor();
|
|
int32_t xla_gpu_memory_limit_slop_factor() const;
|
|
void set_xla_gpu_memory_limit_slop_factor(int32_t value);
|
|
private:
|
|
int32_t _internal_xla_gpu_memory_limit_slop_factor() const;
|
|
void _internal_set_xla_gpu_memory_limit_slop_factor(int32_t value);
|
|
public:
|
|
|
|
// int64 xla_gpu_threshold_for_windowed_einsum_mib = 265;
|
|
void clear_xla_gpu_threshold_for_windowed_einsum_mib();
|
|
int64_t xla_gpu_threshold_for_windowed_einsum_mib() const;
|
|
void set_xla_gpu_threshold_for_windowed_einsum_mib(int64_t value);
|
|
private:
|
|
int64_t _internal_xla_gpu_threshold_for_windowed_einsum_mib() const;
|
|
void _internal_set_xla_gpu_threshold_for_windowed_einsum_mib(int64_t value);
|
|
public:
|
|
|
|
// bool xla_gpu_enable_triton_hopper = 266;
|
|
void clear_xla_gpu_enable_triton_hopper();
|
|
bool xla_gpu_enable_triton_hopper() const;
|
|
void set_xla_gpu_enable_triton_hopper(bool value);
|
|
private:
|
|
bool _internal_xla_gpu_enable_triton_hopper() const;
|
|
void _internal_set_xla_gpu_enable_triton_hopper(bool value);
|
|
public:
|
|
|
|
// bool xla_gpu_enable_nccl_user_buffers = 267;
|
|
void clear_xla_gpu_enable_nccl_user_buffers();
|
|
bool xla_gpu_enable_nccl_user_buffers() const;
|
|
void set_xla_gpu_enable_nccl_user_buffers(bool value);
|
|
private:
|
|
bool _internal_xla_gpu_enable_nccl_user_buffers() const;
|
|
void _internal_set_xla_gpu_enable_nccl_user_buffers(bool value);
|
|
public:
|
|
|
|
// bool xla_gpu_enable_libnvptxcompiler = 269;
|
|
void clear_xla_gpu_enable_libnvptxcompiler();
|
|
bool xla_gpu_enable_libnvptxcompiler() const;
|
|
void set_xla_gpu_enable_libnvptxcompiler(bool value);
|
|
private:
|
|
bool _internal_xla_gpu_enable_libnvptxcompiler() const;
|
|
void _internal_set_xla_gpu_enable_libnvptxcompiler(bool value);
|
|
public:
|
|
|
|
// bool xla_gpu_enable_dot_strength_reduction = 270;
|
|
void clear_xla_gpu_enable_dot_strength_reduction();
|
|
bool xla_gpu_enable_dot_strength_reduction() const;
|
|
void set_xla_gpu_enable_dot_strength_reduction(bool value);
|
|
private:
|
|
bool _internal_xla_gpu_enable_dot_strength_reduction() const;
|
|
void _internal_set_xla_gpu_enable_dot_strength_reduction(bool value);
|
|
public:
|
|
|
|
// @@protoc_insertion_point(class_scope:xla.DebugOptions)
|
|
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<std::string> xla_disable_hlo_passes_;
|
|
::PROTOBUF_NAMESPACE_ID::RepeatedPtrField<std::string> xla_enable_hlo_passes_only_;
|
|
::PROTOBUF_NAMESPACE_ID::RepeatedPtrField<std::string> xla_gpu_ptx_file_;
|
|
::PROTOBUF_NAMESPACE_ID::RepeatedPtrField<std::string> xla_gpu_llvm_ir_file_;
|
|
::PROTOBUF_NAMESPACE_ID::RepeatedField<int> xla_gpu_enable_command_buffer_;
|
|
mutable std::atomic<int> _xla_gpu_enable_command_buffer_cached_byte_size_;
|
|
::PROTOBUF_NAMESPACE_ID::internal::MapField<
|
|
DebugOptions_XlaBackendExtraOptionsEntry_DoNotUse,
|
|
std::string, std::string,
|
|
::PROTOBUF_NAMESPACE_ID::internal::WireFormatLite::TYPE_STRING,
|
|
::PROTOBUF_NAMESPACE_ID::internal::WireFormatLite::TYPE_STRING> xla_backend_extra_options_;
|
|
::PROTOBUF_NAMESPACE_ID::internal::ArenaStringPtr xla_gpu_cuda_data_dir_;
|
|
::PROTOBUF_NAMESPACE_ID::internal::ArenaStringPtr xla_dump_to_;
|
|
::PROTOBUF_NAMESPACE_ID::internal::ArenaStringPtr xla_dump_hlo_module_re_;
|
|
::PROTOBUF_NAMESPACE_ID::internal::ArenaStringPtr xla_dump_hlo_pass_re_;
|
|
::PROTOBUF_NAMESPACE_ID::internal::ArenaStringPtr xla_gpu_algorithm_denylist_path_;
|
|
::PROTOBUF_NAMESPACE_ID::internal::ArenaStringPtr xla_gpu_asm_extra_flags_;
|
|
::PROTOBUF_NAMESPACE_ID::internal::ArenaStringPtr xla_dump_hlo_pipeline_re_;
|
|
::PROTOBUF_NAMESPACE_ID::internal::ArenaStringPtr xla_gpu_pgle_profile_file_or_directory_path_;
|
|
::PROTOBUF_NAMESPACE_ID::internal::ArenaStringPtr xla_gpu_dump_autotune_results_to_;
|
|
::PROTOBUF_NAMESPACE_ID::internal::ArenaStringPtr xla_gpu_load_autotune_results_from_;
|
|
::PROTOBUF_NAMESPACE_ID::internal::ArenaStringPtr xla_gpu_target_config_filename_;
|
|
::PROTOBUF_NAMESPACE_ID::internal::ArenaStringPtr xla_gpu_enable_custom_fusions_re_;
|
|
int32_t xla_backend_optimization_level_;
|
|
bool xla_hlo_graph_addresses_;
|
|
bool xla_hlo_profile_;
|
|
bool xla_disable_all_hlo_passes_;
|
|
bool xla_embed_ir_in_executable_;
|
|
bool xla_eliminate_hlo_implicit_broadcast_;
|
|
bool xla_cpu_multi_thread_eigen_;
|
|
bool xla_gpu_ftz_;
|
|
bool xla_llvm_enable_alias_scope_metadata_;
|
|
bool xla_llvm_enable_noalias_metadata_;
|
|
bool xla_llvm_enable_invariant_load_metadata_;
|
|
bool xla_llvm_disable_expensive_passes_;
|
|
bool xla_test_all_output_layouts_;
|
|
int32_t xla_force_host_platform_device_count_;
|
|
bool xla_gpu_crash_on_verification_failures_;
|
|
bool xla_gpu_disable_gpuasm_optimizations_;
|
|
bool xla_hlo_evaluator_use_fast_path_;
|
|
bool xla_allow_scalar_index_dynamic_ops_;
|
|
int xla_step_marker_location_;
|
|
bool xla_dump_hlo_as_text_;
|
|
bool xla_dump_hlo_as_proto_;
|
|
bool xla_dump_hlo_as_dot_;
|
|
bool xla_dump_hlo_as_url_;
|
|
bool xla_test_all_input_layouts_;
|
|
bool xla_hlo_graph_sharding_color_;
|
|
bool xla_cpu_use_mkl_dnn_;
|
|
bool xla_cpu_use_xla_runtime_;
|
|
bool xla_cpu_enable_fast_math_;
|
|
bool xla_cpu_fast_math_honor_nans_;
|
|
bool xla_cpu_fast_math_honor_infs_;
|
|
bool xla_cpu_fast_math_honor_division_;
|
|
bool xla_cpu_fast_math_honor_functions_;
|
|
bool xla_cpu_enable_fast_min_max_;
|
|
bool xla_gpu_enable_fast_min_max_;
|
|
bool xla_allow_excess_precision_;
|
|
int32_t xla_gpu_autotune_level_;
|
|
bool xla_dump_hlo_as_html_;
|
|
bool xla_dump_fusion_visualization_;
|
|
bool xla_dump_hlo_snapshots_;
|
|
bool xla_dump_include_timestamp_;
|
|
int32_t xla_dump_max_hlo_modules_;
|
|
int32_t xla_multiheap_size_constraint_per_heap_;
|
|
bool xla_dump_module_metadata_;
|
|
bool xla_dump_compress_protos_;
|
|
bool xla_dump_hlo_as_long_text_;
|
|
bool xla_gpu_force_conv_nchw_;
|
|
int32_t xla_gpu_force_compilation_parallelism_;
|
|
bool xla_gpu_force_conv_nhwc_;
|
|
bool xla_gpu_dump_llvmir_;
|
|
bool xla_dump_enable_mlir_pretty_form_;
|
|
bool xla_tpu_detect_nan_;
|
|
int64_t xla_gpu_all_reduce_combine_threshold_bytes_;
|
|
int64_t xla_gpu_nccl_termination_timeout_seconds_;
|
|
int32_t xla_gpu_all_reduce_blueconnect_num_devices_per_host_;
|
|
bool xla_tpu_detect_inf_;
|
|
bool xla_cpu_enable_xprof_traceme_;
|
|
bool xla_gpu_unsafe_fallback_to_driver_on_ptxas_not_found_;
|
|
bool xla_detailed_logging_;
|
|
int64_t xla_gpu_redzone_scratch_max_megabytes_;
|
|
int xla_gpu_shape_checks_;
|
|
bool xla_gpu_enable_xla_runtime_executable_;
|
|
bool xla_gpu_enable_custom_fusions_;
|
|
bool xla_gpu_enable_address_computation_fusion_;
|
|
bool xla_gpu_enable_shared_constants_;
|
|
bool xla_cpu_use_acl_;
|
|
bool xla_cpu_strict_dot_conv_math_;
|
|
bool xla_gpu_use_runtime_fusion_;
|
|
bool xla_dump_latency_hiding_schedule_;
|
|
bool xla_gpu_enable_cublaslt_;
|
|
bool xla_gpu_graph_enable_concurrent_region_;
|
|
bool xla_gpu_simplify_all_fp_conversions_;
|
|
bool xla_gpu_normalize_layouts_;
|
|
bool xla_gpu_enable_async_all_reduce_;
|
|
bool xla_gpu_enable_async_collective_permute_;
|
|
bool xla_gpu_enable_async_all_gather_;
|
|
bool xla_gpu_enable_async_reduce_scatter_;
|
|
int xla_partitioning_algorithm_;
|
|
bool xla_cpu_enable_mlir_tiling_and_fusion_;
|
|
bool xla_cpu_enable_custom_matmul_tiling_;
|
|
bool xla_cpu_enable_mlir_fusion_outlining_;
|
|
bool xla_cpu_enable_experimental_deallocation_;
|
|
bool xla_gpu_enable_reassociation_for_converted_ar_;
|
|
bool xla_gpu_enable_while_loop_reduce_scatter_code_motion_;
|
|
bool xla_gpu_enable_cudnn_frontend_;
|
|
bool xla_gpu_enable_cudnn_fmha_;
|
|
int64_t xla_cpu_matmul_tiling_m_dim_;
|
|
int64_t xla_cpu_matmul_tiling_n_dim_;
|
|
int64_t xla_cpu_matmul_tiling_k_dim_;
|
|
bool xla_gpu_fused_attention_use_cudnn_rng_;
|
|
bool xla_gpu_enable_cudnn_layer_norm_;
|
|
bool xla_dump_disable_metadata_;
|
|
bool xla_gpu_strict_conv_algorithm_picker_;
|
|
int32_t xla_gpu_graph_num_runs_to_instantiate_;
|
|
bool xla_gpu_enable_pipelined_p2p_;
|
|
bool xla_gpu_enable_triton_gemm_;
|
|
bool xla_gpu_enable_cudnn_int8x32_convolution_reordering_;
|
|
bool xla_gpu_triton_gemm_any_;
|
|
int32_t xla_gpu_collective_inflation_factor_;
|
|
int32_t xla_cpu_sparse_cuda_threads_;
|
|
int32_t xla_gpu_graph_min_graph_size_;
|
|
int64_t xla_gpu_all_gather_combine_threshold_bytes_;
|
|
int64_t xla_gpu_reduce_scatter_combine_threshold_bytes_;
|
|
bool xla_gpu_enable_latency_hiding_scheduler_;
|
|
bool xla_gpu_enable_highest_priority_async_stream_;
|
|
bool xla_gpu_enable_analytical_latency_estimator_;
|
|
bool xla_gpu_lhs_enable_gpu_async_tracker_;
|
|
bool xla_gpu_enable_async_all_to_all_;
|
|
bool xla_gpu_enable_all_gather_combine_by_dim_;
|
|
bool xla_gpu_enable_reduce_scatter_combine_by_dim_;
|
|
bool xla_gpu_all_reduce_contiguous_;
|
|
bool xla_gpu_exhaustive_tiling_search_;
|
|
bool xla_gpu_enable_triton_softmax_fusion_;
|
|
bool xla_gpu_enable_priority_fusion_;
|
|
bool xla_gpu_triton_gemm_disable_reduced_precision_reduction_;
|
|
int32_t xla_gpu_auto_spmd_partitioning_memory_budget_gb_;
|
|
float xla_gpu_auto_spmd_partitioning_memory_budget_ratio_;
|
|
bool xla_enable_dumping_;
|
|
bool xla_gpu_enable_llvm_module_compilation_parallelism_;
|
|
bool xla_gpu_deterministic_ops_;
|
|
bool xla_gpu_enable_async_collectives_;
|
|
int64_t xla_gpu_redzone_padding_bytes_;
|
|
bool xla_gpu_enable_pipelined_collectives_;
|
|
bool xla_gpu_enable_pipelined_all_reduce_;
|
|
bool xla_gpu_enable_pipelined_all_gather_;
|
|
bool xla_gpu_enable_pipelined_reduce_scatter_;
|
|
int32_t xla_gpu_triton_fusion_level_;
|
|
int32_t xla_gpu_graph_eviction_timeout_seconds_;
|
|
bool xla_gpu_dump_autotuned_triton_fusions_;
|
|
bool xla_gpu_copy_insertion_use_region_analysis_;
|
|
bool xla_gpu_collect_cost_model_stats_;
|
|
bool xla_gpu_enable_split_k_autotuning_;
|
|
int64_t xla_gpu_collective_permute_decomposer_threshold_;
|
|
bool xla_gpu_enable_reduction_epilogue_fusion_;
|
|
bool xla_gpu_enable_nccl_clique_optimization_;
|
|
bool xla_gpu_mock_custom_calls_;
|
|
bool xla_gpu_cublas_fallback_;
|
|
bool xla_gpu_enable_while_loop_double_buffering_;
|
|
bool xla_gpu_ensure_minor_dot_contraction_dims_;
|
|
bool xla_gpu_filter_kernels_spilling_registers_on_autotuning_;
|
|
bool xla_gpu_enable_cub_radix_sort_;
|
|
int64_t xla_debug_buffer_assignment_show_max_;
|
|
int32_t xla_gpu_llvm_verification_level_;
|
|
int32_t xla_gpu_memory_limit_slop_factor_;
|
|
int64_t xla_gpu_threshold_for_windowed_einsum_mib_;
|
|
bool xla_gpu_enable_triton_hopper_;
|
|
bool xla_gpu_enable_nccl_user_buffers_;
|
|
bool xla_gpu_enable_libnvptxcompiler_;
|
|
bool xla_gpu_enable_dot_strength_reduction_;
|
|
mutable ::PROTOBUF_NAMESPACE_ID::internal::CachedSize _cached_size_;
|
|
};
|
|
union { Impl_ _impl_; };
|
|
friend struct ::TableStruct_xla_2fxla_2eproto;
|
|
};
|
|
// -------------------------------------------------------------------
|
|
|
|
class GpuCompilationEnvironment final :
|
|
public ::PROTOBUF_NAMESPACE_ID::Message /* @@protoc_insertion_point(class_definition:xla.GpuCompilationEnvironment) */ {
|
|
public:
|
|
inline GpuCompilationEnvironment() : GpuCompilationEnvironment(nullptr) {}
|
|
~GpuCompilationEnvironment() override;
|
|
explicit PROTOBUF_CONSTEXPR GpuCompilationEnvironment(::PROTOBUF_NAMESPACE_ID::internal::ConstantInitialized);
|
|
|
|
GpuCompilationEnvironment(const GpuCompilationEnvironment& from);
|
|
GpuCompilationEnvironment(GpuCompilationEnvironment&& from) noexcept
|
|
: GpuCompilationEnvironment() {
|
|
*this = ::std::move(from);
|
|
}
|
|
|
|
inline GpuCompilationEnvironment& operator=(const GpuCompilationEnvironment& from) {
|
|
CopyFrom(from);
|
|
return *this;
|
|
}
|
|
inline GpuCompilationEnvironment& operator=(GpuCompilationEnvironment&& 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 GpuCompilationEnvironment& default_instance() {
|
|
return *internal_default_instance();
|
|
}
|
|
static inline const GpuCompilationEnvironment* internal_default_instance() {
|
|
return reinterpret_cast<const GpuCompilationEnvironment*>(
|
|
&_GpuCompilationEnvironment_default_instance_);
|
|
}
|
|
static constexpr int kIndexInFileMessages =
|
|
3;
|
|
|
|
friend void swap(GpuCompilationEnvironment& a, GpuCompilationEnvironment& b) {
|
|
a.Swap(&b);
|
|
}
|
|
inline void Swap(GpuCompilationEnvironment* 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(GpuCompilationEnvironment* other) {
|
|
if (other == this) return;
|
|
GOOGLE_DCHECK(GetOwningArena() == other->GetOwningArena());
|
|
InternalSwap(other);
|
|
}
|
|
|
|
// implements Message ----------------------------------------------
|
|
|
|
GpuCompilationEnvironment* New(::PROTOBUF_NAMESPACE_ID::Arena* arena = nullptr) const final {
|
|
return CreateMaybeMessage<GpuCompilationEnvironment>(arena);
|
|
}
|
|
using ::PROTOBUF_NAMESPACE_ID::Message::CopyFrom;
|
|
void CopyFrom(const GpuCompilationEnvironment& from);
|
|
using ::PROTOBUF_NAMESPACE_ID::Message::MergeFrom;
|
|
void MergeFrom( const GpuCompilationEnvironment& from) {
|
|
GpuCompilationEnvironment::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(GpuCompilationEnvironment* other);
|
|
|
|
private:
|
|
friend class ::PROTOBUF_NAMESPACE_ID::internal::AnyMetadata;
|
|
static ::PROTOBUF_NAMESPACE_ID::StringPiece FullMessageName() {
|
|
return "xla.GpuCompilationEnvironment";
|
|
}
|
|
protected:
|
|
explicit GpuCompilationEnvironment(::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 {
|
|
kDummyFlagFieldNumber = 1,
|
|
};
|
|
// int64 dummy_flag = 1;
|
|
void clear_dummy_flag();
|
|
int64_t dummy_flag() const;
|
|
void set_dummy_flag(int64_t value);
|
|
private:
|
|
int64_t _internal_dummy_flag() const;
|
|
void _internal_set_dummy_flag(int64_t value);
|
|
public:
|
|
|
|
// @@protoc_insertion_point(class_scope:xla.GpuCompilationEnvironment)
|
|
private:
|
|
class _Internal;
|
|
|
|
template <typename T> friend class ::PROTOBUF_NAMESPACE_ID::Arena::InternalHelper;
|
|
typedef void InternalArenaConstructable_;
|
|
typedef void DestructorSkippable_;
|
|
struct Impl_ {
|
|
int64_t dummy_flag_;
|
|
mutable ::PROTOBUF_NAMESPACE_ID::internal::CachedSize _cached_size_;
|
|
};
|
|
union { Impl_ _impl_; };
|
|
friend struct ::TableStruct_xla_2fxla_2eproto;
|
|
};
|
|
// -------------------------------------------------------------------
|
|
|
|
class ShardableValueUpdatePairProto final :
|
|
public ::PROTOBUF_NAMESPACE_ID::Message /* @@protoc_insertion_point(class_definition:xla.ShardableValueUpdatePairProto) */ {
|
|
public:
|
|
inline ShardableValueUpdatePairProto() : ShardableValueUpdatePairProto(nullptr) {}
|
|
~ShardableValueUpdatePairProto() override;
|
|
explicit PROTOBUF_CONSTEXPR ShardableValueUpdatePairProto(::PROTOBUF_NAMESPACE_ID::internal::ConstantInitialized);
|
|
|
|
ShardableValueUpdatePairProto(const ShardableValueUpdatePairProto& from);
|
|
ShardableValueUpdatePairProto(ShardableValueUpdatePairProto&& from) noexcept
|
|
: ShardableValueUpdatePairProto() {
|
|
*this = ::std::move(from);
|
|
}
|
|
|
|
inline ShardableValueUpdatePairProto& operator=(const ShardableValueUpdatePairProto& from) {
|
|
CopyFrom(from);
|
|
return *this;
|
|
}
|
|
inline ShardableValueUpdatePairProto& operator=(ShardableValueUpdatePairProto&& 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 ShardableValueUpdatePairProto& default_instance() {
|
|
return *internal_default_instance();
|
|
}
|
|
static inline const ShardableValueUpdatePairProto* internal_default_instance() {
|
|
return reinterpret_cast<const ShardableValueUpdatePairProto*>(
|
|
&_ShardableValueUpdatePairProto_default_instance_);
|
|
}
|
|
static constexpr int kIndexInFileMessages =
|
|
4;
|
|
|
|
friend void swap(ShardableValueUpdatePairProto& a, ShardableValueUpdatePairProto& b) {
|
|
a.Swap(&b);
|
|
}
|
|
inline void Swap(ShardableValueUpdatePairProto* 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(ShardableValueUpdatePairProto* other) {
|
|
if (other == this) return;
|
|
GOOGLE_DCHECK(GetOwningArena() == other->GetOwningArena());
|
|
InternalSwap(other);
|
|
}
|
|
|
|
// implements Message ----------------------------------------------
|
|
|
|
ShardableValueUpdatePairProto* New(::PROTOBUF_NAMESPACE_ID::Arena* arena = nullptr) const final {
|
|
return CreateMaybeMessage<ShardableValueUpdatePairProto>(arena);
|
|
}
|
|
using ::PROTOBUF_NAMESPACE_ID::Message::CopyFrom;
|
|
void CopyFrom(const ShardableValueUpdatePairProto& from);
|
|
using ::PROTOBUF_NAMESPACE_ID::Message::MergeFrom;
|
|
void MergeFrom( const ShardableValueUpdatePairProto& from) {
|
|
ShardableValueUpdatePairProto::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(ShardableValueUpdatePairProto* other);
|
|
|
|
private:
|
|
friend class ::PROTOBUF_NAMESPACE_ID::internal::AnyMetadata;
|
|
static ::PROTOBUF_NAMESPACE_ID::StringPiece FullMessageName() {
|
|
return "xla.ShardableValueUpdatePairProto";
|
|
}
|
|
protected:
|
|
explicit ShardableValueUpdatePairProto(::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 {
|
|
kParameterShapeIndexFieldNumber = 2,
|
|
kOutputShapeIndexFieldNumber = 3,
|
|
kInputParameterNumberFieldNumber = 1,
|
|
};
|
|
// repeated int64 parameter_shape_index = 2;
|
|
int parameter_shape_index_size() const;
|
|
private:
|
|
int _internal_parameter_shape_index_size() const;
|
|
public:
|
|
void clear_parameter_shape_index();
|
|
private:
|
|
int64_t _internal_parameter_shape_index(int index) const;
|
|
const ::PROTOBUF_NAMESPACE_ID::RepeatedField< int64_t >&
|
|
_internal_parameter_shape_index() const;
|
|
void _internal_add_parameter_shape_index(int64_t value);
|
|
::PROTOBUF_NAMESPACE_ID::RepeatedField< int64_t >*
|
|
_internal_mutable_parameter_shape_index();
|
|
public:
|
|
int64_t parameter_shape_index(int index) const;
|
|
void set_parameter_shape_index(int index, int64_t value);
|
|
void add_parameter_shape_index(int64_t value);
|
|
const ::PROTOBUF_NAMESPACE_ID::RepeatedField< int64_t >&
|
|
parameter_shape_index() const;
|
|
::PROTOBUF_NAMESPACE_ID::RepeatedField< int64_t >*
|
|
mutable_parameter_shape_index();
|
|
|
|
// repeated int64 output_shape_index = 3;
|
|
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();
|
|
|
|
// int64 input_parameter_number = 1;
|
|
void clear_input_parameter_number();
|
|
int64_t input_parameter_number() const;
|
|
void set_input_parameter_number(int64_t value);
|
|
private:
|
|
int64_t _internal_input_parameter_number() const;
|
|
void _internal_set_input_parameter_number(int64_t value);
|
|
public:
|
|
|
|
// @@protoc_insertion_point(class_scope:xla.ShardableValueUpdatePairProto)
|
|
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 > parameter_shape_index_;
|
|
mutable std::atomic<int> _parameter_shape_index_cached_byte_size_;
|
|
::PROTOBUF_NAMESPACE_ID::RepeatedField< int64_t > output_shape_index_;
|
|
mutable std::atomic<int> _output_shape_index_cached_byte_size_;
|
|
int64_t input_parameter_number_;
|
|
mutable ::PROTOBUF_NAMESPACE_ID::internal::CachedSize _cached_size_;
|
|
};
|
|
union { Impl_ _impl_; };
|
|
friend struct ::TableStruct_xla_2fxla_2eproto;
|
|
};
|
|
// -------------------------------------------------------------------
|
|
|
|
class ExecutionOptions final :
|
|
public ::PROTOBUF_NAMESPACE_ID::Message /* @@protoc_insertion_point(class_definition:xla.ExecutionOptions) */ {
|
|
public:
|
|
inline ExecutionOptions() : ExecutionOptions(nullptr) {}
|
|
~ExecutionOptions() override;
|
|
explicit PROTOBUF_CONSTEXPR ExecutionOptions(::PROTOBUF_NAMESPACE_ID::internal::ConstantInitialized);
|
|
|
|
ExecutionOptions(const ExecutionOptions& from);
|
|
ExecutionOptions(ExecutionOptions&& from) noexcept
|
|
: ExecutionOptions() {
|
|
*this = ::std::move(from);
|
|
}
|
|
|
|
inline ExecutionOptions& operator=(const ExecutionOptions& from) {
|
|
CopyFrom(from);
|
|
return *this;
|
|
}
|
|
inline ExecutionOptions& operator=(ExecutionOptions&& 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 ExecutionOptions& default_instance() {
|
|
return *internal_default_instance();
|
|
}
|
|
static inline const ExecutionOptions* internal_default_instance() {
|
|
return reinterpret_cast<const ExecutionOptions*>(
|
|
&_ExecutionOptions_default_instance_);
|
|
}
|
|
static constexpr int kIndexInFileMessages =
|
|
5;
|
|
|
|
friend void swap(ExecutionOptions& a, ExecutionOptions& b) {
|
|
a.Swap(&b);
|
|
}
|
|
inline void Swap(ExecutionOptions* 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(ExecutionOptions* other) {
|
|
if (other == this) return;
|
|
GOOGLE_DCHECK(GetOwningArena() == other->GetOwningArena());
|
|
InternalSwap(other);
|
|
}
|
|
|
|
// implements Message ----------------------------------------------
|
|
|
|
ExecutionOptions* New(::PROTOBUF_NAMESPACE_ID::Arena* arena = nullptr) const final {
|
|
return CreateMaybeMessage<ExecutionOptions>(arena);
|
|
}
|
|
using ::PROTOBUF_NAMESPACE_ID::Message::CopyFrom;
|
|
void CopyFrom(const ExecutionOptions& from);
|
|
using ::PROTOBUF_NAMESPACE_ID::Message::MergeFrom;
|
|
void MergeFrom( const ExecutionOptions& from) {
|
|
ExecutionOptions::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(ExecutionOptions* other);
|
|
|
|
private:
|
|
friend class ::PROTOBUF_NAMESPACE_ID::internal::AnyMetadata;
|
|
static ::PROTOBUF_NAMESPACE_ID::StringPiece FullMessageName() {
|
|
return "xla.ExecutionOptions";
|
|
}
|
|
protected:
|
|
explicit ExecutionOptions(::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 {
|
|
kDeviceHandlesFieldNumber = 5,
|
|
kAllowSpmdShardingPropagationToOutputFieldNumber = 14,
|
|
kAutoSpmdPartitioningMeshShapeFieldNumber = 16,
|
|
kAutoSpmdPartitioningMeshIdsFieldNumber = 17,
|
|
kParamRequiresBroadcastViaCollectivesFieldNumber = 18,
|
|
kShardableValueUpdatePairsFieldNumber = 20,
|
|
kFdoProfileFieldNumber = 21,
|
|
kShapeWithOutputLayoutFieldNumber = 2,
|
|
kDebugOptionsFieldNumber = 4,
|
|
kDeviceAssignmentFieldNumber = 7,
|
|
kSeedFieldNumber = 3,
|
|
kNumReplicasFieldNumber = 6,
|
|
kNumPartitionsFieldNumber = 9,
|
|
kLaunchIdFieldNumber = 10,
|
|
kAliasPassthroughParamsFieldNumber = 8,
|
|
kUseSpmdPartitioningFieldNumber = 11,
|
|
kUseAutoSpmdPartitioningFieldNumber = 15,
|
|
kDeduplicateHloFieldNumber = 12,
|
|
kDeviceMemorySizeFieldNumber = 22,
|
|
kAllowSeparateShardingProgramsFieldNumber = 19,
|
|
};
|
|
// repeated .xla.DeviceHandle device_handles = 5;
|
|
int device_handles_size() const;
|
|
private:
|
|
int _internal_device_handles_size() const;
|
|
public:
|
|
void clear_device_handles();
|
|
::xla::DeviceHandle* mutable_device_handles(int index);
|
|
::PROTOBUF_NAMESPACE_ID::RepeatedPtrField< ::xla::DeviceHandle >*
|
|
mutable_device_handles();
|
|
private:
|
|
const ::xla::DeviceHandle& _internal_device_handles(int index) const;
|
|
::xla::DeviceHandle* _internal_add_device_handles();
|
|
public:
|
|
const ::xla::DeviceHandle& device_handles(int index) const;
|
|
::xla::DeviceHandle* add_device_handles();
|
|
const ::PROTOBUF_NAMESPACE_ID::RepeatedPtrField< ::xla::DeviceHandle >&
|
|
device_handles() const;
|
|
|
|
// repeated bool allow_spmd_sharding_propagation_to_output = 14;
|
|
int allow_spmd_sharding_propagation_to_output_size() const;
|
|
private:
|
|
int _internal_allow_spmd_sharding_propagation_to_output_size() const;
|
|
public:
|
|
void clear_allow_spmd_sharding_propagation_to_output();
|
|
private:
|
|
bool _internal_allow_spmd_sharding_propagation_to_output(int index) const;
|
|
const ::PROTOBUF_NAMESPACE_ID::RepeatedField< bool >&
|
|
_internal_allow_spmd_sharding_propagation_to_output() const;
|
|
void _internal_add_allow_spmd_sharding_propagation_to_output(bool value);
|
|
::PROTOBUF_NAMESPACE_ID::RepeatedField< bool >*
|
|
_internal_mutable_allow_spmd_sharding_propagation_to_output();
|
|
public:
|
|
bool allow_spmd_sharding_propagation_to_output(int index) const;
|
|
void set_allow_spmd_sharding_propagation_to_output(int index, bool value);
|
|
void add_allow_spmd_sharding_propagation_to_output(bool value);
|
|
const ::PROTOBUF_NAMESPACE_ID::RepeatedField< bool >&
|
|
allow_spmd_sharding_propagation_to_output() const;
|
|
::PROTOBUF_NAMESPACE_ID::RepeatedField< bool >*
|
|
mutable_allow_spmd_sharding_propagation_to_output();
|
|
|
|
// repeated int64 auto_spmd_partitioning_mesh_shape = 16;
|
|
int auto_spmd_partitioning_mesh_shape_size() const;
|
|
private:
|
|
int _internal_auto_spmd_partitioning_mesh_shape_size() const;
|
|
public:
|
|
void clear_auto_spmd_partitioning_mesh_shape();
|
|
private:
|
|
int64_t _internal_auto_spmd_partitioning_mesh_shape(int index) const;
|
|
const ::PROTOBUF_NAMESPACE_ID::RepeatedField< int64_t >&
|
|
_internal_auto_spmd_partitioning_mesh_shape() const;
|
|
void _internal_add_auto_spmd_partitioning_mesh_shape(int64_t value);
|
|
::PROTOBUF_NAMESPACE_ID::RepeatedField< int64_t >*
|
|
_internal_mutable_auto_spmd_partitioning_mesh_shape();
|
|
public:
|
|
int64_t auto_spmd_partitioning_mesh_shape(int index) const;
|
|
void set_auto_spmd_partitioning_mesh_shape(int index, int64_t value);
|
|
void add_auto_spmd_partitioning_mesh_shape(int64_t value);
|
|
const ::PROTOBUF_NAMESPACE_ID::RepeatedField< int64_t >&
|
|
auto_spmd_partitioning_mesh_shape() const;
|
|
::PROTOBUF_NAMESPACE_ID::RepeatedField< int64_t >*
|
|
mutable_auto_spmd_partitioning_mesh_shape();
|
|
|
|
// repeated int64 auto_spmd_partitioning_mesh_ids = 17;
|
|
int auto_spmd_partitioning_mesh_ids_size() const;
|
|
private:
|
|
int _internal_auto_spmd_partitioning_mesh_ids_size() const;
|
|
public:
|
|
void clear_auto_spmd_partitioning_mesh_ids();
|
|
private:
|
|
int64_t _internal_auto_spmd_partitioning_mesh_ids(int index) const;
|
|
const ::PROTOBUF_NAMESPACE_ID::RepeatedField< int64_t >&
|
|
_internal_auto_spmd_partitioning_mesh_ids() const;
|
|
void _internal_add_auto_spmd_partitioning_mesh_ids(int64_t value);
|
|
::PROTOBUF_NAMESPACE_ID::RepeatedField< int64_t >*
|
|
_internal_mutable_auto_spmd_partitioning_mesh_ids();
|
|
public:
|
|
int64_t auto_spmd_partitioning_mesh_ids(int index) const;
|
|
void set_auto_spmd_partitioning_mesh_ids(int index, int64_t value);
|
|
void add_auto_spmd_partitioning_mesh_ids(int64_t value);
|
|
const ::PROTOBUF_NAMESPACE_ID::RepeatedField< int64_t >&
|
|
auto_spmd_partitioning_mesh_ids() const;
|
|
::PROTOBUF_NAMESPACE_ID::RepeatedField< int64_t >*
|
|
mutable_auto_spmd_partitioning_mesh_ids();
|
|
|
|
// repeated bool param_requires_broadcast_via_collectives = 18;
|
|
int param_requires_broadcast_via_collectives_size() const;
|
|
private:
|
|
int _internal_param_requires_broadcast_via_collectives_size() const;
|
|
public:
|
|
void clear_param_requires_broadcast_via_collectives();
|
|
private:
|
|
bool _internal_param_requires_broadcast_via_collectives(int index) const;
|
|
const ::PROTOBUF_NAMESPACE_ID::RepeatedField< bool >&
|
|
_internal_param_requires_broadcast_via_collectives() const;
|
|
void _internal_add_param_requires_broadcast_via_collectives(bool value);
|
|
::PROTOBUF_NAMESPACE_ID::RepeatedField< bool >*
|
|
_internal_mutable_param_requires_broadcast_via_collectives();
|
|
public:
|
|
bool param_requires_broadcast_via_collectives(int index) const;
|
|
void set_param_requires_broadcast_via_collectives(int index, bool value);
|
|
void add_param_requires_broadcast_via_collectives(bool value);
|
|
const ::PROTOBUF_NAMESPACE_ID::RepeatedField< bool >&
|
|
param_requires_broadcast_via_collectives() const;
|
|
::PROTOBUF_NAMESPACE_ID::RepeatedField< bool >*
|
|
mutable_param_requires_broadcast_via_collectives();
|
|
|
|
// repeated .xla.ShardableValueUpdatePairProto shardable_value_update_pairs = 20;
|
|
int shardable_value_update_pairs_size() const;
|
|
private:
|
|
int _internal_shardable_value_update_pairs_size() const;
|
|
public:
|
|
void clear_shardable_value_update_pairs();
|
|
::xla::ShardableValueUpdatePairProto* mutable_shardable_value_update_pairs(int index);
|
|
::PROTOBUF_NAMESPACE_ID::RepeatedPtrField< ::xla::ShardableValueUpdatePairProto >*
|
|
mutable_shardable_value_update_pairs();
|
|
private:
|
|
const ::xla::ShardableValueUpdatePairProto& _internal_shardable_value_update_pairs(int index) const;
|
|
::xla::ShardableValueUpdatePairProto* _internal_add_shardable_value_update_pairs();
|
|
public:
|
|
const ::xla::ShardableValueUpdatePairProto& shardable_value_update_pairs(int index) const;
|
|
::xla::ShardableValueUpdatePairProto* add_shardable_value_update_pairs();
|
|
const ::PROTOBUF_NAMESPACE_ID::RepeatedPtrField< ::xla::ShardableValueUpdatePairProto >&
|
|
shardable_value_update_pairs() const;
|
|
|
|
// bytes fdo_profile = 21;
|
|
void clear_fdo_profile();
|
|
const std::string& fdo_profile() const;
|
|
template <typename ArgT0 = const std::string&, typename... ArgT>
|
|
void set_fdo_profile(ArgT0&& arg0, ArgT... args);
|
|
std::string* mutable_fdo_profile();
|
|
PROTOBUF_NODISCARD std::string* release_fdo_profile();
|
|
void set_allocated_fdo_profile(std::string* fdo_profile);
|
|
private:
|
|
const std::string& _internal_fdo_profile() const;
|
|
inline PROTOBUF_ALWAYS_INLINE void _internal_set_fdo_profile(const std::string& value);
|
|
std::string* _internal_mutable_fdo_profile();
|
|
public:
|
|
|
|
// .xla.ShapeProto shape_with_output_layout = 2;
|
|
bool has_shape_with_output_layout() const;
|
|
private:
|
|
bool _internal_has_shape_with_output_layout() const;
|
|
public:
|
|
void clear_shape_with_output_layout();
|
|
const ::xla::ShapeProto& shape_with_output_layout() const;
|
|
PROTOBUF_NODISCARD ::xla::ShapeProto* release_shape_with_output_layout();
|
|
::xla::ShapeProto* mutable_shape_with_output_layout();
|
|
void set_allocated_shape_with_output_layout(::xla::ShapeProto* shape_with_output_layout);
|
|
private:
|
|
const ::xla::ShapeProto& _internal_shape_with_output_layout() const;
|
|
::xla::ShapeProto* _internal_mutable_shape_with_output_layout();
|
|
public:
|
|
void unsafe_arena_set_allocated_shape_with_output_layout(
|
|
::xla::ShapeProto* shape_with_output_layout);
|
|
::xla::ShapeProto* unsafe_arena_release_shape_with_output_layout();
|
|
|
|
// .xla.DebugOptions debug_options = 4;
|
|
bool has_debug_options() const;
|
|
private:
|
|
bool _internal_has_debug_options() const;
|
|
public:
|
|
void clear_debug_options();
|
|
const ::xla::DebugOptions& debug_options() const;
|
|
PROTOBUF_NODISCARD ::xla::DebugOptions* release_debug_options();
|
|
::xla::DebugOptions* mutable_debug_options();
|
|
void set_allocated_debug_options(::xla::DebugOptions* debug_options);
|
|
private:
|
|
const ::xla::DebugOptions& _internal_debug_options() const;
|
|
::xla::DebugOptions* _internal_mutable_debug_options();
|
|
public:
|
|
void unsafe_arena_set_allocated_debug_options(
|
|
::xla::DebugOptions* debug_options);
|
|
::xla::DebugOptions* unsafe_arena_release_debug_options();
|
|
|
|
// .xla.DeviceAssignmentProto device_assignment = 7;
|
|
bool has_device_assignment() const;
|
|
private:
|
|
bool _internal_has_device_assignment() const;
|
|
public:
|
|
void clear_device_assignment();
|
|
const ::xla::DeviceAssignmentProto& device_assignment() const;
|
|
PROTOBUF_NODISCARD ::xla::DeviceAssignmentProto* release_device_assignment();
|
|
::xla::DeviceAssignmentProto* mutable_device_assignment();
|
|
void set_allocated_device_assignment(::xla::DeviceAssignmentProto* device_assignment);
|
|
private:
|
|
const ::xla::DeviceAssignmentProto& _internal_device_assignment() const;
|
|
::xla::DeviceAssignmentProto* _internal_mutable_device_assignment();
|
|
public:
|
|
void unsafe_arena_set_allocated_device_assignment(
|
|
::xla::DeviceAssignmentProto* device_assignment);
|
|
::xla::DeviceAssignmentProto* unsafe_arena_release_device_assignment();
|
|
|
|
// uint64 seed = 3;
|
|
void clear_seed();
|
|
uint64_t seed() const;
|
|
void set_seed(uint64_t value);
|
|
private:
|
|
uint64_t _internal_seed() const;
|
|
void _internal_set_seed(uint64_t value);
|
|
public:
|
|
|
|
// int32 num_replicas = 6;
|
|
void clear_num_replicas();
|
|
int32_t num_replicas() const;
|
|
void set_num_replicas(int32_t value);
|
|
private:
|
|
int32_t _internal_num_replicas() const;
|
|
void _internal_set_num_replicas(int32_t value);
|
|
public:
|
|
|
|
// int32 num_partitions = 9;
|
|
void clear_num_partitions();
|
|
int32_t num_partitions() const;
|
|
void set_num_partitions(int32_t value);
|
|
private:
|
|
int32_t _internal_num_partitions() const;
|
|
void _internal_set_num_partitions(int32_t value);
|
|
public:
|
|
|
|
// int32 launch_id = 10;
|
|
void clear_launch_id();
|
|
int32_t launch_id() const;
|
|
void set_launch_id(int32_t value);
|
|
private:
|
|
int32_t _internal_launch_id() const;
|
|
void _internal_set_launch_id(int32_t value);
|
|
public:
|
|
|
|
// bool alias_passthrough_params = 8;
|
|
void clear_alias_passthrough_params();
|
|
bool alias_passthrough_params() const;
|
|
void set_alias_passthrough_params(bool value);
|
|
private:
|
|
bool _internal_alias_passthrough_params() const;
|
|
void _internal_set_alias_passthrough_params(bool value);
|
|
public:
|
|
|
|
// bool use_spmd_partitioning = 11;
|
|
void clear_use_spmd_partitioning();
|
|
bool use_spmd_partitioning() const;
|
|
void set_use_spmd_partitioning(bool value);
|
|
private:
|
|
bool _internal_use_spmd_partitioning() const;
|
|
void _internal_set_use_spmd_partitioning(bool value);
|
|
public:
|
|
|
|
// bool use_auto_spmd_partitioning = 15;
|
|
void clear_use_auto_spmd_partitioning();
|
|
bool use_auto_spmd_partitioning() const;
|
|
void set_use_auto_spmd_partitioning(bool value);
|
|
private:
|
|
bool _internal_use_auto_spmd_partitioning() const;
|
|
void _internal_set_use_auto_spmd_partitioning(bool value);
|
|
public:
|
|
|
|
// bool deduplicate_hlo = 12;
|
|
void clear_deduplicate_hlo();
|
|
bool deduplicate_hlo() const;
|
|
void set_deduplicate_hlo(bool value);
|
|
private:
|
|
bool _internal_deduplicate_hlo() const;
|
|
void _internal_set_deduplicate_hlo(bool value);
|
|
public:
|
|
|
|
// int64 device_memory_size = 22;
|
|
void clear_device_memory_size();
|
|
int64_t device_memory_size() const;
|
|
void set_device_memory_size(int64_t value);
|
|
private:
|
|
int64_t _internal_device_memory_size() const;
|
|
void _internal_set_device_memory_size(int64_t value);
|
|
public:
|
|
|
|
// bool allow_separate_sharding_programs = 19;
|
|
void clear_allow_separate_sharding_programs();
|
|
bool allow_separate_sharding_programs() const;
|
|
void set_allow_separate_sharding_programs(bool value);
|
|
private:
|
|
bool _internal_allow_separate_sharding_programs() const;
|
|
void _internal_set_allow_separate_sharding_programs(bool value);
|
|
public:
|
|
|
|
// @@protoc_insertion_point(class_scope:xla.ExecutionOptions)
|
|
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::DeviceHandle > device_handles_;
|
|
::PROTOBUF_NAMESPACE_ID::RepeatedField< bool > allow_spmd_sharding_propagation_to_output_;
|
|
::PROTOBUF_NAMESPACE_ID::RepeatedField< int64_t > auto_spmd_partitioning_mesh_shape_;
|
|
mutable std::atomic<int> _auto_spmd_partitioning_mesh_shape_cached_byte_size_;
|
|
::PROTOBUF_NAMESPACE_ID::RepeatedField< int64_t > auto_spmd_partitioning_mesh_ids_;
|
|
mutable std::atomic<int> _auto_spmd_partitioning_mesh_ids_cached_byte_size_;
|
|
::PROTOBUF_NAMESPACE_ID::RepeatedField< bool > param_requires_broadcast_via_collectives_;
|
|
::PROTOBUF_NAMESPACE_ID::RepeatedPtrField< ::xla::ShardableValueUpdatePairProto > shardable_value_update_pairs_;
|
|
::PROTOBUF_NAMESPACE_ID::internal::ArenaStringPtr fdo_profile_;
|
|
::xla::ShapeProto* shape_with_output_layout_;
|
|
::xla::DebugOptions* debug_options_;
|
|
::xla::DeviceAssignmentProto* device_assignment_;
|
|
uint64_t seed_;
|
|
int32_t num_replicas_;
|
|
int32_t num_partitions_;
|
|
int32_t launch_id_;
|
|
bool alias_passthrough_params_;
|
|
bool use_spmd_partitioning_;
|
|
bool use_auto_spmd_partitioning_;
|
|
bool deduplicate_hlo_;
|
|
int64_t device_memory_size_;
|
|
bool allow_separate_sharding_programs_;
|
|
mutable ::PROTOBUF_NAMESPACE_ID::internal::CachedSize _cached_size_;
|
|
};
|
|
union { Impl_ _impl_; };
|
|
friend struct ::TableStruct_xla_2fxla_2eproto;
|
|
};
|
|
// -------------------------------------------------------------------
|
|
|
|
class HloModuleConfigProto_BoolList final :
|
|
public ::PROTOBUF_NAMESPACE_ID::Message /* @@protoc_insertion_point(class_definition:xla.HloModuleConfigProto.BoolList) */ {
|
|
public:
|
|
inline HloModuleConfigProto_BoolList() : HloModuleConfigProto_BoolList(nullptr) {}
|
|
~HloModuleConfigProto_BoolList() override;
|
|
explicit PROTOBUF_CONSTEXPR HloModuleConfigProto_BoolList(::PROTOBUF_NAMESPACE_ID::internal::ConstantInitialized);
|
|
|
|
HloModuleConfigProto_BoolList(const HloModuleConfigProto_BoolList& from);
|
|
HloModuleConfigProto_BoolList(HloModuleConfigProto_BoolList&& from) noexcept
|
|
: HloModuleConfigProto_BoolList() {
|
|
*this = ::std::move(from);
|
|
}
|
|
|
|
inline HloModuleConfigProto_BoolList& operator=(const HloModuleConfigProto_BoolList& from) {
|
|
CopyFrom(from);
|
|
return *this;
|
|
}
|
|
inline HloModuleConfigProto_BoolList& operator=(HloModuleConfigProto_BoolList&& 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 HloModuleConfigProto_BoolList& default_instance() {
|
|
return *internal_default_instance();
|
|
}
|
|
static inline const HloModuleConfigProto_BoolList* internal_default_instance() {
|
|
return reinterpret_cast<const HloModuleConfigProto_BoolList*>(
|
|
&_HloModuleConfigProto_BoolList_default_instance_);
|
|
}
|
|
static constexpr int kIndexInFileMessages =
|
|
6;
|
|
|
|
friend void swap(HloModuleConfigProto_BoolList& a, HloModuleConfigProto_BoolList& b) {
|
|
a.Swap(&b);
|
|
}
|
|
inline void Swap(HloModuleConfigProto_BoolList* 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(HloModuleConfigProto_BoolList* other) {
|
|
if (other == this) return;
|
|
GOOGLE_DCHECK(GetOwningArena() == other->GetOwningArena());
|
|
InternalSwap(other);
|
|
}
|
|
|
|
// implements Message ----------------------------------------------
|
|
|
|
HloModuleConfigProto_BoolList* New(::PROTOBUF_NAMESPACE_ID::Arena* arena = nullptr) const final {
|
|
return CreateMaybeMessage<HloModuleConfigProto_BoolList>(arena);
|
|
}
|
|
using ::PROTOBUF_NAMESPACE_ID::Message::CopyFrom;
|
|
void CopyFrom(const HloModuleConfigProto_BoolList& from);
|
|
using ::PROTOBUF_NAMESPACE_ID::Message::MergeFrom;
|
|
void MergeFrom( const HloModuleConfigProto_BoolList& from) {
|
|
HloModuleConfigProto_BoolList::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(HloModuleConfigProto_BoolList* other);
|
|
|
|
private:
|
|
friend class ::PROTOBUF_NAMESPACE_ID::internal::AnyMetadata;
|
|
static ::PROTOBUF_NAMESPACE_ID::StringPiece FullMessageName() {
|
|
return "xla.HloModuleConfigProto.BoolList";
|
|
}
|
|
protected:
|
|
explicit HloModuleConfigProto_BoolList(::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 {
|
|
kValsFieldNumber = 1,
|
|
};
|
|
// repeated bool vals = 1;
|
|
int vals_size() const;
|
|
private:
|
|
int _internal_vals_size() const;
|
|
public:
|
|
void clear_vals();
|
|
private:
|
|
bool _internal_vals(int index) const;
|
|
const ::PROTOBUF_NAMESPACE_ID::RepeatedField< bool >&
|
|
_internal_vals() const;
|
|
void _internal_add_vals(bool value);
|
|
::PROTOBUF_NAMESPACE_ID::RepeatedField< bool >*
|
|
_internal_mutable_vals();
|
|
public:
|
|
bool vals(int index) const;
|
|
void set_vals(int index, bool value);
|
|
void add_vals(bool value);
|
|
const ::PROTOBUF_NAMESPACE_ID::RepeatedField< bool >&
|
|
vals() const;
|
|
::PROTOBUF_NAMESPACE_ID::RepeatedField< bool >*
|
|
mutable_vals();
|
|
|
|
// @@protoc_insertion_point(class_scope:xla.HloModuleConfigProto.BoolList)
|
|
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 > vals_;
|
|
mutable ::PROTOBUF_NAMESPACE_ID::internal::CachedSize _cached_size_;
|
|
};
|
|
union { Impl_ _impl_; };
|
|
friend struct ::TableStruct_xla_2fxla_2eproto;
|
|
};
|
|
// -------------------------------------------------------------------
|
|
|
|
class HloModuleConfigProto_Int64List final :
|
|
public ::PROTOBUF_NAMESPACE_ID::Message /* @@protoc_insertion_point(class_definition:xla.HloModuleConfigProto.Int64List) */ {
|
|
public:
|
|
inline HloModuleConfigProto_Int64List() : HloModuleConfigProto_Int64List(nullptr) {}
|
|
~HloModuleConfigProto_Int64List() override;
|
|
explicit PROTOBUF_CONSTEXPR HloModuleConfigProto_Int64List(::PROTOBUF_NAMESPACE_ID::internal::ConstantInitialized);
|
|
|
|
HloModuleConfigProto_Int64List(const HloModuleConfigProto_Int64List& from);
|
|
HloModuleConfigProto_Int64List(HloModuleConfigProto_Int64List&& from) noexcept
|
|
: HloModuleConfigProto_Int64List() {
|
|
*this = ::std::move(from);
|
|
}
|
|
|
|
inline HloModuleConfigProto_Int64List& operator=(const HloModuleConfigProto_Int64List& from) {
|
|
CopyFrom(from);
|
|
return *this;
|
|
}
|
|
inline HloModuleConfigProto_Int64List& operator=(HloModuleConfigProto_Int64List&& 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 HloModuleConfigProto_Int64List& default_instance() {
|
|
return *internal_default_instance();
|
|
}
|
|
static inline const HloModuleConfigProto_Int64List* internal_default_instance() {
|
|
return reinterpret_cast<const HloModuleConfigProto_Int64List*>(
|
|
&_HloModuleConfigProto_Int64List_default_instance_);
|
|
}
|
|
static constexpr int kIndexInFileMessages =
|
|
7;
|
|
|
|
friend void swap(HloModuleConfigProto_Int64List& a, HloModuleConfigProto_Int64List& b) {
|
|
a.Swap(&b);
|
|
}
|
|
inline void Swap(HloModuleConfigProto_Int64List* 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(HloModuleConfigProto_Int64List* other) {
|
|
if (other == this) return;
|
|
GOOGLE_DCHECK(GetOwningArena() == other->GetOwningArena());
|
|
InternalSwap(other);
|
|
}
|
|
|
|
// implements Message ----------------------------------------------
|
|
|
|
HloModuleConfigProto_Int64List* New(::PROTOBUF_NAMESPACE_ID::Arena* arena = nullptr) const final {
|
|
return CreateMaybeMessage<HloModuleConfigProto_Int64List>(arena);
|
|
}
|
|
using ::PROTOBUF_NAMESPACE_ID::Message::CopyFrom;
|
|
void CopyFrom(const HloModuleConfigProto_Int64List& from);
|
|
using ::PROTOBUF_NAMESPACE_ID::Message::MergeFrom;
|
|
void MergeFrom( const HloModuleConfigProto_Int64List& from) {
|
|
HloModuleConfigProto_Int64List::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(HloModuleConfigProto_Int64List* other);
|
|
|
|
private:
|
|
friend class ::PROTOBUF_NAMESPACE_ID::internal::AnyMetadata;
|
|
static ::PROTOBUF_NAMESPACE_ID::StringPiece FullMessageName() {
|
|
return "xla.HloModuleConfigProto.Int64List";
|
|
}
|
|
protected:
|
|
explicit HloModuleConfigProto_Int64List(::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 {
|
|
kValsFieldNumber = 1,
|
|
};
|
|
// repeated int64 vals = 1;
|
|
int vals_size() const;
|
|
private:
|
|
int _internal_vals_size() const;
|
|
public:
|
|
void clear_vals();
|
|
private:
|
|
int64_t _internal_vals(int index) const;
|
|
const ::PROTOBUF_NAMESPACE_ID::RepeatedField< int64_t >&
|
|
_internal_vals() const;
|
|
void _internal_add_vals(int64_t value);
|
|
::PROTOBUF_NAMESPACE_ID::RepeatedField< int64_t >*
|
|
_internal_mutable_vals();
|
|
public:
|
|
int64_t vals(int index) const;
|
|
void set_vals(int index, int64_t value);
|
|
void add_vals(int64_t value);
|
|
const ::PROTOBUF_NAMESPACE_ID::RepeatedField< int64_t >&
|
|
vals() const;
|
|
::PROTOBUF_NAMESPACE_ID::RepeatedField< int64_t >*
|
|
mutable_vals();
|
|
|
|
// @@protoc_insertion_point(class_scope:xla.HloModuleConfigProto.Int64List)
|
|
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 > vals_;
|
|
mutable std::atomic<int> _vals_cached_byte_size_;
|
|
mutable ::PROTOBUF_NAMESPACE_ID::internal::CachedSize _cached_size_;
|
|
};
|
|
union { Impl_ _impl_; };
|
|
friend struct ::TableStruct_xla_2fxla_2eproto;
|
|
};
|
|
// -------------------------------------------------------------------
|
|
|
|
class HloModuleConfigProto_Int64ListList final :
|
|
public ::PROTOBUF_NAMESPACE_ID::Message /* @@protoc_insertion_point(class_definition:xla.HloModuleConfigProto.Int64ListList) */ {
|
|
public:
|
|
inline HloModuleConfigProto_Int64ListList() : HloModuleConfigProto_Int64ListList(nullptr) {}
|
|
~HloModuleConfigProto_Int64ListList() override;
|
|
explicit PROTOBUF_CONSTEXPR HloModuleConfigProto_Int64ListList(::PROTOBUF_NAMESPACE_ID::internal::ConstantInitialized);
|
|
|
|
HloModuleConfigProto_Int64ListList(const HloModuleConfigProto_Int64ListList& from);
|
|
HloModuleConfigProto_Int64ListList(HloModuleConfigProto_Int64ListList&& from) noexcept
|
|
: HloModuleConfigProto_Int64ListList() {
|
|
*this = ::std::move(from);
|
|
}
|
|
|
|
inline HloModuleConfigProto_Int64ListList& operator=(const HloModuleConfigProto_Int64ListList& from) {
|
|
CopyFrom(from);
|
|
return *this;
|
|
}
|
|
inline HloModuleConfigProto_Int64ListList& operator=(HloModuleConfigProto_Int64ListList&& 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 HloModuleConfigProto_Int64ListList& default_instance() {
|
|
return *internal_default_instance();
|
|
}
|
|
static inline const HloModuleConfigProto_Int64ListList* internal_default_instance() {
|
|
return reinterpret_cast<const HloModuleConfigProto_Int64ListList*>(
|
|
&_HloModuleConfigProto_Int64ListList_default_instance_);
|
|
}
|
|
static constexpr int kIndexInFileMessages =
|
|
8;
|
|
|
|
friend void swap(HloModuleConfigProto_Int64ListList& a, HloModuleConfigProto_Int64ListList& b) {
|
|
a.Swap(&b);
|
|
}
|
|
inline void Swap(HloModuleConfigProto_Int64ListList* 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(HloModuleConfigProto_Int64ListList* other) {
|
|
if (other == this) return;
|
|
GOOGLE_DCHECK(GetOwningArena() == other->GetOwningArena());
|
|
InternalSwap(other);
|
|
}
|
|
|
|
// implements Message ----------------------------------------------
|
|
|
|
HloModuleConfigProto_Int64ListList* New(::PROTOBUF_NAMESPACE_ID::Arena* arena = nullptr) const final {
|
|
return CreateMaybeMessage<HloModuleConfigProto_Int64ListList>(arena);
|
|
}
|
|
using ::PROTOBUF_NAMESPACE_ID::Message::CopyFrom;
|
|
void CopyFrom(const HloModuleConfigProto_Int64ListList& from);
|
|
using ::PROTOBUF_NAMESPACE_ID::Message::MergeFrom;
|
|
void MergeFrom( const HloModuleConfigProto_Int64ListList& from) {
|
|
HloModuleConfigProto_Int64ListList::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(HloModuleConfigProto_Int64ListList* other);
|
|
|
|
private:
|
|
friend class ::PROTOBUF_NAMESPACE_ID::internal::AnyMetadata;
|
|
static ::PROTOBUF_NAMESPACE_ID::StringPiece FullMessageName() {
|
|
return "xla.HloModuleConfigProto.Int64ListList";
|
|
}
|
|
protected:
|
|
explicit HloModuleConfigProto_Int64ListList(::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 {
|
|
kListsFieldNumber = 1,
|
|
};
|
|
// repeated .xla.HloModuleConfigProto.Int64List lists = 1;
|
|
int lists_size() const;
|
|
private:
|
|
int _internal_lists_size() const;
|
|
public:
|
|
void clear_lists();
|
|
::xla::HloModuleConfigProto_Int64List* mutable_lists(int index);
|
|
::PROTOBUF_NAMESPACE_ID::RepeatedPtrField< ::xla::HloModuleConfigProto_Int64List >*
|
|
mutable_lists();
|
|
private:
|
|
const ::xla::HloModuleConfigProto_Int64List& _internal_lists(int index) const;
|
|
::xla::HloModuleConfigProto_Int64List* _internal_add_lists();
|
|
public:
|
|
const ::xla::HloModuleConfigProto_Int64List& lists(int index) const;
|
|
::xla::HloModuleConfigProto_Int64List* add_lists();
|
|
const ::PROTOBUF_NAMESPACE_ID::RepeatedPtrField< ::xla::HloModuleConfigProto_Int64List >&
|
|
lists() const;
|
|
|
|
// @@protoc_insertion_point(class_scope:xla.HloModuleConfigProto.Int64ListList)
|
|
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::HloModuleConfigProto_Int64List > lists_;
|
|
mutable ::PROTOBUF_NAMESPACE_ID::internal::CachedSize _cached_size_;
|
|
};
|
|
union { Impl_ _impl_; };
|
|
friend struct ::TableStruct_xla_2fxla_2eproto;
|
|
};
|
|
// -------------------------------------------------------------------
|
|
|
|
class HloModuleConfigProto_DotConfigEntry_DoNotUse : public ::PROTOBUF_NAMESPACE_ID::internal::MapEntry<HloModuleConfigProto_DotConfigEntry_DoNotUse,
|
|
std::string, ::xla::HloModuleConfigProto_Int64List,
|
|
::PROTOBUF_NAMESPACE_ID::internal::WireFormatLite::TYPE_STRING,
|
|
::PROTOBUF_NAMESPACE_ID::internal::WireFormatLite::TYPE_MESSAGE> {
|
|
public:
|
|
typedef ::PROTOBUF_NAMESPACE_ID::internal::MapEntry<HloModuleConfigProto_DotConfigEntry_DoNotUse,
|
|
std::string, ::xla::HloModuleConfigProto_Int64List,
|
|
::PROTOBUF_NAMESPACE_ID::internal::WireFormatLite::TYPE_STRING,
|
|
::PROTOBUF_NAMESPACE_ID::internal::WireFormatLite::TYPE_MESSAGE> SuperType;
|
|
HloModuleConfigProto_DotConfigEntry_DoNotUse();
|
|
explicit PROTOBUF_CONSTEXPR HloModuleConfigProto_DotConfigEntry_DoNotUse(
|
|
::PROTOBUF_NAMESPACE_ID::internal::ConstantInitialized);
|
|
explicit HloModuleConfigProto_DotConfigEntry_DoNotUse(::PROTOBUF_NAMESPACE_ID::Arena* arena);
|
|
void MergeFrom(const HloModuleConfigProto_DotConfigEntry_DoNotUse& other);
|
|
static const HloModuleConfigProto_DotConfigEntry_DoNotUse* internal_default_instance() { return reinterpret_cast<const HloModuleConfigProto_DotConfigEntry_DoNotUse*>(&_HloModuleConfigProto_DotConfigEntry_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.HloModuleConfigProto.DotConfigEntry.key");
|
|
}
|
|
static bool ValidateValue(void*) { return true; }
|
|
using ::PROTOBUF_NAMESPACE_ID::Message::MergeFrom;
|
|
::PROTOBUF_NAMESPACE_ID::Metadata GetMetadata() const final;
|
|
friend struct ::TableStruct_xla_2fxla_2eproto;
|
|
};
|
|
|
|
// -------------------------------------------------------------------
|
|
|
|
class HloModuleConfigProto_AnalysisAllowanceMapEntry_DoNotUse : public ::PROTOBUF_NAMESPACE_ID::internal::MapEntry<HloModuleConfigProto_AnalysisAllowanceMapEntry_DoNotUse,
|
|
std::string, int64_t,
|
|
::PROTOBUF_NAMESPACE_ID::internal::WireFormatLite::TYPE_STRING,
|
|
::PROTOBUF_NAMESPACE_ID::internal::WireFormatLite::TYPE_INT64> {
|
|
public:
|
|
typedef ::PROTOBUF_NAMESPACE_ID::internal::MapEntry<HloModuleConfigProto_AnalysisAllowanceMapEntry_DoNotUse,
|
|
std::string, int64_t,
|
|
::PROTOBUF_NAMESPACE_ID::internal::WireFormatLite::TYPE_STRING,
|
|
::PROTOBUF_NAMESPACE_ID::internal::WireFormatLite::TYPE_INT64> SuperType;
|
|
HloModuleConfigProto_AnalysisAllowanceMapEntry_DoNotUse();
|
|
explicit PROTOBUF_CONSTEXPR HloModuleConfigProto_AnalysisAllowanceMapEntry_DoNotUse(
|
|
::PROTOBUF_NAMESPACE_ID::internal::ConstantInitialized);
|
|
explicit HloModuleConfigProto_AnalysisAllowanceMapEntry_DoNotUse(::PROTOBUF_NAMESPACE_ID::Arena* arena);
|
|
void MergeFrom(const HloModuleConfigProto_AnalysisAllowanceMapEntry_DoNotUse& other);
|
|
static const HloModuleConfigProto_AnalysisAllowanceMapEntry_DoNotUse* internal_default_instance() { return reinterpret_cast<const HloModuleConfigProto_AnalysisAllowanceMapEntry_DoNotUse*>(&_HloModuleConfigProto_AnalysisAllowanceMapEntry_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.HloModuleConfigProto.AnalysisAllowanceMapEntry.key");
|
|
}
|
|
static bool ValidateValue(void*) { return true; }
|
|
using ::PROTOBUF_NAMESPACE_ID::Message::MergeFrom;
|
|
::PROTOBUF_NAMESPACE_ID::Metadata GetMetadata() const final;
|
|
friend struct ::TableStruct_xla_2fxla_2eproto;
|
|
};
|
|
|
|
// -------------------------------------------------------------------
|
|
|
|
class HloModuleConfigProto final :
|
|
public ::PROTOBUF_NAMESPACE_ID::Message /* @@protoc_insertion_point(class_definition:xla.HloModuleConfigProto) */ {
|
|
public:
|
|
inline HloModuleConfigProto() : HloModuleConfigProto(nullptr) {}
|
|
~HloModuleConfigProto() override;
|
|
explicit PROTOBUF_CONSTEXPR HloModuleConfigProto(::PROTOBUF_NAMESPACE_ID::internal::ConstantInitialized);
|
|
|
|
HloModuleConfigProto(const HloModuleConfigProto& from);
|
|
HloModuleConfigProto(HloModuleConfigProto&& from) noexcept
|
|
: HloModuleConfigProto() {
|
|
*this = ::std::move(from);
|
|
}
|
|
|
|
inline HloModuleConfigProto& operator=(const HloModuleConfigProto& from) {
|
|
CopyFrom(from);
|
|
return *this;
|
|
}
|
|
inline HloModuleConfigProto& operator=(HloModuleConfigProto&& 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 HloModuleConfigProto& default_instance() {
|
|
return *internal_default_instance();
|
|
}
|
|
static inline const HloModuleConfigProto* internal_default_instance() {
|
|
return reinterpret_cast<const HloModuleConfigProto*>(
|
|
&_HloModuleConfigProto_default_instance_);
|
|
}
|
|
static constexpr int kIndexInFileMessages =
|
|
11;
|
|
|
|
friend void swap(HloModuleConfigProto& a, HloModuleConfigProto& b) {
|
|
a.Swap(&b);
|
|
}
|
|
inline void Swap(HloModuleConfigProto* 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(HloModuleConfigProto* other) {
|
|
if (other == this) return;
|
|
GOOGLE_DCHECK(GetOwningArena() == other->GetOwningArena());
|
|
InternalSwap(other);
|
|
}
|
|
|
|
// implements Message ----------------------------------------------
|
|
|
|
HloModuleConfigProto* New(::PROTOBUF_NAMESPACE_ID::Arena* arena = nullptr) const final {
|
|
return CreateMaybeMessage<HloModuleConfigProto>(arena);
|
|
}
|
|
using ::PROTOBUF_NAMESPACE_ID::Message::CopyFrom;
|
|
void CopyFrom(const HloModuleConfigProto& from);
|
|
using ::PROTOBUF_NAMESPACE_ID::Message::MergeFrom;
|
|
void MergeFrom( const HloModuleConfigProto& from) {
|
|
HloModuleConfigProto::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(HloModuleConfigProto* other);
|
|
|
|
private:
|
|
friend class ::PROTOBUF_NAMESPACE_ID::internal::AnyMetadata;
|
|
static ::PROTOBUF_NAMESPACE_ID::StringPiece FullMessageName() {
|
|
return "xla.HloModuleConfigProto";
|
|
}
|
|
protected:
|
|
explicit HloModuleConfigProto(::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 ----------------------------------------------------
|
|
|
|
typedef HloModuleConfigProto_BoolList BoolList;
|
|
typedef HloModuleConfigProto_Int64List Int64List;
|
|
typedef HloModuleConfigProto_Int64ListList Int64ListList;
|
|
|
|
typedef HloModuleConfigProto_FusionConfigCollection FusionConfigCollection;
|
|
static constexpr FusionConfigCollection OFF =
|
|
HloModuleConfigProto_FusionConfigCollection_OFF;
|
|
static constexpr FusionConfigCollection PER_EDGE =
|
|
HloModuleConfigProto_FusionConfigCollection_PER_EDGE;
|
|
static constexpr FusionConfigCollection PER_NODE =
|
|
HloModuleConfigProto_FusionConfigCollection_PER_NODE;
|
|
static inline bool FusionConfigCollection_IsValid(int value) {
|
|
return HloModuleConfigProto_FusionConfigCollection_IsValid(value);
|
|
}
|
|
static constexpr FusionConfigCollection FusionConfigCollection_MIN =
|
|
HloModuleConfigProto_FusionConfigCollection_FusionConfigCollection_MIN;
|
|
static constexpr FusionConfigCollection FusionConfigCollection_MAX =
|
|
HloModuleConfigProto_FusionConfigCollection_FusionConfigCollection_MAX;
|
|
static constexpr int FusionConfigCollection_ARRAYSIZE =
|
|
HloModuleConfigProto_FusionConfigCollection_FusionConfigCollection_ARRAYSIZE;
|
|
static inline const ::PROTOBUF_NAMESPACE_ID::EnumDescriptor*
|
|
FusionConfigCollection_descriptor() {
|
|
return HloModuleConfigProto_FusionConfigCollection_descriptor();
|
|
}
|
|
template<typename T>
|
|
static inline const std::string& FusionConfigCollection_Name(T enum_t_value) {
|
|
static_assert(::std::is_same<T, FusionConfigCollection>::value ||
|
|
::std::is_integral<T>::value,
|
|
"Incorrect type passed to function FusionConfigCollection_Name.");
|
|
return HloModuleConfigProto_FusionConfigCollection_Name(enum_t_value);
|
|
}
|
|
static inline bool FusionConfigCollection_Parse(::PROTOBUF_NAMESPACE_ID::ConstStringParam name,
|
|
FusionConfigCollection* value) {
|
|
return HloModuleConfigProto_FusionConfigCollection_Parse(name, value);
|
|
}
|
|
|
|
// accessors -------------------------------------------------------
|
|
|
|
enum : int {
|
|
kParamRequiresBroadcastViaCollectivesFieldNumber = 6,
|
|
kAutoSpmdPartitioningMeshShapeFieldNumber = 9,
|
|
kAutoSpmdPartitioningMeshIdsFieldNumber = 10,
|
|
kShardableValueUpdatePairsFieldNumber = 16,
|
|
kFusionConfigFieldNumber = 20,
|
|
kDotConfigFieldNumber = 21,
|
|
kLayoutConfigFieldNumber = 22,
|
|
kMemorySpaceAssignmentConfigFieldNumber = 23,
|
|
kPhaseOrderingConfigFieldNumber = 24,
|
|
kAllowSpmdShardingPropagationToOutputFieldNumber = 27,
|
|
kAnalysisAllowanceMapFieldNumber = 28,
|
|
kDeviceTypeFieldNumber = 13,
|
|
kFdoProfileFieldNumber = 31,
|
|
kEntryComputationLayoutFieldNumber = 1,
|
|
kDebugOptionsFieldNumber = 14,
|
|
kStaticDeviceAssignmentFieldNumber = 15,
|
|
kSeedFieldNumber = 2,
|
|
kReplicaCountFieldNumber = 4,
|
|
kNumPartitionsFieldNumber = 5,
|
|
kLaunchIdFieldNumber = 3,
|
|
kUseSpmdPartitioningFieldNumber = 7,
|
|
kUseAutoSpmdPartitioningFieldNumber = 8,
|
|
kDeduplicateHloFieldNumber = 11,
|
|
kAllowSeparateShardingProgramsFieldNumber = 30,
|
|
kIntraOpParallelismThreadsFieldNumber = 12,
|
|
kAliasPassthroughParamsFieldNumber = 17,
|
|
kContentAwareComputationSortingFieldNumber = 18,
|
|
kFusionConfigCollectionFieldNumber = 19,
|
|
kPhaseIndexFieldNumber = 25,
|
|
kMatrixUnitOperandPrecisionFieldNumber = 29,
|
|
kDeviceMemorySizeFieldNumber = 32,
|
|
};
|
|
// repeated bool param_requires_broadcast_via_collectives = 6;
|
|
int param_requires_broadcast_via_collectives_size() const;
|
|
private:
|
|
int _internal_param_requires_broadcast_via_collectives_size() const;
|
|
public:
|
|
void clear_param_requires_broadcast_via_collectives();
|
|
private:
|
|
bool _internal_param_requires_broadcast_via_collectives(int index) const;
|
|
const ::PROTOBUF_NAMESPACE_ID::RepeatedField< bool >&
|
|
_internal_param_requires_broadcast_via_collectives() const;
|
|
void _internal_add_param_requires_broadcast_via_collectives(bool value);
|
|
::PROTOBUF_NAMESPACE_ID::RepeatedField< bool >*
|
|
_internal_mutable_param_requires_broadcast_via_collectives();
|
|
public:
|
|
bool param_requires_broadcast_via_collectives(int index) const;
|
|
void set_param_requires_broadcast_via_collectives(int index, bool value);
|
|
void add_param_requires_broadcast_via_collectives(bool value);
|
|
const ::PROTOBUF_NAMESPACE_ID::RepeatedField< bool >&
|
|
param_requires_broadcast_via_collectives() const;
|
|
::PROTOBUF_NAMESPACE_ID::RepeatedField< bool >*
|
|
mutable_param_requires_broadcast_via_collectives();
|
|
|
|
// repeated int64 auto_spmd_partitioning_mesh_shape = 9;
|
|
int auto_spmd_partitioning_mesh_shape_size() const;
|
|
private:
|
|
int _internal_auto_spmd_partitioning_mesh_shape_size() const;
|
|
public:
|
|
void clear_auto_spmd_partitioning_mesh_shape();
|
|
private:
|
|
int64_t _internal_auto_spmd_partitioning_mesh_shape(int index) const;
|
|
const ::PROTOBUF_NAMESPACE_ID::RepeatedField< int64_t >&
|
|
_internal_auto_spmd_partitioning_mesh_shape() const;
|
|
void _internal_add_auto_spmd_partitioning_mesh_shape(int64_t value);
|
|
::PROTOBUF_NAMESPACE_ID::RepeatedField< int64_t >*
|
|
_internal_mutable_auto_spmd_partitioning_mesh_shape();
|
|
public:
|
|
int64_t auto_spmd_partitioning_mesh_shape(int index) const;
|
|
void set_auto_spmd_partitioning_mesh_shape(int index, int64_t value);
|
|
void add_auto_spmd_partitioning_mesh_shape(int64_t value);
|
|
const ::PROTOBUF_NAMESPACE_ID::RepeatedField< int64_t >&
|
|
auto_spmd_partitioning_mesh_shape() const;
|
|
::PROTOBUF_NAMESPACE_ID::RepeatedField< int64_t >*
|
|
mutable_auto_spmd_partitioning_mesh_shape();
|
|
|
|
// repeated int64 auto_spmd_partitioning_mesh_ids = 10;
|
|
int auto_spmd_partitioning_mesh_ids_size() const;
|
|
private:
|
|
int _internal_auto_spmd_partitioning_mesh_ids_size() const;
|
|
public:
|
|
void clear_auto_spmd_partitioning_mesh_ids();
|
|
private:
|
|
int64_t _internal_auto_spmd_partitioning_mesh_ids(int index) const;
|
|
const ::PROTOBUF_NAMESPACE_ID::RepeatedField< int64_t >&
|
|
_internal_auto_spmd_partitioning_mesh_ids() const;
|
|
void _internal_add_auto_spmd_partitioning_mesh_ids(int64_t value);
|
|
::PROTOBUF_NAMESPACE_ID::RepeatedField< int64_t >*
|
|
_internal_mutable_auto_spmd_partitioning_mesh_ids();
|
|
public:
|
|
int64_t auto_spmd_partitioning_mesh_ids(int index) const;
|
|
void set_auto_spmd_partitioning_mesh_ids(int index, int64_t value);
|
|
void add_auto_spmd_partitioning_mesh_ids(int64_t value);
|
|
const ::PROTOBUF_NAMESPACE_ID::RepeatedField< int64_t >&
|
|
auto_spmd_partitioning_mesh_ids() const;
|
|
::PROTOBUF_NAMESPACE_ID::RepeatedField< int64_t >*
|
|
mutable_auto_spmd_partitioning_mesh_ids();
|
|
|
|
// repeated .xla.ShardableValueUpdatePairProto shardable_value_update_pairs = 16;
|
|
int shardable_value_update_pairs_size() const;
|
|
private:
|
|
int _internal_shardable_value_update_pairs_size() const;
|
|
public:
|
|
void clear_shardable_value_update_pairs();
|
|
::xla::ShardableValueUpdatePairProto* mutable_shardable_value_update_pairs(int index);
|
|
::PROTOBUF_NAMESPACE_ID::RepeatedPtrField< ::xla::ShardableValueUpdatePairProto >*
|
|
mutable_shardable_value_update_pairs();
|
|
private:
|
|
const ::xla::ShardableValueUpdatePairProto& _internal_shardable_value_update_pairs(int index) const;
|
|
::xla::ShardableValueUpdatePairProto* _internal_add_shardable_value_update_pairs();
|
|
public:
|
|
const ::xla::ShardableValueUpdatePairProto& shardable_value_update_pairs(int index) const;
|
|
::xla::ShardableValueUpdatePairProto* add_shardable_value_update_pairs();
|
|
const ::PROTOBUF_NAMESPACE_ID::RepeatedPtrField< ::xla::ShardableValueUpdatePairProto >&
|
|
shardable_value_update_pairs() const;
|
|
|
|
// repeated .xla.HloModuleConfigProto.BoolList fusion_config = 20;
|
|
int fusion_config_size() const;
|
|
private:
|
|
int _internal_fusion_config_size() const;
|
|
public:
|
|
void clear_fusion_config();
|
|
::xla::HloModuleConfigProto_BoolList* mutable_fusion_config(int index);
|
|
::PROTOBUF_NAMESPACE_ID::RepeatedPtrField< ::xla::HloModuleConfigProto_BoolList >*
|
|
mutable_fusion_config();
|
|
private:
|
|
const ::xla::HloModuleConfigProto_BoolList& _internal_fusion_config(int index) const;
|
|
::xla::HloModuleConfigProto_BoolList* _internal_add_fusion_config();
|
|
public:
|
|
const ::xla::HloModuleConfigProto_BoolList& fusion_config(int index) const;
|
|
::xla::HloModuleConfigProto_BoolList* add_fusion_config();
|
|
const ::PROTOBUF_NAMESPACE_ID::RepeatedPtrField< ::xla::HloModuleConfigProto_BoolList >&
|
|
fusion_config() const;
|
|
|
|
// map<string, .xla.HloModuleConfigProto.Int64List> dot_config = 21;
|
|
int dot_config_size() const;
|
|
private:
|
|
int _internal_dot_config_size() const;
|
|
public:
|
|
void clear_dot_config();
|
|
private:
|
|
const ::PROTOBUF_NAMESPACE_ID::Map< std::string, ::xla::HloModuleConfigProto_Int64List >&
|
|
_internal_dot_config() const;
|
|
::PROTOBUF_NAMESPACE_ID::Map< std::string, ::xla::HloModuleConfigProto_Int64List >*
|
|
_internal_mutable_dot_config();
|
|
public:
|
|
const ::PROTOBUF_NAMESPACE_ID::Map< std::string, ::xla::HloModuleConfigProto_Int64List >&
|
|
dot_config() const;
|
|
::PROTOBUF_NAMESPACE_ID::Map< std::string, ::xla::HloModuleConfigProto_Int64List >*
|
|
mutable_dot_config();
|
|
|
|
// repeated .xla.HloModuleConfigProto.Int64ListList layout_config = 22;
|
|
int layout_config_size() const;
|
|
private:
|
|
int _internal_layout_config_size() const;
|
|
public:
|
|
void clear_layout_config();
|
|
::xla::HloModuleConfigProto_Int64ListList* mutable_layout_config(int index);
|
|
::PROTOBUF_NAMESPACE_ID::RepeatedPtrField< ::xla::HloModuleConfigProto_Int64ListList >*
|
|
mutable_layout_config();
|
|
private:
|
|
const ::xla::HloModuleConfigProto_Int64ListList& _internal_layout_config(int index) const;
|
|
::xla::HloModuleConfigProto_Int64ListList* _internal_add_layout_config();
|
|
public:
|
|
const ::xla::HloModuleConfigProto_Int64ListList& layout_config(int index) const;
|
|
::xla::HloModuleConfigProto_Int64ListList* add_layout_config();
|
|
const ::PROTOBUF_NAMESPACE_ID::RepeatedPtrField< ::xla::HloModuleConfigProto_Int64ListList >&
|
|
layout_config() const;
|
|
|
|
// repeated uint64 memory_space_assignment_config = 23;
|
|
int memory_space_assignment_config_size() const;
|
|
private:
|
|
int _internal_memory_space_assignment_config_size() const;
|
|
public:
|
|
void clear_memory_space_assignment_config();
|
|
private:
|
|
uint64_t _internal_memory_space_assignment_config(int index) const;
|
|
const ::PROTOBUF_NAMESPACE_ID::RepeatedField< uint64_t >&
|
|
_internal_memory_space_assignment_config() const;
|
|
void _internal_add_memory_space_assignment_config(uint64_t value);
|
|
::PROTOBUF_NAMESPACE_ID::RepeatedField< uint64_t >*
|
|
_internal_mutable_memory_space_assignment_config();
|
|
public:
|
|
uint64_t memory_space_assignment_config(int index) const;
|
|
void set_memory_space_assignment_config(int index, uint64_t value);
|
|
void add_memory_space_assignment_config(uint64_t value);
|
|
const ::PROTOBUF_NAMESPACE_ID::RepeatedField< uint64_t >&
|
|
memory_space_assignment_config() const;
|
|
::PROTOBUF_NAMESPACE_ID::RepeatedField< uint64_t >*
|
|
mutable_memory_space_assignment_config();
|
|
|
|
// repeated .xla.HloModuleConfigProto.BoolList phase_ordering_config = 24;
|
|
int phase_ordering_config_size() const;
|
|
private:
|
|
int _internal_phase_ordering_config_size() const;
|
|
public:
|
|
void clear_phase_ordering_config();
|
|
::xla::HloModuleConfigProto_BoolList* mutable_phase_ordering_config(int index);
|
|
::PROTOBUF_NAMESPACE_ID::RepeatedPtrField< ::xla::HloModuleConfigProto_BoolList >*
|
|
mutable_phase_ordering_config();
|
|
private:
|
|
const ::xla::HloModuleConfigProto_BoolList& _internal_phase_ordering_config(int index) const;
|
|
::xla::HloModuleConfigProto_BoolList* _internal_add_phase_ordering_config();
|
|
public:
|
|
const ::xla::HloModuleConfigProto_BoolList& phase_ordering_config(int index) const;
|
|
::xla::HloModuleConfigProto_BoolList* add_phase_ordering_config();
|
|
const ::PROTOBUF_NAMESPACE_ID::RepeatedPtrField< ::xla::HloModuleConfigProto_BoolList >&
|
|
phase_ordering_config() const;
|
|
|
|
// repeated bool allow_spmd_sharding_propagation_to_output = 27;
|
|
int allow_spmd_sharding_propagation_to_output_size() const;
|
|
private:
|
|
int _internal_allow_spmd_sharding_propagation_to_output_size() const;
|
|
public:
|
|
void clear_allow_spmd_sharding_propagation_to_output();
|
|
private:
|
|
bool _internal_allow_spmd_sharding_propagation_to_output(int index) const;
|
|
const ::PROTOBUF_NAMESPACE_ID::RepeatedField< bool >&
|
|
_internal_allow_spmd_sharding_propagation_to_output() const;
|
|
void _internal_add_allow_spmd_sharding_propagation_to_output(bool value);
|
|
::PROTOBUF_NAMESPACE_ID::RepeatedField< bool >*
|
|
_internal_mutable_allow_spmd_sharding_propagation_to_output();
|
|
public:
|
|
bool allow_spmd_sharding_propagation_to_output(int index) const;
|
|
void set_allow_spmd_sharding_propagation_to_output(int index, bool value);
|
|
void add_allow_spmd_sharding_propagation_to_output(bool value);
|
|
const ::PROTOBUF_NAMESPACE_ID::RepeatedField< bool >&
|
|
allow_spmd_sharding_propagation_to_output() const;
|
|
::PROTOBUF_NAMESPACE_ID::RepeatedField< bool >*
|
|
mutable_allow_spmd_sharding_propagation_to_output();
|
|
|
|
// map<string, int64> analysis_allowance_map = 28;
|
|
int analysis_allowance_map_size() const;
|
|
private:
|
|
int _internal_analysis_allowance_map_size() const;
|
|
public:
|
|
void clear_analysis_allowance_map();
|
|
private:
|
|
const ::PROTOBUF_NAMESPACE_ID::Map< std::string, int64_t >&
|
|
_internal_analysis_allowance_map() const;
|
|
::PROTOBUF_NAMESPACE_ID::Map< std::string, int64_t >*
|
|
_internal_mutable_analysis_allowance_map();
|
|
public:
|
|
const ::PROTOBUF_NAMESPACE_ID::Map< std::string, int64_t >&
|
|
analysis_allowance_map() const;
|
|
::PROTOBUF_NAMESPACE_ID::Map< std::string, int64_t >*
|
|
mutable_analysis_allowance_map();
|
|
|
|
// string device_type = 13;
|
|
void clear_device_type();
|
|
const std::string& device_type() const;
|
|
template <typename ArgT0 = const std::string&, typename... ArgT>
|
|
void set_device_type(ArgT0&& arg0, ArgT... args);
|
|
std::string* mutable_device_type();
|
|
PROTOBUF_NODISCARD std::string* release_device_type();
|
|
void set_allocated_device_type(std::string* device_type);
|
|
private:
|
|
const std::string& _internal_device_type() const;
|
|
inline PROTOBUF_ALWAYS_INLINE void _internal_set_device_type(const std::string& value);
|
|
std::string* _internal_mutable_device_type();
|
|
public:
|
|
|
|
// bytes fdo_profile = 31;
|
|
void clear_fdo_profile();
|
|
const std::string& fdo_profile() const;
|
|
template <typename ArgT0 = const std::string&, typename... ArgT>
|
|
void set_fdo_profile(ArgT0&& arg0, ArgT... args);
|
|
std::string* mutable_fdo_profile();
|
|
PROTOBUF_NODISCARD std::string* release_fdo_profile();
|
|
void set_allocated_fdo_profile(std::string* fdo_profile);
|
|
private:
|
|
const std::string& _internal_fdo_profile() const;
|
|
inline PROTOBUF_ALWAYS_INLINE void _internal_set_fdo_profile(const std::string& value);
|
|
std::string* _internal_mutable_fdo_profile();
|
|
public:
|
|
|
|
// .xla.ProgramShapeProto entry_computation_layout = 1;
|
|
bool has_entry_computation_layout() const;
|
|
private:
|
|
bool _internal_has_entry_computation_layout() const;
|
|
public:
|
|
void clear_entry_computation_layout();
|
|
const ::xla::ProgramShapeProto& entry_computation_layout() const;
|
|
PROTOBUF_NODISCARD ::xla::ProgramShapeProto* release_entry_computation_layout();
|
|
::xla::ProgramShapeProto* mutable_entry_computation_layout();
|
|
void set_allocated_entry_computation_layout(::xla::ProgramShapeProto* entry_computation_layout);
|
|
private:
|
|
const ::xla::ProgramShapeProto& _internal_entry_computation_layout() const;
|
|
::xla::ProgramShapeProto* _internal_mutable_entry_computation_layout();
|
|
public:
|
|
void unsafe_arena_set_allocated_entry_computation_layout(
|
|
::xla::ProgramShapeProto* entry_computation_layout);
|
|
::xla::ProgramShapeProto* unsafe_arena_release_entry_computation_layout();
|
|
|
|
// .xla.DebugOptions debug_options = 14;
|
|
bool has_debug_options() const;
|
|
private:
|
|
bool _internal_has_debug_options() const;
|
|
public:
|
|
void clear_debug_options();
|
|
const ::xla::DebugOptions& debug_options() const;
|
|
PROTOBUF_NODISCARD ::xla::DebugOptions* release_debug_options();
|
|
::xla::DebugOptions* mutable_debug_options();
|
|
void set_allocated_debug_options(::xla::DebugOptions* debug_options);
|
|
private:
|
|
const ::xla::DebugOptions& _internal_debug_options() const;
|
|
::xla::DebugOptions* _internal_mutable_debug_options();
|
|
public:
|
|
void unsafe_arena_set_allocated_debug_options(
|
|
::xla::DebugOptions* debug_options);
|
|
::xla::DebugOptions* unsafe_arena_release_debug_options();
|
|
|
|
// .xla.DeviceAssignmentProto static_device_assignment = 15;
|
|
bool has_static_device_assignment() const;
|
|
private:
|
|
bool _internal_has_static_device_assignment() const;
|
|
public:
|
|
void clear_static_device_assignment();
|
|
const ::xla::DeviceAssignmentProto& static_device_assignment() const;
|
|
PROTOBUF_NODISCARD ::xla::DeviceAssignmentProto* release_static_device_assignment();
|
|
::xla::DeviceAssignmentProto* mutable_static_device_assignment();
|
|
void set_allocated_static_device_assignment(::xla::DeviceAssignmentProto* static_device_assignment);
|
|
private:
|
|
const ::xla::DeviceAssignmentProto& _internal_static_device_assignment() const;
|
|
::xla::DeviceAssignmentProto* _internal_mutable_static_device_assignment();
|
|
public:
|
|
void unsafe_arena_set_allocated_static_device_assignment(
|
|
::xla::DeviceAssignmentProto* static_device_assignment);
|
|
::xla::DeviceAssignmentProto* unsafe_arena_release_static_device_assignment();
|
|
|
|
// uint64 seed = 2;
|
|
void clear_seed();
|
|
uint64_t seed() const;
|
|
void set_seed(uint64_t value);
|
|
private:
|
|
uint64_t _internal_seed() const;
|
|
void _internal_set_seed(uint64_t value);
|
|
public:
|
|
|
|
// int64 replica_count = 4;
|
|
void clear_replica_count();
|
|
int64_t replica_count() const;
|
|
void set_replica_count(int64_t value);
|
|
private:
|
|
int64_t _internal_replica_count() const;
|
|
void _internal_set_replica_count(int64_t value);
|
|
public:
|
|
|
|
// int64 num_partitions = 5;
|
|
void clear_num_partitions();
|
|
int64_t num_partitions() const;
|
|
void set_num_partitions(int64_t value);
|
|
private:
|
|
int64_t _internal_num_partitions() const;
|
|
void _internal_set_num_partitions(int64_t value);
|
|
public:
|
|
|
|
// int32 launch_id = 3;
|
|
void clear_launch_id();
|
|
int32_t launch_id() const;
|
|
void set_launch_id(int32_t value);
|
|
private:
|
|
int32_t _internal_launch_id() const;
|
|
void _internal_set_launch_id(int32_t value);
|
|
public:
|
|
|
|
// bool use_spmd_partitioning = 7;
|
|
void clear_use_spmd_partitioning();
|
|
bool use_spmd_partitioning() const;
|
|
void set_use_spmd_partitioning(bool value);
|
|
private:
|
|
bool _internal_use_spmd_partitioning() const;
|
|
void _internal_set_use_spmd_partitioning(bool value);
|
|
public:
|
|
|
|
// bool use_auto_spmd_partitioning = 8;
|
|
void clear_use_auto_spmd_partitioning();
|
|
bool use_auto_spmd_partitioning() const;
|
|
void set_use_auto_spmd_partitioning(bool value);
|
|
private:
|
|
bool _internal_use_auto_spmd_partitioning() const;
|
|
void _internal_set_use_auto_spmd_partitioning(bool value);
|
|
public:
|
|
|
|
// bool deduplicate_hlo = 11;
|
|
void clear_deduplicate_hlo();
|
|
bool deduplicate_hlo() const;
|
|
void set_deduplicate_hlo(bool value);
|
|
private:
|
|
bool _internal_deduplicate_hlo() const;
|
|
void _internal_set_deduplicate_hlo(bool value);
|
|
public:
|
|
|
|
// bool allow_separate_sharding_programs = 30;
|
|
void clear_allow_separate_sharding_programs();
|
|
bool allow_separate_sharding_programs() const;
|
|
void set_allow_separate_sharding_programs(bool value);
|
|
private:
|
|
bool _internal_allow_separate_sharding_programs() const;
|
|
void _internal_set_allow_separate_sharding_programs(bool value);
|
|
public:
|
|
|
|
// int64 intra_op_parallelism_threads = 12;
|
|
void clear_intra_op_parallelism_threads();
|
|
int64_t intra_op_parallelism_threads() const;
|
|
void set_intra_op_parallelism_threads(int64_t value);
|
|
private:
|
|
int64_t _internal_intra_op_parallelism_threads() const;
|
|
void _internal_set_intra_op_parallelism_threads(int64_t value);
|
|
public:
|
|
|
|
// bool alias_passthrough_params = 17;
|
|
void clear_alias_passthrough_params();
|
|
bool alias_passthrough_params() const;
|
|
void set_alias_passthrough_params(bool value);
|
|
private:
|
|
bool _internal_alias_passthrough_params() const;
|
|
void _internal_set_alias_passthrough_params(bool value);
|
|
public:
|
|
|
|
// bool content_aware_computation_sorting = 18;
|
|
void clear_content_aware_computation_sorting();
|
|
bool content_aware_computation_sorting() const;
|
|
void set_content_aware_computation_sorting(bool value);
|
|
private:
|
|
bool _internal_content_aware_computation_sorting() const;
|
|
void _internal_set_content_aware_computation_sorting(bool value);
|
|
public:
|
|
|
|
// .xla.HloModuleConfigProto.FusionConfigCollection fusion_config_collection = 19;
|
|
void clear_fusion_config_collection();
|
|
::xla::HloModuleConfigProto_FusionConfigCollection fusion_config_collection() const;
|
|
void set_fusion_config_collection(::xla::HloModuleConfigProto_FusionConfigCollection value);
|
|
private:
|
|
::xla::HloModuleConfigProto_FusionConfigCollection _internal_fusion_config_collection() const;
|
|
void _internal_set_fusion_config_collection(::xla::HloModuleConfigProto_FusionConfigCollection value);
|
|
public:
|
|
|
|
// int32 phase_index = 25;
|
|
void clear_phase_index();
|
|
int32_t phase_index() const;
|
|
void set_phase_index(int32_t value);
|
|
private:
|
|
int32_t _internal_phase_index() const;
|
|
void _internal_set_phase_index(int32_t value);
|
|
public:
|
|
|
|
// .xla.PrecisionConfig.Precision matrix_unit_operand_precision = 29;
|
|
void clear_matrix_unit_operand_precision();
|
|
::xla::PrecisionConfig_Precision matrix_unit_operand_precision() const;
|
|
void set_matrix_unit_operand_precision(::xla::PrecisionConfig_Precision value);
|
|
private:
|
|
::xla::PrecisionConfig_Precision _internal_matrix_unit_operand_precision() const;
|
|
void _internal_set_matrix_unit_operand_precision(::xla::PrecisionConfig_Precision value);
|
|
public:
|
|
|
|
// int64 device_memory_size = 32;
|
|
void clear_device_memory_size();
|
|
int64_t device_memory_size() const;
|
|
void set_device_memory_size(int64_t value);
|
|
private:
|
|
int64_t _internal_device_memory_size() const;
|
|
void _internal_set_device_memory_size(int64_t value);
|
|
public:
|
|
|
|
// @@protoc_insertion_point(class_scope:xla.HloModuleConfigProto)
|
|
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 > param_requires_broadcast_via_collectives_;
|
|
::PROTOBUF_NAMESPACE_ID::RepeatedField< int64_t > auto_spmd_partitioning_mesh_shape_;
|
|
mutable std::atomic<int> _auto_spmd_partitioning_mesh_shape_cached_byte_size_;
|
|
::PROTOBUF_NAMESPACE_ID::RepeatedField< int64_t > auto_spmd_partitioning_mesh_ids_;
|
|
mutable std::atomic<int> _auto_spmd_partitioning_mesh_ids_cached_byte_size_;
|
|
::PROTOBUF_NAMESPACE_ID::RepeatedPtrField< ::xla::ShardableValueUpdatePairProto > shardable_value_update_pairs_;
|
|
::PROTOBUF_NAMESPACE_ID::RepeatedPtrField< ::xla::HloModuleConfigProto_BoolList > fusion_config_;
|
|
::PROTOBUF_NAMESPACE_ID::internal::MapField<
|
|
HloModuleConfigProto_DotConfigEntry_DoNotUse,
|
|
std::string, ::xla::HloModuleConfigProto_Int64List,
|
|
::PROTOBUF_NAMESPACE_ID::internal::WireFormatLite::TYPE_STRING,
|
|
::PROTOBUF_NAMESPACE_ID::internal::WireFormatLite::TYPE_MESSAGE> dot_config_;
|
|
::PROTOBUF_NAMESPACE_ID::RepeatedPtrField< ::xla::HloModuleConfigProto_Int64ListList > layout_config_;
|
|
::PROTOBUF_NAMESPACE_ID::RepeatedField< uint64_t > memory_space_assignment_config_;
|
|
mutable std::atomic<int> _memory_space_assignment_config_cached_byte_size_;
|
|
::PROTOBUF_NAMESPACE_ID::RepeatedPtrField< ::xla::HloModuleConfigProto_BoolList > phase_ordering_config_;
|
|
::PROTOBUF_NAMESPACE_ID::RepeatedField< bool > allow_spmd_sharding_propagation_to_output_;
|
|
::PROTOBUF_NAMESPACE_ID::internal::MapField<
|
|
HloModuleConfigProto_AnalysisAllowanceMapEntry_DoNotUse,
|
|
std::string, int64_t,
|
|
::PROTOBUF_NAMESPACE_ID::internal::WireFormatLite::TYPE_STRING,
|
|
::PROTOBUF_NAMESPACE_ID::internal::WireFormatLite::TYPE_INT64> analysis_allowance_map_;
|
|
::PROTOBUF_NAMESPACE_ID::internal::ArenaStringPtr device_type_;
|
|
::PROTOBUF_NAMESPACE_ID::internal::ArenaStringPtr fdo_profile_;
|
|
::xla::ProgramShapeProto* entry_computation_layout_;
|
|
::xla::DebugOptions* debug_options_;
|
|
::xla::DeviceAssignmentProto* static_device_assignment_;
|
|
uint64_t seed_;
|
|
int64_t replica_count_;
|
|
int64_t num_partitions_;
|
|
int32_t launch_id_;
|
|
bool use_spmd_partitioning_;
|
|
bool use_auto_spmd_partitioning_;
|
|
bool deduplicate_hlo_;
|
|
bool allow_separate_sharding_programs_;
|
|
int64_t intra_op_parallelism_threads_;
|
|
bool alias_passthrough_params_;
|
|
bool content_aware_computation_sorting_;
|
|
int fusion_config_collection_;
|
|
int32_t phase_index_;
|
|
int matrix_unit_operand_precision_;
|
|
int64_t device_memory_size_;
|
|
mutable ::PROTOBUF_NAMESPACE_ID::internal::CachedSize _cached_size_;
|
|
};
|
|
union { Impl_ _impl_; };
|
|
friend struct ::TableStruct_xla_2fxla_2eproto;
|
|
};
|
|
// -------------------------------------------------------------------
|
|
|
|
class HloModuleProtoWithConfig final :
|
|
public ::PROTOBUF_NAMESPACE_ID::Message /* @@protoc_insertion_point(class_definition:xla.HloModuleProtoWithConfig) */ {
|
|
public:
|
|
inline HloModuleProtoWithConfig() : HloModuleProtoWithConfig(nullptr) {}
|
|
~HloModuleProtoWithConfig() override;
|
|
explicit PROTOBUF_CONSTEXPR HloModuleProtoWithConfig(::PROTOBUF_NAMESPACE_ID::internal::ConstantInitialized);
|
|
|
|
HloModuleProtoWithConfig(const HloModuleProtoWithConfig& from);
|
|
HloModuleProtoWithConfig(HloModuleProtoWithConfig&& from) noexcept
|
|
: HloModuleProtoWithConfig() {
|
|
*this = ::std::move(from);
|
|
}
|
|
|
|
inline HloModuleProtoWithConfig& operator=(const HloModuleProtoWithConfig& from) {
|
|
CopyFrom(from);
|
|
return *this;
|
|
}
|
|
inline HloModuleProtoWithConfig& operator=(HloModuleProtoWithConfig&& 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 HloModuleProtoWithConfig& default_instance() {
|
|
return *internal_default_instance();
|
|
}
|
|
static inline const HloModuleProtoWithConfig* internal_default_instance() {
|
|
return reinterpret_cast<const HloModuleProtoWithConfig*>(
|
|
&_HloModuleProtoWithConfig_default_instance_);
|
|
}
|
|
static constexpr int kIndexInFileMessages =
|
|
12;
|
|
|
|
friend void swap(HloModuleProtoWithConfig& a, HloModuleProtoWithConfig& b) {
|
|
a.Swap(&b);
|
|
}
|
|
inline void Swap(HloModuleProtoWithConfig* 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(HloModuleProtoWithConfig* other) {
|
|
if (other == this) return;
|
|
GOOGLE_DCHECK(GetOwningArena() == other->GetOwningArena());
|
|
InternalSwap(other);
|
|
}
|
|
|
|
// implements Message ----------------------------------------------
|
|
|
|
HloModuleProtoWithConfig* New(::PROTOBUF_NAMESPACE_ID::Arena* arena = nullptr) const final {
|
|
return CreateMaybeMessage<HloModuleProtoWithConfig>(arena);
|
|
}
|
|
using ::PROTOBUF_NAMESPACE_ID::Message::CopyFrom;
|
|
void CopyFrom(const HloModuleProtoWithConfig& from);
|
|
using ::PROTOBUF_NAMESPACE_ID::Message::MergeFrom;
|
|
void MergeFrom( const HloModuleProtoWithConfig& from) {
|
|
HloModuleProtoWithConfig::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(HloModuleProtoWithConfig* other);
|
|
|
|
private:
|
|
friend class ::PROTOBUF_NAMESPACE_ID::internal::AnyMetadata;
|
|
static ::PROTOBUF_NAMESPACE_ID::StringPiece FullMessageName() {
|
|
return "xla.HloModuleProtoWithConfig";
|
|
}
|
|
protected:
|
|
explicit HloModuleProtoWithConfig(::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 {
|
|
kHloModuleFieldNumber = 1,
|
|
kConfigFieldNumber = 2,
|
|
};
|
|
// .xla.HloModuleProto hlo_module = 1;
|
|
bool has_hlo_module() const;
|
|
private:
|
|
bool _internal_has_hlo_module() const;
|
|
public:
|
|
void clear_hlo_module();
|
|
const ::xla::HloModuleProto& hlo_module() const;
|
|
PROTOBUF_NODISCARD ::xla::HloModuleProto* release_hlo_module();
|
|
::xla::HloModuleProto* mutable_hlo_module();
|
|
void set_allocated_hlo_module(::xla::HloModuleProto* hlo_module);
|
|
private:
|
|
const ::xla::HloModuleProto& _internal_hlo_module() const;
|
|
::xla::HloModuleProto* _internal_mutable_hlo_module();
|
|
public:
|
|
void unsafe_arena_set_allocated_hlo_module(
|
|
::xla::HloModuleProto* hlo_module);
|
|
::xla::HloModuleProto* unsafe_arena_release_hlo_module();
|
|
|
|
// .xla.HloModuleConfigProto config = 2;
|
|
bool has_config() const;
|
|
private:
|
|
bool _internal_has_config() const;
|
|
public:
|
|
void clear_config();
|
|
const ::xla::HloModuleConfigProto& config() const;
|
|
PROTOBUF_NODISCARD ::xla::HloModuleConfigProto* release_config();
|
|
::xla::HloModuleConfigProto* mutable_config();
|
|
void set_allocated_config(::xla::HloModuleConfigProto* config);
|
|
private:
|
|
const ::xla::HloModuleConfigProto& _internal_config() const;
|
|
::xla::HloModuleConfigProto* _internal_mutable_config();
|
|
public:
|
|
void unsafe_arena_set_allocated_config(
|
|
::xla::HloModuleConfigProto* config);
|
|
::xla::HloModuleConfigProto* unsafe_arena_release_config();
|
|
|
|
// @@protoc_insertion_point(class_scope:xla.HloModuleProtoWithConfig)
|
|
private:
|
|
class _Internal;
|
|
|
|
template <typename T> friend class ::PROTOBUF_NAMESPACE_ID::Arena::InternalHelper;
|
|
typedef void InternalArenaConstructable_;
|
|
typedef void DestructorSkippable_;
|
|
struct Impl_ {
|
|
::xla::HloModuleProto* hlo_module_;
|
|
::xla::HloModuleConfigProto* config_;
|
|
mutable ::PROTOBUF_NAMESPACE_ID::internal::CachedSize _cached_size_;
|
|
};
|
|
union { Impl_ _impl_; };
|
|
friend struct ::TableStruct_xla_2fxla_2eproto;
|
|
};
|
|
// -------------------------------------------------------------------
|
|
|
|
class GetDeviceHandlesRequest final :
|
|
public ::PROTOBUF_NAMESPACE_ID::Message /* @@protoc_insertion_point(class_definition:xla.GetDeviceHandlesRequest) */ {
|
|
public:
|
|
inline GetDeviceHandlesRequest() : GetDeviceHandlesRequest(nullptr) {}
|
|
~GetDeviceHandlesRequest() override;
|
|
explicit PROTOBUF_CONSTEXPR GetDeviceHandlesRequest(::PROTOBUF_NAMESPACE_ID::internal::ConstantInitialized);
|
|
|
|
GetDeviceHandlesRequest(const GetDeviceHandlesRequest& from);
|
|
GetDeviceHandlesRequest(GetDeviceHandlesRequest&& from) noexcept
|
|
: GetDeviceHandlesRequest() {
|
|
*this = ::std::move(from);
|
|
}
|
|
|
|
inline GetDeviceHandlesRequest& operator=(const GetDeviceHandlesRequest& from) {
|
|
CopyFrom(from);
|
|
return *this;
|
|
}
|
|
inline GetDeviceHandlesRequest& operator=(GetDeviceHandlesRequest&& 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 GetDeviceHandlesRequest& default_instance() {
|
|
return *internal_default_instance();
|
|
}
|
|
static inline const GetDeviceHandlesRequest* internal_default_instance() {
|
|
return reinterpret_cast<const GetDeviceHandlesRequest*>(
|
|
&_GetDeviceHandlesRequest_default_instance_);
|
|
}
|
|
static constexpr int kIndexInFileMessages =
|
|
13;
|
|
|
|
friend void swap(GetDeviceHandlesRequest& a, GetDeviceHandlesRequest& b) {
|
|
a.Swap(&b);
|
|
}
|
|
inline void Swap(GetDeviceHandlesRequest* 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(GetDeviceHandlesRequest* other) {
|
|
if (other == this) return;
|
|
GOOGLE_DCHECK(GetOwningArena() == other->GetOwningArena());
|
|
InternalSwap(other);
|
|
}
|
|
|
|
// implements Message ----------------------------------------------
|
|
|
|
GetDeviceHandlesRequest* New(::PROTOBUF_NAMESPACE_ID::Arena* arena = nullptr) const final {
|
|
return CreateMaybeMessage<GetDeviceHandlesRequest>(arena);
|
|
}
|
|
using ::PROTOBUF_NAMESPACE_ID::Message::CopyFrom;
|
|
void CopyFrom(const GetDeviceHandlesRequest& from);
|
|
using ::PROTOBUF_NAMESPACE_ID::Message::MergeFrom;
|
|
void MergeFrom( const GetDeviceHandlesRequest& from) {
|
|
GetDeviceHandlesRequest::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(GetDeviceHandlesRequest* other);
|
|
|
|
private:
|
|
friend class ::PROTOBUF_NAMESPACE_ID::internal::AnyMetadata;
|
|
static ::PROTOBUF_NAMESPACE_ID::StringPiece FullMessageName() {
|
|
return "xla.GetDeviceHandlesRequest";
|
|
}
|
|
protected:
|
|
explicit GetDeviceHandlesRequest(::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 {
|
|
kDeviceCountFieldNumber = 1,
|
|
};
|
|
// int64 device_count = 1;
|
|
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.GetDeviceHandlesRequest)
|
|
private:
|
|
class _Internal;
|
|
|
|
template <typename T> friend class ::PROTOBUF_NAMESPACE_ID::Arena::InternalHelper;
|
|
typedef void InternalArenaConstructable_;
|
|
typedef void DestructorSkippable_;
|
|
struct Impl_ {
|
|
int64_t device_count_;
|
|
mutable ::PROTOBUF_NAMESPACE_ID::internal::CachedSize _cached_size_;
|
|
};
|
|
union { Impl_ _impl_; };
|
|
friend struct ::TableStruct_xla_2fxla_2eproto;
|
|
};
|
|
// -------------------------------------------------------------------
|
|
|
|
class GetDeviceHandlesResponse final :
|
|
public ::PROTOBUF_NAMESPACE_ID::Message /* @@protoc_insertion_point(class_definition:xla.GetDeviceHandlesResponse) */ {
|
|
public:
|
|
inline GetDeviceHandlesResponse() : GetDeviceHandlesResponse(nullptr) {}
|
|
~GetDeviceHandlesResponse() override;
|
|
explicit PROTOBUF_CONSTEXPR GetDeviceHandlesResponse(::PROTOBUF_NAMESPACE_ID::internal::ConstantInitialized);
|
|
|
|
GetDeviceHandlesResponse(const GetDeviceHandlesResponse& from);
|
|
GetDeviceHandlesResponse(GetDeviceHandlesResponse&& from) noexcept
|
|
: GetDeviceHandlesResponse() {
|
|
*this = ::std::move(from);
|
|
}
|
|
|
|
inline GetDeviceHandlesResponse& operator=(const GetDeviceHandlesResponse& from) {
|
|
CopyFrom(from);
|
|
return *this;
|
|
}
|
|
inline GetDeviceHandlesResponse& operator=(GetDeviceHandlesResponse&& 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 GetDeviceHandlesResponse& default_instance() {
|
|
return *internal_default_instance();
|
|
}
|
|
static inline const GetDeviceHandlesResponse* internal_default_instance() {
|
|
return reinterpret_cast<const GetDeviceHandlesResponse*>(
|
|
&_GetDeviceHandlesResponse_default_instance_);
|
|
}
|
|
static constexpr int kIndexInFileMessages =
|
|
14;
|
|
|
|
friend void swap(GetDeviceHandlesResponse& a, GetDeviceHandlesResponse& b) {
|
|
a.Swap(&b);
|
|
}
|
|
inline void Swap(GetDeviceHandlesResponse* 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(GetDeviceHandlesResponse* other) {
|
|
if (other == this) return;
|
|
GOOGLE_DCHECK(GetOwningArena() == other->GetOwningArena());
|
|
InternalSwap(other);
|
|
}
|
|
|
|
// implements Message ----------------------------------------------
|
|
|
|
GetDeviceHandlesResponse* New(::PROTOBUF_NAMESPACE_ID::Arena* arena = nullptr) const final {
|
|
return CreateMaybeMessage<GetDeviceHandlesResponse>(arena);
|
|
}
|
|
using ::PROTOBUF_NAMESPACE_ID::Message::CopyFrom;
|
|
void CopyFrom(const GetDeviceHandlesResponse& from);
|
|
using ::PROTOBUF_NAMESPACE_ID::Message::MergeFrom;
|
|
void MergeFrom( const GetDeviceHandlesResponse& from) {
|
|
GetDeviceHandlesResponse::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(GetDeviceHandlesResponse* other);
|
|
|
|
private:
|
|
friend class ::PROTOBUF_NAMESPACE_ID::internal::AnyMetadata;
|
|
static ::PROTOBUF_NAMESPACE_ID::StringPiece FullMessageName() {
|
|
return "xla.GetDeviceHandlesResponse";
|
|
}
|
|
protected:
|
|
explicit GetDeviceHandlesResponse(::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 {
|
|
kDeviceHandlesFieldNumber = 1,
|
|
};
|
|
// repeated .xla.DeviceHandle device_handles = 1;
|
|
int device_handles_size() const;
|
|
private:
|
|
int _internal_device_handles_size() const;
|
|
public:
|
|
void clear_device_handles();
|
|
::xla::DeviceHandle* mutable_device_handles(int index);
|
|
::PROTOBUF_NAMESPACE_ID::RepeatedPtrField< ::xla::DeviceHandle >*
|
|
mutable_device_handles();
|
|
private:
|
|
const ::xla::DeviceHandle& _internal_device_handles(int index) const;
|
|
::xla::DeviceHandle* _internal_add_device_handles();
|
|
public:
|
|
const ::xla::DeviceHandle& device_handles(int index) const;
|
|
::xla::DeviceHandle* add_device_handles();
|
|
const ::PROTOBUF_NAMESPACE_ID::RepeatedPtrField< ::xla::DeviceHandle >&
|
|
device_handles() const;
|
|
|
|
// @@protoc_insertion_point(class_scope:xla.GetDeviceHandlesResponse)
|
|
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::DeviceHandle > device_handles_;
|
|
mutable ::PROTOBUF_NAMESPACE_ID::internal::CachedSize _cached_size_;
|
|
};
|
|
union { Impl_ _impl_; };
|
|
friend struct ::TableStruct_xla_2fxla_2eproto;
|
|
};
|
|
// -------------------------------------------------------------------
|
|
|
|
class TransferToClientRequest final :
|
|
public ::PROTOBUF_NAMESPACE_ID::Message /* @@protoc_insertion_point(class_definition:xla.TransferToClientRequest) */ {
|
|
public:
|
|
inline TransferToClientRequest() : TransferToClientRequest(nullptr) {}
|
|
~TransferToClientRequest() override;
|
|
explicit PROTOBUF_CONSTEXPR TransferToClientRequest(::PROTOBUF_NAMESPACE_ID::internal::ConstantInitialized);
|
|
|
|
TransferToClientRequest(const TransferToClientRequest& from);
|
|
TransferToClientRequest(TransferToClientRequest&& from) noexcept
|
|
: TransferToClientRequest() {
|
|
*this = ::std::move(from);
|
|
}
|
|
|
|
inline TransferToClientRequest& operator=(const TransferToClientRequest& from) {
|
|
CopyFrom(from);
|
|
return *this;
|
|
}
|
|
inline TransferToClientRequest& operator=(TransferToClientRequest&& 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 TransferToClientRequest& default_instance() {
|
|
return *internal_default_instance();
|
|
}
|
|
static inline const TransferToClientRequest* internal_default_instance() {
|
|
return reinterpret_cast<const TransferToClientRequest*>(
|
|
&_TransferToClientRequest_default_instance_);
|
|
}
|
|
static constexpr int kIndexInFileMessages =
|
|
15;
|
|
|
|
friend void swap(TransferToClientRequest& a, TransferToClientRequest& b) {
|
|
a.Swap(&b);
|
|
}
|
|
inline void Swap(TransferToClientRequest* 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(TransferToClientRequest* other) {
|
|
if (other == this) return;
|
|
GOOGLE_DCHECK(GetOwningArena() == other->GetOwningArena());
|
|
InternalSwap(other);
|
|
}
|
|
|
|
// implements Message ----------------------------------------------
|
|
|
|
TransferToClientRequest* New(::PROTOBUF_NAMESPACE_ID::Arena* arena = nullptr) const final {
|
|
return CreateMaybeMessage<TransferToClientRequest>(arena);
|
|
}
|
|
using ::PROTOBUF_NAMESPACE_ID::Message::CopyFrom;
|
|
void CopyFrom(const TransferToClientRequest& from);
|
|
using ::PROTOBUF_NAMESPACE_ID::Message::MergeFrom;
|
|
void MergeFrom( const TransferToClientRequest& from) {
|
|
TransferToClientRequest::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(TransferToClientRequest* other);
|
|
|
|
private:
|
|
friend class ::PROTOBUF_NAMESPACE_ID::internal::AnyMetadata;
|
|
static ::PROTOBUF_NAMESPACE_ID::StringPiece FullMessageName() {
|
|
return "xla.TransferToClientRequest";
|
|
}
|
|
protected:
|
|
explicit TransferToClientRequest(::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 {
|
|
kDataFieldNumber = 1,
|
|
kShapeWithLayoutFieldNumber = 2,
|
|
};
|
|
// .xla.GlobalDataHandle data = 1;
|
|
bool has_data() const;
|
|
private:
|
|
bool _internal_has_data() const;
|
|
public:
|
|
void clear_data();
|
|
const ::xla::GlobalDataHandle& data() const;
|
|
PROTOBUF_NODISCARD ::xla::GlobalDataHandle* release_data();
|
|
::xla::GlobalDataHandle* mutable_data();
|
|
void set_allocated_data(::xla::GlobalDataHandle* data);
|
|
private:
|
|
const ::xla::GlobalDataHandle& _internal_data() const;
|
|
::xla::GlobalDataHandle* _internal_mutable_data();
|
|
public:
|
|
void unsafe_arena_set_allocated_data(
|
|
::xla::GlobalDataHandle* data);
|
|
::xla::GlobalDataHandle* unsafe_arena_release_data();
|
|
|
|
// .xla.ShapeProto shape_with_layout = 2;
|
|
bool has_shape_with_layout() const;
|
|
private:
|
|
bool _internal_has_shape_with_layout() const;
|
|
public:
|
|
void clear_shape_with_layout();
|
|
const ::xla::ShapeProto& shape_with_layout() const;
|
|
PROTOBUF_NODISCARD ::xla::ShapeProto* release_shape_with_layout();
|
|
::xla::ShapeProto* mutable_shape_with_layout();
|
|
void set_allocated_shape_with_layout(::xla::ShapeProto* shape_with_layout);
|
|
private:
|
|
const ::xla::ShapeProto& _internal_shape_with_layout() const;
|
|
::xla::ShapeProto* _internal_mutable_shape_with_layout();
|
|
public:
|
|
void unsafe_arena_set_allocated_shape_with_layout(
|
|
::xla::ShapeProto* shape_with_layout);
|
|
::xla::ShapeProto* unsafe_arena_release_shape_with_layout();
|
|
|
|
// @@protoc_insertion_point(class_scope:xla.TransferToClientRequest)
|
|
private:
|
|
class _Internal;
|
|
|
|
template <typename T> friend class ::PROTOBUF_NAMESPACE_ID::Arena::InternalHelper;
|
|
typedef void InternalArenaConstructable_;
|
|
typedef void DestructorSkippable_;
|
|
struct Impl_ {
|
|
::xla::GlobalDataHandle* data_;
|
|
::xla::ShapeProto* shape_with_layout_;
|
|
mutable ::PROTOBUF_NAMESPACE_ID::internal::CachedSize _cached_size_;
|
|
};
|
|
union { Impl_ _impl_; };
|
|
friend struct ::TableStruct_xla_2fxla_2eproto;
|
|
};
|
|
// -------------------------------------------------------------------
|
|
|
|
class TransferToClientResponse final :
|
|
public ::PROTOBUF_NAMESPACE_ID::Message /* @@protoc_insertion_point(class_definition:xla.TransferToClientResponse) */ {
|
|
public:
|
|
inline TransferToClientResponse() : TransferToClientResponse(nullptr) {}
|
|
~TransferToClientResponse() override;
|
|
explicit PROTOBUF_CONSTEXPR TransferToClientResponse(::PROTOBUF_NAMESPACE_ID::internal::ConstantInitialized);
|
|
|
|
TransferToClientResponse(const TransferToClientResponse& from);
|
|
TransferToClientResponse(TransferToClientResponse&& from) noexcept
|
|
: TransferToClientResponse() {
|
|
*this = ::std::move(from);
|
|
}
|
|
|
|
inline TransferToClientResponse& operator=(const TransferToClientResponse& from) {
|
|
CopyFrom(from);
|
|
return *this;
|
|
}
|
|
inline TransferToClientResponse& operator=(TransferToClientResponse&& 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 TransferToClientResponse& default_instance() {
|
|
return *internal_default_instance();
|
|
}
|
|
static inline const TransferToClientResponse* internal_default_instance() {
|
|
return reinterpret_cast<const TransferToClientResponse*>(
|
|
&_TransferToClientResponse_default_instance_);
|
|
}
|
|
static constexpr int kIndexInFileMessages =
|
|
16;
|
|
|
|
friend void swap(TransferToClientResponse& a, TransferToClientResponse& b) {
|
|
a.Swap(&b);
|
|
}
|
|
inline void Swap(TransferToClientResponse* 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(TransferToClientResponse* other) {
|
|
if (other == this) return;
|
|
GOOGLE_DCHECK(GetOwningArena() == other->GetOwningArena());
|
|
InternalSwap(other);
|
|
}
|
|
|
|
// implements Message ----------------------------------------------
|
|
|
|
TransferToClientResponse* New(::PROTOBUF_NAMESPACE_ID::Arena* arena = nullptr) const final {
|
|
return CreateMaybeMessage<TransferToClientResponse>(arena);
|
|
}
|
|
using ::PROTOBUF_NAMESPACE_ID::Message::CopyFrom;
|
|
void CopyFrom(const TransferToClientResponse& from);
|
|
using ::PROTOBUF_NAMESPACE_ID::Message::MergeFrom;
|
|
void MergeFrom( const TransferToClientResponse& from) {
|
|
TransferToClientResponse::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(TransferToClientResponse* other);
|
|
|
|
private:
|
|
friend class ::PROTOBUF_NAMESPACE_ID::internal::AnyMetadata;
|
|
static ::PROTOBUF_NAMESPACE_ID::StringPiece FullMessageName() {
|
|
return "xla.TransferToClientResponse";
|
|
}
|
|
protected:
|
|
explicit TransferToClientResponse(::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 {
|
|
kLiteralFieldNumber = 1,
|
|
};
|
|
// .xla.LiteralProto literal = 1;
|
|
bool has_literal() const;
|
|
private:
|
|
bool _internal_has_literal() const;
|
|
public:
|
|
void clear_literal();
|
|
const ::xla::LiteralProto& literal() const;
|
|
PROTOBUF_NODISCARD ::xla::LiteralProto* release_literal();
|
|
::xla::LiteralProto* mutable_literal();
|
|
void set_allocated_literal(::xla::LiteralProto* literal);
|
|
private:
|
|
const ::xla::LiteralProto& _internal_literal() const;
|
|
::xla::LiteralProto* _internal_mutable_literal();
|
|
public:
|
|
void unsafe_arena_set_allocated_literal(
|
|
::xla::LiteralProto* literal);
|
|
::xla::LiteralProto* unsafe_arena_release_literal();
|
|
|
|
// @@protoc_insertion_point(class_scope:xla.TransferToClientResponse)
|
|
private:
|
|
class _Internal;
|
|
|
|
template <typename T> friend class ::PROTOBUF_NAMESPACE_ID::Arena::InternalHelper;
|
|
typedef void InternalArenaConstructable_;
|
|
typedef void DestructorSkippable_;
|
|
struct Impl_ {
|
|
::xla::LiteralProto* literal_;
|
|
mutable ::PROTOBUF_NAMESPACE_ID::internal::CachedSize _cached_size_;
|
|
};
|
|
union { Impl_ _impl_; };
|
|
friend struct ::TableStruct_xla_2fxla_2eproto;
|
|
};
|
|
// -------------------------------------------------------------------
|
|
|
|
class TransferToServerRequest final :
|
|
public ::PROTOBUF_NAMESPACE_ID::Message /* @@protoc_insertion_point(class_definition:xla.TransferToServerRequest) */ {
|
|
public:
|
|
inline TransferToServerRequest() : TransferToServerRequest(nullptr) {}
|
|
~TransferToServerRequest() override;
|
|
explicit PROTOBUF_CONSTEXPR TransferToServerRequest(::PROTOBUF_NAMESPACE_ID::internal::ConstantInitialized);
|
|
|
|
TransferToServerRequest(const TransferToServerRequest& from);
|
|
TransferToServerRequest(TransferToServerRequest&& from) noexcept
|
|
: TransferToServerRequest() {
|
|
*this = ::std::move(from);
|
|
}
|
|
|
|
inline TransferToServerRequest& operator=(const TransferToServerRequest& from) {
|
|
CopyFrom(from);
|
|
return *this;
|
|
}
|
|
inline TransferToServerRequest& operator=(TransferToServerRequest&& 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 TransferToServerRequest& default_instance() {
|
|
return *internal_default_instance();
|
|
}
|
|
static inline const TransferToServerRequest* internal_default_instance() {
|
|
return reinterpret_cast<const TransferToServerRequest*>(
|
|
&_TransferToServerRequest_default_instance_);
|
|
}
|
|
static constexpr int kIndexInFileMessages =
|
|
17;
|
|
|
|
friend void swap(TransferToServerRequest& a, TransferToServerRequest& b) {
|
|
a.Swap(&b);
|
|
}
|
|
inline void Swap(TransferToServerRequest* 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(TransferToServerRequest* other) {
|
|
if (other == this) return;
|
|
GOOGLE_DCHECK(GetOwningArena() == other->GetOwningArena());
|
|
InternalSwap(other);
|
|
}
|
|
|
|
// implements Message ----------------------------------------------
|
|
|
|
TransferToServerRequest* New(::PROTOBUF_NAMESPACE_ID::Arena* arena = nullptr) const final {
|
|
return CreateMaybeMessage<TransferToServerRequest>(arena);
|
|
}
|
|
using ::PROTOBUF_NAMESPACE_ID::Message::CopyFrom;
|
|
void CopyFrom(const TransferToServerRequest& from);
|
|
using ::PROTOBUF_NAMESPACE_ID::Message::MergeFrom;
|
|
void MergeFrom( const TransferToServerRequest& from) {
|
|
TransferToServerRequest::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(TransferToServerRequest* other);
|
|
|
|
private:
|
|
friend class ::PROTOBUF_NAMESPACE_ID::internal::AnyMetadata;
|
|
static ::PROTOBUF_NAMESPACE_ID::StringPiece FullMessageName() {
|
|
return "xla.TransferToServerRequest";
|
|
}
|
|
protected:
|
|
explicit TransferToServerRequest(::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 {
|
|
kLiteralFieldNumber = 1,
|
|
kDeviceHandleFieldNumber = 2,
|
|
};
|
|
// .xla.LiteralProto literal = 1;
|
|
bool has_literal() const;
|
|
private:
|
|
bool _internal_has_literal() const;
|
|
public:
|
|
void clear_literal();
|
|
const ::xla::LiteralProto& literal() const;
|
|
PROTOBUF_NODISCARD ::xla::LiteralProto* release_literal();
|
|
::xla::LiteralProto* mutable_literal();
|
|
void set_allocated_literal(::xla::LiteralProto* literal);
|
|
private:
|
|
const ::xla::LiteralProto& _internal_literal() const;
|
|
::xla::LiteralProto* _internal_mutable_literal();
|
|
public:
|
|
void unsafe_arena_set_allocated_literal(
|
|
::xla::LiteralProto* literal);
|
|
::xla::LiteralProto* unsafe_arena_release_literal();
|
|
|
|
// .xla.DeviceHandle device_handle = 2;
|
|
bool has_device_handle() const;
|
|
private:
|
|
bool _internal_has_device_handle() const;
|
|
public:
|
|
void clear_device_handle();
|
|
const ::xla::DeviceHandle& device_handle() const;
|
|
PROTOBUF_NODISCARD ::xla::DeviceHandle* release_device_handle();
|
|
::xla::DeviceHandle* mutable_device_handle();
|
|
void set_allocated_device_handle(::xla::DeviceHandle* device_handle);
|
|
private:
|
|
const ::xla::DeviceHandle& _internal_device_handle() const;
|
|
::xla::DeviceHandle* _internal_mutable_device_handle();
|
|
public:
|
|
void unsafe_arena_set_allocated_device_handle(
|
|
::xla::DeviceHandle* device_handle);
|
|
::xla::DeviceHandle* unsafe_arena_release_device_handle();
|
|
|
|
// @@protoc_insertion_point(class_scope:xla.TransferToServerRequest)
|
|
private:
|
|
class _Internal;
|
|
|
|
template <typename T> friend class ::PROTOBUF_NAMESPACE_ID::Arena::InternalHelper;
|
|
typedef void InternalArenaConstructable_;
|
|
typedef void DestructorSkippable_;
|
|
struct Impl_ {
|
|
::xla::LiteralProto* literal_;
|
|
::xla::DeviceHandle* device_handle_;
|
|
mutable ::PROTOBUF_NAMESPACE_ID::internal::CachedSize _cached_size_;
|
|
};
|
|
union { Impl_ _impl_; };
|
|
friend struct ::TableStruct_xla_2fxla_2eproto;
|
|
};
|
|
// -------------------------------------------------------------------
|
|
|
|
class TransferToServerResponse final :
|
|
public ::PROTOBUF_NAMESPACE_ID::Message /* @@protoc_insertion_point(class_definition:xla.TransferToServerResponse) */ {
|
|
public:
|
|
inline TransferToServerResponse() : TransferToServerResponse(nullptr) {}
|
|
~TransferToServerResponse() override;
|
|
explicit PROTOBUF_CONSTEXPR TransferToServerResponse(::PROTOBUF_NAMESPACE_ID::internal::ConstantInitialized);
|
|
|
|
TransferToServerResponse(const TransferToServerResponse& from);
|
|
TransferToServerResponse(TransferToServerResponse&& from) noexcept
|
|
: TransferToServerResponse() {
|
|
*this = ::std::move(from);
|
|
}
|
|
|
|
inline TransferToServerResponse& operator=(const TransferToServerResponse& from) {
|
|
CopyFrom(from);
|
|
return *this;
|
|
}
|
|
inline TransferToServerResponse& operator=(TransferToServerResponse&& 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 TransferToServerResponse& default_instance() {
|
|
return *internal_default_instance();
|
|
}
|
|
static inline const TransferToServerResponse* internal_default_instance() {
|
|
return reinterpret_cast<const TransferToServerResponse*>(
|
|
&_TransferToServerResponse_default_instance_);
|
|
}
|
|
static constexpr int kIndexInFileMessages =
|
|
18;
|
|
|
|
friend void swap(TransferToServerResponse& a, TransferToServerResponse& b) {
|
|
a.Swap(&b);
|
|
}
|
|
inline void Swap(TransferToServerResponse* 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(TransferToServerResponse* other) {
|
|
if (other == this) return;
|
|
GOOGLE_DCHECK(GetOwningArena() == other->GetOwningArena());
|
|
InternalSwap(other);
|
|
}
|
|
|
|
// implements Message ----------------------------------------------
|
|
|
|
TransferToServerResponse* New(::PROTOBUF_NAMESPACE_ID::Arena* arena = nullptr) const final {
|
|
return CreateMaybeMessage<TransferToServerResponse>(arena);
|
|
}
|
|
using ::PROTOBUF_NAMESPACE_ID::Message::CopyFrom;
|
|
void CopyFrom(const TransferToServerResponse& from);
|
|
using ::PROTOBUF_NAMESPACE_ID::Message::MergeFrom;
|
|
void MergeFrom( const TransferToServerResponse& from) {
|
|
TransferToServerResponse::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(TransferToServerResponse* other);
|
|
|
|
private:
|
|
friend class ::PROTOBUF_NAMESPACE_ID::internal::AnyMetadata;
|
|
static ::PROTOBUF_NAMESPACE_ID::StringPiece FullMessageName() {
|
|
return "xla.TransferToServerResponse";
|
|
}
|
|
protected:
|
|
explicit TransferToServerResponse(::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 {
|
|
kDataFieldNumber = 1,
|
|
};
|
|
// .xla.GlobalDataHandle data = 1;
|
|
bool has_data() const;
|
|
private:
|
|
bool _internal_has_data() const;
|
|
public:
|
|
void clear_data();
|
|
const ::xla::GlobalDataHandle& data() const;
|
|
PROTOBUF_NODISCARD ::xla::GlobalDataHandle* release_data();
|
|
::xla::GlobalDataHandle* mutable_data();
|
|
void set_allocated_data(::xla::GlobalDataHandle* data);
|
|
private:
|
|
const ::xla::GlobalDataHandle& _internal_data() const;
|
|
::xla::GlobalDataHandle* _internal_mutable_data();
|
|
public:
|
|
void unsafe_arena_set_allocated_data(
|
|
::xla::GlobalDataHandle* data);
|
|
::xla::GlobalDataHandle* unsafe_arena_release_data();
|
|
|
|
// @@protoc_insertion_point(class_scope:xla.TransferToServerResponse)
|
|
private:
|
|
class _Internal;
|
|
|
|
template <typename T> friend class ::PROTOBUF_NAMESPACE_ID::Arena::InternalHelper;
|
|
typedef void InternalArenaConstructable_;
|
|
typedef void DestructorSkippable_;
|
|
struct Impl_ {
|
|
::xla::GlobalDataHandle* data_;
|
|
mutable ::PROTOBUF_NAMESPACE_ID::internal::CachedSize _cached_size_;
|
|
};
|
|
union { Impl_ _impl_; };
|
|
friend struct ::TableStruct_xla_2fxla_2eproto;
|
|
};
|
|
// -------------------------------------------------------------------
|
|
|
|
class TransferToInfeedRequest final :
|
|
public ::PROTOBUF_NAMESPACE_ID::Message /* @@protoc_insertion_point(class_definition:xla.TransferToInfeedRequest) */ {
|
|
public:
|
|
inline TransferToInfeedRequest() : TransferToInfeedRequest(nullptr) {}
|
|
~TransferToInfeedRequest() override;
|
|
explicit PROTOBUF_CONSTEXPR TransferToInfeedRequest(::PROTOBUF_NAMESPACE_ID::internal::ConstantInitialized);
|
|
|
|
TransferToInfeedRequest(const TransferToInfeedRequest& from);
|
|
TransferToInfeedRequest(TransferToInfeedRequest&& from) noexcept
|
|
: TransferToInfeedRequest() {
|
|
*this = ::std::move(from);
|
|
}
|
|
|
|
inline TransferToInfeedRequest& operator=(const TransferToInfeedRequest& from) {
|
|
CopyFrom(from);
|
|
return *this;
|
|
}
|
|
inline TransferToInfeedRequest& operator=(TransferToInfeedRequest&& 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 TransferToInfeedRequest& default_instance() {
|
|
return *internal_default_instance();
|
|
}
|
|
static inline const TransferToInfeedRequest* internal_default_instance() {
|
|
return reinterpret_cast<const TransferToInfeedRequest*>(
|
|
&_TransferToInfeedRequest_default_instance_);
|
|
}
|
|
static constexpr int kIndexInFileMessages =
|
|
19;
|
|
|
|
friend void swap(TransferToInfeedRequest& a, TransferToInfeedRequest& b) {
|
|
a.Swap(&b);
|
|
}
|
|
inline void Swap(TransferToInfeedRequest* 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(TransferToInfeedRequest* other) {
|
|
if (other == this) return;
|
|
GOOGLE_DCHECK(GetOwningArena() == other->GetOwningArena());
|
|
InternalSwap(other);
|
|
}
|
|
|
|
// implements Message ----------------------------------------------
|
|
|
|
TransferToInfeedRequest* New(::PROTOBUF_NAMESPACE_ID::Arena* arena = nullptr) const final {
|
|
return CreateMaybeMessage<TransferToInfeedRequest>(arena);
|
|
}
|
|
using ::PROTOBUF_NAMESPACE_ID::Message::CopyFrom;
|
|
void CopyFrom(const TransferToInfeedRequest& from);
|
|
using ::PROTOBUF_NAMESPACE_ID::Message::MergeFrom;
|
|
void MergeFrom( const TransferToInfeedRequest& from) {
|
|
TransferToInfeedRequest::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(TransferToInfeedRequest* other);
|
|
|
|
private:
|
|
friend class ::PROTOBUF_NAMESPACE_ID::internal::AnyMetadata;
|
|
static ::PROTOBUF_NAMESPACE_ID::StringPiece FullMessageName() {
|
|
return "xla.TransferToInfeedRequest";
|
|
}
|
|
protected:
|
|
explicit TransferToInfeedRequest(::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 {
|
|
kLiteralFieldNumber = 1,
|
|
kDeviceHandleFieldNumber = 3,
|
|
kReplicaIdFieldNumber = 2,
|
|
};
|
|
// .xla.LiteralProto literal = 1;
|
|
bool has_literal() const;
|
|
private:
|
|
bool _internal_has_literal() const;
|
|
public:
|
|
void clear_literal();
|
|
const ::xla::LiteralProto& literal() const;
|
|
PROTOBUF_NODISCARD ::xla::LiteralProto* release_literal();
|
|
::xla::LiteralProto* mutable_literal();
|
|
void set_allocated_literal(::xla::LiteralProto* literal);
|
|
private:
|
|
const ::xla::LiteralProto& _internal_literal() const;
|
|
::xla::LiteralProto* _internal_mutable_literal();
|
|
public:
|
|
void unsafe_arena_set_allocated_literal(
|
|
::xla::LiteralProto* literal);
|
|
::xla::LiteralProto* unsafe_arena_release_literal();
|
|
|
|
// .xla.DeviceHandle device_handle = 3;
|
|
bool has_device_handle() const;
|
|
private:
|
|
bool _internal_has_device_handle() const;
|
|
public:
|
|
void clear_device_handle();
|
|
const ::xla::DeviceHandle& device_handle() const;
|
|
PROTOBUF_NODISCARD ::xla::DeviceHandle* release_device_handle();
|
|
::xla::DeviceHandle* mutable_device_handle();
|
|
void set_allocated_device_handle(::xla::DeviceHandle* device_handle);
|
|
private:
|
|
const ::xla::DeviceHandle& _internal_device_handle() const;
|
|
::xla::DeviceHandle* _internal_mutable_device_handle();
|
|
public:
|
|
void unsafe_arena_set_allocated_device_handle(
|
|
::xla::DeviceHandle* device_handle);
|
|
::xla::DeviceHandle* unsafe_arena_release_device_handle();
|
|
|
|
// int64 replica_id = 2;
|
|
void clear_replica_id();
|
|
int64_t replica_id() const;
|
|
void set_replica_id(int64_t value);
|
|
private:
|
|
int64_t _internal_replica_id() const;
|
|
void _internal_set_replica_id(int64_t value);
|
|
public:
|
|
|
|
// @@protoc_insertion_point(class_scope:xla.TransferToInfeedRequest)
|
|
private:
|
|
class _Internal;
|
|
|
|
template <typename T> friend class ::PROTOBUF_NAMESPACE_ID::Arena::InternalHelper;
|
|
typedef void InternalArenaConstructable_;
|
|
typedef void DestructorSkippable_;
|
|
struct Impl_ {
|
|
::xla::LiteralProto* literal_;
|
|
::xla::DeviceHandle* device_handle_;
|
|
int64_t replica_id_;
|
|
mutable ::PROTOBUF_NAMESPACE_ID::internal::CachedSize _cached_size_;
|
|
};
|
|
union { Impl_ _impl_; };
|
|
friend struct ::TableStruct_xla_2fxla_2eproto;
|
|
};
|
|
// -------------------------------------------------------------------
|
|
|
|
class TransferToInfeedResponse final :
|
|
public ::PROTOBUF_NAMESPACE_ID::internal::ZeroFieldsBase /* @@protoc_insertion_point(class_definition:xla.TransferToInfeedResponse) */ {
|
|
public:
|
|
inline TransferToInfeedResponse() : TransferToInfeedResponse(nullptr) {}
|
|
explicit PROTOBUF_CONSTEXPR TransferToInfeedResponse(::PROTOBUF_NAMESPACE_ID::internal::ConstantInitialized);
|
|
|
|
TransferToInfeedResponse(const TransferToInfeedResponse& from);
|
|
TransferToInfeedResponse(TransferToInfeedResponse&& from) noexcept
|
|
: TransferToInfeedResponse() {
|
|
*this = ::std::move(from);
|
|
}
|
|
|
|
inline TransferToInfeedResponse& operator=(const TransferToInfeedResponse& from) {
|
|
CopyFrom(from);
|
|
return *this;
|
|
}
|
|
inline TransferToInfeedResponse& operator=(TransferToInfeedResponse&& 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 TransferToInfeedResponse& default_instance() {
|
|
return *internal_default_instance();
|
|
}
|
|
static inline const TransferToInfeedResponse* internal_default_instance() {
|
|
return reinterpret_cast<const TransferToInfeedResponse*>(
|
|
&_TransferToInfeedResponse_default_instance_);
|
|
}
|
|
static constexpr int kIndexInFileMessages =
|
|
20;
|
|
|
|
friend void swap(TransferToInfeedResponse& a, TransferToInfeedResponse& b) {
|
|
a.Swap(&b);
|
|
}
|
|
inline void Swap(TransferToInfeedResponse* 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(TransferToInfeedResponse* other) {
|
|
if (other == this) return;
|
|
GOOGLE_DCHECK(GetOwningArena() == other->GetOwningArena());
|
|
InternalSwap(other);
|
|
}
|
|
|
|
// implements Message ----------------------------------------------
|
|
|
|
TransferToInfeedResponse* New(::PROTOBUF_NAMESPACE_ID::Arena* arena = nullptr) const final {
|
|
return CreateMaybeMessage<TransferToInfeedResponse>(arena);
|
|
}
|
|
using ::PROTOBUF_NAMESPACE_ID::internal::ZeroFieldsBase::CopyFrom;
|
|
inline void CopyFrom(const TransferToInfeedResponse& from) {
|
|
::PROTOBUF_NAMESPACE_ID::internal::ZeroFieldsBase::CopyImpl(*this, from);
|
|
}
|
|
using ::PROTOBUF_NAMESPACE_ID::internal::ZeroFieldsBase::MergeFrom;
|
|
void MergeFrom(const TransferToInfeedResponse& from) {
|
|
::PROTOBUF_NAMESPACE_ID::internal::ZeroFieldsBase::MergeImpl(*this, from);
|
|
}
|
|
public:
|
|
|
|
private:
|
|
friend class ::PROTOBUF_NAMESPACE_ID::internal::AnyMetadata;
|
|
static ::PROTOBUF_NAMESPACE_ID::StringPiece FullMessageName() {
|
|
return "xla.TransferToInfeedResponse";
|
|
}
|
|
protected:
|
|
explicit TransferToInfeedResponse(::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 -------------------------------------------------------
|
|
|
|
// @@protoc_insertion_point(class_scope:xla.TransferToInfeedResponse)
|
|
private:
|
|
class _Internal;
|
|
|
|
template <typename T> friend class ::PROTOBUF_NAMESPACE_ID::Arena::InternalHelper;
|
|
typedef void InternalArenaConstructable_;
|
|
typedef void DestructorSkippable_;
|
|
struct Impl_ {
|
|
};
|
|
friend struct ::TableStruct_xla_2fxla_2eproto;
|
|
};
|
|
// -------------------------------------------------------------------
|
|
|
|
class TransferFromOutfeedRequest final :
|
|
public ::PROTOBUF_NAMESPACE_ID::Message /* @@protoc_insertion_point(class_definition:xla.TransferFromOutfeedRequest) */ {
|
|
public:
|
|
inline TransferFromOutfeedRequest() : TransferFromOutfeedRequest(nullptr) {}
|
|
~TransferFromOutfeedRequest() override;
|
|
explicit PROTOBUF_CONSTEXPR TransferFromOutfeedRequest(::PROTOBUF_NAMESPACE_ID::internal::ConstantInitialized);
|
|
|
|
TransferFromOutfeedRequest(const TransferFromOutfeedRequest& from);
|
|
TransferFromOutfeedRequest(TransferFromOutfeedRequest&& from) noexcept
|
|
: TransferFromOutfeedRequest() {
|
|
*this = ::std::move(from);
|
|
}
|
|
|
|
inline TransferFromOutfeedRequest& operator=(const TransferFromOutfeedRequest& from) {
|
|
CopyFrom(from);
|
|
return *this;
|
|
}
|
|
inline TransferFromOutfeedRequest& operator=(TransferFromOutfeedRequest&& 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 TransferFromOutfeedRequest& default_instance() {
|
|
return *internal_default_instance();
|
|
}
|
|
static inline const TransferFromOutfeedRequest* internal_default_instance() {
|
|
return reinterpret_cast<const TransferFromOutfeedRequest*>(
|
|
&_TransferFromOutfeedRequest_default_instance_);
|
|
}
|
|
static constexpr int kIndexInFileMessages =
|
|
21;
|
|
|
|
friend void swap(TransferFromOutfeedRequest& a, TransferFromOutfeedRequest& b) {
|
|
a.Swap(&b);
|
|
}
|
|
inline void Swap(TransferFromOutfeedRequest* 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(TransferFromOutfeedRequest* other) {
|
|
if (other == this) return;
|
|
GOOGLE_DCHECK(GetOwningArena() == other->GetOwningArena());
|
|
InternalSwap(other);
|
|
}
|
|
|
|
// implements Message ----------------------------------------------
|
|
|
|
TransferFromOutfeedRequest* New(::PROTOBUF_NAMESPACE_ID::Arena* arena = nullptr) const final {
|
|
return CreateMaybeMessage<TransferFromOutfeedRequest>(arena);
|
|
}
|
|
using ::PROTOBUF_NAMESPACE_ID::Message::CopyFrom;
|
|
void CopyFrom(const TransferFromOutfeedRequest& from);
|
|
using ::PROTOBUF_NAMESPACE_ID::Message::MergeFrom;
|
|
void MergeFrom( const TransferFromOutfeedRequest& from) {
|
|
TransferFromOutfeedRequest::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(TransferFromOutfeedRequest* other);
|
|
|
|
private:
|
|
friend class ::PROTOBUF_NAMESPACE_ID::internal::AnyMetadata;
|
|
static ::PROTOBUF_NAMESPACE_ID::StringPiece FullMessageName() {
|
|
return "xla.TransferFromOutfeedRequest";
|
|
}
|
|
protected:
|
|
explicit TransferFromOutfeedRequest(::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 {
|
|
kShapeWithLayoutFieldNumber = 1,
|
|
kDeviceHandleFieldNumber = 3,
|
|
kReplicaIdFieldNumber = 2,
|
|
};
|
|
// .xla.ShapeProto shape_with_layout = 1;
|
|
bool has_shape_with_layout() const;
|
|
private:
|
|
bool _internal_has_shape_with_layout() const;
|
|
public:
|
|
void clear_shape_with_layout();
|
|
const ::xla::ShapeProto& shape_with_layout() const;
|
|
PROTOBUF_NODISCARD ::xla::ShapeProto* release_shape_with_layout();
|
|
::xla::ShapeProto* mutable_shape_with_layout();
|
|
void set_allocated_shape_with_layout(::xla::ShapeProto* shape_with_layout);
|
|
private:
|
|
const ::xla::ShapeProto& _internal_shape_with_layout() const;
|
|
::xla::ShapeProto* _internal_mutable_shape_with_layout();
|
|
public:
|
|
void unsafe_arena_set_allocated_shape_with_layout(
|
|
::xla::ShapeProto* shape_with_layout);
|
|
::xla::ShapeProto* unsafe_arena_release_shape_with_layout();
|
|
|
|
// .xla.DeviceHandle device_handle = 3;
|
|
bool has_device_handle() const;
|
|
private:
|
|
bool _internal_has_device_handle() const;
|
|
public:
|
|
void clear_device_handle();
|
|
const ::xla::DeviceHandle& device_handle() const;
|
|
PROTOBUF_NODISCARD ::xla::DeviceHandle* release_device_handle();
|
|
::xla::DeviceHandle* mutable_device_handle();
|
|
void set_allocated_device_handle(::xla::DeviceHandle* device_handle);
|
|
private:
|
|
const ::xla::DeviceHandle& _internal_device_handle() const;
|
|
::xla::DeviceHandle* _internal_mutable_device_handle();
|
|
public:
|
|
void unsafe_arena_set_allocated_device_handle(
|
|
::xla::DeviceHandle* device_handle);
|
|
::xla::DeviceHandle* unsafe_arena_release_device_handle();
|
|
|
|
// int64 replica_id = 2;
|
|
void clear_replica_id();
|
|
int64_t replica_id() const;
|
|
void set_replica_id(int64_t value);
|
|
private:
|
|
int64_t _internal_replica_id() const;
|
|
void _internal_set_replica_id(int64_t value);
|
|
public:
|
|
|
|
// @@protoc_insertion_point(class_scope:xla.TransferFromOutfeedRequest)
|
|
private:
|
|
class _Internal;
|
|
|
|
template <typename T> friend class ::PROTOBUF_NAMESPACE_ID::Arena::InternalHelper;
|
|
typedef void InternalArenaConstructable_;
|
|
typedef void DestructorSkippable_;
|
|
struct Impl_ {
|
|
::xla::ShapeProto* shape_with_layout_;
|
|
::xla::DeviceHandle* device_handle_;
|
|
int64_t replica_id_;
|
|
mutable ::PROTOBUF_NAMESPACE_ID::internal::CachedSize _cached_size_;
|
|
};
|
|
union { Impl_ _impl_; };
|
|
friend struct ::TableStruct_xla_2fxla_2eproto;
|
|
};
|
|
// -------------------------------------------------------------------
|
|
|
|
class TransferFromOutfeedResponse final :
|
|
public ::PROTOBUF_NAMESPACE_ID::Message /* @@protoc_insertion_point(class_definition:xla.TransferFromOutfeedResponse) */ {
|
|
public:
|
|
inline TransferFromOutfeedResponse() : TransferFromOutfeedResponse(nullptr) {}
|
|
~TransferFromOutfeedResponse() override;
|
|
explicit PROTOBUF_CONSTEXPR TransferFromOutfeedResponse(::PROTOBUF_NAMESPACE_ID::internal::ConstantInitialized);
|
|
|
|
TransferFromOutfeedResponse(const TransferFromOutfeedResponse& from);
|
|
TransferFromOutfeedResponse(TransferFromOutfeedResponse&& from) noexcept
|
|
: TransferFromOutfeedResponse() {
|
|
*this = ::std::move(from);
|
|
}
|
|
|
|
inline TransferFromOutfeedResponse& operator=(const TransferFromOutfeedResponse& from) {
|
|
CopyFrom(from);
|
|
return *this;
|
|
}
|
|
inline TransferFromOutfeedResponse& operator=(TransferFromOutfeedResponse&& 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 TransferFromOutfeedResponse& default_instance() {
|
|
return *internal_default_instance();
|
|
}
|
|
static inline const TransferFromOutfeedResponse* internal_default_instance() {
|
|
return reinterpret_cast<const TransferFromOutfeedResponse*>(
|
|
&_TransferFromOutfeedResponse_default_instance_);
|
|
}
|
|
static constexpr int kIndexInFileMessages =
|
|
22;
|
|
|
|
friend void swap(TransferFromOutfeedResponse& a, TransferFromOutfeedResponse& b) {
|
|
a.Swap(&b);
|
|
}
|
|
inline void Swap(TransferFromOutfeedResponse* 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(TransferFromOutfeedResponse* other) {
|
|
if (other == this) return;
|
|
GOOGLE_DCHECK(GetOwningArena() == other->GetOwningArena());
|
|
InternalSwap(other);
|
|
}
|
|
|
|
// implements Message ----------------------------------------------
|
|
|
|
TransferFromOutfeedResponse* New(::PROTOBUF_NAMESPACE_ID::Arena* arena = nullptr) const final {
|
|
return CreateMaybeMessage<TransferFromOutfeedResponse>(arena);
|
|
}
|
|
using ::PROTOBUF_NAMESPACE_ID::Message::CopyFrom;
|
|
void CopyFrom(const TransferFromOutfeedResponse& from);
|
|
using ::PROTOBUF_NAMESPACE_ID::Message::MergeFrom;
|
|
void MergeFrom( const TransferFromOutfeedResponse& from) {
|
|
TransferFromOutfeedResponse::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(TransferFromOutfeedResponse* other);
|
|
|
|
private:
|
|
friend class ::PROTOBUF_NAMESPACE_ID::internal::AnyMetadata;
|
|
static ::PROTOBUF_NAMESPACE_ID::StringPiece FullMessageName() {
|
|
return "xla.TransferFromOutfeedResponse";
|
|
}
|
|
protected:
|
|
explicit TransferFromOutfeedResponse(::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 {
|
|
kLiteralFieldNumber = 1,
|
|
};
|
|
// .xla.LiteralProto literal = 1;
|
|
bool has_literal() const;
|
|
private:
|
|
bool _internal_has_literal() const;
|
|
public:
|
|
void clear_literal();
|
|
const ::xla::LiteralProto& literal() const;
|
|
PROTOBUF_NODISCARD ::xla::LiteralProto* release_literal();
|
|
::xla::LiteralProto* mutable_literal();
|
|
void set_allocated_literal(::xla::LiteralProto* literal);
|
|
private:
|
|
const ::xla::LiteralProto& _internal_literal() const;
|
|
::xla::LiteralProto* _internal_mutable_literal();
|
|
public:
|
|
void unsafe_arena_set_allocated_literal(
|
|
::xla::LiteralProto* literal);
|
|
::xla::LiteralProto* unsafe_arena_release_literal();
|
|
|
|
// @@protoc_insertion_point(class_scope:xla.TransferFromOutfeedResponse)
|
|
private:
|
|
class _Internal;
|
|
|
|
template <typename T> friend class ::PROTOBUF_NAMESPACE_ID::Arena::InternalHelper;
|
|
typedef void InternalArenaConstructable_;
|
|
typedef void DestructorSkippable_;
|
|
struct Impl_ {
|
|
::xla::LiteralProto* literal_;
|
|
mutable ::PROTOBUF_NAMESPACE_ID::internal::CachedSize _cached_size_;
|
|
};
|
|
union { Impl_ _impl_; };
|
|
friend struct ::TableStruct_xla_2fxla_2eproto;
|
|
};
|
|
// -------------------------------------------------------------------
|
|
|
|
class ResetDeviceRequest final :
|
|
public ::PROTOBUF_NAMESPACE_ID::Message /* @@protoc_insertion_point(class_definition:xla.ResetDeviceRequest) */ {
|
|
public:
|
|
inline ResetDeviceRequest() : ResetDeviceRequest(nullptr) {}
|
|
~ResetDeviceRequest() override;
|
|
explicit PROTOBUF_CONSTEXPR ResetDeviceRequest(::PROTOBUF_NAMESPACE_ID::internal::ConstantInitialized);
|
|
|
|
ResetDeviceRequest(const ResetDeviceRequest& from);
|
|
ResetDeviceRequest(ResetDeviceRequest&& from) noexcept
|
|
: ResetDeviceRequest() {
|
|
*this = ::std::move(from);
|
|
}
|
|
|
|
inline ResetDeviceRequest& operator=(const ResetDeviceRequest& from) {
|
|
CopyFrom(from);
|
|
return *this;
|
|
}
|
|
inline ResetDeviceRequest& operator=(ResetDeviceRequest&& 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 ResetDeviceRequest& default_instance() {
|
|
return *internal_default_instance();
|
|
}
|
|
static inline const ResetDeviceRequest* internal_default_instance() {
|
|
return reinterpret_cast<const ResetDeviceRequest*>(
|
|
&_ResetDeviceRequest_default_instance_);
|
|
}
|
|
static constexpr int kIndexInFileMessages =
|
|
23;
|
|
|
|
friend void swap(ResetDeviceRequest& a, ResetDeviceRequest& b) {
|
|
a.Swap(&b);
|
|
}
|
|
inline void Swap(ResetDeviceRequest* 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(ResetDeviceRequest* other) {
|
|
if (other == this) return;
|
|
GOOGLE_DCHECK(GetOwningArena() == other->GetOwningArena());
|
|
InternalSwap(other);
|
|
}
|
|
|
|
// implements Message ----------------------------------------------
|
|
|
|
ResetDeviceRequest* New(::PROTOBUF_NAMESPACE_ID::Arena* arena = nullptr) const final {
|
|
return CreateMaybeMessage<ResetDeviceRequest>(arena);
|
|
}
|
|
using ::PROTOBUF_NAMESPACE_ID::Message::CopyFrom;
|
|
void CopyFrom(const ResetDeviceRequest& from);
|
|
using ::PROTOBUF_NAMESPACE_ID::Message::MergeFrom;
|
|
void MergeFrom( const ResetDeviceRequest& from) {
|
|
ResetDeviceRequest::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(ResetDeviceRequest* other);
|
|
|
|
private:
|
|
friend class ::PROTOBUF_NAMESPACE_ID::internal::AnyMetadata;
|
|
static ::PROTOBUF_NAMESPACE_ID::StringPiece FullMessageName() {
|
|
return "xla.ResetDeviceRequest";
|
|
}
|
|
protected:
|
|
explicit ResetDeviceRequest(::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 {
|
|
kDeviceHandleFieldNumber = 1,
|
|
};
|
|
// .xla.DeviceHandle device_handle = 1;
|
|
bool has_device_handle() const;
|
|
private:
|
|
bool _internal_has_device_handle() const;
|
|
public:
|
|
void clear_device_handle();
|
|
const ::xla::DeviceHandle& device_handle() const;
|
|
PROTOBUF_NODISCARD ::xla::DeviceHandle* release_device_handle();
|
|
::xla::DeviceHandle* mutable_device_handle();
|
|
void set_allocated_device_handle(::xla::DeviceHandle* device_handle);
|
|
private:
|
|
const ::xla::DeviceHandle& _internal_device_handle() const;
|
|
::xla::DeviceHandle* _internal_mutable_device_handle();
|
|
public:
|
|
void unsafe_arena_set_allocated_device_handle(
|
|
::xla::DeviceHandle* device_handle);
|
|
::xla::DeviceHandle* unsafe_arena_release_device_handle();
|
|
|
|
// @@protoc_insertion_point(class_scope:xla.ResetDeviceRequest)
|
|
private:
|
|
class _Internal;
|
|
|
|
template <typename T> friend class ::PROTOBUF_NAMESPACE_ID::Arena::InternalHelper;
|
|
typedef void InternalArenaConstructable_;
|
|
typedef void DestructorSkippable_;
|
|
struct Impl_ {
|
|
::xla::DeviceHandle* device_handle_;
|
|
mutable ::PROTOBUF_NAMESPACE_ID::internal::CachedSize _cached_size_;
|
|
};
|
|
union { Impl_ _impl_; };
|
|
friend struct ::TableStruct_xla_2fxla_2eproto;
|
|
};
|
|
// -------------------------------------------------------------------
|
|
|
|
class ResetDeviceResponse final :
|
|
public ::PROTOBUF_NAMESPACE_ID::internal::ZeroFieldsBase /* @@protoc_insertion_point(class_definition:xla.ResetDeviceResponse) */ {
|
|
public:
|
|
inline ResetDeviceResponse() : ResetDeviceResponse(nullptr) {}
|
|
explicit PROTOBUF_CONSTEXPR ResetDeviceResponse(::PROTOBUF_NAMESPACE_ID::internal::ConstantInitialized);
|
|
|
|
ResetDeviceResponse(const ResetDeviceResponse& from);
|
|
ResetDeviceResponse(ResetDeviceResponse&& from) noexcept
|
|
: ResetDeviceResponse() {
|
|
*this = ::std::move(from);
|
|
}
|
|
|
|
inline ResetDeviceResponse& operator=(const ResetDeviceResponse& from) {
|
|
CopyFrom(from);
|
|
return *this;
|
|
}
|
|
inline ResetDeviceResponse& operator=(ResetDeviceResponse&& 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 ResetDeviceResponse& default_instance() {
|
|
return *internal_default_instance();
|
|
}
|
|
static inline const ResetDeviceResponse* internal_default_instance() {
|
|
return reinterpret_cast<const ResetDeviceResponse*>(
|
|
&_ResetDeviceResponse_default_instance_);
|
|
}
|
|
static constexpr int kIndexInFileMessages =
|
|
24;
|
|
|
|
friend void swap(ResetDeviceResponse& a, ResetDeviceResponse& b) {
|
|
a.Swap(&b);
|
|
}
|
|
inline void Swap(ResetDeviceResponse* 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(ResetDeviceResponse* other) {
|
|
if (other == this) return;
|
|
GOOGLE_DCHECK(GetOwningArena() == other->GetOwningArena());
|
|
InternalSwap(other);
|
|
}
|
|
|
|
// implements Message ----------------------------------------------
|
|
|
|
ResetDeviceResponse* New(::PROTOBUF_NAMESPACE_ID::Arena* arena = nullptr) const final {
|
|
return CreateMaybeMessage<ResetDeviceResponse>(arena);
|
|
}
|
|
using ::PROTOBUF_NAMESPACE_ID::internal::ZeroFieldsBase::CopyFrom;
|
|
inline void CopyFrom(const ResetDeviceResponse& from) {
|
|
::PROTOBUF_NAMESPACE_ID::internal::ZeroFieldsBase::CopyImpl(*this, from);
|
|
}
|
|
using ::PROTOBUF_NAMESPACE_ID::internal::ZeroFieldsBase::MergeFrom;
|
|
void MergeFrom(const ResetDeviceResponse& from) {
|
|
::PROTOBUF_NAMESPACE_ID::internal::ZeroFieldsBase::MergeImpl(*this, from);
|
|
}
|
|
public:
|
|
|
|
private:
|
|
friend class ::PROTOBUF_NAMESPACE_ID::internal::AnyMetadata;
|
|
static ::PROTOBUF_NAMESPACE_ID::StringPiece FullMessageName() {
|
|
return "xla.ResetDeviceResponse";
|
|
}
|
|
protected:
|
|
explicit ResetDeviceResponse(::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 -------------------------------------------------------
|
|
|
|
// @@protoc_insertion_point(class_scope:xla.ResetDeviceResponse)
|
|
private:
|
|
class _Internal;
|
|
|
|
template <typename T> friend class ::PROTOBUF_NAMESPACE_ID::Arena::InternalHelper;
|
|
typedef void InternalArenaConstructable_;
|
|
typedef void DestructorSkippable_;
|
|
struct Impl_ {
|
|
};
|
|
friend struct ::TableStruct_xla_2fxla_2eproto;
|
|
};
|
|
// -------------------------------------------------------------------
|
|
|
|
class ComputationGraphStatsRequest final :
|
|
public ::PROTOBUF_NAMESPACE_ID::Message /* @@protoc_insertion_point(class_definition:xla.ComputationGraphStatsRequest) */ {
|
|
public:
|
|
inline ComputationGraphStatsRequest() : ComputationGraphStatsRequest(nullptr) {}
|
|
~ComputationGraphStatsRequest() override;
|
|
explicit PROTOBUF_CONSTEXPR ComputationGraphStatsRequest(::PROTOBUF_NAMESPACE_ID::internal::ConstantInitialized);
|
|
|
|
ComputationGraphStatsRequest(const ComputationGraphStatsRequest& from);
|
|
ComputationGraphStatsRequest(ComputationGraphStatsRequest&& from) noexcept
|
|
: ComputationGraphStatsRequest() {
|
|
*this = ::std::move(from);
|
|
}
|
|
|
|
inline ComputationGraphStatsRequest& operator=(const ComputationGraphStatsRequest& from) {
|
|
CopyFrom(from);
|
|
return *this;
|
|
}
|
|
inline ComputationGraphStatsRequest& operator=(ComputationGraphStatsRequest&& 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 ComputationGraphStatsRequest& default_instance() {
|
|
return *internal_default_instance();
|
|
}
|
|
static inline const ComputationGraphStatsRequest* internal_default_instance() {
|
|
return reinterpret_cast<const ComputationGraphStatsRequest*>(
|
|
&_ComputationGraphStatsRequest_default_instance_);
|
|
}
|
|
static constexpr int kIndexInFileMessages =
|
|
25;
|
|
|
|
friend void swap(ComputationGraphStatsRequest& a, ComputationGraphStatsRequest& b) {
|
|
a.Swap(&b);
|
|
}
|
|
inline void Swap(ComputationGraphStatsRequest* 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(ComputationGraphStatsRequest* other) {
|
|
if (other == this) return;
|
|
GOOGLE_DCHECK(GetOwningArena() == other->GetOwningArena());
|
|
InternalSwap(other);
|
|
}
|
|
|
|
// implements Message ----------------------------------------------
|
|
|
|
ComputationGraphStatsRequest* New(::PROTOBUF_NAMESPACE_ID::Arena* arena = nullptr) const final {
|
|
return CreateMaybeMessage<ComputationGraphStatsRequest>(arena);
|
|
}
|
|
using ::PROTOBUF_NAMESPACE_ID::Message::CopyFrom;
|
|
void CopyFrom(const ComputationGraphStatsRequest& from);
|
|
using ::PROTOBUF_NAMESPACE_ID::Message::MergeFrom;
|
|
void MergeFrom( const ComputationGraphStatsRequest& from) {
|
|
ComputationGraphStatsRequest::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(ComputationGraphStatsRequest* other);
|
|
|
|
private:
|
|
friend class ::PROTOBUF_NAMESPACE_ID::internal::AnyMetadata;
|
|
static ::PROTOBUF_NAMESPACE_ID::StringPiece FullMessageName() {
|
|
return "xla.ComputationGraphStatsRequest";
|
|
}
|
|
protected:
|
|
explicit ComputationGraphStatsRequest(::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 {
|
|
kComputationFieldNumber = 1,
|
|
kDebugOptionsFieldNumber = 2,
|
|
};
|
|
// .xla.HloModuleProto computation = 1;
|
|
bool has_computation() const;
|
|
private:
|
|
bool _internal_has_computation() const;
|
|
public:
|
|
void clear_computation();
|
|
const ::xla::HloModuleProto& computation() const;
|
|
PROTOBUF_NODISCARD ::xla::HloModuleProto* release_computation();
|
|
::xla::HloModuleProto* mutable_computation();
|
|
void set_allocated_computation(::xla::HloModuleProto* computation);
|
|
private:
|
|
const ::xla::HloModuleProto& _internal_computation() const;
|
|
::xla::HloModuleProto* _internal_mutable_computation();
|
|
public:
|
|
void unsafe_arena_set_allocated_computation(
|
|
::xla::HloModuleProto* computation);
|
|
::xla::HloModuleProto* unsafe_arena_release_computation();
|
|
|
|
// .xla.DebugOptions debug_options = 2;
|
|
bool has_debug_options() const;
|
|
private:
|
|
bool _internal_has_debug_options() const;
|
|
public:
|
|
void clear_debug_options();
|
|
const ::xla::DebugOptions& debug_options() const;
|
|
PROTOBUF_NODISCARD ::xla::DebugOptions* release_debug_options();
|
|
::xla::DebugOptions* mutable_debug_options();
|
|
void set_allocated_debug_options(::xla::DebugOptions* debug_options);
|
|
private:
|
|
const ::xla::DebugOptions& _internal_debug_options() const;
|
|
::xla::DebugOptions* _internal_mutable_debug_options();
|
|
public:
|
|
void unsafe_arena_set_allocated_debug_options(
|
|
::xla::DebugOptions* debug_options);
|
|
::xla::DebugOptions* unsafe_arena_release_debug_options();
|
|
|
|
// @@protoc_insertion_point(class_scope:xla.ComputationGraphStatsRequest)
|
|
private:
|
|
class _Internal;
|
|
|
|
template <typename T> friend class ::PROTOBUF_NAMESPACE_ID::Arena::InternalHelper;
|
|
typedef void InternalArenaConstructable_;
|
|
typedef void DestructorSkippable_;
|
|
struct Impl_ {
|
|
::xla::HloModuleProto* computation_;
|
|
::xla::DebugOptions* debug_options_;
|
|
mutable ::PROTOBUF_NAMESPACE_ID::internal::CachedSize _cached_size_;
|
|
};
|
|
union { Impl_ _impl_; };
|
|
friend struct ::TableStruct_xla_2fxla_2eproto;
|
|
};
|
|
// -------------------------------------------------------------------
|
|
|
|
class ComputationStatsResponse final :
|
|
public ::PROTOBUF_NAMESPACE_ID::Message /* @@protoc_insertion_point(class_definition:xla.ComputationStatsResponse) */ {
|
|
public:
|
|
inline ComputationStatsResponse() : ComputationStatsResponse(nullptr) {}
|
|
~ComputationStatsResponse() override;
|
|
explicit PROTOBUF_CONSTEXPR ComputationStatsResponse(::PROTOBUF_NAMESPACE_ID::internal::ConstantInitialized);
|
|
|
|
ComputationStatsResponse(const ComputationStatsResponse& from);
|
|
ComputationStatsResponse(ComputationStatsResponse&& from) noexcept
|
|
: ComputationStatsResponse() {
|
|
*this = ::std::move(from);
|
|
}
|
|
|
|
inline ComputationStatsResponse& operator=(const ComputationStatsResponse& from) {
|
|
CopyFrom(from);
|
|
return *this;
|
|
}
|
|
inline ComputationStatsResponse& operator=(ComputationStatsResponse&& 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 ComputationStatsResponse& default_instance() {
|
|
return *internal_default_instance();
|
|
}
|
|
static inline const ComputationStatsResponse* internal_default_instance() {
|
|
return reinterpret_cast<const ComputationStatsResponse*>(
|
|
&_ComputationStatsResponse_default_instance_);
|
|
}
|
|
static constexpr int kIndexInFileMessages =
|
|
26;
|
|
|
|
friend void swap(ComputationStatsResponse& a, ComputationStatsResponse& b) {
|
|
a.Swap(&b);
|
|
}
|
|
inline void Swap(ComputationStatsResponse* 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(ComputationStatsResponse* other) {
|
|
if (other == this) return;
|
|
GOOGLE_DCHECK(GetOwningArena() == other->GetOwningArena());
|
|
InternalSwap(other);
|
|
}
|
|
|
|
// implements Message ----------------------------------------------
|
|
|
|
ComputationStatsResponse* New(::PROTOBUF_NAMESPACE_ID::Arena* arena = nullptr) const final {
|
|
return CreateMaybeMessage<ComputationStatsResponse>(arena);
|
|
}
|
|
using ::PROTOBUF_NAMESPACE_ID::Message::CopyFrom;
|
|
void CopyFrom(const ComputationStatsResponse& from);
|
|
using ::PROTOBUF_NAMESPACE_ID::Message::MergeFrom;
|
|
void MergeFrom( const ComputationStatsResponse& from) {
|
|
ComputationStatsResponse::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(ComputationStatsResponse* other);
|
|
|
|
private:
|
|
friend class ::PROTOBUF_NAMESPACE_ID::internal::AnyMetadata;
|
|
static ::PROTOBUF_NAMESPACE_ID::StringPiece FullMessageName() {
|
|
return "xla.ComputationStatsResponse";
|
|
}
|
|
protected:
|
|
explicit ComputationStatsResponse(::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 {
|
|
kStatsFieldNumber = 1,
|
|
};
|
|
// .xla.ComputationStats stats = 1;
|
|
bool has_stats() const;
|
|
private:
|
|
bool _internal_has_stats() const;
|
|
public:
|
|
void clear_stats();
|
|
const ::xla::ComputationStats& stats() const;
|
|
PROTOBUF_NODISCARD ::xla::ComputationStats* release_stats();
|
|
::xla::ComputationStats* mutable_stats();
|
|
void set_allocated_stats(::xla::ComputationStats* stats);
|
|
private:
|
|
const ::xla::ComputationStats& _internal_stats() const;
|
|
::xla::ComputationStats* _internal_mutable_stats();
|
|
public:
|
|
void unsafe_arena_set_allocated_stats(
|
|
::xla::ComputationStats* stats);
|
|
::xla::ComputationStats* unsafe_arena_release_stats();
|
|
|
|
// @@protoc_insertion_point(class_scope:xla.ComputationStatsResponse)
|
|
private:
|
|
class _Internal;
|
|
|
|
template <typename T> friend class ::PROTOBUF_NAMESPACE_ID::Arena::InternalHelper;
|
|
typedef void InternalArenaConstructable_;
|
|
typedef void DestructorSkippable_;
|
|
struct Impl_ {
|
|
::xla::ComputationStats* stats_;
|
|
mutable ::PROTOBUF_NAMESPACE_ID::internal::CachedSize _cached_size_;
|
|
};
|
|
union { Impl_ _impl_; };
|
|
friend struct ::TableStruct_xla_2fxla_2eproto;
|
|
};
|
|
// -------------------------------------------------------------------
|
|
|
|
class CreateChannelHandleRequest final :
|
|
public ::PROTOBUF_NAMESPACE_ID::Message /* @@protoc_insertion_point(class_definition:xla.CreateChannelHandleRequest) */ {
|
|
public:
|
|
inline CreateChannelHandleRequest() : CreateChannelHandleRequest(nullptr) {}
|
|
~CreateChannelHandleRequest() override;
|
|
explicit PROTOBUF_CONSTEXPR CreateChannelHandleRequest(::PROTOBUF_NAMESPACE_ID::internal::ConstantInitialized);
|
|
|
|
CreateChannelHandleRequest(const CreateChannelHandleRequest& from);
|
|
CreateChannelHandleRequest(CreateChannelHandleRequest&& from) noexcept
|
|
: CreateChannelHandleRequest() {
|
|
*this = ::std::move(from);
|
|
}
|
|
|
|
inline CreateChannelHandleRequest& operator=(const CreateChannelHandleRequest& from) {
|
|
CopyFrom(from);
|
|
return *this;
|
|
}
|
|
inline CreateChannelHandleRequest& operator=(CreateChannelHandleRequest&& 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 CreateChannelHandleRequest& default_instance() {
|
|
return *internal_default_instance();
|
|
}
|
|
static inline const CreateChannelHandleRequest* internal_default_instance() {
|
|
return reinterpret_cast<const CreateChannelHandleRequest*>(
|
|
&_CreateChannelHandleRequest_default_instance_);
|
|
}
|
|
static constexpr int kIndexInFileMessages =
|
|
27;
|
|
|
|
friend void swap(CreateChannelHandleRequest& a, CreateChannelHandleRequest& b) {
|
|
a.Swap(&b);
|
|
}
|
|
inline void Swap(CreateChannelHandleRequest* 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(CreateChannelHandleRequest* other) {
|
|
if (other == this) return;
|
|
GOOGLE_DCHECK(GetOwningArena() == other->GetOwningArena());
|
|
InternalSwap(other);
|
|
}
|
|
|
|
// implements Message ----------------------------------------------
|
|
|
|
CreateChannelHandleRequest* New(::PROTOBUF_NAMESPACE_ID::Arena* arena = nullptr) const final {
|
|
return CreateMaybeMessage<CreateChannelHandleRequest>(arena);
|
|
}
|
|
using ::PROTOBUF_NAMESPACE_ID::Message::CopyFrom;
|
|
void CopyFrom(const CreateChannelHandleRequest& from);
|
|
using ::PROTOBUF_NAMESPACE_ID::Message::MergeFrom;
|
|
void MergeFrom( const CreateChannelHandleRequest& from) {
|
|
CreateChannelHandleRequest::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(CreateChannelHandleRequest* other);
|
|
|
|
private:
|
|
friend class ::PROTOBUF_NAMESPACE_ID::internal::AnyMetadata;
|
|
static ::PROTOBUF_NAMESPACE_ID::StringPiece FullMessageName() {
|
|
return "xla.CreateChannelHandleRequest";
|
|
}
|
|
protected:
|
|
explicit CreateChannelHandleRequest(::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 {
|
|
kChannelTypeFieldNumber = 1,
|
|
};
|
|
// .xla.ChannelHandle.ChannelType channel_type = 1;
|
|
void clear_channel_type();
|
|
::xla::ChannelHandle_ChannelType channel_type() const;
|
|
void set_channel_type(::xla::ChannelHandle_ChannelType value);
|
|
private:
|
|
::xla::ChannelHandle_ChannelType _internal_channel_type() const;
|
|
void _internal_set_channel_type(::xla::ChannelHandle_ChannelType value);
|
|
public:
|
|
|
|
// @@protoc_insertion_point(class_scope:xla.CreateChannelHandleRequest)
|
|
private:
|
|
class _Internal;
|
|
|
|
template <typename T> friend class ::PROTOBUF_NAMESPACE_ID::Arena::InternalHelper;
|
|
typedef void InternalArenaConstructable_;
|
|
typedef void DestructorSkippable_;
|
|
struct Impl_ {
|
|
int channel_type_;
|
|
mutable ::PROTOBUF_NAMESPACE_ID::internal::CachedSize _cached_size_;
|
|
};
|
|
union { Impl_ _impl_; };
|
|
friend struct ::TableStruct_xla_2fxla_2eproto;
|
|
};
|
|
// -------------------------------------------------------------------
|
|
|
|
class CreateChannelHandleResponse final :
|
|
public ::PROTOBUF_NAMESPACE_ID::Message /* @@protoc_insertion_point(class_definition:xla.CreateChannelHandleResponse) */ {
|
|
public:
|
|
inline CreateChannelHandleResponse() : CreateChannelHandleResponse(nullptr) {}
|
|
~CreateChannelHandleResponse() override;
|
|
explicit PROTOBUF_CONSTEXPR CreateChannelHandleResponse(::PROTOBUF_NAMESPACE_ID::internal::ConstantInitialized);
|
|
|
|
CreateChannelHandleResponse(const CreateChannelHandleResponse& from);
|
|
CreateChannelHandleResponse(CreateChannelHandleResponse&& from) noexcept
|
|
: CreateChannelHandleResponse() {
|
|
*this = ::std::move(from);
|
|
}
|
|
|
|
inline CreateChannelHandleResponse& operator=(const CreateChannelHandleResponse& from) {
|
|
CopyFrom(from);
|
|
return *this;
|
|
}
|
|
inline CreateChannelHandleResponse& operator=(CreateChannelHandleResponse&& 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 CreateChannelHandleResponse& default_instance() {
|
|
return *internal_default_instance();
|
|
}
|
|
static inline const CreateChannelHandleResponse* internal_default_instance() {
|
|
return reinterpret_cast<const CreateChannelHandleResponse*>(
|
|
&_CreateChannelHandleResponse_default_instance_);
|
|
}
|
|
static constexpr int kIndexInFileMessages =
|
|
28;
|
|
|
|
friend void swap(CreateChannelHandleResponse& a, CreateChannelHandleResponse& b) {
|
|
a.Swap(&b);
|
|
}
|
|
inline void Swap(CreateChannelHandleResponse* 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(CreateChannelHandleResponse* other) {
|
|
if (other == this) return;
|
|
GOOGLE_DCHECK(GetOwningArena() == other->GetOwningArena());
|
|
InternalSwap(other);
|
|
}
|
|
|
|
// implements Message ----------------------------------------------
|
|
|
|
CreateChannelHandleResponse* New(::PROTOBUF_NAMESPACE_ID::Arena* arena = nullptr) const final {
|
|
return CreateMaybeMessage<CreateChannelHandleResponse>(arena);
|
|
}
|
|
using ::PROTOBUF_NAMESPACE_ID::Message::CopyFrom;
|
|
void CopyFrom(const CreateChannelHandleResponse& from);
|
|
using ::PROTOBUF_NAMESPACE_ID::Message::MergeFrom;
|
|
void MergeFrom( const CreateChannelHandleResponse& from) {
|
|
CreateChannelHandleResponse::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(CreateChannelHandleResponse* other);
|
|
|
|
private:
|
|
friend class ::PROTOBUF_NAMESPACE_ID::internal::AnyMetadata;
|
|
static ::PROTOBUF_NAMESPACE_ID::StringPiece FullMessageName() {
|
|
return "xla.CreateChannelHandleResponse";
|
|
}
|
|
protected:
|
|
explicit CreateChannelHandleResponse(::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 {
|
|
kChannelFieldNumber = 1,
|
|
};
|
|
// .xla.ChannelHandle channel = 1;
|
|
bool has_channel() const;
|
|
private:
|
|
bool _internal_has_channel() const;
|
|
public:
|
|
void clear_channel();
|
|
const ::xla::ChannelHandle& channel() const;
|
|
PROTOBUF_NODISCARD ::xla::ChannelHandle* release_channel();
|
|
::xla::ChannelHandle* mutable_channel();
|
|
void set_allocated_channel(::xla::ChannelHandle* channel);
|
|
private:
|
|
const ::xla::ChannelHandle& _internal_channel() const;
|
|
::xla::ChannelHandle* _internal_mutable_channel();
|
|
public:
|
|
void unsafe_arena_set_allocated_channel(
|
|
::xla::ChannelHandle* channel);
|
|
::xla::ChannelHandle* unsafe_arena_release_channel();
|
|
|
|
// @@protoc_insertion_point(class_scope:xla.CreateChannelHandleResponse)
|
|
private:
|
|
class _Internal;
|
|
|
|
template <typename T> friend class ::PROTOBUF_NAMESPACE_ID::Arena::InternalHelper;
|
|
typedef void InternalArenaConstructable_;
|
|
typedef void DestructorSkippable_;
|
|
struct Impl_ {
|
|
::xla::ChannelHandle* channel_;
|
|
mutable ::PROTOBUF_NAMESPACE_ID::internal::CachedSize _cached_size_;
|
|
};
|
|
union { Impl_ _impl_; };
|
|
friend struct ::TableStruct_xla_2fxla_2eproto;
|
|
};
|
|
// -------------------------------------------------------------------
|
|
|
|
class UnregisterRequest final :
|
|
public ::PROTOBUF_NAMESPACE_ID::Message /* @@protoc_insertion_point(class_definition:xla.UnregisterRequest) */ {
|
|
public:
|
|
inline UnregisterRequest() : UnregisterRequest(nullptr) {}
|
|
~UnregisterRequest() override;
|
|
explicit PROTOBUF_CONSTEXPR UnregisterRequest(::PROTOBUF_NAMESPACE_ID::internal::ConstantInitialized);
|
|
|
|
UnregisterRequest(const UnregisterRequest& from);
|
|
UnregisterRequest(UnregisterRequest&& from) noexcept
|
|
: UnregisterRequest() {
|
|
*this = ::std::move(from);
|
|
}
|
|
|
|
inline UnregisterRequest& operator=(const UnregisterRequest& from) {
|
|
CopyFrom(from);
|
|
return *this;
|
|
}
|
|
inline UnregisterRequest& operator=(UnregisterRequest&& 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 UnregisterRequest& default_instance() {
|
|
return *internal_default_instance();
|
|
}
|
|
static inline const UnregisterRequest* internal_default_instance() {
|
|
return reinterpret_cast<const UnregisterRequest*>(
|
|
&_UnregisterRequest_default_instance_);
|
|
}
|
|
static constexpr int kIndexInFileMessages =
|
|
29;
|
|
|
|
friend void swap(UnregisterRequest& a, UnregisterRequest& b) {
|
|
a.Swap(&b);
|
|
}
|
|
inline void Swap(UnregisterRequest* 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(UnregisterRequest* other) {
|
|
if (other == this) return;
|
|
GOOGLE_DCHECK(GetOwningArena() == other->GetOwningArena());
|
|
InternalSwap(other);
|
|
}
|
|
|
|
// implements Message ----------------------------------------------
|
|
|
|
UnregisterRequest* New(::PROTOBUF_NAMESPACE_ID::Arena* arena = nullptr) const final {
|
|
return CreateMaybeMessage<UnregisterRequest>(arena);
|
|
}
|
|
using ::PROTOBUF_NAMESPACE_ID::Message::CopyFrom;
|
|
void CopyFrom(const UnregisterRequest& from);
|
|
using ::PROTOBUF_NAMESPACE_ID::Message::MergeFrom;
|
|
void MergeFrom( const UnregisterRequest& from) {
|
|
UnregisterRequest::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(UnregisterRequest* other);
|
|
|
|
private:
|
|
friend class ::PROTOBUF_NAMESPACE_ID::internal::AnyMetadata;
|
|
static ::PROTOBUF_NAMESPACE_ID::StringPiece FullMessageName() {
|
|
return "xla.UnregisterRequest";
|
|
}
|
|
protected:
|
|
explicit UnregisterRequest(::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 {
|
|
kDataFieldNumber = 1,
|
|
};
|
|
// repeated .xla.GlobalDataHandle data = 1;
|
|
int data_size() const;
|
|
private:
|
|
int _internal_data_size() const;
|
|
public:
|
|
void clear_data();
|
|
::xla::GlobalDataHandle* mutable_data(int index);
|
|
::PROTOBUF_NAMESPACE_ID::RepeatedPtrField< ::xla::GlobalDataHandle >*
|
|
mutable_data();
|
|
private:
|
|
const ::xla::GlobalDataHandle& _internal_data(int index) const;
|
|
::xla::GlobalDataHandle* _internal_add_data();
|
|
public:
|
|
const ::xla::GlobalDataHandle& data(int index) const;
|
|
::xla::GlobalDataHandle* add_data();
|
|
const ::PROTOBUF_NAMESPACE_ID::RepeatedPtrField< ::xla::GlobalDataHandle >&
|
|
data() const;
|
|
|
|
// @@protoc_insertion_point(class_scope:xla.UnregisterRequest)
|
|
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::GlobalDataHandle > data_;
|
|
mutable ::PROTOBUF_NAMESPACE_ID::internal::CachedSize _cached_size_;
|
|
};
|
|
union { Impl_ _impl_; };
|
|
friend struct ::TableStruct_xla_2fxla_2eproto;
|
|
};
|
|
// -------------------------------------------------------------------
|
|
|
|
class UnregisterResponse final :
|
|
public ::PROTOBUF_NAMESPACE_ID::internal::ZeroFieldsBase /* @@protoc_insertion_point(class_definition:xla.UnregisterResponse) */ {
|
|
public:
|
|
inline UnregisterResponse() : UnregisterResponse(nullptr) {}
|
|
explicit PROTOBUF_CONSTEXPR UnregisterResponse(::PROTOBUF_NAMESPACE_ID::internal::ConstantInitialized);
|
|
|
|
UnregisterResponse(const UnregisterResponse& from);
|
|
UnregisterResponse(UnregisterResponse&& from) noexcept
|
|
: UnregisterResponse() {
|
|
*this = ::std::move(from);
|
|
}
|
|
|
|
inline UnregisterResponse& operator=(const UnregisterResponse& from) {
|
|
CopyFrom(from);
|
|
return *this;
|
|
}
|
|
inline UnregisterResponse& operator=(UnregisterResponse&& 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 UnregisterResponse& default_instance() {
|
|
return *internal_default_instance();
|
|
}
|
|
static inline const UnregisterResponse* internal_default_instance() {
|
|
return reinterpret_cast<const UnregisterResponse*>(
|
|
&_UnregisterResponse_default_instance_);
|
|
}
|
|
static constexpr int kIndexInFileMessages =
|
|
30;
|
|
|
|
friend void swap(UnregisterResponse& a, UnregisterResponse& b) {
|
|
a.Swap(&b);
|
|
}
|
|
inline void Swap(UnregisterResponse* 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(UnregisterResponse* other) {
|
|
if (other == this) return;
|
|
GOOGLE_DCHECK(GetOwningArena() == other->GetOwningArena());
|
|
InternalSwap(other);
|
|
}
|
|
|
|
// implements Message ----------------------------------------------
|
|
|
|
UnregisterResponse* New(::PROTOBUF_NAMESPACE_ID::Arena* arena = nullptr) const final {
|
|
return CreateMaybeMessage<UnregisterResponse>(arena);
|
|
}
|
|
using ::PROTOBUF_NAMESPACE_ID::internal::ZeroFieldsBase::CopyFrom;
|
|
inline void CopyFrom(const UnregisterResponse& from) {
|
|
::PROTOBUF_NAMESPACE_ID::internal::ZeroFieldsBase::CopyImpl(*this, from);
|
|
}
|
|
using ::PROTOBUF_NAMESPACE_ID::internal::ZeroFieldsBase::MergeFrom;
|
|
void MergeFrom(const UnregisterResponse& from) {
|
|
::PROTOBUF_NAMESPACE_ID::internal::ZeroFieldsBase::MergeImpl(*this, from);
|
|
}
|
|
public:
|
|
|
|
private:
|
|
friend class ::PROTOBUF_NAMESPACE_ID::internal::AnyMetadata;
|
|
static ::PROTOBUF_NAMESPACE_ID::StringPiece FullMessageName() {
|
|
return "xla.UnregisterResponse";
|
|
}
|
|
protected:
|
|
explicit UnregisterResponse(::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 -------------------------------------------------------
|
|
|
|
// @@protoc_insertion_point(class_scope:xla.UnregisterResponse)
|
|
private:
|
|
class _Internal;
|
|
|
|
template <typename T> friend class ::PROTOBUF_NAMESPACE_ID::Arena::InternalHelper;
|
|
typedef void InternalArenaConstructable_;
|
|
typedef void DestructorSkippable_;
|
|
struct Impl_ {
|
|
};
|
|
friend struct ::TableStruct_xla_2fxla_2eproto;
|
|
};
|
|
// -------------------------------------------------------------------
|
|
|
|
class CompileRequest final :
|
|
public ::PROTOBUF_NAMESPACE_ID::Message /* @@protoc_insertion_point(class_definition:xla.CompileRequest) */ {
|
|
public:
|
|
inline CompileRequest() : CompileRequest(nullptr) {}
|
|
~CompileRequest() override;
|
|
explicit PROTOBUF_CONSTEXPR CompileRequest(::PROTOBUF_NAMESPACE_ID::internal::ConstantInitialized);
|
|
|
|
CompileRequest(const CompileRequest& from);
|
|
CompileRequest(CompileRequest&& from) noexcept
|
|
: CompileRequest() {
|
|
*this = ::std::move(from);
|
|
}
|
|
|
|
inline CompileRequest& operator=(const CompileRequest& from) {
|
|
CopyFrom(from);
|
|
return *this;
|
|
}
|
|
inline CompileRequest& operator=(CompileRequest&& 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 CompileRequest& default_instance() {
|
|
return *internal_default_instance();
|
|
}
|
|
static inline const CompileRequest* internal_default_instance() {
|
|
return reinterpret_cast<const CompileRequest*>(
|
|
&_CompileRequest_default_instance_);
|
|
}
|
|
static constexpr int kIndexInFileMessages =
|
|
31;
|
|
|
|
friend void swap(CompileRequest& a, CompileRequest& b) {
|
|
a.Swap(&b);
|
|
}
|
|
inline void Swap(CompileRequest* 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(CompileRequest* other) {
|
|
if (other == this) return;
|
|
GOOGLE_DCHECK(GetOwningArena() == other->GetOwningArena());
|
|
InternalSwap(other);
|
|
}
|
|
|
|
// implements Message ----------------------------------------------
|
|
|
|
CompileRequest* New(::PROTOBUF_NAMESPACE_ID::Arena* arena = nullptr) const final {
|
|
return CreateMaybeMessage<CompileRequest>(arena);
|
|
}
|
|
using ::PROTOBUF_NAMESPACE_ID::Message::CopyFrom;
|
|
void CopyFrom(const CompileRequest& from);
|
|
using ::PROTOBUF_NAMESPACE_ID::Message::MergeFrom;
|
|
void MergeFrom( const CompileRequest& from) {
|
|
CompileRequest::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(CompileRequest* other);
|
|
|
|
private:
|
|
friend class ::PROTOBUF_NAMESPACE_ID::internal::AnyMetadata;
|
|
static ::PROTOBUF_NAMESPACE_ID::StringPiece FullMessageName() {
|
|
return "xla.CompileRequest";
|
|
}
|
|
protected:
|
|
explicit CompileRequest(::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 {
|
|
kInputShapeWithLayoutFieldNumber = 3,
|
|
kComputationFieldNumber = 1,
|
|
kExecutionOptionsFieldNumber = 2,
|
|
};
|
|
// repeated .xla.ShapeProto input_shape_with_layout = 3;
|
|
int input_shape_with_layout_size() const;
|
|
private:
|
|
int _internal_input_shape_with_layout_size() const;
|
|
public:
|
|
void clear_input_shape_with_layout();
|
|
::xla::ShapeProto* mutable_input_shape_with_layout(int index);
|
|
::PROTOBUF_NAMESPACE_ID::RepeatedPtrField< ::xla::ShapeProto >*
|
|
mutable_input_shape_with_layout();
|
|
private:
|
|
const ::xla::ShapeProto& _internal_input_shape_with_layout(int index) const;
|
|
::xla::ShapeProto* _internal_add_input_shape_with_layout();
|
|
public:
|
|
const ::xla::ShapeProto& input_shape_with_layout(int index) const;
|
|
::xla::ShapeProto* add_input_shape_with_layout();
|
|
const ::PROTOBUF_NAMESPACE_ID::RepeatedPtrField< ::xla::ShapeProto >&
|
|
input_shape_with_layout() const;
|
|
|
|
// .xla.HloModuleProto computation = 1;
|
|
bool has_computation() const;
|
|
private:
|
|
bool _internal_has_computation() const;
|
|
public:
|
|
void clear_computation();
|
|
const ::xla::HloModuleProto& computation() const;
|
|
PROTOBUF_NODISCARD ::xla::HloModuleProto* release_computation();
|
|
::xla::HloModuleProto* mutable_computation();
|
|
void set_allocated_computation(::xla::HloModuleProto* computation);
|
|
private:
|
|
const ::xla::HloModuleProto& _internal_computation() const;
|
|
::xla::HloModuleProto* _internal_mutable_computation();
|
|
public:
|
|
void unsafe_arena_set_allocated_computation(
|
|
::xla::HloModuleProto* computation);
|
|
::xla::HloModuleProto* unsafe_arena_release_computation();
|
|
|
|
// .xla.ExecutionOptions execution_options = 2;
|
|
bool has_execution_options() const;
|
|
private:
|
|
bool _internal_has_execution_options() const;
|
|
public:
|
|
void clear_execution_options();
|
|
const ::xla::ExecutionOptions& execution_options() const;
|
|
PROTOBUF_NODISCARD ::xla::ExecutionOptions* release_execution_options();
|
|
::xla::ExecutionOptions* mutable_execution_options();
|
|
void set_allocated_execution_options(::xla::ExecutionOptions* execution_options);
|
|
private:
|
|
const ::xla::ExecutionOptions& _internal_execution_options() const;
|
|
::xla::ExecutionOptions* _internal_mutable_execution_options();
|
|
public:
|
|
void unsafe_arena_set_allocated_execution_options(
|
|
::xla::ExecutionOptions* execution_options);
|
|
::xla::ExecutionOptions* unsafe_arena_release_execution_options();
|
|
|
|
// @@protoc_insertion_point(class_scope:xla.CompileRequest)
|
|
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 > input_shape_with_layout_;
|
|
::xla::HloModuleProto* computation_;
|
|
::xla::ExecutionOptions* execution_options_;
|
|
mutable ::PROTOBUF_NAMESPACE_ID::internal::CachedSize _cached_size_;
|
|
};
|
|
union { Impl_ _impl_; };
|
|
friend struct ::TableStruct_xla_2fxla_2eproto;
|
|
};
|
|
// -------------------------------------------------------------------
|
|
|
|
class CompileResponse final :
|
|
public ::PROTOBUF_NAMESPACE_ID::Message /* @@protoc_insertion_point(class_definition:xla.CompileResponse) */ {
|
|
public:
|
|
inline CompileResponse() : CompileResponse(nullptr) {}
|
|
~CompileResponse() override;
|
|
explicit PROTOBUF_CONSTEXPR CompileResponse(::PROTOBUF_NAMESPACE_ID::internal::ConstantInitialized);
|
|
|
|
CompileResponse(const CompileResponse& from);
|
|
CompileResponse(CompileResponse&& from) noexcept
|
|
: CompileResponse() {
|
|
*this = ::std::move(from);
|
|
}
|
|
|
|
inline CompileResponse& operator=(const CompileResponse& from) {
|
|
CopyFrom(from);
|
|
return *this;
|
|
}
|
|
inline CompileResponse& operator=(CompileResponse&& 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 CompileResponse& default_instance() {
|
|
return *internal_default_instance();
|
|
}
|
|
static inline const CompileResponse* internal_default_instance() {
|
|
return reinterpret_cast<const CompileResponse*>(
|
|
&_CompileResponse_default_instance_);
|
|
}
|
|
static constexpr int kIndexInFileMessages =
|
|
32;
|
|
|
|
friend void swap(CompileResponse& a, CompileResponse& b) {
|
|
a.Swap(&b);
|
|
}
|
|
inline void Swap(CompileResponse* 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(CompileResponse* other) {
|
|
if (other == this) return;
|
|
GOOGLE_DCHECK(GetOwningArena() == other->GetOwningArena());
|
|
InternalSwap(other);
|
|
}
|
|
|
|
// implements Message ----------------------------------------------
|
|
|
|
CompileResponse* New(::PROTOBUF_NAMESPACE_ID::Arena* arena = nullptr) const final {
|
|
return CreateMaybeMessage<CompileResponse>(arena);
|
|
}
|
|
using ::PROTOBUF_NAMESPACE_ID::Message::CopyFrom;
|
|
void CopyFrom(const CompileResponse& from);
|
|
using ::PROTOBUF_NAMESPACE_ID::Message::MergeFrom;
|
|
void MergeFrom( const CompileResponse& from) {
|
|
CompileResponse::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(CompileResponse* other);
|
|
|
|
private:
|
|
friend class ::PROTOBUF_NAMESPACE_ID::internal::AnyMetadata;
|
|
static ::PROTOBUF_NAMESPACE_ID::StringPiece FullMessageName() {
|
|
return "xla.CompileResponse";
|
|
}
|
|
protected:
|
|
explicit CompileResponse(::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,
|
|
};
|
|
// .xla.ExecutionHandle handle = 1;
|
|
bool has_handle() const;
|
|
private:
|
|
bool _internal_has_handle() const;
|
|
public:
|
|
void clear_handle();
|
|
const ::xla::ExecutionHandle& handle() const;
|
|
PROTOBUF_NODISCARD ::xla::ExecutionHandle* release_handle();
|
|
::xla::ExecutionHandle* mutable_handle();
|
|
void set_allocated_handle(::xla::ExecutionHandle* handle);
|
|
private:
|
|
const ::xla::ExecutionHandle& _internal_handle() const;
|
|
::xla::ExecutionHandle* _internal_mutable_handle();
|
|
public:
|
|
void unsafe_arena_set_allocated_handle(
|
|
::xla::ExecutionHandle* handle);
|
|
::xla::ExecutionHandle* unsafe_arena_release_handle();
|
|
|
|
// @@protoc_insertion_point(class_scope:xla.CompileResponse)
|
|
private:
|
|
class _Internal;
|
|
|
|
template <typename T> friend class ::PROTOBUF_NAMESPACE_ID::Arena::InternalHelper;
|
|
typedef void InternalArenaConstructable_;
|
|
typedef void DestructorSkippable_;
|
|
struct Impl_ {
|
|
::xla::ExecutionHandle* handle_;
|
|
mutable ::PROTOBUF_NAMESPACE_ID::internal::CachedSize _cached_size_;
|
|
};
|
|
union { Impl_ _impl_; };
|
|
friend struct ::TableStruct_xla_2fxla_2eproto;
|
|
};
|
|
// -------------------------------------------------------------------
|
|
|
|
class ExecuteRequest final :
|
|
public ::PROTOBUF_NAMESPACE_ID::Message /* @@protoc_insertion_point(class_definition:xla.ExecuteRequest) */ {
|
|
public:
|
|
inline ExecuteRequest() : ExecuteRequest(nullptr) {}
|
|
~ExecuteRequest() override;
|
|
explicit PROTOBUF_CONSTEXPR ExecuteRequest(::PROTOBUF_NAMESPACE_ID::internal::ConstantInitialized);
|
|
|
|
ExecuteRequest(const ExecuteRequest& from);
|
|
ExecuteRequest(ExecuteRequest&& from) noexcept
|
|
: ExecuteRequest() {
|
|
*this = ::std::move(from);
|
|
}
|
|
|
|
inline ExecuteRequest& operator=(const ExecuteRequest& from) {
|
|
CopyFrom(from);
|
|
return *this;
|
|
}
|
|
inline ExecuteRequest& operator=(ExecuteRequest&& 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 ExecuteRequest& default_instance() {
|
|
return *internal_default_instance();
|
|
}
|
|
static inline const ExecuteRequest* internal_default_instance() {
|
|
return reinterpret_cast<const ExecuteRequest*>(
|
|
&_ExecuteRequest_default_instance_);
|
|
}
|
|
static constexpr int kIndexInFileMessages =
|
|
33;
|
|
|
|
friend void swap(ExecuteRequest& a, ExecuteRequest& b) {
|
|
a.Swap(&b);
|
|
}
|
|
inline void Swap(ExecuteRequest* 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(ExecuteRequest* other) {
|
|
if (other == this) return;
|
|
GOOGLE_DCHECK(GetOwningArena() == other->GetOwningArena());
|
|
InternalSwap(other);
|
|
}
|
|
|
|
// implements Message ----------------------------------------------
|
|
|
|
ExecuteRequest* New(::PROTOBUF_NAMESPACE_ID::Arena* arena = nullptr) const final {
|
|
return CreateMaybeMessage<ExecuteRequest>(arena);
|
|
}
|
|
using ::PROTOBUF_NAMESPACE_ID::Message::CopyFrom;
|
|
void CopyFrom(const ExecuteRequest& from);
|
|
using ::PROTOBUF_NAMESPACE_ID::Message::MergeFrom;
|
|
void MergeFrom( const ExecuteRequest& from) {
|
|
ExecuteRequest::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(ExecuteRequest* other);
|
|
|
|
private:
|
|
friend class ::PROTOBUF_NAMESPACE_ID::internal::AnyMetadata;
|
|
static ::PROTOBUF_NAMESPACE_ID::StringPiece FullMessageName() {
|
|
return "xla.ExecuteRequest";
|
|
}
|
|
protected:
|
|
explicit ExecuteRequest(::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 {
|
|
kArgumentsFieldNumber = 2,
|
|
kHandleFieldNumber = 1,
|
|
};
|
|
// repeated .xla.GlobalDataHandle arguments = 2;
|
|
int arguments_size() const;
|
|
private:
|
|
int _internal_arguments_size() const;
|
|
public:
|
|
void clear_arguments();
|
|
::xla::GlobalDataHandle* mutable_arguments(int index);
|
|
::PROTOBUF_NAMESPACE_ID::RepeatedPtrField< ::xla::GlobalDataHandle >*
|
|
mutable_arguments();
|
|
private:
|
|
const ::xla::GlobalDataHandle& _internal_arguments(int index) const;
|
|
::xla::GlobalDataHandle* _internal_add_arguments();
|
|
public:
|
|
const ::xla::GlobalDataHandle& arguments(int index) const;
|
|
::xla::GlobalDataHandle* add_arguments();
|
|
const ::PROTOBUF_NAMESPACE_ID::RepeatedPtrField< ::xla::GlobalDataHandle >&
|
|
arguments() const;
|
|
|
|
// .xla.ExecutionHandle handle = 1;
|
|
bool has_handle() const;
|
|
private:
|
|
bool _internal_has_handle() const;
|
|
public:
|
|
void clear_handle();
|
|
const ::xla::ExecutionHandle& handle() const;
|
|
PROTOBUF_NODISCARD ::xla::ExecutionHandle* release_handle();
|
|
::xla::ExecutionHandle* mutable_handle();
|
|
void set_allocated_handle(::xla::ExecutionHandle* handle);
|
|
private:
|
|
const ::xla::ExecutionHandle& _internal_handle() const;
|
|
::xla::ExecutionHandle* _internal_mutable_handle();
|
|
public:
|
|
void unsafe_arena_set_allocated_handle(
|
|
::xla::ExecutionHandle* handle);
|
|
::xla::ExecutionHandle* unsafe_arena_release_handle();
|
|
|
|
// @@protoc_insertion_point(class_scope:xla.ExecuteRequest)
|
|
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::GlobalDataHandle > arguments_;
|
|
::xla::ExecutionHandle* handle_;
|
|
mutable ::PROTOBUF_NAMESPACE_ID::internal::CachedSize _cached_size_;
|
|
};
|
|
union { Impl_ _impl_; };
|
|
friend struct ::TableStruct_xla_2fxla_2eproto;
|
|
};
|
|
// -------------------------------------------------------------------
|
|
|
|
class ExecuteGraphRequest final :
|
|
public ::PROTOBUF_NAMESPACE_ID::Message /* @@protoc_insertion_point(class_definition:xla.ExecuteGraphRequest) */ {
|
|
public:
|
|
inline ExecuteGraphRequest() : ExecuteGraphRequest(nullptr) {}
|
|
~ExecuteGraphRequest() override;
|
|
explicit PROTOBUF_CONSTEXPR ExecuteGraphRequest(::PROTOBUF_NAMESPACE_ID::internal::ConstantInitialized);
|
|
|
|
ExecuteGraphRequest(const ExecuteGraphRequest& from);
|
|
ExecuteGraphRequest(ExecuteGraphRequest&& from) noexcept
|
|
: ExecuteGraphRequest() {
|
|
*this = ::std::move(from);
|
|
}
|
|
|
|
inline ExecuteGraphRequest& operator=(const ExecuteGraphRequest& from) {
|
|
CopyFrom(from);
|
|
return *this;
|
|
}
|
|
inline ExecuteGraphRequest& operator=(ExecuteGraphRequest&& 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 ExecuteGraphRequest& default_instance() {
|
|
return *internal_default_instance();
|
|
}
|
|
static inline const ExecuteGraphRequest* internal_default_instance() {
|
|
return reinterpret_cast<const ExecuteGraphRequest*>(
|
|
&_ExecuteGraphRequest_default_instance_);
|
|
}
|
|
static constexpr int kIndexInFileMessages =
|
|
34;
|
|
|
|
friend void swap(ExecuteGraphRequest& a, ExecuteGraphRequest& b) {
|
|
a.Swap(&b);
|
|
}
|
|
inline void Swap(ExecuteGraphRequest* 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(ExecuteGraphRequest* other) {
|
|
if (other == this) return;
|
|
GOOGLE_DCHECK(GetOwningArena() == other->GetOwningArena());
|
|
InternalSwap(other);
|
|
}
|
|
|
|
// implements Message ----------------------------------------------
|
|
|
|
ExecuteGraphRequest* New(::PROTOBUF_NAMESPACE_ID::Arena* arena = nullptr) const final {
|
|
return CreateMaybeMessage<ExecuteGraphRequest>(arena);
|
|
}
|
|
using ::PROTOBUF_NAMESPACE_ID::Message::CopyFrom;
|
|
void CopyFrom(const ExecuteGraphRequest& from);
|
|
using ::PROTOBUF_NAMESPACE_ID::Message::MergeFrom;
|
|
void MergeFrom( const ExecuteGraphRequest& from) {
|
|
ExecuteGraphRequest::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(ExecuteGraphRequest* other);
|
|
|
|
private:
|
|
friend class ::PROTOBUF_NAMESPACE_ID::internal::AnyMetadata;
|
|
static ::PROTOBUF_NAMESPACE_ID::StringPiece FullMessageName() {
|
|
return "xla.ExecuteGraphRequest";
|
|
}
|
|
protected:
|
|
explicit ExecuteGraphRequest(::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 {
|
|
kArgumentsFieldNumber = 2,
|
|
kComputationFieldNumber = 1,
|
|
kExecutionOptionsFieldNumber = 3,
|
|
};
|
|
// repeated .xla.GlobalDataHandle arguments = 2;
|
|
int arguments_size() const;
|
|
private:
|
|
int _internal_arguments_size() const;
|
|
public:
|
|
void clear_arguments();
|
|
::xla::GlobalDataHandle* mutable_arguments(int index);
|
|
::PROTOBUF_NAMESPACE_ID::RepeatedPtrField< ::xla::GlobalDataHandle >*
|
|
mutable_arguments();
|
|
private:
|
|
const ::xla::GlobalDataHandle& _internal_arguments(int index) const;
|
|
::xla::GlobalDataHandle* _internal_add_arguments();
|
|
public:
|
|
const ::xla::GlobalDataHandle& arguments(int index) const;
|
|
::xla::GlobalDataHandle* add_arguments();
|
|
const ::PROTOBUF_NAMESPACE_ID::RepeatedPtrField< ::xla::GlobalDataHandle >&
|
|
arguments() const;
|
|
|
|
// .xla.HloModuleProto computation = 1;
|
|
bool has_computation() const;
|
|
private:
|
|
bool _internal_has_computation() const;
|
|
public:
|
|
void clear_computation();
|
|
const ::xla::HloModuleProto& computation() const;
|
|
PROTOBUF_NODISCARD ::xla::HloModuleProto* release_computation();
|
|
::xla::HloModuleProto* mutable_computation();
|
|
void set_allocated_computation(::xla::HloModuleProto* computation);
|
|
private:
|
|
const ::xla::HloModuleProto& _internal_computation() const;
|
|
::xla::HloModuleProto* _internal_mutable_computation();
|
|
public:
|
|
void unsafe_arena_set_allocated_computation(
|
|
::xla::HloModuleProto* computation);
|
|
::xla::HloModuleProto* unsafe_arena_release_computation();
|
|
|
|
// .xla.ExecutionOptions execution_options = 3;
|
|
bool has_execution_options() const;
|
|
private:
|
|
bool _internal_has_execution_options() const;
|
|
public:
|
|
void clear_execution_options();
|
|
const ::xla::ExecutionOptions& execution_options() const;
|
|
PROTOBUF_NODISCARD ::xla::ExecutionOptions* release_execution_options();
|
|
::xla::ExecutionOptions* mutable_execution_options();
|
|
void set_allocated_execution_options(::xla::ExecutionOptions* execution_options);
|
|
private:
|
|
const ::xla::ExecutionOptions& _internal_execution_options() const;
|
|
::xla::ExecutionOptions* _internal_mutable_execution_options();
|
|
public:
|
|
void unsafe_arena_set_allocated_execution_options(
|
|
::xla::ExecutionOptions* execution_options);
|
|
::xla::ExecutionOptions* unsafe_arena_release_execution_options();
|
|
|
|
// @@protoc_insertion_point(class_scope:xla.ExecuteGraphRequest)
|
|
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::GlobalDataHandle > arguments_;
|
|
::xla::HloModuleProto* computation_;
|
|
::xla::ExecutionOptions* execution_options_;
|
|
mutable ::PROTOBUF_NAMESPACE_ID::internal::CachedSize _cached_size_;
|
|
};
|
|
union { Impl_ _impl_; };
|
|
friend struct ::TableStruct_xla_2fxla_2eproto;
|
|
};
|
|
// -------------------------------------------------------------------
|
|
|
|
class ExecuteGraphParallelRequest final :
|
|
public ::PROTOBUF_NAMESPACE_ID::Message /* @@protoc_insertion_point(class_definition:xla.ExecuteGraphParallelRequest) */ {
|
|
public:
|
|
inline ExecuteGraphParallelRequest() : ExecuteGraphParallelRequest(nullptr) {}
|
|
~ExecuteGraphParallelRequest() override;
|
|
explicit PROTOBUF_CONSTEXPR ExecuteGraphParallelRequest(::PROTOBUF_NAMESPACE_ID::internal::ConstantInitialized);
|
|
|
|
ExecuteGraphParallelRequest(const ExecuteGraphParallelRequest& from);
|
|
ExecuteGraphParallelRequest(ExecuteGraphParallelRequest&& from) noexcept
|
|
: ExecuteGraphParallelRequest() {
|
|
*this = ::std::move(from);
|
|
}
|
|
|
|
inline ExecuteGraphParallelRequest& operator=(const ExecuteGraphParallelRequest& from) {
|
|
CopyFrom(from);
|
|
return *this;
|
|
}
|
|
inline ExecuteGraphParallelRequest& operator=(ExecuteGraphParallelRequest&& 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 ExecuteGraphParallelRequest& default_instance() {
|
|
return *internal_default_instance();
|
|
}
|
|
static inline const ExecuteGraphParallelRequest* internal_default_instance() {
|
|
return reinterpret_cast<const ExecuteGraphParallelRequest*>(
|
|
&_ExecuteGraphParallelRequest_default_instance_);
|
|
}
|
|
static constexpr int kIndexInFileMessages =
|
|
35;
|
|
|
|
friend void swap(ExecuteGraphParallelRequest& a, ExecuteGraphParallelRequest& b) {
|
|
a.Swap(&b);
|
|
}
|
|
inline void Swap(ExecuteGraphParallelRequest* 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(ExecuteGraphParallelRequest* other) {
|
|
if (other == this) return;
|
|
GOOGLE_DCHECK(GetOwningArena() == other->GetOwningArena());
|
|
InternalSwap(other);
|
|
}
|
|
|
|
// implements Message ----------------------------------------------
|
|
|
|
ExecuteGraphParallelRequest* New(::PROTOBUF_NAMESPACE_ID::Arena* arena = nullptr) const final {
|
|
return CreateMaybeMessage<ExecuteGraphParallelRequest>(arena);
|
|
}
|
|
using ::PROTOBUF_NAMESPACE_ID::Message::CopyFrom;
|
|
void CopyFrom(const ExecuteGraphParallelRequest& from);
|
|
using ::PROTOBUF_NAMESPACE_ID::Message::MergeFrom;
|
|
void MergeFrom( const ExecuteGraphParallelRequest& from) {
|
|
ExecuteGraphParallelRequest::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(ExecuteGraphParallelRequest* other);
|
|
|
|
private:
|
|
friend class ::PROTOBUF_NAMESPACE_ID::internal::AnyMetadata;
|
|
static ::PROTOBUF_NAMESPACE_ID::StringPiece FullMessageName() {
|
|
return "xla.ExecuteGraphParallelRequest";
|
|
}
|
|
protected:
|
|
explicit ExecuteGraphParallelRequest(::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 {
|
|
kRequestsFieldNumber = 1,
|
|
};
|
|
// repeated .xla.ExecuteGraphRequest requests = 1;
|
|
int requests_size() const;
|
|
private:
|
|
int _internal_requests_size() const;
|
|
public:
|
|
void clear_requests();
|
|
::xla::ExecuteGraphRequest* mutable_requests(int index);
|
|
::PROTOBUF_NAMESPACE_ID::RepeatedPtrField< ::xla::ExecuteGraphRequest >*
|
|
mutable_requests();
|
|
private:
|
|
const ::xla::ExecuteGraphRequest& _internal_requests(int index) const;
|
|
::xla::ExecuteGraphRequest* _internal_add_requests();
|
|
public:
|
|
const ::xla::ExecuteGraphRequest& requests(int index) const;
|
|
::xla::ExecuteGraphRequest* add_requests();
|
|
const ::PROTOBUF_NAMESPACE_ID::RepeatedPtrField< ::xla::ExecuteGraphRequest >&
|
|
requests() const;
|
|
|
|
// @@protoc_insertion_point(class_scope:xla.ExecuteGraphParallelRequest)
|
|
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::ExecuteGraphRequest > requests_;
|
|
mutable ::PROTOBUF_NAMESPACE_ID::internal::CachedSize _cached_size_;
|
|
};
|
|
union { Impl_ _impl_; };
|
|
friend struct ::TableStruct_xla_2fxla_2eproto;
|
|
};
|
|
// -------------------------------------------------------------------
|
|
|
|
class ExecuteResponse final :
|
|
public ::PROTOBUF_NAMESPACE_ID::Message /* @@protoc_insertion_point(class_definition:xla.ExecuteResponse) */ {
|
|
public:
|
|
inline ExecuteResponse() : ExecuteResponse(nullptr) {}
|
|
~ExecuteResponse() override;
|
|
explicit PROTOBUF_CONSTEXPR ExecuteResponse(::PROTOBUF_NAMESPACE_ID::internal::ConstantInitialized);
|
|
|
|
ExecuteResponse(const ExecuteResponse& from);
|
|
ExecuteResponse(ExecuteResponse&& from) noexcept
|
|
: ExecuteResponse() {
|
|
*this = ::std::move(from);
|
|
}
|
|
|
|
inline ExecuteResponse& operator=(const ExecuteResponse& from) {
|
|
CopyFrom(from);
|
|
return *this;
|
|
}
|
|
inline ExecuteResponse& operator=(ExecuteResponse&& 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 ExecuteResponse& default_instance() {
|
|
return *internal_default_instance();
|
|
}
|
|
static inline const ExecuteResponse* internal_default_instance() {
|
|
return reinterpret_cast<const ExecuteResponse*>(
|
|
&_ExecuteResponse_default_instance_);
|
|
}
|
|
static constexpr int kIndexInFileMessages =
|
|
36;
|
|
|
|
friend void swap(ExecuteResponse& a, ExecuteResponse& b) {
|
|
a.Swap(&b);
|
|
}
|
|
inline void Swap(ExecuteResponse* 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(ExecuteResponse* other) {
|
|
if (other == this) return;
|
|
GOOGLE_DCHECK(GetOwningArena() == other->GetOwningArena());
|
|
InternalSwap(other);
|
|
}
|
|
|
|
// implements Message ----------------------------------------------
|
|
|
|
ExecuteResponse* New(::PROTOBUF_NAMESPACE_ID::Arena* arena = nullptr) const final {
|
|
return CreateMaybeMessage<ExecuteResponse>(arena);
|
|
}
|
|
using ::PROTOBUF_NAMESPACE_ID::Message::CopyFrom;
|
|
void CopyFrom(const ExecuteResponse& from);
|
|
using ::PROTOBUF_NAMESPACE_ID::Message::MergeFrom;
|
|
void MergeFrom( const ExecuteResponse& from) {
|
|
ExecuteResponse::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(ExecuteResponse* other);
|
|
|
|
private:
|
|
friend class ::PROTOBUF_NAMESPACE_ID::internal::AnyMetadata;
|
|
static ::PROTOBUF_NAMESPACE_ID::StringPiece FullMessageName() {
|
|
return "xla.ExecuteResponse";
|
|
}
|
|
protected:
|
|
explicit ExecuteResponse(::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 {
|
|
kOutputFieldNumber = 1,
|
|
kProfileFieldNumber = 2,
|
|
};
|
|
// .xla.GlobalDataHandle output = 1;
|
|
bool has_output() const;
|
|
private:
|
|
bool _internal_has_output() const;
|
|
public:
|
|
void clear_output();
|
|
const ::xla::GlobalDataHandle& output() const;
|
|
PROTOBUF_NODISCARD ::xla::GlobalDataHandle* release_output();
|
|
::xla::GlobalDataHandle* mutable_output();
|
|
void set_allocated_output(::xla::GlobalDataHandle* output);
|
|
private:
|
|
const ::xla::GlobalDataHandle& _internal_output() const;
|
|
::xla::GlobalDataHandle* _internal_mutable_output();
|
|
public:
|
|
void unsafe_arena_set_allocated_output(
|
|
::xla::GlobalDataHandle* output);
|
|
::xla::GlobalDataHandle* unsafe_arena_release_output();
|
|
|
|
// .xla.ExecutionProfile profile = 2;
|
|
bool has_profile() const;
|
|
private:
|
|
bool _internal_has_profile() const;
|
|
public:
|
|
void clear_profile();
|
|
const ::xla::ExecutionProfile& profile() const;
|
|
PROTOBUF_NODISCARD ::xla::ExecutionProfile* release_profile();
|
|
::xla::ExecutionProfile* mutable_profile();
|
|
void set_allocated_profile(::xla::ExecutionProfile* profile);
|
|
private:
|
|
const ::xla::ExecutionProfile& _internal_profile() const;
|
|
::xla::ExecutionProfile* _internal_mutable_profile();
|
|
public:
|
|
void unsafe_arena_set_allocated_profile(
|
|
::xla::ExecutionProfile* profile);
|
|
::xla::ExecutionProfile* unsafe_arena_release_profile();
|
|
|
|
// @@protoc_insertion_point(class_scope:xla.ExecuteResponse)
|
|
private:
|
|
class _Internal;
|
|
|
|
template <typename T> friend class ::PROTOBUF_NAMESPACE_ID::Arena::InternalHelper;
|
|
typedef void InternalArenaConstructable_;
|
|
typedef void DestructorSkippable_;
|
|
struct Impl_ {
|
|
::xla::GlobalDataHandle* output_;
|
|
::xla::ExecutionProfile* profile_;
|
|
mutable ::PROTOBUF_NAMESPACE_ID::internal::CachedSize _cached_size_;
|
|
};
|
|
union { Impl_ _impl_; };
|
|
friend struct ::TableStruct_xla_2fxla_2eproto;
|
|
};
|
|
// -------------------------------------------------------------------
|
|
|
|
class ExecuteParallelResponse final :
|
|
public ::PROTOBUF_NAMESPACE_ID::Message /* @@protoc_insertion_point(class_definition:xla.ExecuteParallelResponse) */ {
|
|
public:
|
|
inline ExecuteParallelResponse() : ExecuteParallelResponse(nullptr) {}
|
|
~ExecuteParallelResponse() override;
|
|
explicit PROTOBUF_CONSTEXPR ExecuteParallelResponse(::PROTOBUF_NAMESPACE_ID::internal::ConstantInitialized);
|
|
|
|
ExecuteParallelResponse(const ExecuteParallelResponse& from);
|
|
ExecuteParallelResponse(ExecuteParallelResponse&& from) noexcept
|
|
: ExecuteParallelResponse() {
|
|
*this = ::std::move(from);
|
|
}
|
|
|
|
inline ExecuteParallelResponse& operator=(const ExecuteParallelResponse& from) {
|
|
CopyFrom(from);
|
|
return *this;
|
|
}
|
|
inline ExecuteParallelResponse& operator=(ExecuteParallelResponse&& 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 ExecuteParallelResponse& default_instance() {
|
|
return *internal_default_instance();
|
|
}
|
|
static inline const ExecuteParallelResponse* internal_default_instance() {
|
|
return reinterpret_cast<const ExecuteParallelResponse*>(
|
|
&_ExecuteParallelResponse_default_instance_);
|
|
}
|
|
static constexpr int kIndexInFileMessages =
|
|
37;
|
|
|
|
friend void swap(ExecuteParallelResponse& a, ExecuteParallelResponse& b) {
|
|
a.Swap(&b);
|
|
}
|
|
inline void Swap(ExecuteParallelResponse* 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(ExecuteParallelResponse* other) {
|
|
if (other == this) return;
|
|
GOOGLE_DCHECK(GetOwningArena() == other->GetOwningArena());
|
|
InternalSwap(other);
|
|
}
|
|
|
|
// implements Message ----------------------------------------------
|
|
|
|
ExecuteParallelResponse* New(::PROTOBUF_NAMESPACE_ID::Arena* arena = nullptr) const final {
|
|
return CreateMaybeMessage<ExecuteParallelResponse>(arena);
|
|
}
|
|
using ::PROTOBUF_NAMESPACE_ID::Message::CopyFrom;
|
|
void CopyFrom(const ExecuteParallelResponse& from);
|
|
using ::PROTOBUF_NAMESPACE_ID::Message::MergeFrom;
|
|
void MergeFrom( const ExecuteParallelResponse& from) {
|
|
ExecuteParallelResponse::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(ExecuteParallelResponse* other);
|
|
|
|
private:
|
|
friend class ::PROTOBUF_NAMESPACE_ID::internal::AnyMetadata;
|
|
static ::PROTOBUF_NAMESPACE_ID::StringPiece FullMessageName() {
|
|
return "xla.ExecuteParallelResponse";
|
|
}
|
|
protected:
|
|
explicit ExecuteParallelResponse(::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 {
|
|
kResponsesFieldNumber = 1,
|
|
};
|
|
// repeated .xla.ExecuteResponse responses = 1;
|
|
int responses_size() const;
|
|
private:
|
|
int _internal_responses_size() const;
|
|
public:
|
|
void clear_responses();
|
|
::xla::ExecuteResponse* mutable_responses(int index);
|
|
::PROTOBUF_NAMESPACE_ID::RepeatedPtrField< ::xla::ExecuteResponse >*
|
|
mutable_responses();
|
|
private:
|
|
const ::xla::ExecuteResponse& _internal_responses(int index) const;
|
|
::xla::ExecuteResponse* _internal_add_responses();
|
|
public:
|
|
const ::xla::ExecuteResponse& responses(int index) const;
|
|
::xla::ExecuteResponse* add_responses();
|
|
const ::PROTOBUF_NAMESPACE_ID::RepeatedPtrField< ::xla::ExecuteResponse >&
|
|
responses() const;
|
|
|
|
// @@protoc_insertion_point(class_scope:xla.ExecuteParallelResponse)
|
|
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::ExecuteResponse > responses_;
|
|
mutable ::PROTOBUF_NAMESPACE_ID::internal::CachedSize _cached_size_;
|
|
};
|
|
union { Impl_ _impl_; };
|
|
friend struct ::TableStruct_xla_2fxla_2eproto;
|
|
};
|
|
// -------------------------------------------------------------------
|
|
|
|
class ComputeConstantGraphRequest final :
|
|
public ::PROTOBUF_NAMESPACE_ID::Message /* @@protoc_insertion_point(class_definition:xla.ComputeConstantGraphRequest) */ {
|
|
public:
|
|
inline ComputeConstantGraphRequest() : ComputeConstantGraphRequest(nullptr) {}
|
|
~ComputeConstantGraphRequest() override;
|
|
explicit PROTOBUF_CONSTEXPR ComputeConstantGraphRequest(::PROTOBUF_NAMESPACE_ID::internal::ConstantInitialized);
|
|
|
|
ComputeConstantGraphRequest(const ComputeConstantGraphRequest& from);
|
|
ComputeConstantGraphRequest(ComputeConstantGraphRequest&& from) noexcept
|
|
: ComputeConstantGraphRequest() {
|
|
*this = ::std::move(from);
|
|
}
|
|
|
|
inline ComputeConstantGraphRequest& operator=(const ComputeConstantGraphRequest& from) {
|
|
CopyFrom(from);
|
|
return *this;
|
|
}
|
|
inline ComputeConstantGraphRequest& operator=(ComputeConstantGraphRequest&& 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 ComputeConstantGraphRequest& default_instance() {
|
|
return *internal_default_instance();
|
|
}
|
|
static inline const ComputeConstantGraphRequest* internal_default_instance() {
|
|
return reinterpret_cast<const ComputeConstantGraphRequest*>(
|
|
&_ComputeConstantGraphRequest_default_instance_);
|
|
}
|
|
static constexpr int kIndexInFileMessages =
|
|
38;
|
|
|
|
friend void swap(ComputeConstantGraphRequest& a, ComputeConstantGraphRequest& b) {
|
|
a.Swap(&b);
|
|
}
|
|
inline void Swap(ComputeConstantGraphRequest* 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(ComputeConstantGraphRequest* other) {
|
|
if (other == this) return;
|
|
GOOGLE_DCHECK(GetOwningArena() == other->GetOwningArena());
|
|
InternalSwap(other);
|
|
}
|
|
|
|
// implements Message ----------------------------------------------
|
|
|
|
ComputeConstantGraphRequest* New(::PROTOBUF_NAMESPACE_ID::Arena* arena = nullptr) const final {
|
|
return CreateMaybeMessage<ComputeConstantGraphRequest>(arena);
|
|
}
|
|
using ::PROTOBUF_NAMESPACE_ID::Message::CopyFrom;
|
|
void CopyFrom(const ComputeConstantGraphRequest& from);
|
|
using ::PROTOBUF_NAMESPACE_ID::Message::MergeFrom;
|
|
void MergeFrom( const ComputeConstantGraphRequest& from) {
|
|
ComputeConstantGraphRequest::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(ComputeConstantGraphRequest* other);
|
|
|
|
private:
|
|
friend class ::PROTOBUF_NAMESPACE_ID::internal::AnyMetadata;
|
|
static ::PROTOBUF_NAMESPACE_ID::StringPiece FullMessageName() {
|
|
return "xla.ComputeConstantGraphRequest";
|
|
}
|
|
protected:
|
|
explicit ComputeConstantGraphRequest(::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 {
|
|
kComputationFieldNumber = 1,
|
|
kOutputLayoutFieldNumber = 2,
|
|
};
|
|
// .xla.HloModuleProto computation = 1;
|
|
bool has_computation() const;
|
|
private:
|
|
bool _internal_has_computation() const;
|
|
public:
|
|
void clear_computation();
|
|
const ::xla::HloModuleProto& computation() const;
|
|
PROTOBUF_NODISCARD ::xla::HloModuleProto* release_computation();
|
|
::xla::HloModuleProto* mutable_computation();
|
|
void set_allocated_computation(::xla::HloModuleProto* computation);
|
|
private:
|
|
const ::xla::HloModuleProto& _internal_computation() const;
|
|
::xla::HloModuleProto* _internal_mutable_computation();
|
|
public:
|
|
void unsafe_arena_set_allocated_computation(
|
|
::xla::HloModuleProto* computation);
|
|
::xla::HloModuleProto* unsafe_arena_release_computation();
|
|
|
|
// .xla.LayoutProto output_layout = 2;
|
|
bool has_output_layout() const;
|
|
private:
|
|
bool _internal_has_output_layout() const;
|
|
public:
|
|
void clear_output_layout();
|
|
const ::xla::LayoutProto& output_layout() const;
|
|
PROTOBUF_NODISCARD ::xla::LayoutProto* release_output_layout();
|
|
::xla::LayoutProto* mutable_output_layout();
|
|
void set_allocated_output_layout(::xla::LayoutProto* output_layout);
|
|
private:
|
|
const ::xla::LayoutProto& _internal_output_layout() const;
|
|
::xla::LayoutProto* _internal_mutable_output_layout();
|
|
public:
|
|
void unsafe_arena_set_allocated_output_layout(
|
|
::xla::LayoutProto* output_layout);
|
|
::xla::LayoutProto* unsafe_arena_release_output_layout();
|
|
|
|
// @@protoc_insertion_point(class_scope:xla.ComputeConstantGraphRequest)
|
|
private:
|
|
class _Internal;
|
|
|
|
template <typename T> friend class ::PROTOBUF_NAMESPACE_ID::Arena::InternalHelper;
|
|
typedef void InternalArenaConstructable_;
|
|
typedef void DestructorSkippable_;
|
|
struct Impl_ {
|
|
::xla::HloModuleProto* computation_;
|
|
::xla::LayoutProto* output_layout_;
|
|
mutable ::PROTOBUF_NAMESPACE_ID::internal::CachedSize _cached_size_;
|
|
};
|
|
union { Impl_ _impl_; };
|
|
friend struct ::TableStruct_xla_2fxla_2eproto;
|
|
};
|
|
// -------------------------------------------------------------------
|
|
|
|
class ComputeConstantResponse final :
|
|
public ::PROTOBUF_NAMESPACE_ID::Message /* @@protoc_insertion_point(class_definition:xla.ComputeConstantResponse) */ {
|
|
public:
|
|
inline ComputeConstantResponse() : ComputeConstantResponse(nullptr) {}
|
|
~ComputeConstantResponse() override;
|
|
explicit PROTOBUF_CONSTEXPR ComputeConstantResponse(::PROTOBUF_NAMESPACE_ID::internal::ConstantInitialized);
|
|
|
|
ComputeConstantResponse(const ComputeConstantResponse& from);
|
|
ComputeConstantResponse(ComputeConstantResponse&& from) noexcept
|
|
: ComputeConstantResponse() {
|
|
*this = ::std::move(from);
|
|
}
|
|
|
|
inline ComputeConstantResponse& operator=(const ComputeConstantResponse& from) {
|
|
CopyFrom(from);
|
|
return *this;
|
|
}
|
|
inline ComputeConstantResponse& operator=(ComputeConstantResponse&& 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 ComputeConstantResponse& default_instance() {
|
|
return *internal_default_instance();
|
|
}
|
|
static inline const ComputeConstantResponse* internal_default_instance() {
|
|
return reinterpret_cast<const ComputeConstantResponse*>(
|
|
&_ComputeConstantResponse_default_instance_);
|
|
}
|
|
static constexpr int kIndexInFileMessages =
|
|
39;
|
|
|
|
friend void swap(ComputeConstantResponse& a, ComputeConstantResponse& b) {
|
|
a.Swap(&b);
|
|
}
|
|
inline void Swap(ComputeConstantResponse* 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(ComputeConstantResponse* other) {
|
|
if (other == this) return;
|
|
GOOGLE_DCHECK(GetOwningArena() == other->GetOwningArena());
|
|
InternalSwap(other);
|
|
}
|
|
|
|
// implements Message ----------------------------------------------
|
|
|
|
ComputeConstantResponse* New(::PROTOBUF_NAMESPACE_ID::Arena* arena = nullptr) const final {
|
|
return CreateMaybeMessage<ComputeConstantResponse>(arena);
|
|
}
|
|
using ::PROTOBUF_NAMESPACE_ID::Message::CopyFrom;
|
|
void CopyFrom(const ComputeConstantResponse& from);
|
|
using ::PROTOBUF_NAMESPACE_ID::Message::MergeFrom;
|
|
void MergeFrom( const ComputeConstantResponse& from) {
|
|
ComputeConstantResponse::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(ComputeConstantResponse* other);
|
|
|
|
private:
|
|
friend class ::PROTOBUF_NAMESPACE_ID::internal::AnyMetadata;
|
|
static ::PROTOBUF_NAMESPACE_ID::StringPiece FullMessageName() {
|
|
return "xla.ComputeConstantResponse";
|
|
}
|
|
protected:
|
|
explicit ComputeConstantResponse(::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 {
|
|
kLiteralFieldNumber = 1,
|
|
};
|
|
// .xla.LiteralProto literal = 1;
|
|
bool has_literal() const;
|
|
private:
|
|
bool _internal_has_literal() const;
|
|
public:
|
|
void clear_literal();
|
|
const ::xla::LiteralProto& literal() const;
|
|
PROTOBUF_NODISCARD ::xla::LiteralProto* release_literal();
|
|
::xla::LiteralProto* mutable_literal();
|
|
void set_allocated_literal(::xla::LiteralProto* literal);
|
|
private:
|
|
const ::xla::LiteralProto& _internal_literal() const;
|
|
::xla::LiteralProto* _internal_mutable_literal();
|
|
public:
|
|
void unsafe_arena_set_allocated_literal(
|
|
::xla::LiteralProto* literal);
|
|
::xla::LiteralProto* unsafe_arena_release_literal();
|
|
|
|
// @@protoc_insertion_point(class_scope:xla.ComputeConstantResponse)
|
|
private:
|
|
class _Internal;
|
|
|
|
template <typename T> friend class ::PROTOBUF_NAMESPACE_ID::Arena::InternalHelper;
|
|
typedef void InternalArenaConstructable_;
|
|
typedef void DestructorSkippable_;
|
|
struct Impl_ {
|
|
::xla::LiteralProto* literal_;
|
|
mutable ::PROTOBUF_NAMESPACE_ID::internal::CachedSize _cached_size_;
|
|
};
|
|
union { Impl_ _impl_; };
|
|
friend struct ::TableStruct_xla_2fxla_2eproto;
|
|
};
|
|
// -------------------------------------------------------------------
|
|
|
|
class DeconstructTupleRequest final :
|
|
public ::PROTOBUF_NAMESPACE_ID::Message /* @@protoc_insertion_point(class_definition:xla.DeconstructTupleRequest) */ {
|
|
public:
|
|
inline DeconstructTupleRequest() : DeconstructTupleRequest(nullptr) {}
|
|
~DeconstructTupleRequest() override;
|
|
explicit PROTOBUF_CONSTEXPR DeconstructTupleRequest(::PROTOBUF_NAMESPACE_ID::internal::ConstantInitialized);
|
|
|
|
DeconstructTupleRequest(const DeconstructTupleRequest& from);
|
|
DeconstructTupleRequest(DeconstructTupleRequest&& from) noexcept
|
|
: DeconstructTupleRequest() {
|
|
*this = ::std::move(from);
|
|
}
|
|
|
|
inline DeconstructTupleRequest& operator=(const DeconstructTupleRequest& from) {
|
|
CopyFrom(from);
|
|
return *this;
|
|
}
|
|
inline DeconstructTupleRequest& operator=(DeconstructTupleRequest&& 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 DeconstructTupleRequest& default_instance() {
|
|
return *internal_default_instance();
|
|
}
|
|
static inline const DeconstructTupleRequest* internal_default_instance() {
|
|
return reinterpret_cast<const DeconstructTupleRequest*>(
|
|
&_DeconstructTupleRequest_default_instance_);
|
|
}
|
|
static constexpr int kIndexInFileMessages =
|
|
40;
|
|
|
|
friend void swap(DeconstructTupleRequest& a, DeconstructTupleRequest& b) {
|
|
a.Swap(&b);
|
|
}
|
|
inline void Swap(DeconstructTupleRequest* 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(DeconstructTupleRequest* other) {
|
|
if (other == this) return;
|
|
GOOGLE_DCHECK(GetOwningArena() == other->GetOwningArena());
|
|
InternalSwap(other);
|
|
}
|
|
|
|
// implements Message ----------------------------------------------
|
|
|
|
DeconstructTupleRequest* New(::PROTOBUF_NAMESPACE_ID::Arena* arena = nullptr) const final {
|
|
return CreateMaybeMessage<DeconstructTupleRequest>(arena);
|
|
}
|
|
using ::PROTOBUF_NAMESPACE_ID::Message::CopyFrom;
|
|
void CopyFrom(const DeconstructTupleRequest& from);
|
|
using ::PROTOBUF_NAMESPACE_ID::Message::MergeFrom;
|
|
void MergeFrom( const DeconstructTupleRequest& from) {
|
|
DeconstructTupleRequest::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(DeconstructTupleRequest* other);
|
|
|
|
private:
|
|
friend class ::PROTOBUF_NAMESPACE_ID::internal::AnyMetadata;
|
|
static ::PROTOBUF_NAMESPACE_ID::StringPiece FullMessageName() {
|
|
return "xla.DeconstructTupleRequest";
|
|
}
|
|
protected:
|
|
explicit DeconstructTupleRequest(::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 {
|
|
kTupleHandleFieldNumber = 2,
|
|
};
|
|
// .xla.GlobalDataHandle tuple_handle = 2;
|
|
bool has_tuple_handle() const;
|
|
private:
|
|
bool _internal_has_tuple_handle() const;
|
|
public:
|
|
void clear_tuple_handle();
|
|
const ::xla::GlobalDataHandle& tuple_handle() const;
|
|
PROTOBUF_NODISCARD ::xla::GlobalDataHandle* release_tuple_handle();
|
|
::xla::GlobalDataHandle* mutable_tuple_handle();
|
|
void set_allocated_tuple_handle(::xla::GlobalDataHandle* tuple_handle);
|
|
private:
|
|
const ::xla::GlobalDataHandle& _internal_tuple_handle() const;
|
|
::xla::GlobalDataHandle* _internal_mutable_tuple_handle();
|
|
public:
|
|
void unsafe_arena_set_allocated_tuple_handle(
|
|
::xla::GlobalDataHandle* tuple_handle);
|
|
::xla::GlobalDataHandle* unsafe_arena_release_tuple_handle();
|
|
|
|
// @@protoc_insertion_point(class_scope:xla.DeconstructTupleRequest)
|
|
private:
|
|
class _Internal;
|
|
|
|
template <typename T> friend class ::PROTOBUF_NAMESPACE_ID::Arena::InternalHelper;
|
|
typedef void InternalArenaConstructable_;
|
|
typedef void DestructorSkippable_;
|
|
struct Impl_ {
|
|
::xla::GlobalDataHandle* tuple_handle_;
|
|
mutable ::PROTOBUF_NAMESPACE_ID::internal::CachedSize _cached_size_;
|
|
};
|
|
union { Impl_ _impl_; };
|
|
friend struct ::TableStruct_xla_2fxla_2eproto;
|
|
};
|
|
// -------------------------------------------------------------------
|
|
|
|
class DeconstructTupleResponse final :
|
|
public ::PROTOBUF_NAMESPACE_ID::Message /* @@protoc_insertion_point(class_definition:xla.DeconstructTupleResponse) */ {
|
|
public:
|
|
inline DeconstructTupleResponse() : DeconstructTupleResponse(nullptr) {}
|
|
~DeconstructTupleResponse() override;
|
|
explicit PROTOBUF_CONSTEXPR DeconstructTupleResponse(::PROTOBUF_NAMESPACE_ID::internal::ConstantInitialized);
|
|
|
|
DeconstructTupleResponse(const DeconstructTupleResponse& from);
|
|
DeconstructTupleResponse(DeconstructTupleResponse&& from) noexcept
|
|
: DeconstructTupleResponse() {
|
|
*this = ::std::move(from);
|
|
}
|
|
|
|
inline DeconstructTupleResponse& operator=(const DeconstructTupleResponse& from) {
|
|
CopyFrom(from);
|
|
return *this;
|
|
}
|
|
inline DeconstructTupleResponse& operator=(DeconstructTupleResponse&& 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 DeconstructTupleResponse& default_instance() {
|
|
return *internal_default_instance();
|
|
}
|
|
static inline const DeconstructTupleResponse* internal_default_instance() {
|
|
return reinterpret_cast<const DeconstructTupleResponse*>(
|
|
&_DeconstructTupleResponse_default_instance_);
|
|
}
|
|
static constexpr int kIndexInFileMessages =
|
|
41;
|
|
|
|
friend void swap(DeconstructTupleResponse& a, DeconstructTupleResponse& b) {
|
|
a.Swap(&b);
|
|
}
|
|
inline void Swap(DeconstructTupleResponse* 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(DeconstructTupleResponse* other) {
|
|
if (other == this) return;
|
|
GOOGLE_DCHECK(GetOwningArena() == other->GetOwningArena());
|
|
InternalSwap(other);
|
|
}
|
|
|
|
// implements Message ----------------------------------------------
|
|
|
|
DeconstructTupleResponse* New(::PROTOBUF_NAMESPACE_ID::Arena* arena = nullptr) const final {
|
|
return CreateMaybeMessage<DeconstructTupleResponse>(arena);
|
|
}
|
|
using ::PROTOBUF_NAMESPACE_ID::Message::CopyFrom;
|
|
void CopyFrom(const DeconstructTupleResponse& from);
|
|
using ::PROTOBUF_NAMESPACE_ID::Message::MergeFrom;
|
|
void MergeFrom( const DeconstructTupleResponse& from) {
|
|
DeconstructTupleResponse::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(DeconstructTupleResponse* other);
|
|
|
|
private:
|
|
friend class ::PROTOBUF_NAMESPACE_ID::internal::AnyMetadata;
|
|
static ::PROTOBUF_NAMESPACE_ID::StringPiece FullMessageName() {
|
|
return "xla.DeconstructTupleResponse";
|
|
}
|
|
protected:
|
|
explicit DeconstructTupleResponse(::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 {
|
|
kElementHandlesFieldNumber = 1,
|
|
};
|
|
// repeated .xla.GlobalDataHandle element_handles = 1;
|
|
int element_handles_size() const;
|
|
private:
|
|
int _internal_element_handles_size() const;
|
|
public:
|
|
void clear_element_handles();
|
|
::xla::GlobalDataHandle* mutable_element_handles(int index);
|
|
::PROTOBUF_NAMESPACE_ID::RepeatedPtrField< ::xla::GlobalDataHandle >*
|
|
mutable_element_handles();
|
|
private:
|
|
const ::xla::GlobalDataHandle& _internal_element_handles(int index) const;
|
|
::xla::GlobalDataHandle* _internal_add_element_handles();
|
|
public:
|
|
const ::xla::GlobalDataHandle& element_handles(int index) const;
|
|
::xla::GlobalDataHandle* add_element_handles();
|
|
const ::PROTOBUF_NAMESPACE_ID::RepeatedPtrField< ::xla::GlobalDataHandle >&
|
|
element_handles() const;
|
|
|
|
// @@protoc_insertion_point(class_scope:xla.DeconstructTupleResponse)
|
|
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::GlobalDataHandle > element_handles_;
|
|
mutable ::PROTOBUF_NAMESPACE_ID::internal::CachedSize _cached_size_;
|
|
};
|
|
union { Impl_ _impl_; };
|
|
friend struct ::TableStruct_xla_2fxla_2eproto;
|
|
};
|
|
// -------------------------------------------------------------------
|
|
|
|
class LoadDataRequest final :
|
|
public ::PROTOBUF_NAMESPACE_ID::Message /* @@protoc_insertion_point(class_definition:xla.LoadDataRequest) */ {
|
|
public:
|
|
inline LoadDataRequest() : LoadDataRequest(nullptr) {}
|
|
~LoadDataRequest() override;
|
|
explicit PROTOBUF_CONSTEXPR LoadDataRequest(::PROTOBUF_NAMESPACE_ID::internal::ConstantInitialized);
|
|
|
|
LoadDataRequest(const LoadDataRequest& from);
|
|
LoadDataRequest(LoadDataRequest&& from) noexcept
|
|
: LoadDataRequest() {
|
|
*this = ::std::move(from);
|
|
}
|
|
|
|
inline LoadDataRequest& operator=(const LoadDataRequest& from) {
|
|
CopyFrom(from);
|
|
return *this;
|
|
}
|
|
inline LoadDataRequest& operator=(LoadDataRequest&& 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 LoadDataRequest& default_instance() {
|
|
return *internal_default_instance();
|
|
}
|
|
static inline const LoadDataRequest* internal_default_instance() {
|
|
return reinterpret_cast<const LoadDataRequest*>(
|
|
&_LoadDataRequest_default_instance_);
|
|
}
|
|
static constexpr int kIndexInFileMessages =
|
|
42;
|
|
|
|
friend void swap(LoadDataRequest& a, LoadDataRequest& b) {
|
|
a.Swap(&b);
|
|
}
|
|
inline void Swap(LoadDataRequest* 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(LoadDataRequest* other) {
|
|
if (other == this) return;
|
|
GOOGLE_DCHECK(GetOwningArena() == other->GetOwningArena());
|
|
InternalSwap(other);
|
|
}
|
|
|
|
// implements Message ----------------------------------------------
|
|
|
|
LoadDataRequest* New(::PROTOBUF_NAMESPACE_ID::Arena* arena = nullptr) const final {
|
|
return CreateMaybeMessage<LoadDataRequest>(arena);
|
|
}
|
|
using ::PROTOBUF_NAMESPACE_ID::Message::CopyFrom;
|
|
void CopyFrom(const LoadDataRequest& from);
|
|
using ::PROTOBUF_NAMESPACE_ID::Message::MergeFrom;
|
|
void MergeFrom( const LoadDataRequest& from) {
|
|
LoadDataRequest::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(LoadDataRequest* other);
|
|
|
|
private:
|
|
friend class ::PROTOBUF_NAMESPACE_ID::internal::AnyMetadata;
|
|
static ::PROTOBUF_NAMESPACE_ID::StringPiece FullMessageName() {
|
|
return "xla.LoadDataRequest";
|
|
}
|
|
protected:
|
|
explicit LoadDataRequest(::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 {
|
|
kColumnioTabletPathFieldNumber = 1,
|
|
kColumnioFieldFieldNumber = 2,
|
|
kElementShapeFieldNumber = 3,
|
|
kOffsetFieldNumber = 4,
|
|
kLimitFieldNumber = 5,
|
|
kZipFieldNumber = 6,
|
|
};
|
|
// string columnio_tablet_path = 1;
|
|
void clear_columnio_tablet_path();
|
|
const std::string& columnio_tablet_path() const;
|
|
template <typename ArgT0 = const std::string&, typename... ArgT>
|
|
void set_columnio_tablet_path(ArgT0&& arg0, ArgT... args);
|
|
std::string* mutable_columnio_tablet_path();
|
|
PROTOBUF_NODISCARD std::string* release_columnio_tablet_path();
|
|
void set_allocated_columnio_tablet_path(std::string* columnio_tablet_path);
|
|
private:
|
|
const std::string& _internal_columnio_tablet_path() const;
|
|
inline PROTOBUF_ALWAYS_INLINE void _internal_set_columnio_tablet_path(const std::string& value);
|
|
std::string* _internal_mutable_columnio_tablet_path();
|
|
public:
|
|
|
|
// string columnio_field = 2;
|
|
void clear_columnio_field();
|
|
const std::string& columnio_field() const;
|
|
template <typename ArgT0 = const std::string&, typename... ArgT>
|
|
void set_columnio_field(ArgT0&& arg0, ArgT... args);
|
|
std::string* mutable_columnio_field();
|
|
PROTOBUF_NODISCARD std::string* release_columnio_field();
|
|
void set_allocated_columnio_field(std::string* columnio_field);
|
|
private:
|
|
const std::string& _internal_columnio_field() const;
|
|
inline PROTOBUF_ALWAYS_INLINE void _internal_set_columnio_field(const std::string& value);
|
|
std::string* _internal_mutable_columnio_field();
|
|
public:
|
|
|
|
// .xla.ShapeProto element_shape = 3;
|
|
bool has_element_shape() const;
|
|
private:
|
|
bool _internal_has_element_shape() const;
|
|
public:
|
|
void clear_element_shape();
|
|
const ::xla::ShapeProto& element_shape() const;
|
|
PROTOBUF_NODISCARD ::xla::ShapeProto* release_element_shape();
|
|
::xla::ShapeProto* mutable_element_shape();
|
|
void set_allocated_element_shape(::xla::ShapeProto* element_shape);
|
|
private:
|
|
const ::xla::ShapeProto& _internal_element_shape() const;
|
|
::xla::ShapeProto* _internal_mutable_element_shape();
|
|
public:
|
|
void unsafe_arena_set_allocated_element_shape(
|
|
::xla::ShapeProto* element_shape);
|
|
::xla::ShapeProto* unsafe_arena_release_element_shape();
|
|
|
|
// int64 offset = 4;
|
|
void clear_offset();
|
|
int64_t offset() const;
|
|
void set_offset(int64_t value);
|
|
private:
|
|
int64_t _internal_offset() const;
|
|
void _internal_set_offset(int64_t value);
|
|
public:
|
|
|
|
// int64 limit = 5;
|
|
void clear_limit();
|
|
int64_t limit() const;
|
|
void set_limit(int64_t value);
|
|
private:
|
|
int64_t _internal_limit() const;
|
|
void _internal_set_limit(int64_t value);
|
|
public:
|
|
|
|
// bool zip = 6;
|
|
void clear_zip();
|
|
bool zip() const;
|
|
void set_zip(bool value);
|
|
private:
|
|
bool _internal_zip() const;
|
|
void _internal_set_zip(bool value);
|
|
public:
|
|
|
|
// @@protoc_insertion_point(class_scope:xla.LoadDataRequest)
|
|
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 columnio_tablet_path_;
|
|
::PROTOBUF_NAMESPACE_ID::internal::ArenaStringPtr columnio_field_;
|
|
::xla::ShapeProto* element_shape_;
|
|
int64_t offset_;
|
|
int64_t limit_;
|
|
bool zip_;
|
|
mutable ::PROTOBUF_NAMESPACE_ID::internal::CachedSize _cached_size_;
|
|
};
|
|
union { Impl_ _impl_; };
|
|
friend struct ::TableStruct_xla_2fxla_2eproto;
|
|
};
|
|
// -------------------------------------------------------------------
|
|
|
|
class LoadDataResponse final :
|
|
public ::PROTOBUF_NAMESPACE_ID::Message /* @@protoc_insertion_point(class_definition:xla.LoadDataResponse) */ {
|
|
public:
|
|
inline LoadDataResponse() : LoadDataResponse(nullptr) {}
|
|
~LoadDataResponse() override;
|
|
explicit PROTOBUF_CONSTEXPR LoadDataResponse(::PROTOBUF_NAMESPACE_ID::internal::ConstantInitialized);
|
|
|
|
LoadDataResponse(const LoadDataResponse& from);
|
|
LoadDataResponse(LoadDataResponse&& from) noexcept
|
|
: LoadDataResponse() {
|
|
*this = ::std::move(from);
|
|
}
|
|
|
|
inline LoadDataResponse& operator=(const LoadDataResponse& from) {
|
|
CopyFrom(from);
|
|
return *this;
|
|
}
|
|
inline LoadDataResponse& operator=(LoadDataResponse&& 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 LoadDataResponse& default_instance() {
|
|
return *internal_default_instance();
|
|
}
|
|
static inline const LoadDataResponse* internal_default_instance() {
|
|
return reinterpret_cast<const LoadDataResponse*>(
|
|
&_LoadDataResponse_default_instance_);
|
|
}
|
|
static constexpr int kIndexInFileMessages =
|
|
43;
|
|
|
|
friend void swap(LoadDataResponse& a, LoadDataResponse& b) {
|
|
a.Swap(&b);
|
|
}
|
|
inline void Swap(LoadDataResponse* 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(LoadDataResponse* other) {
|
|
if (other == this) return;
|
|
GOOGLE_DCHECK(GetOwningArena() == other->GetOwningArena());
|
|
InternalSwap(other);
|
|
}
|
|
|
|
// implements Message ----------------------------------------------
|
|
|
|
LoadDataResponse* New(::PROTOBUF_NAMESPACE_ID::Arena* arena = nullptr) const final {
|
|
return CreateMaybeMessage<LoadDataResponse>(arena);
|
|
}
|
|
using ::PROTOBUF_NAMESPACE_ID::Message::CopyFrom;
|
|
void CopyFrom(const LoadDataResponse& from);
|
|
using ::PROTOBUF_NAMESPACE_ID::Message::MergeFrom;
|
|
void MergeFrom( const LoadDataResponse& from) {
|
|
LoadDataResponse::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(LoadDataResponse* other);
|
|
|
|
private:
|
|
friend class ::PROTOBUF_NAMESPACE_ID::internal::AnyMetadata;
|
|
static ::PROTOBUF_NAMESPACE_ID::StringPiece FullMessageName() {
|
|
return "xla.LoadDataResponse";
|
|
}
|
|
protected:
|
|
explicit LoadDataResponse(::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 {
|
|
kDataFieldNumber = 1,
|
|
kDataShapeFieldNumber = 2,
|
|
kAvailableRowsFieldNumber = 3,
|
|
kRowsLoadedFieldNumber = 4,
|
|
kNanosecondsFieldNumber = 5,
|
|
};
|
|
// .xla.GlobalDataHandle data = 1;
|
|
bool has_data() const;
|
|
private:
|
|
bool _internal_has_data() const;
|
|
public:
|
|
void clear_data();
|
|
const ::xla::GlobalDataHandle& data() const;
|
|
PROTOBUF_NODISCARD ::xla::GlobalDataHandle* release_data();
|
|
::xla::GlobalDataHandle* mutable_data();
|
|
void set_allocated_data(::xla::GlobalDataHandle* data);
|
|
private:
|
|
const ::xla::GlobalDataHandle& _internal_data() const;
|
|
::xla::GlobalDataHandle* _internal_mutable_data();
|
|
public:
|
|
void unsafe_arena_set_allocated_data(
|
|
::xla::GlobalDataHandle* data);
|
|
::xla::GlobalDataHandle* unsafe_arena_release_data();
|
|
|
|
// .xla.ShapeProto data_shape = 2;
|
|
bool has_data_shape() const;
|
|
private:
|
|
bool _internal_has_data_shape() const;
|
|
public:
|
|
void clear_data_shape();
|
|
const ::xla::ShapeProto& data_shape() const;
|
|
PROTOBUF_NODISCARD ::xla::ShapeProto* release_data_shape();
|
|
::xla::ShapeProto* mutable_data_shape();
|
|
void set_allocated_data_shape(::xla::ShapeProto* data_shape);
|
|
private:
|
|
const ::xla::ShapeProto& _internal_data_shape() const;
|
|
::xla::ShapeProto* _internal_mutable_data_shape();
|
|
public:
|
|
void unsafe_arena_set_allocated_data_shape(
|
|
::xla::ShapeProto* data_shape);
|
|
::xla::ShapeProto* unsafe_arena_release_data_shape();
|
|
|
|
// int64 available_rows = 3;
|
|
void clear_available_rows();
|
|
int64_t available_rows() const;
|
|
void set_available_rows(int64_t value);
|
|
private:
|
|
int64_t _internal_available_rows() const;
|
|
void _internal_set_available_rows(int64_t value);
|
|
public:
|
|
|
|
// int64 rows_loaded = 4;
|
|
void clear_rows_loaded();
|
|
int64_t rows_loaded() const;
|
|
void set_rows_loaded(int64_t value);
|
|
private:
|
|
int64_t _internal_rows_loaded() const;
|
|
void _internal_set_rows_loaded(int64_t value);
|
|
public:
|
|
|
|
// int64 nanoseconds = 5;
|
|
void clear_nanoseconds();
|
|
int64_t nanoseconds() const;
|
|
void set_nanoseconds(int64_t value);
|
|
private:
|
|
int64_t _internal_nanoseconds() const;
|
|
void _internal_set_nanoseconds(int64_t value);
|
|
public:
|
|
|
|
// @@protoc_insertion_point(class_scope:xla.LoadDataResponse)
|
|
private:
|
|
class _Internal;
|
|
|
|
template <typename T> friend class ::PROTOBUF_NAMESPACE_ID::Arena::InternalHelper;
|
|
typedef void InternalArenaConstructable_;
|
|
typedef void DestructorSkippable_;
|
|
struct Impl_ {
|
|
::xla::GlobalDataHandle* data_;
|
|
::xla::ShapeProto* data_shape_;
|
|
int64_t available_rows_;
|
|
int64_t rows_loaded_;
|
|
int64_t nanoseconds_;
|
|
mutable ::PROTOBUF_NAMESPACE_ID::internal::CachedSize _cached_size_;
|
|
};
|
|
union { Impl_ _impl_; };
|
|
friend struct ::TableStruct_xla_2fxla_2eproto;
|
|
};
|
|
// -------------------------------------------------------------------
|
|
|
|
class GetShapeRequest final :
|
|
public ::PROTOBUF_NAMESPACE_ID::Message /* @@protoc_insertion_point(class_definition:xla.GetShapeRequest) */ {
|
|
public:
|
|
inline GetShapeRequest() : GetShapeRequest(nullptr) {}
|
|
~GetShapeRequest() override;
|
|
explicit PROTOBUF_CONSTEXPR GetShapeRequest(::PROTOBUF_NAMESPACE_ID::internal::ConstantInitialized);
|
|
|
|
GetShapeRequest(const GetShapeRequest& from);
|
|
GetShapeRequest(GetShapeRequest&& from) noexcept
|
|
: GetShapeRequest() {
|
|
*this = ::std::move(from);
|
|
}
|
|
|
|
inline GetShapeRequest& operator=(const GetShapeRequest& from) {
|
|
CopyFrom(from);
|
|
return *this;
|
|
}
|
|
inline GetShapeRequest& operator=(GetShapeRequest&& 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 GetShapeRequest& default_instance() {
|
|
return *internal_default_instance();
|
|
}
|
|
static inline const GetShapeRequest* internal_default_instance() {
|
|
return reinterpret_cast<const GetShapeRequest*>(
|
|
&_GetShapeRequest_default_instance_);
|
|
}
|
|
static constexpr int kIndexInFileMessages =
|
|
44;
|
|
|
|
friend void swap(GetShapeRequest& a, GetShapeRequest& b) {
|
|
a.Swap(&b);
|
|
}
|
|
inline void Swap(GetShapeRequest* 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(GetShapeRequest* other) {
|
|
if (other == this) return;
|
|
GOOGLE_DCHECK(GetOwningArena() == other->GetOwningArena());
|
|
InternalSwap(other);
|
|
}
|
|
|
|
// implements Message ----------------------------------------------
|
|
|
|
GetShapeRequest* New(::PROTOBUF_NAMESPACE_ID::Arena* arena = nullptr) const final {
|
|
return CreateMaybeMessage<GetShapeRequest>(arena);
|
|
}
|
|
using ::PROTOBUF_NAMESPACE_ID::Message::CopyFrom;
|
|
void CopyFrom(const GetShapeRequest& from);
|
|
using ::PROTOBUF_NAMESPACE_ID::Message::MergeFrom;
|
|
void MergeFrom( const GetShapeRequest& from) {
|
|
GetShapeRequest::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(GetShapeRequest* other);
|
|
|
|
private:
|
|
friend class ::PROTOBUF_NAMESPACE_ID::internal::AnyMetadata;
|
|
static ::PROTOBUF_NAMESPACE_ID::StringPiece FullMessageName() {
|
|
return "xla.GetShapeRequest";
|
|
}
|
|
protected:
|
|
explicit GetShapeRequest(::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 {
|
|
kDataFieldNumber = 1,
|
|
};
|
|
// .xla.GlobalDataHandle data = 1;
|
|
bool has_data() const;
|
|
private:
|
|
bool _internal_has_data() const;
|
|
public:
|
|
void clear_data();
|
|
const ::xla::GlobalDataHandle& data() const;
|
|
PROTOBUF_NODISCARD ::xla::GlobalDataHandle* release_data();
|
|
::xla::GlobalDataHandle* mutable_data();
|
|
void set_allocated_data(::xla::GlobalDataHandle* data);
|
|
private:
|
|
const ::xla::GlobalDataHandle& _internal_data() const;
|
|
::xla::GlobalDataHandle* _internal_mutable_data();
|
|
public:
|
|
void unsafe_arena_set_allocated_data(
|
|
::xla::GlobalDataHandle* data);
|
|
::xla::GlobalDataHandle* unsafe_arena_release_data();
|
|
|
|
// @@protoc_insertion_point(class_scope:xla.GetShapeRequest)
|
|
private:
|
|
class _Internal;
|
|
|
|
template <typename T> friend class ::PROTOBUF_NAMESPACE_ID::Arena::InternalHelper;
|
|
typedef void InternalArenaConstructable_;
|
|
typedef void DestructorSkippable_;
|
|
struct Impl_ {
|
|
::xla::GlobalDataHandle* data_;
|
|
mutable ::PROTOBUF_NAMESPACE_ID::internal::CachedSize _cached_size_;
|
|
};
|
|
union { Impl_ _impl_; };
|
|
friend struct ::TableStruct_xla_2fxla_2eproto;
|
|
};
|
|
// -------------------------------------------------------------------
|
|
|
|
class GetShapeResponse final :
|
|
public ::PROTOBUF_NAMESPACE_ID::Message /* @@protoc_insertion_point(class_definition:xla.GetShapeResponse) */ {
|
|
public:
|
|
inline GetShapeResponse() : GetShapeResponse(nullptr) {}
|
|
~GetShapeResponse() override;
|
|
explicit PROTOBUF_CONSTEXPR GetShapeResponse(::PROTOBUF_NAMESPACE_ID::internal::ConstantInitialized);
|
|
|
|
GetShapeResponse(const GetShapeResponse& from);
|
|
GetShapeResponse(GetShapeResponse&& from) noexcept
|
|
: GetShapeResponse() {
|
|
*this = ::std::move(from);
|
|
}
|
|
|
|
inline GetShapeResponse& operator=(const GetShapeResponse& from) {
|
|
CopyFrom(from);
|
|
return *this;
|
|
}
|
|
inline GetShapeResponse& operator=(GetShapeResponse&& 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 GetShapeResponse& default_instance() {
|
|
return *internal_default_instance();
|
|
}
|
|
static inline const GetShapeResponse* internal_default_instance() {
|
|
return reinterpret_cast<const GetShapeResponse*>(
|
|
&_GetShapeResponse_default_instance_);
|
|
}
|
|
static constexpr int kIndexInFileMessages =
|
|
45;
|
|
|
|
friend void swap(GetShapeResponse& a, GetShapeResponse& b) {
|
|
a.Swap(&b);
|
|
}
|
|
inline void Swap(GetShapeResponse* 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(GetShapeResponse* other) {
|
|
if (other == this) return;
|
|
GOOGLE_DCHECK(GetOwningArena() == other->GetOwningArena());
|
|
InternalSwap(other);
|
|
}
|
|
|
|
// implements Message ----------------------------------------------
|
|
|
|
GetShapeResponse* New(::PROTOBUF_NAMESPACE_ID::Arena* arena = nullptr) const final {
|
|
return CreateMaybeMessage<GetShapeResponse>(arena);
|
|
}
|
|
using ::PROTOBUF_NAMESPACE_ID::Message::CopyFrom;
|
|
void CopyFrom(const GetShapeResponse& from);
|
|
using ::PROTOBUF_NAMESPACE_ID::Message::MergeFrom;
|
|
void MergeFrom( const GetShapeResponse& from) {
|
|
GetShapeResponse::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(GetShapeResponse* other);
|
|
|
|
private:
|
|
friend class ::PROTOBUF_NAMESPACE_ID::internal::AnyMetadata;
|
|
static ::PROTOBUF_NAMESPACE_ID::StringPiece FullMessageName() {
|
|
return "xla.GetShapeResponse";
|
|
}
|
|
protected:
|
|
explicit GetShapeResponse(::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 {
|
|
kShapeFieldNumber = 1,
|
|
};
|
|
// .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.GetShapeResponse)
|
|
private:
|
|
class _Internal;
|
|
|
|
template <typename T> friend class ::PROTOBUF_NAMESPACE_ID::Arena::InternalHelper;
|
|
typedef void InternalArenaConstructable_;
|
|
typedef void DestructorSkippable_;
|
|
struct Impl_ {
|
|
::xla::ShapeProto* shape_;
|
|
mutable ::PROTOBUF_NAMESPACE_ID::internal::CachedSize _cached_size_;
|
|
};
|
|
union { Impl_ _impl_; };
|
|
friend struct ::TableStruct_xla_2fxla_2eproto;
|
|
};
|
|
// -------------------------------------------------------------------
|
|
|
|
class UnpackRequest final :
|
|
public ::PROTOBUF_NAMESPACE_ID::Message /* @@protoc_insertion_point(class_definition:xla.UnpackRequest) */ {
|
|
public:
|
|
inline UnpackRequest() : UnpackRequest(nullptr) {}
|
|
~UnpackRequest() override;
|
|
explicit PROTOBUF_CONSTEXPR UnpackRequest(::PROTOBUF_NAMESPACE_ID::internal::ConstantInitialized);
|
|
|
|
UnpackRequest(const UnpackRequest& from);
|
|
UnpackRequest(UnpackRequest&& from) noexcept
|
|
: UnpackRequest() {
|
|
*this = ::std::move(from);
|
|
}
|
|
|
|
inline UnpackRequest& operator=(const UnpackRequest& from) {
|
|
CopyFrom(from);
|
|
return *this;
|
|
}
|
|
inline UnpackRequest& operator=(UnpackRequest&& 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 UnpackRequest& default_instance() {
|
|
return *internal_default_instance();
|
|
}
|
|
static inline const UnpackRequest* internal_default_instance() {
|
|
return reinterpret_cast<const UnpackRequest*>(
|
|
&_UnpackRequest_default_instance_);
|
|
}
|
|
static constexpr int kIndexInFileMessages =
|
|
46;
|
|
|
|
friend void swap(UnpackRequest& a, UnpackRequest& b) {
|
|
a.Swap(&b);
|
|
}
|
|
inline void Swap(UnpackRequest* 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(UnpackRequest* other) {
|
|
if (other == this) return;
|
|
GOOGLE_DCHECK(GetOwningArena() == other->GetOwningArena());
|
|
InternalSwap(other);
|
|
}
|
|
|
|
// implements Message ----------------------------------------------
|
|
|
|
UnpackRequest* New(::PROTOBUF_NAMESPACE_ID::Arena* arena = nullptr) const final {
|
|
return CreateMaybeMessage<UnpackRequest>(arena);
|
|
}
|
|
using ::PROTOBUF_NAMESPACE_ID::Message::CopyFrom;
|
|
void CopyFrom(const UnpackRequest& from);
|
|
using ::PROTOBUF_NAMESPACE_ID::Message::MergeFrom;
|
|
void MergeFrom( const UnpackRequest& from) {
|
|
UnpackRequest::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(UnpackRequest* other);
|
|
|
|
private:
|
|
friend class ::PROTOBUF_NAMESPACE_ID::internal::AnyMetadata;
|
|
static ::PROTOBUF_NAMESPACE_ID::StringPiece FullMessageName() {
|
|
return "xla.UnpackRequest";
|
|
}
|
|
protected:
|
|
explicit UnpackRequest(::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 {
|
|
kDataFieldNumber = 1,
|
|
};
|
|
// .xla.GlobalDataHandle data = 1;
|
|
bool has_data() const;
|
|
private:
|
|
bool _internal_has_data() const;
|
|
public:
|
|
void clear_data();
|
|
const ::xla::GlobalDataHandle& data() const;
|
|
PROTOBUF_NODISCARD ::xla::GlobalDataHandle* release_data();
|
|
::xla::GlobalDataHandle* mutable_data();
|
|
void set_allocated_data(::xla::GlobalDataHandle* data);
|
|
private:
|
|
const ::xla::GlobalDataHandle& _internal_data() const;
|
|
::xla::GlobalDataHandle* _internal_mutable_data();
|
|
public:
|
|
void unsafe_arena_set_allocated_data(
|
|
::xla::GlobalDataHandle* data);
|
|
::xla::GlobalDataHandle* unsafe_arena_release_data();
|
|
|
|
// @@protoc_insertion_point(class_scope:xla.UnpackRequest)
|
|
private:
|
|
class _Internal;
|
|
|
|
template <typename T> friend class ::PROTOBUF_NAMESPACE_ID::Arena::InternalHelper;
|
|
typedef void InternalArenaConstructable_;
|
|
typedef void DestructorSkippable_;
|
|
struct Impl_ {
|
|
::xla::GlobalDataHandle* data_;
|
|
mutable ::PROTOBUF_NAMESPACE_ID::internal::CachedSize _cached_size_;
|
|
};
|
|
union { Impl_ _impl_; };
|
|
friend struct ::TableStruct_xla_2fxla_2eproto;
|
|
};
|
|
// -------------------------------------------------------------------
|
|
|
|
class UnpackResponse final :
|
|
public ::PROTOBUF_NAMESPACE_ID::Message /* @@protoc_insertion_point(class_definition:xla.UnpackResponse) */ {
|
|
public:
|
|
inline UnpackResponse() : UnpackResponse(nullptr) {}
|
|
~UnpackResponse() override;
|
|
explicit PROTOBUF_CONSTEXPR UnpackResponse(::PROTOBUF_NAMESPACE_ID::internal::ConstantInitialized);
|
|
|
|
UnpackResponse(const UnpackResponse& from);
|
|
UnpackResponse(UnpackResponse&& from) noexcept
|
|
: UnpackResponse() {
|
|
*this = ::std::move(from);
|
|
}
|
|
|
|
inline UnpackResponse& operator=(const UnpackResponse& from) {
|
|
CopyFrom(from);
|
|
return *this;
|
|
}
|
|
inline UnpackResponse& operator=(UnpackResponse&& 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 UnpackResponse& default_instance() {
|
|
return *internal_default_instance();
|
|
}
|
|
static inline const UnpackResponse* internal_default_instance() {
|
|
return reinterpret_cast<const UnpackResponse*>(
|
|
&_UnpackResponse_default_instance_);
|
|
}
|
|
static constexpr int kIndexInFileMessages =
|
|
47;
|
|
|
|
friend void swap(UnpackResponse& a, UnpackResponse& b) {
|
|
a.Swap(&b);
|
|
}
|
|
inline void Swap(UnpackResponse* 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(UnpackResponse* other) {
|
|
if (other == this) return;
|
|
GOOGLE_DCHECK(GetOwningArena() == other->GetOwningArena());
|
|
InternalSwap(other);
|
|
}
|
|
|
|
// implements Message ----------------------------------------------
|
|
|
|
UnpackResponse* New(::PROTOBUF_NAMESPACE_ID::Arena* arena = nullptr) const final {
|
|
return CreateMaybeMessage<UnpackResponse>(arena);
|
|
}
|
|
using ::PROTOBUF_NAMESPACE_ID::Message::CopyFrom;
|
|
void CopyFrom(const UnpackResponse& from);
|
|
using ::PROTOBUF_NAMESPACE_ID::Message::MergeFrom;
|
|
void MergeFrom( const UnpackResponse& from) {
|
|
UnpackResponse::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(UnpackResponse* other);
|
|
|
|
private:
|
|
friend class ::PROTOBUF_NAMESPACE_ID::internal::AnyMetadata;
|
|
static ::PROTOBUF_NAMESPACE_ID::StringPiece FullMessageName() {
|
|
return "xla.UnpackResponse";
|
|
}
|
|
protected:
|
|
explicit UnpackResponse(::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 {
|
|
kTiedDataFieldNumber = 1,
|
|
};
|
|
// repeated .xla.GlobalDataHandle tied_data = 1;
|
|
int tied_data_size() const;
|
|
private:
|
|
int _internal_tied_data_size() const;
|
|
public:
|
|
void clear_tied_data();
|
|
::xla::GlobalDataHandle* mutable_tied_data(int index);
|
|
::PROTOBUF_NAMESPACE_ID::RepeatedPtrField< ::xla::GlobalDataHandle >*
|
|
mutable_tied_data();
|
|
private:
|
|
const ::xla::GlobalDataHandle& _internal_tied_data(int index) const;
|
|
::xla::GlobalDataHandle* _internal_add_tied_data();
|
|
public:
|
|
const ::xla::GlobalDataHandle& tied_data(int index) const;
|
|
::xla::GlobalDataHandle* add_tied_data();
|
|
const ::PROTOBUF_NAMESPACE_ID::RepeatedPtrField< ::xla::GlobalDataHandle >&
|
|
tied_data() const;
|
|
|
|
// @@protoc_insertion_point(class_scope:xla.UnpackResponse)
|
|
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::GlobalDataHandle > tied_data_;
|
|
mutable ::PROTOBUF_NAMESPACE_ID::internal::CachedSize _cached_size_;
|
|
};
|
|
union { Impl_ _impl_; };
|
|
friend struct ::TableStruct_xla_2fxla_2eproto;
|
|
};
|
|
// -------------------------------------------------------------------
|
|
|
|
class ScheduleProto_Instruction final :
|
|
public ::PROTOBUF_NAMESPACE_ID::Message /* @@protoc_insertion_point(class_definition:xla.ScheduleProto.Instruction) */ {
|
|
public:
|
|
inline ScheduleProto_Instruction() : ScheduleProto_Instruction(nullptr) {}
|
|
~ScheduleProto_Instruction() override;
|
|
explicit PROTOBUF_CONSTEXPR ScheduleProto_Instruction(::PROTOBUF_NAMESPACE_ID::internal::ConstantInitialized);
|
|
|
|
ScheduleProto_Instruction(const ScheduleProto_Instruction& from);
|
|
ScheduleProto_Instruction(ScheduleProto_Instruction&& from) noexcept
|
|
: ScheduleProto_Instruction() {
|
|
*this = ::std::move(from);
|
|
}
|
|
|
|
inline ScheduleProto_Instruction& operator=(const ScheduleProto_Instruction& from) {
|
|
CopyFrom(from);
|
|
return *this;
|
|
}
|
|
inline ScheduleProto_Instruction& operator=(ScheduleProto_Instruction&& 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 ScheduleProto_Instruction& default_instance() {
|
|
return *internal_default_instance();
|
|
}
|
|
static inline const ScheduleProto_Instruction* internal_default_instance() {
|
|
return reinterpret_cast<const ScheduleProto_Instruction*>(
|
|
&_ScheduleProto_Instruction_default_instance_);
|
|
}
|
|
static constexpr int kIndexInFileMessages =
|
|
48;
|
|
|
|
friend void swap(ScheduleProto_Instruction& a, ScheduleProto_Instruction& b) {
|
|
a.Swap(&b);
|
|
}
|
|
inline void Swap(ScheduleProto_Instruction* 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(ScheduleProto_Instruction* other) {
|
|
if (other == this) return;
|
|
GOOGLE_DCHECK(GetOwningArena() == other->GetOwningArena());
|
|
InternalSwap(other);
|
|
}
|
|
|
|
// implements Message ----------------------------------------------
|
|
|
|
ScheduleProto_Instruction* New(::PROTOBUF_NAMESPACE_ID::Arena* arena = nullptr) const final {
|
|
return CreateMaybeMessage<ScheduleProto_Instruction>(arena);
|
|
}
|
|
using ::PROTOBUF_NAMESPACE_ID::Message::CopyFrom;
|
|
void CopyFrom(const ScheduleProto_Instruction& from);
|
|
using ::PROTOBUF_NAMESPACE_ID::Message::MergeFrom;
|
|
void MergeFrom( const ScheduleProto_Instruction& from) {
|
|
ScheduleProto_Instruction::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(ScheduleProto_Instruction* other);
|
|
|
|
private:
|
|
friend class ::PROTOBUF_NAMESPACE_ID::internal::AnyMetadata;
|
|
static ::PROTOBUF_NAMESPACE_ID::StringPiece FullMessageName() {
|
|
return "xla.ScheduleProto.Instruction";
|
|
}
|
|
protected:
|
|
explicit ScheduleProto_Instruction(::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 {
|
|
kIdFieldNumber = 1,
|
|
kStartTimestampCyclesFieldNumber = 2,
|
|
kEndTimestampCyclesFieldNumber = 3,
|
|
};
|
|
// int64 id = 1;
|
|
void clear_id();
|
|
int64_t id() const;
|
|
void set_id(int64_t value);
|
|
private:
|
|
int64_t _internal_id() const;
|
|
void _internal_set_id(int64_t value);
|
|
public:
|
|
|
|
// double start_timestamp_cycles = 2;
|
|
void clear_start_timestamp_cycles();
|
|
double start_timestamp_cycles() const;
|
|
void set_start_timestamp_cycles(double value);
|
|
private:
|
|
double _internal_start_timestamp_cycles() const;
|
|
void _internal_set_start_timestamp_cycles(double value);
|
|
public:
|
|
|
|
// double end_timestamp_cycles = 3;
|
|
void clear_end_timestamp_cycles();
|
|
double end_timestamp_cycles() const;
|
|
void set_end_timestamp_cycles(double value);
|
|
private:
|
|
double _internal_end_timestamp_cycles() const;
|
|
void _internal_set_end_timestamp_cycles(double value);
|
|
public:
|
|
|
|
// @@protoc_insertion_point(class_scope:xla.ScheduleProto.Instruction)
|
|
private:
|
|
class _Internal;
|
|
|
|
template <typename T> friend class ::PROTOBUF_NAMESPACE_ID::Arena::InternalHelper;
|
|
typedef void InternalArenaConstructable_;
|
|
typedef void DestructorSkippable_;
|
|
struct Impl_ {
|
|
int64_t id_;
|
|
double start_timestamp_cycles_;
|
|
double end_timestamp_cycles_;
|
|
mutable ::PROTOBUF_NAMESPACE_ID::internal::CachedSize _cached_size_;
|
|
};
|
|
union { Impl_ _impl_; };
|
|
friend struct ::TableStruct_xla_2fxla_2eproto;
|
|
};
|
|
// -------------------------------------------------------------------
|
|
|
|
class ScheduleProto final :
|
|
public ::PROTOBUF_NAMESPACE_ID::Message /* @@protoc_insertion_point(class_definition:xla.ScheduleProto) */ {
|
|
public:
|
|
inline ScheduleProto() : ScheduleProto(nullptr) {}
|
|
~ScheduleProto() override;
|
|
explicit PROTOBUF_CONSTEXPR ScheduleProto(::PROTOBUF_NAMESPACE_ID::internal::ConstantInitialized);
|
|
|
|
ScheduleProto(const ScheduleProto& from);
|
|
ScheduleProto(ScheduleProto&& from) noexcept
|
|
: ScheduleProto() {
|
|
*this = ::std::move(from);
|
|
}
|
|
|
|
inline ScheduleProto& operator=(const ScheduleProto& from) {
|
|
CopyFrom(from);
|
|
return *this;
|
|
}
|
|
inline ScheduleProto& operator=(ScheduleProto&& 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 ScheduleProto& default_instance() {
|
|
return *internal_default_instance();
|
|
}
|
|
static inline const ScheduleProto* internal_default_instance() {
|
|
return reinterpret_cast<const ScheduleProto*>(
|
|
&_ScheduleProto_default_instance_);
|
|
}
|
|
static constexpr int kIndexInFileMessages =
|
|
49;
|
|
|
|
friend void swap(ScheduleProto& a, ScheduleProto& b) {
|
|
a.Swap(&b);
|
|
}
|
|
inline void Swap(ScheduleProto* 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(ScheduleProto* other) {
|
|
if (other == this) return;
|
|
GOOGLE_DCHECK(GetOwningArena() == other->GetOwningArena());
|
|
InternalSwap(other);
|
|
}
|
|
|
|
// implements Message ----------------------------------------------
|
|
|
|
ScheduleProto* New(::PROTOBUF_NAMESPACE_ID::Arena* arena = nullptr) const final {
|
|
return CreateMaybeMessage<ScheduleProto>(arena);
|
|
}
|
|
using ::PROTOBUF_NAMESPACE_ID::Message::CopyFrom;
|
|
void CopyFrom(const ScheduleProto& from);
|
|
using ::PROTOBUF_NAMESPACE_ID::Message::MergeFrom;
|
|
void MergeFrom( const ScheduleProto& from) {
|
|
ScheduleProto::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(ScheduleProto* other);
|
|
|
|
private:
|
|
friend class ::PROTOBUF_NAMESPACE_ID::internal::AnyMetadata;
|
|
static ::PROTOBUF_NAMESPACE_ID::StringPiece FullMessageName() {
|
|
return "xla.ScheduleProto";
|
|
}
|
|
protected:
|
|
explicit ScheduleProto(::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 ScheduleProto_Instruction Instruction;
|
|
|
|
// accessors -------------------------------------------------------
|
|
|
|
enum : int {
|
|
kInstructionsFieldNumber = 1,
|
|
kHloModuleFieldNumber = 3,
|
|
kComputationIdFieldNumber = 2,
|
|
kCyclesPerMicrosecondFieldNumber = 4,
|
|
};
|
|
// repeated .xla.ScheduleProto.Instruction instructions = 1;
|
|
int instructions_size() const;
|
|
private:
|
|
int _internal_instructions_size() const;
|
|
public:
|
|
void clear_instructions();
|
|
::xla::ScheduleProto_Instruction* mutable_instructions(int index);
|
|
::PROTOBUF_NAMESPACE_ID::RepeatedPtrField< ::xla::ScheduleProto_Instruction >*
|
|
mutable_instructions();
|
|
private:
|
|
const ::xla::ScheduleProto_Instruction& _internal_instructions(int index) const;
|
|
::xla::ScheduleProto_Instruction* _internal_add_instructions();
|
|
public:
|
|
const ::xla::ScheduleProto_Instruction& instructions(int index) const;
|
|
::xla::ScheduleProto_Instruction* add_instructions();
|
|
const ::PROTOBUF_NAMESPACE_ID::RepeatedPtrField< ::xla::ScheduleProto_Instruction >&
|
|
instructions() const;
|
|
|
|
// .xla.HloModuleProto hlo_module = 3;
|
|
bool has_hlo_module() const;
|
|
private:
|
|
bool _internal_has_hlo_module() const;
|
|
public:
|
|
void clear_hlo_module();
|
|
const ::xla::HloModuleProto& hlo_module() const;
|
|
PROTOBUF_NODISCARD ::xla::HloModuleProto* release_hlo_module();
|
|
::xla::HloModuleProto* mutable_hlo_module();
|
|
void set_allocated_hlo_module(::xla::HloModuleProto* hlo_module);
|
|
private:
|
|
const ::xla::HloModuleProto& _internal_hlo_module() const;
|
|
::xla::HloModuleProto* _internal_mutable_hlo_module();
|
|
public:
|
|
void unsafe_arena_set_allocated_hlo_module(
|
|
::xla::HloModuleProto* hlo_module);
|
|
::xla::HloModuleProto* unsafe_arena_release_hlo_module();
|
|
|
|
// int64 computation_id = 2;
|
|
void clear_computation_id();
|
|
int64_t computation_id() const;
|
|
void set_computation_id(int64_t value);
|
|
private:
|
|
int64_t _internal_computation_id() const;
|
|
void _internal_set_computation_id(int64_t value);
|
|
public:
|
|
|
|
// int64 cycles_per_microsecond = 4;
|
|
void clear_cycles_per_microsecond();
|
|
int64_t cycles_per_microsecond() const;
|
|
void set_cycles_per_microsecond(int64_t value);
|
|
private:
|
|
int64_t _internal_cycles_per_microsecond() const;
|
|
void _internal_set_cycles_per_microsecond(int64_t value);
|
|
public:
|
|
|
|
// @@protoc_insertion_point(class_scope:xla.ScheduleProto)
|
|
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::ScheduleProto_Instruction > instructions_;
|
|
::xla::HloModuleProto* hlo_module_;
|
|
int64_t computation_id_;
|
|
int64_t cycles_per_microsecond_;
|
|
mutable ::PROTOBUF_NAMESPACE_ID::internal::CachedSize _cached_size_;
|
|
};
|
|
union { Impl_ _impl_; };
|
|
friend struct ::TableStruct_xla_2fxla_2eproto;
|
|
};
|
|
// ===================================================================
|
|
|
|
|
|
// ===================================================================
|
|
|
|
#ifdef __GNUC__
|
|
#pragma GCC diagnostic push
|
|
#pragma GCC diagnostic ignored "-Wstrict-aliasing"
|
|
#endif // __GNUC__
|
|
// CompilationEnvironmentsProto
|
|
|
|
// repeated .google.protobuf.Any environments = 1;
|
|
inline int CompilationEnvironmentsProto::_internal_environments_size() const {
|
|
return _impl_.environments_.size();
|
|
}
|
|
inline int CompilationEnvironmentsProto::environments_size() const {
|
|
return _internal_environments_size();
|
|
}
|
|
inline ::PROTOBUF_NAMESPACE_ID::Any* CompilationEnvironmentsProto::mutable_environments(int index) {
|
|
// @@protoc_insertion_point(field_mutable:xla.CompilationEnvironmentsProto.environments)
|
|
return _impl_.environments_.Mutable(index);
|
|
}
|
|
inline ::PROTOBUF_NAMESPACE_ID::RepeatedPtrField< ::PROTOBUF_NAMESPACE_ID::Any >*
|
|
CompilationEnvironmentsProto::mutable_environments() {
|
|
// @@protoc_insertion_point(field_mutable_list:xla.CompilationEnvironmentsProto.environments)
|
|
return &_impl_.environments_;
|
|
}
|
|
inline const ::PROTOBUF_NAMESPACE_ID::Any& CompilationEnvironmentsProto::_internal_environments(int index) const {
|
|
return _impl_.environments_.Get(index);
|
|
}
|
|
inline const ::PROTOBUF_NAMESPACE_ID::Any& CompilationEnvironmentsProto::environments(int index) const {
|
|
// @@protoc_insertion_point(field_get:xla.CompilationEnvironmentsProto.environments)
|
|
return _internal_environments(index);
|
|
}
|
|
inline ::PROTOBUF_NAMESPACE_ID::Any* CompilationEnvironmentsProto::_internal_add_environments() {
|
|
return _impl_.environments_.Add();
|
|
}
|
|
inline ::PROTOBUF_NAMESPACE_ID::Any* CompilationEnvironmentsProto::add_environments() {
|
|
::PROTOBUF_NAMESPACE_ID::Any* _add = _internal_add_environments();
|
|
// @@protoc_insertion_point(field_add:xla.CompilationEnvironmentsProto.environments)
|
|
return _add;
|
|
}
|
|
inline const ::PROTOBUF_NAMESPACE_ID::RepeatedPtrField< ::PROTOBUF_NAMESPACE_ID::Any >&
|
|
CompilationEnvironmentsProto::environments() const {
|
|
// @@protoc_insertion_point(field_list:xla.CompilationEnvironmentsProto.environments)
|
|
return _impl_.environments_;
|
|
}
|
|
|
|
// -------------------------------------------------------------------
|
|
|
|
// -------------------------------------------------------------------
|
|
|
|
// DebugOptions
|
|
|
|
// bool xla_hlo_graph_addresses = 2;
|
|
inline void DebugOptions::clear_xla_hlo_graph_addresses() {
|
|
_impl_.xla_hlo_graph_addresses_ = false;
|
|
}
|
|
inline bool DebugOptions::_internal_xla_hlo_graph_addresses() const {
|
|
return _impl_.xla_hlo_graph_addresses_;
|
|
}
|
|
inline bool DebugOptions::xla_hlo_graph_addresses() const {
|
|
// @@protoc_insertion_point(field_get:xla.DebugOptions.xla_hlo_graph_addresses)
|
|
return _internal_xla_hlo_graph_addresses();
|
|
}
|
|
inline void DebugOptions::_internal_set_xla_hlo_graph_addresses(bool value) {
|
|
|
|
_impl_.xla_hlo_graph_addresses_ = value;
|
|
}
|
|
inline void DebugOptions::set_xla_hlo_graph_addresses(bool value) {
|
|
_internal_set_xla_hlo_graph_addresses(value);
|
|
// @@protoc_insertion_point(field_set:xla.DebugOptions.xla_hlo_graph_addresses)
|
|
}
|
|
|
|
// bool xla_hlo_profile = 9;
|
|
inline void DebugOptions::clear_xla_hlo_profile() {
|
|
_impl_.xla_hlo_profile_ = false;
|
|
}
|
|
inline bool DebugOptions::_internal_xla_hlo_profile() const {
|
|
return _impl_.xla_hlo_profile_;
|
|
}
|
|
inline bool DebugOptions::xla_hlo_profile() const {
|
|
// @@protoc_insertion_point(field_get:xla.DebugOptions.xla_hlo_profile)
|
|
return _internal_xla_hlo_profile();
|
|
}
|
|
inline void DebugOptions::_internal_set_xla_hlo_profile(bool value) {
|
|
|
|
_impl_.xla_hlo_profile_ = value;
|
|
}
|
|
inline void DebugOptions::set_xla_hlo_profile(bool value) {
|
|
_internal_set_xla_hlo_profile(value);
|
|
// @@protoc_insertion_point(field_set:xla.DebugOptions.xla_hlo_profile)
|
|
}
|
|
|
|
// repeated string xla_disable_hlo_passes = 30;
|
|
inline int DebugOptions::_internal_xla_disable_hlo_passes_size() const {
|
|
return _impl_.xla_disable_hlo_passes_.size();
|
|
}
|
|
inline int DebugOptions::xla_disable_hlo_passes_size() const {
|
|
return _internal_xla_disable_hlo_passes_size();
|
|
}
|
|
inline void DebugOptions::clear_xla_disable_hlo_passes() {
|
|
_impl_.xla_disable_hlo_passes_.Clear();
|
|
}
|
|
inline std::string* DebugOptions::add_xla_disable_hlo_passes() {
|
|
std::string* _s = _internal_add_xla_disable_hlo_passes();
|
|
// @@protoc_insertion_point(field_add_mutable:xla.DebugOptions.xla_disable_hlo_passes)
|
|
return _s;
|
|
}
|
|
inline const std::string& DebugOptions::_internal_xla_disable_hlo_passes(int index) const {
|
|
return _impl_.xla_disable_hlo_passes_.Get(index);
|
|
}
|
|
inline const std::string& DebugOptions::xla_disable_hlo_passes(int index) const {
|
|
// @@protoc_insertion_point(field_get:xla.DebugOptions.xla_disable_hlo_passes)
|
|
return _internal_xla_disable_hlo_passes(index);
|
|
}
|
|
inline std::string* DebugOptions::mutable_xla_disable_hlo_passes(int index) {
|
|
// @@protoc_insertion_point(field_mutable:xla.DebugOptions.xla_disable_hlo_passes)
|
|
return _impl_.xla_disable_hlo_passes_.Mutable(index);
|
|
}
|
|
inline void DebugOptions::set_xla_disable_hlo_passes(int index, const std::string& value) {
|
|
_impl_.xla_disable_hlo_passes_.Mutable(index)->assign(value);
|
|
// @@protoc_insertion_point(field_set:xla.DebugOptions.xla_disable_hlo_passes)
|
|
}
|
|
inline void DebugOptions::set_xla_disable_hlo_passes(int index, std::string&& value) {
|
|
_impl_.xla_disable_hlo_passes_.Mutable(index)->assign(std::move(value));
|
|
// @@protoc_insertion_point(field_set:xla.DebugOptions.xla_disable_hlo_passes)
|
|
}
|
|
inline void DebugOptions::set_xla_disable_hlo_passes(int index, const char* value) {
|
|
GOOGLE_DCHECK(value != nullptr);
|
|
_impl_.xla_disable_hlo_passes_.Mutable(index)->assign(value);
|
|
// @@protoc_insertion_point(field_set_char:xla.DebugOptions.xla_disable_hlo_passes)
|
|
}
|
|
inline void DebugOptions::set_xla_disable_hlo_passes(int index, const char* value, size_t size) {
|
|
_impl_.xla_disable_hlo_passes_.Mutable(index)->assign(
|
|
reinterpret_cast<const char*>(value), size);
|
|
// @@protoc_insertion_point(field_set_pointer:xla.DebugOptions.xla_disable_hlo_passes)
|
|
}
|
|
inline std::string* DebugOptions::_internal_add_xla_disable_hlo_passes() {
|
|
return _impl_.xla_disable_hlo_passes_.Add();
|
|
}
|
|
inline void DebugOptions::add_xla_disable_hlo_passes(const std::string& value) {
|
|
_impl_.xla_disable_hlo_passes_.Add()->assign(value);
|
|
// @@protoc_insertion_point(field_add:xla.DebugOptions.xla_disable_hlo_passes)
|
|
}
|
|
inline void DebugOptions::add_xla_disable_hlo_passes(std::string&& value) {
|
|
_impl_.xla_disable_hlo_passes_.Add(std::move(value));
|
|
// @@protoc_insertion_point(field_add:xla.DebugOptions.xla_disable_hlo_passes)
|
|
}
|
|
inline void DebugOptions::add_xla_disable_hlo_passes(const char* value) {
|
|
GOOGLE_DCHECK(value != nullptr);
|
|
_impl_.xla_disable_hlo_passes_.Add()->assign(value);
|
|
// @@protoc_insertion_point(field_add_char:xla.DebugOptions.xla_disable_hlo_passes)
|
|
}
|
|
inline void DebugOptions::add_xla_disable_hlo_passes(const char* value, size_t size) {
|
|
_impl_.xla_disable_hlo_passes_.Add()->assign(reinterpret_cast<const char*>(value), size);
|
|
// @@protoc_insertion_point(field_add_pointer:xla.DebugOptions.xla_disable_hlo_passes)
|
|
}
|
|
inline const ::PROTOBUF_NAMESPACE_ID::RepeatedPtrField<std::string>&
|
|
DebugOptions::xla_disable_hlo_passes() const {
|
|
// @@protoc_insertion_point(field_list:xla.DebugOptions.xla_disable_hlo_passes)
|
|
return _impl_.xla_disable_hlo_passes_;
|
|
}
|
|
inline ::PROTOBUF_NAMESPACE_ID::RepeatedPtrField<std::string>*
|
|
DebugOptions::mutable_xla_disable_hlo_passes() {
|
|
// @@protoc_insertion_point(field_mutable_list:xla.DebugOptions.xla_disable_hlo_passes)
|
|
return &_impl_.xla_disable_hlo_passes_;
|
|
}
|
|
|
|
// repeated string xla_enable_hlo_passes_only = 124;
|
|
inline int DebugOptions::_internal_xla_enable_hlo_passes_only_size() const {
|
|
return _impl_.xla_enable_hlo_passes_only_.size();
|
|
}
|
|
inline int DebugOptions::xla_enable_hlo_passes_only_size() const {
|
|
return _internal_xla_enable_hlo_passes_only_size();
|
|
}
|
|
inline void DebugOptions::clear_xla_enable_hlo_passes_only() {
|
|
_impl_.xla_enable_hlo_passes_only_.Clear();
|
|
}
|
|
inline std::string* DebugOptions::add_xla_enable_hlo_passes_only() {
|
|
std::string* _s = _internal_add_xla_enable_hlo_passes_only();
|
|
// @@protoc_insertion_point(field_add_mutable:xla.DebugOptions.xla_enable_hlo_passes_only)
|
|
return _s;
|
|
}
|
|
inline const std::string& DebugOptions::_internal_xla_enable_hlo_passes_only(int index) const {
|
|
return _impl_.xla_enable_hlo_passes_only_.Get(index);
|
|
}
|
|
inline const std::string& DebugOptions::xla_enable_hlo_passes_only(int index) const {
|
|
// @@protoc_insertion_point(field_get:xla.DebugOptions.xla_enable_hlo_passes_only)
|
|
return _internal_xla_enable_hlo_passes_only(index);
|
|
}
|
|
inline std::string* DebugOptions::mutable_xla_enable_hlo_passes_only(int index) {
|
|
// @@protoc_insertion_point(field_mutable:xla.DebugOptions.xla_enable_hlo_passes_only)
|
|
return _impl_.xla_enable_hlo_passes_only_.Mutable(index);
|
|
}
|
|
inline void DebugOptions::set_xla_enable_hlo_passes_only(int index, const std::string& value) {
|
|
_impl_.xla_enable_hlo_passes_only_.Mutable(index)->assign(value);
|
|
// @@protoc_insertion_point(field_set:xla.DebugOptions.xla_enable_hlo_passes_only)
|
|
}
|
|
inline void DebugOptions::set_xla_enable_hlo_passes_only(int index, std::string&& value) {
|
|
_impl_.xla_enable_hlo_passes_only_.Mutable(index)->assign(std::move(value));
|
|
// @@protoc_insertion_point(field_set:xla.DebugOptions.xla_enable_hlo_passes_only)
|
|
}
|
|
inline void DebugOptions::set_xla_enable_hlo_passes_only(int index, const char* value) {
|
|
GOOGLE_DCHECK(value != nullptr);
|
|
_impl_.xla_enable_hlo_passes_only_.Mutable(index)->assign(value);
|
|
// @@protoc_insertion_point(field_set_char:xla.DebugOptions.xla_enable_hlo_passes_only)
|
|
}
|
|
inline void DebugOptions::set_xla_enable_hlo_passes_only(int index, const char* value, size_t size) {
|
|
_impl_.xla_enable_hlo_passes_only_.Mutable(index)->assign(
|
|
reinterpret_cast<const char*>(value), size);
|
|
// @@protoc_insertion_point(field_set_pointer:xla.DebugOptions.xla_enable_hlo_passes_only)
|
|
}
|
|
inline std::string* DebugOptions::_internal_add_xla_enable_hlo_passes_only() {
|
|
return _impl_.xla_enable_hlo_passes_only_.Add();
|
|
}
|
|
inline void DebugOptions::add_xla_enable_hlo_passes_only(const std::string& value) {
|
|
_impl_.xla_enable_hlo_passes_only_.Add()->assign(value);
|
|
// @@protoc_insertion_point(field_add:xla.DebugOptions.xla_enable_hlo_passes_only)
|
|
}
|
|
inline void DebugOptions::add_xla_enable_hlo_passes_only(std::string&& value) {
|
|
_impl_.xla_enable_hlo_passes_only_.Add(std::move(value));
|
|
// @@protoc_insertion_point(field_add:xla.DebugOptions.xla_enable_hlo_passes_only)
|
|
}
|
|
inline void DebugOptions::add_xla_enable_hlo_passes_only(const char* value) {
|
|
GOOGLE_DCHECK(value != nullptr);
|
|
_impl_.xla_enable_hlo_passes_only_.Add()->assign(value);
|
|
// @@protoc_insertion_point(field_add_char:xla.DebugOptions.xla_enable_hlo_passes_only)
|
|
}
|
|
inline void DebugOptions::add_xla_enable_hlo_passes_only(const char* value, size_t size) {
|
|
_impl_.xla_enable_hlo_passes_only_.Add()->assign(reinterpret_cast<const char*>(value), size);
|
|
// @@protoc_insertion_point(field_add_pointer:xla.DebugOptions.xla_enable_hlo_passes_only)
|
|
}
|
|
inline const ::PROTOBUF_NAMESPACE_ID::RepeatedPtrField<std::string>&
|
|
DebugOptions::xla_enable_hlo_passes_only() const {
|
|
// @@protoc_insertion_point(field_list:xla.DebugOptions.xla_enable_hlo_passes_only)
|
|
return _impl_.xla_enable_hlo_passes_only_;
|
|
}
|
|
inline ::PROTOBUF_NAMESPACE_ID::RepeatedPtrField<std::string>*
|
|
DebugOptions::mutable_xla_enable_hlo_passes_only() {
|
|
// @@protoc_insertion_point(field_mutable_list:xla.DebugOptions.xla_enable_hlo_passes_only)
|
|
return &_impl_.xla_enable_hlo_passes_only_;
|
|
}
|
|
|
|
// bool xla_disable_all_hlo_passes = 104;
|
|
inline void DebugOptions::clear_xla_disable_all_hlo_passes() {
|
|
_impl_.xla_disable_all_hlo_passes_ = false;
|
|
}
|
|
inline bool DebugOptions::_internal_xla_disable_all_hlo_passes() const {
|
|
return _impl_.xla_disable_all_hlo_passes_;
|
|
}
|
|
inline bool DebugOptions::xla_disable_all_hlo_passes() const {
|
|
// @@protoc_insertion_point(field_get:xla.DebugOptions.xla_disable_all_hlo_passes)
|
|
return _internal_xla_disable_all_hlo_passes();
|
|
}
|
|
inline void DebugOptions::_internal_set_xla_disable_all_hlo_passes(bool value) {
|
|
|
|
_impl_.xla_disable_all_hlo_passes_ = value;
|
|
}
|
|
inline void DebugOptions::set_xla_disable_all_hlo_passes(bool value) {
|
|
_internal_set_xla_disable_all_hlo_passes(value);
|
|
// @@protoc_insertion_point(field_set:xla.DebugOptions.xla_disable_all_hlo_passes)
|
|
}
|
|
|
|
// int32 xla_backend_optimization_level = 31;
|
|
inline void DebugOptions::clear_xla_backend_optimization_level() {
|
|
_impl_.xla_backend_optimization_level_ = 0;
|
|
}
|
|
inline int32_t DebugOptions::_internal_xla_backend_optimization_level() const {
|
|
return _impl_.xla_backend_optimization_level_;
|
|
}
|
|
inline int32_t DebugOptions::xla_backend_optimization_level() const {
|
|
// @@protoc_insertion_point(field_get:xla.DebugOptions.xla_backend_optimization_level)
|
|
return _internal_xla_backend_optimization_level();
|
|
}
|
|
inline void DebugOptions::_internal_set_xla_backend_optimization_level(int32_t value) {
|
|
|
|
_impl_.xla_backend_optimization_level_ = value;
|
|
}
|
|
inline void DebugOptions::set_xla_backend_optimization_level(int32_t value) {
|
|
_internal_set_xla_backend_optimization_level(value);
|
|
// @@protoc_insertion_point(field_set:xla.DebugOptions.xla_backend_optimization_level)
|
|
}
|
|
|
|
// bool xla_embed_ir_in_executable = 33;
|
|
inline void DebugOptions::clear_xla_embed_ir_in_executable() {
|
|
_impl_.xla_embed_ir_in_executable_ = false;
|
|
}
|
|
inline bool DebugOptions::_internal_xla_embed_ir_in_executable() const {
|
|
return _impl_.xla_embed_ir_in_executable_;
|
|
}
|
|
inline bool DebugOptions::xla_embed_ir_in_executable() const {
|
|
// @@protoc_insertion_point(field_get:xla.DebugOptions.xla_embed_ir_in_executable)
|
|
return _internal_xla_embed_ir_in_executable();
|
|
}
|
|
inline void DebugOptions::_internal_set_xla_embed_ir_in_executable(bool value) {
|
|
|
|
_impl_.xla_embed_ir_in_executable_ = value;
|
|
}
|
|
inline void DebugOptions::set_xla_embed_ir_in_executable(bool value) {
|
|
_internal_set_xla_embed_ir_in_executable(value);
|
|
// @@protoc_insertion_point(field_set:xla.DebugOptions.xla_embed_ir_in_executable)
|
|
}
|
|
|
|
// bool xla_eliminate_hlo_implicit_broadcast = 35;
|
|
inline void DebugOptions::clear_xla_eliminate_hlo_implicit_broadcast() {
|
|
_impl_.xla_eliminate_hlo_implicit_broadcast_ = false;
|
|
}
|
|
inline bool DebugOptions::_internal_xla_eliminate_hlo_implicit_broadcast() const {
|
|
return _impl_.xla_eliminate_hlo_implicit_broadcast_;
|
|
}
|
|
inline bool DebugOptions::xla_eliminate_hlo_implicit_broadcast() const {
|
|
// @@protoc_insertion_point(field_get:xla.DebugOptions.xla_eliminate_hlo_implicit_broadcast)
|
|
return _internal_xla_eliminate_hlo_implicit_broadcast();
|
|
}
|
|
inline void DebugOptions::_internal_set_xla_eliminate_hlo_implicit_broadcast(bool value) {
|
|
|
|
_impl_.xla_eliminate_hlo_implicit_broadcast_ = value;
|
|
}
|
|
inline void DebugOptions::set_xla_eliminate_hlo_implicit_broadcast(bool value) {
|
|
_internal_set_xla_eliminate_hlo_implicit_broadcast(value);
|
|
// @@protoc_insertion_point(field_set:xla.DebugOptions.xla_eliminate_hlo_implicit_broadcast)
|
|
}
|
|
|
|
// bool xla_cpu_multi_thread_eigen = 60;
|
|
inline void DebugOptions::clear_xla_cpu_multi_thread_eigen() {
|
|
_impl_.xla_cpu_multi_thread_eigen_ = false;
|
|
}
|
|
inline bool DebugOptions::_internal_xla_cpu_multi_thread_eigen() const {
|
|
return _impl_.xla_cpu_multi_thread_eigen_;
|
|
}
|
|
inline bool DebugOptions::xla_cpu_multi_thread_eigen() const {
|
|
// @@protoc_insertion_point(field_get:xla.DebugOptions.xla_cpu_multi_thread_eigen)
|
|
return _internal_xla_cpu_multi_thread_eigen();
|
|
}
|
|
inline void DebugOptions::_internal_set_xla_cpu_multi_thread_eigen(bool value) {
|
|
|
|
_impl_.xla_cpu_multi_thread_eigen_ = value;
|
|
}
|
|
inline void DebugOptions::set_xla_cpu_multi_thread_eigen(bool value) {
|
|
_internal_set_xla_cpu_multi_thread_eigen(value);
|
|
// @@protoc_insertion_point(field_set:xla.DebugOptions.xla_cpu_multi_thread_eigen)
|
|
}
|
|
|
|
// string xla_gpu_cuda_data_dir = 61;
|
|
inline void DebugOptions::clear_xla_gpu_cuda_data_dir() {
|
|
_impl_.xla_gpu_cuda_data_dir_.ClearToEmpty();
|
|
}
|
|
inline const std::string& DebugOptions::xla_gpu_cuda_data_dir() const {
|
|
// @@protoc_insertion_point(field_get:xla.DebugOptions.xla_gpu_cuda_data_dir)
|
|
return _internal_xla_gpu_cuda_data_dir();
|
|
}
|
|
template <typename ArgT0, typename... ArgT>
|
|
inline PROTOBUF_ALWAYS_INLINE
|
|
void DebugOptions::set_xla_gpu_cuda_data_dir(ArgT0&& arg0, ArgT... args) {
|
|
|
|
_impl_.xla_gpu_cuda_data_dir_.Set(static_cast<ArgT0 &&>(arg0), args..., GetArenaForAllocation());
|
|
// @@protoc_insertion_point(field_set:xla.DebugOptions.xla_gpu_cuda_data_dir)
|
|
}
|
|
inline std::string* DebugOptions::mutable_xla_gpu_cuda_data_dir() {
|
|
std::string* _s = _internal_mutable_xla_gpu_cuda_data_dir();
|
|
// @@protoc_insertion_point(field_mutable:xla.DebugOptions.xla_gpu_cuda_data_dir)
|
|
return _s;
|
|
}
|
|
inline const std::string& DebugOptions::_internal_xla_gpu_cuda_data_dir() const {
|
|
return _impl_.xla_gpu_cuda_data_dir_.Get();
|
|
}
|
|
inline void DebugOptions::_internal_set_xla_gpu_cuda_data_dir(const std::string& value) {
|
|
|
|
_impl_.xla_gpu_cuda_data_dir_.Set(value, GetArenaForAllocation());
|
|
}
|
|
inline std::string* DebugOptions::_internal_mutable_xla_gpu_cuda_data_dir() {
|
|
|
|
return _impl_.xla_gpu_cuda_data_dir_.Mutable(GetArenaForAllocation());
|
|
}
|
|
inline std::string* DebugOptions::release_xla_gpu_cuda_data_dir() {
|
|
// @@protoc_insertion_point(field_release:xla.DebugOptions.xla_gpu_cuda_data_dir)
|
|
return _impl_.xla_gpu_cuda_data_dir_.Release();
|
|
}
|
|
inline void DebugOptions::set_allocated_xla_gpu_cuda_data_dir(std::string* xla_gpu_cuda_data_dir) {
|
|
if (xla_gpu_cuda_data_dir != nullptr) {
|
|
|
|
} else {
|
|
|
|
}
|
|
_impl_.xla_gpu_cuda_data_dir_.SetAllocated(xla_gpu_cuda_data_dir, GetArenaForAllocation());
|
|
#ifdef PROTOBUF_FORCE_COPY_DEFAULT_STRING
|
|
if (_impl_.xla_gpu_cuda_data_dir_.IsDefault()) {
|
|
_impl_.xla_gpu_cuda_data_dir_.Set("", GetArenaForAllocation());
|
|
}
|
|
#endif // PROTOBUF_FORCE_COPY_DEFAULT_STRING
|
|
// @@protoc_insertion_point(field_set_allocated:xla.DebugOptions.xla_gpu_cuda_data_dir)
|
|
}
|
|
|
|
// bool xla_gpu_ftz = 62;
|
|
inline void DebugOptions::clear_xla_gpu_ftz() {
|
|
_impl_.xla_gpu_ftz_ = false;
|
|
}
|
|
inline bool DebugOptions::_internal_xla_gpu_ftz() const {
|
|
return _impl_.xla_gpu_ftz_;
|
|
}
|
|
inline bool DebugOptions::xla_gpu_ftz() const {
|
|
// @@protoc_insertion_point(field_get:xla.DebugOptions.xla_gpu_ftz)
|
|
return _internal_xla_gpu_ftz();
|
|
}
|
|
inline void DebugOptions::_internal_set_xla_gpu_ftz(bool value) {
|
|
|
|
_impl_.xla_gpu_ftz_ = value;
|
|
}
|
|
inline void DebugOptions::set_xla_gpu_ftz(bool value) {
|
|
_internal_set_xla_gpu_ftz(value);
|
|
// @@protoc_insertion_point(field_set:xla.DebugOptions.xla_gpu_ftz)
|
|
}
|
|
|
|
// bool xla_llvm_enable_alias_scope_metadata = 70;
|
|
inline void DebugOptions::clear_xla_llvm_enable_alias_scope_metadata() {
|
|
_impl_.xla_llvm_enable_alias_scope_metadata_ = false;
|
|
}
|
|
inline bool DebugOptions::_internal_xla_llvm_enable_alias_scope_metadata() const {
|
|
return _impl_.xla_llvm_enable_alias_scope_metadata_;
|
|
}
|
|
inline bool DebugOptions::xla_llvm_enable_alias_scope_metadata() const {
|
|
// @@protoc_insertion_point(field_get:xla.DebugOptions.xla_llvm_enable_alias_scope_metadata)
|
|
return _internal_xla_llvm_enable_alias_scope_metadata();
|
|
}
|
|
inline void DebugOptions::_internal_set_xla_llvm_enable_alias_scope_metadata(bool value) {
|
|
|
|
_impl_.xla_llvm_enable_alias_scope_metadata_ = value;
|
|
}
|
|
inline void DebugOptions::set_xla_llvm_enable_alias_scope_metadata(bool value) {
|
|
_internal_set_xla_llvm_enable_alias_scope_metadata(value);
|
|
// @@protoc_insertion_point(field_set:xla.DebugOptions.xla_llvm_enable_alias_scope_metadata)
|
|
}
|
|
|
|
// bool xla_llvm_enable_noalias_metadata = 71;
|
|
inline void DebugOptions::clear_xla_llvm_enable_noalias_metadata() {
|
|
_impl_.xla_llvm_enable_noalias_metadata_ = false;
|
|
}
|
|
inline bool DebugOptions::_internal_xla_llvm_enable_noalias_metadata() const {
|
|
return _impl_.xla_llvm_enable_noalias_metadata_;
|
|
}
|
|
inline bool DebugOptions::xla_llvm_enable_noalias_metadata() const {
|
|
// @@protoc_insertion_point(field_get:xla.DebugOptions.xla_llvm_enable_noalias_metadata)
|
|
return _internal_xla_llvm_enable_noalias_metadata();
|
|
}
|
|
inline void DebugOptions::_internal_set_xla_llvm_enable_noalias_metadata(bool value) {
|
|
|
|
_impl_.xla_llvm_enable_noalias_metadata_ = value;
|
|
}
|
|
inline void DebugOptions::set_xla_llvm_enable_noalias_metadata(bool value) {
|
|
_internal_set_xla_llvm_enable_noalias_metadata(value);
|
|
// @@protoc_insertion_point(field_set:xla.DebugOptions.xla_llvm_enable_noalias_metadata)
|
|
}
|
|
|
|
// bool xla_llvm_enable_invariant_load_metadata = 72;
|
|
inline void DebugOptions::clear_xla_llvm_enable_invariant_load_metadata() {
|
|
_impl_.xla_llvm_enable_invariant_load_metadata_ = false;
|
|
}
|
|
inline bool DebugOptions::_internal_xla_llvm_enable_invariant_load_metadata() const {
|
|
return _impl_.xla_llvm_enable_invariant_load_metadata_;
|
|
}
|
|
inline bool DebugOptions::xla_llvm_enable_invariant_load_metadata() const {
|
|
// @@protoc_insertion_point(field_get:xla.DebugOptions.xla_llvm_enable_invariant_load_metadata)
|
|
return _internal_xla_llvm_enable_invariant_load_metadata();
|
|
}
|
|
inline void DebugOptions::_internal_set_xla_llvm_enable_invariant_load_metadata(bool value) {
|
|
|
|
_impl_.xla_llvm_enable_invariant_load_metadata_ = value;
|
|
}
|
|
inline void DebugOptions::set_xla_llvm_enable_invariant_load_metadata(bool value) {
|
|
_internal_set_xla_llvm_enable_invariant_load_metadata(value);
|
|
// @@protoc_insertion_point(field_set:xla.DebugOptions.xla_llvm_enable_invariant_load_metadata)
|
|
}
|
|
|
|
// bool xla_llvm_disable_expensive_passes = 73;
|
|
inline void DebugOptions::clear_xla_llvm_disable_expensive_passes() {
|
|
_impl_.xla_llvm_disable_expensive_passes_ = false;
|
|
}
|
|
inline bool DebugOptions::_internal_xla_llvm_disable_expensive_passes() const {
|
|
return _impl_.xla_llvm_disable_expensive_passes_;
|
|
}
|
|
inline bool DebugOptions::xla_llvm_disable_expensive_passes() const {
|
|
// @@protoc_insertion_point(field_get:xla.DebugOptions.xla_llvm_disable_expensive_passes)
|
|
return _internal_xla_llvm_disable_expensive_passes();
|
|
}
|
|
inline void DebugOptions::_internal_set_xla_llvm_disable_expensive_passes(bool value) {
|
|
|
|
_impl_.xla_llvm_disable_expensive_passes_ = value;
|
|
}
|
|
inline void DebugOptions::set_xla_llvm_disable_expensive_passes(bool value) {
|
|
_internal_set_xla_llvm_disable_expensive_passes(value);
|
|
// @@protoc_insertion_point(field_set:xla.DebugOptions.xla_llvm_disable_expensive_passes)
|
|
}
|
|
|
|
// bool xla_test_all_output_layouts = 90;
|
|
inline void DebugOptions::clear_xla_test_all_output_layouts() {
|
|
_impl_.xla_test_all_output_layouts_ = false;
|
|
}
|
|
inline bool DebugOptions::_internal_xla_test_all_output_layouts() const {
|
|
return _impl_.xla_test_all_output_layouts_;
|
|
}
|
|
inline bool DebugOptions::xla_test_all_output_layouts() const {
|
|
// @@protoc_insertion_point(field_get:xla.DebugOptions.xla_test_all_output_layouts)
|
|
return _internal_xla_test_all_output_layouts();
|
|
}
|
|
inline void DebugOptions::_internal_set_xla_test_all_output_layouts(bool value) {
|
|
|
|
_impl_.xla_test_all_output_layouts_ = value;
|
|
}
|
|
inline void DebugOptions::set_xla_test_all_output_layouts(bool value) {
|
|
_internal_set_xla_test_all_output_layouts(value);
|
|
// @@protoc_insertion_point(field_set:xla.DebugOptions.xla_test_all_output_layouts)
|
|
}
|
|
|
|
// bool xla_test_all_input_layouts = 91;
|
|
inline void DebugOptions::clear_xla_test_all_input_layouts() {
|
|
_impl_.xla_test_all_input_layouts_ = false;
|
|
}
|
|
inline bool DebugOptions::_internal_xla_test_all_input_layouts() const {
|
|
return _impl_.xla_test_all_input_layouts_;
|
|
}
|
|
inline bool DebugOptions::xla_test_all_input_layouts() const {
|
|
// @@protoc_insertion_point(field_get:xla.DebugOptions.xla_test_all_input_layouts)
|
|
return _internal_xla_test_all_input_layouts();
|
|
}
|
|
inline void DebugOptions::_internal_set_xla_test_all_input_layouts(bool value) {
|
|
|
|
_impl_.xla_test_all_input_layouts_ = value;
|
|
}
|
|
inline void DebugOptions::set_xla_test_all_input_layouts(bool value) {
|
|
_internal_set_xla_test_all_input_layouts(value);
|
|
// @@protoc_insertion_point(field_set:xla.DebugOptions.xla_test_all_input_layouts)
|
|
}
|
|
|
|
// bool xla_hlo_graph_sharding_color = 92;
|
|
inline void DebugOptions::clear_xla_hlo_graph_sharding_color() {
|
|
_impl_.xla_hlo_graph_sharding_color_ = false;
|
|
}
|
|
inline bool DebugOptions::_internal_xla_hlo_graph_sharding_color() const {
|
|
return _impl_.xla_hlo_graph_sharding_color_;
|
|
}
|
|
inline bool DebugOptions::xla_hlo_graph_sharding_color() const {
|
|
// @@protoc_insertion_point(field_get:xla.DebugOptions.xla_hlo_graph_sharding_color)
|
|
return _internal_xla_hlo_graph_sharding_color();
|
|
}
|
|
inline void DebugOptions::_internal_set_xla_hlo_graph_sharding_color(bool value) {
|
|
|
|
_impl_.xla_hlo_graph_sharding_color_ = value;
|
|
}
|
|
inline void DebugOptions::set_xla_hlo_graph_sharding_color(bool value) {
|
|
_internal_set_xla_hlo_graph_sharding_color(value);
|
|
// @@protoc_insertion_point(field_set:xla.DebugOptions.xla_hlo_graph_sharding_color)
|
|
}
|
|
|
|
// bool xla_cpu_use_mkl_dnn = 97;
|
|
inline void DebugOptions::clear_xla_cpu_use_mkl_dnn() {
|
|
_impl_.xla_cpu_use_mkl_dnn_ = false;
|
|
}
|
|
inline bool DebugOptions::_internal_xla_cpu_use_mkl_dnn() const {
|
|
return _impl_.xla_cpu_use_mkl_dnn_;
|
|
}
|
|
inline bool DebugOptions::xla_cpu_use_mkl_dnn() const {
|
|
// @@protoc_insertion_point(field_get:xla.DebugOptions.xla_cpu_use_mkl_dnn)
|
|
return _internal_xla_cpu_use_mkl_dnn();
|
|
}
|
|
inline void DebugOptions::_internal_set_xla_cpu_use_mkl_dnn(bool value) {
|
|
|
|
_impl_.xla_cpu_use_mkl_dnn_ = value;
|
|
}
|
|
inline void DebugOptions::set_xla_cpu_use_mkl_dnn(bool value) {
|
|
_internal_set_xla_cpu_use_mkl_dnn(value);
|
|
// @@protoc_insertion_point(field_set:xla.DebugOptions.xla_cpu_use_mkl_dnn)
|
|
}
|
|
|
|
// bool xla_cpu_use_xla_runtime = 177;
|
|
inline void DebugOptions::clear_xla_cpu_use_xla_runtime() {
|
|
_impl_.xla_cpu_use_xla_runtime_ = false;
|
|
}
|
|
inline bool DebugOptions::_internal_xla_cpu_use_xla_runtime() const {
|
|
return _impl_.xla_cpu_use_xla_runtime_;
|
|
}
|
|
inline bool DebugOptions::xla_cpu_use_xla_runtime() const {
|
|
// @@protoc_insertion_point(field_get:xla.DebugOptions.xla_cpu_use_xla_runtime)
|
|
return _internal_xla_cpu_use_xla_runtime();
|
|
}
|
|
inline void DebugOptions::_internal_set_xla_cpu_use_xla_runtime(bool value) {
|
|
|
|
_impl_.xla_cpu_use_xla_runtime_ = value;
|
|
}
|
|
inline void DebugOptions::set_xla_cpu_use_xla_runtime(bool value) {
|
|
_internal_set_xla_cpu_use_xla_runtime(value);
|
|
// @@protoc_insertion_point(field_set:xla.DebugOptions.xla_cpu_use_xla_runtime)
|
|
}
|
|
|
|
// bool xla_cpu_enable_fast_math = 99;
|
|
inline void DebugOptions::clear_xla_cpu_enable_fast_math() {
|
|
_impl_.xla_cpu_enable_fast_math_ = false;
|
|
}
|
|
inline bool DebugOptions::_internal_xla_cpu_enable_fast_math() const {
|
|
return _impl_.xla_cpu_enable_fast_math_;
|
|
}
|
|
inline bool DebugOptions::xla_cpu_enable_fast_math() const {
|
|
// @@protoc_insertion_point(field_get:xla.DebugOptions.xla_cpu_enable_fast_math)
|
|
return _internal_xla_cpu_enable_fast_math();
|
|
}
|
|
inline void DebugOptions::_internal_set_xla_cpu_enable_fast_math(bool value) {
|
|
|
|
_impl_.xla_cpu_enable_fast_math_ = value;
|
|
}
|
|
inline void DebugOptions::set_xla_cpu_enable_fast_math(bool value) {
|
|
_internal_set_xla_cpu_enable_fast_math(value);
|
|
// @@protoc_insertion_point(field_set:xla.DebugOptions.xla_cpu_enable_fast_math)
|
|
}
|
|
|
|
// bool xla_cpu_fast_math_honor_nans = 120;
|
|
inline void DebugOptions::clear_xla_cpu_fast_math_honor_nans() {
|
|
_impl_.xla_cpu_fast_math_honor_nans_ = false;
|
|
}
|
|
inline bool DebugOptions::_internal_xla_cpu_fast_math_honor_nans() const {
|
|
return _impl_.xla_cpu_fast_math_honor_nans_;
|
|
}
|
|
inline bool DebugOptions::xla_cpu_fast_math_honor_nans() const {
|
|
// @@protoc_insertion_point(field_get:xla.DebugOptions.xla_cpu_fast_math_honor_nans)
|
|
return _internal_xla_cpu_fast_math_honor_nans();
|
|
}
|
|
inline void DebugOptions::_internal_set_xla_cpu_fast_math_honor_nans(bool value) {
|
|
|
|
_impl_.xla_cpu_fast_math_honor_nans_ = value;
|
|
}
|
|
inline void DebugOptions::set_xla_cpu_fast_math_honor_nans(bool value) {
|
|
_internal_set_xla_cpu_fast_math_honor_nans(value);
|
|
// @@protoc_insertion_point(field_set:xla.DebugOptions.xla_cpu_fast_math_honor_nans)
|
|
}
|
|
|
|
// bool xla_cpu_fast_math_honor_infs = 121;
|
|
inline void DebugOptions::clear_xla_cpu_fast_math_honor_infs() {
|
|
_impl_.xla_cpu_fast_math_honor_infs_ = false;
|
|
}
|
|
inline bool DebugOptions::_internal_xla_cpu_fast_math_honor_infs() const {
|
|
return _impl_.xla_cpu_fast_math_honor_infs_;
|
|
}
|
|
inline bool DebugOptions::xla_cpu_fast_math_honor_infs() const {
|
|
// @@protoc_insertion_point(field_get:xla.DebugOptions.xla_cpu_fast_math_honor_infs)
|
|
return _internal_xla_cpu_fast_math_honor_infs();
|
|
}
|
|
inline void DebugOptions::_internal_set_xla_cpu_fast_math_honor_infs(bool value) {
|
|
|
|
_impl_.xla_cpu_fast_math_honor_infs_ = value;
|
|
}
|
|
inline void DebugOptions::set_xla_cpu_fast_math_honor_infs(bool value) {
|
|
_internal_set_xla_cpu_fast_math_honor_infs(value);
|
|
// @@protoc_insertion_point(field_set:xla.DebugOptions.xla_cpu_fast_math_honor_infs)
|
|
}
|
|
|
|
// bool xla_cpu_fast_math_honor_division = 126;
|
|
inline void DebugOptions::clear_xla_cpu_fast_math_honor_division() {
|
|
_impl_.xla_cpu_fast_math_honor_division_ = false;
|
|
}
|
|
inline bool DebugOptions::_internal_xla_cpu_fast_math_honor_division() const {
|
|
return _impl_.xla_cpu_fast_math_honor_division_;
|
|
}
|
|
inline bool DebugOptions::xla_cpu_fast_math_honor_division() const {
|
|
// @@protoc_insertion_point(field_get:xla.DebugOptions.xla_cpu_fast_math_honor_division)
|
|
return _internal_xla_cpu_fast_math_honor_division();
|
|
}
|
|
inline void DebugOptions::_internal_set_xla_cpu_fast_math_honor_division(bool value) {
|
|
|
|
_impl_.xla_cpu_fast_math_honor_division_ = value;
|
|
}
|
|
inline void DebugOptions::set_xla_cpu_fast_math_honor_division(bool value) {
|
|
_internal_set_xla_cpu_fast_math_honor_division(value);
|
|
// @@protoc_insertion_point(field_set:xla.DebugOptions.xla_cpu_fast_math_honor_division)
|
|
}
|
|
|
|
// bool xla_cpu_fast_math_honor_functions = 129;
|
|
inline void DebugOptions::clear_xla_cpu_fast_math_honor_functions() {
|
|
_impl_.xla_cpu_fast_math_honor_functions_ = false;
|
|
}
|
|
inline bool DebugOptions::_internal_xla_cpu_fast_math_honor_functions() const {
|
|
return _impl_.xla_cpu_fast_math_honor_functions_;
|
|
}
|
|
inline bool DebugOptions::xla_cpu_fast_math_honor_functions() const {
|
|
// @@protoc_insertion_point(field_get:xla.DebugOptions.xla_cpu_fast_math_honor_functions)
|
|
return _internal_xla_cpu_fast_math_honor_functions();
|
|
}
|
|
inline void DebugOptions::_internal_set_xla_cpu_fast_math_honor_functions(bool value) {
|
|
|
|
_impl_.xla_cpu_fast_math_honor_functions_ = value;
|
|
}
|
|
inline void DebugOptions::set_xla_cpu_fast_math_honor_functions(bool value) {
|
|
_internal_set_xla_cpu_fast_math_honor_functions(value);
|
|
// @@protoc_insertion_point(field_set:xla.DebugOptions.xla_cpu_fast_math_honor_functions)
|
|
}
|
|
|
|
// bool xla_cpu_enable_fast_min_max = 140;
|
|
inline void DebugOptions::clear_xla_cpu_enable_fast_min_max() {
|
|
_impl_.xla_cpu_enable_fast_min_max_ = false;
|
|
}
|
|
inline bool DebugOptions::_internal_xla_cpu_enable_fast_min_max() const {
|
|
return _impl_.xla_cpu_enable_fast_min_max_;
|
|
}
|
|
inline bool DebugOptions::xla_cpu_enable_fast_min_max() const {
|
|
// @@protoc_insertion_point(field_get:xla.DebugOptions.xla_cpu_enable_fast_min_max)
|
|
return _internal_xla_cpu_enable_fast_min_max();
|
|
}
|
|
inline void DebugOptions::_internal_set_xla_cpu_enable_fast_min_max(bool value) {
|
|
|
|
_impl_.xla_cpu_enable_fast_min_max_ = value;
|
|
}
|
|
inline void DebugOptions::set_xla_cpu_enable_fast_min_max(bool value) {
|
|
_internal_set_xla_cpu_enable_fast_min_max(value);
|
|
// @@protoc_insertion_point(field_set:xla.DebugOptions.xla_cpu_enable_fast_min_max)
|
|
}
|
|
|
|
// bool xla_gpu_enable_fast_min_max = 100;
|
|
inline void DebugOptions::clear_xla_gpu_enable_fast_min_max() {
|
|
_impl_.xla_gpu_enable_fast_min_max_ = false;
|
|
}
|
|
inline bool DebugOptions::_internal_xla_gpu_enable_fast_min_max() const {
|
|
return _impl_.xla_gpu_enable_fast_min_max_;
|
|
}
|
|
inline bool DebugOptions::xla_gpu_enable_fast_min_max() const {
|
|
// @@protoc_insertion_point(field_get:xla.DebugOptions.xla_gpu_enable_fast_min_max)
|
|
return _internal_xla_gpu_enable_fast_min_max();
|
|
}
|
|
inline void DebugOptions::_internal_set_xla_gpu_enable_fast_min_max(bool value) {
|
|
|
|
_impl_.xla_gpu_enable_fast_min_max_ = value;
|
|
}
|
|
inline void DebugOptions::set_xla_gpu_enable_fast_min_max(bool value) {
|
|
_internal_set_xla_gpu_enable_fast_min_max(value);
|
|
// @@protoc_insertion_point(field_set:xla.DebugOptions.xla_gpu_enable_fast_min_max)
|
|
}
|
|
|
|
// int32 xla_cpu_sparse_cuda_threads = 207;
|
|
inline void DebugOptions::clear_xla_cpu_sparse_cuda_threads() {
|
|
_impl_.xla_cpu_sparse_cuda_threads_ = 0;
|
|
}
|
|
inline int32_t DebugOptions::_internal_xla_cpu_sparse_cuda_threads() const {
|
|
return _impl_.xla_cpu_sparse_cuda_threads_;
|
|
}
|
|
inline int32_t DebugOptions::xla_cpu_sparse_cuda_threads() const {
|
|
// @@protoc_insertion_point(field_get:xla.DebugOptions.xla_cpu_sparse_cuda_threads)
|
|
return _internal_xla_cpu_sparse_cuda_threads();
|
|
}
|
|
inline void DebugOptions::_internal_set_xla_cpu_sparse_cuda_threads(int32_t value) {
|
|
|
|
_impl_.xla_cpu_sparse_cuda_threads_ = value;
|
|
}
|
|
inline void DebugOptions::set_xla_cpu_sparse_cuda_threads(int32_t value) {
|
|
_internal_set_xla_cpu_sparse_cuda_threads(value);
|
|
// @@protoc_insertion_point(field_set:xla.DebugOptions.xla_cpu_sparse_cuda_threads)
|
|
}
|
|
|
|
// bool xla_allow_excess_precision = 122;
|
|
inline void DebugOptions::clear_xla_allow_excess_precision() {
|
|
_impl_.xla_allow_excess_precision_ = false;
|
|
}
|
|
inline bool DebugOptions::_internal_xla_allow_excess_precision() const {
|
|
return _impl_.xla_allow_excess_precision_;
|
|
}
|
|
inline bool DebugOptions::xla_allow_excess_precision() const {
|
|
// @@protoc_insertion_point(field_get:xla.DebugOptions.xla_allow_excess_precision)
|
|
return _internal_xla_allow_excess_precision();
|
|
}
|
|
inline void DebugOptions::_internal_set_xla_allow_excess_precision(bool value) {
|
|
|
|
_impl_.xla_allow_excess_precision_ = value;
|
|
}
|
|
inline void DebugOptions::set_xla_allow_excess_precision(bool value) {
|
|
_internal_set_xla_allow_excess_precision(value);
|
|
// @@protoc_insertion_point(field_set:xla.DebugOptions.xla_allow_excess_precision)
|
|
}
|
|
|
|
// bool xla_gpu_crash_on_verification_failures = 101;
|
|
inline void DebugOptions::clear_xla_gpu_crash_on_verification_failures() {
|
|
_impl_.xla_gpu_crash_on_verification_failures_ = false;
|
|
}
|
|
inline bool DebugOptions::_internal_xla_gpu_crash_on_verification_failures() const {
|
|
return _impl_.xla_gpu_crash_on_verification_failures_;
|
|
}
|
|
inline bool DebugOptions::xla_gpu_crash_on_verification_failures() const {
|
|
// @@protoc_insertion_point(field_get:xla.DebugOptions.xla_gpu_crash_on_verification_failures)
|
|
return _internal_xla_gpu_crash_on_verification_failures();
|
|
}
|
|
inline void DebugOptions::_internal_set_xla_gpu_crash_on_verification_failures(bool value) {
|
|
|
|
_impl_.xla_gpu_crash_on_verification_failures_ = value;
|
|
}
|
|
inline void DebugOptions::set_xla_gpu_crash_on_verification_failures(bool value) {
|
|
_internal_set_xla_gpu_crash_on_verification_failures(value);
|
|
// @@protoc_insertion_point(field_set:xla.DebugOptions.xla_gpu_crash_on_verification_failures)
|
|
}
|
|
|
|
// int32 xla_gpu_autotune_level = 123;
|
|
inline void DebugOptions::clear_xla_gpu_autotune_level() {
|
|
_impl_.xla_gpu_autotune_level_ = 0;
|
|
}
|
|
inline int32_t DebugOptions::_internal_xla_gpu_autotune_level() const {
|
|
return _impl_.xla_gpu_autotune_level_;
|
|
}
|
|
inline int32_t DebugOptions::xla_gpu_autotune_level() const {
|
|
// @@protoc_insertion_point(field_get:xla.DebugOptions.xla_gpu_autotune_level)
|
|
return _internal_xla_gpu_autotune_level();
|
|
}
|
|
inline void DebugOptions::_internal_set_xla_gpu_autotune_level(int32_t value) {
|
|
|
|
_impl_.xla_gpu_autotune_level_ = value;
|
|
}
|
|
inline void DebugOptions::set_xla_gpu_autotune_level(int32_t value) {
|
|
_internal_set_xla_gpu_autotune_level(value);
|
|
// @@protoc_insertion_point(field_set:xla.DebugOptions.xla_gpu_autotune_level)
|
|
}
|
|
|
|
// int32 xla_force_host_platform_device_count = 102;
|
|
inline void DebugOptions::clear_xla_force_host_platform_device_count() {
|
|
_impl_.xla_force_host_platform_device_count_ = 0;
|
|
}
|
|
inline int32_t DebugOptions::_internal_xla_force_host_platform_device_count() const {
|
|
return _impl_.xla_force_host_platform_device_count_;
|
|
}
|
|
inline int32_t DebugOptions::xla_force_host_platform_device_count() const {
|
|
// @@protoc_insertion_point(field_get:xla.DebugOptions.xla_force_host_platform_device_count)
|
|
return _internal_xla_force_host_platform_device_count();
|
|
}
|
|
inline void DebugOptions::_internal_set_xla_force_host_platform_device_count(int32_t value) {
|
|
|
|
_impl_.xla_force_host_platform_device_count_ = value;
|
|
}
|
|
inline void DebugOptions::set_xla_force_host_platform_device_count(int32_t value) {
|
|
_internal_set_xla_force_host_platform_device_count(value);
|
|
// @@protoc_insertion_point(field_set:xla.DebugOptions.xla_force_host_platform_device_count)
|
|
}
|
|
|
|
// bool xla_gpu_disable_gpuasm_optimizations = 103;
|
|
inline void DebugOptions::clear_xla_gpu_disable_gpuasm_optimizations() {
|
|
_impl_.xla_gpu_disable_gpuasm_optimizations_ = false;
|
|
}
|
|
inline bool DebugOptions::_internal_xla_gpu_disable_gpuasm_optimizations() const {
|
|
return _impl_.xla_gpu_disable_gpuasm_optimizations_;
|
|
}
|
|
inline bool DebugOptions::xla_gpu_disable_gpuasm_optimizations() const {
|
|
// @@protoc_insertion_point(field_get:xla.DebugOptions.xla_gpu_disable_gpuasm_optimizations)
|
|
return _internal_xla_gpu_disable_gpuasm_optimizations();
|
|
}
|
|
inline void DebugOptions::_internal_set_xla_gpu_disable_gpuasm_optimizations(bool value) {
|
|
|
|
_impl_.xla_gpu_disable_gpuasm_optimizations_ = value;
|
|
}
|
|
inline void DebugOptions::set_xla_gpu_disable_gpuasm_optimizations(bool value) {
|
|
_internal_set_xla_gpu_disable_gpuasm_optimizations(value);
|
|
// @@protoc_insertion_point(field_set:xla.DebugOptions.xla_gpu_disable_gpuasm_optimizations)
|
|
}
|
|
|
|
// .xla.DebugOptions.ShapeChecks xla_gpu_shape_checks = 170;
|
|
inline void DebugOptions::clear_xla_gpu_shape_checks() {
|
|
_impl_.xla_gpu_shape_checks_ = 0;
|
|
}
|
|
inline ::xla::DebugOptions_ShapeChecks DebugOptions::_internal_xla_gpu_shape_checks() const {
|
|
return static_cast< ::xla::DebugOptions_ShapeChecks >(_impl_.xla_gpu_shape_checks_);
|
|
}
|
|
inline ::xla::DebugOptions_ShapeChecks DebugOptions::xla_gpu_shape_checks() const {
|
|
// @@protoc_insertion_point(field_get:xla.DebugOptions.xla_gpu_shape_checks)
|
|
return _internal_xla_gpu_shape_checks();
|
|
}
|
|
inline void DebugOptions::_internal_set_xla_gpu_shape_checks(::xla::DebugOptions_ShapeChecks value) {
|
|
|
|
_impl_.xla_gpu_shape_checks_ = value;
|
|
}
|
|
inline void DebugOptions::set_xla_gpu_shape_checks(::xla::DebugOptions_ShapeChecks value) {
|
|
_internal_set_xla_gpu_shape_checks(value);
|
|
// @@protoc_insertion_point(field_set:xla.DebugOptions.xla_gpu_shape_checks)
|
|
}
|
|
|
|
// bool xla_hlo_evaluator_use_fast_path = 106;
|
|
inline void DebugOptions::clear_xla_hlo_evaluator_use_fast_path() {
|
|
_impl_.xla_hlo_evaluator_use_fast_path_ = false;
|
|
}
|
|
inline bool DebugOptions::_internal_xla_hlo_evaluator_use_fast_path() const {
|
|
return _impl_.xla_hlo_evaluator_use_fast_path_;
|
|
}
|
|
inline bool DebugOptions::xla_hlo_evaluator_use_fast_path() const {
|
|
// @@protoc_insertion_point(field_get:xla.DebugOptions.xla_hlo_evaluator_use_fast_path)
|
|
return _internal_xla_hlo_evaluator_use_fast_path();
|
|
}
|
|
inline void DebugOptions::_internal_set_xla_hlo_evaluator_use_fast_path(bool value) {
|
|
|
|
_impl_.xla_hlo_evaluator_use_fast_path_ = value;
|
|
}
|
|
inline void DebugOptions::set_xla_hlo_evaluator_use_fast_path(bool value) {
|
|
_internal_set_xla_hlo_evaluator_use_fast_path(value);
|
|
// @@protoc_insertion_point(field_set:xla.DebugOptions.xla_hlo_evaluator_use_fast_path)
|
|
}
|
|
|
|
// bool xla_allow_scalar_index_dynamic_ops = 107;
|
|
inline void DebugOptions::clear_xla_allow_scalar_index_dynamic_ops() {
|
|
_impl_.xla_allow_scalar_index_dynamic_ops_ = false;
|
|
}
|
|
inline bool DebugOptions::_internal_xla_allow_scalar_index_dynamic_ops() const {
|
|
return _impl_.xla_allow_scalar_index_dynamic_ops_;
|
|
}
|
|
inline bool DebugOptions::xla_allow_scalar_index_dynamic_ops() const {
|
|
// @@protoc_insertion_point(field_get:xla.DebugOptions.xla_allow_scalar_index_dynamic_ops)
|
|
return _internal_xla_allow_scalar_index_dynamic_ops();
|
|
}
|
|
inline void DebugOptions::_internal_set_xla_allow_scalar_index_dynamic_ops(bool value) {
|
|
|
|
_impl_.xla_allow_scalar_index_dynamic_ops_ = value;
|
|
}
|
|
inline void DebugOptions::set_xla_allow_scalar_index_dynamic_ops(bool value) {
|
|
_internal_set_xla_allow_scalar_index_dynamic_ops(value);
|
|
// @@protoc_insertion_point(field_set:xla.DebugOptions.xla_allow_scalar_index_dynamic_ops)
|
|
}
|
|
|
|
// .xla.DebugOptions.StepMarkerLocation xla_step_marker_location = 108;
|
|
inline void DebugOptions::clear_xla_step_marker_location() {
|
|
_impl_.xla_step_marker_location_ = 0;
|
|
}
|
|
inline ::xla::DebugOptions_StepMarkerLocation DebugOptions::_internal_xla_step_marker_location() const {
|
|
return static_cast< ::xla::DebugOptions_StepMarkerLocation >(_impl_.xla_step_marker_location_);
|
|
}
|
|
inline ::xla::DebugOptions_StepMarkerLocation DebugOptions::xla_step_marker_location() const {
|
|
// @@protoc_insertion_point(field_get:xla.DebugOptions.xla_step_marker_location)
|
|
return _internal_xla_step_marker_location();
|
|
}
|
|
inline void DebugOptions::_internal_set_xla_step_marker_location(::xla::DebugOptions_StepMarkerLocation value) {
|
|
|
|
_impl_.xla_step_marker_location_ = value;
|
|
}
|
|
inline void DebugOptions::set_xla_step_marker_location(::xla::DebugOptions_StepMarkerLocation value) {
|
|
_internal_set_xla_step_marker_location(value);
|
|
// @@protoc_insertion_point(field_set:xla.DebugOptions.xla_step_marker_location)
|
|
}
|
|
|
|
// string xla_dump_to = 109;
|
|
inline void DebugOptions::clear_xla_dump_to() {
|
|
_impl_.xla_dump_to_.ClearToEmpty();
|
|
}
|
|
inline const std::string& DebugOptions::xla_dump_to() const {
|
|
// @@protoc_insertion_point(field_get:xla.DebugOptions.xla_dump_to)
|
|
return _internal_xla_dump_to();
|
|
}
|
|
template <typename ArgT0, typename... ArgT>
|
|
inline PROTOBUF_ALWAYS_INLINE
|
|
void DebugOptions::set_xla_dump_to(ArgT0&& arg0, ArgT... args) {
|
|
|
|
_impl_.xla_dump_to_.Set(static_cast<ArgT0 &&>(arg0), args..., GetArenaForAllocation());
|
|
// @@protoc_insertion_point(field_set:xla.DebugOptions.xla_dump_to)
|
|
}
|
|
inline std::string* DebugOptions::mutable_xla_dump_to() {
|
|
std::string* _s = _internal_mutable_xla_dump_to();
|
|
// @@protoc_insertion_point(field_mutable:xla.DebugOptions.xla_dump_to)
|
|
return _s;
|
|
}
|
|
inline const std::string& DebugOptions::_internal_xla_dump_to() const {
|
|
return _impl_.xla_dump_to_.Get();
|
|
}
|
|
inline void DebugOptions::_internal_set_xla_dump_to(const std::string& value) {
|
|
|
|
_impl_.xla_dump_to_.Set(value, GetArenaForAllocation());
|
|
}
|
|
inline std::string* DebugOptions::_internal_mutable_xla_dump_to() {
|
|
|
|
return _impl_.xla_dump_to_.Mutable(GetArenaForAllocation());
|
|
}
|
|
inline std::string* DebugOptions::release_xla_dump_to() {
|
|
// @@protoc_insertion_point(field_release:xla.DebugOptions.xla_dump_to)
|
|
return _impl_.xla_dump_to_.Release();
|
|
}
|
|
inline void DebugOptions::set_allocated_xla_dump_to(std::string* xla_dump_to) {
|
|
if (xla_dump_to != nullptr) {
|
|
|
|
} else {
|
|
|
|
}
|
|
_impl_.xla_dump_to_.SetAllocated(xla_dump_to, GetArenaForAllocation());
|
|
#ifdef PROTOBUF_FORCE_COPY_DEFAULT_STRING
|
|
if (_impl_.xla_dump_to_.IsDefault()) {
|
|
_impl_.xla_dump_to_.Set("", GetArenaForAllocation());
|
|
}
|
|
#endif // PROTOBUF_FORCE_COPY_DEFAULT_STRING
|
|
// @@protoc_insertion_point(field_set_allocated:xla.DebugOptions.xla_dump_to)
|
|
}
|
|
|
|
// string xla_dump_hlo_module_re = 110;
|
|
inline void DebugOptions::clear_xla_dump_hlo_module_re() {
|
|
_impl_.xla_dump_hlo_module_re_.ClearToEmpty();
|
|
}
|
|
inline const std::string& DebugOptions::xla_dump_hlo_module_re() const {
|
|
// @@protoc_insertion_point(field_get:xla.DebugOptions.xla_dump_hlo_module_re)
|
|
return _internal_xla_dump_hlo_module_re();
|
|
}
|
|
template <typename ArgT0, typename... ArgT>
|
|
inline PROTOBUF_ALWAYS_INLINE
|
|
void DebugOptions::set_xla_dump_hlo_module_re(ArgT0&& arg0, ArgT... args) {
|
|
|
|
_impl_.xla_dump_hlo_module_re_.Set(static_cast<ArgT0 &&>(arg0), args..., GetArenaForAllocation());
|
|
// @@protoc_insertion_point(field_set:xla.DebugOptions.xla_dump_hlo_module_re)
|
|
}
|
|
inline std::string* DebugOptions::mutable_xla_dump_hlo_module_re() {
|
|
std::string* _s = _internal_mutable_xla_dump_hlo_module_re();
|
|
// @@protoc_insertion_point(field_mutable:xla.DebugOptions.xla_dump_hlo_module_re)
|
|
return _s;
|
|
}
|
|
inline const std::string& DebugOptions::_internal_xla_dump_hlo_module_re() const {
|
|
return _impl_.xla_dump_hlo_module_re_.Get();
|
|
}
|
|
inline void DebugOptions::_internal_set_xla_dump_hlo_module_re(const std::string& value) {
|
|
|
|
_impl_.xla_dump_hlo_module_re_.Set(value, GetArenaForAllocation());
|
|
}
|
|
inline std::string* DebugOptions::_internal_mutable_xla_dump_hlo_module_re() {
|
|
|
|
return _impl_.xla_dump_hlo_module_re_.Mutable(GetArenaForAllocation());
|
|
}
|
|
inline std::string* DebugOptions::release_xla_dump_hlo_module_re() {
|
|
// @@protoc_insertion_point(field_release:xla.DebugOptions.xla_dump_hlo_module_re)
|
|
return _impl_.xla_dump_hlo_module_re_.Release();
|
|
}
|
|
inline void DebugOptions::set_allocated_xla_dump_hlo_module_re(std::string* xla_dump_hlo_module_re) {
|
|
if (xla_dump_hlo_module_re != nullptr) {
|
|
|
|
} else {
|
|
|
|
}
|
|
_impl_.xla_dump_hlo_module_re_.SetAllocated(xla_dump_hlo_module_re, GetArenaForAllocation());
|
|
#ifdef PROTOBUF_FORCE_COPY_DEFAULT_STRING
|
|
if (_impl_.xla_dump_hlo_module_re_.IsDefault()) {
|
|
_impl_.xla_dump_hlo_module_re_.Set("", GetArenaForAllocation());
|
|
}
|
|
#endif // PROTOBUF_FORCE_COPY_DEFAULT_STRING
|
|
// @@protoc_insertion_point(field_set_allocated:xla.DebugOptions.xla_dump_hlo_module_re)
|
|
}
|
|
|
|
// string xla_dump_hlo_pass_re = 111;
|
|
inline void DebugOptions::clear_xla_dump_hlo_pass_re() {
|
|
_impl_.xla_dump_hlo_pass_re_.ClearToEmpty();
|
|
}
|
|
inline const std::string& DebugOptions::xla_dump_hlo_pass_re() const {
|
|
// @@protoc_insertion_point(field_get:xla.DebugOptions.xla_dump_hlo_pass_re)
|
|
return _internal_xla_dump_hlo_pass_re();
|
|
}
|
|
template <typename ArgT0, typename... ArgT>
|
|
inline PROTOBUF_ALWAYS_INLINE
|
|
void DebugOptions::set_xla_dump_hlo_pass_re(ArgT0&& arg0, ArgT... args) {
|
|
|
|
_impl_.xla_dump_hlo_pass_re_.Set(static_cast<ArgT0 &&>(arg0), args..., GetArenaForAllocation());
|
|
// @@protoc_insertion_point(field_set:xla.DebugOptions.xla_dump_hlo_pass_re)
|
|
}
|
|
inline std::string* DebugOptions::mutable_xla_dump_hlo_pass_re() {
|
|
std::string* _s = _internal_mutable_xla_dump_hlo_pass_re();
|
|
// @@protoc_insertion_point(field_mutable:xla.DebugOptions.xla_dump_hlo_pass_re)
|
|
return _s;
|
|
}
|
|
inline const std::string& DebugOptions::_internal_xla_dump_hlo_pass_re() const {
|
|
return _impl_.xla_dump_hlo_pass_re_.Get();
|
|
}
|
|
inline void DebugOptions::_internal_set_xla_dump_hlo_pass_re(const std::string& value) {
|
|
|
|
_impl_.xla_dump_hlo_pass_re_.Set(value, GetArenaForAllocation());
|
|
}
|
|
inline std::string* DebugOptions::_internal_mutable_xla_dump_hlo_pass_re() {
|
|
|
|
return _impl_.xla_dump_hlo_pass_re_.Mutable(GetArenaForAllocation());
|
|
}
|
|
inline std::string* DebugOptions::release_xla_dump_hlo_pass_re() {
|
|
// @@protoc_insertion_point(field_release:xla.DebugOptions.xla_dump_hlo_pass_re)
|
|
return _impl_.xla_dump_hlo_pass_re_.Release();
|
|
}
|
|
inline void DebugOptions::set_allocated_xla_dump_hlo_pass_re(std::string* xla_dump_hlo_pass_re) {
|
|
if (xla_dump_hlo_pass_re != nullptr) {
|
|
|
|
} else {
|
|
|
|
}
|
|
_impl_.xla_dump_hlo_pass_re_.SetAllocated(xla_dump_hlo_pass_re, GetArenaForAllocation());
|
|
#ifdef PROTOBUF_FORCE_COPY_DEFAULT_STRING
|
|
if (_impl_.xla_dump_hlo_pass_re_.IsDefault()) {
|
|
_impl_.xla_dump_hlo_pass_re_.Set("", GetArenaForAllocation());
|
|
}
|
|
#endif // PROTOBUF_FORCE_COPY_DEFAULT_STRING
|
|
// @@protoc_insertion_point(field_set_allocated:xla.DebugOptions.xla_dump_hlo_pass_re)
|
|
}
|
|
|
|
// bool xla_dump_hlo_as_text = 112;
|
|
inline void DebugOptions::clear_xla_dump_hlo_as_text() {
|
|
_impl_.xla_dump_hlo_as_text_ = false;
|
|
}
|
|
inline bool DebugOptions::_internal_xla_dump_hlo_as_text() const {
|
|
return _impl_.xla_dump_hlo_as_text_;
|
|
}
|
|
inline bool DebugOptions::xla_dump_hlo_as_text() const {
|
|
// @@protoc_insertion_point(field_get:xla.DebugOptions.xla_dump_hlo_as_text)
|
|
return _internal_xla_dump_hlo_as_text();
|
|
}
|
|
inline void DebugOptions::_internal_set_xla_dump_hlo_as_text(bool value) {
|
|
|
|
_impl_.xla_dump_hlo_as_text_ = value;
|
|
}
|
|
inline void DebugOptions::set_xla_dump_hlo_as_text(bool value) {
|
|
_internal_set_xla_dump_hlo_as_text(value);
|
|
// @@protoc_insertion_point(field_set:xla.DebugOptions.xla_dump_hlo_as_text)
|
|
}
|
|
|
|
// bool xla_dump_hlo_as_proto = 113;
|
|
inline void DebugOptions::clear_xla_dump_hlo_as_proto() {
|
|
_impl_.xla_dump_hlo_as_proto_ = false;
|
|
}
|
|
inline bool DebugOptions::_internal_xla_dump_hlo_as_proto() const {
|
|
return _impl_.xla_dump_hlo_as_proto_;
|
|
}
|
|
inline bool DebugOptions::xla_dump_hlo_as_proto() const {
|
|
// @@protoc_insertion_point(field_get:xla.DebugOptions.xla_dump_hlo_as_proto)
|
|
return _internal_xla_dump_hlo_as_proto();
|
|
}
|
|
inline void DebugOptions::_internal_set_xla_dump_hlo_as_proto(bool value) {
|
|
|
|
_impl_.xla_dump_hlo_as_proto_ = value;
|
|
}
|
|
inline void DebugOptions::set_xla_dump_hlo_as_proto(bool value) {
|
|
_internal_set_xla_dump_hlo_as_proto(value);
|
|
// @@protoc_insertion_point(field_set:xla.DebugOptions.xla_dump_hlo_as_proto)
|
|
}
|
|
|
|
// bool xla_dump_hlo_as_dot = 114;
|
|
inline void DebugOptions::clear_xla_dump_hlo_as_dot() {
|
|
_impl_.xla_dump_hlo_as_dot_ = false;
|
|
}
|
|
inline bool DebugOptions::_internal_xla_dump_hlo_as_dot() const {
|
|
return _impl_.xla_dump_hlo_as_dot_;
|
|
}
|
|
inline bool DebugOptions::xla_dump_hlo_as_dot() const {
|
|
// @@protoc_insertion_point(field_get:xla.DebugOptions.xla_dump_hlo_as_dot)
|
|
return _internal_xla_dump_hlo_as_dot();
|
|
}
|
|
inline void DebugOptions::_internal_set_xla_dump_hlo_as_dot(bool value) {
|
|
|
|
_impl_.xla_dump_hlo_as_dot_ = value;
|
|
}
|
|
inline void DebugOptions::set_xla_dump_hlo_as_dot(bool value) {
|
|
_internal_set_xla_dump_hlo_as_dot(value);
|
|
// @@protoc_insertion_point(field_set:xla.DebugOptions.xla_dump_hlo_as_dot)
|
|
}
|
|
|
|
// bool xla_dump_hlo_as_url = 115;
|
|
inline void DebugOptions::clear_xla_dump_hlo_as_url() {
|
|
_impl_.xla_dump_hlo_as_url_ = false;
|
|
}
|
|
inline bool DebugOptions::_internal_xla_dump_hlo_as_url() const {
|
|
return _impl_.xla_dump_hlo_as_url_;
|
|
}
|
|
inline bool DebugOptions::xla_dump_hlo_as_url() const {
|
|
// @@protoc_insertion_point(field_get:xla.DebugOptions.xla_dump_hlo_as_url)
|
|
return _internal_xla_dump_hlo_as_url();
|
|
}
|
|
inline void DebugOptions::_internal_set_xla_dump_hlo_as_url(bool value) {
|
|
|
|
_impl_.xla_dump_hlo_as_url_ = value;
|
|
}
|
|
inline void DebugOptions::set_xla_dump_hlo_as_url(bool value) {
|
|
_internal_set_xla_dump_hlo_as_url(value);
|
|
// @@protoc_insertion_point(field_set:xla.DebugOptions.xla_dump_hlo_as_url)
|
|
}
|
|
|
|
// bool xla_dump_hlo_as_html = 116;
|
|
inline void DebugOptions::clear_xla_dump_hlo_as_html() {
|
|
_impl_.xla_dump_hlo_as_html_ = false;
|
|
}
|
|
inline bool DebugOptions::_internal_xla_dump_hlo_as_html() const {
|
|
return _impl_.xla_dump_hlo_as_html_;
|
|
}
|
|
inline bool DebugOptions::xla_dump_hlo_as_html() const {
|
|
// @@protoc_insertion_point(field_get:xla.DebugOptions.xla_dump_hlo_as_html)
|
|
return _internal_xla_dump_hlo_as_html();
|
|
}
|
|
inline void DebugOptions::_internal_set_xla_dump_hlo_as_html(bool value) {
|
|
|
|
_impl_.xla_dump_hlo_as_html_ = value;
|
|
}
|
|
inline void DebugOptions::set_xla_dump_hlo_as_html(bool value) {
|
|
_internal_set_xla_dump_hlo_as_html(value);
|
|
// @@protoc_insertion_point(field_set:xla.DebugOptions.xla_dump_hlo_as_html)
|
|
}
|
|
|
|
// bool xla_dump_fusion_visualization = 149;
|
|
inline void DebugOptions::clear_xla_dump_fusion_visualization() {
|
|
_impl_.xla_dump_fusion_visualization_ = false;
|
|
}
|
|
inline bool DebugOptions::_internal_xla_dump_fusion_visualization() const {
|
|
return _impl_.xla_dump_fusion_visualization_;
|
|
}
|
|
inline bool DebugOptions::xla_dump_fusion_visualization() const {
|
|
// @@protoc_insertion_point(field_get:xla.DebugOptions.xla_dump_fusion_visualization)
|
|
return _internal_xla_dump_fusion_visualization();
|
|
}
|
|
inline void DebugOptions::_internal_set_xla_dump_fusion_visualization(bool value) {
|
|
|
|
_impl_.xla_dump_fusion_visualization_ = value;
|
|
}
|
|
inline void DebugOptions::set_xla_dump_fusion_visualization(bool value) {
|
|
_internal_set_xla_dump_fusion_visualization(value);
|
|
// @@protoc_insertion_point(field_set:xla.DebugOptions.xla_dump_fusion_visualization)
|
|
}
|
|
|
|
// bool xla_dump_hlo_snapshots = 118;
|
|
inline void DebugOptions::clear_xla_dump_hlo_snapshots() {
|
|
_impl_.xla_dump_hlo_snapshots_ = false;
|
|
}
|
|
inline bool DebugOptions::_internal_xla_dump_hlo_snapshots() const {
|
|
return _impl_.xla_dump_hlo_snapshots_;
|
|
}
|
|
inline bool DebugOptions::xla_dump_hlo_snapshots() const {
|
|
// @@protoc_insertion_point(field_get:xla.DebugOptions.xla_dump_hlo_snapshots)
|
|
return _internal_xla_dump_hlo_snapshots();
|
|
}
|
|
inline void DebugOptions::_internal_set_xla_dump_hlo_snapshots(bool value) {
|
|
|
|
_impl_.xla_dump_hlo_snapshots_ = value;
|
|
}
|
|
inline void DebugOptions::set_xla_dump_hlo_snapshots(bool value) {
|
|
_internal_set_xla_dump_hlo_snapshots(value);
|
|
// @@protoc_insertion_point(field_set:xla.DebugOptions.xla_dump_hlo_snapshots)
|
|
}
|
|
|
|
// bool xla_dump_include_timestamp = 131;
|
|
inline void DebugOptions::clear_xla_dump_include_timestamp() {
|
|
_impl_.xla_dump_include_timestamp_ = false;
|
|
}
|
|
inline bool DebugOptions::_internal_xla_dump_include_timestamp() const {
|
|
return _impl_.xla_dump_include_timestamp_;
|
|
}
|
|
inline bool DebugOptions::xla_dump_include_timestamp() const {
|
|
// @@protoc_insertion_point(field_get:xla.DebugOptions.xla_dump_include_timestamp)
|
|
return _internal_xla_dump_include_timestamp();
|
|
}
|
|
inline void DebugOptions::_internal_set_xla_dump_include_timestamp(bool value) {
|
|
|
|
_impl_.xla_dump_include_timestamp_ = value;
|
|
}
|
|
inline void DebugOptions::set_xla_dump_include_timestamp(bool value) {
|
|
_internal_set_xla_dump_include_timestamp(value);
|
|
// @@protoc_insertion_point(field_set:xla.DebugOptions.xla_dump_include_timestamp)
|
|
}
|
|
|
|
// int32 xla_dump_max_hlo_modules = 132;
|
|
inline void DebugOptions::clear_xla_dump_max_hlo_modules() {
|
|
_impl_.xla_dump_max_hlo_modules_ = 0;
|
|
}
|
|
inline int32_t DebugOptions::_internal_xla_dump_max_hlo_modules() const {
|
|
return _impl_.xla_dump_max_hlo_modules_;
|
|
}
|
|
inline int32_t DebugOptions::xla_dump_max_hlo_modules() const {
|
|
// @@protoc_insertion_point(field_get:xla.DebugOptions.xla_dump_max_hlo_modules)
|
|
return _internal_xla_dump_max_hlo_modules();
|
|
}
|
|
inline void DebugOptions::_internal_set_xla_dump_max_hlo_modules(int32_t value) {
|
|
|
|
_impl_.xla_dump_max_hlo_modules_ = value;
|
|
}
|
|
inline void DebugOptions::set_xla_dump_max_hlo_modules(int32_t value) {
|
|
_internal_set_xla_dump_max_hlo_modules(value);
|
|
// @@protoc_insertion_point(field_set:xla.DebugOptions.xla_dump_max_hlo_modules)
|
|
}
|
|
|
|
// bool xla_dump_module_metadata = 144;
|
|
inline void DebugOptions::clear_xla_dump_module_metadata() {
|
|
_impl_.xla_dump_module_metadata_ = false;
|
|
}
|
|
inline bool DebugOptions::_internal_xla_dump_module_metadata() const {
|
|
return _impl_.xla_dump_module_metadata_;
|
|
}
|
|
inline bool DebugOptions::xla_dump_module_metadata() const {
|
|
// @@protoc_insertion_point(field_get:xla.DebugOptions.xla_dump_module_metadata)
|
|
return _internal_xla_dump_module_metadata();
|
|
}
|
|
inline void DebugOptions::_internal_set_xla_dump_module_metadata(bool value) {
|
|
|
|
_impl_.xla_dump_module_metadata_ = value;
|
|
}
|
|
inline void DebugOptions::set_xla_dump_module_metadata(bool value) {
|
|
_internal_set_xla_dump_module_metadata(value);
|
|
// @@protoc_insertion_point(field_set:xla.DebugOptions.xla_dump_module_metadata)
|
|
}
|
|
|
|
// bool xla_dump_compress_protos = 151;
|
|
inline void DebugOptions::clear_xla_dump_compress_protos() {
|
|
_impl_.xla_dump_compress_protos_ = false;
|
|
}
|
|
inline bool DebugOptions::_internal_xla_dump_compress_protos() const {
|
|
return _impl_.xla_dump_compress_protos_;
|
|
}
|
|
inline bool DebugOptions::xla_dump_compress_protos() const {
|
|
// @@protoc_insertion_point(field_get:xla.DebugOptions.xla_dump_compress_protos)
|
|
return _internal_xla_dump_compress_protos();
|
|
}
|
|
inline void DebugOptions::_internal_set_xla_dump_compress_protos(bool value) {
|
|
|
|
_impl_.xla_dump_compress_protos_ = value;
|
|
}
|
|
inline void DebugOptions::set_xla_dump_compress_protos(bool value) {
|
|
_internal_set_xla_dump_compress_protos(value);
|
|
// @@protoc_insertion_point(field_set:xla.DebugOptions.xla_dump_compress_protos)
|
|
}
|
|
|
|
// bool xla_dump_hlo_as_long_text = 164;
|
|
inline void DebugOptions::clear_xla_dump_hlo_as_long_text() {
|
|
_impl_.xla_dump_hlo_as_long_text_ = false;
|
|
}
|
|
inline bool DebugOptions::_internal_xla_dump_hlo_as_long_text() const {
|
|
return _impl_.xla_dump_hlo_as_long_text_;
|
|
}
|
|
inline bool DebugOptions::xla_dump_hlo_as_long_text() const {
|
|
// @@protoc_insertion_point(field_get:xla.DebugOptions.xla_dump_hlo_as_long_text)
|
|
return _internal_xla_dump_hlo_as_long_text();
|
|
}
|
|
inline void DebugOptions::_internal_set_xla_dump_hlo_as_long_text(bool value) {
|
|
|
|
_impl_.xla_dump_hlo_as_long_text_ = value;
|
|
}
|
|
inline void DebugOptions::set_xla_dump_hlo_as_long_text(bool value) {
|
|
_internal_set_xla_dump_hlo_as_long_text(value);
|
|
// @@protoc_insertion_point(field_set:xla.DebugOptions.xla_dump_hlo_as_long_text)
|
|
}
|
|
|
|
// bool xla_gpu_force_conv_nchw = 125;
|
|
inline void DebugOptions::clear_xla_gpu_force_conv_nchw() {
|
|
_impl_.xla_gpu_force_conv_nchw_ = false;
|
|
}
|
|
inline bool DebugOptions::_internal_xla_gpu_force_conv_nchw() const {
|
|
return _impl_.xla_gpu_force_conv_nchw_;
|
|
}
|
|
inline bool DebugOptions::xla_gpu_force_conv_nchw() const {
|
|
// @@protoc_insertion_point(field_get:xla.DebugOptions.xla_gpu_force_conv_nchw)
|
|
return _internal_xla_gpu_force_conv_nchw();
|
|
}
|
|
inline void DebugOptions::_internal_set_xla_gpu_force_conv_nchw(bool value) {
|
|
|
|
_impl_.xla_gpu_force_conv_nchw_ = value;
|
|
}
|
|
inline void DebugOptions::set_xla_gpu_force_conv_nchw(bool value) {
|
|
_internal_set_xla_gpu_force_conv_nchw(value);
|
|
// @@protoc_insertion_point(field_set:xla.DebugOptions.xla_gpu_force_conv_nchw)
|
|
}
|
|
|
|
// bool xla_gpu_force_conv_nhwc = 146;
|
|
inline void DebugOptions::clear_xla_gpu_force_conv_nhwc() {
|
|
_impl_.xla_gpu_force_conv_nhwc_ = false;
|
|
}
|
|
inline bool DebugOptions::_internal_xla_gpu_force_conv_nhwc() const {
|
|
return _impl_.xla_gpu_force_conv_nhwc_;
|
|
}
|
|
inline bool DebugOptions::xla_gpu_force_conv_nhwc() const {
|
|
// @@protoc_insertion_point(field_get:xla.DebugOptions.xla_gpu_force_conv_nhwc)
|
|
return _internal_xla_gpu_force_conv_nhwc();
|
|
}
|
|
inline void DebugOptions::_internal_set_xla_gpu_force_conv_nhwc(bool value) {
|
|
|
|
_impl_.xla_gpu_force_conv_nhwc_ = value;
|
|
}
|
|
inline void DebugOptions::set_xla_gpu_force_conv_nhwc(bool value) {
|
|
_internal_set_xla_gpu_force_conv_nhwc(value);
|
|
// @@protoc_insertion_point(field_set:xla.DebugOptions.xla_gpu_force_conv_nhwc)
|
|
}
|
|
|
|
// repeated string xla_gpu_ptx_file = 127;
|
|
inline int DebugOptions::_internal_xla_gpu_ptx_file_size() const {
|
|
return _impl_.xla_gpu_ptx_file_.size();
|
|
}
|
|
inline int DebugOptions::xla_gpu_ptx_file_size() const {
|
|
return _internal_xla_gpu_ptx_file_size();
|
|
}
|
|
inline void DebugOptions::clear_xla_gpu_ptx_file() {
|
|
_impl_.xla_gpu_ptx_file_.Clear();
|
|
}
|
|
inline std::string* DebugOptions::add_xla_gpu_ptx_file() {
|
|
std::string* _s = _internal_add_xla_gpu_ptx_file();
|
|
// @@protoc_insertion_point(field_add_mutable:xla.DebugOptions.xla_gpu_ptx_file)
|
|
return _s;
|
|
}
|
|
inline const std::string& DebugOptions::_internal_xla_gpu_ptx_file(int index) const {
|
|
return _impl_.xla_gpu_ptx_file_.Get(index);
|
|
}
|
|
inline const std::string& DebugOptions::xla_gpu_ptx_file(int index) const {
|
|
// @@protoc_insertion_point(field_get:xla.DebugOptions.xla_gpu_ptx_file)
|
|
return _internal_xla_gpu_ptx_file(index);
|
|
}
|
|
inline std::string* DebugOptions::mutable_xla_gpu_ptx_file(int index) {
|
|
// @@protoc_insertion_point(field_mutable:xla.DebugOptions.xla_gpu_ptx_file)
|
|
return _impl_.xla_gpu_ptx_file_.Mutable(index);
|
|
}
|
|
inline void DebugOptions::set_xla_gpu_ptx_file(int index, const std::string& value) {
|
|
_impl_.xla_gpu_ptx_file_.Mutable(index)->assign(value);
|
|
// @@protoc_insertion_point(field_set:xla.DebugOptions.xla_gpu_ptx_file)
|
|
}
|
|
inline void DebugOptions::set_xla_gpu_ptx_file(int index, std::string&& value) {
|
|
_impl_.xla_gpu_ptx_file_.Mutable(index)->assign(std::move(value));
|
|
// @@protoc_insertion_point(field_set:xla.DebugOptions.xla_gpu_ptx_file)
|
|
}
|
|
inline void DebugOptions::set_xla_gpu_ptx_file(int index, const char* value) {
|
|
GOOGLE_DCHECK(value != nullptr);
|
|
_impl_.xla_gpu_ptx_file_.Mutable(index)->assign(value);
|
|
// @@protoc_insertion_point(field_set_char:xla.DebugOptions.xla_gpu_ptx_file)
|
|
}
|
|
inline void DebugOptions::set_xla_gpu_ptx_file(int index, const char* value, size_t size) {
|
|
_impl_.xla_gpu_ptx_file_.Mutable(index)->assign(
|
|
reinterpret_cast<const char*>(value), size);
|
|
// @@protoc_insertion_point(field_set_pointer:xla.DebugOptions.xla_gpu_ptx_file)
|
|
}
|
|
inline std::string* DebugOptions::_internal_add_xla_gpu_ptx_file() {
|
|
return _impl_.xla_gpu_ptx_file_.Add();
|
|
}
|
|
inline void DebugOptions::add_xla_gpu_ptx_file(const std::string& value) {
|
|
_impl_.xla_gpu_ptx_file_.Add()->assign(value);
|
|
// @@protoc_insertion_point(field_add:xla.DebugOptions.xla_gpu_ptx_file)
|
|
}
|
|
inline void DebugOptions::add_xla_gpu_ptx_file(std::string&& value) {
|
|
_impl_.xla_gpu_ptx_file_.Add(std::move(value));
|
|
// @@protoc_insertion_point(field_add:xla.DebugOptions.xla_gpu_ptx_file)
|
|
}
|
|
inline void DebugOptions::add_xla_gpu_ptx_file(const char* value) {
|
|
GOOGLE_DCHECK(value != nullptr);
|
|
_impl_.xla_gpu_ptx_file_.Add()->assign(value);
|
|
// @@protoc_insertion_point(field_add_char:xla.DebugOptions.xla_gpu_ptx_file)
|
|
}
|
|
inline void DebugOptions::add_xla_gpu_ptx_file(const char* value, size_t size) {
|
|
_impl_.xla_gpu_ptx_file_.Add()->assign(reinterpret_cast<const char*>(value), size);
|
|
// @@protoc_insertion_point(field_add_pointer:xla.DebugOptions.xla_gpu_ptx_file)
|
|
}
|
|
inline const ::PROTOBUF_NAMESPACE_ID::RepeatedPtrField<std::string>&
|
|
DebugOptions::xla_gpu_ptx_file() const {
|
|
// @@protoc_insertion_point(field_list:xla.DebugOptions.xla_gpu_ptx_file)
|
|
return _impl_.xla_gpu_ptx_file_;
|
|
}
|
|
inline ::PROTOBUF_NAMESPACE_ID::RepeatedPtrField<std::string>*
|
|
DebugOptions::mutable_xla_gpu_ptx_file() {
|
|
// @@protoc_insertion_point(field_mutable_list:xla.DebugOptions.xla_gpu_ptx_file)
|
|
return &_impl_.xla_gpu_ptx_file_;
|
|
}
|
|
|
|
// bool xla_gpu_dump_llvmir = 155;
|
|
inline void DebugOptions::clear_xla_gpu_dump_llvmir() {
|
|
_impl_.xla_gpu_dump_llvmir_ = false;
|
|
}
|
|
inline bool DebugOptions::_internal_xla_gpu_dump_llvmir() const {
|
|
return _impl_.xla_gpu_dump_llvmir_;
|
|
}
|
|
inline bool DebugOptions::xla_gpu_dump_llvmir() const {
|
|
// @@protoc_insertion_point(field_get:xla.DebugOptions.xla_gpu_dump_llvmir)
|
|
return _internal_xla_gpu_dump_llvmir();
|
|
}
|
|
inline void DebugOptions::_internal_set_xla_gpu_dump_llvmir(bool value) {
|
|
|
|
_impl_.xla_gpu_dump_llvmir_ = value;
|
|
}
|
|
inline void DebugOptions::set_xla_gpu_dump_llvmir(bool value) {
|
|
_internal_set_xla_gpu_dump_llvmir(value);
|
|
// @@protoc_insertion_point(field_set:xla.DebugOptions.xla_gpu_dump_llvmir)
|
|
}
|
|
|
|
// bool xla_dump_enable_mlir_pretty_form = 185;
|
|
inline void DebugOptions::clear_xla_dump_enable_mlir_pretty_form() {
|
|
_impl_.xla_dump_enable_mlir_pretty_form_ = false;
|
|
}
|
|
inline bool DebugOptions::_internal_xla_dump_enable_mlir_pretty_form() const {
|
|
return _impl_.xla_dump_enable_mlir_pretty_form_;
|
|
}
|
|
inline bool DebugOptions::xla_dump_enable_mlir_pretty_form() const {
|
|
// @@protoc_insertion_point(field_get:xla.DebugOptions.xla_dump_enable_mlir_pretty_form)
|
|
return _internal_xla_dump_enable_mlir_pretty_form();
|
|
}
|
|
inline void DebugOptions::_internal_set_xla_dump_enable_mlir_pretty_form(bool value) {
|
|
|
|
_impl_.xla_dump_enable_mlir_pretty_form_ = value;
|
|
}
|
|
inline void DebugOptions::set_xla_dump_enable_mlir_pretty_form(bool value) {
|
|
_internal_set_xla_dump_enable_mlir_pretty_form(value);
|
|
// @@protoc_insertion_point(field_set:xla.DebugOptions.xla_dump_enable_mlir_pretty_form)
|
|
}
|
|
|
|
// string xla_gpu_algorithm_denylist_path = 128;
|
|
inline void DebugOptions::clear_xla_gpu_algorithm_denylist_path() {
|
|
_impl_.xla_gpu_algorithm_denylist_path_.ClearToEmpty();
|
|
}
|
|
inline const std::string& DebugOptions::xla_gpu_algorithm_denylist_path() const {
|
|
// @@protoc_insertion_point(field_get:xla.DebugOptions.xla_gpu_algorithm_denylist_path)
|
|
return _internal_xla_gpu_algorithm_denylist_path();
|
|
}
|
|
template <typename ArgT0, typename... ArgT>
|
|
inline PROTOBUF_ALWAYS_INLINE
|
|
void DebugOptions::set_xla_gpu_algorithm_denylist_path(ArgT0&& arg0, ArgT... args) {
|
|
|
|
_impl_.xla_gpu_algorithm_denylist_path_.Set(static_cast<ArgT0 &&>(arg0), args..., GetArenaForAllocation());
|
|
// @@protoc_insertion_point(field_set:xla.DebugOptions.xla_gpu_algorithm_denylist_path)
|
|
}
|
|
inline std::string* DebugOptions::mutable_xla_gpu_algorithm_denylist_path() {
|
|
std::string* _s = _internal_mutable_xla_gpu_algorithm_denylist_path();
|
|
// @@protoc_insertion_point(field_mutable:xla.DebugOptions.xla_gpu_algorithm_denylist_path)
|
|
return _s;
|
|
}
|
|
inline const std::string& DebugOptions::_internal_xla_gpu_algorithm_denylist_path() const {
|
|
return _impl_.xla_gpu_algorithm_denylist_path_.Get();
|
|
}
|
|
inline void DebugOptions::_internal_set_xla_gpu_algorithm_denylist_path(const std::string& value) {
|
|
|
|
_impl_.xla_gpu_algorithm_denylist_path_.Set(value, GetArenaForAllocation());
|
|
}
|
|
inline std::string* DebugOptions::_internal_mutable_xla_gpu_algorithm_denylist_path() {
|
|
|
|
return _impl_.xla_gpu_algorithm_denylist_path_.Mutable(GetArenaForAllocation());
|
|
}
|
|
inline std::string* DebugOptions::release_xla_gpu_algorithm_denylist_path() {
|
|
// @@protoc_insertion_point(field_release:xla.DebugOptions.xla_gpu_algorithm_denylist_path)
|
|
return _impl_.xla_gpu_algorithm_denylist_path_.Release();
|
|
}
|
|
inline void DebugOptions::set_allocated_xla_gpu_algorithm_denylist_path(std::string* xla_gpu_algorithm_denylist_path) {
|
|
if (xla_gpu_algorithm_denylist_path != nullptr) {
|
|
|
|
} else {
|
|
|
|
}
|
|
_impl_.xla_gpu_algorithm_denylist_path_.SetAllocated(xla_gpu_algorithm_denylist_path, GetArenaForAllocation());
|
|
#ifdef PROTOBUF_FORCE_COPY_DEFAULT_STRING
|
|
if (_impl_.xla_gpu_algorithm_denylist_path_.IsDefault()) {
|
|
_impl_.xla_gpu_algorithm_denylist_path_.Set("", GetArenaForAllocation());
|
|
}
|
|
#endif // PROTOBUF_FORCE_COPY_DEFAULT_STRING
|
|
// @@protoc_insertion_point(field_set_allocated:xla.DebugOptions.xla_gpu_algorithm_denylist_path)
|
|
}
|
|
|
|
// bool xla_tpu_detect_nan = 135;
|
|
inline void DebugOptions::clear_xla_tpu_detect_nan() {
|
|
_impl_.xla_tpu_detect_nan_ = false;
|
|
}
|
|
inline bool DebugOptions::_internal_xla_tpu_detect_nan() const {
|
|
return _impl_.xla_tpu_detect_nan_;
|
|
}
|
|
inline bool DebugOptions::xla_tpu_detect_nan() const {
|
|
// @@protoc_insertion_point(field_get:xla.DebugOptions.xla_tpu_detect_nan)
|
|
return _internal_xla_tpu_detect_nan();
|
|
}
|
|
inline void DebugOptions::_internal_set_xla_tpu_detect_nan(bool value) {
|
|
|
|
_impl_.xla_tpu_detect_nan_ = value;
|
|
}
|
|
inline void DebugOptions::set_xla_tpu_detect_nan(bool value) {
|
|
_internal_set_xla_tpu_detect_nan(value);
|
|
// @@protoc_insertion_point(field_set:xla.DebugOptions.xla_tpu_detect_nan)
|
|
}
|
|
|
|
// bool xla_tpu_detect_inf = 136;
|
|
inline void DebugOptions::clear_xla_tpu_detect_inf() {
|
|
_impl_.xla_tpu_detect_inf_ = false;
|
|
}
|
|
inline bool DebugOptions::_internal_xla_tpu_detect_inf() const {
|
|
return _impl_.xla_tpu_detect_inf_;
|
|
}
|
|
inline bool DebugOptions::xla_tpu_detect_inf() const {
|
|
// @@protoc_insertion_point(field_get:xla.DebugOptions.xla_tpu_detect_inf)
|
|
return _internal_xla_tpu_detect_inf();
|
|
}
|
|
inline void DebugOptions::_internal_set_xla_tpu_detect_inf(bool value) {
|
|
|
|
_impl_.xla_tpu_detect_inf_ = value;
|
|
}
|
|
inline void DebugOptions::set_xla_tpu_detect_inf(bool value) {
|
|
_internal_set_xla_tpu_detect_inf(value);
|
|
// @@protoc_insertion_point(field_set:xla.DebugOptions.xla_tpu_detect_inf)
|
|
}
|
|
|
|
// bool xla_cpu_enable_xprof_traceme = 137;
|
|
inline void DebugOptions::clear_xla_cpu_enable_xprof_traceme() {
|
|
_impl_.xla_cpu_enable_xprof_traceme_ = false;
|
|
}
|
|
inline bool DebugOptions::_internal_xla_cpu_enable_xprof_traceme() const {
|
|
return _impl_.xla_cpu_enable_xprof_traceme_;
|
|
}
|
|
inline bool DebugOptions::xla_cpu_enable_xprof_traceme() const {
|
|
// @@protoc_insertion_point(field_get:xla.DebugOptions.xla_cpu_enable_xprof_traceme)
|
|
return _internal_xla_cpu_enable_xprof_traceme();
|
|
}
|
|
inline void DebugOptions::_internal_set_xla_cpu_enable_xprof_traceme(bool value) {
|
|
|
|
_impl_.xla_cpu_enable_xprof_traceme_ = value;
|
|
}
|
|
inline void DebugOptions::set_xla_cpu_enable_xprof_traceme(bool value) {
|
|
_internal_set_xla_cpu_enable_xprof_traceme(value);
|
|
// @@protoc_insertion_point(field_set:xla.DebugOptions.xla_cpu_enable_xprof_traceme)
|
|
}
|
|
|
|
// bool xla_gpu_unsafe_fallback_to_driver_on_ptxas_not_found = 138;
|
|
inline void DebugOptions::clear_xla_gpu_unsafe_fallback_to_driver_on_ptxas_not_found() {
|
|
_impl_.xla_gpu_unsafe_fallback_to_driver_on_ptxas_not_found_ = false;
|
|
}
|
|
inline bool DebugOptions::_internal_xla_gpu_unsafe_fallback_to_driver_on_ptxas_not_found() const {
|
|
return _impl_.xla_gpu_unsafe_fallback_to_driver_on_ptxas_not_found_;
|
|
}
|
|
inline bool DebugOptions::xla_gpu_unsafe_fallback_to_driver_on_ptxas_not_found() const {
|
|
// @@protoc_insertion_point(field_get:xla.DebugOptions.xla_gpu_unsafe_fallback_to_driver_on_ptxas_not_found)
|
|
return _internal_xla_gpu_unsafe_fallback_to_driver_on_ptxas_not_found();
|
|
}
|
|
inline void DebugOptions::_internal_set_xla_gpu_unsafe_fallback_to_driver_on_ptxas_not_found(bool value) {
|
|
|
|
_impl_.xla_gpu_unsafe_fallback_to_driver_on_ptxas_not_found_ = value;
|
|
}
|
|
inline void DebugOptions::set_xla_gpu_unsafe_fallback_to_driver_on_ptxas_not_found(bool value) {
|
|
_internal_set_xla_gpu_unsafe_fallback_to_driver_on_ptxas_not_found(value);
|
|
// @@protoc_insertion_point(field_set:xla.DebugOptions.xla_gpu_unsafe_fallback_to_driver_on_ptxas_not_found)
|
|
}
|
|
|
|
// string xla_gpu_asm_extra_flags = 141;
|
|
inline void DebugOptions::clear_xla_gpu_asm_extra_flags() {
|
|
_impl_.xla_gpu_asm_extra_flags_.ClearToEmpty();
|
|
}
|
|
inline const std::string& DebugOptions::xla_gpu_asm_extra_flags() const {
|
|
// @@protoc_insertion_point(field_get:xla.DebugOptions.xla_gpu_asm_extra_flags)
|
|
return _internal_xla_gpu_asm_extra_flags();
|
|
}
|
|
template <typename ArgT0, typename... ArgT>
|
|
inline PROTOBUF_ALWAYS_INLINE
|
|
void DebugOptions::set_xla_gpu_asm_extra_flags(ArgT0&& arg0, ArgT... args) {
|
|
|
|
_impl_.xla_gpu_asm_extra_flags_.Set(static_cast<ArgT0 &&>(arg0), args..., GetArenaForAllocation());
|
|
// @@protoc_insertion_point(field_set:xla.DebugOptions.xla_gpu_asm_extra_flags)
|
|
}
|
|
inline std::string* DebugOptions::mutable_xla_gpu_asm_extra_flags() {
|
|
std::string* _s = _internal_mutable_xla_gpu_asm_extra_flags();
|
|
// @@protoc_insertion_point(field_mutable:xla.DebugOptions.xla_gpu_asm_extra_flags)
|
|
return _s;
|
|
}
|
|
inline const std::string& DebugOptions::_internal_xla_gpu_asm_extra_flags() const {
|
|
return _impl_.xla_gpu_asm_extra_flags_.Get();
|
|
}
|
|
inline void DebugOptions::_internal_set_xla_gpu_asm_extra_flags(const std::string& value) {
|
|
|
|
_impl_.xla_gpu_asm_extra_flags_.Set(value, GetArenaForAllocation());
|
|
}
|
|
inline std::string* DebugOptions::_internal_mutable_xla_gpu_asm_extra_flags() {
|
|
|
|
return _impl_.xla_gpu_asm_extra_flags_.Mutable(GetArenaForAllocation());
|
|
}
|
|
inline std::string* DebugOptions::release_xla_gpu_asm_extra_flags() {
|
|
// @@protoc_insertion_point(field_release:xla.DebugOptions.xla_gpu_asm_extra_flags)
|
|
return _impl_.xla_gpu_asm_extra_flags_.Release();
|
|
}
|
|
inline void DebugOptions::set_allocated_xla_gpu_asm_extra_flags(std::string* xla_gpu_asm_extra_flags) {
|
|
if (xla_gpu_asm_extra_flags != nullptr) {
|
|
|
|
} else {
|
|
|
|
}
|
|
_impl_.xla_gpu_asm_extra_flags_.SetAllocated(xla_gpu_asm_extra_flags, GetArenaForAllocation());
|
|
#ifdef PROTOBUF_FORCE_COPY_DEFAULT_STRING
|
|
if (_impl_.xla_gpu_asm_extra_flags_.IsDefault()) {
|
|
_impl_.xla_gpu_asm_extra_flags_.Set("", GetArenaForAllocation());
|
|
}
|
|
#endif // PROTOBUF_FORCE_COPY_DEFAULT_STRING
|
|
// @@protoc_insertion_point(field_set_allocated:xla.DebugOptions.xla_gpu_asm_extra_flags)
|
|
}
|
|
|
|
// int32 xla_multiheap_size_constraint_per_heap = 142;
|
|
inline void DebugOptions::clear_xla_multiheap_size_constraint_per_heap() {
|
|
_impl_.xla_multiheap_size_constraint_per_heap_ = 0;
|
|
}
|
|
inline int32_t DebugOptions::_internal_xla_multiheap_size_constraint_per_heap() const {
|
|
return _impl_.xla_multiheap_size_constraint_per_heap_;
|
|
}
|
|
inline int32_t DebugOptions::xla_multiheap_size_constraint_per_heap() const {
|
|
// @@protoc_insertion_point(field_get:xla.DebugOptions.xla_multiheap_size_constraint_per_heap)
|
|
return _internal_xla_multiheap_size_constraint_per_heap();
|
|
}
|
|
inline void DebugOptions::_internal_set_xla_multiheap_size_constraint_per_heap(int32_t value) {
|
|
|
|
_impl_.xla_multiheap_size_constraint_per_heap_ = value;
|
|
}
|
|
inline void DebugOptions::set_xla_multiheap_size_constraint_per_heap(int32_t value) {
|
|
_internal_set_xla_multiheap_size_constraint_per_heap(value);
|
|
// @@protoc_insertion_point(field_set:xla.DebugOptions.xla_multiheap_size_constraint_per_heap)
|
|
}
|
|
|
|
// bool xla_detailed_logging = 252;
|
|
inline void DebugOptions::clear_xla_detailed_logging() {
|
|
_impl_.xla_detailed_logging_ = false;
|
|
}
|
|
inline bool DebugOptions::_internal_xla_detailed_logging() const {
|
|
return _impl_.xla_detailed_logging_;
|
|
}
|
|
inline bool DebugOptions::xla_detailed_logging() const {
|
|
// @@protoc_insertion_point(field_get:xla.DebugOptions.xla_detailed_logging)
|
|
return _internal_xla_detailed_logging();
|
|
}
|
|
inline void DebugOptions::_internal_set_xla_detailed_logging(bool value) {
|
|
|
|
_impl_.xla_detailed_logging_ = value;
|
|
}
|
|
inline void DebugOptions::set_xla_detailed_logging(bool value) {
|
|
_internal_set_xla_detailed_logging(value);
|
|
// @@protoc_insertion_point(field_set:xla.DebugOptions.xla_detailed_logging)
|
|
}
|
|
|
|
// bool xla_enable_dumping = 253;
|
|
inline void DebugOptions::clear_xla_enable_dumping() {
|
|
_impl_.xla_enable_dumping_ = false;
|
|
}
|
|
inline bool DebugOptions::_internal_xla_enable_dumping() const {
|
|
return _impl_.xla_enable_dumping_;
|
|
}
|
|
inline bool DebugOptions::xla_enable_dumping() const {
|
|
// @@protoc_insertion_point(field_get:xla.DebugOptions.xla_enable_dumping)
|
|
return _internal_xla_enable_dumping();
|
|
}
|
|
inline void DebugOptions::_internal_set_xla_enable_dumping(bool value) {
|
|
|
|
_impl_.xla_enable_dumping_ = value;
|
|
}
|
|
inline void DebugOptions::set_xla_enable_dumping(bool value) {
|
|
_internal_set_xla_enable_dumping(value);
|
|
// @@protoc_insertion_point(field_set:xla.DebugOptions.xla_enable_dumping)
|
|
}
|
|
|
|
// int32 xla_gpu_force_compilation_parallelism = 147;
|
|
inline void DebugOptions::clear_xla_gpu_force_compilation_parallelism() {
|
|
_impl_.xla_gpu_force_compilation_parallelism_ = 0;
|
|
}
|
|
inline int32_t DebugOptions::_internal_xla_gpu_force_compilation_parallelism() const {
|
|
return _impl_.xla_gpu_force_compilation_parallelism_;
|
|
}
|
|
inline int32_t DebugOptions::xla_gpu_force_compilation_parallelism() const {
|
|
// @@protoc_insertion_point(field_get:xla.DebugOptions.xla_gpu_force_compilation_parallelism)
|
|
return _internal_xla_gpu_force_compilation_parallelism();
|
|
}
|
|
inline void DebugOptions::_internal_set_xla_gpu_force_compilation_parallelism(int32_t value) {
|
|
|
|
_impl_.xla_gpu_force_compilation_parallelism_ = value;
|
|
}
|
|
inline void DebugOptions::set_xla_gpu_force_compilation_parallelism(int32_t value) {
|
|
_internal_set_xla_gpu_force_compilation_parallelism(value);
|
|
// @@protoc_insertion_point(field_set:xla.DebugOptions.xla_gpu_force_compilation_parallelism)
|
|
}
|
|
|
|
// bool xla_gpu_enable_llvm_module_compilation_parallelism = 268;
|
|
inline void DebugOptions::clear_xla_gpu_enable_llvm_module_compilation_parallelism() {
|
|
_impl_.xla_gpu_enable_llvm_module_compilation_parallelism_ = false;
|
|
}
|
|
inline bool DebugOptions::_internal_xla_gpu_enable_llvm_module_compilation_parallelism() const {
|
|
return _impl_.xla_gpu_enable_llvm_module_compilation_parallelism_;
|
|
}
|
|
inline bool DebugOptions::xla_gpu_enable_llvm_module_compilation_parallelism() const {
|
|
// @@protoc_insertion_point(field_get:xla.DebugOptions.xla_gpu_enable_llvm_module_compilation_parallelism)
|
|
return _internal_xla_gpu_enable_llvm_module_compilation_parallelism();
|
|
}
|
|
inline void DebugOptions::_internal_set_xla_gpu_enable_llvm_module_compilation_parallelism(bool value) {
|
|
|
|
_impl_.xla_gpu_enable_llvm_module_compilation_parallelism_ = value;
|
|
}
|
|
inline void DebugOptions::set_xla_gpu_enable_llvm_module_compilation_parallelism(bool value) {
|
|
_internal_set_xla_gpu_enable_llvm_module_compilation_parallelism(value);
|
|
// @@protoc_insertion_point(field_set:xla.DebugOptions.xla_gpu_enable_llvm_module_compilation_parallelism)
|
|
}
|
|
|
|
// bool xla_gpu_deterministic_ops = 148;
|
|
inline void DebugOptions::clear_xla_gpu_deterministic_ops() {
|
|
_impl_.xla_gpu_deterministic_ops_ = false;
|
|
}
|
|
inline bool DebugOptions::_internal_xla_gpu_deterministic_ops() const {
|
|
return _impl_.xla_gpu_deterministic_ops_;
|
|
}
|
|
inline bool DebugOptions::xla_gpu_deterministic_ops() const {
|
|
// @@protoc_insertion_point(field_get:xla.DebugOptions.xla_gpu_deterministic_ops)
|
|
return _internal_xla_gpu_deterministic_ops();
|
|
}
|
|
inline void DebugOptions::_internal_set_xla_gpu_deterministic_ops(bool value) {
|
|
|
|
_impl_.xla_gpu_deterministic_ops_ = value;
|
|
}
|
|
inline void DebugOptions::set_xla_gpu_deterministic_ops(bool value) {
|
|
_internal_set_xla_gpu_deterministic_ops(value);
|
|
// @@protoc_insertion_point(field_set:xla.DebugOptions.xla_gpu_deterministic_ops)
|
|
}
|
|
|
|
// repeated string xla_gpu_llvm_ir_file = 150;
|
|
inline int DebugOptions::_internal_xla_gpu_llvm_ir_file_size() const {
|
|
return _impl_.xla_gpu_llvm_ir_file_.size();
|
|
}
|
|
inline int DebugOptions::xla_gpu_llvm_ir_file_size() const {
|
|
return _internal_xla_gpu_llvm_ir_file_size();
|
|
}
|
|
inline void DebugOptions::clear_xla_gpu_llvm_ir_file() {
|
|
_impl_.xla_gpu_llvm_ir_file_.Clear();
|
|
}
|
|
inline std::string* DebugOptions::add_xla_gpu_llvm_ir_file() {
|
|
std::string* _s = _internal_add_xla_gpu_llvm_ir_file();
|
|
// @@protoc_insertion_point(field_add_mutable:xla.DebugOptions.xla_gpu_llvm_ir_file)
|
|
return _s;
|
|
}
|
|
inline const std::string& DebugOptions::_internal_xla_gpu_llvm_ir_file(int index) const {
|
|
return _impl_.xla_gpu_llvm_ir_file_.Get(index);
|
|
}
|
|
inline const std::string& DebugOptions::xla_gpu_llvm_ir_file(int index) const {
|
|
// @@protoc_insertion_point(field_get:xla.DebugOptions.xla_gpu_llvm_ir_file)
|
|
return _internal_xla_gpu_llvm_ir_file(index);
|
|
}
|
|
inline std::string* DebugOptions::mutable_xla_gpu_llvm_ir_file(int index) {
|
|
// @@protoc_insertion_point(field_mutable:xla.DebugOptions.xla_gpu_llvm_ir_file)
|
|
return _impl_.xla_gpu_llvm_ir_file_.Mutable(index);
|
|
}
|
|
inline void DebugOptions::set_xla_gpu_llvm_ir_file(int index, const std::string& value) {
|
|
_impl_.xla_gpu_llvm_ir_file_.Mutable(index)->assign(value);
|
|
// @@protoc_insertion_point(field_set:xla.DebugOptions.xla_gpu_llvm_ir_file)
|
|
}
|
|
inline void DebugOptions::set_xla_gpu_llvm_ir_file(int index, std::string&& value) {
|
|
_impl_.xla_gpu_llvm_ir_file_.Mutable(index)->assign(std::move(value));
|
|
// @@protoc_insertion_point(field_set:xla.DebugOptions.xla_gpu_llvm_ir_file)
|
|
}
|
|
inline void DebugOptions::set_xla_gpu_llvm_ir_file(int index, const char* value) {
|
|
GOOGLE_DCHECK(value != nullptr);
|
|
_impl_.xla_gpu_llvm_ir_file_.Mutable(index)->assign(value);
|
|
// @@protoc_insertion_point(field_set_char:xla.DebugOptions.xla_gpu_llvm_ir_file)
|
|
}
|
|
inline void DebugOptions::set_xla_gpu_llvm_ir_file(int index, const char* value, size_t size) {
|
|
_impl_.xla_gpu_llvm_ir_file_.Mutable(index)->assign(
|
|
reinterpret_cast<const char*>(value), size);
|
|
// @@protoc_insertion_point(field_set_pointer:xla.DebugOptions.xla_gpu_llvm_ir_file)
|
|
}
|
|
inline std::string* DebugOptions::_internal_add_xla_gpu_llvm_ir_file() {
|
|
return _impl_.xla_gpu_llvm_ir_file_.Add();
|
|
}
|
|
inline void DebugOptions::add_xla_gpu_llvm_ir_file(const std::string& value) {
|
|
_impl_.xla_gpu_llvm_ir_file_.Add()->assign(value);
|
|
// @@protoc_insertion_point(field_add:xla.DebugOptions.xla_gpu_llvm_ir_file)
|
|
}
|
|
inline void DebugOptions::add_xla_gpu_llvm_ir_file(std::string&& value) {
|
|
_impl_.xla_gpu_llvm_ir_file_.Add(std::move(value));
|
|
// @@protoc_insertion_point(field_add:xla.DebugOptions.xla_gpu_llvm_ir_file)
|
|
}
|
|
inline void DebugOptions::add_xla_gpu_llvm_ir_file(const char* value) {
|
|
GOOGLE_DCHECK(value != nullptr);
|
|
_impl_.xla_gpu_llvm_ir_file_.Add()->assign(value);
|
|
// @@protoc_insertion_point(field_add_char:xla.DebugOptions.xla_gpu_llvm_ir_file)
|
|
}
|
|
inline void DebugOptions::add_xla_gpu_llvm_ir_file(const char* value, size_t size) {
|
|
_impl_.xla_gpu_llvm_ir_file_.Add()->assign(reinterpret_cast<const char*>(value), size);
|
|
// @@protoc_insertion_point(field_add_pointer:xla.DebugOptions.xla_gpu_llvm_ir_file)
|
|
}
|
|
inline const ::PROTOBUF_NAMESPACE_ID::RepeatedPtrField<std::string>&
|
|
DebugOptions::xla_gpu_llvm_ir_file() const {
|
|
// @@protoc_insertion_point(field_list:xla.DebugOptions.xla_gpu_llvm_ir_file)
|
|
return _impl_.xla_gpu_llvm_ir_file_;
|
|
}
|
|
inline ::PROTOBUF_NAMESPACE_ID::RepeatedPtrField<std::string>*
|
|
DebugOptions::mutable_xla_gpu_llvm_ir_file() {
|
|
// @@protoc_insertion_point(field_mutable_list:xla.DebugOptions.xla_gpu_llvm_ir_file)
|
|
return &_impl_.xla_gpu_llvm_ir_file_;
|
|
}
|
|
|
|
// bool xla_gpu_enable_async_collectives = 238;
|
|
inline void DebugOptions::clear_xla_gpu_enable_async_collectives() {
|
|
_impl_.xla_gpu_enable_async_collectives_ = false;
|
|
}
|
|
inline bool DebugOptions::_internal_xla_gpu_enable_async_collectives() const {
|
|
return _impl_.xla_gpu_enable_async_collectives_;
|
|
}
|
|
inline bool DebugOptions::xla_gpu_enable_async_collectives() const {
|
|
// @@protoc_insertion_point(field_get:xla.DebugOptions.xla_gpu_enable_async_collectives)
|
|
return _internal_xla_gpu_enable_async_collectives();
|
|
}
|
|
inline void DebugOptions::_internal_set_xla_gpu_enable_async_collectives(bool value) {
|
|
|
|
_impl_.xla_gpu_enable_async_collectives_ = value;
|
|
}
|
|
inline void DebugOptions::set_xla_gpu_enable_async_collectives(bool value) {
|
|
_internal_set_xla_gpu_enable_async_collectives(value);
|
|
// @@protoc_insertion_point(field_set:xla.DebugOptions.xla_gpu_enable_async_collectives)
|
|
}
|
|
|
|
// bool xla_gpu_enable_async_all_reduce = 152;
|
|
inline void DebugOptions::clear_xla_gpu_enable_async_all_reduce() {
|
|
_impl_.xla_gpu_enable_async_all_reduce_ = false;
|
|
}
|
|
inline bool DebugOptions::_internal_xla_gpu_enable_async_all_reduce() const {
|
|
return _impl_.xla_gpu_enable_async_all_reduce_;
|
|
}
|
|
inline bool DebugOptions::xla_gpu_enable_async_all_reduce() const {
|
|
// @@protoc_insertion_point(field_get:xla.DebugOptions.xla_gpu_enable_async_all_reduce)
|
|
return _internal_xla_gpu_enable_async_all_reduce();
|
|
}
|
|
inline void DebugOptions::_internal_set_xla_gpu_enable_async_all_reduce(bool value) {
|
|
|
|
_impl_.xla_gpu_enable_async_all_reduce_ = value;
|
|
}
|
|
inline void DebugOptions::set_xla_gpu_enable_async_all_reduce(bool value) {
|
|
_internal_set_xla_gpu_enable_async_all_reduce(value);
|
|
// @@protoc_insertion_point(field_set:xla.DebugOptions.xla_gpu_enable_async_all_reduce)
|
|
}
|
|
|
|
// bool xla_gpu_enable_async_collective_permute = 183;
|
|
inline void DebugOptions::clear_xla_gpu_enable_async_collective_permute() {
|
|
_impl_.xla_gpu_enable_async_collective_permute_ = false;
|
|
}
|
|
inline bool DebugOptions::_internal_xla_gpu_enable_async_collective_permute() const {
|
|
return _impl_.xla_gpu_enable_async_collective_permute_;
|
|
}
|
|
inline bool DebugOptions::xla_gpu_enable_async_collective_permute() const {
|
|
// @@protoc_insertion_point(field_get:xla.DebugOptions.xla_gpu_enable_async_collective_permute)
|
|
return _internal_xla_gpu_enable_async_collective_permute();
|
|
}
|
|
inline void DebugOptions::_internal_set_xla_gpu_enable_async_collective_permute(bool value) {
|
|
|
|
_impl_.xla_gpu_enable_async_collective_permute_ = value;
|
|
}
|
|
inline void DebugOptions::set_xla_gpu_enable_async_collective_permute(bool value) {
|
|
_internal_set_xla_gpu_enable_async_collective_permute(value);
|
|
// @@protoc_insertion_point(field_set:xla.DebugOptions.xla_gpu_enable_async_collective_permute)
|
|
}
|
|
|
|
// bool xla_gpu_enable_async_all_gather = 199;
|
|
inline void DebugOptions::clear_xla_gpu_enable_async_all_gather() {
|
|
_impl_.xla_gpu_enable_async_all_gather_ = false;
|
|
}
|
|
inline bool DebugOptions::_internal_xla_gpu_enable_async_all_gather() const {
|
|
return _impl_.xla_gpu_enable_async_all_gather_;
|
|
}
|
|
inline bool DebugOptions::xla_gpu_enable_async_all_gather() const {
|
|
// @@protoc_insertion_point(field_get:xla.DebugOptions.xla_gpu_enable_async_all_gather)
|
|
return _internal_xla_gpu_enable_async_all_gather();
|
|
}
|
|
inline void DebugOptions::_internal_set_xla_gpu_enable_async_all_gather(bool value) {
|
|
|
|
_impl_.xla_gpu_enable_async_all_gather_ = value;
|
|
}
|
|
inline void DebugOptions::set_xla_gpu_enable_async_all_gather(bool value) {
|
|
_internal_set_xla_gpu_enable_async_all_gather(value);
|
|
// @@protoc_insertion_point(field_set:xla.DebugOptions.xla_gpu_enable_async_all_gather)
|
|
}
|
|
|
|
// bool xla_gpu_enable_async_reduce_scatter = 200;
|
|
inline void DebugOptions::clear_xla_gpu_enable_async_reduce_scatter() {
|
|
_impl_.xla_gpu_enable_async_reduce_scatter_ = false;
|
|
}
|
|
inline bool DebugOptions::_internal_xla_gpu_enable_async_reduce_scatter() const {
|
|
return _impl_.xla_gpu_enable_async_reduce_scatter_;
|
|
}
|
|
inline bool DebugOptions::xla_gpu_enable_async_reduce_scatter() const {
|
|
// @@protoc_insertion_point(field_get:xla.DebugOptions.xla_gpu_enable_async_reduce_scatter)
|
|
return _internal_xla_gpu_enable_async_reduce_scatter();
|
|
}
|
|
inline void DebugOptions::_internal_set_xla_gpu_enable_async_reduce_scatter(bool value) {
|
|
|
|
_impl_.xla_gpu_enable_async_reduce_scatter_ = value;
|
|
}
|
|
inline void DebugOptions::set_xla_gpu_enable_async_reduce_scatter(bool value) {
|
|
_internal_set_xla_gpu_enable_async_reduce_scatter(value);
|
|
// @@protoc_insertion_point(field_set:xla.DebugOptions.xla_gpu_enable_async_reduce_scatter)
|
|
}
|
|
|
|
// bool xla_gpu_enable_async_all_to_all = 201;
|
|
inline void DebugOptions::clear_xla_gpu_enable_async_all_to_all() {
|
|
_impl_.xla_gpu_enable_async_all_to_all_ = false;
|
|
}
|
|
inline bool DebugOptions::_internal_xla_gpu_enable_async_all_to_all() const {
|
|
return _impl_.xla_gpu_enable_async_all_to_all_;
|
|
}
|
|
inline bool DebugOptions::xla_gpu_enable_async_all_to_all() const {
|
|
// @@protoc_insertion_point(field_get:xla.DebugOptions.xla_gpu_enable_async_all_to_all)
|
|
return _internal_xla_gpu_enable_async_all_to_all();
|
|
}
|
|
inline void DebugOptions::_internal_set_xla_gpu_enable_async_all_to_all(bool value) {
|
|
|
|
_impl_.xla_gpu_enable_async_all_to_all_ = value;
|
|
}
|
|
inline void DebugOptions::set_xla_gpu_enable_async_all_to_all(bool value) {
|
|
_internal_set_xla_gpu_enable_async_all_to_all(value);
|
|
// @@protoc_insertion_point(field_set:xla.DebugOptions.xla_gpu_enable_async_all_to_all)
|
|
}
|
|
|
|
// int64 xla_gpu_all_reduce_combine_threshold_bytes = 157;
|
|
inline void DebugOptions::clear_xla_gpu_all_reduce_combine_threshold_bytes() {
|
|
_impl_.xla_gpu_all_reduce_combine_threshold_bytes_ = int64_t{0};
|
|
}
|
|
inline int64_t DebugOptions::_internal_xla_gpu_all_reduce_combine_threshold_bytes() const {
|
|
return _impl_.xla_gpu_all_reduce_combine_threshold_bytes_;
|
|
}
|
|
inline int64_t DebugOptions::xla_gpu_all_reduce_combine_threshold_bytes() const {
|
|
// @@protoc_insertion_point(field_get:xla.DebugOptions.xla_gpu_all_reduce_combine_threshold_bytes)
|
|
return _internal_xla_gpu_all_reduce_combine_threshold_bytes();
|
|
}
|
|
inline void DebugOptions::_internal_set_xla_gpu_all_reduce_combine_threshold_bytes(int64_t value) {
|
|
|
|
_impl_.xla_gpu_all_reduce_combine_threshold_bytes_ = value;
|
|
}
|
|
inline void DebugOptions::set_xla_gpu_all_reduce_combine_threshold_bytes(int64_t value) {
|
|
_internal_set_xla_gpu_all_reduce_combine_threshold_bytes(value);
|
|
// @@protoc_insertion_point(field_set:xla.DebugOptions.xla_gpu_all_reduce_combine_threshold_bytes)
|
|
}
|
|
|
|
// int64 xla_gpu_all_gather_combine_threshold_bytes = 212;
|
|
inline void DebugOptions::clear_xla_gpu_all_gather_combine_threshold_bytes() {
|
|
_impl_.xla_gpu_all_gather_combine_threshold_bytes_ = int64_t{0};
|
|
}
|
|
inline int64_t DebugOptions::_internal_xla_gpu_all_gather_combine_threshold_bytes() const {
|
|
return _impl_.xla_gpu_all_gather_combine_threshold_bytes_;
|
|
}
|
|
inline int64_t DebugOptions::xla_gpu_all_gather_combine_threshold_bytes() const {
|
|
// @@protoc_insertion_point(field_get:xla.DebugOptions.xla_gpu_all_gather_combine_threshold_bytes)
|
|
return _internal_xla_gpu_all_gather_combine_threshold_bytes();
|
|
}
|
|
inline void DebugOptions::_internal_set_xla_gpu_all_gather_combine_threshold_bytes(int64_t value) {
|
|
|
|
_impl_.xla_gpu_all_gather_combine_threshold_bytes_ = value;
|
|
}
|
|
inline void DebugOptions::set_xla_gpu_all_gather_combine_threshold_bytes(int64_t value) {
|
|
_internal_set_xla_gpu_all_gather_combine_threshold_bytes(value);
|
|
// @@protoc_insertion_point(field_set:xla.DebugOptions.xla_gpu_all_gather_combine_threshold_bytes)
|
|
}
|
|
|
|
// int64 xla_gpu_reduce_scatter_combine_threshold_bytes = 213;
|
|
inline void DebugOptions::clear_xla_gpu_reduce_scatter_combine_threshold_bytes() {
|
|
_impl_.xla_gpu_reduce_scatter_combine_threshold_bytes_ = int64_t{0};
|
|
}
|
|
inline int64_t DebugOptions::_internal_xla_gpu_reduce_scatter_combine_threshold_bytes() const {
|
|
return _impl_.xla_gpu_reduce_scatter_combine_threshold_bytes_;
|
|
}
|
|
inline int64_t DebugOptions::xla_gpu_reduce_scatter_combine_threshold_bytes() const {
|
|
// @@protoc_insertion_point(field_get:xla.DebugOptions.xla_gpu_reduce_scatter_combine_threshold_bytes)
|
|
return _internal_xla_gpu_reduce_scatter_combine_threshold_bytes();
|
|
}
|
|
inline void DebugOptions::_internal_set_xla_gpu_reduce_scatter_combine_threshold_bytes(int64_t value) {
|
|
|
|
_impl_.xla_gpu_reduce_scatter_combine_threshold_bytes_ = value;
|
|
}
|
|
inline void DebugOptions::set_xla_gpu_reduce_scatter_combine_threshold_bytes(int64_t value) {
|
|
_internal_set_xla_gpu_reduce_scatter_combine_threshold_bytes(value);
|
|
// @@protoc_insertion_point(field_set:xla.DebugOptions.xla_gpu_reduce_scatter_combine_threshold_bytes)
|
|
}
|
|
|
|
// bool xla_gpu_enable_all_gather_combine_by_dim = 254;
|
|
inline void DebugOptions::clear_xla_gpu_enable_all_gather_combine_by_dim() {
|
|
_impl_.xla_gpu_enable_all_gather_combine_by_dim_ = false;
|
|
}
|
|
inline bool DebugOptions::_internal_xla_gpu_enable_all_gather_combine_by_dim() const {
|
|
return _impl_.xla_gpu_enable_all_gather_combine_by_dim_;
|
|
}
|
|
inline bool DebugOptions::xla_gpu_enable_all_gather_combine_by_dim() const {
|
|
// @@protoc_insertion_point(field_get:xla.DebugOptions.xla_gpu_enable_all_gather_combine_by_dim)
|
|
return _internal_xla_gpu_enable_all_gather_combine_by_dim();
|
|
}
|
|
inline void DebugOptions::_internal_set_xla_gpu_enable_all_gather_combine_by_dim(bool value) {
|
|
|
|
_impl_.xla_gpu_enable_all_gather_combine_by_dim_ = value;
|
|
}
|
|
inline void DebugOptions::set_xla_gpu_enable_all_gather_combine_by_dim(bool value) {
|
|
_internal_set_xla_gpu_enable_all_gather_combine_by_dim(value);
|
|
// @@protoc_insertion_point(field_set:xla.DebugOptions.xla_gpu_enable_all_gather_combine_by_dim)
|
|
}
|
|
|
|
// bool xla_gpu_enable_reduce_scatter_combine_by_dim = 257;
|
|
inline void DebugOptions::clear_xla_gpu_enable_reduce_scatter_combine_by_dim() {
|
|
_impl_.xla_gpu_enable_reduce_scatter_combine_by_dim_ = false;
|
|
}
|
|
inline bool DebugOptions::_internal_xla_gpu_enable_reduce_scatter_combine_by_dim() const {
|
|
return _impl_.xla_gpu_enable_reduce_scatter_combine_by_dim_;
|
|
}
|
|
inline bool DebugOptions::xla_gpu_enable_reduce_scatter_combine_by_dim() const {
|
|
// @@protoc_insertion_point(field_get:xla.DebugOptions.xla_gpu_enable_reduce_scatter_combine_by_dim)
|
|
return _internal_xla_gpu_enable_reduce_scatter_combine_by_dim();
|
|
}
|
|
inline void DebugOptions::_internal_set_xla_gpu_enable_reduce_scatter_combine_by_dim(bool value) {
|
|
|
|
_impl_.xla_gpu_enable_reduce_scatter_combine_by_dim_ = value;
|
|
}
|
|
inline void DebugOptions::set_xla_gpu_enable_reduce_scatter_combine_by_dim(bool value) {
|
|
_internal_set_xla_gpu_enable_reduce_scatter_combine_by_dim(value);
|
|
// @@protoc_insertion_point(field_set:xla.DebugOptions.xla_gpu_enable_reduce_scatter_combine_by_dim)
|
|
}
|
|
|
|
// bool xla_gpu_all_reduce_contiguous = 158;
|
|
inline void DebugOptions::clear_xla_gpu_all_reduce_contiguous() {
|
|
_impl_.xla_gpu_all_reduce_contiguous_ = false;
|
|
}
|
|
inline bool DebugOptions::_internal_xla_gpu_all_reduce_contiguous() const {
|
|
return _impl_.xla_gpu_all_reduce_contiguous_;
|
|
}
|
|
inline bool DebugOptions::xla_gpu_all_reduce_contiguous() const {
|
|
// @@protoc_insertion_point(field_get:xla.DebugOptions.xla_gpu_all_reduce_contiguous)
|
|
return _internal_xla_gpu_all_reduce_contiguous();
|
|
}
|
|
inline void DebugOptions::_internal_set_xla_gpu_all_reduce_contiguous(bool value) {
|
|
|
|
_impl_.xla_gpu_all_reduce_contiguous_ = value;
|
|
}
|
|
inline void DebugOptions::set_xla_gpu_all_reduce_contiguous(bool value) {
|
|
_internal_set_xla_gpu_all_reduce_contiguous(value);
|
|
// @@protoc_insertion_point(field_set:xla.DebugOptions.xla_gpu_all_reduce_contiguous)
|
|
}
|
|
|
|
// bool xla_gpu_enable_reassociation_for_converted_ar = 209;
|
|
inline void DebugOptions::clear_xla_gpu_enable_reassociation_for_converted_ar() {
|
|
_impl_.xla_gpu_enable_reassociation_for_converted_ar_ = false;
|
|
}
|
|
inline bool DebugOptions::_internal_xla_gpu_enable_reassociation_for_converted_ar() const {
|
|
return _impl_.xla_gpu_enable_reassociation_for_converted_ar_;
|
|
}
|
|
inline bool DebugOptions::xla_gpu_enable_reassociation_for_converted_ar() const {
|
|
// @@protoc_insertion_point(field_get:xla.DebugOptions.xla_gpu_enable_reassociation_for_converted_ar)
|
|
return _internal_xla_gpu_enable_reassociation_for_converted_ar();
|
|
}
|
|
inline void DebugOptions::_internal_set_xla_gpu_enable_reassociation_for_converted_ar(bool value) {
|
|
|
|
_impl_.xla_gpu_enable_reassociation_for_converted_ar_ = value;
|
|
}
|
|
inline void DebugOptions::set_xla_gpu_enable_reassociation_for_converted_ar(bool value) {
|
|
_internal_set_xla_gpu_enable_reassociation_for_converted_ar(value);
|
|
// @@protoc_insertion_point(field_set:xla.DebugOptions.xla_gpu_enable_reassociation_for_converted_ar)
|
|
}
|
|
|
|
// int32 xla_gpu_all_reduce_blueconnect_num_devices_per_host = 159;
|
|
inline void DebugOptions::clear_xla_gpu_all_reduce_blueconnect_num_devices_per_host() {
|
|
_impl_.xla_gpu_all_reduce_blueconnect_num_devices_per_host_ = 0;
|
|
}
|
|
inline int32_t DebugOptions::_internal_xla_gpu_all_reduce_blueconnect_num_devices_per_host() const {
|
|
return _impl_.xla_gpu_all_reduce_blueconnect_num_devices_per_host_;
|
|
}
|
|
inline int32_t DebugOptions::xla_gpu_all_reduce_blueconnect_num_devices_per_host() const {
|
|
// @@protoc_insertion_point(field_get:xla.DebugOptions.xla_gpu_all_reduce_blueconnect_num_devices_per_host)
|
|
return _internal_xla_gpu_all_reduce_blueconnect_num_devices_per_host();
|
|
}
|
|
inline void DebugOptions::_internal_set_xla_gpu_all_reduce_blueconnect_num_devices_per_host(int32_t value) {
|
|
|
|
_impl_.xla_gpu_all_reduce_blueconnect_num_devices_per_host_ = value;
|
|
}
|
|
inline void DebugOptions::set_xla_gpu_all_reduce_blueconnect_num_devices_per_host(int32_t value) {
|
|
_internal_set_xla_gpu_all_reduce_blueconnect_num_devices_per_host(value);
|
|
// @@protoc_insertion_point(field_set:xla.DebugOptions.xla_gpu_all_reduce_blueconnect_num_devices_per_host)
|
|
}
|
|
|
|
// bool xla_gpu_enable_while_loop_reduce_scatter_code_motion = 203;
|
|
inline void DebugOptions::clear_xla_gpu_enable_while_loop_reduce_scatter_code_motion() {
|
|
_impl_.xla_gpu_enable_while_loop_reduce_scatter_code_motion_ = false;
|
|
}
|
|
inline bool DebugOptions::_internal_xla_gpu_enable_while_loop_reduce_scatter_code_motion() const {
|
|
return _impl_.xla_gpu_enable_while_loop_reduce_scatter_code_motion_;
|
|
}
|
|
inline bool DebugOptions::xla_gpu_enable_while_loop_reduce_scatter_code_motion() const {
|
|
// @@protoc_insertion_point(field_get:xla.DebugOptions.xla_gpu_enable_while_loop_reduce_scatter_code_motion)
|
|
return _internal_xla_gpu_enable_while_loop_reduce_scatter_code_motion();
|
|
}
|
|
inline void DebugOptions::_internal_set_xla_gpu_enable_while_loop_reduce_scatter_code_motion(bool value) {
|
|
|
|
_impl_.xla_gpu_enable_while_loop_reduce_scatter_code_motion_ = value;
|
|
}
|
|
inline void DebugOptions::set_xla_gpu_enable_while_loop_reduce_scatter_code_motion(bool value) {
|
|
_internal_set_xla_gpu_enable_while_loop_reduce_scatter_code_motion(value);
|
|
// @@protoc_insertion_point(field_set:xla.DebugOptions.xla_gpu_enable_while_loop_reduce_scatter_code_motion)
|
|
}
|
|
|
|
// int32 xla_gpu_collective_inflation_factor = 205;
|
|
inline void DebugOptions::clear_xla_gpu_collective_inflation_factor() {
|
|
_impl_.xla_gpu_collective_inflation_factor_ = 0;
|
|
}
|
|
inline int32_t DebugOptions::_internal_xla_gpu_collective_inflation_factor() const {
|
|
return _impl_.xla_gpu_collective_inflation_factor_;
|
|
}
|
|
inline int32_t DebugOptions::xla_gpu_collective_inflation_factor() const {
|
|
// @@protoc_insertion_point(field_get:xla.DebugOptions.xla_gpu_collective_inflation_factor)
|
|
return _internal_xla_gpu_collective_inflation_factor();
|
|
}
|
|
inline void DebugOptions::_internal_set_xla_gpu_collective_inflation_factor(int32_t value) {
|
|
|
|
_impl_.xla_gpu_collective_inflation_factor_ = value;
|
|
}
|
|
inline void DebugOptions::set_xla_gpu_collective_inflation_factor(int32_t value) {
|
|
_internal_set_xla_gpu_collective_inflation_factor(value);
|
|
// @@protoc_insertion_point(field_set:xla.DebugOptions.xla_gpu_collective_inflation_factor)
|
|
}
|
|
|
|
// bool xla_gpu_enable_cudnn_frontend = 160;
|
|
inline void DebugOptions::clear_xla_gpu_enable_cudnn_frontend() {
|
|
_impl_.xla_gpu_enable_cudnn_frontend_ = false;
|
|
}
|
|
inline bool DebugOptions::_internal_xla_gpu_enable_cudnn_frontend() const {
|
|
return _impl_.xla_gpu_enable_cudnn_frontend_;
|
|
}
|
|
inline bool DebugOptions::xla_gpu_enable_cudnn_frontend() const {
|
|
// @@protoc_insertion_point(field_get:xla.DebugOptions.xla_gpu_enable_cudnn_frontend)
|
|
return _internal_xla_gpu_enable_cudnn_frontend();
|
|
}
|
|
inline void DebugOptions::_internal_set_xla_gpu_enable_cudnn_frontend(bool value) {
|
|
|
|
_impl_.xla_gpu_enable_cudnn_frontend_ = value;
|
|
}
|
|
inline void DebugOptions::set_xla_gpu_enable_cudnn_frontend(bool value) {
|
|
_internal_set_xla_gpu_enable_cudnn_frontend(value);
|
|
// @@protoc_insertion_point(field_set:xla.DebugOptions.xla_gpu_enable_cudnn_frontend)
|
|
}
|
|
|
|
// bool xla_gpu_enable_cudnn_fmha = 218;
|
|
inline void DebugOptions::clear_xla_gpu_enable_cudnn_fmha() {
|
|
_impl_.xla_gpu_enable_cudnn_fmha_ = false;
|
|
}
|
|
inline bool DebugOptions::_internal_xla_gpu_enable_cudnn_fmha() const {
|
|
return _impl_.xla_gpu_enable_cudnn_fmha_;
|
|
}
|
|
inline bool DebugOptions::xla_gpu_enable_cudnn_fmha() const {
|
|
// @@protoc_insertion_point(field_get:xla.DebugOptions.xla_gpu_enable_cudnn_fmha)
|
|
return _internal_xla_gpu_enable_cudnn_fmha();
|
|
}
|
|
inline void DebugOptions::_internal_set_xla_gpu_enable_cudnn_fmha(bool value) {
|
|
|
|
_impl_.xla_gpu_enable_cudnn_fmha_ = value;
|
|
}
|
|
inline void DebugOptions::set_xla_gpu_enable_cudnn_fmha(bool value) {
|
|
_internal_set_xla_gpu_enable_cudnn_fmha(value);
|
|
// @@protoc_insertion_point(field_set:xla.DebugOptions.xla_gpu_enable_cudnn_fmha)
|
|
}
|
|
|
|
// bool xla_gpu_fused_attention_use_cudnn_rng = 235;
|
|
inline void DebugOptions::clear_xla_gpu_fused_attention_use_cudnn_rng() {
|
|
_impl_.xla_gpu_fused_attention_use_cudnn_rng_ = false;
|
|
}
|
|
inline bool DebugOptions::_internal_xla_gpu_fused_attention_use_cudnn_rng() const {
|
|
return _impl_.xla_gpu_fused_attention_use_cudnn_rng_;
|
|
}
|
|
inline bool DebugOptions::xla_gpu_fused_attention_use_cudnn_rng() const {
|
|
// @@protoc_insertion_point(field_get:xla.DebugOptions.xla_gpu_fused_attention_use_cudnn_rng)
|
|
return _internal_xla_gpu_fused_attention_use_cudnn_rng();
|
|
}
|
|
inline void DebugOptions::_internal_set_xla_gpu_fused_attention_use_cudnn_rng(bool value) {
|
|
|
|
_impl_.xla_gpu_fused_attention_use_cudnn_rng_ = value;
|
|
}
|
|
inline void DebugOptions::set_xla_gpu_fused_attention_use_cudnn_rng(bool value) {
|
|
_internal_set_xla_gpu_fused_attention_use_cudnn_rng(value);
|
|
// @@protoc_insertion_point(field_set:xla.DebugOptions.xla_gpu_fused_attention_use_cudnn_rng)
|
|
}
|
|
|
|
// bool xla_gpu_enable_cudnn_layer_norm = 262;
|
|
inline void DebugOptions::clear_xla_gpu_enable_cudnn_layer_norm() {
|
|
_impl_.xla_gpu_enable_cudnn_layer_norm_ = false;
|
|
}
|
|
inline bool DebugOptions::_internal_xla_gpu_enable_cudnn_layer_norm() const {
|
|
return _impl_.xla_gpu_enable_cudnn_layer_norm_;
|
|
}
|
|
inline bool DebugOptions::xla_gpu_enable_cudnn_layer_norm() const {
|
|
// @@protoc_insertion_point(field_get:xla.DebugOptions.xla_gpu_enable_cudnn_layer_norm)
|
|
return _internal_xla_gpu_enable_cudnn_layer_norm();
|
|
}
|
|
inline void DebugOptions::_internal_set_xla_gpu_enable_cudnn_layer_norm(bool value) {
|
|
|
|
_impl_.xla_gpu_enable_cudnn_layer_norm_ = value;
|
|
}
|
|
inline void DebugOptions::set_xla_gpu_enable_cudnn_layer_norm(bool value) {
|
|
_internal_set_xla_gpu_enable_cudnn_layer_norm(value);
|
|
// @@protoc_insertion_point(field_set:xla.DebugOptions.xla_gpu_enable_cudnn_layer_norm)
|
|
}
|
|
|
|
// bool xla_dump_disable_metadata = 153;
|
|
inline void DebugOptions::clear_xla_dump_disable_metadata() {
|
|
_impl_.xla_dump_disable_metadata_ = false;
|
|
}
|
|
inline bool DebugOptions::_internal_xla_dump_disable_metadata() const {
|
|
return _impl_.xla_dump_disable_metadata_;
|
|
}
|
|
inline bool DebugOptions::xla_dump_disable_metadata() const {
|
|
// @@protoc_insertion_point(field_get:xla.DebugOptions.xla_dump_disable_metadata)
|
|
return _internal_xla_dump_disable_metadata();
|
|
}
|
|
inline void DebugOptions::_internal_set_xla_dump_disable_metadata(bool value) {
|
|
|
|
_impl_.xla_dump_disable_metadata_ = value;
|
|
}
|
|
inline void DebugOptions::set_xla_dump_disable_metadata(bool value) {
|
|
_internal_set_xla_dump_disable_metadata(value);
|
|
// @@protoc_insertion_point(field_set:xla.DebugOptions.xla_dump_disable_metadata)
|
|
}
|
|
|
|
// string xla_dump_hlo_pipeline_re = 154;
|
|
inline void DebugOptions::clear_xla_dump_hlo_pipeline_re() {
|
|
_impl_.xla_dump_hlo_pipeline_re_.ClearToEmpty();
|
|
}
|
|
inline const std::string& DebugOptions::xla_dump_hlo_pipeline_re() const {
|
|
// @@protoc_insertion_point(field_get:xla.DebugOptions.xla_dump_hlo_pipeline_re)
|
|
return _internal_xla_dump_hlo_pipeline_re();
|
|
}
|
|
template <typename ArgT0, typename... ArgT>
|
|
inline PROTOBUF_ALWAYS_INLINE
|
|
void DebugOptions::set_xla_dump_hlo_pipeline_re(ArgT0&& arg0, ArgT... args) {
|
|
|
|
_impl_.xla_dump_hlo_pipeline_re_.Set(static_cast<ArgT0 &&>(arg0), args..., GetArenaForAllocation());
|
|
// @@protoc_insertion_point(field_set:xla.DebugOptions.xla_dump_hlo_pipeline_re)
|
|
}
|
|
inline std::string* DebugOptions::mutable_xla_dump_hlo_pipeline_re() {
|
|
std::string* _s = _internal_mutable_xla_dump_hlo_pipeline_re();
|
|
// @@protoc_insertion_point(field_mutable:xla.DebugOptions.xla_dump_hlo_pipeline_re)
|
|
return _s;
|
|
}
|
|
inline const std::string& DebugOptions::_internal_xla_dump_hlo_pipeline_re() const {
|
|
return _impl_.xla_dump_hlo_pipeline_re_.Get();
|
|
}
|
|
inline void DebugOptions::_internal_set_xla_dump_hlo_pipeline_re(const std::string& value) {
|
|
|
|
_impl_.xla_dump_hlo_pipeline_re_.Set(value, GetArenaForAllocation());
|
|
}
|
|
inline std::string* DebugOptions::_internal_mutable_xla_dump_hlo_pipeline_re() {
|
|
|
|
return _impl_.xla_dump_hlo_pipeline_re_.Mutable(GetArenaForAllocation());
|
|
}
|
|
inline std::string* DebugOptions::release_xla_dump_hlo_pipeline_re() {
|
|
// @@protoc_insertion_point(field_release:xla.DebugOptions.xla_dump_hlo_pipeline_re)
|
|
return _impl_.xla_dump_hlo_pipeline_re_.Release();
|
|
}
|
|
inline void DebugOptions::set_allocated_xla_dump_hlo_pipeline_re(std::string* xla_dump_hlo_pipeline_re) {
|
|
if (xla_dump_hlo_pipeline_re != nullptr) {
|
|
|
|
} else {
|
|
|
|
}
|
|
_impl_.xla_dump_hlo_pipeline_re_.SetAllocated(xla_dump_hlo_pipeline_re, GetArenaForAllocation());
|
|
#ifdef PROTOBUF_FORCE_COPY_DEFAULT_STRING
|
|
if (_impl_.xla_dump_hlo_pipeline_re_.IsDefault()) {
|
|
_impl_.xla_dump_hlo_pipeline_re_.Set("", GetArenaForAllocation());
|
|
}
|
|
#endif // PROTOBUF_FORCE_COPY_DEFAULT_STRING
|
|
// @@protoc_insertion_point(field_set_allocated:xla.DebugOptions.xla_dump_hlo_pipeline_re)
|
|
}
|
|
|
|
// bool xla_gpu_strict_conv_algorithm_picker = 156;
|
|
inline void DebugOptions::clear_xla_gpu_strict_conv_algorithm_picker() {
|
|
_impl_.xla_gpu_strict_conv_algorithm_picker_ = false;
|
|
}
|
|
inline bool DebugOptions::_internal_xla_gpu_strict_conv_algorithm_picker() const {
|
|
return _impl_.xla_gpu_strict_conv_algorithm_picker_;
|
|
}
|
|
inline bool DebugOptions::xla_gpu_strict_conv_algorithm_picker() const {
|
|
// @@protoc_insertion_point(field_get:xla.DebugOptions.xla_gpu_strict_conv_algorithm_picker)
|
|
return _internal_xla_gpu_strict_conv_algorithm_picker();
|
|
}
|
|
inline void DebugOptions::_internal_set_xla_gpu_strict_conv_algorithm_picker(bool value) {
|
|
|
|
_impl_.xla_gpu_strict_conv_algorithm_picker_ = value;
|
|
}
|
|
inline void DebugOptions::set_xla_gpu_strict_conv_algorithm_picker(bool value) {
|
|
_internal_set_xla_gpu_strict_conv_algorithm_picker(value);
|
|
// @@protoc_insertion_point(field_set:xla.DebugOptions.xla_gpu_strict_conv_algorithm_picker)
|
|
}
|
|
|
|
// bool xla_gpu_enable_xla_runtime_executable = 169;
|
|
inline void DebugOptions::clear_xla_gpu_enable_xla_runtime_executable() {
|
|
_impl_.xla_gpu_enable_xla_runtime_executable_ = false;
|
|
}
|
|
inline bool DebugOptions::_internal_xla_gpu_enable_xla_runtime_executable() const {
|
|
return _impl_.xla_gpu_enable_xla_runtime_executable_;
|
|
}
|
|
inline bool DebugOptions::xla_gpu_enable_xla_runtime_executable() const {
|
|
// @@protoc_insertion_point(field_get:xla.DebugOptions.xla_gpu_enable_xla_runtime_executable)
|
|
return _internal_xla_gpu_enable_xla_runtime_executable();
|
|
}
|
|
inline void DebugOptions::_internal_set_xla_gpu_enable_xla_runtime_executable(bool value) {
|
|
|
|
_impl_.xla_gpu_enable_xla_runtime_executable_ = value;
|
|
}
|
|
inline void DebugOptions::set_xla_gpu_enable_xla_runtime_executable(bool value) {
|
|
_internal_set_xla_gpu_enable_xla_runtime_executable(value);
|
|
// @@protoc_insertion_point(field_set:xla.DebugOptions.xla_gpu_enable_xla_runtime_executable)
|
|
}
|
|
|
|
// bool xla_gpu_enable_custom_fusions = 263;
|
|
inline void DebugOptions::clear_xla_gpu_enable_custom_fusions() {
|
|
_impl_.xla_gpu_enable_custom_fusions_ = false;
|
|
}
|
|
inline bool DebugOptions::_internal_xla_gpu_enable_custom_fusions() const {
|
|
return _impl_.xla_gpu_enable_custom_fusions_;
|
|
}
|
|
inline bool DebugOptions::xla_gpu_enable_custom_fusions() const {
|
|
// @@protoc_insertion_point(field_get:xla.DebugOptions.xla_gpu_enable_custom_fusions)
|
|
return _internal_xla_gpu_enable_custom_fusions();
|
|
}
|
|
inline void DebugOptions::_internal_set_xla_gpu_enable_custom_fusions(bool value) {
|
|
|
|
_impl_.xla_gpu_enable_custom_fusions_ = value;
|
|
}
|
|
inline void DebugOptions::set_xla_gpu_enable_custom_fusions(bool value) {
|
|
_internal_set_xla_gpu_enable_custom_fusions(value);
|
|
// @@protoc_insertion_point(field_set:xla.DebugOptions.xla_gpu_enable_custom_fusions)
|
|
}
|
|
|
|
// string xla_gpu_enable_custom_fusions_re = 264;
|
|
inline void DebugOptions::clear_xla_gpu_enable_custom_fusions_re() {
|
|
_impl_.xla_gpu_enable_custom_fusions_re_.ClearToEmpty();
|
|
}
|
|
inline const std::string& DebugOptions::xla_gpu_enable_custom_fusions_re() const {
|
|
// @@protoc_insertion_point(field_get:xla.DebugOptions.xla_gpu_enable_custom_fusions_re)
|
|
return _internal_xla_gpu_enable_custom_fusions_re();
|
|
}
|
|
template <typename ArgT0, typename... ArgT>
|
|
inline PROTOBUF_ALWAYS_INLINE
|
|
void DebugOptions::set_xla_gpu_enable_custom_fusions_re(ArgT0&& arg0, ArgT... args) {
|
|
|
|
_impl_.xla_gpu_enable_custom_fusions_re_.Set(static_cast<ArgT0 &&>(arg0), args..., GetArenaForAllocation());
|
|
// @@protoc_insertion_point(field_set:xla.DebugOptions.xla_gpu_enable_custom_fusions_re)
|
|
}
|
|
inline std::string* DebugOptions::mutable_xla_gpu_enable_custom_fusions_re() {
|
|
std::string* _s = _internal_mutable_xla_gpu_enable_custom_fusions_re();
|
|
// @@protoc_insertion_point(field_mutable:xla.DebugOptions.xla_gpu_enable_custom_fusions_re)
|
|
return _s;
|
|
}
|
|
inline const std::string& DebugOptions::_internal_xla_gpu_enable_custom_fusions_re() const {
|
|
return _impl_.xla_gpu_enable_custom_fusions_re_.Get();
|
|
}
|
|
inline void DebugOptions::_internal_set_xla_gpu_enable_custom_fusions_re(const std::string& value) {
|
|
|
|
_impl_.xla_gpu_enable_custom_fusions_re_.Set(value, GetArenaForAllocation());
|
|
}
|
|
inline std::string* DebugOptions::_internal_mutable_xla_gpu_enable_custom_fusions_re() {
|
|
|
|
return _impl_.xla_gpu_enable_custom_fusions_re_.Mutable(GetArenaForAllocation());
|
|
}
|
|
inline std::string* DebugOptions::release_xla_gpu_enable_custom_fusions_re() {
|
|
// @@protoc_insertion_point(field_release:xla.DebugOptions.xla_gpu_enable_custom_fusions_re)
|
|
return _impl_.xla_gpu_enable_custom_fusions_re_.Release();
|
|
}
|
|
inline void DebugOptions::set_allocated_xla_gpu_enable_custom_fusions_re(std::string* xla_gpu_enable_custom_fusions_re) {
|
|
if (xla_gpu_enable_custom_fusions_re != nullptr) {
|
|
|
|
} else {
|
|
|
|
}
|
|
_impl_.xla_gpu_enable_custom_fusions_re_.SetAllocated(xla_gpu_enable_custom_fusions_re, GetArenaForAllocation());
|
|
#ifdef PROTOBUF_FORCE_COPY_DEFAULT_STRING
|
|
if (_impl_.xla_gpu_enable_custom_fusions_re_.IsDefault()) {
|
|
_impl_.xla_gpu_enable_custom_fusions_re_.Set("", GetArenaForAllocation());
|
|
}
|
|
#endif // PROTOBUF_FORCE_COPY_DEFAULT_STRING
|
|
// @@protoc_insertion_point(field_set_allocated:xla.DebugOptions.xla_gpu_enable_custom_fusions_re)
|
|
}
|
|
|
|
// bool xla_gpu_enable_address_computation_fusion = 105;
|
|
inline void DebugOptions::clear_xla_gpu_enable_address_computation_fusion() {
|
|
_impl_.xla_gpu_enable_address_computation_fusion_ = false;
|
|
}
|
|
inline bool DebugOptions::_internal_xla_gpu_enable_address_computation_fusion() const {
|
|
return _impl_.xla_gpu_enable_address_computation_fusion_;
|
|
}
|
|
inline bool DebugOptions::xla_gpu_enable_address_computation_fusion() const {
|
|
// @@protoc_insertion_point(field_get:xla.DebugOptions.xla_gpu_enable_address_computation_fusion)
|
|
return _internal_xla_gpu_enable_address_computation_fusion();
|
|
}
|
|
inline void DebugOptions::_internal_set_xla_gpu_enable_address_computation_fusion(bool value) {
|
|
|
|
_impl_.xla_gpu_enable_address_computation_fusion_ = value;
|
|
}
|
|
inline void DebugOptions::set_xla_gpu_enable_address_computation_fusion(bool value) {
|
|
_internal_set_xla_gpu_enable_address_computation_fusion(value);
|
|
// @@protoc_insertion_point(field_set:xla.DebugOptions.xla_gpu_enable_address_computation_fusion)
|
|
}
|
|
|
|
// int64 xla_gpu_nccl_termination_timeout_seconds = 163;
|
|
inline void DebugOptions::clear_xla_gpu_nccl_termination_timeout_seconds() {
|
|
_impl_.xla_gpu_nccl_termination_timeout_seconds_ = int64_t{0};
|
|
}
|
|
inline int64_t DebugOptions::_internal_xla_gpu_nccl_termination_timeout_seconds() const {
|
|
return _impl_.xla_gpu_nccl_termination_timeout_seconds_;
|
|
}
|
|
inline int64_t DebugOptions::xla_gpu_nccl_termination_timeout_seconds() const {
|
|
// @@protoc_insertion_point(field_get:xla.DebugOptions.xla_gpu_nccl_termination_timeout_seconds)
|
|
return _internal_xla_gpu_nccl_termination_timeout_seconds();
|
|
}
|
|
inline void DebugOptions::_internal_set_xla_gpu_nccl_termination_timeout_seconds(int64_t value) {
|
|
|
|
_impl_.xla_gpu_nccl_termination_timeout_seconds_ = value;
|
|
}
|
|
inline void DebugOptions::set_xla_gpu_nccl_termination_timeout_seconds(int64_t value) {
|
|
_internal_set_xla_gpu_nccl_termination_timeout_seconds(value);
|
|
// @@protoc_insertion_point(field_set:xla.DebugOptions.xla_gpu_nccl_termination_timeout_seconds)
|
|
}
|
|
|
|
// bool xla_gpu_enable_shared_constants = 165;
|
|
inline void DebugOptions::clear_xla_gpu_enable_shared_constants() {
|
|
_impl_.xla_gpu_enable_shared_constants_ = false;
|
|
}
|
|
inline bool DebugOptions::_internal_xla_gpu_enable_shared_constants() const {
|
|
return _impl_.xla_gpu_enable_shared_constants_;
|
|
}
|
|
inline bool DebugOptions::xla_gpu_enable_shared_constants() const {
|
|
// @@protoc_insertion_point(field_get:xla.DebugOptions.xla_gpu_enable_shared_constants)
|
|
return _internal_xla_gpu_enable_shared_constants();
|
|
}
|
|
inline void DebugOptions::_internal_set_xla_gpu_enable_shared_constants(bool value) {
|
|
|
|
_impl_.xla_gpu_enable_shared_constants_ = value;
|
|
}
|
|
inline void DebugOptions::set_xla_gpu_enable_shared_constants(bool value) {
|
|
_internal_set_xla_gpu_enable_shared_constants(value);
|
|
// @@protoc_insertion_point(field_set:xla.DebugOptions.xla_gpu_enable_shared_constants)
|
|
}
|
|
|
|
// bool xla_gpu_enable_cublaslt = 166;
|
|
inline void DebugOptions::clear_xla_gpu_enable_cublaslt() {
|
|
_impl_.xla_gpu_enable_cublaslt_ = false;
|
|
}
|
|
inline bool DebugOptions::_internal_xla_gpu_enable_cublaslt() const {
|
|
return _impl_.xla_gpu_enable_cublaslt_;
|
|
}
|
|
inline bool DebugOptions::xla_gpu_enable_cublaslt() const {
|
|
// @@protoc_insertion_point(field_get:xla.DebugOptions.xla_gpu_enable_cublaslt)
|
|
return _internal_xla_gpu_enable_cublaslt();
|
|
}
|
|
inline void DebugOptions::_internal_set_xla_gpu_enable_cublaslt(bool value) {
|
|
|
|
_impl_.xla_gpu_enable_cublaslt_ = value;
|
|
}
|
|
inline void DebugOptions::set_xla_gpu_enable_cublaslt(bool value) {
|
|
_internal_set_xla_gpu_enable_cublaslt(value);
|
|
// @@protoc_insertion_point(field_set:xla.DebugOptions.xla_gpu_enable_cublaslt)
|
|
}
|
|
|
|
// repeated .xla.DebugOptions.CommandBufferCmdType xla_gpu_enable_command_buffer = 258;
|
|
inline int DebugOptions::_internal_xla_gpu_enable_command_buffer_size() const {
|
|
return _impl_.xla_gpu_enable_command_buffer_.size();
|
|
}
|
|
inline int DebugOptions::xla_gpu_enable_command_buffer_size() const {
|
|
return _internal_xla_gpu_enable_command_buffer_size();
|
|
}
|
|
inline void DebugOptions::clear_xla_gpu_enable_command_buffer() {
|
|
_impl_.xla_gpu_enable_command_buffer_.Clear();
|
|
}
|
|
inline ::xla::DebugOptions_CommandBufferCmdType DebugOptions::_internal_xla_gpu_enable_command_buffer(int index) const {
|
|
return static_cast< ::xla::DebugOptions_CommandBufferCmdType >(_impl_.xla_gpu_enable_command_buffer_.Get(index));
|
|
}
|
|
inline ::xla::DebugOptions_CommandBufferCmdType DebugOptions::xla_gpu_enable_command_buffer(int index) const {
|
|
// @@protoc_insertion_point(field_get:xla.DebugOptions.xla_gpu_enable_command_buffer)
|
|
return _internal_xla_gpu_enable_command_buffer(index);
|
|
}
|
|
inline void DebugOptions::set_xla_gpu_enable_command_buffer(int index, ::xla::DebugOptions_CommandBufferCmdType value) {
|
|
_impl_.xla_gpu_enable_command_buffer_.Set(index, value);
|
|
// @@protoc_insertion_point(field_set:xla.DebugOptions.xla_gpu_enable_command_buffer)
|
|
}
|
|
inline void DebugOptions::_internal_add_xla_gpu_enable_command_buffer(::xla::DebugOptions_CommandBufferCmdType value) {
|
|
_impl_.xla_gpu_enable_command_buffer_.Add(value);
|
|
}
|
|
inline void DebugOptions::add_xla_gpu_enable_command_buffer(::xla::DebugOptions_CommandBufferCmdType value) {
|
|
_internal_add_xla_gpu_enable_command_buffer(value);
|
|
// @@protoc_insertion_point(field_add:xla.DebugOptions.xla_gpu_enable_command_buffer)
|
|
}
|
|
inline const ::PROTOBUF_NAMESPACE_ID::RepeatedField<int>&
|
|
DebugOptions::xla_gpu_enable_command_buffer() const {
|
|
// @@protoc_insertion_point(field_list:xla.DebugOptions.xla_gpu_enable_command_buffer)
|
|
return _impl_.xla_gpu_enable_command_buffer_;
|
|
}
|
|
inline ::PROTOBUF_NAMESPACE_ID::RepeatedField<int>*
|
|
DebugOptions::_internal_mutable_xla_gpu_enable_command_buffer() {
|
|
return &_impl_.xla_gpu_enable_command_buffer_;
|
|
}
|
|
inline ::PROTOBUF_NAMESPACE_ID::RepeatedField<int>*
|
|
DebugOptions::mutable_xla_gpu_enable_command_buffer() {
|
|
// @@protoc_insertion_point(field_mutable_list:xla.DebugOptions.xla_gpu_enable_command_buffer)
|
|
return _internal_mutable_xla_gpu_enable_command_buffer();
|
|
}
|
|
|
|
// int32 xla_gpu_graph_num_runs_to_instantiate = 202;
|
|
inline void DebugOptions::clear_xla_gpu_graph_num_runs_to_instantiate() {
|
|
_impl_.xla_gpu_graph_num_runs_to_instantiate_ = 0;
|
|
}
|
|
inline int32_t DebugOptions::_internal_xla_gpu_graph_num_runs_to_instantiate() const {
|
|
return _impl_.xla_gpu_graph_num_runs_to_instantiate_;
|
|
}
|
|
inline int32_t DebugOptions::xla_gpu_graph_num_runs_to_instantiate() const {
|
|
// @@protoc_insertion_point(field_get:xla.DebugOptions.xla_gpu_graph_num_runs_to_instantiate)
|
|
return _internal_xla_gpu_graph_num_runs_to_instantiate();
|
|
}
|
|
inline void DebugOptions::_internal_set_xla_gpu_graph_num_runs_to_instantiate(int32_t value) {
|
|
|
|
_impl_.xla_gpu_graph_num_runs_to_instantiate_ = value;
|
|
}
|
|
inline void DebugOptions::set_xla_gpu_graph_num_runs_to_instantiate(int32_t value) {
|
|
_internal_set_xla_gpu_graph_num_runs_to_instantiate(value);
|
|
// @@protoc_insertion_point(field_set:xla.DebugOptions.xla_gpu_graph_num_runs_to_instantiate)
|
|
}
|
|
|
|
// int32 xla_gpu_graph_min_graph_size = 208;
|
|
inline void DebugOptions::clear_xla_gpu_graph_min_graph_size() {
|
|
_impl_.xla_gpu_graph_min_graph_size_ = 0;
|
|
}
|
|
inline int32_t DebugOptions::_internal_xla_gpu_graph_min_graph_size() const {
|
|
return _impl_.xla_gpu_graph_min_graph_size_;
|
|
}
|
|
inline int32_t DebugOptions::xla_gpu_graph_min_graph_size() const {
|
|
// @@protoc_insertion_point(field_get:xla.DebugOptions.xla_gpu_graph_min_graph_size)
|
|
return _internal_xla_gpu_graph_min_graph_size();
|
|
}
|
|
inline void DebugOptions::_internal_set_xla_gpu_graph_min_graph_size(int32_t value) {
|
|
|
|
_impl_.xla_gpu_graph_min_graph_size_ = value;
|
|
}
|
|
inline void DebugOptions::set_xla_gpu_graph_min_graph_size(int32_t value) {
|
|
_internal_set_xla_gpu_graph_min_graph_size(value);
|
|
// @@protoc_insertion_point(field_set:xla.DebugOptions.xla_gpu_graph_min_graph_size)
|
|
}
|
|
|
|
// bool xla_gpu_graph_enable_concurrent_region = 215;
|
|
inline void DebugOptions::clear_xla_gpu_graph_enable_concurrent_region() {
|
|
_impl_.xla_gpu_graph_enable_concurrent_region_ = false;
|
|
}
|
|
inline bool DebugOptions::_internal_xla_gpu_graph_enable_concurrent_region() const {
|
|
return _impl_.xla_gpu_graph_enable_concurrent_region_;
|
|
}
|
|
inline bool DebugOptions::xla_gpu_graph_enable_concurrent_region() const {
|
|
// @@protoc_insertion_point(field_get:xla.DebugOptions.xla_gpu_graph_enable_concurrent_region)
|
|
return _internal_xla_gpu_graph_enable_concurrent_region();
|
|
}
|
|
inline void DebugOptions::_internal_set_xla_gpu_graph_enable_concurrent_region(bool value) {
|
|
|
|
_impl_.xla_gpu_graph_enable_concurrent_region_ = value;
|
|
}
|
|
inline void DebugOptions::set_xla_gpu_graph_enable_concurrent_region(bool value) {
|
|
_internal_set_xla_gpu_graph_enable_concurrent_region(value);
|
|
// @@protoc_insertion_point(field_set:xla.DebugOptions.xla_gpu_graph_enable_concurrent_region)
|
|
}
|
|
|
|
// int32 xla_gpu_graph_eviction_timeout_seconds = 230;
|
|
inline void DebugOptions::clear_xla_gpu_graph_eviction_timeout_seconds() {
|
|
_impl_.xla_gpu_graph_eviction_timeout_seconds_ = 0;
|
|
}
|
|
inline int32_t DebugOptions::_internal_xla_gpu_graph_eviction_timeout_seconds() const {
|
|
return _impl_.xla_gpu_graph_eviction_timeout_seconds_;
|
|
}
|
|
inline int32_t DebugOptions::xla_gpu_graph_eviction_timeout_seconds() const {
|
|
// @@protoc_insertion_point(field_get:xla.DebugOptions.xla_gpu_graph_eviction_timeout_seconds)
|
|
return _internal_xla_gpu_graph_eviction_timeout_seconds();
|
|
}
|
|
inline void DebugOptions::_internal_set_xla_gpu_graph_eviction_timeout_seconds(int32_t value) {
|
|
|
|
_impl_.xla_gpu_graph_eviction_timeout_seconds_ = value;
|
|
}
|
|
inline void DebugOptions::set_xla_gpu_graph_eviction_timeout_seconds(int32_t value) {
|
|
_internal_set_xla_gpu_graph_eviction_timeout_seconds(value);
|
|
// @@protoc_insertion_point(field_set:xla.DebugOptions.xla_gpu_graph_eviction_timeout_seconds)
|
|
}
|
|
|
|
// int64 xla_gpu_redzone_scratch_max_megabytes = 167;
|
|
inline void DebugOptions::clear_xla_gpu_redzone_scratch_max_megabytes() {
|
|
_impl_.xla_gpu_redzone_scratch_max_megabytes_ = int64_t{0};
|
|
}
|
|
inline int64_t DebugOptions::_internal_xla_gpu_redzone_scratch_max_megabytes() const {
|
|
return _impl_.xla_gpu_redzone_scratch_max_megabytes_;
|
|
}
|
|
inline int64_t DebugOptions::xla_gpu_redzone_scratch_max_megabytes() const {
|
|
// @@protoc_insertion_point(field_get:xla.DebugOptions.xla_gpu_redzone_scratch_max_megabytes)
|
|
return _internal_xla_gpu_redzone_scratch_max_megabytes();
|
|
}
|
|
inline void DebugOptions::_internal_set_xla_gpu_redzone_scratch_max_megabytes(int64_t value) {
|
|
|
|
_impl_.xla_gpu_redzone_scratch_max_megabytes_ = value;
|
|
}
|
|
inline void DebugOptions::set_xla_gpu_redzone_scratch_max_megabytes(int64_t value) {
|
|
_internal_set_xla_gpu_redzone_scratch_max_megabytes(value);
|
|
// @@protoc_insertion_point(field_set:xla.DebugOptions.xla_gpu_redzone_scratch_max_megabytes)
|
|
}
|
|
|
|
// int64 xla_gpu_redzone_padding_bytes = 228;
|
|
inline void DebugOptions::clear_xla_gpu_redzone_padding_bytes() {
|
|
_impl_.xla_gpu_redzone_padding_bytes_ = int64_t{0};
|
|
}
|
|
inline int64_t DebugOptions::_internal_xla_gpu_redzone_padding_bytes() const {
|
|
return _impl_.xla_gpu_redzone_padding_bytes_;
|
|
}
|
|
inline int64_t DebugOptions::xla_gpu_redzone_padding_bytes() const {
|
|
// @@protoc_insertion_point(field_get:xla.DebugOptions.xla_gpu_redzone_padding_bytes)
|
|
return _internal_xla_gpu_redzone_padding_bytes();
|
|
}
|
|
inline void DebugOptions::_internal_set_xla_gpu_redzone_padding_bytes(int64_t value) {
|
|
|
|
_impl_.xla_gpu_redzone_padding_bytes_ = value;
|
|
}
|
|
inline void DebugOptions::set_xla_gpu_redzone_padding_bytes(int64_t value) {
|
|
_internal_set_xla_gpu_redzone_padding_bytes(value);
|
|
// @@protoc_insertion_point(field_set:xla.DebugOptions.xla_gpu_redzone_padding_bytes)
|
|
}
|
|
|
|
// bool xla_gpu_simplify_all_fp_conversions = 168;
|
|
inline void DebugOptions::clear_xla_gpu_simplify_all_fp_conversions() {
|
|
_impl_.xla_gpu_simplify_all_fp_conversions_ = false;
|
|
}
|
|
inline bool DebugOptions::_internal_xla_gpu_simplify_all_fp_conversions() const {
|
|
return _impl_.xla_gpu_simplify_all_fp_conversions_;
|
|
}
|
|
inline bool DebugOptions::xla_gpu_simplify_all_fp_conversions() const {
|
|
// @@protoc_insertion_point(field_get:xla.DebugOptions.xla_gpu_simplify_all_fp_conversions)
|
|
return _internal_xla_gpu_simplify_all_fp_conversions();
|
|
}
|
|
inline void DebugOptions::_internal_set_xla_gpu_simplify_all_fp_conversions(bool value) {
|
|
|
|
_impl_.xla_gpu_simplify_all_fp_conversions_ = value;
|
|
}
|
|
inline void DebugOptions::set_xla_gpu_simplify_all_fp_conversions(bool value) {
|
|
_internal_set_xla_gpu_simplify_all_fp_conversions(value);
|
|
// @@protoc_insertion_point(field_set:xla.DebugOptions.xla_gpu_simplify_all_fp_conversions)
|
|
}
|
|
|
|
// bool xla_gpu_normalize_layouts = 172;
|
|
inline void DebugOptions::clear_xla_gpu_normalize_layouts() {
|
|
_impl_.xla_gpu_normalize_layouts_ = false;
|
|
}
|
|
inline bool DebugOptions::_internal_xla_gpu_normalize_layouts() const {
|
|
return _impl_.xla_gpu_normalize_layouts_;
|
|
}
|
|
inline bool DebugOptions::xla_gpu_normalize_layouts() const {
|
|
// @@protoc_insertion_point(field_get:xla.DebugOptions.xla_gpu_normalize_layouts)
|
|
return _internal_xla_gpu_normalize_layouts();
|
|
}
|
|
inline void DebugOptions::_internal_set_xla_gpu_normalize_layouts(bool value) {
|
|
|
|
_impl_.xla_gpu_normalize_layouts_ = value;
|
|
}
|
|
inline void DebugOptions::set_xla_gpu_normalize_layouts(bool value) {
|
|
_internal_set_xla_gpu_normalize_layouts(value);
|
|
// @@protoc_insertion_point(field_set:xla.DebugOptions.xla_gpu_normalize_layouts)
|
|
}
|
|
|
|
// bool xla_cpu_use_acl = 174;
|
|
inline void DebugOptions::clear_xla_cpu_use_acl() {
|
|
_impl_.xla_cpu_use_acl_ = false;
|
|
}
|
|
inline bool DebugOptions::_internal_xla_cpu_use_acl() const {
|
|
return _impl_.xla_cpu_use_acl_;
|
|
}
|
|
inline bool DebugOptions::xla_cpu_use_acl() const {
|
|
// @@protoc_insertion_point(field_get:xla.DebugOptions.xla_cpu_use_acl)
|
|
return _internal_xla_cpu_use_acl();
|
|
}
|
|
inline void DebugOptions::_internal_set_xla_cpu_use_acl(bool value) {
|
|
|
|
_impl_.xla_cpu_use_acl_ = value;
|
|
}
|
|
inline void DebugOptions::set_xla_cpu_use_acl(bool value) {
|
|
_internal_set_xla_cpu_use_acl(value);
|
|
// @@protoc_insertion_point(field_set:xla.DebugOptions.xla_cpu_use_acl)
|
|
}
|
|
|
|
// bool xla_cpu_strict_dot_conv_math = 175;
|
|
inline void DebugOptions::clear_xla_cpu_strict_dot_conv_math() {
|
|
_impl_.xla_cpu_strict_dot_conv_math_ = false;
|
|
}
|
|
inline bool DebugOptions::_internal_xla_cpu_strict_dot_conv_math() const {
|
|
return _impl_.xla_cpu_strict_dot_conv_math_;
|
|
}
|
|
inline bool DebugOptions::xla_cpu_strict_dot_conv_math() const {
|
|
// @@protoc_insertion_point(field_get:xla.DebugOptions.xla_cpu_strict_dot_conv_math)
|
|
return _internal_xla_cpu_strict_dot_conv_math();
|
|
}
|
|
inline void DebugOptions::_internal_set_xla_cpu_strict_dot_conv_math(bool value) {
|
|
|
|
_impl_.xla_cpu_strict_dot_conv_math_ = value;
|
|
}
|
|
inline void DebugOptions::set_xla_cpu_strict_dot_conv_math(bool value) {
|
|
_internal_set_xla_cpu_strict_dot_conv_math(value);
|
|
// @@protoc_insertion_point(field_set:xla.DebugOptions.xla_cpu_strict_dot_conv_math)
|
|
}
|
|
|
|
// bool xla_gpu_use_runtime_fusion = 181;
|
|
inline void DebugOptions::clear_xla_gpu_use_runtime_fusion() {
|
|
_impl_.xla_gpu_use_runtime_fusion_ = false;
|
|
}
|
|
inline bool DebugOptions::_internal_xla_gpu_use_runtime_fusion() const {
|
|
return _impl_.xla_gpu_use_runtime_fusion_;
|
|
}
|
|
inline bool DebugOptions::xla_gpu_use_runtime_fusion() const {
|
|
// @@protoc_insertion_point(field_get:xla.DebugOptions.xla_gpu_use_runtime_fusion)
|
|
return _internal_xla_gpu_use_runtime_fusion();
|
|
}
|
|
inline void DebugOptions::_internal_set_xla_gpu_use_runtime_fusion(bool value) {
|
|
|
|
_impl_.xla_gpu_use_runtime_fusion_ = value;
|
|
}
|
|
inline void DebugOptions::set_xla_gpu_use_runtime_fusion(bool value) {
|
|
_internal_set_xla_gpu_use_runtime_fusion(value);
|
|
// @@protoc_insertion_point(field_set:xla.DebugOptions.xla_gpu_use_runtime_fusion)
|
|
}
|
|
|
|
// bool xla_dump_latency_hiding_schedule = 182;
|
|
inline void DebugOptions::clear_xla_dump_latency_hiding_schedule() {
|
|
_impl_.xla_dump_latency_hiding_schedule_ = false;
|
|
}
|
|
inline bool DebugOptions::_internal_xla_dump_latency_hiding_schedule() const {
|
|
return _impl_.xla_dump_latency_hiding_schedule_;
|
|
}
|
|
inline bool DebugOptions::xla_dump_latency_hiding_schedule() const {
|
|
// @@protoc_insertion_point(field_get:xla.DebugOptions.xla_dump_latency_hiding_schedule)
|
|
return _internal_xla_dump_latency_hiding_schedule();
|
|
}
|
|
inline void DebugOptions::_internal_set_xla_dump_latency_hiding_schedule(bool value) {
|
|
|
|
_impl_.xla_dump_latency_hiding_schedule_ = value;
|
|
}
|
|
inline void DebugOptions::set_xla_dump_latency_hiding_schedule(bool value) {
|
|
_internal_set_xla_dump_latency_hiding_schedule(value);
|
|
// @@protoc_insertion_point(field_set:xla.DebugOptions.xla_dump_latency_hiding_schedule)
|
|
}
|
|
|
|
// bool xla_cpu_enable_mlir_tiling_and_fusion = 184;
|
|
inline void DebugOptions::clear_xla_cpu_enable_mlir_tiling_and_fusion() {
|
|
_impl_.xla_cpu_enable_mlir_tiling_and_fusion_ = false;
|
|
}
|
|
inline bool DebugOptions::_internal_xla_cpu_enable_mlir_tiling_and_fusion() const {
|
|
return _impl_.xla_cpu_enable_mlir_tiling_and_fusion_;
|
|
}
|
|
inline bool DebugOptions::xla_cpu_enable_mlir_tiling_and_fusion() const {
|
|
// @@protoc_insertion_point(field_get:xla.DebugOptions.xla_cpu_enable_mlir_tiling_and_fusion)
|
|
return _internal_xla_cpu_enable_mlir_tiling_and_fusion();
|
|
}
|
|
inline void DebugOptions::_internal_set_xla_cpu_enable_mlir_tiling_and_fusion(bool value) {
|
|
|
|
_impl_.xla_cpu_enable_mlir_tiling_and_fusion_ = value;
|
|
}
|
|
inline void DebugOptions::set_xla_cpu_enable_mlir_tiling_and_fusion(bool value) {
|
|
_internal_set_xla_cpu_enable_mlir_tiling_and_fusion(value);
|
|
// @@protoc_insertion_point(field_set:xla.DebugOptions.xla_cpu_enable_mlir_tiling_and_fusion)
|
|
}
|
|
|
|
// bool xla_cpu_enable_custom_matmul_tiling = 195;
|
|
inline void DebugOptions::clear_xla_cpu_enable_custom_matmul_tiling() {
|
|
_impl_.xla_cpu_enable_custom_matmul_tiling_ = false;
|
|
}
|
|
inline bool DebugOptions::_internal_xla_cpu_enable_custom_matmul_tiling() const {
|
|
return _impl_.xla_cpu_enable_custom_matmul_tiling_;
|
|
}
|
|
inline bool DebugOptions::xla_cpu_enable_custom_matmul_tiling() const {
|
|
// @@protoc_insertion_point(field_get:xla.DebugOptions.xla_cpu_enable_custom_matmul_tiling)
|
|
return _internal_xla_cpu_enable_custom_matmul_tiling();
|
|
}
|
|
inline void DebugOptions::_internal_set_xla_cpu_enable_custom_matmul_tiling(bool value) {
|
|
|
|
_impl_.xla_cpu_enable_custom_matmul_tiling_ = value;
|
|
}
|
|
inline void DebugOptions::set_xla_cpu_enable_custom_matmul_tiling(bool value) {
|
|
_internal_set_xla_cpu_enable_custom_matmul_tiling(value);
|
|
// @@protoc_insertion_point(field_set:xla.DebugOptions.xla_cpu_enable_custom_matmul_tiling)
|
|
}
|
|
|
|
// int64 xla_cpu_matmul_tiling_m_dim = 196;
|
|
inline void DebugOptions::clear_xla_cpu_matmul_tiling_m_dim() {
|
|
_impl_.xla_cpu_matmul_tiling_m_dim_ = int64_t{0};
|
|
}
|
|
inline int64_t DebugOptions::_internal_xla_cpu_matmul_tiling_m_dim() const {
|
|
return _impl_.xla_cpu_matmul_tiling_m_dim_;
|
|
}
|
|
inline int64_t DebugOptions::xla_cpu_matmul_tiling_m_dim() const {
|
|
// @@protoc_insertion_point(field_get:xla.DebugOptions.xla_cpu_matmul_tiling_m_dim)
|
|
return _internal_xla_cpu_matmul_tiling_m_dim();
|
|
}
|
|
inline void DebugOptions::_internal_set_xla_cpu_matmul_tiling_m_dim(int64_t value) {
|
|
|
|
_impl_.xla_cpu_matmul_tiling_m_dim_ = value;
|
|
}
|
|
inline void DebugOptions::set_xla_cpu_matmul_tiling_m_dim(int64_t value) {
|
|
_internal_set_xla_cpu_matmul_tiling_m_dim(value);
|
|
// @@protoc_insertion_point(field_set:xla.DebugOptions.xla_cpu_matmul_tiling_m_dim)
|
|
}
|
|
|
|
// int64 xla_cpu_matmul_tiling_n_dim = 197;
|
|
inline void DebugOptions::clear_xla_cpu_matmul_tiling_n_dim() {
|
|
_impl_.xla_cpu_matmul_tiling_n_dim_ = int64_t{0};
|
|
}
|
|
inline int64_t DebugOptions::_internal_xla_cpu_matmul_tiling_n_dim() const {
|
|
return _impl_.xla_cpu_matmul_tiling_n_dim_;
|
|
}
|
|
inline int64_t DebugOptions::xla_cpu_matmul_tiling_n_dim() const {
|
|
// @@protoc_insertion_point(field_get:xla.DebugOptions.xla_cpu_matmul_tiling_n_dim)
|
|
return _internal_xla_cpu_matmul_tiling_n_dim();
|
|
}
|
|
inline void DebugOptions::_internal_set_xla_cpu_matmul_tiling_n_dim(int64_t value) {
|
|
|
|
_impl_.xla_cpu_matmul_tiling_n_dim_ = value;
|
|
}
|
|
inline void DebugOptions::set_xla_cpu_matmul_tiling_n_dim(int64_t value) {
|
|
_internal_set_xla_cpu_matmul_tiling_n_dim(value);
|
|
// @@protoc_insertion_point(field_set:xla.DebugOptions.xla_cpu_matmul_tiling_n_dim)
|
|
}
|
|
|
|
// int64 xla_cpu_matmul_tiling_k_dim = 198;
|
|
inline void DebugOptions::clear_xla_cpu_matmul_tiling_k_dim() {
|
|
_impl_.xla_cpu_matmul_tiling_k_dim_ = int64_t{0};
|
|
}
|
|
inline int64_t DebugOptions::_internal_xla_cpu_matmul_tiling_k_dim() const {
|
|
return _impl_.xla_cpu_matmul_tiling_k_dim_;
|
|
}
|
|
inline int64_t DebugOptions::xla_cpu_matmul_tiling_k_dim() const {
|
|
// @@protoc_insertion_point(field_get:xla.DebugOptions.xla_cpu_matmul_tiling_k_dim)
|
|
return _internal_xla_cpu_matmul_tiling_k_dim();
|
|
}
|
|
inline void DebugOptions::_internal_set_xla_cpu_matmul_tiling_k_dim(int64_t value) {
|
|
|
|
_impl_.xla_cpu_matmul_tiling_k_dim_ = value;
|
|
}
|
|
inline void DebugOptions::set_xla_cpu_matmul_tiling_k_dim(int64_t value) {
|
|
_internal_set_xla_cpu_matmul_tiling_k_dim(value);
|
|
// @@protoc_insertion_point(field_set:xla.DebugOptions.xla_cpu_matmul_tiling_k_dim)
|
|
}
|
|
|
|
// bool xla_cpu_enable_mlir_fusion_outlining = 192;
|
|
inline void DebugOptions::clear_xla_cpu_enable_mlir_fusion_outlining() {
|
|
_impl_.xla_cpu_enable_mlir_fusion_outlining_ = false;
|
|
}
|
|
inline bool DebugOptions::_internal_xla_cpu_enable_mlir_fusion_outlining() const {
|
|
return _impl_.xla_cpu_enable_mlir_fusion_outlining_;
|
|
}
|
|
inline bool DebugOptions::xla_cpu_enable_mlir_fusion_outlining() const {
|
|
// @@protoc_insertion_point(field_get:xla.DebugOptions.xla_cpu_enable_mlir_fusion_outlining)
|
|
return _internal_xla_cpu_enable_mlir_fusion_outlining();
|
|
}
|
|
inline void DebugOptions::_internal_set_xla_cpu_enable_mlir_fusion_outlining(bool value) {
|
|
|
|
_impl_.xla_cpu_enable_mlir_fusion_outlining_ = value;
|
|
}
|
|
inline void DebugOptions::set_xla_cpu_enable_mlir_fusion_outlining(bool value) {
|
|
_internal_set_xla_cpu_enable_mlir_fusion_outlining(value);
|
|
// @@protoc_insertion_point(field_set:xla.DebugOptions.xla_cpu_enable_mlir_fusion_outlining)
|
|
}
|
|
|
|
// bool xla_cpu_enable_experimental_deallocation = 191;
|
|
inline void DebugOptions::clear_xla_cpu_enable_experimental_deallocation() {
|
|
_impl_.xla_cpu_enable_experimental_deallocation_ = false;
|
|
}
|
|
inline bool DebugOptions::_internal_xla_cpu_enable_experimental_deallocation() const {
|
|
return _impl_.xla_cpu_enable_experimental_deallocation_;
|
|
}
|
|
inline bool DebugOptions::xla_cpu_enable_experimental_deallocation() const {
|
|
// @@protoc_insertion_point(field_get:xla.DebugOptions.xla_cpu_enable_experimental_deallocation)
|
|
return _internal_xla_cpu_enable_experimental_deallocation();
|
|
}
|
|
inline void DebugOptions::_internal_set_xla_cpu_enable_experimental_deallocation(bool value) {
|
|
|
|
_impl_.xla_cpu_enable_experimental_deallocation_ = value;
|
|
}
|
|
inline void DebugOptions::set_xla_cpu_enable_experimental_deallocation(bool value) {
|
|
_internal_set_xla_cpu_enable_experimental_deallocation(value);
|
|
// @@protoc_insertion_point(field_set:xla.DebugOptions.xla_cpu_enable_experimental_deallocation)
|
|
}
|
|
|
|
// bool xla_gpu_enable_latency_hiding_scheduler = 186;
|
|
inline void DebugOptions::clear_xla_gpu_enable_latency_hiding_scheduler() {
|
|
_impl_.xla_gpu_enable_latency_hiding_scheduler_ = false;
|
|
}
|
|
inline bool DebugOptions::_internal_xla_gpu_enable_latency_hiding_scheduler() const {
|
|
return _impl_.xla_gpu_enable_latency_hiding_scheduler_;
|
|
}
|
|
inline bool DebugOptions::xla_gpu_enable_latency_hiding_scheduler() const {
|
|
// @@protoc_insertion_point(field_get:xla.DebugOptions.xla_gpu_enable_latency_hiding_scheduler)
|
|
return _internal_xla_gpu_enable_latency_hiding_scheduler();
|
|
}
|
|
inline void DebugOptions::_internal_set_xla_gpu_enable_latency_hiding_scheduler(bool value) {
|
|
|
|
_impl_.xla_gpu_enable_latency_hiding_scheduler_ = value;
|
|
}
|
|
inline void DebugOptions::set_xla_gpu_enable_latency_hiding_scheduler(bool value) {
|
|
_internal_set_xla_gpu_enable_latency_hiding_scheduler(value);
|
|
// @@protoc_insertion_point(field_set:xla.DebugOptions.xla_gpu_enable_latency_hiding_scheduler)
|
|
}
|
|
|
|
// bool xla_gpu_enable_highest_priority_async_stream = 216;
|
|
inline void DebugOptions::clear_xla_gpu_enable_highest_priority_async_stream() {
|
|
_impl_.xla_gpu_enable_highest_priority_async_stream_ = false;
|
|
}
|
|
inline bool DebugOptions::_internal_xla_gpu_enable_highest_priority_async_stream() const {
|
|
return _impl_.xla_gpu_enable_highest_priority_async_stream_;
|
|
}
|
|
inline bool DebugOptions::xla_gpu_enable_highest_priority_async_stream() const {
|
|
// @@protoc_insertion_point(field_get:xla.DebugOptions.xla_gpu_enable_highest_priority_async_stream)
|
|
return _internal_xla_gpu_enable_highest_priority_async_stream();
|
|
}
|
|
inline void DebugOptions::_internal_set_xla_gpu_enable_highest_priority_async_stream(bool value) {
|
|
|
|
_impl_.xla_gpu_enable_highest_priority_async_stream_ = value;
|
|
}
|
|
inline void DebugOptions::set_xla_gpu_enable_highest_priority_async_stream(bool value) {
|
|
_internal_set_xla_gpu_enable_highest_priority_async_stream(value);
|
|
// @@protoc_insertion_point(field_set:xla.DebugOptions.xla_gpu_enable_highest_priority_async_stream)
|
|
}
|
|
|
|
// bool xla_gpu_enable_analytical_latency_estimator = 255;
|
|
inline void DebugOptions::clear_xla_gpu_enable_analytical_latency_estimator() {
|
|
_impl_.xla_gpu_enable_analytical_latency_estimator_ = false;
|
|
}
|
|
inline bool DebugOptions::_internal_xla_gpu_enable_analytical_latency_estimator() const {
|
|
return _impl_.xla_gpu_enable_analytical_latency_estimator_;
|
|
}
|
|
inline bool DebugOptions::xla_gpu_enable_analytical_latency_estimator() const {
|
|
// @@protoc_insertion_point(field_get:xla.DebugOptions.xla_gpu_enable_analytical_latency_estimator)
|
|
return _internal_xla_gpu_enable_analytical_latency_estimator();
|
|
}
|
|
inline void DebugOptions::_internal_set_xla_gpu_enable_analytical_latency_estimator(bool value) {
|
|
|
|
_impl_.xla_gpu_enable_analytical_latency_estimator_ = value;
|
|
}
|
|
inline void DebugOptions::set_xla_gpu_enable_analytical_latency_estimator(bool value) {
|
|
_internal_set_xla_gpu_enable_analytical_latency_estimator(value);
|
|
// @@protoc_insertion_point(field_set:xla.DebugOptions.xla_gpu_enable_analytical_latency_estimator)
|
|
}
|
|
|
|
// bool xla_gpu_lhs_enable_gpu_async_tracker = 204;
|
|
inline void DebugOptions::clear_xla_gpu_lhs_enable_gpu_async_tracker() {
|
|
_impl_.xla_gpu_lhs_enable_gpu_async_tracker_ = false;
|
|
}
|
|
inline bool DebugOptions::_internal_xla_gpu_lhs_enable_gpu_async_tracker() const {
|
|
return _impl_.xla_gpu_lhs_enable_gpu_async_tracker_;
|
|
}
|
|
inline bool DebugOptions::xla_gpu_lhs_enable_gpu_async_tracker() const {
|
|
// @@protoc_insertion_point(field_get:xla.DebugOptions.xla_gpu_lhs_enable_gpu_async_tracker)
|
|
return _internal_xla_gpu_lhs_enable_gpu_async_tracker();
|
|
}
|
|
inline void DebugOptions::_internal_set_xla_gpu_lhs_enable_gpu_async_tracker(bool value) {
|
|
|
|
_impl_.xla_gpu_lhs_enable_gpu_async_tracker_ = value;
|
|
}
|
|
inline void DebugOptions::set_xla_gpu_lhs_enable_gpu_async_tracker(bool value) {
|
|
_internal_set_xla_gpu_lhs_enable_gpu_async_tracker(value);
|
|
// @@protoc_insertion_point(field_set:xla.DebugOptions.xla_gpu_lhs_enable_gpu_async_tracker)
|
|
}
|
|
|
|
// string xla_gpu_pgle_profile_file_or_directory_path = 210;
|
|
inline void DebugOptions::clear_xla_gpu_pgle_profile_file_or_directory_path() {
|
|
_impl_.xla_gpu_pgle_profile_file_or_directory_path_.ClearToEmpty();
|
|
}
|
|
inline const std::string& DebugOptions::xla_gpu_pgle_profile_file_or_directory_path() const {
|
|
// @@protoc_insertion_point(field_get:xla.DebugOptions.xla_gpu_pgle_profile_file_or_directory_path)
|
|
return _internal_xla_gpu_pgle_profile_file_or_directory_path();
|
|
}
|
|
template <typename ArgT0, typename... ArgT>
|
|
inline PROTOBUF_ALWAYS_INLINE
|
|
void DebugOptions::set_xla_gpu_pgle_profile_file_or_directory_path(ArgT0&& arg0, ArgT... args) {
|
|
|
|
_impl_.xla_gpu_pgle_profile_file_or_directory_path_.Set(static_cast<ArgT0 &&>(arg0), args..., GetArenaForAllocation());
|
|
// @@protoc_insertion_point(field_set:xla.DebugOptions.xla_gpu_pgle_profile_file_or_directory_path)
|
|
}
|
|
inline std::string* DebugOptions::mutable_xla_gpu_pgle_profile_file_or_directory_path() {
|
|
std::string* _s = _internal_mutable_xla_gpu_pgle_profile_file_or_directory_path();
|
|
// @@protoc_insertion_point(field_mutable:xla.DebugOptions.xla_gpu_pgle_profile_file_or_directory_path)
|
|
return _s;
|
|
}
|
|
inline const std::string& DebugOptions::_internal_xla_gpu_pgle_profile_file_or_directory_path() const {
|
|
return _impl_.xla_gpu_pgle_profile_file_or_directory_path_.Get();
|
|
}
|
|
inline void DebugOptions::_internal_set_xla_gpu_pgle_profile_file_or_directory_path(const std::string& value) {
|
|
|
|
_impl_.xla_gpu_pgle_profile_file_or_directory_path_.Set(value, GetArenaForAllocation());
|
|
}
|
|
inline std::string* DebugOptions::_internal_mutable_xla_gpu_pgle_profile_file_or_directory_path() {
|
|
|
|
return _impl_.xla_gpu_pgle_profile_file_or_directory_path_.Mutable(GetArenaForAllocation());
|
|
}
|
|
inline std::string* DebugOptions::release_xla_gpu_pgle_profile_file_or_directory_path() {
|
|
// @@protoc_insertion_point(field_release:xla.DebugOptions.xla_gpu_pgle_profile_file_or_directory_path)
|
|
return _impl_.xla_gpu_pgle_profile_file_or_directory_path_.Release();
|
|
}
|
|
inline void DebugOptions::set_allocated_xla_gpu_pgle_profile_file_or_directory_path(std::string* xla_gpu_pgle_profile_file_or_directory_path) {
|
|
if (xla_gpu_pgle_profile_file_or_directory_path != nullptr) {
|
|
|
|
} else {
|
|
|
|
}
|
|
_impl_.xla_gpu_pgle_profile_file_or_directory_path_.SetAllocated(xla_gpu_pgle_profile_file_or_directory_path, GetArenaForAllocation());
|
|
#ifdef PROTOBUF_FORCE_COPY_DEFAULT_STRING
|
|
if (_impl_.xla_gpu_pgle_profile_file_or_directory_path_.IsDefault()) {
|
|
_impl_.xla_gpu_pgle_profile_file_or_directory_path_.Set("", GetArenaForAllocation());
|
|
}
|
|
#endif // PROTOBUF_FORCE_COPY_DEFAULT_STRING
|
|
// @@protoc_insertion_point(field_set_allocated:xla.DebugOptions.xla_gpu_pgle_profile_file_or_directory_path)
|
|
}
|
|
|
|
// int32 xla_gpu_memory_limit_slop_factor = 260;
|
|
inline void DebugOptions::clear_xla_gpu_memory_limit_slop_factor() {
|
|
_impl_.xla_gpu_memory_limit_slop_factor_ = 0;
|
|
}
|
|
inline int32_t DebugOptions::_internal_xla_gpu_memory_limit_slop_factor() const {
|
|
return _impl_.xla_gpu_memory_limit_slop_factor_;
|
|
}
|
|
inline int32_t DebugOptions::xla_gpu_memory_limit_slop_factor() const {
|
|
// @@protoc_insertion_point(field_get:xla.DebugOptions.xla_gpu_memory_limit_slop_factor)
|
|
return _internal_xla_gpu_memory_limit_slop_factor();
|
|
}
|
|
inline void DebugOptions::_internal_set_xla_gpu_memory_limit_slop_factor(int32_t value) {
|
|
|
|
_impl_.xla_gpu_memory_limit_slop_factor_ = value;
|
|
}
|
|
inline void DebugOptions::set_xla_gpu_memory_limit_slop_factor(int32_t value) {
|
|
_internal_set_xla_gpu_memory_limit_slop_factor(value);
|
|
// @@protoc_insertion_point(field_set:xla.DebugOptions.xla_gpu_memory_limit_slop_factor)
|
|
}
|
|
|
|
// bool xla_gpu_enable_pipelined_collectives = 239;
|
|
inline void DebugOptions::clear_xla_gpu_enable_pipelined_collectives() {
|
|
_impl_.xla_gpu_enable_pipelined_collectives_ = false;
|
|
}
|
|
inline bool DebugOptions::_internal_xla_gpu_enable_pipelined_collectives() const {
|
|
return _impl_.xla_gpu_enable_pipelined_collectives_;
|
|
}
|
|
inline bool DebugOptions::xla_gpu_enable_pipelined_collectives() const {
|
|
// @@protoc_insertion_point(field_get:xla.DebugOptions.xla_gpu_enable_pipelined_collectives)
|
|
return _internal_xla_gpu_enable_pipelined_collectives();
|
|
}
|
|
inline void DebugOptions::_internal_set_xla_gpu_enable_pipelined_collectives(bool value) {
|
|
|
|
_impl_.xla_gpu_enable_pipelined_collectives_ = value;
|
|
}
|
|
inline void DebugOptions::set_xla_gpu_enable_pipelined_collectives(bool value) {
|
|
_internal_set_xla_gpu_enable_pipelined_collectives(value);
|
|
// @@protoc_insertion_point(field_set:xla.DebugOptions.xla_gpu_enable_pipelined_collectives)
|
|
}
|
|
|
|
// bool xla_gpu_enable_pipelined_all_reduce = 217;
|
|
inline void DebugOptions::clear_xla_gpu_enable_pipelined_all_reduce() {
|
|
_impl_.xla_gpu_enable_pipelined_all_reduce_ = false;
|
|
}
|
|
inline bool DebugOptions::_internal_xla_gpu_enable_pipelined_all_reduce() const {
|
|
return _impl_.xla_gpu_enable_pipelined_all_reduce_;
|
|
}
|
|
inline bool DebugOptions::xla_gpu_enable_pipelined_all_reduce() const {
|
|
// @@protoc_insertion_point(field_get:xla.DebugOptions.xla_gpu_enable_pipelined_all_reduce)
|
|
return _internal_xla_gpu_enable_pipelined_all_reduce();
|
|
}
|
|
inline void DebugOptions::_internal_set_xla_gpu_enable_pipelined_all_reduce(bool value) {
|
|
|
|
_impl_.xla_gpu_enable_pipelined_all_reduce_ = value;
|
|
}
|
|
inline void DebugOptions::set_xla_gpu_enable_pipelined_all_reduce(bool value) {
|
|
_internal_set_xla_gpu_enable_pipelined_all_reduce(value);
|
|
// @@protoc_insertion_point(field_set:xla.DebugOptions.xla_gpu_enable_pipelined_all_reduce)
|
|
}
|
|
|
|
// bool xla_gpu_enable_pipelined_all_gather = 227;
|
|
inline void DebugOptions::clear_xla_gpu_enable_pipelined_all_gather() {
|
|
_impl_.xla_gpu_enable_pipelined_all_gather_ = false;
|
|
}
|
|
inline bool DebugOptions::_internal_xla_gpu_enable_pipelined_all_gather() const {
|
|
return _impl_.xla_gpu_enable_pipelined_all_gather_;
|
|
}
|
|
inline bool DebugOptions::xla_gpu_enable_pipelined_all_gather() const {
|
|
// @@protoc_insertion_point(field_get:xla.DebugOptions.xla_gpu_enable_pipelined_all_gather)
|
|
return _internal_xla_gpu_enable_pipelined_all_gather();
|
|
}
|
|
inline void DebugOptions::_internal_set_xla_gpu_enable_pipelined_all_gather(bool value) {
|
|
|
|
_impl_.xla_gpu_enable_pipelined_all_gather_ = value;
|
|
}
|
|
inline void DebugOptions::set_xla_gpu_enable_pipelined_all_gather(bool value) {
|
|
_internal_set_xla_gpu_enable_pipelined_all_gather(value);
|
|
// @@protoc_insertion_point(field_set:xla.DebugOptions.xla_gpu_enable_pipelined_all_gather)
|
|
}
|
|
|
|
// bool xla_gpu_enable_pipelined_reduce_scatter = 231;
|
|
inline void DebugOptions::clear_xla_gpu_enable_pipelined_reduce_scatter() {
|
|
_impl_.xla_gpu_enable_pipelined_reduce_scatter_ = false;
|
|
}
|
|
inline bool DebugOptions::_internal_xla_gpu_enable_pipelined_reduce_scatter() const {
|
|
return _impl_.xla_gpu_enable_pipelined_reduce_scatter_;
|
|
}
|
|
inline bool DebugOptions::xla_gpu_enable_pipelined_reduce_scatter() const {
|
|
// @@protoc_insertion_point(field_get:xla.DebugOptions.xla_gpu_enable_pipelined_reduce_scatter)
|
|
return _internal_xla_gpu_enable_pipelined_reduce_scatter();
|
|
}
|
|
inline void DebugOptions::_internal_set_xla_gpu_enable_pipelined_reduce_scatter(bool value) {
|
|
|
|
_impl_.xla_gpu_enable_pipelined_reduce_scatter_ = value;
|
|
}
|
|
inline void DebugOptions::set_xla_gpu_enable_pipelined_reduce_scatter(bool value) {
|
|
_internal_set_xla_gpu_enable_pipelined_reduce_scatter(value);
|
|
// @@protoc_insertion_point(field_set:xla.DebugOptions.xla_gpu_enable_pipelined_reduce_scatter)
|
|
}
|
|
|
|
// bool xla_gpu_enable_pipelined_p2p = 246;
|
|
inline void DebugOptions::clear_xla_gpu_enable_pipelined_p2p() {
|
|
_impl_.xla_gpu_enable_pipelined_p2p_ = false;
|
|
}
|
|
inline bool DebugOptions::_internal_xla_gpu_enable_pipelined_p2p() const {
|
|
return _impl_.xla_gpu_enable_pipelined_p2p_;
|
|
}
|
|
inline bool DebugOptions::xla_gpu_enable_pipelined_p2p() const {
|
|
// @@protoc_insertion_point(field_get:xla.DebugOptions.xla_gpu_enable_pipelined_p2p)
|
|
return _internal_xla_gpu_enable_pipelined_p2p();
|
|
}
|
|
inline void DebugOptions::_internal_set_xla_gpu_enable_pipelined_p2p(bool value) {
|
|
|
|
_impl_.xla_gpu_enable_pipelined_p2p_ = value;
|
|
}
|
|
inline void DebugOptions::set_xla_gpu_enable_pipelined_p2p(bool value) {
|
|
_internal_set_xla_gpu_enable_pipelined_p2p(value);
|
|
// @@protoc_insertion_point(field_set:xla.DebugOptions.xla_gpu_enable_pipelined_p2p)
|
|
}
|
|
|
|
// int64 xla_gpu_collective_permute_decomposer_threshold = 237;
|
|
inline void DebugOptions::clear_xla_gpu_collective_permute_decomposer_threshold() {
|
|
_impl_.xla_gpu_collective_permute_decomposer_threshold_ = int64_t{0};
|
|
}
|
|
inline int64_t DebugOptions::_internal_xla_gpu_collective_permute_decomposer_threshold() const {
|
|
return _impl_.xla_gpu_collective_permute_decomposer_threshold_;
|
|
}
|
|
inline int64_t DebugOptions::xla_gpu_collective_permute_decomposer_threshold() const {
|
|
// @@protoc_insertion_point(field_get:xla.DebugOptions.xla_gpu_collective_permute_decomposer_threshold)
|
|
return _internal_xla_gpu_collective_permute_decomposer_threshold();
|
|
}
|
|
inline void DebugOptions::_internal_set_xla_gpu_collective_permute_decomposer_threshold(int64_t value) {
|
|
|
|
_impl_.xla_gpu_collective_permute_decomposer_threshold_ = value;
|
|
}
|
|
inline void DebugOptions::set_xla_gpu_collective_permute_decomposer_threshold(int64_t value) {
|
|
_internal_set_xla_gpu_collective_permute_decomposer_threshold(value);
|
|
// @@protoc_insertion_point(field_set:xla.DebugOptions.xla_gpu_collective_permute_decomposer_threshold)
|
|
}
|
|
|
|
// .xla.DebugOptions.PartitioningAlgorithm xla_partitioning_algorithm = 187;
|
|
inline void DebugOptions::clear_xla_partitioning_algorithm() {
|
|
_impl_.xla_partitioning_algorithm_ = 0;
|
|
}
|
|
inline ::xla::DebugOptions_PartitioningAlgorithm DebugOptions::_internal_xla_partitioning_algorithm() const {
|
|
return static_cast< ::xla::DebugOptions_PartitioningAlgorithm >(_impl_.xla_partitioning_algorithm_);
|
|
}
|
|
inline ::xla::DebugOptions_PartitioningAlgorithm DebugOptions::xla_partitioning_algorithm() const {
|
|
// @@protoc_insertion_point(field_get:xla.DebugOptions.xla_partitioning_algorithm)
|
|
return _internal_xla_partitioning_algorithm();
|
|
}
|
|
inline void DebugOptions::_internal_set_xla_partitioning_algorithm(::xla::DebugOptions_PartitioningAlgorithm value) {
|
|
|
|
_impl_.xla_partitioning_algorithm_ = value;
|
|
}
|
|
inline void DebugOptions::set_xla_partitioning_algorithm(::xla::DebugOptions_PartitioningAlgorithm value) {
|
|
_internal_set_xla_partitioning_algorithm(value);
|
|
// @@protoc_insertion_point(field_set:xla.DebugOptions.xla_partitioning_algorithm)
|
|
}
|
|
|
|
// bool xla_gpu_enable_triton_gemm = 188;
|
|
inline void DebugOptions::clear_xla_gpu_enable_triton_gemm() {
|
|
_impl_.xla_gpu_enable_triton_gemm_ = false;
|
|
}
|
|
inline bool DebugOptions::_internal_xla_gpu_enable_triton_gemm() const {
|
|
return _impl_.xla_gpu_enable_triton_gemm_;
|
|
}
|
|
inline bool DebugOptions::xla_gpu_enable_triton_gemm() const {
|
|
// @@protoc_insertion_point(field_get:xla.DebugOptions.xla_gpu_enable_triton_gemm)
|
|
return _internal_xla_gpu_enable_triton_gemm();
|
|
}
|
|
inline void DebugOptions::_internal_set_xla_gpu_enable_triton_gemm(bool value) {
|
|
|
|
_impl_.xla_gpu_enable_triton_gemm_ = value;
|
|
}
|
|
inline void DebugOptions::set_xla_gpu_enable_triton_gemm(bool value) {
|
|
_internal_set_xla_gpu_enable_triton_gemm(value);
|
|
// @@protoc_insertion_point(field_set:xla.DebugOptions.xla_gpu_enable_triton_gemm)
|
|
}
|
|
|
|
// bool xla_gpu_enable_cudnn_int8x32_convolution_reordering = 189;
|
|
inline void DebugOptions::clear_xla_gpu_enable_cudnn_int8x32_convolution_reordering() {
|
|
_impl_.xla_gpu_enable_cudnn_int8x32_convolution_reordering_ = false;
|
|
}
|
|
inline bool DebugOptions::_internal_xla_gpu_enable_cudnn_int8x32_convolution_reordering() const {
|
|
return _impl_.xla_gpu_enable_cudnn_int8x32_convolution_reordering_;
|
|
}
|
|
inline bool DebugOptions::xla_gpu_enable_cudnn_int8x32_convolution_reordering() const {
|
|
// @@protoc_insertion_point(field_get:xla.DebugOptions.xla_gpu_enable_cudnn_int8x32_convolution_reordering)
|
|
return _internal_xla_gpu_enable_cudnn_int8x32_convolution_reordering();
|
|
}
|
|
inline void DebugOptions::_internal_set_xla_gpu_enable_cudnn_int8x32_convolution_reordering(bool value) {
|
|
|
|
_impl_.xla_gpu_enable_cudnn_int8x32_convolution_reordering_ = value;
|
|
}
|
|
inline void DebugOptions::set_xla_gpu_enable_cudnn_int8x32_convolution_reordering(bool value) {
|
|
_internal_set_xla_gpu_enable_cudnn_int8x32_convolution_reordering(value);
|
|
// @@protoc_insertion_point(field_set:xla.DebugOptions.xla_gpu_enable_cudnn_int8x32_convolution_reordering)
|
|
}
|
|
|
|
// bool xla_gpu_triton_gemm_any = 190;
|
|
inline void DebugOptions::clear_xla_gpu_triton_gemm_any() {
|
|
_impl_.xla_gpu_triton_gemm_any_ = false;
|
|
}
|
|
inline bool DebugOptions::_internal_xla_gpu_triton_gemm_any() const {
|
|
return _impl_.xla_gpu_triton_gemm_any_;
|
|
}
|
|
inline bool DebugOptions::xla_gpu_triton_gemm_any() const {
|
|
// @@protoc_insertion_point(field_get:xla.DebugOptions.xla_gpu_triton_gemm_any)
|
|
return _internal_xla_gpu_triton_gemm_any();
|
|
}
|
|
inline void DebugOptions::_internal_set_xla_gpu_triton_gemm_any(bool value) {
|
|
|
|
_impl_.xla_gpu_triton_gemm_any_ = value;
|
|
}
|
|
inline void DebugOptions::set_xla_gpu_triton_gemm_any(bool value) {
|
|
_internal_set_xla_gpu_triton_gemm_any(value);
|
|
// @@protoc_insertion_point(field_set:xla.DebugOptions.xla_gpu_triton_gemm_any)
|
|
}
|
|
|
|
// bool xla_gpu_exhaustive_tiling_search = 219;
|
|
inline void DebugOptions::clear_xla_gpu_exhaustive_tiling_search() {
|
|
_impl_.xla_gpu_exhaustive_tiling_search_ = false;
|
|
}
|
|
inline bool DebugOptions::_internal_xla_gpu_exhaustive_tiling_search() const {
|
|
return _impl_.xla_gpu_exhaustive_tiling_search_;
|
|
}
|
|
inline bool DebugOptions::xla_gpu_exhaustive_tiling_search() const {
|
|
// @@protoc_insertion_point(field_get:xla.DebugOptions.xla_gpu_exhaustive_tiling_search)
|
|
return _internal_xla_gpu_exhaustive_tiling_search();
|
|
}
|
|
inline void DebugOptions::_internal_set_xla_gpu_exhaustive_tiling_search(bool value) {
|
|
|
|
_impl_.xla_gpu_exhaustive_tiling_search_ = value;
|
|
}
|
|
inline void DebugOptions::set_xla_gpu_exhaustive_tiling_search(bool value) {
|
|
_internal_set_xla_gpu_exhaustive_tiling_search(value);
|
|
// @@protoc_insertion_point(field_set:xla.DebugOptions.xla_gpu_exhaustive_tiling_search)
|
|
}
|
|
|
|
// bool xla_gpu_enable_triton_softmax_fusion = 220;
|
|
inline void DebugOptions::clear_xla_gpu_enable_triton_softmax_fusion() {
|
|
_impl_.xla_gpu_enable_triton_softmax_fusion_ = false;
|
|
}
|
|
inline bool DebugOptions::_internal_xla_gpu_enable_triton_softmax_fusion() const {
|
|
return _impl_.xla_gpu_enable_triton_softmax_fusion_;
|
|
}
|
|
inline bool DebugOptions::xla_gpu_enable_triton_softmax_fusion() const {
|
|
// @@protoc_insertion_point(field_get:xla.DebugOptions.xla_gpu_enable_triton_softmax_fusion)
|
|
return _internal_xla_gpu_enable_triton_softmax_fusion();
|
|
}
|
|
inline void DebugOptions::_internal_set_xla_gpu_enable_triton_softmax_fusion(bool value) {
|
|
|
|
_impl_.xla_gpu_enable_triton_softmax_fusion_ = value;
|
|
}
|
|
inline void DebugOptions::set_xla_gpu_enable_triton_softmax_fusion(bool value) {
|
|
_internal_set_xla_gpu_enable_triton_softmax_fusion(value);
|
|
// @@protoc_insertion_point(field_set:xla.DebugOptions.xla_gpu_enable_triton_softmax_fusion)
|
|
}
|
|
|
|
// bool xla_gpu_enable_priority_fusion = 221;
|
|
inline void DebugOptions::clear_xla_gpu_enable_priority_fusion() {
|
|
_impl_.xla_gpu_enable_priority_fusion_ = false;
|
|
}
|
|
inline bool DebugOptions::_internal_xla_gpu_enable_priority_fusion() const {
|
|
return _impl_.xla_gpu_enable_priority_fusion_;
|
|
}
|
|
inline bool DebugOptions::xla_gpu_enable_priority_fusion() const {
|
|
// @@protoc_insertion_point(field_get:xla.DebugOptions.xla_gpu_enable_priority_fusion)
|
|
return _internal_xla_gpu_enable_priority_fusion();
|
|
}
|
|
inline void DebugOptions::_internal_set_xla_gpu_enable_priority_fusion(bool value) {
|
|
|
|
_impl_.xla_gpu_enable_priority_fusion_ = value;
|
|
}
|
|
inline void DebugOptions::set_xla_gpu_enable_priority_fusion(bool value) {
|
|
_internal_set_xla_gpu_enable_priority_fusion(value);
|
|
// @@protoc_insertion_point(field_set:xla.DebugOptions.xla_gpu_enable_priority_fusion)
|
|
}
|
|
|
|
// string xla_gpu_dump_autotune_results_to = 222;
|
|
inline void DebugOptions::clear_xla_gpu_dump_autotune_results_to() {
|
|
_impl_.xla_gpu_dump_autotune_results_to_.ClearToEmpty();
|
|
}
|
|
inline const std::string& DebugOptions::xla_gpu_dump_autotune_results_to() const {
|
|
// @@protoc_insertion_point(field_get:xla.DebugOptions.xla_gpu_dump_autotune_results_to)
|
|
return _internal_xla_gpu_dump_autotune_results_to();
|
|
}
|
|
template <typename ArgT0, typename... ArgT>
|
|
inline PROTOBUF_ALWAYS_INLINE
|
|
void DebugOptions::set_xla_gpu_dump_autotune_results_to(ArgT0&& arg0, ArgT... args) {
|
|
|
|
_impl_.xla_gpu_dump_autotune_results_to_.Set(static_cast<ArgT0 &&>(arg0), args..., GetArenaForAllocation());
|
|
// @@protoc_insertion_point(field_set:xla.DebugOptions.xla_gpu_dump_autotune_results_to)
|
|
}
|
|
inline std::string* DebugOptions::mutable_xla_gpu_dump_autotune_results_to() {
|
|
std::string* _s = _internal_mutable_xla_gpu_dump_autotune_results_to();
|
|
// @@protoc_insertion_point(field_mutable:xla.DebugOptions.xla_gpu_dump_autotune_results_to)
|
|
return _s;
|
|
}
|
|
inline const std::string& DebugOptions::_internal_xla_gpu_dump_autotune_results_to() const {
|
|
return _impl_.xla_gpu_dump_autotune_results_to_.Get();
|
|
}
|
|
inline void DebugOptions::_internal_set_xla_gpu_dump_autotune_results_to(const std::string& value) {
|
|
|
|
_impl_.xla_gpu_dump_autotune_results_to_.Set(value, GetArenaForAllocation());
|
|
}
|
|
inline std::string* DebugOptions::_internal_mutable_xla_gpu_dump_autotune_results_to() {
|
|
|
|
return _impl_.xla_gpu_dump_autotune_results_to_.Mutable(GetArenaForAllocation());
|
|
}
|
|
inline std::string* DebugOptions::release_xla_gpu_dump_autotune_results_to() {
|
|
// @@protoc_insertion_point(field_release:xla.DebugOptions.xla_gpu_dump_autotune_results_to)
|
|
return _impl_.xla_gpu_dump_autotune_results_to_.Release();
|
|
}
|
|
inline void DebugOptions::set_allocated_xla_gpu_dump_autotune_results_to(std::string* xla_gpu_dump_autotune_results_to) {
|
|
if (xla_gpu_dump_autotune_results_to != nullptr) {
|
|
|
|
} else {
|
|
|
|
}
|
|
_impl_.xla_gpu_dump_autotune_results_to_.SetAllocated(xla_gpu_dump_autotune_results_to, GetArenaForAllocation());
|
|
#ifdef PROTOBUF_FORCE_COPY_DEFAULT_STRING
|
|
if (_impl_.xla_gpu_dump_autotune_results_to_.IsDefault()) {
|
|
_impl_.xla_gpu_dump_autotune_results_to_.Set("", GetArenaForAllocation());
|
|
}
|
|
#endif // PROTOBUF_FORCE_COPY_DEFAULT_STRING
|
|
// @@protoc_insertion_point(field_set_allocated:xla.DebugOptions.xla_gpu_dump_autotune_results_to)
|
|
}
|
|
|
|
// string xla_gpu_load_autotune_results_from = 223;
|
|
inline void DebugOptions::clear_xla_gpu_load_autotune_results_from() {
|
|
_impl_.xla_gpu_load_autotune_results_from_.ClearToEmpty();
|
|
}
|
|
inline const std::string& DebugOptions::xla_gpu_load_autotune_results_from() const {
|
|
// @@protoc_insertion_point(field_get:xla.DebugOptions.xla_gpu_load_autotune_results_from)
|
|
return _internal_xla_gpu_load_autotune_results_from();
|
|
}
|
|
template <typename ArgT0, typename... ArgT>
|
|
inline PROTOBUF_ALWAYS_INLINE
|
|
void DebugOptions::set_xla_gpu_load_autotune_results_from(ArgT0&& arg0, ArgT... args) {
|
|
|
|
_impl_.xla_gpu_load_autotune_results_from_.Set(static_cast<ArgT0 &&>(arg0), args..., GetArenaForAllocation());
|
|
// @@protoc_insertion_point(field_set:xla.DebugOptions.xla_gpu_load_autotune_results_from)
|
|
}
|
|
inline std::string* DebugOptions::mutable_xla_gpu_load_autotune_results_from() {
|
|
std::string* _s = _internal_mutable_xla_gpu_load_autotune_results_from();
|
|
// @@protoc_insertion_point(field_mutable:xla.DebugOptions.xla_gpu_load_autotune_results_from)
|
|
return _s;
|
|
}
|
|
inline const std::string& DebugOptions::_internal_xla_gpu_load_autotune_results_from() const {
|
|
return _impl_.xla_gpu_load_autotune_results_from_.Get();
|
|
}
|
|
inline void DebugOptions::_internal_set_xla_gpu_load_autotune_results_from(const std::string& value) {
|
|
|
|
_impl_.xla_gpu_load_autotune_results_from_.Set(value, GetArenaForAllocation());
|
|
}
|
|
inline std::string* DebugOptions::_internal_mutable_xla_gpu_load_autotune_results_from() {
|
|
|
|
return _impl_.xla_gpu_load_autotune_results_from_.Mutable(GetArenaForAllocation());
|
|
}
|
|
inline std::string* DebugOptions::release_xla_gpu_load_autotune_results_from() {
|
|
// @@protoc_insertion_point(field_release:xla.DebugOptions.xla_gpu_load_autotune_results_from)
|
|
return _impl_.xla_gpu_load_autotune_results_from_.Release();
|
|
}
|
|
inline void DebugOptions::set_allocated_xla_gpu_load_autotune_results_from(std::string* xla_gpu_load_autotune_results_from) {
|
|
if (xla_gpu_load_autotune_results_from != nullptr) {
|
|
|
|
} else {
|
|
|
|
}
|
|
_impl_.xla_gpu_load_autotune_results_from_.SetAllocated(xla_gpu_load_autotune_results_from, GetArenaForAllocation());
|
|
#ifdef PROTOBUF_FORCE_COPY_DEFAULT_STRING
|
|
if (_impl_.xla_gpu_load_autotune_results_from_.IsDefault()) {
|
|
_impl_.xla_gpu_load_autotune_results_from_.Set("", GetArenaForAllocation());
|
|
}
|
|
#endif // PROTOBUF_FORCE_COPY_DEFAULT_STRING
|
|
// @@protoc_insertion_point(field_set_allocated:xla.DebugOptions.xla_gpu_load_autotune_results_from)
|
|
}
|
|
|
|
// string xla_gpu_target_config_filename = 261;
|
|
inline void DebugOptions::clear_xla_gpu_target_config_filename() {
|
|
_impl_.xla_gpu_target_config_filename_.ClearToEmpty();
|
|
}
|
|
inline const std::string& DebugOptions::xla_gpu_target_config_filename() const {
|
|
// @@protoc_insertion_point(field_get:xla.DebugOptions.xla_gpu_target_config_filename)
|
|
return _internal_xla_gpu_target_config_filename();
|
|
}
|
|
template <typename ArgT0, typename... ArgT>
|
|
inline PROTOBUF_ALWAYS_INLINE
|
|
void DebugOptions::set_xla_gpu_target_config_filename(ArgT0&& arg0, ArgT... args) {
|
|
|
|
_impl_.xla_gpu_target_config_filename_.Set(static_cast<ArgT0 &&>(arg0), args..., GetArenaForAllocation());
|
|
// @@protoc_insertion_point(field_set:xla.DebugOptions.xla_gpu_target_config_filename)
|
|
}
|
|
inline std::string* DebugOptions::mutable_xla_gpu_target_config_filename() {
|
|
std::string* _s = _internal_mutable_xla_gpu_target_config_filename();
|
|
// @@protoc_insertion_point(field_mutable:xla.DebugOptions.xla_gpu_target_config_filename)
|
|
return _s;
|
|
}
|
|
inline const std::string& DebugOptions::_internal_xla_gpu_target_config_filename() const {
|
|
return _impl_.xla_gpu_target_config_filename_.Get();
|
|
}
|
|
inline void DebugOptions::_internal_set_xla_gpu_target_config_filename(const std::string& value) {
|
|
|
|
_impl_.xla_gpu_target_config_filename_.Set(value, GetArenaForAllocation());
|
|
}
|
|
inline std::string* DebugOptions::_internal_mutable_xla_gpu_target_config_filename() {
|
|
|
|
return _impl_.xla_gpu_target_config_filename_.Mutable(GetArenaForAllocation());
|
|
}
|
|
inline std::string* DebugOptions::release_xla_gpu_target_config_filename() {
|
|
// @@protoc_insertion_point(field_release:xla.DebugOptions.xla_gpu_target_config_filename)
|
|
return _impl_.xla_gpu_target_config_filename_.Release();
|
|
}
|
|
inline void DebugOptions::set_allocated_xla_gpu_target_config_filename(std::string* xla_gpu_target_config_filename) {
|
|
if (xla_gpu_target_config_filename != nullptr) {
|
|
|
|
} else {
|
|
|
|
}
|
|
_impl_.xla_gpu_target_config_filename_.SetAllocated(xla_gpu_target_config_filename, GetArenaForAllocation());
|
|
#ifdef PROTOBUF_FORCE_COPY_DEFAULT_STRING
|
|
if (_impl_.xla_gpu_target_config_filename_.IsDefault()) {
|
|
_impl_.xla_gpu_target_config_filename_.Set("", GetArenaForAllocation());
|
|
}
|
|
#endif // PROTOBUF_FORCE_COPY_DEFAULT_STRING
|
|
// @@protoc_insertion_point(field_set_allocated:xla.DebugOptions.xla_gpu_target_config_filename)
|
|
}
|
|
|
|
// int32 xla_gpu_auto_spmd_partitioning_memory_budget_gb = 224;
|
|
inline void DebugOptions::clear_xla_gpu_auto_spmd_partitioning_memory_budget_gb() {
|
|
_impl_.xla_gpu_auto_spmd_partitioning_memory_budget_gb_ = 0;
|
|
}
|
|
inline int32_t DebugOptions::_internal_xla_gpu_auto_spmd_partitioning_memory_budget_gb() const {
|
|
return _impl_.xla_gpu_auto_spmd_partitioning_memory_budget_gb_;
|
|
}
|
|
inline int32_t DebugOptions::xla_gpu_auto_spmd_partitioning_memory_budget_gb() const {
|
|
// @@protoc_insertion_point(field_get:xla.DebugOptions.xla_gpu_auto_spmd_partitioning_memory_budget_gb)
|
|
return _internal_xla_gpu_auto_spmd_partitioning_memory_budget_gb();
|
|
}
|
|
inline void DebugOptions::_internal_set_xla_gpu_auto_spmd_partitioning_memory_budget_gb(int32_t value) {
|
|
|
|
_impl_.xla_gpu_auto_spmd_partitioning_memory_budget_gb_ = value;
|
|
}
|
|
inline void DebugOptions::set_xla_gpu_auto_spmd_partitioning_memory_budget_gb(int32_t value) {
|
|
_internal_set_xla_gpu_auto_spmd_partitioning_memory_budget_gb(value);
|
|
// @@protoc_insertion_point(field_set:xla.DebugOptions.xla_gpu_auto_spmd_partitioning_memory_budget_gb)
|
|
}
|
|
|
|
// float xla_gpu_auto_spmd_partitioning_memory_budget_ratio = 225;
|
|
inline void DebugOptions::clear_xla_gpu_auto_spmd_partitioning_memory_budget_ratio() {
|
|
_impl_.xla_gpu_auto_spmd_partitioning_memory_budget_ratio_ = 0;
|
|
}
|
|
inline float DebugOptions::_internal_xla_gpu_auto_spmd_partitioning_memory_budget_ratio() const {
|
|
return _impl_.xla_gpu_auto_spmd_partitioning_memory_budget_ratio_;
|
|
}
|
|
inline float DebugOptions::xla_gpu_auto_spmd_partitioning_memory_budget_ratio() const {
|
|
// @@protoc_insertion_point(field_get:xla.DebugOptions.xla_gpu_auto_spmd_partitioning_memory_budget_ratio)
|
|
return _internal_xla_gpu_auto_spmd_partitioning_memory_budget_ratio();
|
|
}
|
|
inline void DebugOptions::_internal_set_xla_gpu_auto_spmd_partitioning_memory_budget_ratio(float value) {
|
|
|
|
_impl_.xla_gpu_auto_spmd_partitioning_memory_budget_ratio_ = value;
|
|
}
|
|
inline void DebugOptions::set_xla_gpu_auto_spmd_partitioning_memory_budget_ratio(float value) {
|
|
_internal_set_xla_gpu_auto_spmd_partitioning_memory_budget_ratio(value);
|
|
// @@protoc_insertion_point(field_set:xla.DebugOptions.xla_gpu_auto_spmd_partitioning_memory_budget_ratio)
|
|
}
|
|
|
|
// bool xla_gpu_triton_gemm_disable_reduced_precision_reduction = 226;
|
|
inline void DebugOptions::clear_xla_gpu_triton_gemm_disable_reduced_precision_reduction() {
|
|
_impl_.xla_gpu_triton_gemm_disable_reduced_precision_reduction_ = false;
|
|
}
|
|
inline bool DebugOptions::_internal_xla_gpu_triton_gemm_disable_reduced_precision_reduction() const {
|
|
return _impl_.xla_gpu_triton_gemm_disable_reduced_precision_reduction_;
|
|
}
|
|
inline bool DebugOptions::xla_gpu_triton_gemm_disable_reduced_precision_reduction() const {
|
|
// @@protoc_insertion_point(field_get:xla.DebugOptions.xla_gpu_triton_gemm_disable_reduced_precision_reduction)
|
|
return _internal_xla_gpu_triton_gemm_disable_reduced_precision_reduction();
|
|
}
|
|
inline void DebugOptions::_internal_set_xla_gpu_triton_gemm_disable_reduced_precision_reduction(bool value) {
|
|
|
|
_impl_.xla_gpu_triton_gemm_disable_reduced_precision_reduction_ = value;
|
|
}
|
|
inline void DebugOptions::set_xla_gpu_triton_gemm_disable_reduced_precision_reduction(bool value) {
|
|
_internal_set_xla_gpu_triton_gemm_disable_reduced_precision_reduction(value);
|
|
// @@protoc_insertion_point(field_set:xla.DebugOptions.xla_gpu_triton_gemm_disable_reduced_precision_reduction)
|
|
}
|
|
|
|
// int32 xla_gpu_triton_fusion_level = 229;
|
|
inline void DebugOptions::clear_xla_gpu_triton_fusion_level() {
|
|
_impl_.xla_gpu_triton_fusion_level_ = 0;
|
|
}
|
|
inline int32_t DebugOptions::_internal_xla_gpu_triton_fusion_level() const {
|
|
return _impl_.xla_gpu_triton_fusion_level_;
|
|
}
|
|
inline int32_t DebugOptions::xla_gpu_triton_fusion_level() const {
|
|
// @@protoc_insertion_point(field_get:xla.DebugOptions.xla_gpu_triton_fusion_level)
|
|
return _internal_xla_gpu_triton_fusion_level();
|
|
}
|
|
inline void DebugOptions::_internal_set_xla_gpu_triton_fusion_level(int32_t value) {
|
|
|
|
_impl_.xla_gpu_triton_fusion_level_ = value;
|
|
}
|
|
inline void DebugOptions::set_xla_gpu_triton_fusion_level(int32_t value) {
|
|
_internal_set_xla_gpu_triton_fusion_level(value);
|
|
// @@protoc_insertion_point(field_set:xla.DebugOptions.xla_gpu_triton_fusion_level)
|
|
}
|
|
|
|
// bool xla_gpu_dump_autotuned_triton_fusions = 232;
|
|
inline void DebugOptions::clear_xla_gpu_dump_autotuned_triton_fusions() {
|
|
_impl_.xla_gpu_dump_autotuned_triton_fusions_ = false;
|
|
}
|
|
inline bool DebugOptions::_internal_xla_gpu_dump_autotuned_triton_fusions() const {
|
|
return _impl_.xla_gpu_dump_autotuned_triton_fusions_;
|
|
}
|
|
inline bool DebugOptions::xla_gpu_dump_autotuned_triton_fusions() const {
|
|
// @@protoc_insertion_point(field_get:xla.DebugOptions.xla_gpu_dump_autotuned_triton_fusions)
|
|
return _internal_xla_gpu_dump_autotuned_triton_fusions();
|
|
}
|
|
inline void DebugOptions::_internal_set_xla_gpu_dump_autotuned_triton_fusions(bool value) {
|
|
|
|
_impl_.xla_gpu_dump_autotuned_triton_fusions_ = value;
|
|
}
|
|
inline void DebugOptions::set_xla_gpu_dump_autotuned_triton_fusions(bool value) {
|
|
_internal_set_xla_gpu_dump_autotuned_triton_fusions(value);
|
|
// @@protoc_insertion_point(field_set:xla.DebugOptions.xla_gpu_dump_autotuned_triton_fusions)
|
|
}
|
|
|
|
// bool xla_gpu_copy_insertion_use_region_analysis = 236;
|
|
inline void DebugOptions::clear_xla_gpu_copy_insertion_use_region_analysis() {
|
|
_impl_.xla_gpu_copy_insertion_use_region_analysis_ = false;
|
|
}
|
|
inline bool DebugOptions::_internal_xla_gpu_copy_insertion_use_region_analysis() const {
|
|
return _impl_.xla_gpu_copy_insertion_use_region_analysis_;
|
|
}
|
|
inline bool DebugOptions::xla_gpu_copy_insertion_use_region_analysis() const {
|
|
// @@protoc_insertion_point(field_get:xla.DebugOptions.xla_gpu_copy_insertion_use_region_analysis)
|
|
return _internal_xla_gpu_copy_insertion_use_region_analysis();
|
|
}
|
|
inline void DebugOptions::_internal_set_xla_gpu_copy_insertion_use_region_analysis(bool value) {
|
|
|
|
_impl_.xla_gpu_copy_insertion_use_region_analysis_ = value;
|
|
}
|
|
inline void DebugOptions::set_xla_gpu_copy_insertion_use_region_analysis(bool value) {
|
|
_internal_set_xla_gpu_copy_insertion_use_region_analysis(value);
|
|
// @@protoc_insertion_point(field_set:xla.DebugOptions.xla_gpu_copy_insertion_use_region_analysis)
|
|
}
|
|
|
|
// bool xla_gpu_collect_cost_model_stats = 240;
|
|
inline void DebugOptions::clear_xla_gpu_collect_cost_model_stats() {
|
|
_impl_.xla_gpu_collect_cost_model_stats_ = false;
|
|
}
|
|
inline bool DebugOptions::_internal_xla_gpu_collect_cost_model_stats() const {
|
|
return _impl_.xla_gpu_collect_cost_model_stats_;
|
|
}
|
|
inline bool DebugOptions::xla_gpu_collect_cost_model_stats() const {
|
|
// @@protoc_insertion_point(field_get:xla.DebugOptions.xla_gpu_collect_cost_model_stats)
|
|
return _internal_xla_gpu_collect_cost_model_stats();
|
|
}
|
|
inline void DebugOptions::_internal_set_xla_gpu_collect_cost_model_stats(bool value) {
|
|
|
|
_impl_.xla_gpu_collect_cost_model_stats_ = value;
|
|
}
|
|
inline void DebugOptions::set_xla_gpu_collect_cost_model_stats(bool value) {
|
|
_internal_set_xla_gpu_collect_cost_model_stats(value);
|
|
// @@protoc_insertion_point(field_set:xla.DebugOptions.xla_gpu_collect_cost_model_stats)
|
|
}
|
|
|
|
// bool xla_gpu_enable_split_k_autotuning = 241;
|
|
inline void DebugOptions::clear_xla_gpu_enable_split_k_autotuning() {
|
|
_impl_.xla_gpu_enable_split_k_autotuning_ = false;
|
|
}
|
|
inline bool DebugOptions::_internal_xla_gpu_enable_split_k_autotuning() const {
|
|
return _impl_.xla_gpu_enable_split_k_autotuning_;
|
|
}
|
|
inline bool DebugOptions::xla_gpu_enable_split_k_autotuning() const {
|
|
// @@protoc_insertion_point(field_get:xla.DebugOptions.xla_gpu_enable_split_k_autotuning)
|
|
return _internal_xla_gpu_enable_split_k_autotuning();
|
|
}
|
|
inline void DebugOptions::_internal_set_xla_gpu_enable_split_k_autotuning(bool value) {
|
|
|
|
_impl_.xla_gpu_enable_split_k_autotuning_ = value;
|
|
}
|
|
inline void DebugOptions::set_xla_gpu_enable_split_k_autotuning(bool value) {
|
|
_internal_set_xla_gpu_enable_split_k_autotuning(value);
|
|
// @@protoc_insertion_point(field_set:xla.DebugOptions.xla_gpu_enable_split_k_autotuning)
|
|
}
|
|
|
|
// bool xla_gpu_enable_reduction_epilogue_fusion = 243;
|
|
inline void DebugOptions::clear_xla_gpu_enable_reduction_epilogue_fusion() {
|
|
_impl_.xla_gpu_enable_reduction_epilogue_fusion_ = false;
|
|
}
|
|
inline bool DebugOptions::_internal_xla_gpu_enable_reduction_epilogue_fusion() const {
|
|
return _impl_.xla_gpu_enable_reduction_epilogue_fusion_;
|
|
}
|
|
inline bool DebugOptions::xla_gpu_enable_reduction_epilogue_fusion() const {
|
|
// @@protoc_insertion_point(field_get:xla.DebugOptions.xla_gpu_enable_reduction_epilogue_fusion)
|
|
return _internal_xla_gpu_enable_reduction_epilogue_fusion();
|
|
}
|
|
inline void DebugOptions::_internal_set_xla_gpu_enable_reduction_epilogue_fusion(bool value) {
|
|
|
|
_impl_.xla_gpu_enable_reduction_epilogue_fusion_ = value;
|
|
}
|
|
inline void DebugOptions::set_xla_gpu_enable_reduction_epilogue_fusion(bool value) {
|
|
_internal_set_xla_gpu_enable_reduction_epilogue_fusion(value);
|
|
// @@protoc_insertion_point(field_set:xla.DebugOptions.xla_gpu_enable_reduction_epilogue_fusion)
|
|
}
|
|
|
|
// bool xla_gpu_enable_nccl_clique_optimization = 244;
|
|
inline void DebugOptions::clear_xla_gpu_enable_nccl_clique_optimization() {
|
|
_impl_.xla_gpu_enable_nccl_clique_optimization_ = false;
|
|
}
|
|
inline bool DebugOptions::_internal_xla_gpu_enable_nccl_clique_optimization() const {
|
|
return _impl_.xla_gpu_enable_nccl_clique_optimization_;
|
|
}
|
|
inline bool DebugOptions::xla_gpu_enable_nccl_clique_optimization() const {
|
|
// @@protoc_insertion_point(field_get:xla.DebugOptions.xla_gpu_enable_nccl_clique_optimization)
|
|
return _internal_xla_gpu_enable_nccl_clique_optimization();
|
|
}
|
|
inline void DebugOptions::_internal_set_xla_gpu_enable_nccl_clique_optimization(bool value) {
|
|
|
|
_impl_.xla_gpu_enable_nccl_clique_optimization_ = value;
|
|
}
|
|
inline void DebugOptions::set_xla_gpu_enable_nccl_clique_optimization(bool value) {
|
|
_internal_set_xla_gpu_enable_nccl_clique_optimization(value);
|
|
// @@protoc_insertion_point(field_set:xla.DebugOptions.xla_gpu_enable_nccl_clique_optimization)
|
|
}
|
|
|
|
// bool xla_gpu_mock_custom_calls = 245;
|
|
inline void DebugOptions::clear_xla_gpu_mock_custom_calls() {
|
|
_impl_.xla_gpu_mock_custom_calls_ = false;
|
|
}
|
|
inline bool DebugOptions::_internal_xla_gpu_mock_custom_calls() const {
|
|
return _impl_.xla_gpu_mock_custom_calls_;
|
|
}
|
|
inline bool DebugOptions::xla_gpu_mock_custom_calls() const {
|
|
// @@protoc_insertion_point(field_get:xla.DebugOptions.xla_gpu_mock_custom_calls)
|
|
return _internal_xla_gpu_mock_custom_calls();
|
|
}
|
|
inline void DebugOptions::_internal_set_xla_gpu_mock_custom_calls(bool value) {
|
|
|
|
_impl_.xla_gpu_mock_custom_calls_ = value;
|
|
}
|
|
inline void DebugOptions::set_xla_gpu_mock_custom_calls(bool value) {
|
|
_internal_set_xla_gpu_mock_custom_calls(value);
|
|
// @@protoc_insertion_point(field_set:xla.DebugOptions.xla_gpu_mock_custom_calls)
|
|
}
|
|
|
|
// bool xla_gpu_cublas_fallback = 247;
|
|
inline void DebugOptions::clear_xla_gpu_cublas_fallback() {
|
|
_impl_.xla_gpu_cublas_fallback_ = false;
|
|
}
|
|
inline bool DebugOptions::_internal_xla_gpu_cublas_fallback() const {
|
|
return _impl_.xla_gpu_cublas_fallback_;
|
|
}
|
|
inline bool DebugOptions::xla_gpu_cublas_fallback() const {
|
|
// @@protoc_insertion_point(field_get:xla.DebugOptions.xla_gpu_cublas_fallback)
|
|
return _internal_xla_gpu_cublas_fallback();
|
|
}
|
|
inline void DebugOptions::_internal_set_xla_gpu_cublas_fallback(bool value) {
|
|
|
|
_impl_.xla_gpu_cublas_fallback_ = value;
|
|
}
|
|
inline void DebugOptions::set_xla_gpu_cublas_fallback(bool value) {
|
|
_internal_set_xla_gpu_cublas_fallback(value);
|
|
// @@protoc_insertion_point(field_set:xla.DebugOptions.xla_gpu_cublas_fallback)
|
|
}
|
|
|
|
// bool xla_gpu_enable_while_loop_double_buffering = 248;
|
|
inline void DebugOptions::clear_xla_gpu_enable_while_loop_double_buffering() {
|
|
_impl_.xla_gpu_enable_while_loop_double_buffering_ = false;
|
|
}
|
|
inline bool DebugOptions::_internal_xla_gpu_enable_while_loop_double_buffering() const {
|
|
return _impl_.xla_gpu_enable_while_loop_double_buffering_;
|
|
}
|
|
inline bool DebugOptions::xla_gpu_enable_while_loop_double_buffering() const {
|
|
// @@protoc_insertion_point(field_get:xla.DebugOptions.xla_gpu_enable_while_loop_double_buffering)
|
|
return _internal_xla_gpu_enable_while_loop_double_buffering();
|
|
}
|
|
inline void DebugOptions::_internal_set_xla_gpu_enable_while_loop_double_buffering(bool value) {
|
|
|
|
_impl_.xla_gpu_enable_while_loop_double_buffering_ = value;
|
|
}
|
|
inline void DebugOptions::set_xla_gpu_enable_while_loop_double_buffering(bool value) {
|
|
_internal_set_xla_gpu_enable_while_loop_double_buffering(value);
|
|
// @@protoc_insertion_point(field_set:xla.DebugOptions.xla_gpu_enable_while_loop_double_buffering)
|
|
}
|
|
|
|
// bool xla_gpu_ensure_minor_dot_contraction_dims = 249;
|
|
inline void DebugOptions::clear_xla_gpu_ensure_minor_dot_contraction_dims() {
|
|
_impl_.xla_gpu_ensure_minor_dot_contraction_dims_ = false;
|
|
}
|
|
inline bool DebugOptions::_internal_xla_gpu_ensure_minor_dot_contraction_dims() const {
|
|
return _impl_.xla_gpu_ensure_minor_dot_contraction_dims_;
|
|
}
|
|
inline bool DebugOptions::xla_gpu_ensure_minor_dot_contraction_dims() const {
|
|
// @@protoc_insertion_point(field_get:xla.DebugOptions.xla_gpu_ensure_minor_dot_contraction_dims)
|
|
return _internal_xla_gpu_ensure_minor_dot_contraction_dims();
|
|
}
|
|
inline void DebugOptions::_internal_set_xla_gpu_ensure_minor_dot_contraction_dims(bool value) {
|
|
|
|
_impl_.xla_gpu_ensure_minor_dot_contraction_dims_ = value;
|
|
}
|
|
inline void DebugOptions::set_xla_gpu_ensure_minor_dot_contraction_dims(bool value) {
|
|
_internal_set_xla_gpu_ensure_minor_dot_contraction_dims(value);
|
|
// @@protoc_insertion_point(field_set:xla.DebugOptions.xla_gpu_ensure_minor_dot_contraction_dims)
|
|
}
|
|
|
|
// bool xla_gpu_filter_kernels_spilling_registers_on_autotuning = 250;
|
|
inline void DebugOptions::clear_xla_gpu_filter_kernels_spilling_registers_on_autotuning() {
|
|
_impl_.xla_gpu_filter_kernels_spilling_registers_on_autotuning_ = false;
|
|
}
|
|
inline bool DebugOptions::_internal_xla_gpu_filter_kernels_spilling_registers_on_autotuning() const {
|
|
return _impl_.xla_gpu_filter_kernels_spilling_registers_on_autotuning_;
|
|
}
|
|
inline bool DebugOptions::xla_gpu_filter_kernels_spilling_registers_on_autotuning() const {
|
|
// @@protoc_insertion_point(field_get:xla.DebugOptions.xla_gpu_filter_kernels_spilling_registers_on_autotuning)
|
|
return _internal_xla_gpu_filter_kernels_spilling_registers_on_autotuning();
|
|
}
|
|
inline void DebugOptions::_internal_set_xla_gpu_filter_kernels_spilling_registers_on_autotuning(bool value) {
|
|
|
|
_impl_.xla_gpu_filter_kernels_spilling_registers_on_autotuning_ = value;
|
|
}
|
|
inline void DebugOptions::set_xla_gpu_filter_kernels_spilling_registers_on_autotuning(bool value) {
|
|
_internal_set_xla_gpu_filter_kernels_spilling_registers_on_autotuning(value);
|
|
// @@protoc_insertion_point(field_set:xla.DebugOptions.xla_gpu_filter_kernels_spilling_registers_on_autotuning)
|
|
}
|
|
|
|
// int64 xla_debug_buffer_assignment_show_max = 251;
|
|
inline void DebugOptions::clear_xla_debug_buffer_assignment_show_max() {
|
|
_impl_.xla_debug_buffer_assignment_show_max_ = int64_t{0};
|
|
}
|
|
inline int64_t DebugOptions::_internal_xla_debug_buffer_assignment_show_max() const {
|
|
return _impl_.xla_debug_buffer_assignment_show_max_;
|
|
}
|
|
inline int64_t DebugOptions::xla_debug_buffer_assignment_show_max() const {
|
|
// @@protoc_insertion_point(field_get:xla.DebugOptions.xla_debug_buffer_assignment_show_max)
|
|
return _internal_xla_debug_buffer_assignment_show_max();
|
|
}
|
|
inline void DebugOptions::_internal_set_xla_debug_buffer_assignment_show_max(int64_t value) {
|
|
|
|
_impl_.xla_debug_buffer_assignment_show_max_ = value;
|
|
}
|
|
inline void DebugOptions::set_xla_debug_buffer_assignment_show_max(int64_t value) {
|
|
_internal_set_xla_debug_buffer_assignment_show_max(value);
|
|
// @@protoc_insertion_point(field_set:xla.DebugOptions.xla_debug_buffer_assignment_show_max)
|
|
}
|
|
|
|
// int32 xla_gpu_llvm_verification_level = 256;
|
|
inline void DebugOptions::clear_xla_gpu_llvm_verification_level() {
|
|
_impl_.xla_gpu_llvm_verification_level_ = 0;
|
|
}
|
|
inline int32_t DebugOptions::_internal_xla_gpu_llvm_verification_level() const {
|
|
return _impl_.xla_gpu_llvm_verification_level_;
|
|
}
|
|
inline int32_t DebugOptions::xla_gpu_llvm_verification_level() const {
|
|
// @@protoc_insertion_point(field_get:xla.DebugOptions.xla_gpu_llvm_verification_level)
|
|
return _internal_xla_gpu_llvm_verification_level();
|
|
}
|
|
inline void DebugOptions::_internal_set_xla_gpu_llvm_verification_level(int32_t value) {
|
|
|
|
_impl_.xla_gpu_llvm_verification_level_ = value;
|
|
}
|
|
inline void DebugOptions::set_xla_gpu_llvm_verification_level(int32_t value) {
|
|
_internal_set_xla_gpu_llvm_verification_level(value);
|
|
// @@protoc_insertion_point(field_set:xla.DebugOptions.xla_gpu_llvm_verification_level)
|
|
}
|
|
|
|
// bool xla_gpu_enable_cub_radix_sort = 259;
|
|
inline void DebugOptions::clear_xla_gpu_enable_cub_radix_sort() {
|
|
_impl_.xla_gpu_enable_cub_radix_sort_ = false;
|
|
}
|
|
inline bool DebugOptions::_internal_xla_gpu_enable_cub_radix_sort() const {
|
|
return _impl_.xla_gpu_enable_cub_radix_sort_;
|
|
}
|
|
inline bool DebugOptions::xla_gpu_enable_cub_radix_sort() const {
|
|
// @@protoc_insertion_point(field_get:xla.DebugOptions.xla_gpu_enable_cub_radix_sort)
|
|
return _internal_xla_gpu_enable_cub_radix_sort();
|
|
}
|
|
inline void DebugOptions::_internal_set_xla_gpu_enable_cub_radix_sort(bool value) {
|
|
|
|
_impl_.xla_gpu_enable_cub_radix_sort_ = value;
|
|
}
|
|
inline void DebugOptions::set_xla_gpu_enable_cub_radix_sort(bool value) {
|
|
_internal_set_xla_gpu_enable_cub_radix_sort(value);
|
|
// @@protoc_insertion_point(field_set:xla.DebugOptions.xla_gpu_enable_cub_radix_sort)
|
|
}
|
|
|
|
// int64 xla_gpu_threshold_for_windowed_einsum_mib = 265;
|
|
inline void DebugOptions::clear_xla_gpu_threshold_for_windowed_einsum_mib() {
|
|
_impl_.xla_gpu_threshold_for_windowed_einsum_mib_ = int64_t{0};
|
|
}
|
|
inline int64_t DebugOptions::_internal_xla_gpu_threshold_for_windowed_einsum_mib() const {
|
|
return _impl_.xla_gpu_threshold_for_windowed_einsum_mib_;
|
|
}
|
|
inline int64_t DebugOptions::xla_gpu_threshold_for_windowed_einsum_mib() const {
|
|
// @@protoc_insertion_point(field_get:xla.DebugOptions.xla_gpu_threshold_for_windowed_einsum_mib)
|
|
return _internal_xla_gpu_threshold_for_windowed_einsum_mib();
|
|
}
|
|
inline void DebugOptions::_internal_set_xla_gpu_threshold_for_windowed_einsum_mib(int64_t value) {
|
|
|
|
_impl_.xla_gpu_threshold_for_windowed_einsum_mib_ = value;
|
|
}
|
|
inline void DebugOptions::set_xla_gpu_threshold_for_windowed_einsum_mib(int64_t value) {
|
|
_internal_set_xla_gpu_threshold_for_windowed_einsum_mib(value);
|
|
// @@protoc_insertion_point(field_set:xla.DebugOptions.xla_gpu_threshold_for_windowed_einsum_mib)
|
|
}
|
|
|
|
// bool xla_gpu_enable_triton_hopper = 266;
|
|
inline void DebugOptions::clear_xla_gpu_enable_triton_hopper() {
|
|
_impl_.xla_gpu_enable_triton_hopper_ = false;
|
|
}
|
|
inline bool DebugOptions::_internal_xla_gpu_enable_triton_hopper() const {
|
|
return _impl_.xla_gpu_enable_triton_hopper_;
|
|
}
|
|
inline bool DebugOptions::xla_gpu_enable_triton_hopper() const {
|
|
// @@protoc_insertion_point(field_get:xla.DebugOptions.xla_gpu_enable_triton_hopper)
|
|
return _internal_xla_gpu_enable_triton_hopper();
|
|
}
|
|
inline void DebugOptions::_internal_set_xla_gpu_enable_triton_hopper(bool value) {
|
|
|
|
_impl_.xla_gpu_enable_triton_hopper_ = value;
|
|
}
|
|
inline void DebugOptions::set_xla_gpu_enable_triton_hopper(bool value) {
|
|
_internal_set_xla_gpu_enable_triton_hopper(value);
|
|
// @@protoc_insertion_point(field_set:xla.DebugOptions.xla_gpu_enable_triton_hopper)
|
|
}
|
|
|
|
// bool xla_gpu_enable_nccl_user_buffers = 267;
|
|
inline void DebugOptions::clear_xla_gpu_enable_nccl_user_buffers() {
|
|
_impl_.xla_gpu_enable_nccl_user_buffers_ = false;
|
|
}
|
|
inline bool DebugOptions::_internal_xla_gpu_enable_nccl_user_buffers() const {
|
|
return _impl_.xla_gpu_enable_nccl_user_buffers_;
|
|
}
|
|
inline bool DebugOptions::xla_gpu_enable_nccl_user_buffers() const {
|
|
// @@protoc_insertion_point(field_get:xla.DebugOptions.xla_gpu_enable_nccl_user_buffers)
|
|
return _internal_xla_gpu_enable_nccl_user_buffers();
|
|
}
|
|
inline void DebugOptions::_internal_set_xla_gpu_enable_nccl_user_buffers(bool value) {
|
|
|
|
_impl_.xla_gpu_enable_nccl_user_buffers_ = value;
|
|
}
|
|
inline void DebugOptions::set_xla_gpu_enable_nccl_user_buffers(bool value) {
|
|
_internal_set_xla_gpu_enable_nccl_user_buffers(value);
|
|
// @@protoc_insertion_point(field_set:xla.DebugOptions.xla_gpu_enable_nccl_user_buffers)
|
|
}
|
|
|
|
// bool xla_gpu_enable_libnvptxcompiler = 269;
|
|
inline void DebugOptions::clear_xla_gpu_enable_libnvptxcompiler() {
|
|
_impl_.xla_gpu_enable_libnvptxcompiler_ = false;
|
|
}
|
|
inline bool DebugOptions::_internal_xla_gpu_enable_libnvptxcompiler() const {
|
|
return _impl_.xla_gpu_enable_libnvptxcompiler_;
|
|
}
|
|
inline bool DebugOptions::xla_gpu_enable_libnvptxcompiler() const {
|
|
// @@protoc_insertion_point(field_get:xla.DebugOptions.xla_gpu_enable_libnvptxcompiler)
|
|
return _internal_xla_gpu_enable_libnvptxcompiler();
|
|
}
|
|
inline void DebugOptions::_internal_set_xla_gpu_enable_libnvptxcompiler(bool value) {
|
|
|
|
_impl_.xla_gpu_enable_libnvptxcompiler_ = value;
|
|
}
|
|
inline void DebugOptions::set_xla_gpu_enable_libnvptxcompiler(bool value) {
|
|
_internal_set_xla_gpu_enable_libnvptxcompiler(value);
|
|
// @@protoc_insertion_point(field_set:xla.DebugOptions.xla_gpu_enable_libnvptxcompiler)
|
|
}
|
|
|
|
// bool xla_gpu_enable_dot_strength_reduction = 270;
|
|
inline void DebugOptions::clear_xla_gpu_enable_dot_strength_reduction() {
|
|
_impl_.xla_gpu_enable_dot_strength_reduction_ = false;
|
|
}
|
|
inline bool DebugOptions::_internal_xla_gpu_enable_dot_strength_reduction() const {
|
|
return _impl_.xla_gpu_enable_dot_strength_reduction_;
|
|
}
|
|
inline bool DebugOptions::xla_gpu_enable_dot_strength_reduction() const {
|
|
// @@protoc_insertion_point(field_get:xla.DebugOptions.xla_gpu_enable_dot_strength_reduction)
|
|
return _internal_xla_gpu_enable_dot_strength_reduction();
|
|
}
|
|
inline void DebugOptions::_internal_set_xla_gpu_enable_dot_strength_reduction(bool value) {
|
|
|
|
_impl_.xla_gpu_enable_dot_strength_reduction_ = value;
|
|
}
|
|
inline void DebugOptions::set_xla_gpu_enable_dot_strength_reduction(bool value) {
|
|
_internal_set_xla_gpu_enable_dot_strength_reduction(value);
|
|
// @@protoc_insertion_point(field_set:xla.DebugOptions.xla_gpu_enable_dot_strength_reduction)
|
|
}
|
|
|
|
// map<string, string> xla_backend_extra_options = 500;
|
|
inline int DebugOptions::_internal_xla_backend_extra_options_size() const {
|
|
return _impl_.xla_backend_extra_options_.size();
|
|
}
|
|
inline int DebugOptions::xla_backend_extra_options_size() const {
|
|
return _internal_xla_backend_extra_options_size();
|
|
}
|
|
inline void DebugOptions::clear_xla_backend_extra_options() {
|
|
_impl_.xla_backend_extra_options_.Clear();
|
|
}
|
|
inline const ::PROTOBUF_NAMESPACE_ID::Map< std::string, std::string >&
|
|
DebugOptions::_internal_xla_backend_extra_options() const {
|
|
return _impl_.xla_backend_extra_options_.GetMap();
|
|
}
|
|
inline const ::PROTOBUF_NAMESPACE_ID::Map< std::string, std::string >&
|
|
DebugOptions::xla_backend_extra_options() const {
|
|
// @@protoc_insertion_point(field_map:xla.DebugOptions.xla_backend_extra_options)
|
|
return _internal_xla_backend_extra_options();
|
|
}
|
|
inline ::PROTOBUF_NAMESPACE_ID::Map< std::string, std::string >*
|
|
DebugOptions::_internal_mutable_xla_backend_extra_options() {
|
|
return _impl_.xla_backend_extra_options_.MutableMap();
|
|
}
|
|
inline ::PROTOBUF_NAMESPACE_ID::Map< std::string, std::string >*
|
|
DebugOptions::mutable_xla_backend_extra_options() {
|
|
// @@protoc_insertion_point(field_mutable_map:xla.DebugOptions.xla_backend_extra_options)
|
|
return _internal_mutable_xla_backend_extra_options();
|
|
}
|
|
|
|
// -------------------------------------------------------------------
|
|
|
|
// GpuCompilationEnvironment
|
|
|
|
// int64 dummy_flag = 1;
|
|
inline void GpuCompilationEnvironment::clear_dummy_flag() {
|
|
_impl_.dummy_flag_ = int64_t{0};
|
|
}
|
|
inline int64_t GpuCompilationEnvironment::_internal_dummy_flag() const {
|
|
return _impl_.dummy_flag_;
|
|
}
|
|
inline int64_t GpuCompilationEnvironment::dummy_flag() const {
|
|
// @@protoc_insertion_point(field_get:xla.GpuCompilationEnvironment.dummy_flag)
|
|
return _internal_dummy_flag();
|
|
}
|
|
inline void GpuCompilationEnvironment::_internal_set_dummy_flag(int64_t value) {
|
|
|
|
_impl_.dummy_flag_ = value;
|
|
}
|
|
inline void GpuCompilationEnvironment::set_dummy_flag(int64_t value) {
|
|
_internal_set_dummy_flag(value);
|
|
// @@protoc_insertion_point(field_set:xla.GpuCompilationEnvironment.dummy_flag)
|
|
}
|
|
|
|
// -------------------------------------------------------------------
|
|
|
|
// ShardableValueUpdatePairProto
|
|
|
|
// int64 input_parameter_number = 1;
|
|
inline void ShardableValueUpdatePairProto::clear_input_parameter_number() {
|
|
_impl_.input_parameter_number_ = int64_t{0};
|
|
}
|
|
inline int64_t ShardableValueUpdatePairProto::_internal_input_parameter_number() const {
|
|
return _impl_.input_parameter_number_;
|
|
}
|
|
inline int64_t ShardableValueUpdatePairProto::input_parameter_number() const {
|
|
// @@protoc_insertion_point(field_get:xla.ShardableValueUpdatePairProto.input_parameter_number)
|
|
return _internal_input_parameter_number();
|
|
}
|
|
inline void ShardableValueUpdatePairProto::_internal_set_input_parameter_number(int64_t value) {
|
|
|
|
_impl_.input_parameter_number_ = value;
|
|
}
|
|
inline void ShardableValueUpdatePairProto::set_input_parameter_number(int64_t value) {
|
|
_internal_set_input_parameter_number(value);
|
|
// @@protoc_insertion_point(field_set:xla.ShardableValueUpdatePairProto.input_parameter_number)
|
|
}
|
|
|
|
// repeated int64 parameter_shape_index = 2;
|
|
inline int ShardableValueUpdatePairProto::_internal_parameter_shape_index_size() const {
|
|
return _impl_.parameter_shape_index_.size();
|
|
}
|
|
inline int ShardableValueUpdatePairProto::parameter_shape_index_size() const {
|
|
return _internal_parameter_shape_index_size();
|
|
}
|
|
inline void ShardableValueUpdatePairProto::clear_parameter_shape_index() {
|
|
_impl_.parameter_shape_index_.Clear();
|
|
}
|
|
inline int64_t ShardableValueUpdatePairProto::_internal_parameter_shape_index(int index) const {
|
|
return _impl_.parameter_shape_index_.Get(index);
|
|
}
|
|
inline int64_t ShardableValueUpdatePairProto::parameter_shape_index(int index) const {
|
|
// @@protoc_insertion_point(field_get:xla.ShardableValueUpdatePairProto.parameter_shape_index)
|
|
return _internal_parameter_shape_index(index);
|
|
}
|
|
inline void ShardableValueUpdatePairProto::set_parameter_shape_index(int index, int64_t value) {
|
|
_impl_.parameter_shape_index_.Set(index, value);
|
|
// @@protoc_insertion_point(field_set:xla.ShardableValueUpdatePairProto.parameter_shape_index)
|
|
}
|
|
inline void ShardableValueUpdatePairProto::_internal_add_parameter_shape_index(int64_t value) {
|
|
_impl_.parameter_shape_index_.Add(value);
|
|
}
|
|
inline void ShardableValueUpdatePairProto::add_parameter_shape_index(int64_t value) {
|
|
_internal_add_parameter_shape_index(value);
|
|
// @@protoc_insertion_point(field_add:xla.ShardableValueUpdatePairProto.parameter_shape_index)
|
|
}
|
|
inline const ::PROTOBUF_NAMESPACE_ID::RepeatedField< int64_t >&
|
|
ShardableValueUpdatePairProto::_internal_parameter_shape_index() const {
|
|
return _impl_.parameter_shape_index_;
|
|
}
|
|
inline const ::PROTOBUF_NAMESPACE_ID::RepeatedField< int64_t >&
|
|
ShardableValueUpdatePairProto::parameter_shape_index() const {
|
|
// @@protoc_insertion_point(field_list:xla.ShardableValueUpdatePairProto.parameter_shape_index)
|
|
return _internal_parameter_shape_index();
|
|
}
|
|
inline ::PROTOBUF_NAMESPACE_ID::RepeatedField< int64_t >*
|
|
ShardableValueUpdatePairProto::_internal_mutable_parameter_shape_index() {
|
|
return &_impl_.parameter_shape_index_;
|
|
}
|
|
inline ::PROTOBUF_NAMESPACE_ID::RepeatedField< int64_t >*
|
|
ShardableValueUpdatePairProto::mutable_parameter_shape_index() {
|
|
// @@protoc_insertion_point(field_mutable_list:xla.ShardableValueUpdatePairProto.parameter_shape_index)
|
|
return _internal_mutable_parameter_shape_index();
|
|
}
|
|
|
|
// repeated int64 output_shape_index = 3;
|
|
inline int ShardableValueUpdatePairProto::_internal_output_shape_index_size() const {
|
|
return _impl_.output_shape_index_.size();
|
|
}
|
|
inline int ShardableValueUpdatePairProto::output_shape_index_size() const {
|
|
return _internal_output_shape_index_size();
|
|
}
|
|
inline void ShardableValueUpdatePairProto::clear_output_shape_index() {
|
|
_impl_.output_shape_index_.Clear();
|
|
}
|
|
inline int64_t ShardableValueUpdatePairProto::_internal_output_shape_index(int index) const {
|
|
return _impl_.output_shape_index_.Get(index);
|
|
}
|
|
inline int64_t ShardableValueUpdatePairProto::output_shape_index(int index) const {
|
|
// @@protoc_insertion_point(field_get:xla.ShardableValueUpdatePairProto.output_shape_index)
|
|
return _internal_output_shape_index(index);
|
|
}
|
|
inline void ShardableValueUpdatePairProto::set_output_shape_index(int index, int64_t value) {
|
|
_impl_.output_shape_index_.Set(index, value);
|
|
// @@protoc_insertion_point(field_set:xla.ShardableValueUpdatePairProto.output_shape_index)
|
|
}
|
|
inline void ShardableValueUpdatePairProto::_internal_add_output_shape_index(int64_t value) {
|
|
_impl_.output_shape_index_.Add(value);
|
|
}
|
|
inline void ShardableValueUpdatePairProto::add_output_shape_index(int64_t value) {
|
|
_internal_add_output_shape_index(value);
|
|
// @@protoc_insertion_point(field_add:xla.ShardableValueUpdatePairProto.output_shape_index)
|
|
}
|
|
inline const ::PROTOBUF_NAMESPACE_ID::RepeatedField< int64_t >&
|
|
ShardableValueUpdatePairProto::_internal_output_shape_index() const {
|
|
return _impl_.output_shape_index_;
|
|
}
|
|
inline const ::PROTOBUF_NAMESPACE_ID::RepeatedField< int64_t >&
|
|
ShardableValueUpdatePairProto::output_shape_index() const {
|
|
// @@protoc_insertion_point(field_list:xla.ShardableValueUpdatePairProto.output_shape_index)
|
|
return _internal_output_shape_index();
|
|
}
|
|
inline ::PROTOBUF_NAMESPACE_ID::RepeatedField< int64_t >*
|
|
ShardableValueUpdatePairProto::_internal_mutable_output_shape_index() {
|
|
return &_impl_.output_shape_index_;
|
|
}
|
|
inline ::PROTOBUF_NAMESPACE_ID::RepeatedField< int64_t >*
|
|
ShardableValueUpdatePairProto::mutable_output_shape_index() {
|
|
// @@protoc_insertion_point(field_mutable_list:xla.ShardableValueUpdatePairProto.output_shape_index)
|
|
return _internal_mutable_output_shape_index();
|
|
}
|
|
|
|
// -------------------------------------------------------------------
|
|
|
|
// ExecutionOptions
|
|
|
|
// .xla.ShapeProto shape_with_output_layout = 2;
|
|
inline bool ExecutionOptions::_internal_has_shape_with_output_layout() const {
|
|
return this != internal_default_instance() && _impl_.shape_with_output_layout_ != nullptr;
|
|
}
|
|
inline bool ExecutionOptions::has_shape_with_output_layout() const {
|
|
return _internal_has_shape_with_output_layout();
|
|
}
|
|
inline const ::xla::ShapeProto& ExecutionOptions::_internal_shape_with_output_layout() const {
|
|
const ::xla::ShapeProto* p = _impl_.shape_with_output_layout_;
|
|
return p != nullptr ? *p : reinterpret_cast<const ::xla::ShapeProto&>(
|
|
::xla::_ShapeProto_default_instance_);
|
|
}
|
|
inline const ::xla::ShapeProto& ExecutionOptions::shape_with_output_layout() const {
|
|
// @@protoc_insertion_point(field_get:xla.ExecutionOptions.shape_with_output_layout)
|
|
return _internal_shape_with_output_layout();
|
|
}
|
|
inline void ExecutionOptions::unsafe_arena_set_allocated_shape_with_output_layout(
|
|
::xla::ShapeProto* shape_with_output_layout) {
|
|
if (GetArenaForAllocation() == nullptr) {
|
|
delete reinterpret_cast<::PROTOBUF_NAMESPACE_ID::MessageLite*>(_impl_.shape_with_output_layout_);
|
|
}
|
|
_impl_.shape_with_output_layout_ = shape_with_output_layout;
|
|
if (shape_with_output_layout) {
|
|
|
|
} else {
|
|
|
|
}
|
|
// @@protoc_insertion_point(field_unsafe_arena_set_allocated:xla.ExecutionOptions.shape_with_output_layout)
|
|
}
|
|
inline ::xla::ShapeProto* ExecutionOptions::release_shape_with_output_layout() {
|
|
|
|
::xla::ShapeProto* temp = _impl_.shape_with_output_layout_;
|
|
_impl_.shape_with_output_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::ShapeProto* ExecutionOptions::unsafe_arena_release_shape_with_output_layout() {
|
|
// @@protoc_insertion_point(field_release:xla.ExecutionOptions.shape_with_output_layout)
|
|
|
|
::xla::ShapeProto* temp = _impl_.shape_with_output_layout_;
|
|
_impl_.shape_with_output_layout_ = nullptr;
|
|
return temp;
|
|
}
|
|
inline ::xla::ShapeProto* ExecutionOptions::_internal_mutable_shape_with_output_layout() {
|
|
|
|
if (_impl_.shape_with_output_layout_ == nullptr) {
|
|
auto* p = CreateMaybeMessage<::xla::ShapeProto>(GetArenaForAllocation());
|
|
_impl_.shape_with_output_layout_ = p;
|
|
}
|
|
return _impl_.shape_with_output_layout_;
|
|
}
|
|
inline ::xla::ShapeProto* ExecutionOptions::mutable_shape_with_output_layout() {
|
|
::xla::ShapeProto* _msg = _internal_mutable_shape_with_output_layout();
|
|
// @@protoc_insertion_point(field_mutable:xla.ExecutionOptions.shape_with_output_layout)
|
|
return _msg;
|
|
}
|
|
inline void ExecutionOptions::set_allocated_shape_with_output_layout(::xla::ShapeProto* shape_with_output_layout) {
|
|
::PROTOBUF_NAMESPACE_ID::Arena* message_arena = GetArenaForAllocation();
|
|
if (message_arena == nullptr) {
|
|
delete reinterpret_cast< ::PROTOBUF_NAMESPACE_ID::MessageLite*>(_impl_.shape_with_output_layout_);
|
|
}
|
|
if (shape_with_output_layout) {
|
|
::PROTOBUF_NAMESPACE_ID::Arena* submessage_arena =
|
|
::PROTOBUF_NAMESPACE_ID::Arena::InternalGetOwningArena(
|
|
reinterpret_cast<::PROTOBUF_NAMESPACE_ID::MessageLite*>(shape_with_output_layout));
|
|
if (message_arena != submessage_arena) {
|
|
shape_with_output_layout = ::PROTOBUF_NAMESPACE_ID::internal::GetOwnedMessage(
|
|
message_arena, shape_with_output_layout, submessage_arena);
|
|
}
|
|
|
|
} else {
|
|
|
|
}
|
|
_impl_.shape_with_output_layout_ = shape_with_output_layout;
|
|
// @@protoc_insertion_point(field_set_allocated:xla.ExecutionOptions.shape_with_output_layout)
|
|
}
|
|
|
|
// uint64 seed = 3;
|
|
inline void ExecutionOptions::clear_seed() {
|
|
_impl_.seed_ = uint64_t{0u};
|
|
}
|
|
inline uint64_t ExecutionOptions::_internal_seed() const {
|
|
return _impl_.seed_;
|
|
}
|
|
inline uint64_t ExecutionOptions::seed() const {
|
|
// @@protoc_insertion_point(field_get:xla.ExecutionOptions.seed)
|
|
return _internal_seed();
|
|
}
|
|
inline void ExecutionOptions::_internal_set_seed(uint64_t value) {
|
|
|
|
_impl_.seed_ = value;
|
|
}
|
|
inline void ExecutionOptions::set_seed(uint64_t value) {
|
|
_internal_set_seed(value);
|
|
// @@protoc_insertion_point(field_set:xla.ExecutionOptions.seed)
|
|
}
|
|
|
|
// .xla.DebugOptions debug_options = 4;
|
|
inline bool ExecutionOptions::_internal_has_debug_options() const {
|
|
return this != internal_default_instance() && _impl_.debug_options_ != nullptr;
|
|
}
|
|
inline bool ExecutionOptions::has_debug_options() const {
|
|
return _internal_has_debug_options();
|
|
}
|
|
inline void ExecutionOptions::clear_debug_options() {
|
|
if (GetArenaForAllocation() == nullptr && _impl_.debug_options_ != nullptr) {
|
|
delete _impl_.debug_options_;
|
|
}
|
|
_impl_.debug_options_ = nullptr;
|
|
}
|
|
inline const ::xla::DebugOptions& ExecutionOptions::_internal_debug_options() const {
|
|
const ::xla::DebugOptions* p = _impl_.debug_options_;
|
|
return p != nullptr ? *p : reinterpret_cast<const ::xla::DebugOptions&>(
|
|
::xla::_DebugOptions_default_instance_);
|
|
}
|
|
inline const ::xla::DebugOptions& ExecutionOptions::debug_options() const {
|
|
// @@protoc_insertion_point(field_get:xla.ExecutionOptions.debug_options)
|
|
return _internal_debug_options();
|
|
}
|
|
inline void ExecutionOptions::unsafe_arena_set_allocated_debug_options(
|
|
::xla::DebugOptions* debug_options) {
|
|
if (GetArenaForAllocation() == nullptr) {
|
|
delete reinterpret_cast<::PROTOBUF_NAMESPACE_ID::MessageLite*>(_impl_.debug_options_);
|
|
}
|
|
_impl_.debug_options_ = debug_options;
|
|
if (debug_options) {
|
|
|
|
} else {
|
|
|
|
}
|
|
// @@protoc_insertion_point(field_unsafe_arena_set_allocated:xla.ExecutionOptions.debug_options)
|
|
}
|
|
inline ::xla::DebugOptions* ExecutionOptions::release_debug_options() {
|
|
|
|
::xla::DebugOptions* temp = _impl_.debug_options_;
|
|
_impl_.debug_options_ = 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::DebugOptions* ExecutionOptions::unsafe_arena_release_debug_options() {
|
|
// @@protoc_insertion_point(field_release:xla.ExecutionOptions.debug_options)
|
|
|
|
::xla::DebugOptions* temp = _impl_.debug_options_;
|
|
_impl_.debug_options_ = nullptr;
|
|
return temp;
|
|
}
|
|
inline ::xla::DebugOptions* ExecutionOptions::_internal_mutable_debug_options() {
|
|
|
|
if (_impl_.debug_options_ == nullptr) {
|
|
auto* p = CreateMaybeMessage<::xla::DebugOptions>(GetArenaForAllocation());
|
|
_impl_.debug_options_ = p;
|
|
}
|
|
return _impl_.debug_options_;
|
|
}
|
|
inline ::xla::DebugOptions* ExecutionOptions::mutable_debug_options() {
|
|
::xla::DebugOptions* _msg = _internal_mutable_debug_options();
|
|
// @@protoc_insertion_point(field_mutable:xla.ExecutionOptions.debug_options)
|
|
return _msg;
|
|
}
|
|
inline void ExecutionOptions::set_allocated_debug_options(::xla::DebugOptions* debug_options) {
|
|
::PROTOBUF_NAMESPACE_ID::Arena* message_arena = GetArenaForAllocation();
|
|
if (message_arena == nullptr) {
|
|
delete _impl_.debug_options_;
|
|
}
|
|
if (debug_options) {
|
|
::PROTOBUF_NAMESPACE_ID::Arena* submessage_arena =
|
|
::PROTOBUF_NAMESPACE_ID::Arena::InternalGetOwningArena(debug_options);
|
|
if (message_arena != submessage_arena) {
|
|
debug_options = ::PROTOBUF_NAMESPACE_ID::internal::GetOwnedMessage(
|
|
message_arena, debug_options, submessage_arena);
|
|
}
|
|
|
|
} else {
|
|
|
|
}
|
|
_impl_.debug_options_ = debug_options;
|
|
// @@protoc_insertion_point(field_set_allocated:xla.ExecutionOptions.debug_options)
|
|
}
|
|
|
|
// repeated .xla.DeviceHandle device_handles = 5;
|
|
inline int ExecutionOptions::_internal_device_handles_size() const {
|
|
return _impl_.device_handles_.size();
|
|
}
|
|
inline int ExecutionOptions::device_handles_size() const {
|
|
return _internal_device_handles_size();
|
|
}
|
|
inline ::xla::DeviceHandle* ExecutionOptions::mutable_device_handles(int index) {
|
|
// @@protoc_insertion_point(field_mutable:xla.ExecutionOptions.device_handles)
|
|
return _impl_.device_handles_.Mutable(index);
|
|
}
|
|
inline ::PROTOBUF_NAMESPACE_ID::RepeatedPtrField< ::xla::DeviceHandle >*
|
|
ExecutionOptions::mutable_device_handles() {
|
|
// @@protoc_insertion_point(field_mutable_list:xla.ExecutionOptions.device_handles)
|
|
return &_impl_.device_handles_;
|
|
}
|
|
inline const ::xla::DeviceHandle& ExecutionOptions::_internal_device_handles(int index) const {
|
|
return _impl_.device_handles_.Get(index);
|
|
}
|
|
inline const ::xla::DeviceHandle& ExecutionOptions::device_handles(int index) const {
|
|
// @@protoc_insertion_point(field_get:xla.ExecutionOptions.device_handles)
|
|
return _internal_device_handles(index);
|
|
}
|
|
inline ::xla::DeviceHandle* ExecutionOptions::_internal_add_device_handles() {
|
|
return _impl_.device_handles_.Add();
|
|
}
|
|
inline ::xla::DeviceHandle* ExecutionOptions::add_device_handles() {
|
|
::xla::DeviceHandle* _add = _internal_add_device_handles();
|
|
// @@protoc_insertion_point(field_add:xla.ExecutionOptions.device_handles)
|
|
return _add;
|
|
}
|
|
inline const ::PROTOBUF_NAMESPACE_ID::RepeatedPtrField< ::xla::DeviceHandle >&
|
|
ExecutionOptions::device_handles() const {
|
|
// @@protoc_insertion_point(field_list:xla.ExecutionOptions.device_handles)
|
|
return _impl_.device_handles_;
|
|
}
|
|
|
|
// int32 num_replicas = 6;
|
|
inline void ExecutionOptions::clear_num_replicas() {
|
|
_impl_.num_replicas_ = 0;
|
|
}
|
|
inline int32_t ExecutionOptions::_internal_num_replicas() const {
|
|
return _impl_.num_replicas_;
|
|
}
|
|
inline int32_t ExecutionOptions::num_replicas() const {
|
|
// @@protoc_insertion_point(field_get:xla.ExecutionOptions.num_replicas)
|
|
return _internal_num_replicas();
|
|
}
|
|
inline void ExecutionOptions::_internal_set_num_replicas(int32_t value) {
|
|
|
|
_impl_.num_replicas_ = value;
|
|
}
|
|
inline void ExecutionOptions::set_num_replicas(int32_t value) {
|
|
_internal_set_num_replicas(value);
|
|
// @@protoc_insertion_point(field_set:xla.ExecutionOptions.num_replicas)
|
|
}
|
|
|
|
// .xla.DeviceAssignmentProto device_assignment = 7;
|
|
inline bool ExecutionOptions::_internal_has_device_assignment() const {
|
|
return this != internal_default_instance() && _impl_.device_assignment_ != nullptr;
|
|
}
|
|
inline bool ExecutionOptions::has_device_assignment() const {
|
|
return _internal_has_device_assignment();
|
|
}
|
|
inline const ::xla::DeviceAssignmentProto& ExecutionOptions::_internal_device_assignment() const {
|
|
const ::xla::DeviceAssignmentProto* p = _impl_.device_assignment_;
|
|
return p != nullptr ? *p : reinterpret_cast<const ::xla::DeviceAssignmentProto&>(
|
|
::xla::_DeviceAssignmentProto_default_instance_);
|
|
}
|
|
inline const ::xla::DeviceAssignmentProto& ExecutionOptions::device_assignment() const {
|
|
// @@protoc_insertion_point(field_get:xla.ExecutionOptions.device_assignment)
|
|
return _internal_device_assignment();
|
|
}
|
|
inline void ExecutionOptions::unsafe_arena_set_allocated_device_assignment(
|
|
::xla::DeviceAssignmentProto* device_assignment) {
|
|
if (GetArenaForAllocation() == nullptr) {
|
|
delete reinterpret_cast<::PROTOBUF_NAMESPACE_ID::MessageLite*>(_impl_.device_assignment_);
|
|
}
|
|
_impl_.device_assignment_ = device_assignment;
|
|
if (device_assignment) {
|
|
|
|
} else {
|
|
|
|
}
|
|
// @@protoc_insertion_point(field_unsafe_arena_set_allocated:xla.ExecutionOptions.device_assignment)
|
|
}
|
|
inline ::xla::DeviceAssignmentProto* ExecutionOptions::release_device_assignment() {
|
|
|
|
::xla::DeviceAssignmentProto* temp = _impl_.device_assignment_;
|
|
_impl_.device_assignment_ = 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::DeviceAssignmentProto* ExecutionOptions::unsafe_arena_release_device_assignment() {
|
|
// @@protoc_insertion_point(field_release:xla.ExecutionOptions.device_assignment)
|
|
|
|
::xla::DeviceAssignmentProto* temp = _impl_.device_assignment_;
|
|
_impl_.device_assignment_ = nullptr;
|
|
return temp;
|
|
}
|
|
inline ::xla::DeviceAssignmentProto* ExecutionOptions::_internal_mutable_device_assignment() {
|
|
|
|
if (_impl_.device_assignment_ == nullptr) {
|
|
auto* p = CreateMaybeMessage<::xla::DeviceAssignmentProto>(GetArenaForAllocation());
|
|
_impl_.device_assignment_ = p;
|
|
}
|
|
return _impl_.device_assignment_;
|
|
}
|
|
inline ::xla::DeviceAssignmentProto* ExecutionOptions::mutable_device_assignment() {
|
|
::xla::DeviceAssignmentProto* _msg = _internal_mutable_device_assignment();
|
|
// @@protoc_insertion_point(field_mutable:xla.ExecutionOptions.device_assignment)
|
|
return _msg;
|
|
}
|
|
inline void ExecutionOptions::set_allocated_device_assignment(::xla::DeviceAssignmentProto* device_assignment) {
|
|
::PROTOBUF_NAMESPACE_ID::Arena* message_arena = GetArenaForAllocation();
|
|
if (message_arena == nullptr) {
|
|
delete reinterpret_cast< ::PROTOBUF_NAMESPACE_ID::MessageLite*>(_impl_.device_assignment_);
|
|
}
|
|
if (device_assignment) {
|
|
::PROTOBUF_NAMESPACE_ID::Arena* submessage_arena =
|
|
::PROTOBUF_NAMESPACE_ID::Arena::InternalGetOwningArena(
|
|
reinterpret_cast<::PROTOBUF_NAMESPACE_ID::MessageLite*>(device_assignment));
|
|
if (message_arena != submessage_arena) {
|
|
device_assignment = ::PROTOBUF_NAMESPACE_ID::internal::GetOwnedMessage(
|
|
message_arena, device_assignment, submessage_arena);
|
|
}
|
|
|
|
} else {
|
|
|
|
}
|
|
_impl_.device_assignment_ = device_assignment;
|
|
// @@protoc_insertion_point(field_set_allocated:xla.ExecutionOptions.device_assignment)
|
|
}
|
|
|
|
// bool alias_passthrough_params = 8;
|
|
inline void ExecutionOptions::clear_alias_passthrough_params() {
|
|
_impl_.alias_passthrough_params_ = false;
|
|
}
|
|
inline bool ExecutionOptions::_internal_alias_passthrough_params() const {
|
|
return _impl_.alias_passthrough_params_;
|
|
}
|
|
inline bool ExecutionOptions::alias_passthrough_params() const {
|
|
// @@protoc_insertion_point(field_get:xla.ExecutionOptions.alias_passthrough_params)
|
|
return _internal_alias_passthrough_params();
|
|
}
|
|
inline void ExecutionOptions::_internal_set_alias_passthrough_params(bool value) {
|
|
|
|
_impl_.alias_passthrough_params_ = value;
|
|
}
|
|
inline void ExecutionOptions::set_alias_passthrough_params(bool value) {
|
|
_internal_set_alias_passthrough_params(value);
|
|
// @@protoc_insertion_point(field_set:xla.ExecutionOptions.alias_passthrough_params)
|
|
}
|
|
|
|
// int32 num_partitions = 9;
|
|
inline void ExecutionOptions::clear_num_partitions() {
|
|
_impl_.num_partitions_ = 0;
|
|
}
|
|
inline int32_t ExecutionOptions::_internal_num_partitions() const {
|
|
return _impl_.num_partitions_;
|
|
}
|
|
inline int32_t ExecutionOptions::num_partitions() const {
|
|
// @@protoc_insertion_point(field_get:xla.ExecutionOptions.num_partitions)
|
|
return _internal_num_partitions();
|
|
}
|
|
inline void ExecutionOptions::_internal_set_num_partitions(int32_t value) {
|
|
|
|
_impl_.num_partitions_ = value;
|
|
}
|
|
inline void ExecutionOptions::set_num_partitions(int32_t value) {
|
|
_internal_set_num_partitions(value);
|
|
// @@protoc_insertion_point(field_set:xla.ExecutionOptions.num_partitions)
|
|
}
|
|
|
|
// int32 launch_id = 10;
|
|
inline void ExecutionOptions::clear_launch_id() {
|
|
_impl_.launch_id_ = 0;
|
|
}
|
|
inline int32_t ExecutionOptions::_internal_launch_id() const {
|
|
return _impl_.launch_id_;
|
|
}
|
|
inline int32_t ExecutionOptions::launch_id() const {
|
|
// @@protoc_insertion_point(field_get:xla.ExecutionOptions.launch_id)
|
|
return _internal_launch_id();
|
|
}
|
|
inline void ExecutionOptions::_internal_set_launch_id(int32_t value) {
|
|
|
|
_impl_.launch_id_ = value;
|
|
}
|
|
inline void ExecutionOptions::set_launch_id(int32_t value) {
|
|
_internal_set_launch_id(value);
|
|
// @@protoc_insertion_point(field_set:xla.ExecutionOptions.launch_id)
|
|
}
|
|
|
|
// bool use_spmd_partitioning = 11;
|
|
inline void ExecutionOptions::clear_use_spmd_partitioning() {
|
|
_impl_.use_spmd_partitioning_ = false;
|
|
}
|
|
inline bool ExecutionOptions::_internal_use_spmd_partitioning() const {
|
|
return _impl_.use_spmd_partitioning_;
|
|
}
|
|
inline bool ExecutionOptions::use_spmd_partitioning() const {
|
|
// @@protoc_insertion_point(field_get:xla.ExecutionOptions.use_spmd_partitioning)
|
|
return _internal_use_spmd_partitioning();
|
|
}
|
|
inline void ExecutionOptions::_internal_set_use_spmd_partitioning(bool value) {
|
|
|
|
_impl_.use_spmd_partitioning_ = value;
|
|
}
|
|
inline void ExecutionOptions::set_use_spmd_partitioning(bool value) {
|
|
_internal_set_use_spmd_partitioning(value);
|
|
// @@protoc_insertion_point(field_set:xla.ExecutionOptions.use_spmd_partitioning)
|
|
}
|
|
|
|
// bool use_auto_spmd_partitioning = 15;
|
|
inline void ExecutionOptions::clear_use_auto_spmd_partitioning() {
|
|
_impl_.use_auto_spmd_partitioning_ = false;
|
|
}
|
|
inline bool ExecutionOptions::_internal_use_auto_spmd_partitioning() const {
|
|
return _impl_.use_auto_spmd_partitioning_;
|
|
}
|
|
inline bool ExecutionOptions::use_auto_spmd_partitioning() const {
|
|
// @@protoc_insertion_point(field_get:xla.ExecutionOptions.use_auto_spmd_partitioning)
|
|
return _internal_use_auto_spmd_partitioning();
|
|
}
|
|
inline void ExecutionOptions::_internal_set_use_auto_spmd_partitioning(bool value) {
|
|
|
|
_impl_.use_auto_spmd_partitioning_ = value;
|
|
}
|
|
inline void ExecutionOptions::set_use_auto_spmd_partitioning(bool value) {
|
|
_internal_set_use_auto_spmd_partitioning(value);
|
|
// @@protoc_insertion_point(field_set:xla.ExecutionOptions.use_auto_spmd_partitioning)
|
|
}
|
|
|
|
// repeated int64 auto_spmd_partitioning_mesh_shape = 16;
|
|
inline int ExecutionOptions::_internal_auto_spmd_partitioning_mesh_shape_size() const {
|
|
return _impl_.auto_spmd_partitioning_mesh_shape_.size();
|
|
}
|
|
inline int ExecutionOptions::auto_spmd_partitioning_mesh_shape_size() const {
|
|
return _internal_auto_spmd_partitioning_mesh_shape_size();
|
|
}
|
|
inline void ExecutionOptions::clear_auto_spmd_partitioning_mesh_shape() {
|
|
_impl_.auto_spmd_partitioning_mesh_shape_.Clear();
|
|
}
|
|
inline int64_t ExecutionOptions::_internal_auto_spmd_partitioning_mesh_shape(int index) const {
|
|
return _impl_.auto_spmd_partitioning_mesh_shape_.Get(index);
|
|
}
|
|
inline int64_t ExecutionOptions::auto_spmd_partitioning_mesh_shape(int index) const {
|
|
// @@protoc_insertion_point(field_get:xla.ExecutionOptions.auto_spmd_partitioning_mesh_shape)
|
|
return _internal_auto_spmd_partitioning_mesh_shape(index);
|
|
}
|
|
inline void ExecutionOptions::set_auto_spmd_partitioning_mesh_shape(int index, int64_t value) {
|
|
_impl_.auto_spmd_partitioning_mesh_shape_.Set(index, value);
|
|
// @@protoc_insertion_point(field_set:xla.ExecutionOptions.auto_spmd_partitioning_mesh_shape)
|
|
}
|
|
inline void ExecutionOptions::_internal_add_auto_spmd_partitioning_mesh_shape(int64_t value) {
|
|
_impl_.auto_spmd_partitioning_mesh_shape_.Add(value);
|
|
}
|
|
inline void ExecutionOptions::add_auto_spmd_partitioning_mesh_shape(int64_t value) {
|
|
_internal_add_auto_spmd_partitioning_mesh_shape(value);
|
|
// @@protoc_insertion_point(field_add:xla.ExecutionOptions.auto_spmd_partitioning_mesh_shape)
|
|
}
|
|
inline const ::PROTOBUF_NAMESPACE_ID::RepeatedField< int64_t >&
|
|
ExecutionOptions::_internal_auto_spmd_partitioning_mesh_shape() const {
|
|
return _impl_.auto_spmd_partitioning_mesh_shape_;
|
|
}
|
|
inline const ::PROTOBUF_NAMESPACE_ID::RepeatedField< int64_t >&
|
|
ExecutionOptions::auto_spmd_partitioning_mesh_shape() const {
|
|
// @@protoc_insertion_point(field_list:xla.ExecutionOptions.auto_spmd_partitioning_mesh_shape)
|
|
return _internal_auto_spmd_partitioning_mesh_shape();
|
|
}
|
|
inline ::PROTOBUF_NAMESPACE_ID::RepeatedField< int64_t >*
|
|
ExecutionOptions::_internal_mutable_auto_spmd_partitioning_mesh_shape() {
|
|
return &_impl_.auto_spmd_partitioning_mesh_shape_;
|
|
}
|
|
inline ::PROTOBUF_NAMESPACE_ID::RepeatedField< int64_t >*
|
|
ExecutionOptions::mutable_auto_spmd_partitioning_mesh_shape() {
|
|
// @@protoc_insertion_point(field_mutable_list:xla.ExecutionOptions.auto_spmd_partitioning_mesh_shape)
|
|
return _internal_mutable_auto_spmd_partitioning_mesh_shape();
|
|
}
|
|
|
|
// repeated int64 auto_spmd_partitioning_mesh_ids = 17;
|
|
inline int ExecutionOptions::_internal_auto_spmd_partitioning_mesh_ids_size() const {
|
|
return _impl_.auto_spmd_partitioning_mesh_ids_.size();
|
|
}
|
|
inline int ExecutionOptions::auto_spmd_partitioning_mesh_ids_size() const {
|
|
return _internal_auto_spmd_partitioning_mesh_ids_size();
|
|
}
|
|
inline void ExecutionOptions::clear_auto_spmd_partitioning_mesh_ids() {
|
|
_impl_.auto_spmd_partitioning_mesh_ids_.Clear();
|
|
}
|
|
inline int64_t ExecutionOptions::_internal_auto_spmd_partitioning_mesh_ids(int index) const {
|
|
return _impl_.auto_spmd_partitioning_mesh_ids_.Get(index);
|
|
}
|
|
inline int64_t ExecutionOptions::auto_spmd_partitioning_mesh_ids(int index) const {
|
|
// @@protoc_insertion_point(field_get:xla.ExecutionOptions.auto_spmd_partitioning_mesh_ids)
|
|
return _internal_auto_spmd_partitioning_mesh_ids(index);
|
|
}
|
|
inline void ExecutionOptions::set_auto_spmd_partitioning_mesh_ids(int index, int64_t value) {
|
|
_impl_.auto_spmd_partitioning_mesh_ids_.Set(index, value);
|
|
// @@protoc_insertion_point(field_set:xla.ExecutionOptions.auto_spmd_partitioning_mesh_ids)
|
|
}
|
|
inline void ExecutionOptions::_internal_add_auto_spmd_partitioning_mesh_ids(int64_t value) {
|
|
_impl_.auto_spmd_partitioning_mesh_ids_.Add(value);
|
|
}
|
|
inline void ExecutionOptions::add_auto_spmd_partitioning_mesh_ids(int64_t value) {
|
|
_internal_add_auto_spmd_partitioning_mesh_ids(value);
|
|
// @@protoc_insertion_point(field_add:xla.ExecutionOptions.auto_spmd_partitioning_mesh_ids)
|
|
}
|
|
inline const ::PROTOBUF_NAMESPACE_ID::RepeatedField< int64_t >&
|
|
ExecutionOptions::_internal_auto_spmd_partitioning_mesh_ids() const {
|
|
return _impl_.auto_spmd_partitioning_mesh_ids_;
|
|
}
|
|
inline const ::PROTOBUF_NAMESPACE_ID::RepeatedField< int64_t >&
|
|
ExecutionOptions::auto_spmd_partitioning_mesh_ids() const {
|
|
// @@protoc_insertion_point(field_list:xla.ExecutionOptions.auto_spmd_partitioning_mesh_ids)
|
|
return _internal_auto_spmd_partitioning_mesh_ids();
|
|
}
|
|
inline ::PROTOBUF_NAMESPACE_ID::RepeatedField< int64_t >*
|
|
ExecutionOptions::_internal_mutable_auto_spmd_partitioning_mesh_ids() {
|
|
return &_impl_.auto_spmd_partitioning_mesh_ids_;
|
|
}
|
|
inline ::PROTOBUF_NAMESPACE_ID::RepeatedField< int64_t >*
|
|
ExecutionOptions::mutable_auto_spmd_partitioning_mesh_ids() {
|
|
// @@protoc_insertion_point(field_mutable_list:xla.ExecutionOptions.auto_spmd_partitioning_mesh_ids)
|
|
return _internal_mutable_auto_spmd_partitioning_mesh_ids();
|
|
}
|
|
|
|
// bool deduplicate_hlo = 12;
|
|
inline void ExecutionOptions::clear_deduplicate_hlo() {
|
|
_impl_.deduplicate_hlo_ = false;
|
|
}
|
|
inline bool ExecutionOptions::_internal_deduplicate_hlo() const {
|
|
return _impl_.deduplicate_hlo_;
|
|
}
|
|
inline bool ExecutionOptions::deduplicate_hlo() const {
|
|
// @@protoc_insertion_point(field_get:xla.ExecutionOptions.deduplicate_hlo)
|
|
return _internal_deduplicate_hlo();
|
|
}
|
|
inline void ExecutionOptions::_internal_set_deduplicate_hlo(bool value) {
|
|
|
|
_impl_.deduplicate_hlo_ = value;
|
|
}
|
|
inline void ExecutionOptions::set_deduplicate_hlo(bool value) {
|
|
_internal_set_deduplicate_hlo(value);
|
|
// @@protoc_insertion_point(field_set:xla.ExecutionOptions.deduplicate_hlo)
|
|
}
|
|
|
|
// repeated bool allow_spmd_sharding_propagation_to_output = 14;
|
|
inline int ExecutionOptions::_internal_allow_spmd_sharding_propagation_to_output_size() const {
|
|
return _impl_.allow_spmd_sharding_propagation_to_output_.size();
|
|
}
|
|
inline int ExecutionOptions::allow_spmd_sharding_propagation_to_output_size() const {
|
|
return _internal_allow_spmd_sharding_propagation_to_output_size();
|
|
}
|
|
inline void ExecutionOptions::clear_allow_spmd_sharding_propagation_to_output() {
|
|
_impl_.allow_spmd_sharding_propagation_to_output_.Clear();
|
|
}
|
|
inline bool ExecutionOptions::_internal_allow_spmd_sharding_propagation_to_output(int index) const {
|
|
return _impl_.allow_spmd_sharding_propagation_to_output_.Get(index);
|
|
}
|
|
inline bool ExecutionOptions::allow_spmd_sharding_propagation_to_output(int index) const {
|
|
// @@protoc_insertion_point(field_get:xla.ExecutionOptions.allow_spmd_sharding_propagation_to_output)
|
|
return _internal_allow_spmd_sharding_propagation_to_output(index);
|
|
}
|
|
inline void ExecutionOptions::set_allow_spmd_sharding_propagation_to_output(int index, bool value) {
|
|
_impl_.allow_spmd_sharding_propagation_to_output_.Set(index, value);
|
|
// @@protoc_insertion_point(field_set:xla.ExecutionOptions.allow_spmd_sharding_propagation_to_output)
|
|
}
|
|
inline void ExecutionOptions::_internal_add_allow_spmd_sharding_propagation_to_output(bool value) {
|
|
_impl_.allow_spmd_sharding_propagation_to_output_.Add(value);
|
|
}
|
|
inline void ExecutionOptions::add_allow_spmd_sharding_propagation_to_output(bool value) {
|
|
_internal_add_allow_spmd_sharding_propagation_to_output(value);
|
|
// @@protoc_insertion_point(field_add:xla.ExecutionOptions.allow_spmd_sharding_propagation_to_output)
|
|
}
|
|
inline const ::PROTOBUF_NAMESPACE_ID::RepeatedField< bool >&
|
|
ExecutionOptions::_internal_allow_spmd_sharding_propagation_to_output() const {
|
|
return _impl_.allow_spmd_sharding_propagation_to_output_;
|
|
}
|
|
inline const ::PROTOBUF_NAMESPACE_ID::RepeatedField< bool >&
|
|
ExecutionOptions::allow_spmd_sharding_propagation_to_output() const {
|
|
// @@protoc_insertion_point(field_list:xla.ExecutionOptions.allow_spmd_sharding_propagation_to_output)
|
|
return _internal_allow_spmd_sharding_propagation_to_output();
|
|
}
|
|
inline ::PROTOBUF_NAMESPACE_ID::RepeatedField< bool >*
|
|
ExecutionOptions::_internal_mutable_allow_spmd_sharding_propagation_to_output() {
|
|
return &_impl_.allow_spmd_sharding_propagation_to_output_;
|
|
}
|
|
inline ::PROTOBUF_NAMESPACE_ID::RepeatedField< bool >*
|
|
ExecutionOptions::mutable_allow_spmd_sharding_propagation_to_output() {
|
|
// @@protoc_insertion_point(field_mutable_list:xla.ExecutionOptions.allow_spmd_sharding_propagation_to_output)
|
|
return _internal_mutable_allow_spmd_sharding_propagation_to_output();
|
|
}
|
|
|
|
// repeated bool param_requires_broadcast_via_collectives = 18;
|
|
inline int ExecutionOptions::_internal_param_requires_broadcast_via_collectives_size() const {
|
|
return _impl_.param_requires_broadcast_via_collectives_.size();
|
|
}
|
|
inline int ExecutionOptions::param_requires_broadcast_via_collectives_size() const {
|
|
return _internal_param_requires_broadcast_via_collectives_size();
|
|
}
|
|
inline void ExecutionOptions::clear_param_requires_broadcast_via_collectives() {
|
|
_impl_.param_requires_broadcast_via_collectives_.Clear();
|
|
}
|
|
inline bool ExecutionOptions::_internal_param_requires_broadcast_via_collectives(int index) const {
|
|
return _impl_.param_requires_broadcast_via_collectives_.Get(index);
|
|
}
|
|
inline bool ExecutionOptions::param_requires_broadcast_via_collectives(int index) const {
|
|
// @@protoc_insertion_point(field_get:xla.ExecutionOptions.param_requires_broadcast_via_collectives)
|
|
return _internal_param_requires_broadcast_via_collectives(index);
|
|
}
|
|
inline void ExecutionOptions::set_param_requires_broadcast_via_collectives(int index, bool value) {
|
|
_impl_.param_requires_broadcast_via_collectives_.Set(index, value);
|
|
// @@protoc_insertion_point(field_set:xla.ExecutionOptions.param_requires_broadcast_via_collectives)
|
|
}
|
|
inline void ExecutionOptions::_internal_add_param_requires_broadcast_via_collectives(bool value) {
|
|
_impl_.param_requires_broadcast_via_collectives_.Add(value);
|
|
}
|
|
inline void ExecutionOptions::add_param_requires_broadcast_via_collectives(bool value) {
|
|
_internal_add_param_requires_broadcast_via_collectives(value);
|
|
// @@protoc_insertion_point(field_add:xla.ExecutionOptions.param_requires_broadcast_via_collectives)
|
|
}
|
|
inline const ::PROTOBUF_NAMESPACE_ID::RepeatedField< bool >&
|
|
ExecutionOptions::_internal_param_requires_broadcast_via_collectives() const {
|
|
return _impl_.param_requires_broadcast_via_collectives_;
|
|
}
|
|
inline const ::PROTOBUF_NAMESPACE_ID::RepeatedField< bool >&
|
|
ExecutionOptions::param_requires_broadcast_via_collectives() const {
|
|
// @@protoc_insertion_point(field_list:xla.ExecutionOptions.param_requires_broadcast_via_collectives)
|
|
return _internal_param_requires_broadcast_via_collectives();
|
|
}
|
|
inline ::PROTOBUF_NAMESPACE_ID::RepeatedField< bool >*
|
|
ExecutionOptions::_internal_mutable_param_requires_broadcast_via_collectives() {
|
|
return &_impl_.param_requires_broadcast_via_collectives_;
|
|
}
|
|
inline ::PROTOBUF_NAMESPACE_ID::RepeatedField< bool >*
|
|
ExecutionOptions::mutable_param_requires_broadcast_via_collectives() {
|
|
// @@protoc_insertion_point(field_mutable_list:xla.ExecutionOptions.param_requires_broadcast_via_collectives)
|
|
return _internal_mutable_param_requires_broadcast_via_collectives();
|
|
}
|
|
|
|
// bool allow_separate_sharding_programs = 19;
|
|
inline void ExecutionOptions::clear_allow_separate_sharding_programs() {
|
|
_impl_.allow_separate_sharding_programs_ = false;
|
|
}
|
|
inline bool ExecutionOptions::_internal_allow_separate_sharding_programs() const {
|
|
return _impl_.allow_separate_sharding_programs_;
|
|
}
|
|
inline bool ExecutionOptions::allow_separate_sharding_programs() const {
|
|
// @@protoc_insertion_point(field_get:xla.ExecutionOptions.allow_separate_sharding_programs)
|
|
return _internal_allow_separate_sharding_programs();
|
|
}
|
|
inline void ExecutionOptions::_internal_set_allow_separate_sharding_programs(bool value) {
|
|
|
|
_impl_.allow_separate_sharding_programs_ = value;
|
|
}
|
|
inline void ExecutionOptions::set_allow_separate_sharding_programs(bool value) {
|
|
_internal_set_allow_separate_sharding_programs(value);
|
|
// @@protoc_insertion_point(field_set:xla.ExecutionOptions.allow_separate_sharding_programs)
|
|
}
|
|
|
|
// repeated .xla.ShardableValueUpdatePairProto shardable_value_update_pairs = 20;
|
|
inline int ExecutionOptions::_internal_shardable_value_update_pairs_size() const {
|
|
return _impl_.shardable_value_update_pairs_.size();
|
|
}
|
|
inline int ExecutionOptions::shardable_value_update_pairs_size() const {
|
|
return _internal_shardable_value_update_pairs_size();
|
|
}
|
|
inline void ExecutionOptions::clear_shardable_value_update_pairs() {
|
|
_impl_.shardable_value_update_pairs_.Clear();
|
|
}
|
|
inline ::xla::ShardableValueUpdatePairProto* ExecutionOptions::mutable_shardable_value_update_pairs(int index) {
|
|
// @@protoc_insertion_point(field_mutable:xla.ExecutionOptions.shardable_value_update_pairs)
|
|
return _impl_.shardable_value_update_pairs_.Mutable(index);
|
|
}
|
|
inline ::PROTOBUF_NAMESPACE_ID::RepeatedPtrField< ::xla::ShardableValueUpdatePairProto >*
|
|
ExecutionOptions::mutable_shardable_value_update_pairs() {
|
|
// @@protoc_insertion_point(field_mutable_list:xla.ExecutionOptions.shardable_value_update_pairs)
|
|
return &_impl_.shardable_value_update_pairs_;
|
|
}
|
|
inline const ::xla::ShardableValueUpdatePairProto& ExecutionOptions::_internal_shardable_value_update_pairs(int index) const {
|
|
return _impl_.shardable_value_update_pairs_.Get(index);
|
|
}
|
|
inline const ::xla::ShardableValueUpdatePairProto& ExecutionOptions::shardable_value_update_pairs(int index) const {
|
|
// @@protoc_insertion_point(field_get:xla.ExecutionOptions.shardable_value_update_pairs)
|
|
return _internal_shardable_value_update_pairs(index);
|
|
}
|
|
inline ::xla::ShardableValueUpdatePairProto* ExecutionOptions::_internal_add_shardable_value_update_pairs() {
|
|
return _impl_.shardable_value_update_pairs_.Add();
|
|
}
|
|
inline ::xla::ShardableValueUpdatePairProto* ExecutionOptions::add_shardable_value_update_pairs() {
|
|
::xla::ShardableValueUpdatePairProto* _add = _internal_add_shardable_value_update_pairs();
|
|
// @@protoc_insertion_point(field_add:xla.ExecutionOptions.shardable_value_update_pairs)
|
|
return _add;
|
|
}
|
|
inline const ::PROTOBUF_NAMESPACE_ID::RepeatedPtrField< ::xla::ShardableValueUpdatePairProto >&
|
|
ExecutionOptions::shardable_value_update_pairs() const {
|
|
// @@protoc_insertion_point(field_list:xla.ExecutionOptions.shardable_value_update_pairs)
|
|
return _impl_.shardable_value_update_pairs_;
|
|
}
|
|
|
|
// bytes fdo_profile = 21;
|
|
inline void ExecutionOptions::clear_fdo_profile() {
|
|
_impl_.fdo_profile_.ClearToEmpty();
|
|
}
|
|
inline const std::string& ExecutionOptions::fdo_profile() const {
|
|
// @@protoc_insertion_point(field_get:xla.ExecutionOptions.fdo_profile)
|
|
return _internal_fdo_profile();
|
|
}
|
|
template <typename ArgT0, typename... ArgT>
|
|
inline PROTOBUF_ALWAYS_INLINE
|
|
void ExecutionOptions::set_fdo_profile(ArgT0&& arg0, ArgT... args) {
|
|
|
|
_impl_.fdo_profile_.SetBytes(static_cast<ArgT0 &&>(arg0), args..., GetArenaForAllocation());
|
|
// @@protoc_insertion_point(field_set:xla.ExecutionOptions.fdo_profile)
|
|
}
|
|
inline std::string* ExecutionOptions::mutable_fdo_profile() {
|
|
std::string* _s = _internal_mutable_fdo_profile();
|
|
// @@protoc_insertion_point(field_mutable:xla.ExecutionOptions.fdo_profile)
|
|
return _s;
|
|
}
|
|
inline const std::string& ExecutionOptions::_internal_fdo_profile() const {
|
|
return _impl_.fdo_profile_.Get();
|
|
}
|
|
inline void ExecutionOptions::_internal_set_fdo_profile(const std::string& value) {
|
|
|
|
_impl_.fdo_profile_.Set(value, GetArenaForAllocation());
|
|
}
|
|
inline std::string* ExecutionOptions::_internal_mutable_fdo_profile() {
|
|
|
|
return _impl_.fdo_profile_.Mutable(GetArenaForAllocation());
|
|
}
|
|
inline std::string* ExecutionOptions::release_fdo_profile() {
|
|
// @@protoc_insertion_point(field_release:xla.ExecutionOptions.fdo_profile)
|
|
return _impl_.fdo_profile_.Release();
|
|
}
|
|
inline void ExecutionOptions::set_allocated_fdo_profile(std::string* fdo_profile) {
|
|
if (fdo_profile != nullptr) {
|
|
|
|
} else {
|
|
|
|
}
|
|
_impl_.fdo_profile_.SetAllocated(fdo_profile, GetArenaForAllocation());
|
|
#ifdef PROTOBUF_FORCE_COPY_DEFAULT_STRING
|
|
if (_impl_.fdo_profile_.IsDefault()) {
|
|
_impl_.fdo_profile_.Set("", GetArenaForAllocation());
|
|
}
|
|
#endif // PROTOBUF_FORCE_COPY_DEFAULT_STRING
|
|
// @@protoc_insertion_point(field_set_allocated:xla.ExecutionOptions.fdo_profile)
|
|
}
|
|
|
|
// int64 device_memory_size = 22;
|
|
inline void ExecutionOptions::clear_device_memory_size() {
|
|
_impl_.device_memory_size_ = int64_t{0};
|
|
}
|
|
inline int64_t ExecutionOptions::_internal_device_memory_size() const {
|
|
return _impl_.device_memory_size_;
|
|
}
|
|
inline int64_t ExecutionOptions::device_memory_size() const {
|
|
// @@protoc_insertion_point(field_get:xla.ExecutionOptions.device_memory_size)
|
|
return _internal_device_memory_size();
|
|
}
|
|
inline void ExecutionOptions::_internal_set_device_memory_size(int64_t value) {
|
|
|
|
_impl_.device_memory_size_ = value;
|
|
}
|
|
inline void ExecutionOptions::set_device_memory_size(int64_t value) {
|
|
_internal_set_device_memory_size(value);
|
|
// @@protoc_insertion_point(field_set:xla.ExecutionOptions.device_memory_size)
|
|
}
|
|
|
|
// -------------------------------------------------------------------
|
|
|
|
// HloModuleConfigProto_BoolList
|
|
|
|
// repeated bool vals = 1;
|
|
inline int HloModuleConfigProto_BoolList::_internal_vals_size() const {
|
|
return _impl_.vals_.size();
|
|
}
|
|
inline int HloModuleConfigProto_BoolList::vals_size() const {
|
|
return _internal_vals_size();
|
|
}
|
|
inline void HloModuleConfigProto_BoolList::clear_vals() {
|
|
_impl_.vals_.Clear();
|
|
}
|
|
inline bool HloModuleConfigProto_BoolList::_internal_vals(int index) const {
|
|
return _impl_.vals_.Get(index);
|
|
}
|
|
inline bool HloModuleConfigProto_BoolList::vals(int index) const {
|
|
// @@protoc_insertion_point(field_get:xla.HloModuleConfigProto.BoolList.vals)
|
|
return _internal_vals(index);
|
|
}
|
|
inline void HloModuleConfigProto_BoolList::set_vals(int index, bool value) {
|
|
_impl_.vals_.Set(index, value);
|
|
// @@protoc_insertion_point(field_set:xla.HloModuleConfigProto.BoolList.vals)
|
|
}
|
|
inline void HloModuleConfigProto_BoolList::_internal_add_vals(bool value) {
|
|
_impl_.vals_.Add(value);
|
|
}
|
|
inline void HloModuleConfigProto_BoolList::add_vals(bool value) {
|
|
_internal_add_vals(value);
|
|
// @@protoc_insertion_point(field_add:xla.HloModuleConfigProto.BoolList.vals)
|
|
}
|
|
inline const ::PROTOBUF_NAMESPACE_ID::RepeatedField< bool >&
|
|
HloModuleConfigProto_BoolList::_internal_vals() const {
|
|
return _impl_.vals_;
|
|
}
|
|
inline const ::PROTOBUF_NAMESPACE_ID::RepeatedField< bool >&
|
|
HloModuleConfigProto_BoolList::vals() const {
|
|
// @@protoc_insertion_point(field_list:xla.HloModuleConfigProto.BoolList.vals)
|
|
return _internal_vals();
|
|
}
|
|
inline ::PROTOBUF_NAMESPACE_ID::RepeatedField< bool >*
|
|
HloModuleConfigProto_BoolList::_internal_mutable_vals() {
|
|
return &_impl_.vals_;
|
|
}
|
|
inline ::PROTOBUF_NAMESPACE_ID::RepeatedField< bool >*
|
|
HloModuleConfigProto_BoolList::mutable_vals() {
|
|
// @@protoc_insertion_point(field_mutable_list:xla.HloModuleConfigProto.BoolList.vals)
|
|
return _internal_mutable_vals();
|
|
}
|
|
|
|
// -------------------------------------------------------------------
|
|
|
|
// HloModuleConfigProto_Int64List
|
|
|
|
// repeated int64 vals = 1;
|
|
inline int HloModuleConfigProto_Int64List::_internal_vals_size() const {
|
|
return _impl_.vals_.size();
|
|
}
|
|
inline int HloModuleConfigProto_Int64List::vals_size() const {
|
|
return _internal_vals_size();
|
|
}
|
|
inline void HloModuleConfigProto_Int64List::clear_vals() {
|
|
_impl_.vals_.Clear();
|
|
}
|
|
inline int64_t HloModuleConfigProto_Int64List::_internal_vals(int index) const {
|
|
return _impl_.vals_.Get(index);
|
|
}
|
|
inline int64_t HloModuleConfigProto_Int64List::vals(int index) const {
|
|
// @@protoc_insertion_point(field_get:xla.HloModuleConfigProto.Int64List.vals)
|
|
return _internal_vals(index);
|
|
}
|
|
inline void HloModuleConfigProto_Int64List::set_vals(int index, int64_t value) {
|
|
_impl_.vals_.Set(index, value);
|
|
// @@protoc_insertion_point(field_set:xla.HloModuleConfigProto.Int64List.vals)
|
|
}
|
|
inline void HloModuleConfigProto_Int64List::_internal_add_vals(int64_t value) {
|
|
_impl_.vals_.Add(value);
|
|
}
|
|
inline void HloModuleConfigProto_Int64List::add_vals(int64_t value) {
|
|
_internal_add_vals(value);
|
|
// @@protoc_insertion_point(field_add:xla.HloModuleConfigProto.Int64List.vals)
|
|
}
|
|
inline const ::PROTOBUF_NAMESPACE_ID::RepeatedField< int64_t >&
|
|
HloModuleConfigProto_Int64List::_internal_vals() const {
|
|
return _impl_.vals_;
|
|
}
|
|
inline const ::PROTOBUF_NAMESPACE_ID::RepeatedField< int64_t >&
|
|
HloModuleConfigProto_Int64List::vals() const {
|
|
// @@protoc_insertion_point(field_list:xla.HloModuleConfigProto.Int64List.vals)
|
|
return _internal_vals();
|
|
}
|
|
inline ::PROTOBUF_NAMESPACE_ID::RepeatedField< int64_t >*
|
|
HloModuleConfigProto_Int64List::_internal_mutable_vals() {
|
|
return &_impl_.vals_;
|
|
}
|
|
inline ::PROTOBUF_NAMESPACE_ID::RepeatedField< int64_t >*
|
|
HloModuleConfigProto_Int64List::mutable_vals() {
|
|
// @@protoc_insertion_point(field_mutable_list:xla.HloModuleConfigProto.Int64List.vals)
|
|
return _internal_mutable_vals();
|
|
}
|
|
|
|
// -------------------------------------------------------------------
|
|
|
|
// HloModuleConfigProto_Int64ListList
|
|
|
|
// repeated .xla.HloModuleConfigProto.Int64List lists = 1;
|
|
inline int HloModuleConfigProto_Int64ListList::_internal_lists_size() const {
|
|
return _impl_.lists_.size();
|
|
}
|
|
inline int HloModuleConfigProto_Int64ListList::lists_size() const {
|
|
return _internal_lists_size();
|
|
}
|
|
inline void HloModuleConfigProto_Int64ListList::clear_lists() {
|
|
_impl_.lists_.Clear();
|
|
}
|
|
inline ::xla::HloModuleConfigProto_Int64List* HloModuleConfigProto_Int64ListList::mutable_lists(int index) {
|
|
// @@protoc_insertion_point(field_mutable:xla.HloModuleConfigProto.Int64ListList.lists)
|
|
return _impl_.lists_.Mutable(index);
|
|
}
|
|
inline ::PROTOBUF_NAMESPACE_ID::RepeatedPtrField< ::xla::HloModuleConfigProto_Int64List >*
|
|
HloModuleConfigProto_Int64ListList::mutable_lists() {
|
|
// @@protoc_insertion_point(field_mutable_list:xla.HloModuleConfigProto.Int64ListList.lists)
|
|
return &_impl_.lists_;
|
|
}
|
|
inline const ::xla::HloModuleConfigProto_Int64List& HloModuleConfigProto_Int64ListList::_internal_lists(int index) const {
|
|
return _impl_.lists_.Get(index);
|
|
}
|
|
inline const ::xla::HloModuleConfigProto_Int64List& HloModuleConfigProto_Int64ListList::lists(int index) const {
|
|
// @@protoc_insertion_point(field_get:xla.HloModuleConfigProto.Int64ListList.lists)
|
|
return _internal_lists(index);
|
|
}
|
|
inline ::xla::HloModuleConfigProto_Int64List* HloModuleConfigProto_Int64ListList::_internal_add_lists() {
|
|
return _impl_.lists_.Add();
|
|
}
|
|
inline ::xla::HloModuleConfigProto_Int64List* HloModuleConfigProto_Int64ListList::add_lists() {
|
|
::xla::HloModuleConfigProto_Int64List* _add = _internal_add_lists();
|
|
// @@protoc_insertion_point(field_add:xla.HloModuleConfigProto.Int64ListList.lists)
|
|
return _add;
|
|
}
|
|
inline const ::PROTOBUF_NAMESPACE_ID::RepeatedPtrField< ::xla::HloModuleConfigProto_Int64List >&
|
|
HloModuleConfigProto_Int64ListList::lists() const {
|
|
// @@protoc_insertion_point(field_list:xla.HloModuleConfigProto.Int64ListList.lists)
|
|
return _impl_.lists_;
|
|
}
|
|
|
|
// -------------------------------------------------------------------
|
|
|
|
// -------------------------------------------------------------------
|
|
|
|
// -------------------------------------------------------------------
|
|
|
|
// HloModuleConfigProto
|
|
|
|
// .xla.ProgramShapeProto entry_computation_layout = 1;
|
|
inline bool HloModuleConfigProto::_internal_has_entry_computation_layout() const {
|
|
return this != internal_default_instance() && _impl_.entry_computation_layout_ != nullptr;
|
|
}
|
|
inline bool HloModuleConfigProto::has_entry_computation_layout() const {
|
|
return _internal_has_entry_computation_layout();
|
|
}
|
|
inline const ::xla::ProgramShapeProto& HloModuleConfigProto::_internal_entry_computation_layout() const {
|
|
const ::xla::ProgramShapeProto* p = _impl_.entry_computation_layout_;
|
|
return p != nullptr ? *p : reinterpret_cast<const ::xla::ProgramShapeProto&>(
|
|
::xla::_ProgramShapeProto_default_instance_);
|
|
}
|
|
inline const ::xla::ProgramShapeProto& HloModuleConfigProto::entry_computation_layout() const {
|
|
// @@protoc_insertion_point(field_get:xla.HloModuleConfigProto.entry_computation_layout)
|
|
return _internal_entry_computation_layout();
|
|
}
|
|
inline void HloModuleConfigProto::unsafe_arena_set_allocated_entry_computation_layout(
|
|
::xla::ProgramShapeProto* entry_computation_layout) {
|
|
if (GetArenaForAllocation() == nullptr) {
|
|
delete reinterpret_cast<::PROTOBUF_NAMESPACE_ID::MessageLite*>(_impl_.entry_computation_layout_);
|
|
}
|
|
_impl_.entry_computation_layout_ = entry_computation_layout;
|
|
if (entry_computation_layout) {
|
|
|
|
} else {
|
|
|
|
}
|
|
// @@protoc_insertion_point(field_unsafe_arena_set_allocated:xla.HloModuleConfigProto.entry_computation_layout)
|
|
}
|
|
inline ::xla::ProgramShapeProto* HloModuleConfigProto::release_entry_computation_layout() {
|
|
|
|
::xla::ProgramShapeProto* temp = _impl_.entry_computation_layout_;
|
|
_impl_.entry_computation_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::ProgramShapeProto* HloModuleConfigProto::unsafe_arena_release_entry_computation_layout() {
|
|
// @@protoc_insertion_point(field_release:xla.HloModuleConfigProto.entry_computation_layout)
|
|
|
|
::xla::ProgramShapeProto* temp = _impl_.entry_computation_layout_;
|
|
_impl_.entry_computation_layout_ = nullptr;
|
|
return temp;
|
|
}
|
|
inline ::xla::ProgramShapeProto* HloModuleConfigProto::_internal_mutable_entry_computation_layout() {
|
|
|
|
if (_impl_.entry_computation_layout_ == nullptr) {
|
|
auto* p = CreateMaybeMessage<::xla::ProgramShapeProto>(GetArenaForAllocation());
|
|
_impl_.entry_computation_layout_ = p;
|
|
}
|
|
return _impl_.entry_computation_layout_;
|
|
}
|
|
inline ::xla::ProgramShapeProto* HloModuleConfigProto::mutable_entry_computation_layout() {
|
|
::xla::ProgramShapeProto* _msg = _internal_mutable_entry_computation_layout();
|
|
// @@protoc_insertion_point(field_mutable:xla.HloModuleConfigProto.entry_computation_layout)
|
|
return _msg;
|
|
}
|
|
inline void HloModuleConfigProto::set_allocated_entry_computation_layout(::xla::ProgramShapeProto* entry_computation_layout) {
|
|
::PROTOBUF_NAMESPACE_ID::Arena* message_arena = GetArenaForAllocation();
|
|
if (message_arena == nullptr) {
|
|
delete reinterpret_cast< ::PROTOBUF_NAMESPACE_ID::MessageLite*>(_impl_.entry_computation_layout_);
|
|
}
|
|
if (entry_computation_layout) {
|
|
::PROTOBUF_NAMESPACE_ID::Arena* submessage_arena =
|
|
::PROTOBUF_NAMESPACE_ID::Arena::InternalGetOwningArena(
|
|
reinterpret_cast<::PROTOBUF_NAMESPACE_ID::MessageLite*>(entry_computation_layout));
|
|
if (message_arena != submessage_arena) {
|
|
entry_computation_layout = ::PROTOBUF_NAMESPACE_ID::internal::GetOwnedMessage(
|
|
message_arena, entry_computation_layout, submessage_arena);
|
|
}
|
|
|
|
} else {
|
|
|
|
}
|
|
_impl_.entry_computation_layout_ = entry_computation_layout;
|
|
// @@protoc_insertion_point(field_set_allocated:xla.HloModuleConfigProto.entry_computation_layout)
|
|
}
|
|
|
|
// uint64 seed = 2;
|
|
inline void HloModuleConfigProto::clear_seed() {
|
|
_impl_.seed_ = uint64_t{0u};
|
|
}
|
|
inline uint64_t HloModuleConfigProto::_internal_seed() const {
|
|
return _impl_.seed_;
|
|
}
|
|
inline uint64_t HloModuleConfigProto::seed() const {
|
|
// @@protoc_insertion_point(field_get:xla.HloModuleConfigProto.seed)
|
|
return _internal_seed();
|
|
}
|
|
inline void HloModuleConfigProto::_internal_set_seed(uint64_t value) {
|
|
|
|
_impl_.seed_ = value;
|
|
}
|
|
inline void HloModuleConfigProto::set_seed(uint64_t value) {
|
|
_internal_set_seed(value);
|
|
// @@protoc_insertion_point(field_set:xla.HloModuleConfigProto.seed)
|
|
}
|
|
|
|
// int32 launch_id = 3;
|
|
inline void HloModuleConfigProto::clear_launch_id() {
|
|
_impl_.launch_id_ = 0;
|
|
}
|
|
inline int32_t HloModuleConfigProto::_internal_launch_id() const {
|
|
return _impl_.launch_id_;
|
|
}
|
|
inline int32_t HloModuleConfigProto::launch_id() const {
|
|
// @@protoc_insertion_point(field_get:xla.HloModuleConfigProto.launch_id)
|
|
return _internal_launch_id();
|
|
}
|
|
inline void HloModuleConfigProto::_internal_set_launch_id(int32_t value) {
|
|
|
|
_impl_.launch_id_ = value;
|
|
}
|
|
inline void HloModuleConfigProto::set_launch_id(int32_t value) {
|
|
_internal_set_launch_id(value);
|
|
// @@protoc_insertion_point(field_set:xla.HloModuleConfigProto.launch_id)
|
|
}
|
|
|
|
// int64 replica_count = 4;
|
|
inline void HloModuleConfigProto::clear_replica_count() {
|
|
_impl_.replica_count_ = int64_t{0};
|
|
}
|
|
inline int64_t HloModuleConfigProto::_internal_replica_count() const {
|
|
return _impl_.replica_count_;
|
|
}
|
|
inline int64_t HloModuleConfigProto::replica_count() const {
|
|
// @@protoc_insertion_point(field_get:xla.HloModuleConfigProto.replica_count)
|
|
return _internal_replica_count();
|
|
}
|
|
inline void HloModuleConfigProto::_internal_set_replica_count(int64_t value) {
|
|
|
|
_impl_.replica_count_ = value;
|
|
}
|
|
inline void HloModuleConfigProto::set_replica_count(int64_t value) {
|
|
_internal_set_replica_count(value);
|
|
// @@protoc_insertion_point(field_set:xla.HloModuleConfigProto.replica_count)
|
|
}
|
|
|
|
// int64 num_partitions = 5;
|
|
inline void HloModuleConfigProto::clear_num_partitions() {
|
|
_impl_.num_partitions_ = int64_t{0};
|
|
}
|
|
inline int64_t HloModuleConfigProto::_internal_num_partitions() const {
|
|
return _impl_.num_partitions_;
|
|
}
|
|
inline int64_t HloModuleConfigProto::num_partitions() const {
|
|
// @@protoc_insertion_point(field_get:xla.HloModuleConfigProto.num_partitions)
|
|
return _internal_num_partitions();
|
|
}
|
|
inline void HloModuleConfigProto::_internal_set_num_partitions(int64_t value) {
|
|
|
|
_impl_.num_partitions_ = value;
|
|
}
|
|
inline void HloModuleConfigProto::set_num_partitions(int64_t value) {
|
|
_internal_set_num_partitions(value);
|
|
// @@protoc_insertion_point(field_set:xla.HloModuleConfigProto.num_partitions)
|
|
}
|
|
|
|
// repeated bool param_requires_broadcast_via_collectives = 6;
|
|
inline int HloModuleConfigProto::_internal_param_requires_broadcast_via_collectives_size() const {
|
|
return _impl_.param_requires_broadcast_via_collectives_.size();
|
|
}
|
|
inline int HloModuleConfigProto::param_requires_broadcast_via_collectives_size() const {
|
|
return _internal_param_requires_broadcast_via_collectives_size();
|
|
}
|
|
inline void HloModuleConfigProto::clear_param_requires_broadcast_via_collectives() {
|
|
_impl_.param_requires_broadcast_via_collectives_.Clear();
|
|
}
|
|
inline bool HloModuleConfigProto::_internal_param_requires_broadcast_via_collectives(int index) const {
|
|
return _impl_.param_requires_broadcast_via_collectives_.Get(index);
|
|
}
|
|
inline bool HloModuleConfigProto::param_requires_broadcast_via_collectives(int index) const {
|
|
// @@protoc_insertion_point(field_get:xla.HloModuleConfigProto.param_requires_broadcast_via_collectives)
|
|
return _internal_param_requires_broadcast_via_collectives(index);
|
|
}
|
|
inline void HloModuleConfigProto::set_param_requires_broadcast_via_collectives(int index, bool value) {
|
|
_impl_.param_requires_broadcast_via_collectives_.Set(index, value);
|
|
// @@protoc_insertion_point(field_set:xla.HloModuleConfigProto.param_requires_broadcast_via_collectives)
|
|
}
|
|
inline void HloModuleConfigProto::_internal_add_param_requires_broadcast_via_collectives(bool value) {
|
|
_impl_.param_requires_broadcast_via_collectives_.Add(value);
|
|
}
|
|
inline void HloModuleConfigProto::add_param_requires_broadcast_via_collectives(bool value) {
|
|
_internal_add_param_requires_broadcast_via_collectives(value);
|
|
// @@protoc_insertion_point(field_add:xla.HloModuleConfigProto.param_requires_broadcast_via_collectives)
|
|
}
|
|
inline const ::PROTOBUF_NAMESPACE_ID::RepeatedField< bool >&
|
|
HloModuleConfigProto::_internal_param_requires_broadcast_via_collectives() const {
|
|
return _impl_.param_requires_broadcast_via_collectives_;
|
|
}
|
|
inline const ::PROTOBUF_NAMESPACE_ID::RepeatedField< bool >&
|
|
HloModuleConfigProto::param_requires_broadcast_via_collectives() const {
|
|
// @@protoc_insertion_point(field_list:xla.HloModuleConfigProto.param_requires_broadcast_via_collectives)
|
|
return _internal_param_requires_broadcast_via_collectives();
|
|
}
|
|
inline ::PROTOBUF_NAMESPACE_ID::RepeatedField< bool >*
|
|
HloModuleConfigProto::_internal_mutable_param_requires_broadcast_via_collectives() {
|
|
return &_impl_.param_requires_broadcast_via_collectives_;
|
|
}
|
|
inline ::PROTOBUF_NAMESPACE_ID::RepeatedField< bool >*
|
|
HloModuleConfigProto::mutable_param_requires_broadcast_via_collectives() {
|
|
// @@protoc_insertion_point(field_mutable_list:xla.HloModuleConfigProto.param_requires_broadcast_via_collectives)
|
|
return _internal_mutable_param_requires_broadcast_via_collectives();
|
|
}
|
|
|
|
// bool use_spmd_partitioning = 7;
|
|
inline void HloModuleConfigProto::clear_use_spmd_partitioning() {
|
|
_impl_.use_spmd_partitioning_ = false;
|
|
}
|
|
inline bool HloModuleConfigProto::_internal_use_spmd_partitioning() const {
|
|
return _impl_.use_spmd_partitioning_;
|
|
}
|
|
inline bool HloModuleConfigProto::use_spmd_partitioning() const {
|
|
// @@protoc_insertion_point(field_get:xla.HloModuleConfigProto.use_spmd_partitioning)
|
|
return _internal_use_spmd_partitioning();
|
|
}
|
|
inline void HloModuleConfigProto::_internal_set_use_spmd_partitioning(bool value) {
|
|
|
|
_impl_.use_spmd_partitioning_ = value;
|
|
}
|
|
inline void HloModuleConfigProto::set_use_spmd_partitioning(bool value) {
|
|
_internal_set_use_spmd_partitioning(value);
|
|
// @@protoc_insertion_point(field_set:xla.HloModuleConfigProto.use_spmd_partitioning)
|
|
}
|
|
|
|
// bool use_auto_spmd_partitioning = 8;
|
|
inline void HloModuleConfigProto::clear_use_auto_spmd_partitioning() {
|
|
_impl_.use_auto_spmd_partitioning_ = false;
|
|
}
|
|
inline bool HloModuleConfigProto::_internal_use_auto_spmd_partitioning() const {
|
|
return _impl_.use_auto_spmd_partitioning_;
|
|
}
|
|
inline bool HloModuleConfigProto::use_auto_spmd_partitioning() const {
|
|
// @@protoc_insertion_point(field_get:xla.HloModuleConfigProto.use_auto_spmd_partitioning)
|
|
return _internal_use_auto_spmd_partitioning();
|
|
}
|
|
inline void HloModuleConfigProto::_internal_set_use_auto_spmd_partitioning(bool value) {
|
|
|
|
_impl_.use_auto_spmd_partitioning_ = value;
|
|
}
|
|
inline void HloModuleConfigProto::set_use_auto_spmd_partitioning(bool value) {
|
|
_internal_set_use_auto_spmd_partitioning(value);
|
|
// @@protoc_insertion_point(field_set:xla.HloModuleConfigProto.use_auto_spmd_partitioning)
|
|
}
|
|
|
|
// repeated int64 auto_spmd_partitioning_mesh_shape = 9;
|
|
inline int HloModuleConfigProto::_internal_auto_spmd_partitioning_mesh_shape_size() const {
|
|
return _impl_.auto_spmd_partitioning_mesh_shape_.size();
|
|
}
|
|
inline int HloModuleConfigProto::auto_spmd_partitioning_mesh_shape_size() const {
|
|
return _internal_auto_spmd_partitioning_mesh_shape_size();
|
|
}
|
|
inline void HloModuleConfigProto::clear_auto_spmd_partitioning_mesh_shape() {
|
|
_impl_.auto_spmd_partitioning_mesh_shape_.Clear();
|
|
}
|
|
inline int64_t HloModuleConfigProto::_internal_auto_spmd_partitioning_mesh_shape(int index) const {
|
|
return _impl_.auto_spmd_partitioning_mesh_shape_.Get(index);
|
|
}
|
|
inline int64_t HloModuleConfigProto::auto_spmd_partitioning_mesh_shape(int index) const {
|
|
// @@protoc_insertion_point(field_get:xla.HloModuleConfigProto.auto_spmd_partitioning_mesh_shape)
|
|
return _internal_auto_spmd_partitioning_mesh_shape(index);
|
|
}
|
|
inline void HloModuleConfigProto::set_auto_spmd_partitioning_mesh_shape(int index, int64_t value) {
|
|
_impl_.auto_spmd_partitioning_mesh_shape_.Set(index, value);
|
|
// @@protoc_insertion_point(field_set:xla.HloModuleConfigProto.auto_spmd_partitioning_mesh_shape)
|
|
}
|
|
inline void HloModuleConfigProto::_internal_add_auto_spmd_partitioning_mesh_shape(int64_t value) {
|
|
_impl_.auto_spmd_partitioning_mesh_shape_.Add(value);
|
|
}
|
|
inline void HloModuleConfigProto::add_auto_spmd_partitioning_mesh_shape(int64_t value) {
|
|
_internal_add_auto_spmd_partitioning_mesh_shape(value);
|
|
// @@protoc_insertion_point(field_add:xla.HloModuleConfigProto.auto_spmd_partitioning_mesh_shape)
|
|
}
|
|
inline const ::PROTOBUF_NAMESPACE_ID::RepeatedField< int64_t >&
|
|
HloModuleConfigProto::_internal_auto_spmd_partitioning_mesh_shape() const {
|
|
return _impl_.auto_spmd_partitioning_mesh_shape_;
|
|
}
|
|
inline const ::PROTOBUF_NAMESPACE_ID::RepeatedField< int64_t >&
|
|
HloModuleConfigProto::auto_spmd_partitioning_mesh_shape() const {
|
|
// @@protoc_insertion_point(field_list:xla.HloModuleConfigProto.auto_spmd_partitioning_mesh_shape)
|
|
return _internal_auto_spmd_partitioning_mesh_shape();
|
|
}
|
|
inline ::PROTOBUF_NAMESPACE_ID::RepeatedField< int64_t >*
|
|
HloModuleConfigProto::_internal_mutable_auto_spmd_partitioning_mesh_shape() {
|
|
return &_impl_.auto_spmd_partitioning_mesh_shape_;
|
|
}
|
|
inline ::PROTOBUF_NAMESPACE_ID::RepeatedField< int64_t >*
|
|
HloModuleConfigProto::mutable_auto_spmd_partitioning_mesh_shape() {
|
|
// @@protoc_insertion_point(field_mutable_list:xla.HloModuleConfigProto.auto_spmd_partitioning_mesh_shape)
|
|
return _internal_mutable_auto_spmd_partitioning_mesh_shape();
|
|
}
|
|
|
|
// repeated int64 auto_spmd_partitioning_mesh_ids = 10;
|
|
inline int HloModuleConfigProto::_internal_auto_spmd_partitioning_mesh_ids_size() const {
|
|
return _impl_.auto_spmd_partitioning_mesh_ids_.size();
|
|
}
|
|
inline int HloModuleConfigProto::auto_spmd_partitioning_mesh_ids_size() const {
|
|
return _internal_auto_spmd_partitioning_mesh_ids_size();
|
|
}
|
|
inline void HloModuleConfigProto::clear_auto_spmd_partitioning_mesh_ids() {
|
|
_impl_.auto_spmd_partitioning_mesh_ids_.Clear();
|
|
}
|
|
inline int64_t HloModuleConfigProto::_internal_auto_spmd_partitioning_mesh_ids(int index) const {
|
|
return _impl_.auto_spmd_partitioning_mesh_ids_.Get(index);
|
|
}
|
|
inline int64_t HloModuleConfigProto::auto_spmd_partitioning_mesh_ids(int index) const {
|
|
// @@protoc_insertion_point(field_get:xla.HloModuleConfigProto.auto_spmd_partitioning_mesh_ids)
|
|
return _internal_auto_spmd_partitioning_mesh_ids(index);
|
|
}
|
|
inline void HloModuleConfigProto::set_auto_spmd_partitioning_mesh_ids(int index, int64_t value) {
|
|
_impl_.auto_spmd_partitioning_mesh_ids_.Set(index, value);
|
|
// @@protoc_insertion_point(field_set:xla.HloModuleConfigProto.auto_spmd_partitioning_mesh_ids)
|
|
}
|
|
inline void HloModuleConfigProto::_internal_add_auto_spmd_partitioning_mesh_ids(int64_t value) {
|
|
_impl_.auto_spmd_partitioning_mesh_ids_.Add(value);
|
|
}
|
|
inline void HloModuleConfigProto::add_auto_spmd_partitioning_mesh_ids(int64_t value) {
|
|
_internal_add_auto_spmd_partitioning_mesh_ids(value);
|
|
// @@protoc_insertion_point(field_add:xla.HloModuleConfigProto.auto_spmd_partitioning_mesh_ids)
|
|
}
|
|
inline const ::PROTOBUF_NAMESPACE_ID::RepeatedField< int64_t >&
|
|
HloModuleConfigProto::_internal_auto_spmd_partitioning_mesh_ids() const {
|
|
return _impl_.auto_spmd_partitioning_mesh_ids_;
|
|
}
|
|
inline const ::PROTOBUF_NAMESPACE_ID::RepeatedField< int64_t >&
|
|
HloModuleConfigProto::auto_spmd_partitioning_mesh_ids() const {
|
|
// @@protoc_insertion_point(field_list:xla.HloModuleConfigProto.auto_spmd_partitioning_mesh_ids)
|
|
return _internal_auto_spmd_partitioning_mesh_ids();
|
|
}
|
|
inline ::PROTOBUF_NAMESPACE_ID::RepeatedField< int64_t >*
|
|
HloModuleConfigProto::_internal_mutable_auto_spmd_partitioning_mesh_ids() {
|
|
return &_impl_.auto_spmd_partitioning_mesh_ids_;
|
|
}
|
|
inline ::PROTOBUF_NAMESPACE_ID::RepeatedField< int64_t >*
|
|
HloModuleConfigProto::mutable_auto_spmd_partitioning_mesh_ids() {
|
|
// @@protoc_insertion_point(field_mutable_list:xla.HloModuleConfigProto.auto_spmd_partitioning_mesh_ids)
|
|
return _internal_mutable_auto_spmd_partitioning_mesh_ids();
|
|
}
|
|
|
|
// bool deduplicate_hlo = 11;
|
|
inline void HloModuleConfigProto::clear_deduplicate_hlo() {
|
|
_impl_.deduplicate_hlo_ = false;
|
|
}
|
|
inline bool HloModuleConfigProto::_internal_deduplicate_hlo() const {
|
|
return _impl_.deduplicate_hlo_;
|
|
}
|
|
inline bool HloModuleConfigProto::deduplicate_hlo() const {
|
|
// @@protoc_insertion_point(field_get:xla.HloModuleConfigProto.deduplicate_hlo)
|
|
return _internal_deduplicate_hlo();
|
|
}
|
|
inline void HloModuleConfigProto::_internal_set_deduplicate_hlo(bool value) {
|
|
|
|
_impl_.deduplicate_hlo_ = value;
|
|
}
|
|
inline void HloModuleConfigProto::set_deduplicate_hlo(bool value) {
|
|
_internal_set_deduplicate_hlo(value);
|
|
// @@protoc_insertion_point(field_set:xla.HloModuleConfigProto.deduplicate_hlo)
|
|
}
|
|
|
|
// int64 intra_op_parallelism_threads = 12;
|
|
inline void HloModuleConfigProto::clear_intra_op_parallelism_threads() {
|
|
_impl_.intra_op_parallelism_threads_ = int64_t{0};
|
|
}
|
|
inline int64_t HloModuleConfigProto::_internal_intra_op_parallelism_threads() const {
|
|
return _impl_.intra_op_parallelism_threads_;
|
|
}
|
|
inline int64_t HloModuleConfigProto::intra_op_parallelism_threads() const {
|
|
// @@protoc_insertion_point(field_get:xla.HloModuleConfigProto.intra_op_parallelism_threads)
|
|
return _internal_intra_op_parallelism_threads();
|
|
}
|
|
inline void HloModuleConfigProto::_internal_set_intra_op_parallelism_threads(int64_t value) {
|
|
|
|
_impl_.intra_op_parallelism_threads_ = value;
|
|
}
|
|
inline void HloModuleConfigProto::set_intra_op_parallelism_threads(int64_t value) {
|
|
_internal_set_intra_op_parallelism_threads(value);
|
|
// @@protoc_insertion_point(field_set:xla.HloModuleConfigProto.intra_op_parallelism_threads)
|
|
}
|
|
|
|
// string device_type = 13;
|
|
inline void HloModuleConfigProto::clear_device_type() {
|
|
_impl_.device_type_.ClearToEmpty();
|
|
}
|
|
inline const std::string& HloModuleConfigProto::device_type() const {
|
|
// @@protoc_insertion_point(field_get:xla.HloModuleConfigProto.device_type)
|
|
return _internal_device_type();
|
|
}
|
|
template <typename ArgT0, typename... ArgT>
|
|
inline PROTOBUF_ALWAYS_INLINE
|
|
void HloModuleConfigProto::set_device_type(ArgT0&& arg0, ArgT... args) {
|
|
|
|
_impl_.device_type_.Set(static_cast<ArgT0 &&>(arg0), args..., GetArenaForAllocation());
|
|
// @@protoc_insertion_point(field_set:xla.HloModuleConfigProto.device_type)
|
|
}
|
|
inline std::string* HloModuleConfigProto::mutable_device_type() {
|
|
std::string* _s = _internal_mutable_device_type();
|
|
// @@protoc_insertion_point(field_mutable:xla.HloModuleConfigProto.device_type)
|
|
return _s;
|
|
}
|
|
inline const std::string& HloModuleConfigProto::_internal_device_type() const {
|
|
return _impl_.device_type_.Get();
|
|
}
|
|
inline void HloModuleConfigProto::_internal_set_device_type(const std::string& value) {
|
|
|
|
_impl_.device_type_.Set(value, GetArenaForAllocation());
|
|
}
|
|
inline std::string* HloModuleConfigProto::_internal_mutable_device_type() {
|
|
|
|
return _impl_.device_type_.Mutable(GetArenaForAllocation());
|
|
}
|
|
inline std::string* HloModuleConfigProto::release_device_type() {
|
|
// @@protoc_insertion_point(field_release:xla.HloModuleConfigProto.device_type)
|
|
return _impl_.device_type_.Release();
|
|
}
|
|
inline void HloModuleConfigProto::set_allocated_device_type(std::string* device_type) {
|
|
if (device_type != nullptr) {
|
|
|
|
} else {
|
|
|
|
}
|
|
_impl_.device_type_.SetAllocated(device_type, GetArenaForAllocation());
|
|
#ifdef PROTOBUF_FORCE_COPY_DEFAULT_STRING
|
|
if (_impl_.device_type_.IsDefault()) {
|
|
_impl_.device_type_.Set("", GetArenaForAllocation());
|
|
}
|
|
#endif // PROTOBUF_FORCE_COPY_DEFAULT_STRING
|
|
// @@protoc_insertion_point(field_set_allocated:xla.HloModuleConfigProto.device_type)
|
|
}
|
|
|
|
// .xla.DebugOptions debug_options = 14;
|
|
inline bool HloModuleConfigProto::_internal_has_debug_options() const {
|
|
return this != internal_default_instance() && _impl_.debug_options_ != nullptr;
|
|
}
|
|
inline bool HloModuleConfigProto::has_debug_options() const {
|
|
return _internal_has_debug_options();
|
|
}
|
|
inline void HloModuleConfigProto::clear_debug_options() {
|
|
if (GetArenaForAllocation() == nullptr && _impl_.debug_options_ != nullptr) {
|
|
delete _impl_.debug_options_;
|
|
}
|
|
_impl_.debug_options_ = nullptr;
|
|
}
|
|
inline const ::xla::DebugOptions& HloModuleConfigProto::_internal_debug_options() const {
|
|
const ::xla::DebugOptions* p = _impl_.debug_options_;
|
|
return p != nullptr ? *p : reinterpret_cast<const ::xla::DebugOptions&>(
|
|
::xla::_DebugOptions_default_instance_);
|
|
}
|
|
inline const ::xla::DebugOptions& HloModuleConfigProto::debug_options() const {
|
|
// @@protoc_insertion_point(field_get:xla.HloModuleConfigProto.debug_options)
|
|
return _internal_debug_options();
|
|
}
|
|
inline void HloModuleConfigProto::unsafe_arena_set_allocated_debug_options(
|
|
::xla::DebugOptions* debug_options) {
|
|
if (GetArenaForAllocation() == nullptr) {
|
|
delete reinterpret_cast<::PROTOBUF_NAMESPACE_ID::MessageLite*>(_impl_.debug_options_);
|
|
}
|
|
_impl_.debug_options_ = debug_options;
|
|
if (debug_options) {
|
|
|
|
} else {
|
|
|
|
}
|
|
// @@protoc_insertion_point(field_unsafe_arena_set_allocated:xla.HloModuleConfigProto.debug_options)
|
|
}
|
|
inline ::xla::DebugOptions* HloModuleConfigProto::release_debug_options() {
|
|
|
|
::xla::DebugOptions* temp = _impl_.debug_options_;
|
|
_impl_.debug_options_ = 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::DebugOptions* HloModuleConfigProto::unsafe_arena_release_debug_options() {
|
|
// @@protoc_insertion_point(field_release:xla.HloModuleConfigProto.debug_options)
|
|
|
|
::xla::DebugOptions* temp = _impl_.debug_options_;
|
|
_impl_.debug_options_ = nullptr;
|
|
return temp;
|
|
}
|
|
inline ::xla::DebugOptions* HloModuleConfigProto::_internal_mutable_debug_options() {
|
|
|
|
if (_impl_.debug_options_ == nullptr) {
|
|
auto* p = CreateMaybeMessage<::xla::DebugOptions>(GetArenaForAllocation());
|
|
_impl_.debug_options_ = p;
|
|
}
|
|
return _impl_.debug_options_;
|
|
}
|
|
inline ::xla::DebugOptions* HloModuleConfigProto::mutable_debug_options() {
|
|
::xla::DebugOptions* _msg = _internal_mutable_debug_options();
|
|
// @@protoc_insertion_point(field_mutable:xla.HloModuleConfigProto.debug_options)
|
|
return _msg;
|
|
}
|
|
inline void HloModuleConfigProto::set_allocated_debug_options(::xla::DebugOptions* debug_options) {
|
|
::PROTOBUF_NAMESPACE_ID::Arena* message_arena = GetArenaForAllocation();
|
|
if (message_arena == nullptr) {
|
|
delete _impl_.debug_options_;
|
|
}
|
|
if (debug_options) {
|
|
::PROTOBUF_NAMESPACE_ID::Arena* submessage_arena =
|
|
::PROTOBUF_NAMESPACE_ID::Arena::InternalGetOwningArena(debug_options);
|
|
if (message_arena != submessage_arena) {
|
|
debug_options = ::PROTOBUF_NAMESPACE_ID::internal::GetOwnedMessage(
|
|
message_arena, debug_options, submessage_arena);
|
|
}
|
|
|
|
} else {
|
|
|
|
}
|
|
_impl_.debug_options_ = debug_options;
|
|
// @@protoc_insertion_point(field_set_allocated:xla.HloModuleConfigProto.debug_options)
|
|
}
|
|
|
|
// .xla.DeviceAssignmentProto static_device_assignment = 15;
|
|
inline bool HloModuleConfigProto::_internal_has_static_device_assignment() const {
|
|
return this != internal_default_instance() && _impl_.static_device_assignment_ != nullptr;
|
|
}
|
|
inline bool HloModuleConfigProto::has_static_device_assignment() const {
|
|
return _internal_has_static_device_assignment();
|
|
}
|
|
inline const ::xla::DeviceAssignmentProto& HloModuleConfigProto::_internal_static_device_assignment() const {
|
|
const ::xla::DeviceAssignmentProto* p = _impl_.static_device_assignment_;
|
|
return p != nullptr ? *p : reinterpret_cast<const ::xla::DeviceAssignmentProto&>(
|
|
::xla::_DeviceAssignmentProto_default_instance_);
|
|
}
|
|
inline const ::xla::DeviceAssignmentProto& HloModuleConfigProto::static_device_assignment() const {
|
|
// @@protoc_insertion_point(field_get:xla.HloModuleConfigProto.static_device_assignment)
|
|
return _internal_static_device_assignment();
|
|
}
|
|
inline void HloModuleConfigProto::unsafe_arena_set_allocated_static_device_assignment(
|
|
::xla::DeviceAssignmentProto* static_device_assignment) {
|
|
if (GetArenaForAllocation() == nullptr) {
|
|
delete reinterpret_cast<::PROTOBUF_NAMESPACE_ID::MessageLite*>(_impl_.static_device_assignment_);
|
|
}
|
|
_impl_.static_device_assignment_ = static_device_assignment;
|
|
if (static_device_assignment) {
|
|
|
|
} else {
|
|
|
|
}
|
|
// @@protoc_insertion_point(field_unsafe_arena_set_allocated:xla.HloModuleConfigProto.static_device_assignment)
|
|
}
|
|
inline ::xla::DeviceAssignmentProto* HloModuleConfigProto::release_static_device_assignment() {
|
|
|
|
::xla::DeviceAssignmentProto* temp = _impl_.static_device_assignment_;
|
|
_impl_.static_device_assignment_ = 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::DeviceAssignmentProto* HloModuleConfigProto::unsafe_arena_release_static_device_assignment() {
|
|
// @@protoc_insertion_point(field_release:xla.HloModuleConfigProto.static_device_assignment)
|
|
|
|
::xla::DeviceAssignmentProto* temp = _impl_.static_device_assignment_;
|
|
_impl_.static_device_assignment_ = nullptr;
|
|
return temp;
|
|
}
|
|
inline ::xla::DeviceAssignmentProto* HloModuleConfigProto::_internal_mutable_static_device_assignment() {
|
|
|
|
if (_impl_.static_device_assignment_ == nullptr) {
|
|
auto* p = CreateMaybeMessage<::xla::DeviceAssignmentProto>(GetArenaForAllocation());
|
|
_impl_.static_device_assignment_ = p;
|
|
}
|
|
return _impl_.static_device_assignment_;
|
|
}
|
|
inline ::xla::DeviceAssignmentProto* HloModuleConfigProto::mutable_static_device_assignment() {
|
|
::xla::DeviceAssignmentProto* _msg = _internal_mutable_static_device_assignment();
|
|
// @@protoc_insertion_point(field_mutable:xla.HloModuleConfigProto.static_device_assignment)
|
|
return _msg;
|
|
}
|
|
inline void HloModuleConfigProto::set_allocated_static_device_assignment(::xla::DeviceAssignmentProto* static_device_assignment) {
|
|
::PROTOBUF_NAMESPACE_ID::Arena* message_arena = GetArenaForAllocation();
|
|
if (message_arena == nullptr) {
|
|
delete reinterpret_cast< ::PROTOBUF_NAMESPACE_ID::MessageLite*>(_impl_.static_device_assignment_);
|
|
}
|
|
if (static_device_assignment) {
|
|
::PROTOBUF_NAMESPACE_ID::Arena* submessage_arena =
|
|
::PROTOBUF_NAMESPACE_ID::Arena::InternalGetOwningArena(
|
|
reinterpret_cast<::PROTOBUF_NAMESPACE_ID::MessageLite*>(static_device_assignment));
|
|
if (message_arena != submessage_arena) {
|
|
static_device_assignment = ::PROTOBUF_NAMESPACE_ID::internal::GetOwnedMessage(
|
|
message_arena, static_device_assignment, submessage_arena);
|
|
}
|
|
|
|
} else {
|
|
|
|
}
|
|
_impl_.static_device_assignment_ = static_device_assignment;
|
|
// @@protoc_insertion_point(field_set_allocated:xla.HloModuleConfigProto.static_device_assignment)
|
|
}
|
|
|
|
// bool allow_separate_sharding_programs = 30;
|
|
inline void HloModuleConfigProto::clear_allow_separate_sharding_programs() {
|
|
_impl_.allow_separate_sharding_programs_ = false;
|
|
}
|
|
inline bool HloModuleConfigProto::_internal_allow_separate_sharding_programs() const {
|
|
return _impl_.allow_separate_sharding_programs_;
|
|
}
|
|
inline bool HloModuleConfigProto::allow_separate_sharding_programs() const {
|
|
// @@protoc_insertion_point(field_get:xla.HloModuleConfigProto.allow_separate_sharding_programs)
|
|
return _internal_allow_separate_sharding_programs();
|
|
}
|
|
inline void HloModuleConfigProto::_internal_set_allow_separate_sharding_programs(bool value) {
|
|
|
|
_impl_.allow_separate_sharding_programs_ = value;
|
|
}
|
|
inline void HloModuleConfigProto::set_allow_separate_sharding_programs(bool value) {
|
|
_internal_set_allow_separate_sharding_programs(value);
|
|
// @@protoc_insertion_point(field_set:xla.HloModuleConfigProto.allow_separate_sharding_programs)
|
|
}
|
|
|
|
// repeated .xla.ShardableValueUpdatePairProto shardable_value_update_pairs = 16;
|
|
inline int HloModuleConfigProto::_internal_shardable_value_update_pairs_size() const {
|
|
return _impl_.shardable_value_update_pairs_.size();
|
|
}
|
|
inline int HloModuleConfigProto::shardable_value_update_pairs_size() const {
|
|
return _internal_shardable_value_update_pairs_size();
|
|
}
|
|
inline void HloModuleConfigProto::clear_shardable_value_update_pairs() {
|
|
_impl_.shardable_value_update_pairs_.Clear();
|
|
}
|
|
inline ::xla::ShardableValueUpdatePairProto* HloModuleConfigProto::mutable_shardable_value_update_pairs(int index) {
|
|
// @@protoc_insertion_point(field_mutable:xla.HloModuleConfigProto.shardable_value_update_pairs)
|
|
return _impl_.shardable_value_update_pairs_.Mutable(index);
|
|
}
|
|
inline ::PROTOBUF_NAMESPACE_ID::RepeatedPtrField< ::xla::ShardableValueUpdatePairProto >*
|
|
HloModuleConfigProto::mutable_shardable_value_update_pairs() {
|
|
// @@protoc_insertion_point(field_mutable_list:xla.HloModuleConfigProto.shardable_value_update_pairs)
|
|
return &_impl_.shardable_value_update_pairs_;
|
|
}
|
|
inline const ::xla::ShardableValueUpdatePairProto& HloModuleConfigProto::_internal_shardable_value_update_pairs(int index) const {
|
|
return _impl_.shardable_value_update_pairs_.Get(index);
|
|
}
|
|
inline const ::xla::ShardableValueUpdatePairProto& HloModuleConfigProto::shardable_value_update_pairs(int index) const {
|
|
// @@protoc_insertion_point(field_get:xla.HloModuleConfigProto.shardable_value_update_pairs)
|
|
return _internal_shardable_value_update_pairs(index);
|
|
}
|
|
inline ::xla::ShardableValueUpdatePairProto* HloModuleConfigProto::_internal_add_shardable_value_update_pairs() {
|
|
return _impl_.shardable_value_update_pairs_.Add();
|
|
}
|
|
inline ::xla::ShardableValueUpdatePairProto* HloModuleConfigProto::add_shardable_value_update_pairs() {
|
|
::xla::ShardableValueUpdatePairProto* _add = _internal_add_shardable_value_update_pairs();
|
|
// @@protoc_insertion_point(field_add:xla.HloModuleConfigProto.shardable_value_update_pairs)
|
|
return _add;
|
|
}
|
|
inline const ::PROTOBUF_NAMESPACE_ID::RepeatedPtrField< ::xla::ShardableValueUpdatePairProto >&
|
|
HloModuleConfigProto::shardable_value_update_pairs() const {
|
|
// @@protoc_insertion_point(field_list:xla.HloModuleConfigProto.shardable_value_update_pairs)
|
|
return _impl_.shardable_value_update_pairs_;
|
|
}
|
|
|
|
// bool alias_passthrough_params = 17;
|
|
inline void HloModuleConfigProto::clear_alias_passthrough_params() {
|
|
_impl_.alias_passthrough_params_ = false;
|
|
}
|
|
inline bool HloModuleConfigProto::_internal_alias_passthrough_params() const {
|
|
return _impl_.alias_passthrough_params_;
|
|
}
|
|
inline bool HloModuleConfigProto::alias_passthrough_params() const {
|
|
// @@protoc_insertion_point(field_get:xla.HloModuleConfigProto.alias_passthrough_params)
|
|
return _internal_alias_passthrough_params();
|
|
}
|
|
inline void HloModuleConfigProto::_internal_set_alias_passthrough_params(bool value) {
|
|
|
|
_impl_.alias_passthrough_params_ = value;
|
|
}
|
|
inline void HloModuleConfigProto::set_alias_passthrough_params(bool value) {
|
|
_internal_set_alias_passthrough_params(value);
|
|
// @@protoc_insertion_point(field_set:xla.HloModuleConfigProto.alias_passthrough_params)
|
|
}
|
|
|
|
// bool content_aware_computation_sorting = 18;
|
|
inline void HloModuleConfigProto::clear_content_aware_computation_sorting() {
|
|
_impl_.content_aware_computation_sorting_ = false;
|
|
}
|
|
inline bool HloModuleConfigProto::_internal_content_aware_computation_sorting() const {
|
|
return _impl_.content_aware_computation_sorting_;
|
|
}
|
|
inline bool HloModuleConfigProto::content_aware_computation_sorting() const {
|
|
// @@protoc_insertion_point(field_get:xla.HloModuleConfigProto.content_aware_computation_sorting)
|
|
return _internal_content_aware_computation_sorting();
|
|
}
|
|
inline void HloModuleConfigProto::_internal_set_content_aware_computation_sorting(bool value) {
|
|
|
|
_impl_.content_aware_computation_sorting_ = value;
|
|
}
|
|
inline void HloModuleConfigProto::set_content_aware_computation_sorting(bool value) {
|
|
_internal_set_content_aware_computation_sorting(value);
|
|
// @@protoc_insertion_point(field_set:xla.HloModuleConfigProto.content_aware_computation_sorting)
|
|
}
|
|
|
|
// .xla.HloModuleConfigProto.FusionConfigCollection fusion_config_collection = 19;
|
|
inline void HloModuleConfigProto::clear_fusion_config_collection() {
|
|
_impl_.fusion_config_collection_ = 0;
|
|
}
|
|
inline ::xla::HloModuleConfigProto_FusionConfigCollection HloModuleConfigProto::_internal_fusion_config_collection() const {
|
|
return static_cast< ::xla::HloModuleConfigProto_FusionConfigCollection >(_impl_.fusion_config_collection_);
|
|
}
|
|
inline ::xla::HloModuleConfigProto_FusionConfigCollection HloModuleConfigProto::fusion_config_collection() const {
|
|
// @@protoc_insertion_point(field_get:xla.HloModuleConfigProto.fusion_config_collection)
|
|
return _internal_fusion_config_collection();
|
|
}
|
|
inline void HloModuleConfigProto::_internal_set_fusion_config_collection(::xla::HloModuleConfigProto_FusionConfigCollection value) {
|
|
|
|
_impl_.fusion_config_collection_ = value;
|
|
}
|
|
inline void HloModuleConfigProto::set_fusion_config_collection(::xla::HloModuleConfigProto_FusionConfigCollection value) {
|
|
_internal_set_fusion_config_collection(value);
|
|
// @@protoc_insertion_point(field_set:xla.HloModuleConfigProto.fusion_config_collection)
|
|
}
|
|
|
|
// repeated .xla.HloModuleConfigProto.BoolList fusion_config = 20;
|
|
inline int HloModuleConfigProto::_internal_fusion_config_size() const {
|
|
return _impl_.fusion_config_.size();
|
|
}
|
|
inline int HloModuleConfigProto::fusion_config_size() const {
|
|
return _internal_fusion_config_size();
|
|
}
|
|
inline void HloModuleConfigProto::clear_fusion_config() {
|
|
_impl_.fusion_config_.Clear();
|
|
}
|
|
inline ::xla::HloModuleConfigProto_BoolList* HloModuleConfigProto::mutable_fusion_config(int index) {
|
|
// @@protoc_insertion_point(field_mutable:xla.HloModuleConfigProto.fusion_config)
|
|
return _impl_.fusion_config_.Mutable(index);
|
|
}
|
|
inline ::PROTOBUF_NAMESPACE_ID::RepeatedPtrField< ::xla::HloModuleConfigProto_BoolList >*
|
|
HloModuleConfigProto::mutable_fusion_config() {
|
|
// @@protoc_insertion_point(field_mutable_list:xla.HloModuleConfigProto.fusion_config)
|
|
return &_impl_.fusion_config_;
|
|
}
|
|
inline const ::xla::HloModuleConfigProto_BoolList& HloModuleConfigProto::_internal_fusion_config(int index) const {
|
|
return _impl_.fusion_config_.Get(index);
|
|
}
|
|
inline const ::xla::HloModuleConfigProto_BoolList& HloModuleConfigProto::fusion_config(int index) const {
|
|
// @@protoc_insertion_point(field_get:xla.HloModuleConfigProto.fusion_config)
|
|
return _internal_fusion_config(index);
|
|
}
|
|
inline ::xla::HloModuleConfigProto_BoolList* HloModuleConfigProto::_internal_add_fusion_config() {
|
|
return _impl_.fusion_config_.Add();
|
|
}
|
|
inline ::xla::HloModuleConfigProto_BoolList* HloModuleConfigProto::add_fusion_config() {
|
|
::xla::HloModuleConfigProto_BoolList* _add = _internal_add_fusion_config();
|
|
// @@protoc_insertion_point(field_add:xla.HloModuleConfigProto.fusion_config)
|
|
return _add;
|
|
}
|
|
inline const ::PROTOBUF_NAMESPACE_ID::RepeatedPtrField< ::xla::HloModuleConfigProto_BoolList >&
|
|
HloModuleConfigProto::fusion_config() const {
|
|
// @@protoc_insertion_point(field_list:xla.HloModuleConfigProto.fusion_config)
|
|
return _impl_.fusion_config_;
|
|
}
|
|
|
|
// map<string, .xla.HloModuleConfigProto.Int64List> dot_config = 21;
|
|
inline int HloModuleConfigProto::_internal_dot_config_size() const {
|
|
return _impl_.dot_config_.size();
|
|
}
|
|
inline int HloModuleConfigProto::dot_config_size() const {
|
|
return _internal_dot_config_size();
|
|
}
|
|
inline void HloModuleConfigProto::clear_dot_config() {
|
|
_impl_.dot_config_.Clear();
|
|
}
|
|
inline const ::PROTOBUF_NAMESPACE_ID::Map< std::string, ::xla::HloModuleConfigProto_Int64List >&
|
|
HloModuleConfigProto::_internal_dot_config() const {
|
|
return _impl_.dot_config_.GetMap();
|
|
}
|
|
inline const ::PROTOBUF_NAMESPACE_ID::Map< std::string, ::xla::HloModuleConfigProto_Int64List >&
|
|
HloModuleConfigProto::dot_config() const {
|
|
// @@protoc_insertion_point(field_map:xla.HloModuleConfigProto.dot_config)
|
|
return _internal_dot_config();
|
|
}
|
|
inline ::PROTOBUF_NAMESPACE_ID::Map< std::string, ::xla::HloModuleConfigProto_Int64List >*
|
|
HloModuleConfigProto::_internal_mutable_dot_config() {
|
|
return _impl_.dot_config_.MutableMap();
|
|
}
|
|
inline ::PROTOBUF_NAMESPACE_ID::Map< std::string, ::xla::HloModuleConfigProto_Int64List >*
|
|
HloModuleConfigProto::mutable_dot_config() {
|
|
// @@protoc_insertion_point(field_mutable_map:xla.HloModuleConfigProto.dot_config)
|
|
return _internal_mutable_dot_config();
|
|
}
|
|
|
|
// repeated .xla.HloModuleConfigProto.Int64ListList layout_config = 22;
|
|
inline int HloModuleConfigProto::_internal_layout_config_size() const {
|
|
return _impl_.layout_config_.size();
|
|
}
|
|
inline int HloModuleConfigProto::layout_config_size() const {
|
|
return _internal_layout_config_size();
|
|
}
|
|
inline void HloModuleConfigProto::clear_layout_config() {
|
|
_impl_.layout_config_.Clear();
|
|
}
|
|
inline ::xla::HloModuleConfigProto_Int64ListList* HloModuleConfigProto::mutable_layout_config(int index) {
|
|
// @@protoc_insertion_point(field_mutable:xla.HloModuleConfigProto.layout_config)
|
|
return _impl_.layout_config_.Mutable(index);
|
|
}
|
|
inline ::PROTOBUF_NAMESPACE_ID::RepeatedPtrField< ::xla::HloModuleConfigProto_Int64ListList >*
|
|
HloModuleConfigProto::mutable_layout_config() {
|
|
// @@protoc_insertion_point(field_mutable_list:xla.HloModuleConfigProto.layout_config)
|
|
return &_impl_.layout_config_;
|
|
}
|
|
inline const ::xla::HloModuleConfigProto_Int64ListList& HloModuleConfigProto::_internal_layout_config(int index) const {
|
|
return _impl_.layout_config_.Get(index);
|
|
}
|
|
inline const ::xla::HloModuleConfigProto_Int64ListList& HloModuleConfigProto::layout_config(int index) const {
|
|
// @@protoc_insertion_point(field_get:xla.HloModuleConfigProto.layout_config)
|
|
return _internal_layout_config(index);
|
|
}
|
|
inline ::xla::HloModuleConfigProto_Int64ListList* HloModuleConfigProto::_internal_add_layout_config() {
|
|
return _impl_.layout_config_.Add();
|
|
}
|
|
inline ::xla::HloModuleConfigProto_Int64ListList* HloModuleConfigProto::add_layout_config() {
|
|
::xla::HloModuleConfigProto_Int64ListList* _add = _internal_add_layout_config();
|
|
// @@protoc_insertion_point(field_add:xla.HloModuleConfigProto.layout_config)
|
|
return _add;
|
|
}
|
|
inline const ::PROTOBUF_NAMESPACE_ID::RepeatedPtrField< ::xla::HloModuleConfigProto_Int64ListList >&
|
|
HloModuleConfigProto::layout_config() const {
|
|
// @@protoc_insertion_point(field_list:xla.HloModuleConfigProto.layout_config)
|
|
return _impl_.layout_config_;
|
|
}
|
|
|
|
// repeated uint64 memory_space_assignment_config = 23;
|
|
inline int HloModuleConfigProto::_internal_memory_space_assignment_config_size() const {
|
|
return _impl_.memory_space_assignment_config_.size();
|
|
}
|
|
inline int HloModuleConfigProto::memory_space_assignment_config_size() const {
|
|
return _internal_memory_space_assignment_config_size();
|
|
}
|
|
inline void HloModuleConfigProto::clear_memory_space_assignment_config() {
|
|
_impl_.memory_space_assignment_config_.Clear();
|
|
}
|
|
inline uint64_t HloModuleConfigProto::_internal_memory_space_assignment_config(int index) const {
|
|
return _impl_.memory_space_assignment_config_.Get(index);
|
|
}
|
|
inline uint64_t HloModuleConfigProto::memory_space_assignment_config(int index) const {
|
|
// @@protoc_insertion_point(field_get:xla.HloModuleConfigProto.memory_space_assignment_config)
|
|
return _internal_memory_space_assignment_config(index);
|
|
}
|
|
inline void HloModuleConfigProto::set_memory_space_assignment_config(int index, uint64_t value) {
|
|
_impl_.memory_space_assignment_config_.Set(index, value);
|
|
// @@protoc_insertion_point(field_set:xla.HloModuleConfigProto.memory_space_assignment_config)
|
|
}
|
|
inline void HloModuleConfigProto::_internal_add_memory_space_assignment_config(uint64_t value) {
|
|
_impl_.memory_space_assignment_config_.Add(value);
|
|
}
|
|
inline void HloModuleConfigProto::add_memory_space_assignment_config(uint64_t value) {
|
|
_internal_add_memory_space_assignment_config(value);
|
|
// @@protoc_insertion_point(field_add:xla.HloModuleConfigProto.memory_space_assignment_config)
|
|
}
|
|
inline const ::PROTOBUF_NAMESPACE_ID::RepeatedField< uint64_t >&
|
|
HloModuleConfigProto::_internal_memory_space_assignment_config() const {
|
|
return _impl_.memory_space_assignment_config_;
|
|
}
|
|
inline const ::PROTOBUF_NAMESPACE_ID::RepeatedField< uint64_t >&
|
|
HloModuleConfigProto::memory_space_assignment_config() const {
|
|
// @@protoc_insertion_point(field_list:xla.HloModuleConfigProto.memory_space_assignment_config)
|
|
return _internal_memory_space_assignment_config();
|
|
}
|
|
inline ::PROTOBUF_NAMESPACE_ID::RepeatedField< uint64_t >*
|
|
HloModuleConfigProto::_internal_mutable_memory_space_assignment_config() {
|
|
return &_impl_.memory_space_assignment_config_;
|
|
}
|
|
inline ::PROTOBUF_NAMESPACE_ID::RepeatedField< uint64_t >*
|
|
HloModuleConfigProto::mutable_memory_space_assignment_config() {
|
|
// @@protoc_insertion_point(field_mutable_list:xla.HloModuleConfigProto.memory_space_assignment_config)
|
|
return _internal_mutable_memory_space_assignment_config();
|
|
}
|
|
|
|
// repeated .xla.HloModuleConfigProto.BoolList phase_ordering_config = 24;
|
|
inline int HloModuleConfigProto::_internal_phase_ordering_config_size() const {
|
|
return _impl_.phase_ordering_config_.size();
|
|
}
|
|
inline int HloModuleConfigProto::phase_ordering_config_size() const {
|
|
return _internal_phase_ordering_config_size();
|
|
}
|
|
inline void HloModuleConfigProto::clear_phase_ordering_config() {
|
|
_impl_.phase_ordering_config_.Clear();
|
|
}
|
|
inline ::xla::HloModuleConfigProto_BoolList* HloModuleConfigProto::mutable_phase_ordering_config(int index) {
|
|
// @@protoc_insertion_point(field_mutable:xla.HloModuleConfigProto.phase_ordering_config)
|
|
return _impl_.phase_ordering_config_.Mutable(index);
|
|
}
|
|
inline ::PROTOBUF_NAMESPACE_ID::RepeatedPtrField< ::xla::HloModuleConfigProto_BoolList >*
|
|
HloModuleConfigProto::mutable_phase_ordering_config() {
|
|
// @@protoc_insertion_point(field_mutable_list:xla.HloModuleConfigProto.phase_ordering_config)
|
|
return &_impl_.phase_ordering_config_;
|
|
}
|
|
inline const ::xla::HloModuleConfigProto_BoolList& HloModuleConfigProto::_internal_phase_ordering_config(int index) const {
|
|
return _impl_.phase_ordering_config_.Get(index);
|
|
}
|
|
inline const ::xla::HloModuleConfigProto_BoolList& HloModuleConfigProto::phase_ordering_config(int index) const {
|
|
// @@protoc_insertion_point(field_get:xla.HloModuleConfigProto.phase_ordering_config)
|
|
return _internal_phase_ordering_config(index);
|
|
}
|
|
inline ::xla::HloModuleConfigProto_BoolList* HloModuleConfigProto::_internal_add_phase_ordering_config() {
|
|
return _impl_.phase_ordering_config_.Add();
|
|
}
|
|
inline ::xla::HloModuleConfigProto_BoolList* HloModuleConfigProto::add_phase_ordering_config() {
|
|
::xla::HloModuleConfigProto_BoolList* _add = _internal_add_phase_ordering_config();
|
|
// @@protoc_insertion_point(field_add:xla.HloModuleConfigProto.phase_ordering_config)
|
|
return _add;
|
|
}
|
|
inline const ::PROTOBUF_NAMESPACE_ID::RepeatedPtrField< ::xla::HloModuleConfigProto_BoolList >&
|
|
HloModuleConfigProto::phase_ordering_config() const {
|
|
// @@protoc_insertion_point(field_list:xla.HloModuleConfigProto.phase_ordering_config)
|
|
return _impl_.phase_ordering_config_;
|
|
}
|
|
|
|
// int32 phase_index = 25;
|
|
inline void HloModuleConfigProto::clear_phase_index() {
|
|
_impl_.phase_index_ = 0;
|
|
}
|
|
inline int32_t HloModuleConfigProto::_internal_phase_index() const {
|
|
return _impl_.phase_index_;
|
|
}
|
|
inline int32_t HloModuleConfigProto::phase_index() const {
|
|
// @@protoc_insertion_point(field_get:xla.HloModuleConfigProto.phase_index)
|
|
return _internal_phase_index();
|
|
}
|
|
inline void HloModuleConfigProto::_internal_set_phase_index(int32_t value) {
|
|
|
|
_impl_.phase_index_ = value;
|
|
}
|
|
inline void HloModuleConfigProto::set_phase_index(int32_t value) {
|
|
_internal_set_phase_index(value);
|
|
// @@protoc_insertion_point(field_set:xla.HloModuleConfigProto.phase_index)
|
|
}
|
|
|
|
// repeated bool allow_spmd_sharding_propagation_to_output = 27;
|
|
inline int HloModuleConfigProto::_internal_allow_spmd_sharding_propagation_to_output_size() const {
|
|
return _impl_.allow_spmd_sharding_propagation_to_output_.size();
|
|
}
|
|
inline int HloModuleConfigProto::allow_spmd_sharding_propagation_to_output_size() const {
|
|
return _internal_allow_spmd_sharding_propagation_to_output_size();
|
|
}
|
|
inline void HloModuleConfigProto::clear_allow_spmd_sharding_propagation_to_output() {
|
|
_impl_.allow_spmd_sharding_propagation_to_output_.Clear();
|
|
}
|
|
inline bool HloModuleConfigProto::_internal_allow_spmd_sharding_propagation_to_output(int index) const {
|
|
return _impl_.allow_spmd_sharding_propagation_to_output_.Get(index);
|
|
}
|
|
inline bool HloModuleConfigProto::allow_spmd_sharding_propagation_to_output(int index) const {
|
|
// @@protoc_insertion_point(field_get:xla.HloModuleConfigProto.allow_spmd_sharding_propagation_to_output)
|
|
return _internal_allow_spmd_sharding_propagation_to_output(index);
|
|
}
|
|
inline void HloModuleConfigProto::set_allow_spmd_sharding_propagation_to_output(int index, bool value) {
|
|
_impl_.allow_spmd_sharding_propagation_to_output_.Set(index, value);
|
|
// @@protoc_insertion_point(field_set:xla.HloModuleConfigProto.allow_spmd_sharding_propagation_to_output)
|
|
}
|
|
inline void HloModuleConfigProto::_internal_add_allow_spmd_sharding_propagation_to_output(bool value) {
|
|
_impl_.allow_spmd_sharding_propagation_to_output_.Add(value);
|
|
}
|
|
inline void HloModuleConfigProto::add_allow_spmd_sharding_propagation_to_output(bool value) {
|
|
_internal_add_allow_spmd_sharding_propagation_to_output(value);
|
|
// @@protoc_insertion_point(field_add:xla.HloModuleConfigProto.allow_spmd_sharding_propagation_to_output)
|
|
}
|
|
inline const ::PROTOBUF_NAMESPACE_ID::RepeatedField< bool >&
|
|
HloModuleConfigProto::_internal_allow_spmd_sharding_propagation_to_output() const {
|
|
return _impl_.allow_spmd_sharding_propagation_to_output_;
|
|
}
|
|
inline const ::PROTOBUF_NAMESPACE_ID::RepeatedField< bool >&
|
|
HloModuleConfigProto::allow_spmd_sharding_propagation_to_output() const {
|
|
// @@protoc_insertion_point(field_list:xla.HloModuleConfigProto.allow_spmd_sharding_propagation_to_output)
|
|
return _internal_allow_spmd_sharding_propagation_to_output();
|
|
}
|
|
inline ::PROTOBUF_NAMESPACE_ID::RepeatedField< bool >*
|
|
HloModuleConfigProto::_internal_mutable_allow_spmd_sharding_propagation_to_output() {
|
|
return &_impl_.allow_spmd_sharding_propagation_to_output_;
|
|
}
|
|
inline ::PROTOBUF_NAMESPACE_ID::RepeatedField< bool >*
|
|
HloModuleConfigProto::mutable_allow_spmd_sharding_propagation_to_output() {
|
|
// @@protoc_insertion_point(field_mutable_list:xla.HloModuleConfigProto.allow_spmd_sharding_propagation_to_output)
|
|
return _internal_mutable_allow_spmd_sharding_propagation_to_output();
|
|
}
|
|
|
|
// map<string, int64> analysis_allowance_map = 28;
|
|
inline int HloModuleConfigProto::_internal_analysis_allowance_map_size() const {
|
|
return _impl_.analysis_allowance_map_.size();
|
|
}
|
|
inline int HloModuleConfigProto::analysis_allowance_map_size() const {
|
|
return _internal_analysis_allowance_map_size();
|
|
}
|
|
inline void HloModuleConfigProto::clear_analysis_allowance_map() {
|
|
_impl_.analysis_allowance_map_.Clear();
|
|
}
|
|
inline const ::PROTOBUF_NAMESPACE_ID::Map< std::string, int64_t >&
|
|
HloModuleConfigProto::_internal_analysis_allowance_map() const {
|
|
return _impl_.analysis_allowance_map_.GetMap();
|
|
}
|
|
inline const ::PROTOBUF_NAMESPACE_ID::Map< std::string, int64_t >&
|
|
HloModuleConfigProto::analysis_allowance_map() const {
|
|
// @@protoc_insertion_point(field_map:xla.HloModuleConfigProto.analysis_allowance_map)
|
|
return _internal_analysis_allowance_map();
|
|
}
|
|
inline ::PROTOBUF_NAMESPACE_ID::Map< std::string, int64_t >*
|
|
HloModuleConfigProto::_internal_mutable_analysis_allowance_map() {
|
|
return _impl_.analysis_allowance_map_.MutableMap();
|
|
}
|
|
inline ::PROTOBUF_NAMESPACE_ID::Map< std::string, int64_t >*
|
|
HloModuleConfigProto::mutable_analysis_allowance_map() {
|
|
// @@protoc_insertion_point(field_mutable_map:xla.HloModuleConfigProto.analysis_allowance_map)
|
|
return _internal_mutable_analysis_allowance_map();
|
|
}
|
|
|
|
// .xla.PrecisionConfig.Precision matrix_unit_operand_precision = 29;
|
|
inline void HloModuleConfigProto::clear_matrix_unit_operand_precision() {
|
|
_impl_.matrix_unit_operand_precision_ = 0;
|
|
}
|
|
inline ::xla::PrecisionConfig_Precision HloModuleConfigProto::_internal_matrix_unit_operand_precision() const {
|
|
return static_cast< ::xla::PrecisionConfig_Precision >(_impl_.matrix_unit_operand_precision_);
|
|
}
|
|
inline ::xla::PrecisionConfig_Precision HloModuleConfigProto::matrix_unit_operand_precision() const {
|
|
// @@protoc_insertion_point(field_get:xla.HloModuleConfigProto.matrix_unit_operand_precision)
|
|
return _internal_matrix_unit_operand_precision();
|
|
}
|
|
inline void HloModuleConfigProto::_internal_set_matrix_unit_operand_precision(::xla::PrecisionConfig_Precision value) {
|
|
|
|
_impl_.matrix_unit_operand_precision_ = value;
|
|
}
|
|
inline void HloModuleConfigProto::set_matrix_unit_operand_precision(::xla::PrecisionConfig_Precision value) {
|
|
_internal_set_matrix_unit_operand_precision(value);
|
|
// @@protoc_insertion_point(field_set:xla.HloModuleConfigProto.matrix_unit_operand_precision)
|
|
}
|
|
|
|
// bytes fdo_profile = 31;
|
|
inline void HloModuleConfigProto::clear_fdo_profile() {
|
|
_impl_.fdo_profile_.ClearToEmpty();
|
|
}
|
|
inline const std::string& HloModuleConfigProto::fdo_profile() const {
|
|
// @@protoc_insertion_point(field_get:xla.HloModuleConfigProto.fdo_profile)
|
|
return _internal_fdo_profile();
|
|
}
|
|
template <typename ArgT0, typename... ArgT>
|
|
inline PROTOBUF_ALWAYS_INLINE
|
|
void HloModuleConfigProto::set_fdo_profile(ArgT0&& arg0, ArgT... args) {
|
|
|
|
_impl_.fdo_profile_.SetBytes(static_cast<ArgT0 &&>(arg0), args..., GetArenaForAllocation());
|
|
// @@protoc_insertion_point(field_set:xla.HloModuleConfigProto.fdo_profile)
|
|
}
|
|
inline std::string* HloModuleConfigProto::mutable_fdo_profile() {
|
|
std::string* _s = _internal_mutable_fdo_profile();
|
|
// @@protoc_insertion_point(field_mutable:xla.HloModuleConfigProto.fdo_profile)
|
|
return _s;
|
|
}
|
|
inline const std::string& HloModuleConfigProto::_internal_fdo_profile() const {
|
|
return _impl_.fdo_profile_.Get();
|
|
}
|
|
inline void HloModuleConfigProto::_internal_set_fdo_profile(const std::string& value) {
|
|
|
|
_impl_.fdo_profile_.Set(value, GetArenaForAllocation());
|
|
}
|
|
inline std::string* HloModuleConfigProto::_internal_mutable_fdo_profile() {
|
|
|
|
return _impl_.fdo_profile_.Mutable(GetArenaForAllocation());
|
|
}
|
|
inline std::string* HloModuleConfigProto::release_fdo_profile() {
|
|
// @@protoc_insertion_point(field_release:xla.HloModuleConfigProto.fdo_profile)
|
|
return _impl_.fdo_profile_.Release();
|
|
}
|
|
inline void HloModuleConfigProto::set_allocated_fdo_profile(std::string* fdo_profile) {
|
|
if (fdo_profile != nullptr) {
|
|
|
|
} else {
|
|
|
|
}
|
|
_impl_.fdo_profile_.SetAllocated(fdo_profile, GetArenaForAllocation());
|
|
#ifdef PROTOBUF_FORCE_COPY_DEFAULT_STRING
|
|
if (_impl_.fdo_profile_.IsDefault()) {
|
|
_impl_.fdo_profile_.Set("", GetArenaForAllocation());
|
|
}
|
|
#endif // PROTOBUF_FORCE_COPY_DEFAULT_STRING
|
|
// @@protoc_insertion_point(field_set_allocated:xla.HloModuleConfigProto.fdo_profile)
|
|
}
|
|
|
|
// int64 device_memory_size = 32;
|
|
inline void HloModuleConfigProto::clear_device_memory_size() {
|
|
_impl_.device_memory_size_ = int64_t{0};
|
|
}
|
|
inline int64_t HloModuleConfigProto::_internal_device_memory_size() const {
|
|
return _impl_.device_memory_size_;
|
|
}
|
|
inline int64_t HloModuleConfigProto::device_memory_size() const {
|
|
// @@protoc_insertion_point(field_get:xla.HloModuleConfigProto.device_memory_size)
|
|
return _internal_device_memory_size();
|
|
}
|
|
inline void HloModuleConfigProto::_internal_set_device_memory_size(int64_t value) {
|
|
|
|
_impl_.device_memory_size_ = value;
|
|
}
|
|
inline void HloModuleConfigProto::set_device_memory_size(int64_t value) {
|
|
_internal_set_device_memory_size(value);
|
|
// @@protoc_insertion_point(field_set:xla.HloModuleConfigProto.device_memory_size)
|
|
}
|
|
|
|
// -------------------------------------------------------------------
|
|
|
|
// HloModuleProtoWithConfig
|
|
|
|
// .xla.HloModuleProto hlo_module = 1;
|
|
inline bool HloModuleProtoWithConfig::_internal_has_hlo_module() const {
|
|
return this != internal_default_instance() && _impl_.hlo_module_ != nullptr;
|
|
}
|
|
inline bool HloModuleProtoWithConfig::has_hlo_module() const {
|
|
return _internal_has_hlo_module();
|
|
}
|
|
inline const ::xla::HloModuleProto& HloModuleProtoWithConfig::_internal_hlo_module() const {
|
|
const ::xla::HloModuleProto* p = _impl_.hlo_module_;
|
|
return p != nullptr ? *p : reinterpret_cast<const ::xla::HloModuleProto&>(
|
|
::xla::_HloModuleProto_default_instance_);
|
|
}
|
|
inline const ::xla::HloModuleProto& HloModuleProtoWithConfig::hlo_module() const {
|
|
// @@protoc_insertion_point(field_get:xla.HloModuleProtoWithConfig.hlo_module)
|
|
return _internal_hlo_module();
|
|
}
|
|
inline void HloModuleProtoWithConfig::unsafe_arena_set_allocated_hlo_module(
|
|
::xla::HloModuleProto* hlo_module) {
|
|
if (GetArenaForAllocation() == nullptr) {
|
|
delete reinterpret_cast<::PROTOBUF_NAMESPACE_ID::MessageLite*>(_impl_.hlo_module_);
|
|
}
|
|
_impl_.hlo_module_ = hlo_module;
|
|
if (hlo_module) {
|
|
|
|
} else {
|
|
|
|
}
|
|
// @@protoc_insertion_point(field_unsafe_arena_set_allocated:xla.HloModuleProtoWithConfig.hlo_module)
|
|
}
|
|
inline ::xla::HloModuleProto* HloModuleProtoWithConfig::release_hlo_module() {
|
|
|
|
::xla::HloModuleProto* temp = _impl_.hlo_module_;
|
|
_impl_.hlo_module_ = 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::HloModuleProto* HloModuleProtoWithConfig::unsafe_arena_release_hlo_module() {
|
|
// @@protoc_insertion_point(field_release:xla.HloModuleProtoWithConfig.hlo_module)
|
|
|
|
::xla::HloModuleProto* temp = _impl_.hlo_module_;
|
|
_impl_.hlo_module_ = nullptr;
|
|
return temp;
|
|
}
|
|
inline ::xla::HloModuleProto* HloModuleProtoWithConfig::_internal_mutable_hlo_module() {
|
|
|
|
if (_impl_.hlo_module_ == nullptr) {
|
|
auto* p = CreateMaybeMessage<::xla::HloModuleProto>(GetArenaForAllocation());
|
|
_impl_.hlo_module_ = p;
|
|
}
|
|
return _impl_.hlo_module_;
|
|
}
|
|
inline ::xla::HloModuleProto* HloModuleProtoWithConfig::mutable_hlo_module() {
|
|
::xla::HloModuleProto* _msg = _internal_mutable_hlo_module();
|
|
// @@protoc_insertion_point(field_mutable:xla.HloModuleProtoWithConfig.hlo_module)
|
|
return _msg;
|
|
}
|
|
inline void HloModuleProtoWithConfig::set_allocated_hlo_module(::xla::HloModuleProto* hlo_module) {
|
|
::PROTOBUF_NAMESPACE_ID::Arena* message_arena = GetArenaForAllocation();
|
|
if (message_arena == nullptr) {
|
|
delete reinterpret_cast< ::PROTOBUF_NAMESPACE_ID::MessageLite*>(_impl_.hlo_module_);
|
|
}
|
|
if (hlo_module) {
|
|
::PROTOBUF_NAMESPACE_ID::Arena* submessage_arena =
|
|
::PROTOBUF_NAMESPACE_ID::Arena::InternalGetOwningArena(
|
|
reinterpret_cast<::PROTOBUF_NAMESPACE_ID::MessageLite*>(hlo_module));
|
|
if (message_arena != submessage_arena) {
|
|
hlo_module = ::PROTOBUF_NAMESPACE_ID::internal::GetOwnedMessage(
|
|
message_arena, hlo_module, submessage_arena);
|
|
}
|
|
|
|
} else {
|
|
|
|
}
|
|
_impl_.hlo_module_ = hlo_module;
|
|
// @@protoc_insertion_point(field_set_allocated:xla.HloModuleProtoWithConfig.hlo_module)
|
|
}
|
|
|
|
// .xla.HloModuleConfigProto config = 2;
|
|
inline bool HloModuleProtoWithConfig::_internal_has_config() const {
|
|
return this != internal_default_instance() && _impl_.config_ != nullptr;
|
|
}
|
|
inline bool HloModuleProtoWithConfig::has_config() const {
|
|
return _internal_has_config();
|
|
}
|
|
inline void HloModuleProtoWithConfig::clear_config() {
|
|
if (GetArenaForAllocation() == nullptr && _impl_.config_ != nullptr) {
|
|
delete _impl_.config_;
|
|
}
|
|
_impl_.config_ = nullptr;
|
|
}
|
|
inline const ::xla::HloModuleConfigProto& HloModuleProtoWithConfig::_internal_config() const {
|
|
const ::xla::HloModuleConfigProto* p = _impl_.config_;
|
|
return p != nullptr ? *p : reinterpret_cast<const ::xla::HloModuleConfigProto&>(
|
|
::xla::_HloModuleConfigProto_default_instance_);
|
|
}
|
|
inline const ::xla::HloModuleConfigProto& HloModuleProtoWithConfig::config() const {
|
|
// @@protoc_insertion_point(field_get:xla.HloModuleProtoWithConfig.config)
|
|
return _internal_config();
|
|
}
|
|
inline void HloModuleProtoWithConfig::unsafe_arena_set_allocated_config(
|
|
::xla::HloModuleConfigProto* config) {
|
|
if (GetArenaForAllocation() == nullptr) {
|
|
delete reinterpret_cast<::PROTOBUF_NAMESPACE_ID::MessageLite*>(_impl_.config_);
|
|
}
|
|
_impl_.config_ = config;
|
|
if (config) {
|
|
|
|
} else {
|
|
|
|
}
|
|
// @@protoc_insertion_point(field_unsafe_arena_set_allocated:xla.HloModuleProtoWithConfig.config)
|
|
}
|
|
inline ::xla::HloModuleConfigProto* HloModuleProtoWithConfig::release_config() {
|
|
|
|
::xla::HloModuleConfigProto* temp = _impl_.config_;
|
|
_impl_.config_ = 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::HloModuleConfigProto* HloModuleProtoWithConfig::unsafe_arena_release_config() {
|
|
// @@protoc_insertion_point(field_release:xla.HloModuleProtoWithConfig.config)
|
|
|
|
::xla::HloModuleConfigProto* temp = _impl_.config_;
|
|
_impl_.config_ = nullptr;
|
|
return temp;
|
|
}
|
|
inline ::xla::HloModuleConfigProto* HloModuleProtoWithConfig::_internal_mutable_config() {
|
|
|
|
if (_impl_.config_ == nullptr) {
|
|
auto* p = CreateMaybeMessage<::xla::HloModuleConfigProto>(GetArenaForAllocation());
|
|
_impl_.config_ = p;
|
|
}
|
|
return _impl_.config_;
|
|
}
|
|
inline ::xla::HloModuleConfigProto* HloModuleProtoWithConfig::mutable_config() {
|
|
::xla::HloModuleConfigProto* _msg = _internal_mutable_config();
|
|
// @@protoc_insertion_point(field_mutable:xla.HloModuleProtoWithConfig.config)
|
|
return _msg;
|
|
}
|
|
inline void HloModuleProtoWithConfig::set_allocated_config(::xla::HloModuleConfigProto* config) {
|
|
::PROTOBUF_NAMESPACE_ID::Arena* message_arena = GetArenaForAllocation();
|
|
if (message_arena == nullptr) {
|
|
delete _impl_.config_;
|
|
}
|
|
if (config) {
|
|
::PROTOBUF_NAMESPACE_ID::Arena* submessage_arena =
|
|
::PROTOBUF_NAMESPACE_ID::Arena::InternalGetOwningArena(config);
|
|
if (message_arena != submessage_arena) {
|
|
config = ::PROTOBUF_NAMESPACE_ID::internal::GetOwnedMessage(
|
|
message_arena, config, submessage_arena);
|
|
}
|
|
|
|
} else {
|
|
|
|
}
|
|
_impl_.config_ = config;
|
|
// @@protoc_insertion_point(field_set_allocated:xla.HloModuleProtoWithConfig.config)
|
|
}
|
|
|
|
// -------------------------------------------------------------------
|
|
|
|
// GetDeviceHandlesRequest
|
|
|
|
// int64 device_count = 1;
|
|
inline void GetDeviceHandlesRequest::clear_device_count() {
|
|
_impl_.device_count_ = int64_t{0};
|
|
}
|
|
inline int64_t GetDeviceHandlesRequest::_internal_device_count() const {
|
|
return _impl_.device_count_;
|
|
}
|
|
inline int64_t GetDeviceHandlesRequest::device_count() const {
|
|
// @@protoc_insertion_point(field_get:xla.GetDeviceHandlesRequest.device_count)
|
|
return _internal_device_count();
|
|
}
|
|
inline void GetDeviceHandlesRequest::_internal_set_device_count(int64_t value) {
|
|
|
|
_impl_.device_count_ = value;
|
|
}
|
|
inline void GetDeviceHandlesRequest::set_device_count(int64_t value) {
|
|
_internal_set_device_count(value);
|
|
// @@protoc_insertion_point(field_set:xla.GetDeviceHandlesRequest.device_count)
|
|
}
|
|
|
|
// -------------------------------------------------------------------
|
|
|
|
// GetDeviceHandlesResponse
|
|
|
|
// repeated .xla.DeviceHandle device_handles = 1;
|
|
inline int GetDeviceHandlesResponse::_internal_device_handles_size() const {
|
|
return _impl_.device_handles_.size();
|
|
}
|
|
inline int GetDeviceHandlesResponse::device_handles_size() const {
|
|
return _internal_device_handles_size();
|
|
}
|
|
inline ::xla::DeviceHandle* GetDeviceHandlesResponse::mutable_device_handles(int index) {
|
|
// @@protoc_insertion_point(field_mutable:xla.GetDeviceHandlesResponse.device_handles)
|
|
return _impl_.device_handles_.Mutable(index);
|
|
}
|
|
inline ::PROTOBUF_NAMESPACE_ID::RepeatedPtrField< ::xla::DeviceHandle >*
|
|
GetDeviceHandlesResponse::mutable_device_handles() {
|
|
// @@protoc_insertion_point(field_mutable_list:xla.GetDeviceHandlesResponse.device_handles)
|
|
return &_impl_.device_handles_;
|
|
}
|
|
inline const ::xla::DeviceHandle& GetDeviceHandlesResponse::_internal_device_handles(int index) const {
|
|
return _impl_.device_handles_.Get(index);
|
|
}
|
|
inline const ::xla::DeviceHandle& GetDeviceHandlesResponse::device_handles(int index) const {
|
|
// @@protoc_insertion_point(field_get:xla.GetDeviceHandlesResponse.device_handles)
|
|
return _internal_device_handles(index);
|
|
}
|
|
inline ::xla::DeviceHandle* GetDeviceHandlesResponse::_internal_add_device_handles() {
|
|
return _impl_.device_handles_.Add();
|
|
}
|
|
inline ::xla::DeviceHandle* GetDeviceHandlesResponse::add_device_handles() {
|
|
::xla::DeviceHandle* _add = _internal_add_device_handles();
|
|
// @@protoc_insertion_point(field_add:xla.GetDeviceHandlesResponse.device_handles)
|
|
return _add;
|
|
}
|
|
inline const ::PROTOBUF_NAMESPACE_ID::RepeatedPtrField< ::xla::DeviceHandle >&
|
|
GetDeviceHandlesResponse::device_handles() const {
|
|
// @@protoc_insertion_point(field_list:xla.GetDeviceHandlesResponse.device_handles)
|
|
return _impl_.device_handles_;
|
|
}
|
|
|
|
// -------------------------------------------------------------------
|
|
|
|
// TransferToClientRequest
|
|
|
|
// .xla.GlobalDataHandle data = 1;
|
|
inline bool TransferToClientRequest::_internal_has_data() const {
|
|
return this != internal_default_instance() && _impl_.data_ != nullptr;
|
|
}
|
|
inline bool TransferToClientRequest::has_data() const {
|
|
return _internal_has_data();
|
|
}
|
|
inline const ::xla::GlobalDataHandle& TransferToClientRequest::_internal_data() const {
|
|
const ::xla::GlobalDataHandle* p = _impl_.data_;
|
|
return p != nullptr ? *p : reinterpret_cast<const ::xla::GlobalDataHandle&>(
|
|
::xla::_GlobalDataHandle_default_instance_);
|
|
}
|
|
inline const ::xla::GlobalDataHandle& TransferToClientRequest::data() const {
|
|
// @@protoc_insertion_point(field_get:xla.TransferToClientRequest.data)
|
|
return _internal_data();
|
|
}
|
|
inline void TransferToClientRequest::unsafe_arena_set_allocated_data(
|
|
::xla::GlobalDataHandle* data) {
|
|
if (GetArenaForAllocation() == nullptr) {
|
|
delete reinterpret_cast<::PROTOBUF_NAMESPACE_ID::MessageLite*>(_impl_.data_);
|
|
}
|
|
_impl_.data_ = data;
|
|
if (data) {
|
|
|
|
} else {
|
|
|
|
}
|
|
// @@protoc_insertion_point(field_unsafe_arena_set_allocated:xla.TransferToClientRequest.data)
|
|
}
|
|
inline ::xla::GlobalDataHandle* TransferToClientRequest::release_data() {
|
|
|
|
::xla::GlobalDataHandle* temp = _impl_.data_;
|
|
_impl_.data_ = 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::GlobalDataHandle* TransferToClientRequest::unsafe_arena_release_data() {
|
|
// @@protoc_insertion_point(field_release:xla.TransferToClientRequest.data)
|
|
|
|
::xla::GlobalDataHandle* temp = _impl_.data_;
|
|
_impl_.data_ = nullptr;
|
|
return temp;
|
|
}
|
|
inline ::xla::GlobalDataHandle* TransferToClientRequest::_internal_mutable_data() {
|
|
|
|
if (_impl_.data_ == nullptr) {
|
|
auto* p = CreateMaybeMessage<::xla::GlobalDataHandle>(GetArenaForAllocation());
|
|
_impl_.data_ = p;
|
|
}
|
|
return _impl_.data_;
|
|
}
|
|
inline ::xla::GlobalDataHandle* TransferToClientRequest::mutable_data() {
|
|
::xla::GlobalDataHandle* _msg = _internal_mutable_data();
|
|
// @@protoc_insertion_point(field_mutable:xla.TransferToClientRequest.data)
|
|
return _msg;
|
|
}
|
|
inline void TransferToClientRequest::set_allocated_data(::xla::GlobalDataHandle* data) {
|
|
::PROTOBUF_NAMESPACE_ID::Arena* message_arena = GetArenaForAllocation();
|
|
if (message_arena == nullptr) {
|
|
delete reinterpret_cast< ::PROTOBUF_NAMESPACE_ID::MessageLite*>(_impl_.data_);
|
|
}
|
|
if (data) {
|
|
::PROTOBUF_NAMESPACE_ID::Arena* submessage_arena =
|
|
::PROTOBUF_NAMESPACE_ID::Arena::InternalGetOwningArena(
|
|
reinterpret_cast<::PROTOBUF_NAMESPACE_ID::MessageLite*>(data));
|
|
if (message_arena != submessage_arena) {
|
|
data = ::PROTOBUF_NAMESPACE_ID::internal::GetOwnedMessage(
|
|
message_arena, data, submessage_arena);
|
|
}
|
|
|
|
} else {
|
|
|
|
}
|
|
_impl_.data_ = data;
|
|
// @@protoc_insertion_point(field_set_allocated:xla.TransferToClientRequest.data)
|
|
}
|
|
|
|
// .xla.ShapeProto shape_with_layout = 2;
|
|
inline bool TransferToClientRequest::_internal_has_shape_with_layout() const {
|
|
return this != internal_default_instance() && _impl_.shape_with_layout_ != nullptr;
|
|
}
|
|
inline bool TransferToClientRequest::has_shape_with_layout() const {
|
|
return _internal_has_shape_with_layout();
|
|
}
|
|
inline const ::xla::ShapeProto& TransferToClientRequest::_internal_shape_with_layout() const {
|
|
const ::xla::ShapeProto* p = _impl_.shape_with_layout_;
|
|
return p != nullptr ? *p : reinterpret_cast<const ::xla::ShapeProto&>(
|
|
::xla::_ShapeProto_default_instance_);
|
|
}
|
|
inline const ::xla::ShapeProto& TransferToClientRequest::shape_with_layout() const {
|
|
// @@protoc_insertion_point(field_get:xla.TransferToClientRequest.shape_with_layout)
|
|
return _internal_shape_with_layout();
|
|
}
|
|
inline void TransferToClientRequest::unsafe_arena_set_allocated_shape_with_layout(
|
|
::xla::ShapeProto* shape_with_layout) {
|
|
if (GetArenaForAllocation() == nullptr) {
|
|
delete reinterpret_cast<::PROTOBUF_NAMESPACE_ID::MessageLite*>(_impl_.shape_with_layout_);
|
|
}
|
|
_impl_.shape_with_layout_ = shape_with_layout;
|
|
if (shape_with_layout) {
|
|
|
|
} else {
|
|
|
|
}
|
|
// @@protoc_insertion_point(field_unsafe_arena_set_allocated:xla.TransferToClientRequest.shape_with_layout)
|
|
}
|
|
inline ::xla::ShapeProto* TransferToClientRequest::release_shape_with_layout() {
|
|
|
|
::xla::ShapeProto* temp = _impl_.shape_with_layout_;
|
|
_impl_.shape_with_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::ShapeProto* TransferToClientRequest::unsafe_arena_release_shape_with_layout() {
|
|
// @@protoc_insertion_point(field_release:xla.TransferToClientRequest.shape_with_layout)
|
|
|
|
::xla::ShapeProto* temp = _impl_.shape_with_layout_;
|
|
_impl_.shape_with_layout_ = nullptr;
|
|
return temp;
|
|
}
|
|
inline ::xla::ShapeProto* TransferToClientRequest::_internal_mutable_shape_with_layout() {
|
|
|
|
if (_impl_.shape_with_layout_ == nullptr) {
|
|
auto* p = CreateMaybeMessage<::xla::ShapeProto>(GetArenaForAllocation());
|
|
_impl_.shape_with_layout_ = p;
|
|
}
|
|
return _impl_.shape_with_layout_;
|
|
}
|
|
inline ::xla::ShapeProto* TransferToClientRequest::mutable_shape_with_layout() {
|
|
::xla::ShapeProto* _msg = _internal_mutable_shape_with_layout();
|
|
// @@protoc_insertion_point(field_mutable:xla.TransferToClientRequest.shape_with_layout)
|
|
return _msg;
|
|
}
|
|
inline void TransferToClientRequest::set_allocated_shape_with_layout(::xla::ShapeProto* shape_with_layout) {
|
|
::PROTOBUF_NAMESPACE_ID::Arena* message_arena = GetArenaForAllocation();
|
|
if (message_arena == nullptr) {
|
|
delete reinterpret_cast< ::PROTOBUF_NAMESPACE_ID::MessageLite*>(_impl_.shape_with_layout_);
|
|
}
|
|
if (shape_with_layout) {
|
|
::PROTOBUF_NAMESPACE_ID::Arena* submessage_arena =
|
|
::PROTOBUF_NAMESPACE_ID::Arena::InternalGetOwningArena(
|
|
reinterpret_cast<::PROTOBUF_NAMESPACE_ID::MessageLite*>(shape_with_layout));
|
|
if (message_arena != submessage_arena) {
|
|
shape_with_layout = ::PROTOBUF_NAMESPACE_ID::internal::GetOwnedMessage(
|
|
message_arena, shape_with_layout, submessage_arena);
|
|
}
|
|
|
|
} else {
|
|
|
|
}
|
|
_impl_.shape_with_layout_ = shape_with_layout;
|
|
// @@protoc_insertion_point(field_set_allocated:xla.TransferToClientRequest.shape_with_layout)
|
|
}
|
|
|
|
// -------------------------------------------------------------------
|
|
|
|
// TransferToClientResponse
|
|
|
|
// .xla.LiteralProto literal = 1;
|
|
inline bool TransferToClientResponse::_internal_has_literal() const {
|
|
return this != internal_default_instance() && _impl_.literal_ != nullptr;
|
|
}
|
|
inline bool TransferToClientResponse::has_literal() const {
|
|
return _internal_has_literal();
|
|
}
|
|
inline const ::xla::LiteralProto& TransferToClientResponse::_internal_literal() const {
|
|
const ::xla::LiteralProto* p = _impl_.literal_;
|
|
return p != nullptr ? *p : reinterpret_cast<const ::xla::LiteralProto&>(
|
|
::xla::_LiteralProto_default_instance_);
|
|
}
|
|
inline const ::xla::LiteralProto& TransferToClientResponse::literal() const {
|
|
// @@protoc_insertion_point(field_get:xla.TransferToClientResponse.literal)
|
|
return _internal_literal();
|
|
}
|
|
inline void TransferToClientResponse::unsafe_arena_set_allocated_literal(
|
|
::xla::LiteralProto* literal) {
|
|
if (GetArenaForAllocation() == nullptr) {
|
|
delete reinterpret_cast<::PROTOBUF_NAMESPACE_ID::MessageLite*>(_impl_.literal_);
|
|
}
|
|
_impl_.literal_ = literal;
|
|
if (literal) {
|
|
|
|
} else {
|
|
|
|
}
|
|
// @@protoc_insertion_point(field_unsafe_arena_set_allocated:xla.TransferToClientResponse.literal)
|
|
}
|
|
inline ::xla::LiteralProto* TransferToClientResponse::release_literal() {
|
|
|
|
::xla::LiteralProto* temp = _impl_.literal_;
|
|
_impl_.literal_ = 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::LiteralProto* TransferToClientResponse::unsafe_arena_release_literal() {
|
|
// @@protoc_insertion_point(field_release:xla.TransferToClientResponse.literal)
|
|
|
|
::xla::LiteralProto* temp = _impl_.literal_;
|
|
_impl_.literal_ = nullptr;
|
|
return temp;
|
|
}
|
|
inline ::xla::LiteralProto* TransferToClientResponse::_internal_mutable_literal() {
|
|
|
|
if (_impl_.literal_ == nullptr) {
|
|
auto* p = CreateMaybeMessage<::xla::LiteralProto>(GetArenaForAllocation());
|
|
_impl_.literal_ = p;
|
|
}
|
|
return _impl_.literal_;
|
|
}
|
|
inline ::xla::LiteralProto* TransferToClientResponse::mutable_literal() {
|
|
::xla::LiteralProto* _msg = _internal_mutable_literal();
|
|
// @@protoc_insertion_point(field_mutable:xla.TransferToClientResponse.literal)
|
|
return _msg;
|
|
}
|
|
inline void TransferToClientResponse::set_allocated_literal(::xla::LiteralProto* literal) {
|
|
::PROTOBUF_NAMESPACE_ID::Arena* message_arena = GetArenaForAllocation();
|
|
if (message_arena == nullptr) {
|
|
delete reinterpret_cast< ::PROTOBUF_NAMESPACE_ID::MessageLite*>(_impl_.literal_);
|
|
}
|
|
if (literal) {
|
|
::PROTOBUF_NAMESPACE_ID::Arena* submessage_arena =
|
|
::PROTOBUF_NAMESPACE_ID::Arena::InternalGetOwningArena(
|
|
reinterpret_cast<::PROTOBUF_NAMESPACE_ID::MessageLite*>(literal));
|
|
if (message_arena != submessage_arena) {
|
|
literal = ::PROTOBUF_NAMESPACE_ID::internal::GetOwnedMessage(
|
|
message_arena, literal, submessage_arena);
|
|
}
|
|
|
|
} else {
|
|
|
|
}
|
|
_impl_.literal_ = literal;
|
|
// @@protoc_insertion_point(field_set_allocated:xla.TransferToClientResponse.literal)
|
|
}
|
|
|
|
// -------------------------------------------------------------------
|
|
|
|
// TransferToServerRequest
|
|
|
|
// .xla.LiteralProto literal = 1;
|
|
inline bool TransferToServerRequest::_internal_has_literal() const {
|
|
return this != internal_default_instance() && _impl_.literal_ != nullptr;
|
|
}
|
|
inline bool TransferToServerRequest::has_literal() const {
|
|
return _internal_has_literal();
|
|
}
|
|
inline const ::xla::LiteralProto& TransferToServerRequest::_internal_literal() const {
|
|
const ::xla::LiteralProto* p = _impl_.literal_;
|
|
return p != nullptr ? *p : reinterpret_cast<const ::xla::LiteralProto&>(
|
|
::xla::_LiteralProto_default_instance_);
|
|
}
|
|
inline const ::xla::LiteralProto& TransferToServerRequest::literal() const {
|
|
// @@protoc_insertion_point(field_get:xla.TransferToServerRequest.literal)
|
|
return _internal_literal();
|
|
}
|
|
inline void TransferToServerRequest::unsafe_arena_set_allocated_literal(
|
|
::xla::LiteralProto* literal) {
|
|
if (GetArenaForAllocation() == nullptr) {
|
|
delete reinterpret_cast<::PROTOBUF_NAMESPACE_ID::MessageLite*>(_impl_.literal_);
|
|
}
|
|
_impl_.literal_ = literal;
|
|
if (literal) {
|
|
|
|
} else {
|
|
|
|
}
|
|
// @@protoc_insertion_point(field_unsafe_arena_set_allocated:xla.TransferToServerRequest.literal)
|
|
}
|
|
inline ::xla::LiteralProto* TransferToServerRequest::release_literal() {
|
|
|
|
::xla::LiteralProto* temp = _impl_.literal_;
|
|
_impl_.literal_ = 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::LiteralProto* TransferToServerRequest::unsafe_arena_release_literal() {
|
|
// @@protoc_insertion_point(field_release:xla.TransferToServerRequest.literal)
|
|
|
|
::xla::LiteralProto* temp = _impl_.literal_;
|
|
_impl_.literal_ = nullptr;
|
|
return temp;
|
|
}
|
|
inline ::xla::LiteralProto* TransferToServerRequest::_internal_mutable_literal() {
|
|
|
|
if (_impl_.literal_ == nullptr) {
|
|
auto* p = CreateMaybeMessage<::xla::LiteralProto>(GetArenaForAllocation());
|
|
_impl_.literal_ = p;
|
|
}
|
|
return _impl_.literal_;
|
|
}
|
|
inline ::xla::LiteralProto* TransferToServerRequest::mutable_literal() {
|
|
::xla::LiteralProto* _msg = _internal_mutable_literal();
|
|
// @@protoc_insertion_point(field_mutable:xla.TransferToServerRequest.literal)
|
|
return _msg;
|
|
}
|
|
inline void TransferToServerRequest::set_allocated_literal(::xla::LiteralProto* literal) {
|
|
::PROTOBUF_NAMESPACE_ID::Arena* message_arena = GetArenaForAllocation();
|
|
if (message_arena == nullptr) {
|
|
delete reinterpret_cast< ::PROTOBUF_NAMESPACE_ID::MessageLite*>(_impl_.literal_);
|
|
}
|
|
if (literal) {
|
|
::PROTOBUF_NAMESPACE_ID::Arena* submessage_arena =
|
|
::PROTOBUF_NAMESPACE_ID::Arena::InternalGetOwningArena(
|
|
reinterpret_cast<::PROTOBUF_NAMESPACE_ID::MessageLite*>(literal));
|
|
if (message_arena != submessage_arena) {
|
|
literal = ::PROTOBUF_NAMESPACE_ID::internal::GetOwnedMessage(
|
|
message_arena, literal, submessage_arena);
|
|
}
|
|
|
|
} else {
|
|
|
|
}
|
|
_impl_.literal_ = literal;
|
|
// @@protoc_insertion_point(field_set_allocated:xla.TransferToServerRequest.literal)
|
|
}
|
|
|
|
// .xla.DeviceHandle device_handle = 2;
|
|
inline bool TransferToServerRequest::_internal_has_device_handle() const {
|
|
return this != internal_default_instance() && _impl_.device_handle_ != nullptr;
|
|
}
|
|
inline bool TransferToServerRequest::has_device_handle() const {
|
|
return _internal_has_device_handle();
|
|
}
|
|
inline const ::xla::DeviceHandle& TransferToServerRequest::_internal_device_handle() const {
|
|
const ::xla::DeviceHandle* p = _impl_.device_handle_;
|
|
return p != nullptr ? *p : reinterpret_cast<const ::xla::DeviceHandle&>(
|
|
::xla::_DeviceHandle_default_instance_);
|
|
}
|
|
inline const ::xla::DeviceHandle& TransferToServerRequest::device_handle() const {
|
|
// @@protoc_insertion_point(field_get:xla.TransferToServerRequest.device_handle)
|
|
return _internal_device_handle();
|
|
}
|
|
inline void TransferToServerRequest::unsafe_arena_set_allocated_device_handle(
|
|
::xla::DeviceHandle* device_handle) {
|
|
if (GetArenaForAllocation() == nullptr) {
|
|
delete reinterpret_cast<::PROTOBUF_NAMESPACE_ID::MessageLite*>(_impl_.device_handle_);
|
|
}
|
|
_impl_.device_handle_ = device_handle;
|
|
if (device_handle) {
|
|
|
|
} else {
|
|
|
|
}
|
|
// @@protoc_insertion_point(field_unsafe_arena_set_allocated:xla.TransferToServerRequest.device_handle)
|
|
}
|
|
inline ::xla::DeviceHandle* TransferToServerRequest::release_device_handle() {
|
|
|
|
::xla::DeviceHandle* temp = _impl_.device_handle_;
|
|
_impl_.device_handle_ = 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::DeviceHandle* TransferToServerRequest::unsafe_arena_release_device_handle() {
|
|
// @@protoc_insertion_point(field_release:xla.TransferToServerRequest.device_handle)
|
|
|
|
::xla::DeviceHandle* temp = _impl_.device_handle_;
|
|
_impl_.device_handle_ = nullptr;
|
|
return temp;
|
|
}
|
|
inline ::xla::DeviceHandle* TransferToServerRequest::_internal_mutable_device_handle() {
|
|
|
|
if (_impl_.device_handle_ == nullptr) {
|
|
auto* p = CreateMaybeMessage<::xla::DeviceHandle>(GetArenaForAllocation());
|
|
_impl_.device_handle_ = p;
|
|
}
|
|
return _impl_.device_handle_;
|
|
}
|
|
inline ::xla::DeviceHandle* TransferToServerRequest::mutable_device_handle() {
|
|
::xla::DeviceHandle* _msg = _internal_mutable_device_handle();
|
|
// @@protoc_insertion_point(field_mutable:xla.TransferToServerRequest.device_handle)
|
|
return _msg;
|
|
}
|
|
inline void TransferToServerRequest::set_allocated_device_handle(::xla::DeviceHandle* device_handle) {
|
|
::PROTOBUF_NAMESPACE_ID::Arena* message_arena = GetArenaForAllocation();
|
|
if (message_arena == nullptr) {
|
|
delete reinterpret_cast< ::PROTOBUF_NAMESPACE_ID::MessageLite*>(_impl_.device_handle_);
|
|
}
|
|
if (device_handle) {
|
|
::PROTOBUF_NAMESPACE_ID::Arena* submessage_arena =
|
|
::PROTOBUF_NAMESPACE_ID::Arena::InternalGetOwningArena(
|
|
reinterpret_cast<::PROTOBUF_NAMESPACE_ID::MessageLite*>(device_handle));
|
|
if (message_arena != submessage_arena) {
|
|
device_handle = ::PROTOBUF_NAMESPACE_ID::internal::GetOwnedMessage(
|
|
message_arena, device_handle, submessage_arena);
|
|
}
|
|
|
|
} else {
|
|
|
|
}
|
|
_impl_.device_handle_ = device_handle;
|
|
// @@protoc_insertion_point(field_set_allocated:xla.TransferToServerRequest.device_handle)
|
|
}
|
|
|
|
// -------------------------------------------------------------------
|
|
|
|
// TransferToServerResponse
|
|
|
|
// .xla.GlobalDataHandle data = 1;
|
|
inline bool TransferToServerResponse::_internal_has_data() const {
|
|
return this != internal_default_instance() && _impl_.data_ != nullptr;
|
|
}
|
|
inline bool TransferToServerResponse::has_data() const {
|
|
return _internal_has_data();
|
|
}
|
|
inline const ::xla::GlobalDataHandle& TransferToServerResponse::_internal_data() const {
|
|
const ::xla::GlobalDataHandle* p = _impl_.data_;
|
|
return p != nullptr ? *p : reinterpret_cast<const ::xla::GlobalDataHandle&>(
|
|
::xla::_GlobalDataHandle_default_instance_);
|
|
}
|
|
inline const ::xla::GlobalDataHandle& TransferToServerResponse::data() const {
|
|
// @@protoc_insertion_point(field_get:xla.TransferToServerResponse.data)
|
|
return _internal_data();
|
|
}
|
|
inline void TransferToServerResponse::unsafe_arena_set_allocated_data(
|
|
::xla::GlobalDataHandle* data) {
|
|
if (GetArenaForAllocation() == nullptr) {
|
|
delete reinterpret_cast<::PROTOBUF_NAMESPACE_ID::MessageLite*>(_impl_.data_);
|
|
}
|
|
_impl_.data_ = data;
|
|
if (data) {
|
|
|
|
} else {
|
|
|
|
}
|
|
// @@protoc_insertion_point(field_unsafe_arena_set_allocated:xla.TransferToServerResponse.data)
|
|
}
|
|
inline ::xla::GlobalDataHandle* TransferToServerResponse::release_data() {
|
|
|
|
::xla::GlobalDataHandle* temp = _impl_.data_;
|
|
_impl_.data_ = 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::GlobalDataHandle* TransferToServerResponse::unsafe_arena_release_data() {
|
|
// @@protoc_insertion_point(field_release:xla.TransferToServerResponse.data)
|
|
|
|
::xla::GlobalDataHandle* temp = _impl_.data_;
|
|
_impl_.data_ = nullptr;
|
|
return temp;
|
|
}
|
|
inline ::xla::GlobalDataHandle* TransferToServerResponse::_internal_mutable_data() {
|
|
|
|
if (_impl_.data_ == nullptr) {
|
|
auto* p = CreateMaybeMessage<::xla::GlobalDataHandle>(GetArenaForAllocation());
|
|
_impl_.data_ = p;
|
|
}
|
|
return _impl_.data_;
|
|
}
|
|
inline ::xla::GlobalDataHandle* TransferToServerResponse::mutable_data() {
|
|
::xla::GlobalDataHandle* _msg = _internal_mutable_data();
|
|
// @@protoc_insertion_point(field_mutable:xla.TransferToServerResponse.data)
|
|
return _msg;
|
|
}
|
|
inline void TransferToServerResponse::set_allocated_data(::xla::GlobalDataHandle* data) {
|
|
::PROTOBUF_NAMESPACE_ID::Arena* message_arena = GetArenaForAllocation();
|
|
if (message_arena == nullptr) {
|
|
delete reinterpret_cast< ::PROTOBUF_NAMESPACE_ID::MessageLite*>(_impl_.data_);
|
|
}
|
|
if (data) {
|
|
::PROTOBUF_NAMESPACE_ID::Arena* submessage_arena =
|
|
::PROTOBUF_NAMESPACE_ID::Arena::InternalGetOwningArena(
|
|
reinterpret_cast<::PROTOBUF_NAMESPACE_ID::MessageLite*>(data));
|
|
if (message_arena != submessage_arena) {
|
|
data = ::PROTOBUF_NAMESPACE_ID::internal::GetOwnedMessage(
|
|
message_arena, data, submessage_arena);
|
|
}
|
|
|
|
} else {
|
|
|
|
}
|
|
_impl_.data_ = data;
|
|
// @@protoc_insertion_point(field_set_allocated:xla.TransferToServerResponse.data)
|
|
}
|
|
|
|
// -------------------------------------------------------------------
|
|
|
|
// TransferToInfeedRequest
|
|
|
|
// .xla.LiteralProto literal = 1;
|
|
inline bool TransferToInfeedRequest::_internal_has_literal() const {
|
|
return this != internal_default_instance() && _impl_.literal_ != nullptr;
|
|
}
|
|
inline bool TransferToInfeedRequest::has_literal() const {
|
|
return _internal_has_literal();
|
|
}
|
|
inline const ::xla::LiteralProto& TransferToInfeedRequest::_internal_literal() const {
|
|
const ::xla::LiteralProto* p = _impl_.literal_;
|
|
return p != nullptr ? *p : reinterpret_cast<const ::xla::LiteralProto&>(
|
|
::xla::_LiteralProto_default_instance_);
|
|
}
|
|
inline const ::xla::LiteralProto& TransferToInfeedRequest::literal() const {
|
|
// @@protoc_insertion_point(field_get:xla.TransferToInfeedRequest.literal)
|
|
return _internal_literal();
|
|
}
|
|
inline void TransferToInfeedRequest::unsafe_arena_set_allocated_literal(
|
|
::xla::LiteralProto* literal) {
|
|
if (GetArenaForAllocation() == nullptr) {
|
|
delete reinterpret_cast<::PROTOBUF_NAMESPACE_ID::MessageLite*>(_impl_.literal_);
|
|
}
|
|
_impl_.literal_ = literal;
|
|
if (literal) {
|
|
|
|
} else {
|
|
|
|
}
|
|
// @@protoc_insertion_point(field_unsafe_arena_set_allocated:xla.TransferToInfeedRequest.literal)
|
|
}
|
|
inline ::xla::LiteralProto* TransferToInfeedRequest::release_literal() {
|
|
|
|
::xla::LiteralProto* temp = _impl_.literal_;
|
|
_impl_.literal_ = 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::LiteralProto* TransferToInfeedRequest::unsafe_arena_release_literal() {
|
|
// @@protoc_insertion_point(field_release:xla.TransferToInfeedRequest.literal)
|
|
|
|
::xla::LiteralProto* temp = _impl_.literal_;
|
|
_impl_.literal_ = nullptr;
|
|
return temp;
|
|
}
|
|
inline ::xla::LiteralProto* TransferToInfeedRequest::_internal_mutable_literal() {
|
|
|
|
if (_impl_.literal_ == nullptr) {
|
|
auto* p = CreateMaybeMessage<::xla::LiteralProto>(GetArenaForAllocation());
|
|
_impl_.literal_ = p;
|
|
}
|
|
return _impl_.literal_;
|
|
}
|
|
inline ::xla::LiteralProto* TransferToInfeedRequest::mutable_literal() {
|
|
::xla::LiteralProto* _msg = _internal_mutable_literal();
|
|
// @@protoc_insertion_point(field_mutable:xla.TransferToInfeedRequest.literal)
|
|
return _msg;
|
|
}
|
|
inline void TransferToInfeedRequest::set_allocated_literal(::xla::LiteralProto* literal) {
|
|
::PROTOBUF_NAMESPACE_ID::Arena* message_arena = GetArenaForAllocation();
|
|
if (message_arena == nullptr) {
|
|
delete reinterpret_cast< ::PROTOBUF_NAMESPACE_ID::MessageLite*>(_impl_.literal_);
|
|
}
|
|
if (literal) {
|
|
::PROTOBUF_NAMESPACE_ID::Arena* submessage_arena =
|
|
::PROTOBUF_NAMESPACE_ID::Arena::InternalGetOwningArena(
|
|
reinterpret_cast<::PROTOBUF_NAMESPACE_ID::MessageLite*>(literal));
|
|
if (message_arena != submessage_arena) {
|
|
literal = ::PROTOBUF_NAMESPACE_ID::internal::GetOwnedMessage(
|
|
message_arena, literal, submessage_arena);
|
|
}
|
|
|
|
} else {
|
|
|
|
}
|
|
_impl_.literal_ = literal;
|
|
// @@protoc_insertion_point(field_set_allocated:xla.TransferToInfeedRequest.literal)
|
|
}
|
|
|
|
// int64 replica_id = 2;
|
|
inline void TransferToInfeedRequest::clear_replica_id() {
|
|
_impl_.replica_id_ = int64_t{0};
|
|
}
|
|
inline int64_t TransferToInfeedRequest::_internal_replica_id() const {
|
|
return _impl_.replica_id_;
|
|
}
|
|
inline int64_t TransferToInfeedRequest::replica_id() const {
|
|
// @@protoc_insertion_point(field_get:xla.TransferToInfeedRequest.replica_id)
|
|
return _internal_replica_id();
|
|
}
|
|
inline void TransferToInfeedRequest::_internal_set_replica_id(int64_t value) {
|
|
|
|
_impl_.replica_id_ = value;
|
|
}
|
|
inline void TransferToInfeedRequest::set_replica_id(int64_t value) {
|
|
_internal_set_replica_id(value);
|
|
// @@protoc_insertion_point(field_set:xla.TransferToInfeedRequest.replica_id)
|
|
}
|
|
|
|
// .xla.DeviceHandle device_handle = 3;
|
|
inline bool TransferToInfeedRequest::_internal_has_device_handle() const {
|
|
return this != internal_default_instance() && _impl_.device_handle_ != nullptr;
|
|
}
|
|
inline bool TransferToInfeedRequest::has_device_handle() const {
|
|
return _internal_has_device_handle();
|
|
}
|
|
inline const ::xla::DeviceHandle& TransferToInfeedRequest::_internal_device_handle() const {
|
|
const ::xla::DeviceHandle* p = _impl_.device_handle_;
|
|
return p != nullptr ? *p : reinterpret_cast<const ::xla::DeviceHandle&>(
|
|
::xla::_DeviceHandle_default_instance_);
|
|
}
|
|
inline const ::xla::DeviceHandle& TransferToInfeedRequest::device_handle() const {
|
|
// @@protoc_insertion_point(field_get:xla.TransferToInfeedRequest.device_handle)
|
|
return _internal_device_handle();
|
|
}
|
|
inline void TransferToInfeedRequest::unsafe_arena_set_allocated_device_handle(
|
|
::xla::DeviceHandle* device_handle) {
|
|
if (GetArenaForAllocation() == nullptr) {
|
|
delete reinterpret_cast<::PROTOBUF_NAMESPACE_ID::MessageLite*>(_impl_.device_handle_);
|
|
}
|
|
_impl_.device_handle_ = device_handle;
|
|
if (device_handle) {
|
|
|
|
} else {
|
|
|
|
}
|
|
// @@protoc_insertion_point(field_unsafe_arena_set_allocated:xla.TransferToInfeedRequest.device_handle)
|
|
}
|
|
inline ::xla::DeviceHandle* TransferToInfeedRequest::release_device_handle() {
|
|
|
|
::xla::DeviceHandle* temp = _impl_.device_handle_;
|
|
_impl_.device_handle_ = 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::DeviceHandle* TransferToInfeedRequest::unsafe_arena_release_device_handle() {
|
|
// @@protoc_insertion_point(field_release:xla.TransferToInfeedRequest.device_handle)
|
|
|
|
::xla::DeviceHandle* temp = _impl_.device_handle_;
|
|
_impl_.device_handle_ = nullptr;
|
|
return temp;
|
|
}
|
|
inline ::xla::DeviceHandle* TransferToInfeedRequest::_internal_mutable_device_handle() {
|
|
|
|
if (_impl_.device_handle_ == nullptr) {
|
|
auto* p = CreateMaybeMessage<::xla::DeviceHandle>(GetArenaForAllocation());
|
|
_impl_.device_handle_ = p;
|
|
}
|
|
return _impl_.device_handle_;
|
|
}
|
|
inline ::xla::DeviceHandle* TransferToInfeedRequest::mutable_device_handle() {
|
|
::xla::DeviceHandle* _msg = _internal_mutable_device_handle();
|
|
// @@protoc_insertion_point(field_mutable:xla.TransferToInfeedRequest.device_handle)
|
|
return _msg;
|
|
}
|
|
inline void TransferToInfeedRequest::set_allocated_device_handle(::xla::DeviceHandle* device_handle) {
|
|
::PROTOBUF_NAMESPACE_ID::Arena* message_arena = GetArenaForAllocation();
|
|
if (message_arena == nullptr) {
|
|
delete reinterpret_cast< ::PROTOBUF_NAMESPACE_ID::MessageLite*>(_impl_.device_handle_);
|
|
}
|
|
if (device_handle) {
|
|
::PROTOBUF_NAMESPACE_ID::Arena* submessage_arena =
|
|
::PROTOBUF_NAMESPACE_ID::Arena::InternalGetOwningArena(
|
|
reinterpret_cast<::PROTOBUF_NAMESPACE_ID::MessageLite*>(device_handle));
|
|
if (message_arena != submessage_arena) {
|
|
device_handle = ::PROTOBUF_NAMESPACE_ID::internal::GetOwnedMessage(
|
|
message_arena, device_handle, submessage_arena);
|
|
}
|
|
|
|
} else {
|
|
|
|
}
|
|
_impl_.device_handle_ = device_handle;
|
|
// @@protoc_insertion_point(field_set_allocated:xla.TransferToInfeedRequest.device_handle)
|
|
}
|
|
|
|
// -------------------------------------------------------------------
|
|
|
|
// TransferToInfeedResponse
|
|
|
|
// -------------------------------------------------------------------
|
|
|
|
// TransferFromOutfeedRequest
|
|
|
|
// .xla.ShapeProto shape_with_layout = 1;
|
|
inline bool TransferFromOutfeedRequest::_internal_has_shape_with_layout() const {
|
|
return this != internal_default_instance() && _impl_.shape_with_layout_ != nullptr;
|
|
}
|
|
inline bool TransferFromOutfeedRequest::has_shape_with_layout() const {
|
|
return _internal_has_shape_with_layout();
|
|
}
|
|
inline const ::xla::ShapeProto& TransferFromOutfeedRequest::_internal_shape_with_layout() const {
|
|
const ::xla::ShapeProto* p = _impl_.shape_with_layout_;
|
|
return p != nullptr ? *p : reinterpret_cast<const ::xla::ShapeProto&>(
|
|
::xla::_ShapeProto_default_instance_);
|
|
}
|
|
inline const ::xla::ShapeProto& TransferFromOutfeedRequest::shape_with_layout() const {
|
|
// @@protoc_insertion_point(field_get:xla.TransferFromOutfeedRequest.shape_with_layout)
|
|
return _internal_shape_with_layout();
|
|
}
|
|
inline void TransferFromOutfeedRequest::unsafe_arena_set_allocated_shape_with_layout(
|
|
::xla::ShapeProto* shape_with_layout) {
|
|
if (GetArenaForAllocation() == nullptr) {
|
|
delete reinterpret_cast<::PROTOBUF_NAMESPACE_ID::MessageLite*>(_impl_.shape_with_layout_);
|
|
}
|
|
_impl_.shape_with_layout_ = shape_with_layout;
|
|
if (shape_with_layout) {
|
|
|
|
} else {
|
|
|
|
}
|
|
// @@protoc_insertion_point(field_unsafe_arena_set_allocated:xla.TransferFromOutfeedRequest.shape_with_layout)
|
|
}
|
|
inline ::xla::ShapeProto* TransferFromOutfeedRequest::release_shape_with_layout() {
|
|
|
|
::xla::ShapeProto* temp = _impl_.shape_with_layout_;
|
|
_impl_.shape_with_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::ShapeProto* TransferFromOutfeedRequest::unsafe_arena_release_shape_with_layout() {
|
|
// @@protoc_insertion_point(field_release:xla.TransferFromOutfeedRequest.shape_with_layout)
|
|
|
|
::xla::ShapeProto* temp = _impl_.shape_with_layout_;
|
|
_impl_.shape_with_layout_ = nullptr;
|
|
return temp;
|
|
}
|
|
inline ::xla::ShapeProto* TransferFromOutfeedRequest::_internal_mutable_shape_with_layout() {
|
|
|
|
if (_impl_.shape_with_layout_ == nullptr) {
|
|
auto* p = CreateMaybeMessage<::xla::ShapeProto>(GetArenaForAllocation());
|
|
_impl_.shape_with_layout_ = p;
|
|
}
|
|
return _impl_.shape_with_layout_;
|
|
}
|
|
inline ::xla::ShapeProto* TransferFromOutfeedRequest::mutable_shape_with_layout() {
|
|
::xla::ShapeProto* _msg = _internal_mutable_shape_with_layout();
|
|
// @@protoc_insertion_point(field_mutable:xla.TransferFromOutfeedRequest.shape_with_layout)
|
|
return _msg;
|
|
}
|
|
inline void TransferFromOutfeedRequest::set_allocated_shape_with_layout(::xla::ShapeProto* shape_with_layout) {
|
|
::PROTOBUF_NAMESPACE_ID::Arena* message_arena = GetArenaForAllocation();
|
|
if (message_arena == nullptr) {
|
|
delete reinterpret_cast< ::PROTOBUF_NAMESPACE_ID::MessageLite*>(_impl_.shape_with_layout_);
|
|
}
|
|
if (shape_with_layout) {
|
|
::PROTOBUF_NAMESPACE_ID::Arena* submessage_arena =
|
|
::PROTOBUF_NAMESPACE_ID::Arena::InternalGetOwningArena(
|
|
reinterpret_cast<::PROTOBUF_NAMESPACE_ID::MessageLite*>(shape_with_layout));
|
|
if (message_arena != submessage_arena) {
|
|
shape_with_layout = ::PROTOBUF_NAMESPACE_ID::internal::GetOwnedMessage(
|
|
message_arena, shape_with_layout, submessage_arena);
|
|
}
|
|
|
|
} else {
|
|
|
|
}
|
|
_impl_.shape_with_layout_ = shape_with_layout;
|
|
// @@protoc_insertion_point(field_set_allocated:xla.TransferFromOutfeedRequest.shape_with_layout)
|
|
}
|
|
|
|
// int64 replica_id = 2;
|
|
inline void TransferFromOutfeedRequest::clear_replica_id() {
|
|
_impl_.replica_id_ = int64_t{0};
|
|
}
|
|
inline int64_t TransferFromOutfeedRequest::_internal_replica_id() const {
|
|
return _impl_.replica_id_;
|
|
}
|
|
inline int64_t TransferFromOutfeedRequest::replica_id() const {
|
|
// @@protoc_insertion_point(field_get:xla.TransferFromOutfeedRequest.replica_id)
|
|
return _internal_replica_id();
|
|
}
|
|
inline void TransferFromOutfeedRequest::_internal_set_replica_id(int64_t value) {
|
|
|
|
_impl_.replica_id_ = value;
|
|
}
|
|
inline void TransferFromOutfeedRequest::set_replica_id(int64_t value) {
|
|
_internal_set_replica_id(value);
|
|
// @@protoc_insertion_point(field_set:xla.TransferFromOutfeedRequest.replica_id)
|
|
}
|
|
|
|
// .xla.DeviceHandle device_handle = 3;
|
|
inline bool TransferFromOutfeedRequest::_internal_has_device_handle() const {
|
|
return this != internal_default_instance() && _impl_.device_handle_ != nullptr;
|
|
}
|
|
inline bool TransferFromOutfeedRequest::has_device_handle() const {
|
|
return _internal_has_device_handle();
|
|
}
|
|
inline const ::xla::DeviceHandle& TransferFromOutfeedRequest::_internal_device_handle() const {
|
|
const ::xla::DeviceHandle* p = _impl_.device_handle_;
|
|
return p != nullptr ? *p : reinterpret_cast<const ::xla::DeviceHandle&>(
|
|
::xla::_DeviceHandle_default_instance_);
|
|
}
|
|
inline const ::xla::DeviceHandle& TransferFromOutfeedRequest::device_handle() const {
|
|
// @@protoc_insertion_point(field_get:xla.TransferFromOutfeedRequest.device_handle)
|
|
return _internal_device_handle();
|
|
}
|
|
inline void TransferFromOutfeedRequest::unsafe_arena_set_allocated_device_handle(
|
|
::xla::DeviceHandle* device_handle) {
|
|
if (GetArenaForAllocation() == nullptr) {
|
|
delete reinterpret_cast<::PROTOBUF_NAMESPACE_ID::MessageLite*>(_impl_.device_handle_);
|
|
}
|
|
_impl_.device_handle_ = device_handle;
|
|
if (device_handle) {
|
|
|
|
} else {
|
|
|
|
}
|
|
// @@protoc_insertion_point(field_unsafe_arena_set_allocated:xla.TransferFromOutfeedRequest.device_handle)
|
|
}
|
|
inline ::xla::DeviceHandle* TransferFromOutfeedRequest::release_device_handle() {
|
|
|
|
::xla::DeviceHandle* temp = _impl_.device_handle_;
|
|
_impl_.device_handle_ = 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::DeviceHandle* TransferFromOutfeedRequest::unsafe_arena_release_device_handle() {
|
|
// @@protoc_insertion_point(field_release:xla.TransferFromOutfeedRequest.device_handle)
|
|
|
|
::xla::DeviceHandle* temp = _impl_.device_handle_;
|
|
_impl_.device_handle_ = nullptr;
|
|
return temp;
|
|
}
|
|
inline ::xla::DeviceHandle* TransferFromOutfeedRequest::_internal_mutable_device_handle() {
|
|
|
|
if (_impl_.device_handle_ == nullptr) {
|
|
auto* p = CreateMaybeMessage<::xla::DeviceHandle>(GetArenaForAllocation());
|
|
_impl_.device_handle_ = p;
|
|
}
|
|
return _impl_.device_handle_;
|
|
}
|
|
inline ::xla::DeviceHandle* TransferFromOutfeedRequest::mutable_device_handle() {
|
|
::xla::DeviceHandle* _msg = _internal_mutable_device_handle();
|
|
// @@protoc_insertion_point(field_mutable:xla.TransferFromOutfeedRequest.device_handle)
|
|
return _msg;
|
|
}
|
|
inline void TransferFromOutfeedRequest::set_allocated_device_handle(::xla::DeviceHandle* device_handle) {
|
|
::PROTOBUF_NAMESPACE_ID::Arena* message_arena = GetArenaForAllocation();
|
|
if (message_arena == nullptr) {
|
|
delete reinterpret_cast< ::PROTOBUF_NAMESPACE_ID::MessageLite*>(_impl_.device_handle_);
|
|
}
|
|
if (device_handle) {
|
|
::PROTOBUF_NAMESPACE_ID::Arena* submessage_arena =
|
|
::PROTOBUF_NAMESPACE_ID::Arena::InternalGetOwningArena(
|
|
reinterpret_cast<::PROTOBUF_NAMESPACE_ID::MessageLite*>(device_handle));
|
|
if (message_arena != submessage_arena) {
|
|
device_handle = ::PROTOBUF_NAMESPACE_ID::internal::GetOwnedMessage(
|
|
message_arena, device_handle, submessage_arena);
|
|
}
|
|
|
|
} else {
|
|
|
|
}
|
|
_impl_.device_handle_ = device_handle;
|
|
// @@protoc_insertion_point(field_set_allocated:xla.TransferFromOutfeedRequest.device_handle)
|
|
}
|
|
|
|
// -------------------------------------------------------------------
|
|
|
|
// TransferFromOutfeedResponse
|
|
|
|
// .xla.LiteralProto literal = 1;
|
|
inline bool TransferFromOutfeedResponse::_internal_has_literal() const {
|
|
return this != internal_default_instance() && _impl_.literal_ != nullptr;
|
|
}
|
|
inline bool TransferFromOutfeedResponse::has_literal() const {
|
|
return _internal_has_literal();
|
|
}
|
|
inline const ::xla::LiteralProto& TransferFromOutfeedResponse::_internal_literal() const {
|
|
const ::xla::LiteralProto* p = _impl_.literal_;
|
|
return p != nullptr ? *p : reinterpret_cast<const ::xla::LiteralProto&>(
|
|
::xla::_LiteralProto_default_instance_);
|
|
}
|
|
inline const ::xla::LiteralProto& TransferFromOutfeedResponse::literal() const {
|
|
// @@protoc_insertion_point(field_get:xla.TransferFromOutfeedResponse.literal)
|
|
return _internal_literal();
|
|
}
|
|
inline void TransferFromOutfeedResponse::unsafe_arena_set_allocated_literal(
|
|
::xla::LiteralProto* literal) {
|
|
if (GetArenaForAllocation() == nullptr) {
|
|
delete reinterpret_cast<::PROTOBUF_NAMESPACE_ID::MessageLite*>(_impl_.literal_);
|
|
}
|
|
_impl_.literal_ = literal;
|
|
if (literal) {
|
|
|
|
} else {
|
|
|
|
}
|
|
// @@protoc_insertion_point(field_unsafe_arena_set_allocated:xla.TransferFromOutfeedResponse.literal)
|
|
}
|
|
inline ::xla::LiteralProto* TransferFromOutfeedResponse::release_literal() {
|
|
|
|
::xla::LiteralProto* temp = _impl_.literal_;
|
|
_impl_.literal_ = 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::LiteralProto* TransferFromOutfeedResponse::unsafe_arena_release_literal() {
|
|
// @@protoc_insertion_point(field_release:xla.TransferFromOutfeedResponse.literal)
|
|
|
|
::xla::LiteralProto* temp = _impl_.literal_;
|
|
_impl_.literal_ = nullptr;
|
|
return temp;
|
|
}
|
|
inline ::xla::LiteralProto* TransferFromOutfeedResponse::_internal_mutable_literal() {
|
|
|
|
if (_impl_.literal_ == nullptr) {
|
|
auto* p = CreateMaybeMessage<::xla::LiteralProto>(GetArenaForAllocation());
|
|
_impl_.literal_ = p;
|
|
}
|
|
return _impl_.literal_;
|
|
}
|
|
inline ::xla::LiteralProto* TransferFromOutfeedResponse::mutable_literal() {
|
|
::xla::LiteralProto* _msg = _internal_mutable_literal();
|
|
// @@protoc_insertion_point(field_mutable:xla.TransferFromOutfeedResponse.literal)
|
|
return _msg;
|
|
}
|
|
inline void TransferFromOutfeedResponse::set_allocated_literal(::xla::LiteralProto* literal) {
|
|
::PROTOBUF_NAMESPACE_ID::Arena* message_arena = GetArenaForAllocation();
|
|
if (message_arena == nullptr) {
|
|
delete reinterpret_cast< ::PROTOBUF_NAMESPACE_ID::MessageLite*>(_impl_.literal_);
|
|
}
|
|
if (literal) {
|
|
::PROTOBUF_NAMESPACE_ID::Arena* submessage_arena =
|
|
::PROTOBUF_NAMESPACE_ID::Arena::InternalGetOwningArena(
|
|
reinterpret_cast<::PROTOBUF_NAMESPACE_ID::MessageLite*>(literal));
|
|
if (message_arena != submessage_arena) {
|
|
literal = ::PROTOBUF_NAMESPACE_ID::internal::GetOwnedMessage(
|
|
message_arena, literal, submessage_arena);
|
|
}
|
|
|
|
} else {
|
|
|
|
}
|
|
_impl_.literal_ = literal;
|
|
// @@protoc_insertion_point(field_set_allocated:xla.TransferFromOutfeedResponse.literal)
|
|
}
|
|
|
|
// -------------------------------------------------------------------
|
|
|
|
// ResetDeviceRequest
|
|
|
|
// .xla.DeviceHandle device_handle = 1;
|
|
inline bool ResetDeviceRequest::_internal_has_device_handle() const {
|
|
return this != internal_default_instance() && _impl_.device_handle_ != nullptr;
|
|
}
|
|
inline bool ResetDeviceRequest::has_device_handle() const {
|
|
return _internal_has_device_handle();
|
|
}
|
|
inline const ::xla::DeviceHandle& ResetDeviceRequest::_internal_device_handle() const {
|
|
const ::xla::DeviceHandle* p = _impl_.device_handle_;
|
|
return p != nullptr ? *p : reinterpret_cast<const ::xla::DeviceHandle&>(
|
|
::xla::_DeviceHandle_default_instance_);
|
|
}
|
|
inline const ::xla::DeviceHandle& ResetDeviceRequest::device_handle() const {
|
|
// @@protoc_insertion_point(field_get:xla.ResetDeviceRequest.device_handle)
|
|
return _internal_device_handle();
|
|
}
|
|
inline void ResetDeviceRequest::unsafe_arena_set_allocated_device_handle(
|
|
::xla::DeviceHandle* device_handle) {
|
|
if (GetArenaForAllocation() == nullptr) {
|
|
delete reinterpret_cast<::PROTOBUF_NAMESPACE_ID::MessageLite*>(_impl_.device_handle_);
|
|
}
|
|
_impl_.device_handle_ = device_handle;
|
|
if (device_handle) {
|
|
|
|
} else {
|
|
|
|
}
|
|
// @@protoc_insertion_point(field_unsafe_arena_set_allocated:xla.ResetDeviceRequest.device_handle)
|
|
}
|
|
inline ::xla::DeviceHandle* ResetDeviceRequest::release_device_handle() {
|
|
|
|
::xla::DeviceHandle* temp = _impl_.device_handle_;
|
|
_impl_.device_handle_ = 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::DeviceHandle* ResetDeviceRequest::unsafe_arena_release_device_handle() {
|
|
// @@protoc_insertion_point(field_release:xla.ResetDeviceRequest.device_handle)
|
|
|
|
::xla::DeviceHandle* temp = _impl_.device_handle_;
|
|
_impl_.device_handle_ = nullptr;
|
|
return temp;
|
|
}
|
|
inline ::xla::DeviceHandle* ResetDeviceRequest::_internal_mutable_device_handle() {
|
|
|
|
if (_impl_.device_handle_ == nullptr) {
|
|
auto* p = CreateMaybeMessage<::xla::DeviceHandle>(GetArenaForAllocation());
|
|
_impl_.device_handle_ = p;
|
|
}
|
|
return _impl_.device_handle_;
|
|
}
|
|
inline ::xla::DeviceHandle* ResetDeviceRequest::mutable_device_handle() {
|
|
::xla::DeviceHandle* _msg = _internal_mutable_device_handle();
|
|
// @@protoc_insertion_point(field_mutable:xla.ResetDeviceRequest.device_handle)
|
|
return _msg;
|
|
}
|
|
inline void ResetDeviceRequest::set_allocated_device_handle(::xla::DeviceHandle* device_handle) {
|
|
::PROTOBUF_NAMESPACE_ID::Arena* message_arena = GetArenaForAllocation();
|
|
if (message_arena == nullptr) {
|
|
delete reinterpret_cast< ::PROTOBUF_NAMESPACE_ID::MessageLite*>(_impl_.device_handle_);
|
|
}
|
|
if (device_handle) {
|
|
::PROTOBUF_NAMESPACE_ID::Arena* submessage_arena =
|
|
::PROTOBUF_NAMESPACE_ID::Arena::InternalGetOwningArena(
|
|
reinterpret_cast<::PROTOBUF_NAMESPACE_ID::MessageLite*>(device_handle));
|
|
if (message_arena != submessage_arena) {
|
|
device_handle = ::PROTOBUF_NAMESPACE_ID::internal::GetOwnedMessage(
|
|
message_arena, device_handle, submessage_arena);
|
|
}
|
|
|
|
} else {
|
|
|
|
}
|
|
_impl_.device_handle_ = device_handle;
|
|
// @@protoc_insertion_point(field_set_allocated:xla.ResetDeviceRequest.device_handle)
|
|
}
|
|
|
|
// -------------------------------------------------------------------
|
|
|
|
// ResetDeviceResponse
|
|
|
|
// -------------------------------------------------------------------
|
|
|
|
// ComputationGraphStatsRequest
|
|
|
|
// .xla.HloModuleProto computation = 1;
|
|
inline bool ComputationGraphStatsRequest::_internal_has_computation() const {
|
|
return this != internal_default_instance() && _impl_.computation_ != nullptr;
|
|
}
|
|
inline bool ComputationGraphStatsRequest::has_computation() const {
|
|
return _internal_has_computation();
|
|
}
|
|
inline const ::xla::HloModuleProto& ComputationGraphStatsRequest::_internal_computation() const {
|
|
const ::xla::HloModuleProto* p = _impl_.computation_;
|
|
return p != nullptr ? *p : reinterpret_cast<const ::xla::HloModuleProto&>(
|
|
::xla::_HloModuleProto_default_instance_);
|
|
}
|
|
inline const ::xla::HloModuleProto& ComputationGraphStatsRequest::computation() const {
|
|
// @@protoc_insertion_point(field_get:xla.ComputationGraphStatsRequest.computation)
|
|
return _internal_computation();
|
|
}
|
|
inline void ComputationGraphStatsRequest::unsafe_arena_set_allocated_computation(
|
|
::xla::HloModuleProto* computation) {
|
|
if (GetArenaForAllocation() == nullptr) {
|
|
delete reinterpret_cast<::PROTOBUF_NAMESPACE_ID::MessageLite*>(_impl_.computation_);
|
|
}
|
|
_impl_.computation_ = computation;
|
|
if (computation) {
|
|
|
|
} else {
|
|
|
|
}
|
|
// @@protoc_insertion_point(field_unsafe_arena_set_allocated:xla.ComputationGraphStatsRequest.computation)
|
|
}
|
|
inline ::xla::HloModuleProto* ComputationGraphStatsRequest::release_computation() {
|
|
|
|
::xla::HloModuleProto* temp = _impl_.computation_;
|
|
_impl_.computation_ = 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::HloModuleProto* ComputationGraphStatsRequest::unsafe_arena_release_computation() {
|
|
// @@protoc_insertion_point(field_release:xla.ComputationGraphStatsRequest.computation)
|
|
|
|
::xla::HloModuleProto* temp = _impl_.computation_;
|
|
_impl_.computation_ = nullptr;
|
|
return temp;
|
|
}
|
|
inline ::xla::HloModuleProto* ComputationGraphStatsRequest::_internal_mutable_computation() {
|
|
|
|
if (_impl_.computation_ == nullptr) {
|
|
auto* p = CreateMaybeMessage<::xla::HloModuleProto>(GetArenaForAllocation());
|
|
_impl_.computation_ = p;
|
|
}
|
|
return _impl_.computation_;
|
|
}
|
|
inline ::xla::HloModuleProto* ComputationGraphStatsRequest::mutable_computation() {
|
|
::xla::HloModuleProto* _msg = _internal_mutable_computation();
|
|
// @@protoc_insertion_point(field_mutable:xla.ComputationGraphStatsRequest.computation)
|
|
return _msg;
|
|
}
|
|
inline void ComputationGraphStatsRequest::set_allocated_computation(::xla::HloModuleProto* computation) {
|
|
::PROTOBUF_NAMESPACE_ID::Arena* message_arena = GetArenaForAllocation();
|
|
if (message_arena == nullptr) {
|
|
delete reinterpret_cast< ::PROTOBUF_NAMESPACE_ID::MessageLite*>(_impl_.computation_);
|
|
}
|
|
if (computation) {
|
|
::PROTOBUF_NAMESPACE_ID::Arena* submessage_arena =
|
|
::PROTOBUF_NAMESPACE_ID::Arena::InternalGetOwningArena(
|
|
reinterpret_cast<::PROTOBUF_NAMESPACE_ID::MessageLite*>(computation));
|
|
if (message_arena != submessage_arena) {
|
|
computation = ::PROTOBUF_NAMESPACE_ID::internal::GetOwnedMessage(
|
|
message_arena, computation, submessage_arena);
|
|
}
|
|
|
|
} else {
|
|
|
|
}
|
|
_impl_.computation_ = computation;
|
|
// @@protoc_insertion_point(field_set_allocated:xla.ComputationGraphStatsRequest.computation)
|
|
}
|
|
|
|
// .xla.DebugOptions debug_options = 2;
|
|
inline bool ComputationGraphStatsRequest::_internal_has_debug_options() const {
|
|
return this != internal_default_instance() && _impl_.debug_options_ != nullptr;
|
|
}
|
|
inline bool ComputationGraphStatsRequest::has_debug_options() const {
|
|
return _internal_has_debug_options();
|
|
}
|
|
inline void ComputationGraphStatsRequest::clear_debug_options() {
|
|
if (GetArenaForAllocation() == nullptr && _impl_.debug_options_ != nullptr) {
|
|
delete _impl_.debug_options_;
|
|
}
|
|
_impl_.debug_options_ = nullptr;
|
|
}
|
|
inline const ::xla::DebugOptions& ComputationGraphStatsRequest::_internal_debug_options() const {
|
|
const ::xla::DebugOptions* p = _impl_.debug_options_;
|
|
return p != nullptr ? *p : reinterpret_cast<const ::xla::DebugOptions&>(
|
|
::xla::_DebugOptions_default_instance_);
|
|
}
|
|
inline const ::xla::DebugOptions& ComputationGraphStatsRequest::debug_options() const {
|
|
// @@protoc_insertion_point(field_get:xla.ComputationGraphStatsRequest.debug_options)
|
|
return _internal_debug_options();
|
|
}
|
|
inline void ComputationGraphStatsRequest::unsafe_arena_set_allocated_debug_options(
|
|
::xla::DebugOptions* debug_options) {
|
|
if (GetArenaForAllocation() == nullptr) {
|
|
delete reinterpret_cast<::PROTOBUF_NAMESPACE_ID::MessageLite*>(_impl_.debug_options_);
|
|
}
|
|
_impl_.debug_options_ = debug_options;
|
|
if (debug_options) {
|
|
|
|
} else {
|
|
|
|
}
|
|
// @@protoc_insertion_point(field_unsafe_arena_set_allocated:xla.ComputationGraphStatsRequest.debug_options)
|
|
}
|
|
inline ::xla::DebugOptions* ComputationGraphStatsRequest::release_debug_options() {
|
|
|
|
::xla::DebugOptions* temp = _impl_.debug_options_;
|
|
_impl_.debug_options_ = 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::DebugOptions* ComputationGraphStatsRequest::unsafe_arena_release_debug_options() {
|
|
// @@protoc_insertion_point(field_release:xla.ComputationGraphStatsRequest.debug_options)
|
|
|
|
::xla::DebugOptions* temp = _impl_.debug_options_;
|
|
_impl_.debug_options_ = nullptr;
|
|
return temp;
|
|
}
|
|
inline ::xla::DebugOptions* ComputationGraphStatsRequest::_internal_mutable_debug_options() {
|
|
|
|
if (_impl_.debug_options_ == nullptr) {
|
|
auto* p = CreateMaybeMessage<::xla::DebugOptions>(GetArenaForAllocation());
|
|
_impl_.debug_options_ = p;
|
|
}
|
|
return _impl_.debug_options_;
|
|
}
|
|
inline ::xla::DebugOptions* ComputationGraphStatsRequest::mutable_debug_options() {
|
|
::xla::DebugOptions* _msg = _internal_mutable_debug_options();
|
|
// @@protoc_insertion_point(field_mutable:xla.ComputationGraphStatsRequest.debug_options)
|
|
return _msg;
|
|
}
|
|
inline void ComputationGraphStatsRequest::set_allocated_debug_options(::xla::DebugOptions* debug_options) {
|
|
::PROTOBUF_NAMESPACE_ID::Arena* message_arena = GetArenaForAllocation();
|
|
if (message_arena == nullptr) {
|
|
delete _impl_.debug_options_;
|
|
}
|
|
if (debug_options) {
|
|
::PROTOBUF_NAMESPACE_ID::Arena* submessage_arena =
|
|
::PROTOBUF_NAMESPACE_ID::Arena::InternalGetOwningArena(debug_options);
|
|
if (message_arena != submessage_arena) {
|
|
debug_options = ::PROTOBUF_NAMESPACE_ID::internal::GetOwnedMessage(
|
|
message_arena, debug_options, submessage_arena);
|
|
}
|
|
|
|
} else {
|
|
|
|
}
|
|
_impl_.debug_options_ = debug_options;
|
|
// @@protoc_insertion_point(field_set_allocated:xla.ComputationGraphStatsRequest.debug_options)
|
|
}
|
|
|
|
// -------------------------------------------------------------------
|
|
|
|
// ComputationStatsResponse
|
|
|
|
// .xla.ComputationStats stats = 1;
|
|
inline bool ComputationStatsResponse::_internal_has_stats() const {
|
|
return this != internal_default_instance() && _impl_.stats_ != nullptr;
|
|
}
|
|
inline bool ComputationStatsResponse::has_stats() const {
|
|
return _internal_has_stats();
|
|
}
|
|
inline const ::xla::ComputationStats& ComputationStatsResponse::_internal_stats() const {
|
|
const ::xla::ComputationStats* p = _impl_.stats_;
|
|
return p != nullptr ? *p : reinterpret_cast<const ::xla::ComputationStats&>(
|
|
::xla::_ComputationStats_default_instance_);
|
|
}
|
|
inline const ::xla::ComputationStats& ComputationStatsResponse::stats() const {
|
|
// @@protoc_insertion_point(field_get:xla.ComputationStatsResponse.stats)
|
|
return _internal_stats();
|
|
}
|
|
inline void ComputationStatsResponse::unsafe_arena_set_allocated_stats(
|
|
::xla::ComputationStats* stats) {
|
|
if (GetArenaForAllocation() == nullptr) {
|
|
delete reinterpret_cast<::PROTOBUF_NAMESPACE_ID::MessageLite*>(_impl_.stats_);
|
|
}
|
|
_impl_.stats_ = stats;
|
|
if (stats) {
|
|
|
|
} else {
|
|
|
|
}
|
|
// @@protoc_insertion_point(field_unsafe_arena_set_allocated:xla.ComputationStatsResponse.stats)
|
|
}
|
|
inline ::xla::ComputationStats* ComputationStatsResponse::release_stats() {
|
|
|
|
::xla::ComputationStats* temp = _impl_.stats_;
|
|
_impl_.stats_ = 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::ComputationStats* ComputationStatsResponse::unsafe_arena_release_stats() {
|
|
// @@protoc_insertion_point(field_release:xla.ComputationStatsResponse.stats)
|
|
|
|
::xla::ComputationStats* temp = _impl_.stats_;
|
|
_impl_.stats_ = nullptr;
|
|
return temp;
|
|
}
|
|
inline ::xla::ComputationStats* ComputationStatsResponse::_internal_mutable_stats() {
|
|
|
|
if (_impl_.stats_ == nullptr) {
|
|
auto* p = CreateMaybeMessage<::xla::ComputationStats>(GetArenaForAllocation());
|
|
_impl_.stats_ = p;
|
|
}
|
|
return _impl_.stats_;
|
|
}
|
|
inline ::xla::ComputationStats* ComputationStatsResponse::mutable_stats() {
|
|
::xla::ComputationStats* _msg = _internal_mutable_stats();
|
|
// @@protoc_insertion_point(field_mutable:xla.ComputationStatsResponse.stats)
|
|
return _msg;
|
|
}
|
|
inline void ComputationStatsResponse::set_allocated_stats(::xla::ComputationStats* stats) {
|
|
::PROTOBUF_NAMESPACE_ID::Arena* message_arena = GetArenaForAllocation();
|
|
if (message_arena == nullptr) {
|
|
delete reinterpret_cast< ::PROTOBUF_NAMESPACE_ID::MessageLite*>(_impl_.stats_);
|
|
}
|
|
if (stats) {
|
|
::PROTOBUF_NAMESPACE_ID::Arena* submessage_arena =
|
|
::PROTOBUF_NAMESPACE_ID::Arena::InternalGetOwningArena(
|
|
reinterpret_cast<::PROTOBUF_NAMESPACE_ID::MessageLite*>(stats));
|
|
if (message_arena != submessage_arena) {
|
|
stats = ::PROTOBUF_NAMESPACE_ID::internal::GetOwnedMessage(
|
|
message_arena, stats, submessage_arena);
|
|
}
|
|
|
|
} else {
|
|
|
|
}
|
|
_impl_.stats_ = stats;
|
|
// @@protoc_insertion_point(field_set_allocated:xla.ComputationStatsResponse.stats)
|
|
}
|
|
|
|
// -------------------------------------------------------------------
|
|
|
|
// CreateChannelHandleRequest
|
|
|
|
// .xla.ChannelHandle.ChannelType channel_type = 1;
|
|
inline void CreateChannelHandleRequest::clear_channel_type() {
|
|
_impl_.channel_type_ = 0;
|
|
}
|
|
inline ::xla::ChannelHandle_ChannelType CreateChannelHandleRequest::_internal_channel_type() const {
|
|
return static_cast< ::xla::ChannelHandle_ChannelType >(_impl_.channel_type_);
|
|
}
|
|
inline ::xla::ChannelHandle_ChannelType CreateChannelHandleRequest::channel_type() const {
|
|
// @@protoc_insertion_point(field_get:xla.CreateChannelHandleRequest.channel_type)
|
|
return _internal_channel_type();
|
|
}
|
|
inline void CreateChannelHandleRequest::_internal_set_channel_type(::xla::ChannelHandle_ChannelType value) {
|
|
|
|
_impl_.channel_type_ = value;
|
|
}
|
|
inline void CreateChannelHandleRequest::set_channel_type(::xla::ChannelHandle_ChannelType value) {
|
|
_internal_set_channel_type(value);
|
|
// @@protoc_insertion_point(field_set:xla.CreateChannelHandleRequest.channel_type)
|
|
}
|
|
|
|
// -------------------------------------------------------------------
|
|
|
|
// CreateChannelHandleResponse
|
|
|
|
// .xla.ChannelHandle channel = 1;
|
|
inline bool CreateChannelHandleResponse::_internal_has_channel() const {
|
|
return this != internal_default_instance() && _impl_.channel_ != nullptr;
|
|
}
|
|
inline bool CreateChannelHandleResponse::has_channel() const {
|
|
return _internal_has_channel();
|
|
}
|
|
inline const ::xla::ChannelHandle& CreateChannelHandleResponse::_internal_channel() const {
|
|
const ::xla::ChannelHandle* p = _impl_.channel_;
|
|
return p != nullptr ? *p : reinterpret_cast<const ::xla::ChannelHandle&>(
|
|
::xla::_ChannelHandle_default_instance_);
|
|
}
|
|
inline const ::xla::ChannelHandle& CreateChannelHandleResponse::channel() const {
|
|
// @@protoc_insertion_point(field_get:xla.CreateChannelHandleResponse.channel)
|
|
return _internal_channel();
|
|
}
|
|
inline void CreateChannelHandleResponse::unsafe_arena_set_allocated_channel(
|
|
::xla::ChannelHandle* channel) {
|
|
if (GetArenaForAllocation() == nullptr) {
|
|
delete reinterpret_cast<::PROTOBUF_NAMESPACE_ID::MessageLite*>(_impl_.channel_);
|
|
}
|
|
_impl_.channel_ = channel;
|
|
if (channel) {
|
|
|
|
} else {
|
|
|
|
}
|
|
// @@protoc_insertion_point(field_unsafe_arena_set_allocated:xla.CreateChannelHandleResponse.channel)
|
|
}
|
|
inline ::xla::ChannelHandle* CreateChannelHandleResponse::release_channel() {
|
|
|
|
::xla::ChannelHandle* temp = _impl_.channel_;
|
|
_impl_.channel_ = 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::ChannelHandle* CreateChannelHandleResponse::unsafe_arena_release_channel() {
|
|
// @@protoc_insertion_point(field_release:xla.CreateChannelHandleResponse.channel)
|
|
|
|
::xla::ChannelHandle* temp = _impl_.channel_;
|
|
_impl_.channel_ = nullptr;
|
|
return temp;
|
|
}
|
|
inline ::xla::ChannelHandle* CreateChannelHandleResponse::_internal_mutable_channel() {
|
|
|
|
if (_impl_.channel_ == nullptr) {
|
|
auto* p = CreateMaybeMessage<::xla::ChannelHandle>(GetArenaForAllocation());
|
|
_impl_.channel_ = p;
|
|
}
|
|
return _impl_.channel_;
|
|
}
|
|
inline ::xla::ChannelHandle* CreateChannelHandleResponse::mutable_channel() {
|
|
::xla::ChannelHandle* _msg = _internal_mutable_channel();
|
|
// @@protoc_insertion_point(field_mutable:xla.CreateChannelHandleResponse.channel)
|
|
return _msg;
|
|
}
|
|
inline void CreateChannelHandleResponse::set_allocated_channel(::xla::ChannelHandle* channel) {
|
|
::PROTOBUF_NAMESPACE_ID::Arena* message_arena = GetArenaForAllocation();
|
|
if (message_arena == nullptr) {
|
|
delete reinterpret_cast< ::PROTOBUF_NAMESPACE_ID::MessageLite*>(_impl_.channel_);
|
|
}
|
|
if (channel) {
|
|
::PROTOBUF_NAMESPACE_ID::Arena* submessage_arena =
|
|
::PROTOBUF_NAMESPACE_ID::Arena::InternalGetOwningArena(
|
|
reinterpret_cast<::PROTOBUF_NAMESPACE_ID::MessageLite*>(channel));
|
|
if (message_arena != submessage_arena) {
|
|
channel = ::PROTOBUF_NAMESPACE_ID::internal::GetOwnedMessage(
|
|
message_arena, channel, submessage_arena);
|
|
}
|
|
|
|
} else {
|
|
|
|
}
|
|
_impl_.channel_ = channel;
|
|
// @@protoc_insertion_point(field_set_allocated:xla.CreateChannelHandleResponse.channel)
|
|
}
|
|
|
|
// -------------------------------------------------------------------
|
|
|
|
// UnregisterRequest
|
|
|
|
// repeated .xla.GlobalDataHandle data = 1;
|
|
inline int UnregisterRequest::_internal_data_size() const {
|
|
return _impl_.data_.size();
|
|
}
|
|
inline int UnregisterRequest::data_size() const {
|
|
return _internal_data_size();
|
|
}
|
|
inline ::xla::GlobalDataHandle* UnregisterRequest::mutable_data(int index) {
|
|
// @@protoc_insertion_point(field_mutable:xla.UnregisterRequest.data)
|
|
return _impl_.data_.Mutable(index);
|
|
}
|
|
inline ::PROTOBUF_NAMESPACE_ID::RepeatedPtrField< ::xla::GlobalDataHandle >*
|
|
UnregisterRequest::mutable_data() {
|
|
// @@protoc_insertion_point(field_mutable_list:xla.UnregisterRequest.data)
|
|
return &_impl_.data_;
|
|
}
|
|
inline const ::xla::GlobalDataHandle& UnregisterRequest::_internal_data(int index) const {
|
|
return _impl_.data_.Get(index);
|
|
}
|
|
inline const ::xla::GlobalDataHandle& UnregisterRequest::data(int index) const {
|
|
// @@protoc_insertion_point(field_get:xla.UnregisterRequest.data)
|
|
return _internal_data(index);
|
|
}
|
|
inline ::xla::GlobalDataHandle* UnregisterRequest::_internal_add_data() {
|
|
return _impl_.data_.Add();
|
|
}
|
|
inline ::xla::GlobalDataHandle* UnregisterRequest::add_data() {
|
|
::xla::GlobalDataHandle* _add = _internal_add_data();
|
|
// @@protoc_insertion_point(field_add:xla.UnregisterRequest.data)
|
|
return _add;
|
|
}
|
|
inline const ::PROTOBUF_NAMESPACE_ID::RepeatedPtrField< ::xla::GlobalDataHandle >&
|
|
UnregisterRequest::data() const {
|
|
// @@protoc_insertion_point(field_list:xla.UnregisterRequest.data)
|
|
return _impl_.data_;
|
|
}
|
|
|
|
// -------------------------------------------------------------------
|
|
|
|
// UnregisterResponse
|
|
|
|
// -------------------------------------------------------------------
|
|
|
|
// CompileRequest
|
|
|
|
// .xla.HloModuleProto computation = 1;
|
|
inline bool CompileRequest::_internal_has_computation() const {
|
|
return this != internal_default_instance() && _impl_.computation_ != nullptr;
|
|
}
|
|
inline bool CompileRequest::has_computation() const {
|
|
return _internal_has_computation();
|
|
}
|
|
inline const ::xla::HloModuleProto& CompileRequest::_internal_computation() const {
|
|
const ::xla::HloModuleProto* p = _impl_.computation_;
|
|
return p != nullptr ? *p : reinterpret_cast<const ::xla::HloModuleProto&>(
|
|
::xla::_HloModuleProto_default_instance_);
|
|
}
|
|
inline const ::xla::HloModuleProto& CompileRequest::computation() const {
|
|
// @@protoc_insertion_point(field_get:xla.CompileRequest.computation)
|
|
return _internal_computation();
|
|
}
|
|
inline void CompileRequest::unsafe_arena_set_allocated_computation(
|
|
::xla::HloModuleProto* computation) {
|
|
if (GetArenaForAllocation() == nullptr) {
|
|
delete reinterpret_cast<::PROTOBUF_NAMESPACE_ID::MessageLite*>(_impl_.computation_);
|
|
}
|
|
_impl_.computation_ = computation;
|
|
if (computation) {
|
|
|
|
} else {
|
|
|
|
}
|
|
// @@protoc_insertion_point(field_unsafe_arena_set_allocated:xla.CompileRequest.computation)
|
|
}
|
|
inline ::xla::HloModuleProto* CompileRequest::release_computation() {
|
|
|
|
::xla::HloModuleProto* temp = _impl_.computation_;
|
|
_impl_.computation_ = 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::HloModuleProto* CompileRequest::unsafe_arena_release_computation() {
|
|
// @@protoc_insertion_point(field_release:xla.CompileRequest.computation)
|
|
|
|
::xla::HloModuleProto* temp = _impl_.computation_;
|
|
_impl_.computation_ = nullptr;
|
|
return temp;
|
|
}
|
|
inline ::xla::HloModuleProto* CompileRequest::_internal_mutable_computation() {
|
|
|
|
if (_impl_.computation_ == nullptr) {
|
|
auto* p = CreateMaybeMessage<::xla::HloModuleProto>(GetArenaForAllocation());
|
|
_impl_.computation_ = p;
|
|
}
|
|
return _impl_.computation_;
|
|
}
|
|
inline ::xla::HloModuleProto* CompileRequest::mutable_computation() {
|
|
::xla::HloModuleProto* _msg = _internal_mutable_computation();
|
|
// @@protoc_insertion_point(field_mutable:xla.CompileRequest.computation)
|
|
return _msg;
|
|
}
|
|
inline void CompileRequest::set_allocated_computation(::xla::HloModuleProto* computation) {
|
|
::PROTOBUF_NAMESPACE_ID::Arena* message_arena = GetArenaForAllocation();
|
|
if (message_arena == nullptr) {
|
|
delete reinterpret_cast< ::PROTOBUF_NAMESPACE_ID::MessageLite*>(_impl_.computation_);
|
|
}
|
|
if (computation) {
|
|
::PROTOBUF_NAMESPACE_ID::Arena* submessage_arena =
|
|
::PROTOBUF_NAMESPACE_ID::Arena::InternalGetOwningArena(
|
|
reinterpret_cast<::PROTOBUF_NAMESPACE_ID::MessageLite*>(computation));
|
|
if (message_arena != submessage_arena) {
|
|
computation = ::PROTOBUF_NAMESPACE_ID::internal::GetOwnedMessage(
|
|
message_arena, computation, submessage_arena);
|
|
}
|
|
|
|
} else {
|
|
|
|
}
|
|
_impl_.computation_ = computation;
|
|
// @@protoc_insertion_point(field_set_allocated:xla.CompileRequest.computation)
|
|
}
|
|
|
|
// .xla.ExecutionOptions execution_options = 2;
|
|
inline bool CompileRequest::_internal_has_execution_options() const {
|
|
return this != internal_default_instance() && _impl_.execution_options_ != nullptr;
|
|
}
|
|
inline bool CompileRequest::has_execution_options() const {
|
|
return _internal_has_execution_options();
|
|
}
|
|
inline void CompileRequest::clear_execution_options() {
|
|
if (GetArenaForAllocation() == nullptr && _impl_.execution_options_ != nullptr) {
|
|
delete _impl_.execution_options_;
|
|
}
|
|
_impl_.execution_options_ = nullptr;
|
|
}
|
|
inline const ::xla::ExecutionOptions& CompileRequest::_internal_execution_options() const {
|
|
const ::xla::ExecutionOptions* p = _impl_.execution_options_;
|
|
return p != nullptr ? *p : reinterpret_cast<const ::xla::ExecutionOptions&>(
|
|
::xla::_ExecutionOptions_default_instance_);
|
|
}
|
|
inline const ::xla::ExecutionOptions& CompileRequest::execution_options() const {
|
|
// @@protoc_insertion_point(field_get:xla.CompileRequest.execution_options)
|
|
return _internal_execution_options();
|
|
}
|
|
inline void CompileRequest::unsafe_arena_set_allocated_execution_options(
|
|
::xla::ExecutionOptions* execution_options) {
|
|
if (GetArenaForAllocation() == nullptr) {
|
|
delete reinterpret_cast<::PROTOBUF_NAMESPACE_ID::MessageLite*>(_impl_.execution_options_);
|
|
}
|
|
_impl_.execution_options_ = execution_options;
|
|
if (execution_options) {
|
|
|
|
} else {
|
|
|
|
}
|
|
// @@protoc_insertion_point(field_unsafe_arena_set_allocated:xla.CompileRequest.execution_options)
|
|
}
|
|
inline ::xla::ExecutionOptions* CompileRequest::release_execution_options() {
|
|
|
|
::xla::ExecutionOptions* temp = _impl_.execution_options_;
|
|
_impl_.execution_options_ = 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::ExecutionOptions* CompileRequest::unsafe_arena_release_execution_options() {
|
|
// @@protoc_insertion_point(field_release:xla.CompileRequest.execution_options)
|
|
|
|
::xla::ExecutionOptions* temp = _impl_.execution_options_;
|
|
_impl_.execution_options_ = nullptr;
|
|
return temp;
|
|
}
|
|
inline ::xla::ExecutionOptions* CompileRequest::_internal_mutable_execution_options() {
|
|
|
|
if (_impl_.execution_options_ == nullptr) {
|
|
auto* p = CreateMaybeMessage<::xla::ExecutionOptions>(GetArenaForAllocation());
|
|
_impl_.execution_options_ = p;
|
|
}
|
|
return _impl_.execution_options_;
|
|
}
|
|
inline ::xla::ExecutionOptions* CompileRequest::mutable_execution_options() {
|
|
::xla::ExecutionOptions* _msg = _internal_mutable_execution_options();
|
|
// @@protoc_insertion_point(field_mutable:xla.CompileRequest.execution_options)
|
|
return _msg;
|
|
}
|
|
inline void CompileRequest::set_allocated_execution_options(::xla::ExecutionOptions* execution_options) {
|
|
::PROTOBUF_NAMESPACE_ID::Arena* message_arena = GetArenaForAllocation();
|
|
if (message_arena == nullptr) {
|
|
delete _impl_.execution_options_;
|
|
}
|
|
if (execution_options) {
|
|
::PROTOBUF_NAMESPACE_ID::Arena* submessage_arena =
|
|
::PROTOBUF_NAMESPACE_ID::Arena::InternalGetOwningArena(execution_options);
|
|
if (message_arena != submessage_arena) {
|
|
execution_options = ::PROTOBUF_NAMESPACE_ID::internal::GetOwnedMessage(
|
|
message_arena, execution_options, submessage_arena);
|
|
}
|
|
|
|
} else {
|
|
|
|
}
|
|
_impl_.execution_options_ = execution_options;
|
|
// @@protoc_insertion_point(field_set_allocated:xla.CompileRequest.execution_options)
|
|
}
|
|
|
|
// repeated .xla.ShapeProto input_shape_with_layout = 3;
|
|
inline int CompileRequest::_internal_input_shape_with_layout_size() const {
|
|
return _impl_.input_shape_with_layout_.size();
|
|
}
|
|
inline int CompileRequest::input_shape_with_layout_size() const {
|
|
return _internal_input_shape_with_layout_size();
|
|
}
|
|
inline ::xla::ShapeProto* CompileRequest::mutable_input_shape_with_layout(int index) {
|
|
// @@protoc_insertion_point(field_mutable:xla.CompileRequest.input_shape_with_layout)
|
|
return _impl_.input_shape_with_layout_.Mutable(index);
|
|
}
|
|
inline ::PROTOBUF_NAMESPACE_ID::RepeatedPtrField< ::xla::ShapeProto >*
|
|
CompileRequest::mutable_input_shape_with_layout() {
|
|
// @@protoc_insertion_point(field_mutable_list:xla.CompileRequest.input_shape_with_layout)
|
|
return &_impl_.input_shape_with_layout_;
|
|
}
|
|
inline const ::xla::ShapeProto& CompileRequest::_internal_input_shape_with_layout(int index) const {
|
|
return _impl_.input_shape_with_layout_.Get(index);
|
|
}
|
|
inline const ::xla::ShapeProto& CompileRequest::input_shape_with_layout(int index) const {
|
|
// @@protoc_insertion_point(field_get:xla.CompileRequest.input_shape_with_layout)
|
|
return _internal_input_shape_with_layout(index);
|
|
}
|
|
inline ::xla::ShapeProto* CompileRequest::_internal_add_input_shape_with_layout() {
|
|
return _impl_.input_shape_with_layout_.Add();
|
|
}
|
|
inline ::xla::ShapeProto* CompileRequest::add_input_shape_with_layout() {
|
|
::xla::ShapeProto* _add = _internal_add_input_shape_with_layout();
|
|
// @@protoc_insertion_point(field_add:xla.CompileRequest.input_shape_with_layout)
|
|
return _add;
|
|
}
|
|
inline const ::PROTOBUF_NAMESPACE_ID::RepeatedPtrField< ::xla::ShapeProto >&
|
|
CompileRequest::input_shape_with_layout() const {
|
|
// @@protoc_insertion_point(field_list:xla.CompileRequest.input_shape_with_layout)
|
|
return _impl_.input_shape_with_layout_;
|
|
}
|
|
|
|
// -------------------------------------------------------------------
|
|
|
|
// CompileResponse
|
|
|
|
// .xla.ExecutionHandle handle = 1;
|
|
inline bool CompileResponse::_internal_has_handle() const {
|
|
return this != internal_default_instance() && _impl_.handle_ != nullptr;
|
|
}
|
|
inline bool CompileResponse::has_handle() const {
|
|
return _internal_has_handle();
|
|
}
|
|
inline const ::xla::ExecutionHandle& CompileResponse::_internal_handle() const {
|
|
const ::xla::ExecutionHandle* p = _impl_.handle_;
|
|
return p != nullptr ? *p : reinterpret_cast<const ::xla::ExecutionHandle&>(
|
|
::xla::_ExecutionHandle_default_instance_);
|
|
}
|
|
inline const ::xla::ExecutionHandle& CompileResponse::handle() const {
|
|
// @@protoc_insertion_point(field_get:xla.CompileResponse.handle)
|
|
return _internal_handle();
|
|
}
|
|
inline void CompileResponse::unsafe_arena_set_allocated_handle(
|
|
::xla::ExecutionHandle* handle) {
|
|
if (GetArenaForAllocation() == nullptr) {
|
|
delete reinterpret_cast<::PROTOBUF_NAMESPACE_ID::MessageLite*>(_impl_.handle_);
|
|
}
|
|
_impl_.handle_ = handle;
|
|
if (handle) {
|
|
|
|
} else {
|
|
|
|
}
|
|
// @@protoc_insertion_point(field_unsafe_arena_set_allocated:xla.CompileResponse.handle)
|
|
}
|
|
inline ::xla::ExecutionHandle* CompileResponse::release_handle() {
|
|
|
|
::xla::ExecutionHandle* temp = _impl_.handle_;
|
|
_impl_.handle_ = 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::ExecutionHandle* CompileResponse::unsafe_arena_release_handle() {
|
|
// @@protoc_insertion_point(field_release:xla.CompileResponse.handle)
|
|
|
|
::xla::ExecutionHandle* temp = _impl_.handle_;
|
|
_impl_.handle_ = nullptr;
|
|
return temp;
|
|
}
|
|
inline ::xla::ExecutionHandle* CompileResponse::_internal_mutable_handle() {
|
|
|
|
if (_impl_.handle_ == nullptr) {
|
|
auto* p = CreateMaybeMessage<::xla::ExecutionHandle>(GetArenaForAllocation());
|
|
_impl_.handle_ = p;
|
|
}
|
|
return _impl_.handle_;
|
|
}
|
|
inline ::xla::ExecutionHandle* CompileResponse::mutable_handle() {
|
|
::xla::ExecutionHandle* _msg = _internal_mutable_handle();
|
|
// @@protoc_insertion_point(field_mutable:xla.CompileResponse.handle)
|
|
return _msg;
|
|
}
|
|
inline void CompileResponse::set_allocated_handle(::xla::ExecutionHandle* handle) {
|
|
::PROTOBUF_NAMESPACE_ID::Arena* message_arena = GetArenaForAllocation();
|
|
if (message_arena == nullptr) {
|
|
delete reinterpret_cast< ::PROTOBUF_NAMESPACE_ID::MessageLite*>(_impl_.handle_);
|
|
}
|
|
if (handle) {
|
|
::PROTOBUF_NAMESPACE_ID::Arena* submessage_arena =
|
|
::PROTOBUF_NAMESPACE_ID::Arena::InternalGetOwningArena(
|
|
reinterpret_cast<::PROTOBUF_NAMESPACE_ID::MessageLite*>(handle));
|
|
if (message_arena != submessage_arena) {
|
|
handle = ::PROTOBUF_NAMESPACE_ID::internal::GetOwnedMessage(
|
|
message_arena, handle, submessage_arena);
|
|
}
|
|
|
|
} else {
|
|
|
|
}
|
|
_impl_.handle_ = handle;
|
|
// @@protoc_insertion_point(field_set_allocated:xla.CompileResponse.handle)
|
|
}
|
|
|
|
// -------------------------------------------------------------------
|
|
|
|
// ExecuteRequest
|
|
|
|
// .xla.ExecutionHandle handle = 1;
|
|
inline bool ExecuteRequest::_internal_has_handle() const {
|
|
return this != internal_default_instance() && _impl_.handle_ != nullptr;
|
|
}
|
|
inline bool ExecuteRequest::has_handle() const {
|
|
return _internal_has_handle();
|
|
}
|
|
inline const ::xla::ExecutionHandle& ExecuteRequest::_internal_handle() const {
|
|
const ::xla::ExecutionHandle* p = _impl_.handle_;
|
|
return p != nullptr ? *p : reinterpret_cast<const ::xla::ExecutionHandle&>(
|
|
::xla::_ExecutionHandle_default_instance_);
|
|
}
|
|
inline const ::xla::ExecutionHandle& ExecuteRequest::handle() const {
|
|
// @@protoc_insertion_point(field_get:xla.ExecuteRequest.handle)
|
|
return _internal_handle();
|
|
}
|
|
inline void ExecuteRequest::unsafe_arena_set_allocated_handle(
|
|
::xla::ExecutionHandle* handle) {
|
|
if (GetArenaForAllocation() == nullptr) {
|
|
delete reinterpret_cast<::PROTOBUF_NAMESPACE_ID::MessageLite*>(_impl_.handle_);
|
|
}
|
|
_impl_.handle_ = handle;
|
|
if (handle) {
|
|
|
|
} else {
|
|
|
|
}
|
|
// @@protoc_insertion_point(field_unsafe_arena_set_allocated:xla.ExecuteRequest.handle)
|
|
}
|
|
inline ::xla::ExecutionHandle* ExecuteRequest::release_handle() {
|
|
|
|
::xla::ExecutionHandle* temp = _impl_.handle_;
|
|
_impl_.handle_ = 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::ExecutionHandle* ExecuteRequest::unsafe_arena_release_handle() {
|
|
// @@protoc_insertion_point(field_release:xla.ExecuteRequest.handle)
|
|
|
|
::xla::ExecutionHandle* temp = _impl_.handle_;
|
|
_impl_.handle_ = nullptr;
|
|
return temp;
|
|
}
|
|
inline ::xla::ExecutionHandle* ExecuteRequest::_internal_mutable_handle() {
|
|
|
|
if (_impl_.handle_ == nullptr) {
|
|
auto* p = CreateMaybeMessage<::xla::ExecutionHandle>(GetArenaForAllocation());
|
|
_impl_.handle_ = p;
|
|
}
|
|
return _impl_.handle_;
|
|
}
|
|
inline ::xla::ExecutionHandle* ExecuteRequest::mutable_handle() {
|
|
::xla::ExecutionHandle* _msg = _internal_mutable_handle();
|
|
// @@protoc_insertion_point(field_mutable:xla.ExecuteRequest.handle)
|
|
return _msg;
|
|
}
|
|
inline void ExecuteRequest::set_allocated_handle(::xla::ExecutionHandle* handle) {
|
|
::PROTOBUF_NAMESPACE_ID::Arena* message_arena = GetArenaForAllocation();
|
|
if (message_arena == nullptr) {
|
|
delete reinterpret_cast< ::PROTOBUF_NAMESPACE_ID::MessageLite*>(_impl_.handle_);
|
|
}
|
|
if (handle) {
|
|
::PROTOBUF_NAMESPACE_ID::Arena* submessage_arena =
|
|
::PROTOBUF_NAMESPACE_ID::Arena::InternalGetOwningArena(
|
|
reinterpret_cast<::PROTOBUF_NAMESPACE_ID::MessageLite*>(handle));
|
|
if (message_arena != submessage_arena) {
|
|
handle = ::PROTOBUF_NAMESPACE_ID::internal::GetOwnedMessage(
|
|
message_arena, handle, submessage_arena);
|
|
}
|
|
|
|
} else {
|
|
|
|
}
|
|
_impl_.handle_ = handle;
|
|
// @@protoc_insertion_point(field_set_allocated:xla.ExecuteRequest.handle)
|
|
}
|
|
|
|
// repeated .xla.GlobalDataHandle arguments = 2;
|
|
inline int ExecuteRequest::_internal_arguments_size() const {
|
|
return _impl_.arguments_.size();
|
|
}
|
|
inline int ExecuteRequest::arguments_size() const {
|
|
return _internal_arguments_size();
|
|
}
|
|
inline ::xla::GlobalDataHandle* ExecuteRequest::mutable_arguments(int index) {
|
|
// @@protoc_insertion_point(field_mutable:xla.ExecuteRequest.arguments)
|
|
return _impl_.arguments_.Mutable(index);
|
|
}
|
|
inline ::PROTOBUF_NAMESPACE_ID::RepeatedPtrField< ::xla::GlobalDataHandle >*
|
|
ExecuteRequest::mutable_arguments() {
|
|
// @@protoc_insertion_point(field_mutable_list:xla.ExecuteRequest.arguments)
|
|
return &_impl_.arguments_;
|
|
}
|
|
inline const ::xla::GlobalDataHandle& ExecuteRequest::_internal_arguments(int index) const {
|
|
return _impl_.arguments_.Get(index);
|
|
}
|
|
inline const ::xla::GlobalDataHandle& ExecuteRequest::arguments(int index) const {
|
|
// @@protoc_insertion_point(field_get:xla.ExecuteRequest.arguments)
|
|
return _internal_arguments(index);
|
|
}
|
|
inline ::xla::GlobalDataHandle* ExecuteRequest::_internal_add_arguments() {
|
|
return _impl_.arguments_.Add();
|
|
}
|
|
inline ::xla::GlobalDataHandle* ExecuteRequest::add_arguments() {
|
|
::xla::GlobalDataHandle* _add = _internal_add_arguments();
|
|
// @@protoc_insertion_point(field_add:xla.ExecuteRequest.arguments)
|
|
return _add;
|
|
}
|
|
inline const ::PROTOBUF_NAMESPACE_ID::RepeatedPtrField< ::xla::GlobalDataHandle >&
|
|
ExecuteRequest::arguments() const {
|
|
// @@protoc_insertion_point(field_list:xla.ExecuteRequest.arguments)
|
|
return _impl_.arguments_;
|
|
}
|
|
|
|
// -------------------------------------------------------------------
|
|
|
|
// ExecuteGraphRequest
|
|
|
|
// .xla.HloModuleProto computation = 1;
|
|
inline bool ExecuteGraphRequest::_internal_has_computation() const {
|
|
return this != internal_default_instance() && _impl_.computation_ != nullptr;
|
|
}
|
|
inline bool ExecuteGraphRequest::has_computation() const {
|
|
return _internal_has_computation();
|
|
}
|
|
inline const ::xla::HloModuleProto& ExecuteGraphRequest::_internal_computation() const {
|
|
const ::xla::HloModuleProto* p = _impl_.computation_;
|
|
return p != nullptr ? *p : reinterpret_cast<const ::xla::HloModuleProto&>(
|
|
::xla::_HloModuleProto_default_instance_);
|
|
}
|
|
inline const ::xla::HloModuleProto& ExecuteGraphRequest::computation() const {
|
|
// @@protoc_insertion_point(field_get:xla.ExecuteGraphRequest.computation)
|
|
return _internal_computation();
|
|
}
|
|
inline void ExecuteGraphRequest::unsafe_arena_set_allocated_computation(
|
|
::xla::HloModuleProto* computation) {
|
|
if (GetArenaForAllocation() == nullptr) {
|
|
delete reinterpret_cast<::PROTOBUF_NAMESPACE_ID::MessageLite*>(_impl_.computation_);
|
|
}
|
|
_impl_.computation_ = computation;
|
|
if (computation) {
|
|
|
|
} else {
|
|
|
|
}
|
|
// @@protoc_insertion_point(field_unsafe_arena_set_allocated:xla.ExecuteGraphRequest.computation)
|
|
}
|
|
inline ::xla::HloModuleProto* ExecuteGraphRequest::release_computation() {
|
|
|
|
::xla::HloModuleProto* temp = _impl_.computation_;
|
|
_impl_.computation_ = 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::HloModuleProto* ExecuteGraphRequest::unsafe_arena_release_computation() {
|
|
// @@protoc_insertion_point(field_release:xla.ExecuteGraphRequest.computation)
|
|
|
|
::xla::HloModuleProto* temp = _impl_.computation_;
|
|
_impl_.computation_ = nullptr;
|
|
return temp;
|
|
}
|
|
inline ::xla::HloModuleProto* ExecuteGraphRequest::_internal_mutable_computation() {
|
|
|
|
if (_impl_.computation_ == nullptr) {
|
|
auto* p = CreateMaybeMessage<::xla::HloModuleProto>(GetArenaForAllocation());
|
|
_impl_.computation_ = p;
|
|
}
|
|
return _impl_.computation_;
|
|
}
|
|
inline ::xla::HloModuleProto* ExecuteGraphRequest::mutable_computation() {
|
|
::xla::HloModuleProto* _msg = _internal_mutable_computation();
|
|
// @@protoc_insertion_point(field_mutable:xla.ExecuteGraphRequest.computation)
|
|
return _msg;
|
|
}
|
|
inline void ExecuteGraphRequest::set_allocated_computation(::xla::HloModuleProto* computation) {
|
|
::PROTOBUF_NAMESPACE_ID::Arena* message_arena = GetArenaForAllocation();
|
|
if (message_arena == nullptr) {
|
|
delete reinterpret_cast< ::PROTOBUF_NAMESPACE_ID::MessageLite*>(_impl_.computation_);
|
|
}
|
|
if (computation) {
|
|
::PROTOBUF_NAMESPACE_ID::Arena* submessage_arena =
|
|
::PROTOBUF_NAMESPACE_ID::Arena::InternalGetOwningArena(
|
|
reinterpret_cast<::PROTOBUF_NAMESPACE_ID::MessageLite*>(computation));
|
|
if (message_arena != submessage_arena) {
|
|
computation = ::PROTOBUF_NAMESPACE_ID::internal::GetOwnedMessage(
|
|
message_arena, computation, submessage_arena);
|
|
}
|
|
|
|
} else {
|
|
|
|
}
|
|
_impl_.computation_ = computation;
|
|
// @@protoc_insertion_point(field_set_allocated:xla.ExecuteGraphRequest.computation)
|
|
}
|
|
|
|
// repeated .xla.GlobalDataHandle arguments = 2;
|
|
inline int ExecuteGraphRequest::_internal_arguments_size() const {
|
|
return _impl_.arguments_.size();
|
|
}
|
|
inline int ExecuteGraphRequest::arguments_size() const {
|
|
return _internal_arguments_size();
|
|
}
|
|
inline ::xla::GlobalDataHandle* ExecuteGraphRequest::mutable_arguments(int index) {
|
|
// @@protoc_insertion_point(field_mutable:xla.ExecuteGraphRequest.arguments)
|
|
return _impl_.arguments_.Mutable(index);
|
|
}
|
|
inline ::PROTOBUF_NAMESPACE_ID::RepeatedPtrField< ::xla::GlobalDataHandle >*
|
|
ExecuteGraphRequest::mutable_arguments() {
|
|
// @@protoc_insertion_point(field_mutable_list:xla.ExecuteGraphRequest.arguments)
|
|
return &_impl_.arguments_;
|
|
}
|
|
inline const ::xla::GlobalDataHandle& ExecuteGraphRequest::_internal_arguments(int index) const {
|
|
return _impl_.arguments_.Get(index);
|
|
}
|
|
inline const ::xla::GlobalDataHandle& ExecuteGraphRequest::arguments(int index) const {
|
|
// @@protoc_insertion_point(field_get:xla.ExecuteGraphRequest.arguments)
|
|
return _internal_arguments(index);
|
|
}
|
|
inline ::xla::GlobalDataHandle* ExecuteGraphRequest::_internal_add_arguments() {
|
|
return _impl_.arguments_.Add();
|
|
}
|
|
inline ::xla::GlobalDataHandle* ExecuteGraphRequest::add_arguments() {
|
|
::xla::GlobalDataHandle* _add = _internal_add_arguments();
|
|
// @@protoc_insertion_point(field_add:xla.ExecuteGraphRequest.arguments)
|
|
return _add;
|
|
}
|
|
inline const ::PROTOBUF_NAMESPACE_ID::RepeatedPtrField< ::xla::GlobalDataHandle >&
|
|
ExecuteGraphRequest::arguments() const {
|
|
// @@protoc_insertion_point(field_list:xla.ExecuteGraphRequest.arguments)
|
|
return _impl_.arguments_;
|
|
}
|
|
|
|
// .xla.ExecutionOptions execution_options = 3;
|
|
inline bool ExecuteGraphRequest::_internal_has_execution_options() const {
|
|
return this != internal_default_instance() && _impl_.execution_options_ != nullptr;
|
|
}
|
|
inline bool ExecuteGraphRequest::has_execution_options() const {
|
|
return _internal_has_execution_options();
|
|
}
|
|
inline void ExecuteGraphRequest::clear_execution_options() {
|
|
if (GetArenaForAllocation() == nullptr && _impl_.execution_options_ != nullptr) {
|
|
delete _impl_.execution_options_;
|
|
}
|
|
_impl_.execution_options_ = nullptr;
|
|
}
|
|
inline const ::xla::ExecutionOptions& ExecuteGraphRequest::_internal_execution_options() const {
|
|
const ::xla::ExecutionOptions* p = _impl_.execution_options_;
|
|
return p != nullptr ? *p : reinterpret_cast<const ::xla::ExecutionOptions&>(
|
|
::xla::_ExecutionOptions_default_instance_);
|
|
}
|
|
inline const ::xla::ExecutionOptions& ExecuteGraphRequest::execution_options() const {
|
|
// @@protoc_insertion_point(field_get:xla.ExecuteGraphRequest.execution_options)
|
|
return _internal_execution_options();
|
|
}
|
|
inline void ExecuteGraphRequest::unsafe_arena_set_allocated_execution_options(
|
|
::xla::ExecutionOptions* execution_options) {
|
|
if (GetArenaForAllocation() == nullptr) {
|
|
delete reinterpret_cast<::PROTOBUF_NAMESPACE_ID::MessageLite*>(_impl_.execution_options_);
|
|
}
|
|
_impl_.execution_options_ = execution_options;
|
|
if (execution_options) {
|
|
|
|
} else {
|
|
|
|
}
|
|
// @@protoc_insertion_point(field_unsafe_arena_set_allocated:xla.ExecuteGraphRequest.execution_options)
|
|
}
|
|
inline ::xla::ExecutionOptions* ExecuteGraphRequest::release_execution_options() {
|
|
|
|
::xla::ExecutionOptions* temp = _impl_.execution_options_;
|
|
_impl_.execution_options_ = 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::ExecutionOptions* ExecuteGraphRequest::unsafe_arena_release_execution_options() {
|
|
// @@protoc_insertion_point(field_release:xla.ExecuteGraphRequest.execution_options)
|
|
|
|
::xla::ExecutionOptions* temp = _impl_.execution_options_;
|
|
_impl_.execution_options_ = nullptr;
|
|
return temp;
|
|
}
|
|
inline ::xla::ExecutionOptions* ExecuteGraphRequest::_internal_mutable_execution_options() {
|
|
|
|
if (_impl_.execution_options_ == nullptr) {
|
|
auto* p = CreateMaybeMessage<::xla::ExecutionOptions>(GetArenaForAllocation());
|
|
_impl_.execution_options_ = p;
|
|
}
|
|
return _impl_.execution_options_;
|
|
}
|
|
inline ::xla::ExecutionOptions* ExecuteGraphRequest::mutable_execution_options() {
|
|
::xla::ExecutionOptions* _msg = _internal_mutable_execution_options();
|
|
// @@protoc_insertion_point(field_mutable:xla.ExecuteGraphRequest.execution_options)
|
|
return _msg;
|
|
}
|
|
inline void ExecuteGraphRequest::set_allocated_execution_options(::xla::ExecutionOptions* execution_options) {
|
|
::PROTOBUF_NAMESPACE_ID::Arena* message_arena = GetArenaForAllocation();
|
|
if (message_arena == nullptr) {
|
|
delete _impl_.execution_options_;
|
|
}
|
|
if (execution_options) {
|
|
::PROTOBUF_NAMESPACE_ID::Arena* submessage_arena =
|
|
::PROTOBUF_NAMESPACE_ID::Arena::InternalGetOwningArena(execution_options);
|
|
if (message_arena != submessage_arena) {
|
|
execution_options = ::PROTOBUF_NAMESPACE_ID::internal::GetOwnedMessage(
|
|
message_arena, execution_options, submessage_arena);
|
|
}
|
|
|
|
} else {
|
|
|
|
}
|
|
_impl_.execution_options_ = execution_options;
|
|
// @@protoc_insertion_point(field_set_allocated:xla.ExecuteGraphRequest.execution_options)
|
|
}
|
|
|
|
// -------------------------------------------------------------------
|
|
|
|
// ExecuteGraphParallelRequest
|
|
|
|
// repeated .xla.ExecuteGraphRequest requests = 1;
|
|
inline int ExecuteGraphParallelRequest::_internal_requests_size() const {
|
|
return _impl_.requests_.size();
|
|
}
|
|
inline int ExecuteGraphParallelRequest::requests_size() const {
|
|
return _internal_requests_size();
|
|
}
|
|
inline void ExecuteGraphParallelRequest::clear_requests() {
|
|
_impl_.requests_.Clear();
|
|
}
|
|
inline ::xla::ExecuteGraphRequest* ExecuteGraphParallelRequest::mutable_requests(int index) {
|
|
// @@protoc_insertion_point(field_mutable:xla.ExecuteGraphParallelRequest.requests)
|
|
return _impl_.requests_.Mutable(index);
|
|
}
|
|
inline ::PROTOBUF_NAMESPACE_ID::RepeatedPtrField< ::xla::ExecuteGraphRequest >*
|
|
ExecuteGraphParallelRequest::mutable_requests() {
|
|
// @@protoc_insertion_point(field_mutable_list:xla.ExecuteGraphParallelRequest.requests)
|
|
return &_impl_.requests_;
|
|
}
|
|
inline const ::xla::ExecuteGraphRequest& ExecuteGraphParallelRequest::_internal_requests(int index) const {
|
|
return _impl_.requests_.Get(index);
|
|
}
|
|
inline const ::xla::ExecuteGraphRequest& ExecuteGraphParallelRequest::requests(int index) const {
|
|
// @@protoc_insertion_point(field_get:xla.ExecuteGraphParallelRequest.requests)
|
|
return _internal_requests(index);
|
|
}
|
|
inline ::xla::ExecuteGraphRequest* ExecuteGraphParallelRequest::_internal_add_requests() {
|
|
return _impl_.requests_.Add();
|
|
}
|
|
inline ::xla::ExecuteGraphRequest* ExecuteGraphParallelRequest::add_requests() {
|
|
::xla::ExecuteGraphRequest* _add = _internal_add_requests();
|
|
// @@protoc_insertion_point(field_add:xla.ExecuteGraphParallelRequest.requests)
|
|
return _add;
|
|
}
|
|
inline const ::PROTOBUF_NAMESPACE_ID::RepeatedPtrField< ::xla::ExecuteGraphRequest >&
|
|
ExecuteGraphParallelRequest::requests() const {
|
|
// @@protoc_insertion_point(field_list:xla.ExecuteGraphParallelRequest.requests)
|
|
return _impl_.requests_;
|
|
}
|
|
|
|
// -------------------------------------------------------------------
|
|
|
|
// ExecuteResponse
|
|
|
|
// .xla.GlobalDataHandle output = 1;
|
|
inline bool ExecuteResponse::_internal_has_output() const {
|
|
return this != internal_default_instance() && _impl_.output_ != nullptr;
|
|
}
|
|
inline bool ExecuteResponse::has_output() const {
|
|
return _internal_has_output();
|
|
}
|
|
inline const ::xla::GlobalDataHandle& ExecuteResponse::_internal_output() const {
|
|
const ::xla::GlobalDataHandle* p = _impl_.output_;
|
|
return p != nullptr ? *p : reinterpret_cast<const ::xla::GlobalDataHandle&>(
|
|
::xla::_GlobalDataHandle_default_instance_);
|
|
}
|
|
inline const ::xla::GlobalDataHandle& ExecuteResponse::output() const {
|
|
// @@protoc_insertion_point(field_get:xla.ExecuteResponse.output)
|
|
return _internal_output();
|
|
}
|
|
inline void ExecuteResponse::unsafe_arena_set_allocated_output(
|
|
::xla::GlobalDataHandle* output) {
|
|
if (GetArenaForAllocation() == nullptr) {
|
|
delete reinterpret_cast<::PROTOBUF_NAMESPACE_ID::MessageLite*>(_impl_.output_);
|
|
}
|
|
_impl_.output_ = output;
|
|
if (output) {
|
|
|
|
} else {
|
|
|
|
}
|
|
// @@protoc_insertion_point(field_unsafe_arena_set_allocated:xla.ExecuteResponse.output)
|
|
}
|
|
inline ::xla::GlobalDataHandle* ExecuteResponse::release_output() {
|
|
|
|
::xla::GlobalDataHandle* temp = _impl_.output_;
|
|
_impl_.output_ = 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::GlobalDataHandle* ExecuteResponse::unsafe_arena_release_output() {
|
|
// @@protoc_insertion_point(field_release:xla.ExecuteResponse.output)
|
|
|
|
::xla::GlobalDataHandle* temp = _impl_.output_;
|
|
_impl_.output_ = nullptr;
|
|
return temp;
|
|
}
|
|
inline ::xla::GlobalDataHandle* ExecuteResponse::_internal_mutable_output() {
|
|
|
|
if (_impl_.output_ == nullptr) {
|
|
auto* p = CreateMaybeMessage<::xla::GlobalDataHandle>(GetArenaForAllocation());
|
|
_impl_.output_ = p;
|
|
}
|
|
return _impl_.output_;
|
|
}
|
|
inline ::xla::GlobalDataHandle* ExecuteResponse::mutable_output() {
|
|
::xla::GlobalDataHandle* _msg = _internal_mutable_output();
|
|
// @@protoc_insertion_point(field_mutable:xla.ExecuteResponse.output)
|
|
return _msg;
|
|
}
|
|
inline void ExecuteResponse::set_allocated_output(::xla::GlobalDataHandle* output) {
|
|
::PROTOBUF_NAMESPACE_ID::Arena* message_arena = GetArenaForAllocation();
|
|
if (message_arena == nullptr) {
|
|
delete reinterpret_cast< ::PROTOBUF_NAMESPACE_ID::MessageLite*>(_impl_.output_);
|
|
}
|
|
if (output) {
|
|
::PROTOBUF_NAMESPACE_ID::Arena* submessage_arena =
|
|
::PROTOBUF_NAMESPACE_ID::Arena::InternalGetOwningArena(
|
|
reinterpret_cast<::PROTOBUF_NAMESPACE_ID::MessageLite*>(output));
|
|
if (message_arena != submessage_arena) {
|
|
output = ::PROTOBUF_NAMESPACE_ID::internal::GetOwnedMessage(
|
|
message_arena, output, submessage_arena);
|
|
}
|
|
|
|
} else {
|
|
|
|
}
|
|
_impl_.output_ = output;
|
|
// @@protoc_insertion_point(field_set_allocated:xla.ExecuteResponse.output)
|
|
}
|
|
|
|
// .xla.ExecutionProfile profile = 2;
|
|
inline bool ExecuteResponse::_internal_has_profile() const {
|
|
return this != internal_default_instance() && _impl_.profile_ != nullptr;
|
|
}
|
|
inline bool ExecuteResponse::has_profile() const {
|
|
return _internal_has_profile();
|
|
}
|
|
inline const ::xla::ExecutionProfile& ExecuteResponse::_internal_profile() const {
|
|
const ::xla::ExecutionProfile* p = _impl_.profile_;
|
|
return p != nullptr ? *p : reinterpret_cast<const ::xla::ExecutionProfile&>(
|
|
::xla::_ExecutionProfile_default_instance_);
|
|
}
|
|
inline const ::xla::ExecutionProfile& ExecuteResponse::profile() const {
|
|
// @@protoc_insertion_point(field_get:xla.ExecuteResponse.profile)
|
|
return _internal_profile();
|
|
}
|
|
inline void ExecuteResponse::unsafe_arena_set_allocated_profile(
|
|
::xla::ExecutionProfile* profile) {
|
|
if (GetArenaForAllocation() == nullptr) {
|
|
delete reinterpret_cast<::PROTOBUF_NAMESPACE_ID::MessageLite*>(_impl_.profile_);
|
|
}
|
|
_impl_.profile_ = profile;
|
|
if (profile) {
|
|
|
|
} else {
|
|
|
|
}
|
|
// @@protoc_insertion_point(field_unsafe_arena_set_allocated:xla.ExecuteResponse.profile)
|
|
}
|
|
inline ::xla::ExecutionProfile* ExecuteResponse::release_profile() {
|
|
|
|
::xla::ExecutionProfile* temp = _impl_.profile_;
|
|
_impl_.profile_ = 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::ExecutionProfile* ExecuteResponse::unsafe_arena_release_profile() {
|
|
// @@protoc_insertion_point(field_release:xla.ExecuteResponse.profile)
|
|
|
|
::xla::ExecutionProfile* temp = _impl_.profile_;
|
|
_impl_.profile_ = nullptr;
|
|
return temp;
|
|
}
|
|
inline ::xla::ExecutionProfile* ExecuteResponse::_internal_mutable_profile() {
|
|
|
|
if (_impl_.profile_ == nullptr) {
|
|
auto* p = CreateMaybeMessage<::xla::ExecutionProfile>(GetArenaForAllocation());
|
|
_impl_.profile_ = p;
|
|
}
|
|
return _impl_.profile_;
|
|
}
|
|
inline ::xla::ExecutionProfile* ExecuteResponse::mutable_profile() {
|
|
::xla::ExecutionProfile* _msg = _internal_mutable_profile();
|
|
// @@protoc_insertion_point(field_mutable:xla.ExecuteResponse.profile)
|
|
return _msg;
|
|
}
|
|
inline void ExecuteResponse::set_allocated_profile(::xla::ExecutionProfile* profile) {
|
|
::PROTOBUF_NAMESPACE_ID::Arena* message_arena = GetArenaForAllocation();
|
|
if (message_arena == nullptr) {
|
|
delete reinterpret_cast< ::PROTOBUF_NAMESPACE_ID::MessageLite*>(_impl_.profile_);
|
|
}
|
|
if (profile) {
|
|
::PROTOBUF_NAMESPACE_ID::Arena* submessage_arena =
|
|
::PROTOBUF_NAMESPACE_ID::Arena::InternalGetOwningArena(
|
|
reinterpret_cast<::PROTOBUF_NAMESPACE_ID::MessageLite*>(profile));
|
|
if (message_arena != submessage_arena) {
|
|
profile = ::PROTOBUF_NAMESPACE_ID::internal::GetOwnedMessage(
|
|
message_arena, profile, submessage_arena);
|
|
}
|
|
|
|
} else {
|
|
|
|
}
|
|
_impl_.profile_ = profile;
|
|
// @@protoc_insertion_point(field_set_allocated:xla.ExecuteResponse.profile)
|
|
}
|
|
|
|
// -------------------------------------------------------------------
|
|
|
|
// ExecuteParallelResponse
|
|
|
|
// repeated .xla.ExecuteResponse responses = 1;
|
|
inline int ExecuteParallelResponse::_internal_responses_size() const {
|
|
return _impl_.responses_.size();
|
|
}
|
|
inline int ExecuteParallelResponse::responses_size() const {
|
|
return _internal_responses_size();
|
|
}
|
|
inline void ExecuteParallelResponse::clear_responses() {
|
|
_impl_.responses_.Clear();
|
|
}
|
|
inline ::xla::ExecuteResponse* ExecuteParallelResponse::mutable_responses(int index) {
|
|
// @@protoc_insertion_point(field_mutable:xla.ExecuteParallelResponse.responses)
|
|
return _impl_.responses_.Mutable(index);
|
|
}
|
|
inline ::PROTOBUF_NAMESPACE_ID::RepeatedPtrField< ::xla::ExecuteResponse >*
|
|
ExecuteParallelResponse::mutable_responses() {
|
|
// @@protoc_insertion_point(field_mutable_list:xla.ExecuteParallelResponse.responses)
|
|
return &_impl_.responses_;
|
|
}
|
|
inline const ::xla::ExecuteResponse& ExecuteParallelResponse::_internal_responses(int index) const {
|
|
return _impl_.responses_.Get(index);
|
|
}
|
|
inline const ::xla::ExecuteResponse& ExecuteParallelResponse::responses(int index) const {
|
|
// @@protoc_insertion_point(field_get:xla.ExecuteParallelResponse.responses)
|
|
return _internal_responses(index);
|
|
}
|
|
inline ::xla::ExecuteResponse* ExecuteParallelResponse::_internal_add_responses() {
|
|
return _impl_.responses_.Add();
|
|
}
|
|
inline ::xla::ExecuteResponse* ExecuteParallelResponse::add_responses() {
|
|
::xla::ExecuteResponse* _add = _internal_add_responses();
|
|
// @@protoc_insertion_point(field_add:xla.ExecuteParallelResponse.responses)
|
|
return _add;
|
|
}
|
|
inline const ::PROTOBUF_NAMESPACE_ID::RepeatedPtrField< ::xla::ExecuteResponse >&
|
|
ExecuteParallelResponse::responses() const {
|
|
// @@protoc_insertion_point(field_list:xla.ExecuteParallelResponse.responses)
|
|
return _impl_.responses_;
|
|
}
|
|
|
|
// -------------------------------------------------------------------
|
|
|
|
// ComputeConstantGraphRequest
|
|
|
|
// .xla.HloModuleProto computation = 1;
|
|
inline bool ComputeConstantGraphRequest::_internal_has_computation() const {
|
|
return this != internal_default_instance() && _impl_.computation_ != nullptr;
|
|
}
|
|
inline bool ComputeConstantGraphRequest::has_computation() const {
|
|
return _internal_has_computation();
|
|
}
|
|
inline const ::xla::HloModuleProto& ComputeConstantGraphRequest::_internal_computation() const {
|
|
const ::xla::HloModuleProto* p = _impl_.computation_;
|
|
return p != nullptr ? *p : reinterpret_cast<const ::xla::HloModuleProto&>(
|
|
::xla::_HloModuleProto_default_instance_);
|
|
}
|
|
inline const ::xla::HloModuleProto& ComputeConstantGraphRequest::computation() const {
|
|
// @@protoc_insertion_point(field_get:xla.ComputeConstantGraphRequest.computation)
|
|
return _internal_computation();
|
|
}
|
|
inline void ComputeConstantGraphRequest::unsafe_arena_set_allocated_computation(
|
|
::xla::HloModuleProto* computation) {
|
|
if (GetArenaForAllocation() == nullptr) {
|
|
delete reinterpret_cast<::PROTOBUF_NAMESPACE_ID::MessageLite*>(_impl_.computation_);
|
|
}
|
|
_impl_.computation_ = computation;
|
|
if (computation) {
|
|
|
|
} else {
|
|
|
|
}
|
|
// @@protoc_insertion_point(field_unsafe_arena_set_allocated:xla.ComputeConstantGraphRequest.computation)
|
|
}
|
|
inline ::xla::HloModuleProto* ComputeConstantGraphRequest::release_computation() {
|
|
|
|
::xla::HloModuleProto* temp = _impl_.computation_;
|
|
_impl_.computation_ = 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::HloModuleProto* ComputeConstantGraphRequest::unsafe_arena_release_computation() {
|
|
// @@protoc_insertion_point(field_release:xla.ComputeConstantGraphRequest.computation)
|
|
|
|
::xla::HloModuleProto* temp = _impl_.computation_;
|
|
_impl_.computation_ = nullptr;
|
|
return temp;
|
|
}
|
|
inline ::xla::HloModuleProto* ComputeConstantGraphRequest::_internal_mutable_computation() {
|
|
|
|
if (_impl_.computation_ == nullptr) {
|
|
auto* p = CreateMaybeMessage<::xla::HloModuleProto>(GetArenaForAllocation());
|
|
_impl_.computation_ = p;
|
|
}
|
|
return _impl_.computation_;
|
|
}
|
|
inline ::xla::HloModuleProto* ComputeConstantGraphRequest::mutable_computation() {
|
|
::xla::HloModuleProto* _msg = _internal_mutable_computation();
|
|
// @@protoc_insertion_point(field_mutable:xla.ComputeConstantGraphRequest.computation)
|
|
return _msg;
|
|
}
|
|
inline void ComputeConstantGraphRequest::set_allocated_computation(::xla::HloModuleProto* computation) {
|
|
::PROTOBUF_NAMESPACE_ID::Arena* message_arena = GetArenaForAllocation();
|
|
if (message_arena == nullptr) {
|
|
delete reinterpret_cast< ::PROTOBUF_NAMESPACE_ID::MessageLite*>(_impl_.computation_);
|
|
}
|
|
if (computation) {
|
|
::PROTOBUF_NAMESPACE_ID::Arena* submessage_arena =
|
|
::PROTOBUF_NAMESPACE_ID::Arena::InternalGetOwningArena(
|
|
reinterpret_cast<::PROTOBUF_NAMESPACE_ID::MessageLite*>(computation));
|
|
if (message_arena != submessage_arena) {
|
|
computation = ::PROTOBUF_NAMESPACE_ID::internal::GetOwnedMessage(
|
|
message_arena, computation, submessage_arena);
|
|
}
|
|
|
|
} else {
|
|
|
|
}
|
|
_impl_.computation_ = computation;
|
|
// @@protoc_insertion_point(field_set_allocated:xla.ComputeConstantGraphRequest.computation)
|
|
}
|
|
|
|
// .xla.LayoutProto output_layout = 2;
|
|
inline bool ComputeConstantGraphRequest::_internal_has_output_layout() const {
|
|
return this != internal_default_instance() && _impl_.output_layout_ != nullptr;
|
|
}
|
|
inline bool ComputeConstantGraphRequest::has_output_layout() const {
|
|
return _internal_has_output_layout();
|
|
}
|
|
inline const ::xla::LayoutProto& ComputeConstantGraphRequest::_internal_output_layout() const {
|
|
const ::xla::LayoutProto* p = _impl_.output_layout_;
|
|
return p != nullptr ? *p : reinterpret_cast<const ::xla::LayoutProto&>(
|
|
::xla::_LayoutProto_default_instance_);
|
|
}
|
|
inline const ::xla::LayoutProto& ComputeConstantGraphRequest::output_layout() const {
|
|
// @@protoc_insertion_point(field_get:xla.ComputeConstantGraphRequest.output_layout)
|
|
return _internal_output_layout();
|
|
}
|
|
inline void ComputeConstantGraphRequest::unsafe_arena_set_allocated_output_layout(
|
|
::xla::LayoutProto* output_layout) {
|
|
if (GetArenaForAllocation() == nullptr) {
|
|
delete reinterpret_cast<::PROTOBUF_NAMESPACE_ID::MessageLite*>(_impl_.output_layout_);
|
|
}
|
|
_impl_.output_layout_ = output_layout;
|
|
if (output_layout) {
|
|
|
|
} else {
|
|
|
|
}
|
|
// @@protoc_insertion_point(field_unsafe_arena_set_allocated:xla.ComputeConstantGraphRequest.output_layout)
|
|
}
|
|
inline ::xla::LayoutProto* ComputeConstantGraphRequest::release_output_layout() {
|
|
|
|
::xla::LayoutProto* temp = _impl_.output_layout_;
|
|
_impl_.output_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* ComputeConstantGraphRequest::unsafe_arena_release_output_layout() {
|
|
// @@protoc_insertion_point(field_release:xla.ComputeConstantGraphRequest.output_layout)
|
|
|
|
::xla::LayoutProto* temp = _impl_.output_layout_;
|
|
_impl_.output_layout_ = nullptr;
|
|
return temp;
|
|
}
|
|
inline ::xla::LayoutProto* ComputeConstantGraphRequest::_internal_mutable_output_layout() {
|
|
|
|
if (_impl_.output_layout_ == nullptr) {
|
|
auto* p = CreateMaybeMessage<::xla::LayoutProto>(GetArenaForAllocation());
|
|
_impl_.output_layout_ = p;
|
|
}
|
|
return _impl_.output_layout_;
|
|
}
|
|
inline ::xla::LayoutProto* ComputeConstantGraphRequest::mutable_output_layout() {
|
|
::xla::LayoutProto* _msg = _internal_mutable_output_layout();
|
|
// @@protoc_insertion_point(field_mutable:xla.ComputeConstantGraphRequest.output_layout)
|
|
return _msg;
|
|
}
|
|
inline void ComputeConstantGraphRequest::set_allocated_output_layout(::xla::LayoutProto* output_layout) {
|
|
::PROTOBUF_NAMESPACE_ID::Arena* message_arena = GetArenaForAllocation();
|
|
if (message_arena == nullptr) {
|
|
delete reinterpret_cast< ::PROTOBUF_NAMESPACE_ID::MessageLite*>(_impl_.output_layout_);
|
|
}
|
|
if (output_layout) {
|
|
::PROTOBUF_NAMESPACE_ID::Arena* submessage_arena =
|
|
::PROTOBUF_NAMESPACE_ID::Arena::InternalGetOwningArena(
|
|
reinterpret_cast<::PROTOBUF_NAMESPACE_ID::MessageLite*>(output_layout));
|
|
if (message_arena != submessage_arena) {
|
|
output_layout = ::PROTOBUF_NAMESPACE_ID::internal::GetOwnedMessage(
|
|
message_arena, output_layout, submessage_arena);
|
|
}
|
|
|
|
} else {
|
|
|
|
}
|
|
_impl_.output_layout_ = output_layout;
|
|
// @@protoc_insertion_point(field_set_allocated:xla.ComputeConstantGraphRequest.output_layout)
|
|
}
|
|
|
|
// -------------------------------------------------------------------
|
|
|
|
// ComputeConstantResponse
|
|
|
|
// .xla.LiteralProto literal = 1;
|
|
inline bool ComputeConstantResponse::_internal_has_literal() const {
|
|
return this != internal_default_instance() && _impl_.literal_ != nullptr;
|
|
}
|
|
inline bool ComputeConstantResponse::has_literal() const {
|
|
return _internal_has_literal();
|
|
}
|
|
inline const ::xla::LiteralProto& ComputeConstantResponse::_internal_literal() const {
|
|
const ::xla::LiteralProto* p = _impl_.literal_;
|
|
return p != nullptr ? *p : reinterpret_cast<const ::xla::LiteralProto&>(
|
|
::xla::_LiteralProto_default_instance_);
|
|
}
|
|
inline const ::xla::LiteralProto& ComputeConstantResponse::literal() const {
|
|
// @@protoc_insertion_point(field_get:xla.ComputeConstantResponse.literal)
|
|
return _internal_literal();
|
|
}
|
|
inline void ComputeConstantResponse::unsafe_arena_set_allocated_literal(
|
|
::xla::LiteralProto* literal) {
|
|
if (GetArenaForAllocation() == nullptr) {
|
|
delete reinterpret_cast<::PROTOBUF_NAMESPACE_ID::MessageLite*>(_impl_.literal_);
|
|
}
|
|
_impl_.literal_ = literal;
|
|
if (literal) {
|
|
|
|
} else {
|
|
|
|
}
|
|
// @@protoc_insertion_point(field_unsafe_arena_set_allocated:xla.ComputeConstantResponse.literal)
|
|
}
|
|
inline ::xla::LiteralProto* ComputeConstantResponse::release_literal() {
|
|
|
|
::xla::LiteralProto* temp = _impl_.literal_;
|
|
_impl_.literal_ = 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::LiteralProto* ComputeConstantResponse::unsafe_arena_release_literal() {
|
|
// @@protoc_insertion_point(field_release:xla.ComputeConstantResponse.literal)
|
|
|
|
::xla::LiteralProto* temp = _impl_.literal_;
|
|
_impl_.literal_ = nullptr;
|
|
return temp;
|
|
}
|
|
inline ::xla::LiteralProto* ComputeConstantResponse::_internal_mutable_literal() {
|
|
|
|
if (_impl_.literal_ == nullptr) {
|
|
auto* p = CreateMaybeMessage<::xla::LiteralProto>(GetArenaForAllocation());
|
|
_impl_.literal_ = p;
|
|
}
|
|
return _impl_.literal_;
|
|
}
|
|
inline ::xla::LiteralProto* ComputeConstantResponse::mutable_literal() {
|
|
::xla::LiteralProto* _msg = _internal_mutable_literal();
|
|
// @@protoc_insertion_point(field_mutable:xla.ComputeConstantResponse.literal)
|
|
return _msg;
|
|
}
|
|
inline void ComputeConstantResponse::set_allocated_literal(::xla::LiteralProto* literal) {
|
|
::PROTOBUF_NAMESPACE_ID::Arena* message_arena = GetArenaForAllocation();
|
|
if (message_arena == nullptr) {
|
|
delete reinterpret_cast< ::PROTOBUF_NAMESPACE_ID::MessageLite*>(_impl_.literal_);
|
|
}
|
|
if (literal) {
|
|
::PROTOBUF_NAMESPACE_ID::Arena* submessage_arena =
|
|
::PROTOBUF_NAMESPACE_ID::Arena::InternalGetOwningArena(
|
|
reinterpret_cast<::PROTOBUF_NAMESPACE_ID::MessageLite*>(literal));
|
|
if (message_arena != submessage_arena) {
|
|
literal = ::PROTOBUF_NAMESPACE_ID::internal::GetOwnedMessage(
|
|
message_arena, literal, submessage_arena);
|
|
}
|
|
|
|
} else {
|
|
|
|
}
|
|
_impl_.literal_ = literal;
|
|
// @@protoc_insertion_point(field_set_allocated:xla.ComputeConstantResponse.literal)
|
|
}
|
|
|
|
// -------------------------------------------------------------------
|
|
|
|
// DeconstructTupleRequest
|
|
|
|
// .xla.GlobalDataHandle tuple_handle = 2;
|
|
inline bool DeconstructTupleRequest::_internal_has_tuple_handle() const {
|
|
return this != internal_default_instance() && _impl_.tuple_handle_ != nullptr;
|
|
}
|
|
inline bool DeconstructTupleRequest::has_tuple_handle() const {
|
|
return _internal_has_tuple_handle();
|
|
}
|
|
inline const ::xla::GlobalDataHandle& DeconstructTupleRequest::_internal_tuple_handle() const {
|
|
const ::xla::GlobalDataHandle* p = _impl_.tuple_handle_;
|
|
return p != nullptr ? *p : reinterpret_cast<const ::xla::GlobalDataHandle&>(
|
|
::xla::_GlobalDataHandle_default_instance_);
|
|
}
|
|
inline const ::xla::GlobalDataHandle& DeconstructTupleRequest::tuple_handle() const {
|
|
// @@protoc_insertion_point(field_get:xla.DeconstructTupleRequest.tuple_handle)
|
|
return _internal_tuple_handle();
|
|
}
|
|
inline void DeconstructTupleRequest::unsafe_arena_set_allocated_tuple_handle(
|
|
::xla::GlobalDataHandle* tuple_handle) {
|
|
if (GetArenaForAllocation() == nullptr) {
|
|
delete reinterpret_cast<::PROTOBUF_NAMESPACE_ID::MessageLite*>(_impl_.tuple_handle_);
|
|
}
|
|
_impl_.tuple_handle_ = tuple_handle;
|
|
if (tuple_handle) {
|
|
|
|
} else {
|
|
|
|
}
|
|
// @@protoc_insertion_point(field_unsafe_arena_set_allocated:xla.DeconstructTupleRequest.tuple_handle)
|
|
}
|
|
inline ::xla::GlobalDataHandle* DeconstructTupleRequest::release_tuple_handle() {
|
|
|
|
::xla::GlobalDataHandle* temp = _impl_.tuple_handle_;
|
|
_impl_.tuple_handle_ = 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::GlobalDataHandle* DeconstructTupleRequest::unsafe_arena_release_tuple_handle() {
|
|
// @@protoc_insertion_point(field_release:xla.DeconstructTupleRequest.tuple_handle)
|
|
|
|
::xla::GlobalDataHandle* temp = _impl_.tuple_handle_;
|
|
_impl_.tuple_handle_ = nullptr;
|
|
return temp;
|
|
}
|
|
inline ::xla::GlobalDataHandle* DeconstructTupleRequest::_internal_mutable_tuple_handle() {
|
|
|
|
if (_impl_.tuple_handle_ == nullptr) {
|
|
auto* p = CreateMaybeMessage<::xla::GlobalDataHandle>(GetArenaForAllocation());
|
|
_impl_.tuple_handle_ = p;
|
|
}
|
|
return _impl_.tuple_handle_;
|
|
}
|
|
inline ::xla::GlobalDataHandle* DeconstructTupleRequest::mutable_tuple_handle() {
|
|
::xla::GlobalDataHandle* _msg = _internal_mutable_tuple_handle();
|
|
// @@protoc_insertion_point(field_mutable:xla.DeconstructTupleRequest.tuple_handle)
|
|
return _msg;
|
|
}
|
|
inline void DeconstructTupleRequest::set_allocated_tuple_handle(::xla::GlobalDataHandle* tuple_handle) {
|
|
::PROTOBUF_NAMESPACE_ID::Arena* message_arena = GetArenaForAllocation();
|
|
if (message_arena == nullptr) {
|
|
delete reinterpret_cast< ::PROTOBUF_NAMESPACE_ID::MessageLite*>(_impl_.tuple_handle_);
|
|
}
|
|
if (tuple_handle) {
|
|
::PROTOBUF_NAMESPACE_ID::Arena* submessage_arena =
|
|
::PROTOBUF_NAMESPACE_ID::Arena::InternalGetOwningArena(
|
|
reinterpret_cast<::PROTOBUF_NAMESPACE_ID::MessageLite*>(tuple_handle));
|
|
if (message_arena != submessage_arena) {
|
|
tuple_handle = ::PROTOBUF_NAMESPACE_ID::internal::GetOwnedMessage(
|
|
message_arena, tuple_handle, submessage_arena);
|
|
}
|
|
|
|
} else {
|
|
|
|
}
|
|
_impl_.tuple_handle_ = tuple_handle;
|
|
// @@protoc_insertion_point(field_set_allocated:xla.DeconstructTupleRequest.tuple_handle)
|
|
}
|
|
|
|
// -------------------------------------------------------------------
|
|
|
|
// DeconstructTupleResponse
|
|
|
|
// repeated .xla.GlobalDataHandle element_handles = 1;
|
|
inline int DeconstructTupleResponse::_internal_element_handles_size() const {
|
|
return _impl_.element_handles_.size();
|
|
}
|
|
inline int DeconstructTupleResponse::element_handles_size() const {
|
|
return _internal_element_handles_size();
|
|
}
|
|
inline ::xla::GlobalDataHandle* DeconstructTupleResponse::mutable_element_handles(int index) {
|
|
// @@protoc_insertion_point(field_mutable:xla.DeconstructTupleResponse.element_handles)
|
|
return _impl_.element_handles_.Mutable(index);
|
|
}
|
|
inline ::PROTOBUF_NAMESPACE_ID::RepeatedPtrField< ::xla::GlobalDataHandle >*
|
|
DeconstructTupleResponse::mutable_element_handles() {
|
|
// @@protoc_insertion_point(field_mutable_list:xla.DeconstructTupleResponse.element_handles)
|
|
return &_impl_.element_handles_;
|
|
}
|
|
inline const ::xla::GlobalDataHandle& DeconstructTupleResponse::_internal_element_handles(int index) const {
|
|
return _impl_.element_handles_.Get(index);
|
|
}
|
|
inline const ::xla::GlobalDataHandle& DeconstructTupleResponse::element_handles(int index) const {
|
|
// @@protoc_insertion_point(field_get:xla.DeconstructTupleResponse.element_handles)
|
|
return _internal_element_handles(index);
|
|
}
|
|
inline ::xla::GlobalDataHandle* DeconstructTupleResponse::_internal_add_element_handles() {
|
|
return _impl_.element_handles_.Add();
|
|
}
|
|
inline ::xla::GlobalDataHandle* DeconstructTupleResponse::add_element_handles() {
|
|
::xla::GlobalDataHandle* _add = _internal_add_element_handles();
|
|
// @@protoc_insertion_point(field_add:xla.DeconstructTupleResponse.element_handles)
|
|
return _add;
|
|
}
|
|
inline const ::PROTOBUF_NAMESPACE_ID::RepeatedPtrField< ::xla::GlobalDataHandle >&
|
|
DeconstructTupleResponse::element_handles() const {
|
|
// @@protoc_insertion_point(field_list:xla.DeconstructTupleResponse.element_handles)
|
|
return _impl_.element_handles_;
|
|
}
|
|
|
|
// -------------------------------------------------------------------
|
|
|
|
// LoadDataRequest
|
|
|
|
// string columnio_tablet_path = 1;
|
|
inline void LoadDataRequest::clear_columnio_tablet_path() {
|
|
_impl_.columnio_tablet_path_.ClearToEmpty();
|
|
}
|
|
inline const std::string& LoadDataRequest::columnio_tablet_path() const {
|
|
// @@protoc_insertion_point(field_get:xla.LoadDataRequest.columnio_tablet_path)
|
|
return _internal_columnio_tablet_path();
|
|
}
|
|
template <typename ArgT0, typename... ArgT>
|
|
inline PROTOBUF_ALWAYS_INLINE
|
|
void LoadDataRequest::set_columnio_tablet_path(ArgT0&& arg0, ArgT... args) {
|
|
|
|
_impl_.columnio_tablet_path_.Set(static_cast<ArgT0 &&>(arg0), args..., GetArenaForAllocation());
|
|
// @@protoc_insertion_point(field_set:xla.LoadDataRequest.columnio_tablet_path)
|
|
}
|
|
inline std::string* LoadDataRequest::mutable_columnio_tablet_path() {
|
|
std::string* _s = _internal_mutable_columnio_tablet_path();
|
|
// @@protoc_insertion_point(field_mutable:xla.LoadDataRequest.columnio_tablet_path)
|
|
return _s;
|
|
}
|
|
inline const std::string& LoadDataRequest::_internal_columnio_tablet_path() const {
|
|
return _impl_.columnio_tablet_path_.Get();
|
|
}
|
|
inline void LoadDataRequest::_internal_set_columnio_tablet_path(const std::string& value) {
|
|
|
|
_impl_.columnio_tablet_path_.Set(value, GetArenaForAllocation());
|
|
}
|
|
inline std::string* LoadDataRequest::_internal_mutable_columnio_tablet_path() {
|
|
|
|
return _impl_.columnio_tablet_path_.Mutable(GetArenaForAllocation());
|
|
}
|
|
inline std::string* LoadDataRequest::release_columnio_tablet_path() {
|
|
// @@protoc_insertion_point(field_release:xla.LoadDataRequest.columnio_tablet_path)
|
|
return _impl_.columnio_tablet_path_.Release();
|
|
}
|
|
inline void LoadDataRequest::set_allocated_columnio_tablet_path(std::string* columnio_tablet_path) {
|
|
if (columnio_tablet_path != nullptr) {
|
|
|
|
} else {
|
|
|
|
}
|
|
_impl_.columnio_tablet_path_.SetAllocated(columnio_tablet_path, GetArenaForAllocation());
|
|
#ifdef PROTOBUF_FORCE_COPY_DEFAULT_STRING
|
|
if (_impl_.columnio_tablet_path_.IsDefault()) {
|
|
_impl_.columnio_tablet_path_.Set("", GetArenaForAllocation());
|
|
}
|
|
#endif // PROTOBUF_FORCE_COPY_DEFAULT_STRING
|
|
// @@protoc_insertion_point(field_set_allocated:xla.LoadDataRequest.columnio_tablet_path)
|
|
}
|
|
|
|
// string columnio_field = 2;
|
|
inline void LoadDataRequest::clear_columnio_field() {
|
|
_impl_.columnio_field_.ClearToEmpty();
|
|
}
|
|
inline const std::string& LoadDataRequest::columnio_field() const {
|
|
// @@protoc_insertion_point(field_get:xla.LoadDataRequest.columnio_field)
|
|
return _internal_columnio_field();
|
|
}
|
|
template <typename ArgT0, typename... ArgT>
|
|
inline PROTOBUF_ALWAYS_INLINE
|
|
void LoadDataRequest::set_columnio_field(ArgT0&& arg0, ArgT... args) {
|
|
|
|
_impl_.columnio_field_.Set(static_cast<ArgT0 &&>(arg0), args..., GetArenaForAllocation());
|
|
// @@protoc_insertion_point(field_set:xla.LoadDataRequest.columnio_field)
|
|
}
|
|
inline std::string* LoadDataRequest::mutable_columnio_field() {
|
|
std::string* _s = _internal_mutable_columnio_field();
|
|
// @@protoc_insertion_point(field_mutable:xla.LoadDataRequest.columnio_field)
|
|
return _s;
|
|
}
|
|
inline const std::string& LoadDataRequest::_internal_columnio_field() const {
|
|
return _impl_.columnio_field_.Get();
|
|
}
|
|
inline void LoadDataRequest::_internal_set_columnio_field(const std::string& value) {
|
|
|
|
_impl_.columnio_field_.Set(value, GetArenaForAllocation());
|
|
}
|
|
inline std::string* LoadDataRequest::_internal_mutable_columnio_field() {
|
|
|
|
return _impl_.columnio_field_.Mutable(GetArenaForAllocation());
|
|
}
|
|
inline std::string* LoadDataRequest::release_columnio_field() {
|
|
// @@protoc_insertion_point(field_release:xla.LoadDataRequest.columnio_field)
|
|
return _impl_.columnio_field_.Release();
|
|
}
|
|
inline void LoadDataRequest::set_allocated_columnio_field(std::string* columnio_field) {
|
|
if (columnio_field != nullptr) {
|
|
|
|
} else {
|
|
|
|
}
|
|
_impl_.columnio_field_.SetAllocated(columnio_field, GetArenaForAllocation());
|
|
#ifdef PROTOBUF_FORCE_COPY_DEFAULT_STRING
|
|
if (_impl_.columnio_field_.IsDefault()) {
|
|
_impl_.columnio_field_.Set("", GetArenaForAllocation());
|
|
}
|
|
#endif // PROTOBUF_FORCE_COPY_DEFAULT_STRING
|
|
// @@protoc_insertion_point(field_set_allocated:xla.LoadDataRequest.columnio_field)
|
|
}
|
|
|
|
// .xla.ShapeProto element_shape = 3;
|
|
inline bool LoadDataRequest::_internal_has_element_shape() const {
|
|
return this != internal_default_instance() && _impl_.element_shape_ != nullptr;
|
|
}
|
|
inline bool LoadDataRequest::has_element_shape() const {
|
|
return _internal_has_element_shape();
|
|
}
|
|
inline const ::xla::ShapeProto& LoadDataRequest::_internal_element_shape() const {
|
|
const ::xla::ShapeProto* p = _impl_.element_shape_;
|
|
return p != nullptr ? *p : reinterpret_cast<const ::xla::ShapeProto&>(
|
|
::xla::_ShapeProto_default_instance_);
|
|
}
|
|
inline const ::xla::ShapeProto& LoadDataRequest::element_shape() const {
|
|
// @@protoc_insertion_point(field_get:xla.LoadDataRequest.element_shape)
|
|
return _internal_element_shape();
|
|
}
|
|
inline void LoadDataRequest::unsafe_arena_set_allocated_element_shape(
|
|
::xla::ShapeProto* element_shape) {
|
|
if (GetArenaForAllocation() == nullptr) {
|
|
delete reinterpret_cast<::PROTOBUF_NAMESPACE_ID::MessageLite*>(_impl_.element_shape_);
|
|
}
|
|
_impl_.element_shape_ = element_shape;
|
|
if (element_shape) {
|
|
|
|
} else {
|
|
|
|
}
|
|
// @@protoc_insertion_point(field_unsafe_arena_set_allocated:xla.LoadDataRequest.element_shape)
|
|
}
|
|
inline ::xla::ShapeProto* LoadDataRequest::release_element_shape() {
|
|
|
|
::xla::ShapeProto* temp = _impl_.element_shape_;
|
|
_impl_.element_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* LoadDataRequest::unsafe_arena_release_element_shape() {
|
|
// @@protoc_insertion_point(field_release:xla.LoadDataRequest.element_shape)
|
|
|
|
::xla::ShapeProto* temp = _impl_.element_shape_;
|
|
_impl_.element_shape_ = nullptr;
|
|
return temp;
|
|
}
|
|
inline ::xla::ShapeProto* LoadDataRequest::_internal_mutable_element_shape() {
|
|
|
|
if (_impl_.element_shape_ == nullptr) {
|
|
auto* p = CreateMaybeMessage<::xla::ShapeProto>(GetArenaForAllocation());
|
|
_impl_.element_shape_ = p;
|
|
}
|
|
return _impl_.element_shape_;
|
|
}
|
|
inline ::xla::ShapeProto* LoadDataRequest::mutable_element_shape() {
|
|
::xla::ShapeProto* _msg = _internal_mutable_element_shape();
|
|
// @@protoc_insertion_point(field_mutable:xla.LoadDataRequest.element_shape)
|
|
return _msg;
|
|
}
|
|
inline void LoadDataRequest::set_allocated_element_shape(::xla::ShapeProto* element_shape) {
|
|
::PROTOBUF_NAMESPACE_ID::Arena* message_arena = GetArenaForAllocation();
|
|
if (message_arena == nullptr) {
|
|
delete reinterpret_cast< ::PROTOBUF_NAMESPACE_ID::MessageLite*>(_impl_.element_shape_);
|
|
}
|
|
if (element_shape) {
|
|
::PROTOBUF_NAMESPACE_ID::Arena* submessage_arena =
|
|
::PROTOBUF_NAMESPACE_ID::Arena::InternalGetOwningArena(
|
|
reinterpret_cast<::PROTOBUF_NAMESPACE_ID::MessageLite*>(element_shape));
|
|
if (message_arena != submessage_arena) {
|
|
element_shape = ::PROTOBUF_NAMESPACE_ID::internal::GetOwnedMessage(
|
|
message_arena, element_shape, submessage_arena);
|
|
}
|
|
|
|
} else {
|
|
|
|
}
|
|
_impl_.element_shape_ = element_shape;
|
|
// @@protoc_insertion_point(field_set_allocated:xla.LoadDataRequest.element_shape)
|
|
}
|
|
|
|
// int64 offset = 4;
|
|
inline void LoadDataRequest::clear_offset() {
|
|
_impl_.offset_ = int64_t{0};
|
|
}
|
|
inline int64_t LoadDataRequest::_internal_offset() const {
|
|
return _impl_.offset_;
|
|
}
|
|
inline int64_t LoadDataRequest::offset() const {
|
|
// @@protoc_insertion_point(field_get:xla.LoadDataRequest.offset)
|
|
return _internal_offset();
|
|
}
|
|
inline void LoadDataRequest::_internal_set_offset(int64_t value) {
|
|
|
|
_impl_.offset_ = value;
|
|
}
|
|
inline void LoadDataRequest::set_offset(int64_t value) {
|
|
_internal_set_offset(value);
|
|
// @@protoc_insertion_point(field_set:xla.LoadDataRequest.offset)
|
|
}
|
|
|
|
// int64 limit = 5;
|
|
inline void LoadDataRequest::clear_limit() {
|
|
_impl_.limit_ = int64_t{0};
|
|
}
|
|
inline int64_t LoadDataRequest::_internal_limit() const {
|
|
return _impl_.limit_;
|
|
}
|
|
inline int64_t LoadDataRequest::limit() const {
|
|
// @@protoc_insertion_point(field_get:xla.LoadDataRequest.limit)
|
|
return _internal_limit();
|
|
}
|
|
inline void LoadDataRequest::_internal_set_limit(int64_t value) {
|
|
|
|
_impl_.limit_ = value;
|
|
}
|
|
inline void LoadDataRequest::set_limit(int64_t value) {
|
|
_internal_set_limit(value);
|
|
// @@protoc_insertion_point(field_set:xla.LoadDataRequest.limit)
|
|
}
|
|
|
|
// bool zip = 6;
|
|
inline void LoadDataRequest::clear_zip() {
|
|
_impl_.zip_ = false;
|
|
}
|
|
inline bool LoadDataRequest::_internal_zip() const {
|
|
return _impl_.zip_;
|
|
}
|
|
inline bool LoadDataRequest::zip() const {
|
|
// @@protoc_insertion_point(field_get:xla.LoadDataRequest.zip)
|
|
return _internal_zip();
|
|
}
|
|
inline void LoadDataRequest::_internal_set_zip(bool value) {
|
|
|
|
_impl_.zip_ = value;
|
|
}
|
|
inline void LoadDataRequest::set_zip(bool value) {
|
|
_internal_set_zip(value);
|
|
// @@protoc_insertion_point(field_set:xla.LoadDataRequest.zip)
|
|
}
|
|
|
|
// -------------------------------------------------------------------
|
|
|
|
// LoadDataResponse
|
|
|
|
// .xla.GlobalDataHandle data = 1;
|
|
inline bool LoadDataResponse::_internal_has_data() const {
|
|
return this != internal_default_instance() && _impl_.data_ != nullptr;
|
|
}
|
|
inline bool LoadDataResponse::has_data() const {
|
|
return _internal_has_data();
|
|
}
|
|
inline const ::xla::GlobalDataHandle& LoadDataResponse::_internal_data() const {
|
|
const ::xla::GlobalDataHandle* p = _impl_.data_;
|
|
return p != nullptr ? *p : reinterpret_cast<const ::xla::GlobalDataHandle&>(
|
|
::xla::_GlobalDataHandle_default_instance_);
|
|
}
|
|
inline const ::xla::GlobalDataHandle& LoadDataResponse::data() const {
|
|
// @@protoc_insertion_point(field_get:xla.LoadDataResponse.data)
|
|
return _internal_data();
|
|
}
|
|
inline void LoadDataResponse::unsafe_arena_set_allocated_data(
|
|
::xla::GlobalDataHandle* data) {
|
|
if (GetArenaForAllocation() == nullptr) {
|
|
delete reinterpret_cast<::PROTOBUF_NAMESPACE_ID::MessageLite*>(_impl_.data_);
|
|
}
|
|
_impl_.data_ = data;
|
|
if (data) {
|
|
|
|
} else {
|
|
|
|
}
|
|
// @@protoc_insertion_point(field_unsafe_arena_set_allocated:xla.LoadDataResponse.data)
|
|
}
|
|
inline ::xla::GlobalDataHandle* LoadDataResponse::release_data() {
|
|
|
|
::xla::GlobalDataHandle* temp = _impl_.data_;
|
|
_impl_.data_ = 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::GlobalDataHandle* LoadDataResponse::unsafe_arena_release_data() {
|
|
// @@protoc_insertion_point(field_release:xla.LoadDataResponse.data)
|
|
|
|
::xla::GlobalDataHandle* temp = _impl_.data_;
|
|
_impl_.data_ = nullptr;
|
|
return temp;
|
|
}
|
|
inline ::xla::GlobalDataHandle* LoadDataResponse::_internal_mutable_data() {
|
|
|
|
if (_impl_.data_ == nullptr) {
|
|
auto* p = CreateMaybeMessage<::xla::GlobalDataHandle>(GetArenaForAllocation());
|
|
_impl_.data_ = p;
|
|
}
|
|
return _impl_.data_;
|
|
}
|
|
inline ::xla::GlobalDataHandle* LoadDataResponse::mutable_data() {
|
|
::xla::GlobalDataHandle* _msg = _internal_mutable_data();
|
|
// @@protoc_insertion_point(field_mutable:xla.LoadDataResponse.data)
|
|
return _msg;
|
|
}
|
|
inline void LoadDataResponse::set_allocated_data(::xla::GlobalDataHandle* data) {
|
|
::PROTOBUF_NAMESPACE_ID::Arena* message_arena = GetArenaForAllocation();
|
|
if (message_arena == nullptr) {
|
|
delete reinterpret_cast< ::PROTOBUF_NAMESPACE_ID::MessageLite*>(_impl_.data_);
|
|
}
|
|
if (data) {
|
|
::PROTOBUF_NAMESPACE_ID::Arena* submessage_arena =
|
|
::PROTOBUF_NAMESPACE_ID::Arena::InternalGetOwningArena(
|
|
reinterpret_cast<::PROTOBUF_NAMESPACE_ID::MessageLite*>(data));
|
|
if (message_arena != submessage_arena) {
|
|
data = ::PROTOBUF_NAMESPACE_ID::internal::GetOwnedMessage(
|
|
message_arena, data, submessage_arena);
|
|
}
|
|
|
|
} else {
|
|
|
|
}
|
|
_impl_.data_ = data;
|
|
// @@protoc_insertion_point(field_set_allocated:xla.LoadDataResponse.data)
|
|
}
|
|
|
|
// .xla.ShapeProto data_shape = 2;
|
|
inline bool LoadDataResponse::_internal_has_data_shape() const {
|
|
return this != internal_default_instance() && _impl_.data_shape_ != nullptr;
|
|
}
|
|
inline bool LoadDataResponse::has_data_shape() const {
|
|
return _internal_has_data_shape();
|
|
}
|
|
inline const ::xla::ShapeProto& LoadDataResponse::_internal_data_shape() const {
|
|
const ::xla::ShapeProto* p = _impl_.data_shape_;
|
|
return p != nullptr ? *p : reinterpret_cast<const ::xla::ShapeProto&>(
|
|
::xla::_ShapeProto_default_instance_);
|
|
}
|
|
inline const ::xla::ShapeProto& LoadDataResponse::data_shape() const {
|
|
// @@protoc_insertion_point(field_get:xla.LoadDataResponse.data_shape)
|
|
return _internal_data_shape();
|
|
}
|
|
inline void LoadDataResponse::unsafe_arena_set_allocated_data_shape(
|
|
::xla::ShapeProto* data_shape) {
|
|
if (GetArenaForAllocation() == nullptr) {
|
|
delete reinterpret_cast<::PROTOBUF_NAMESPACE_ID::MessageLite*>(_impl_.data_shape_);
|
|
}
|
|
_impl_.data_shape_ = data_shape;
|
|
if (data_shape) {
|
|
|
|
} else {
|
|
|
|
}
|
|
// @@protoc_insertion_point(field_unsafe_arena_set_allocated:xla.LoadDataResponse.data_shape)
|
|
}
|
|
inline ::xla::ShapeProto* LoadDataResponse::release_data_shape() {
|
|
|
|
::xla::ShapeProto* temp = _impl_.data_shape_;
|
|
_impl_.data_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* LoadDataResponse::unsafe_arena_release_data_shape() {
|
|
// @@protoc_insertion_point(field_release:xla.LoadDataResponse.data_shape)
|
|
|
|
::xla::ShapeProto* temp = _impl_.data_shape_;
|
|
_impl_.data_shape_ = nullptr;
|
|
return temp;
|
|
}
|
|
inline ::xla::ShapeProto* LoadDataResponse::_internal_mutable_data_shape() {
|
|
|
|
if (_impl_.data_shape_ == nullptr) {
|
|
auto* p = CreateMaybeMessage<::xla::ShapeProto>(GetArenaForAllocation());
|
|
_impl_.data_shape_ = p;
|
|
}
|
|
return _impl_.data_shape_;
|
|
}
|
|
inline ::xla::ShapeProto* LoadDataResponse::mutable_data_shape() {
|
|
::xla::ShapeProto* _msg = _internal_mutable_data_shape();
|
|
// @@protoc_insertion_point(field_mutable:xla.LoadDataResponse.data_shape)
|
|
return _msg;
|
|
}
|
|
inline void LoadDataResponse::set_allocated_data_shape(::xla::ShapeProto* data_shape) {
|
|
::PROTOBUF_NAMESPACE_ID::Arena* message_arena = GetArenaForAllocation();
|
|
if (message_arena == nullptr) {
|
|
delete reinterpret_cast< ::PROTOBUF_NAMESPACE_ID::MessageLite*>(_impl_.data_shape_);
|
|
}
|
|
if (data_shape) {
|
|
::PROTOBUF_NAMESPACE_ID::Arena* submessage_arena =
|
|
::PROTOBUF_NAMESPACE_ID::Arena::InternalGetOwningArena(
|
|
reinterpret_cast<::PROTOBUF_NAMESPACE_ID::MessageLite*>(data_shape));
|
|
if (message_arena != submessage_arena) {
|
|
data_shape = ::PROTOBUF_NAMESPACE_ID::internal::GetOwnedMessage(
|
|
message_arena, data_shape, submessage_arena);
|
|
}
|
|
|
|
} else {
|
|
|
|
}
|
|
_impl_.data_shape_ = data_shape;
|
|
// @@protoc_insertion_point(field_set_allocated:xla.LoadDataResponse.data_shape)
|
|
}
|
|
|
|
// int64 available_rows = 3;
|
|
inline void LoadDataResponse::clear_available_rows() {
|
|
_impl_.available_rows_ = int64_t{0};
|
|
}
|
|
inline int64_t LoadDataResponse::_internal_available_rows() const {
|
|
return _impl_.available_rows_;
|
|
}
|
|
inline int64_t LoadDataResponse::available_rows() const {
|
|
// @@protoc_insertion_point(field_get:xla.LoadDataResponse.available_rows)
|
|
return _internal_available_rows();
|
|
}
|
|
inline void LoadDataResponse::_internal_set_available_rows(int64_t value) {
|
|
|
|
_impl_.available_rows_ = value;
|
|
}
|
|
inline void LoadDataResponse::set_available_rows(int64_t value) {
|
|
_internal_set_available_rows(value);
|
|
// @@protoc_insertion_point(field_set:xla.LoadDataResponse.available_rows)
|
|
}
|
|
|
|
// int64 rows_loaded = 4;
|
|
inline void LoadDataResponse::clear_rows_loaded() {
|
|
_impl_.rows_loaded_ = int64_t{0};
|
|
}
|
|
inline int64_t LoadDataResponse::_internal_rows_loaded() const {
|
|
return _impl_.rows_loaded_;
|
|
}
|
|
inline int64_t LoadDataResponse::rows_loaded() const {
|
|
// @@protoc_insertion_point(field_get:xla.LoadDataResponse.rows_loaded)
|
|
return _internal_rows_loaded();
|
|
}
|
|
inline void LoadDataResponse::_internal_set_rows_loaded(int64_t value) {
|
|
|
|
_impl_.rows_loaded_ = value;
|
|
}
|
|
inline void LoadDataResponse::set_rows_loaded(int64_t value) {
|
|
_internal_set_rows_loaded(value);
|
|
// @@protoc_insertion_point(field_set:xla.LoadDataResponse.rows_loaded)
|
|
}
|
|
|
|
// int64 nanoseconds = 5;
|
|
inline void LoadDataResponse::clear_nanoseconds() {
|
|
_impl_.nanoseconds_ = int64_t{0};
|
|
}
|
|
inline int64_t LoadDataResponse::_internal_nanoseconds() const {
|
|
return _impl_.nanoseconds_;
|
|
}
|
|
inline int64_t LoadDataResponse::nanoseconds() const {
|
|
// @@protoc_insertion_point(field_get:xla.LoadDataResponse.nanoseconds)
|
|
return _internal_nanoseconds();
|
|
}
|
|
inline void LoadDataResponse::_internal_set_nanoseconds(int64_t value) {
|
|
|
|
_impl_.nanoseconds_ = value;
|
|
}
|
|
inline void LoadDataResponse::set_nanoseconds(int64_t value) {
|
|
_internal_set_nanoseconds(value);
|
|
// @@protoc_insertion_point(field_set:xla.LoadDataResponse.nanoseconds)
|
|
}
|
|
|
|
// -------------------------------------------------------------------
|
|
|
|
// GetShapeRequest
|
|
|
|
// .xla.GlobalDataHandle data = 1;
|
|
inline bool GetShapeRequest::_internal_has_data() const {
|
|
return this != internal_default_instance() && _impl_.data_ != nullptr;
|
|
}
|
|
inline bool GetShapeRequest::has_data() const {
|
|
return _internal_has_data();
|
|
}
|
|
inline const ::xla::GlobalDataHandle& GetShapeRequest::_internal_data() const {
|
|
const ::xla::GlobalDataHandle* p = _impl_.data_;
|
|
return p != nullptr ? *p : reinterpret_cast<const ::xla::GlobalDataHandle&>(
|
|
::xla::_GlobalDataHandle_default_instance_);
|
|
}
|
|
inline const ::xla::GlobalDataHandle& GetShapeRequest::data() const {
|
|
// @@protoc_insertion_point(field_get:xla.GetShapeRequest.data)
|
|
return _internal_data();
|
|
}
|
|
inline void GetShapeRequest::unsafe_arena_set_allocated_data(
|
|
::xla::GlobalDataHandle* data) {
|
|
if (GetArenaForAllocation() == nullptr) {
|
|
delete reinterpret_cast<::PROTOBUF_NAMESPACE_ID::MessageLite*>(_impl_.data_);
|
|
}
|
|
_impl_.data_ = data;
|
|
if (data) {
|
|
|
|
} else {
|
|
|
|
}
|
|
// @@protoc_insertion_point(field_unsafe_arena_set_allocated:xla.GetShapeRequest.data)
|
|
}
|
|
inline ::xla::GlobalDataHandle* GetShapeRequest::release_data() {
|
|
|
|
::xla::GlobalDataHandle* temp = _impl_.data_;
|
|
_impl_.data_ = 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::GlobalDataHandle* GetShapeRequest::unsafe_arena_release_data() {
|
|
// @@protoc_insertion_point(field_release:xla.GetShapeRequest.data)
|
|
|
|
::xla::GlobalDataHandle* temp = _impl_.data_;
|
|
_impl_.data_ = nullptr;
|
|
return temp;
|
|
}
|
|
inline ::xla::GlobalDataHandle* GetShapeRequest::_internal_mutable_data() {
|
|
|
|
if (_impl_.data_ == nullptr) {
|
|
auto* p = CreateMaybeMessage<::xla::GlobalDataHandle>(GetArenaForAllocation());
|
|
_impl_.data_ = p;
|
|
}
|
|
return _impl_.data_;
|
|
}
|
|
inline ::xla::GlobalDataHandle* GetShapeRequest::mutable_data() {
|
|
::xla::GlobalDataHandle* _msg = _internal_mutable_data();
|
|
// @@protoc_insertion_point(field_mutable:xla.GetShapeRequest.data)
|
|
return _msg;
|
|
}
|
|
inline void GetShapeRequest::set_allocated_data(::xla::GlobalDataHandle* data) {
|
|
::PROTOBUF_NAMESPACE_ID::Arena* message_arena = GetArenaForAllocation();
|
|
if (message_arena == nullptr) {
|
|
delete reinterpret_cast< ::PROTOBUF_NAMESPACE_ID::MessageLite*>(_impl_.data_);
|
|
}
|
|
if (data) {
|
|
::PROTOBUF_NAMESPACE_ID::Arena* submessage_arena =
|
|
::PROTOBUF_NAMESPACE_ID::Arena::InternalGetOwningArena(
|
|
reinterpret_cast<::PROTOBUF_NAMESPACE_ID::MessageLite*>(data));
|
|
if (message_arena != submessage_arena) {
|
|
data = ::PROTOBUF_NAMESPACE_ID::internal::GetOwnedMessage(
|
|
message_arena, data, submessage_arena);
|
|
}
|
|
|
|
} else {
|
|
|
|
}
|
|
_impl_.data_ = data;
|
|
// @@protoc_insertion_point(field_set_allocated:xla.GetShapeRequest.data)
|
|
}
|
|
|
|
// -------------------------------------------------------------------
|
|
|
|
// GetShapeResponse
|
|
|
|
// .xla.ShapeProto shape = 1;
|
|
inline bool GetShapeResponse::_internal_has_shape() const {
|
|
return this != internal_default_instance() && _impl_.shape_ != nullptr;
|
|
}
|
|
inline bool GetShapeResponse::has_shape() const {
|
|
return _internal_has_shape();
|
|
}
|
|
inline const ::xla::ShapeProto& GetShapeResponse::_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& GetShapeResponse::shape() const {
|
|
// @@protoc_insertion_point(field_get:xla.GetShapeResponse.shape)
|
|
return _internal_shape();
|
|
}
|
|
inline void GetShapeResponse::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.GetShapeResponse.shape)
|
|
}
|
|
inline ::xla::ShapeProto* GetShapeResponse::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* GetShapeResponse::unsafe_arena_release_shape() {
|
|
// @@protoc_insertion_point(field_release:xla.GetShapeResponse.shape)
|
|
|
|
::xla::ShapeProto* temp = _impl_.shape_;
|
|
_impl_.shape_ = nullptr;
|
|
return temp;
|
|
}
|
|
inline ::xla::ShapeProto* GetShapeResponse::_internal_mutable_shape() {
|
|
|
|
if (_impl_.shape_ == nullptr) {
|
|
auto* p = CreateMaybeMessage<::xla::ShapeProto>(GetArenaForAllocation());
|
|
_impl_.shape_ = p;
|
|
}
|
|
return _impl_.shape_;
|
|
}
|
|
inline ::xla::ShapeProto* GetShapeResponse::mutable_shape() {
|
|
::xla::ShapeProto* _msg = _internal_mutable_shape();
|
|
// @@protoc_insertion_point(field_mutable:xla.GetShapeResponse.shape)
|
|
return _msg;
|
|
}
|
|
inline void GetShapeResponse::set_allocated_shape(::xla::ShapeProto* shape) {
|
|
::PROTOBUF_NAMESPACE_ID::Arena* message_arena = GetArenaForAllocation();
|
|
if (message_arena == nullptr) {
|
|
delete reinterpret_cast< ::PROTOBUF_NAMESPACE_ID::MessageLite*>(_impl_.shape_);
|
|
}
|
|
if (shape) {
|
|
::PROTOBUF_NAMESPACE_ID::Arena* submessage_arena =
|
|
::PROTOBUF_NAMESPACE_ID::Arena::InternalGetOwningArena(
|
|
reinterpret_cast<::PROTOBUF_NAMESPACE_ID::MessageLite*>(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.GetShapeResponse.shape)
|
|
}
|
|
|
|
// -------------------------------------------------------------------
|
|
|
|
// UnpackRequest
|
|
|
|
// .xla.GlobalDataHandle data = 1;
|
|
inline bool UnpackRequest::_internal_has_data() const {
|
|
return this != internal_default_instance() && _impl_.data_ != nullptr;
|
|
}
|
|
inline bool UnpackRequest::has_data() const {
|
|
return _internal_has_data();
|
|
}
|
|
inline const ::xla::GlobalDataHandle& UnpackRequest::_internal_data() const {
|
|
const ::xla::GlobalDataHandle* p = _impl_.data_;
|
|
return p != nullptr ? *p : reinterpret_cast<const ::xla::GlobalDataHandle&>(
|
|
::xla::_GlobalDataHandle_default_instance_);
|
|
}
|
|
inline const ::xla::GlobalDataHandle& UnpackRequest::data() const {
|
|
// @@protoc_insertion_point(field_get:xla.UnpackRequest.data)
|
|
return _internal_data();
|
|
}
|
|
inline void UnpackRequest::unsafe_arena_set_allocated_data(
|
|
::xla::GlobalDataHandle* data) {
|
|
if (GetArenaForAllocation() == nullptr) {
|
|
delete reinterpret_cast<::PROTOBUF_NAMESPACE_ID::MessageLite*>(_impl_.data_);
|
|
}
|
|
_impl_.data_ = data;
|
|
if (data) {
|
|
|
|
} else {
|
|
|
|
}
|
|
// @@protoc_insertion_point(field_unsafe_arena_set_allocated:xla.UnpackRequest.data)
|
|
}
|
|
inline ::xla::GlobalDataHandle* UnpackRequest::release_data() {
|
|
|
|
::xla::GlobalDataHandle* temp = _impl_.data_;
|
|
_impl_.data_ = 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::GlobalDataHandle* UnpackRequest::unsafe_arena_release_data() {
|
|
// @@protoc_insertion_point(field_release:xla.UnpackRequest.data)
|
|
|
|
::xla::GlobalDataHandle* temp = _impl_.data_;
|
|
_impl_.data_ = nullptr;
|
|
return temp;
|
|
}
|
|
inline ::xla::GlobalDataHandle* UnpackRequest::_internal_mutable_data() {
|
|
|
|
if (_impl_.data_ == nullptr) {
|
|
auto* p = CreateMaybeMessage<::xla::GlobalDataHandle>(GetArenaForAllocation());
|
|
_impl_.data_ = p;
|
|
}
|
|
return _impl_.data_;
|
|
}
|
|
inline ::xla::GlobalDataHandle* UnpackRequest::mutable_data() {
|
|
::xla::GlobalDataHandle* _msg = _internal_mutable_data();
|
|
// @@protoc_insertion_point(field_mutable:xla.UnpackRequest.data)
|
|
return _msg;
|
|
}
|
|
inline void UnpackRequest::set_allocated_data(::xla::GlobalDataHandle* data) {
|
|
::PROTOBUF_NAMESPACE_ID::Arena* message_arena = GetArenaForAllocation();
|
|
if (message_arena == nullptr) {
|
|
delete reinterpret_cast< ::PROTOBUF_NAMESPACE_ID::MessageLite*>(_impl_.data_);
|
|
}
|
|
if (data) {
|
|
::PROTOBUF_NAMESPACE_ID::Arena* submessage_arena =
|
|
::PROTOBUF_NAMESPACE_ID::Arena::InternalGetOwningArena(
|
|
reinterpret_cast<::PROTOBUF_NAMESPACE_ID::MessageLite*>(data));
|
|
if (message_arena != submessage_arena) {
|
|
data = ::PROTOBUF_NAMESPACE_ID::internal::GetOwnedMessage(
|
|
message_arena, data, submessage_arena);
|
|
}
|
|
|
|
} else {
|
|
|
|
}
|
|
_impl_.data_ = data;
|
|
// @@protoc_insertion_point(field_set_allocated:xla.UnpackRequest.data)
|
|
}
|
|
|
|
// -------------------------------------------------------------------
|
|
|
|
// UnpackResponse
|
|
|
|
// repeated .xla.GlobalDataHandle tied_data = 1;
|
|
inline int UnpackResponse::_internal_tied_data_size() const {
|
|
return _impl_.tied_data_.size();
|
|
}
|
|
inline int UnpackResponse::tied_data_size() const {
|
|
return _internal_tied_data_size();
|
|
}
|
|
inline ::xla::GlobalDataHandle* UnpackResponse::mutable_tied_data(int index) {
|
|
// @@protoc_insertion_point(field_mutable:xla.UnpackResponse.tied_data)
|
|
return _impl_.tied_data_.Mutable(index);
|
|
}
|
|
inline ::PROTOBUF_NAMESPACE_ID::RepeatedPtrField< ::xla::GlobalDataHandle >*
|
|
UnpackResponse::mutable_tied_data() {
|
|
// @@protoc_insertion_point(field_mutable_list:xla.UnpackResponse.tied_data)
|
|
return &_impl_.tied_data_;
|
|
}
|
|
inline const ::xla::GlobalDataHandle& UnpackResponse::_internal_tied_data(int index) const {
|
|
return _impl_.tied_data_.Get(index);
|
|
}
|
|
inline const ::xla::GlobalDataHandle& UnpackResponse::tied_data(int index) const {
|
|
// @@protoc_insertion_point(field_get:xla.UnpackResponse.tied_data)
|
|
return _internal_tied_data(index);
|
|
}
|
|
inline ::xla::GlobalDataHandle* UnpackResponse::_internal_add_tied_data() {
|
|
return _impl_.tied_data_.Add();
|
|
}
|
|
inline ::xla::GlobalDataHandle* UnpackResponse::add_tied_data() {
|
|
::xla::GlobalDataHandle* _add = _internal_add_tied_data();
|
|
// @@protoc_insertion_point(field_add:xla.UnpackResponse.tied_data)
|
|
return _add;
|
|
}
|
|
inline const ::PROTOBUF_NAMESPACE_ID::RepeatedPtrField< ::xla::GlobalDataHandle >&
|
|
UnpackResponse::tied_data() const {
|
|
// @@protoc_insertion_point(field_list:xla.UnpackResponse.tied_data)
|
|
return _impl_.tied_data_;
|
|
}
|
|
|
|
// -------------------------------------------------------------------
|
|
|
|
// ScheduleProto_Instruction
|
|
|
|
// int64 id = 1;
|
|
inline void ScheduleProto_Instruction::clear_id() {
|
|
_impl_.id_ = int64_t{0};
|
|
}
|
|
inline int64_t ScheduleProto_Instruction::_internal_id() const {
|
|
return _impl_.id_;
|
|
}
|
|
inline int64_t ScheduleProto_Instruction::id() const {
|
|
// @@protoc_insertion_point(field_get:xla.ScheduleProto.Instruction.id)
|
|
return _internal_id();
|
|
}
|
|
inline void ScheduleProto_Instruction::_internal_set_id(int64_t value) {
|
|
|
|
_impl_.id_ = value;
|
|
}
|
|
inline void ScheduleProto_Instruction::set_id(int64_t value) {
|
|
_internal_set_id(value);
|
|
// @@protoc_insertion_point(field_set:xla.ScheduleProto.Instruction.id)
|
|
}
|
|
|
|
// double start_timestamp_cycles = 2;
|
|
inline void ScheduleProto_Instruction::clear_start_timestamp_cycles() {
|
|
_impl_.start_timestamp_cycles_ = 0;
|
|
}
|
|
inline double ScheduleProto_Instruction::_internal_start_timestamp_cycles() const {
|
|
return _impl_.start_timestamp_cycles_;
|
|
}
|
|
inline double ScheduleProto_Instruction::start_timestamp_cycles() const {
|
|
// @@protoc_insertion_point(field_get:xla.ScheduleProto.Instruction.start_timestamp_cycles)
|
|
return _internal_start_timestamp_cycles();
|
|
}
|
|
inline void ScheduleProto_Instruction::_internal_set_start_timestamp_cycles(double value) {
|
|
|
|
_impl_.start_timestamp_cycles_ = value;
|
|
}
|
|
inline void ScheduleProto_Instruction::set_start_timestamp_cycles(double value) {
|
|
_internal_set_start_timestamp_cycles(value);
|
|
// @@protoc_insertion_point(field_set:xla.ScheduleProto.Instruction.start_timestamp_cycles)
|
|
}
|
|
|
|
// double end_timestamp_cycles = 3;
|
|
inline void ScheduleProto_Instruction::clear_end_timestamp_cycles() {
|
|
_impl_.end_timestamp_cycles_ = 0;
|
|
}
|
|
inline double ScheduleProto_Instruction::_internal_end_timestamp_cycles() const {
|
|
return _impl_.end_timestamp_cycles_;
|
|
}
|
|
inline double ScheduleProto_Instruction::end_timestamp_cycles() const {
|
|
// @@protoc_insertion_point(field_get:xla.ScheduleProto.Instruction.end_timestamp_cycles)
|
|
return _internal_end_timestamp_cycles();
|
|
}
|
|
inline void ScheduleProto_Instruction::_internal_set_end_timestamp_cycles(double value) {
|
|
|
|
_impl_.end_timestamp_cycles_ = value;
|
|
}
|
|
inline void ScheduleProto_Instruction::set_end_timestamp_cycles(double value) {
|
|
_internal_set_end_timestamp_cycles(value);
|
|
// @@protoc_insertion_point(field_set:xla.ScheduleProto.Instruction.end_timestamp_cycles)
|
|
}
|
|
|
|
// -------------------------------------------------------------------
|
|
|
|
// ScheduleProto
|
|
|
|
// repeated .xla.ScheduleProto.Instruction instructions = 1;
|
|
inline int ScheduleProto::_internal_instructions_size() const {
|
|
return _impl_.instructions_.size();
|
|
}
|
|
inline int ScheduleProto::instructions_size() const {
|
|
return _internal_instructions_size();
|
|
}
|
|
inline void ScheduleProto::clear_instructions() {
|
|
_impl_.instructions_.Clear();
|
|
}
|
|
inline ::xla::ScheduleProto_Instruction* ScheduleProto::mutable_instructions(int index) {
|
|
// @@protoc_insertion_point(field_mutable:xla.ScheduleProto.instructions)
|
|
return _impl_.instructions_.Mutable(index);
|
|
}
|
|
inline ::PROTOBUF_NAMESPACE_ID::RepeatedPtrField< ::xla::ScheduleProto_Instruction >*
|
|
ScheduleProto::mutable_instructions() {
|
|
// @@protoc_insertion_point(field_mutable_list:xla.ScheduleProto.instructions)
|
|
return &_impl_.instructions_;
|
|
}
|
|
inline const ::xla::ScheduleProto_Instruction& ScheduleProto::_internal_instructions(int index) const {
|
|
return _impl_.instructions_.Get(index);
|
|
}
|
|
inline const ::xla::ScheduleProto_Instruction& ScheduleProto::instructions(int index) const {
|
|
// @@protoc_insertion_point(field_get:xla.ScheduleProto.instructions)
|
|
return _internal_instructions(index);
|
|
}
|
|
inline ::xla::ScheduleProto_Instruction* ScheduleProto::_internal_add_instructions() {
|
|
return _impl_.instructions_.Add();
|
|
}
|
|
inline ::xla::ScheduleProto_Instruction* ScheduleProto::add_instructions() {
|
|
::xla::ScheduleProto_Instruction* _add = _internal_add_instructions();
|
|
// @@protoc_insertion_point(field_add:xla.ScheduleProto.instructions)
|
|
return _add;
|
|
}
|
|
inline const ::PROTOBUF_NAMESPACE_ID::RepeatedPtrField< ::xla::ScheduleProto_Instruction >&
|
|
ScheduleProto::instructions() const {
|
|
// @@protoc_insertion_point(field_list:xla.ScheduleProto.instructions)
|
|
return _impl_.instructions_;
|
|
}
|
|
|
|
// int64 computation_id = 2;
|
|
inline void ScheduleProto::clear_computation_id() {
|
|
_impl_.computation_id_ = int64_t{0};
|
|
}
|
|
inline int64_t ScheduleProto::_internal_computation_id() const {
|
|
return _impl_.computation_id_;
|
|
}
|
|
inline int64_t ScheduleProto::computation_id() const {
|
|
// @@protoc_insertion_point(field_get:xla.ScheduleProto.computation_id)
|
|
return _internal_computation_id();
|
|
}
|
|
inline void ScheduleProto::_internal_set_computation_id(int64_t value) {
|
|
|
|
_impl_.computation_id_ = value;
|
|
}
|
|
inline void ScheduleProto::set_computation_id(int64_t value) {
|
|
_internal_set_computation_id(value);
|
|
// @@protoc_insertion_point(field_set:xla.ScheduleProto.computation_id)
|
|
}
|
|
|
|
// .xla.HloModuleProto hlo_module = 3;
|
|
inline bool ScheduleProto::_internal_has_hlo_module() const {
|
|
return this != internal_default_instance() && _impl_.hlo_module_ != nullptr;
|
|
}
|
|
inline bool ScheduleProto::has_hlo_module() const {
|
|
return _internal_has_hlo_module();
|
|
}
|
|
inline const ::xla::HloModuleProto& ScheduleProto::_internal_hlo_module() const {
|
|
const ::xla::HloModuleProto* p = _impl_.hlo_module_;
|
|
return p != nullptr ? *p : reinterpret_cast<const ::xla::HloModuleProto&>(
|
|
::xla::_HloModuleProto_default_instance_);
|
|
}
|
|
inline const ::xla::HloModuleProto& ScheduleProto::hlo_module() const {
|
|
// @@protoc_insertion_point(field_get:xla.ScheduleProto.hlo_module)
|
|
return _internal_hlo_module();
|
|
}
|
|
inline void ScheduleProto::unsafe_arena_set_allocated_hlo_module(
|
|
::xla::HloModuleProto* hlo_module) {
|
|
if (GetArenaForAllocation() == nullptr) {
|
|
delete reinterpret_cast<::PROTOBUF_NAMESPACE_ID::MessageLite*>(_impl_.hlo_module_);
|
|
}
|
|
_impl_.hlo_module_ = hlo_module;
|
|
if (hlo_module) {
|
|
|
|
} else {
|
|
|
|
}
|
|
// @@protoc_insertion_point(field_unsafe_arena_set_allocated:xla.ScheduleProto.hlo_module)
|
|
}
|
|
inline ::xla::HloModuleProto* ScheduleProto::release_hlo_module() {
|
|
|
|
::xla::HloModuleProto* temp = _impl_.hlo_module_;
|
|
_impl_.hlo_module_ = 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::HloModuleProto* ScheduleProto::unsafe_arena_release_hlo_module() {
|
|
// @@protoc_insertion_point(field_release:xla.ScheduleProto.hlo_module)
|
|
|
|
::xla::HloModuleProto* temp = _impl_.hlo_module_;
|
|
_impl_.hlo_module_ = nullptr;
|
|
return temp;
|
|
}
|
|
inline ::xla::HloModuleProto* ScheduleProto::_internal_mutable_hlo_module() {
|
|
|
|
if (_impl_.hlo_module_ == nullptr) {
|
|
auto* p = CreateMaybeMessage<::xla::HloModuleProto>(GetArenaForAllocation());
|
|
_impl_.hlo_module_ = p;
|
|
}
|
|
return _impl_.hlo_module_;
|
|
}
|
|
inline ::xla::HloModuleProto* ScheduleProto::mutable_hlo_module() {
|
|
::xla::HloModuleProto* _msg = _internal_mutable_hlo_module();
|
|
// @@protoc_insertion_point(field_mutable:xla.ScheduleProto.hlo_module)
|
|
return _msg;
|
|
}
|
|
inline void ScheduleProto::set_allocated_hlo_module(::xla::HloModuleProto* hlo_module) {
|
|
::PROTOBUF_NAMESPACE_ID::Arena* message_arena = GetArenaForAllocation();
|
|
if (message_arena == nullptr) {
|
|
delete reinterpret_cast< ::PROTOBUF_NAMESPACE_ID::MessageLite*>(_impl_.hlo_module_);
|
|
}
|
|
if (hlo_module) {
|
|
::PROTOBUF_NAMESPACE_ID::Arena* submessage_arena =
|
|
::PROTOBUF_NAMESPACE_ID::Arena::InternalGetOwningArena(
|
|
reinterpret_cast<::PROTOBUF_NAMESPACE_ID::MessageLite*>(hlo_module));
|
|
if (message_arena != submessage_arena) {
|
|
hlo_module = ::PROTOBUF_NAMESPACE_ID::internal::GetOwnedMessage(
|
|
message_arena, hlo_module, submessage_arena);
|
|
}
|
|
|
|
} else {
|
|
|
|
}
|
|
_impl_.hlo_module_ = hlo_module;
|
|
// @@protoc_insertion_point(field_set_allocated:xla.ScheduleProto.hlo_module)
|
|
}
|
|
|
|
// int64 cycles_per_microsecond = 4;
|
|
inline void ScheduleProto::clear_cycles_per_microsecond() {
|
|
_impl_.cycles_per_microsecond_ = int64_t{0};
|
|
}
|
|
inline int64_t ScheduleProto::_internal_cycles_per_microsecond() const {
|
|
return _impl_.cycles_per_microsecond_;
|
|
}
|
|
inline int64_t ScheduleProto::cycles_per_microsecond() const {
|
|
// @@protoc_insertion_point(field_get:xla.ScheduleProto.cycles_per_microsecond)
|
|
return _internal_cycles_per_microsecond();
|
|
}
|
|
inline void ScheduleProto::_internal_set_cycles_per_microsecond(int64_t value) {
|
|
|
|
_impl_.cycles_per_microsecond_ = value;
|
|
}
|
|
inline void ScheduleProto::set_cycles_per_microsecond(int64_t value) {
|
|
_internal_set_cycles_per_microsecond(value);
|
|
// @@protoc_insertion_point(field_set:xla.ScheduleProto.cycles_per_microsecond)
|
|
}
|
|
|
|
#ifdef __GNUC__
|
|
#pragma GCC diagnostic pop
|
|
#endif // __GNUC__
|
|
// -------------------------------------------------------------------
|
|
|
|
// -------------------------------------------------------------------
|
|
|
|
// -------------------------------------------------------------------
|
|
|
|
// -------------------------------------------------------------------
|
|
|
|
// -------------------------------------------------------------------
|
|
|
|
// -------------------------------------------------------------------
|
|
|
|
// -------------------------------------------------------------------
|
|
|
|
// -------------------------------------------------------------------
|
|
|
|
// -------------------------------------------------------------------
|
|
|
|
// -------------------------------------------------------------------
|
|
|
|
// -------------------------------------------------------------------
|
|
|
|
// -------------------------------------------------------------------
|
|
|
|
// -------------------------------------------------------------------
|
|
|
|
// -------------------------------------------------------------------
|
|
|
|
// -------------------------------------------------------------------
|
|
|
|
// -------------------------------------------------------------------
|
|
|
|
// -------------------------------------------------------------------
|
|
|
|
// -------------------------------------------------------------------
|
|
|
|
// -------------------------------------------------------------------
|
|
|
|
// -------------------------------------------------------------------
|
|
|
|
// -------------------------------------------------------------------
|
|
|
|
// -------------------------------------------------------------------
|
|
|
|
// -------------------------------------------------------------------
|
|
|
|
// -------------------------------------------------------------------
|
|
|
|
// -------------------------------------------------------------------
|
|
|
|
// -------------------------------------------------------------------
|
|
|
|
// -------------------------------------------------------------------
|
|
|
|
// -------------------------------------------------------------------
|
|
|
|
// -------------------------------------------------------------------
|
|
|
|
// -------------------------------------------------------------------
|
|
|
|
// -------------------------------------------------------------------
|
|
|
|
// -------------------------------------------------------------------
|
|
|
|
// -------------------------------------------------------------------
|
|
|
|
// -------------------------------------------------------------------
|
|
|
|
// -------------------------------------------------------------------
|
|
|
|
// -------------------------------------------------------------------
|
|
|
|
// -------------------------------------------------------------------
|
|
|
|
// -------------------------------------------------------------------
|
|
|
|
// -------------------------------------------------------------------
|
|
|
|
// -------------------------------------------------------------------
|
|
|
|
// -------------------------------------------------------------------
|
|
|
|
// -------------------------------------------------------------------
|
|
|
|
// -------------------------------------------------------------------
|
|
|
|
// -------------------------------------------------------------------
|
|
|
|
// -------------------------------------------------------------------
|
|
|
|
// -------------------------------------------------------------------
|
|
|
|
// -------------------------------------------------------------------
|
|
|
|
// -------------------------------------------------------------------
|
|
|
|
// -------------------------------------------------------------------
|
|
|
|
|
|
// @@protoc_insertion_point(namespace_scope)
|
|
|
|
} // namespace xla
|
|
|
|
PROTOBUF_NAMESPACE_OPEN
|
|
|
|
template <> struct is_proto_enum< ::xla::DebugOptions_ShapeChecks> : ::std::true_type {};
|
|
template <>
|
|
inline const EnumDescriptor* GetEnumDescriptor< ::xla::DebugOptions_ShapeChecks>() {
|
|
return ::xla::DebugOptions_ShapeChecks_descriptor();
|
|
}
|
|
template <> struct is_proto_enum< ::xla::DebugOptions_StepMarkerLocation> : ::std::true_type {};
|
|
template <>
|
|
inline const EnumDescriptor* GetEnumDescriptor< ::xla::DebugOptions_StepMarkerLocation>() {
|
|
return ::xla::DebugOptions_StepMarkerLocation_descriptor();
|
|
}
|
|
template <> struct is_proto_enum< ::xla::DebugOptions_CommandBufferCmdType> : ::std::true_type {};
|
|
template <>
|
|
inline const EnumDescriptor* GetEnumDescriptor< ::xla::DebugOptions_CommandBufferCmdType>() {
|
|
return ::xla::DebugOptions_CommandBufferCmdType_descriptor();
|
|
}
|
|
template <> struct is_proto_enum< ::xla::DebugOptions_PartitioningAlgorithm> : ::std::true_type {};
|
|
template <>
|
|
inline const EnumDescriptor* GetEnumDescriptor< ::xla::DebugOptions_PartitioningAlgorithm>() {
|
|
return ::xla::DebugOptions_PartitioningAlgorithm_descriptor();
|
|
}
|
|
template <> struct is_proto_enum< ::xla::HloModuleConfigProto_FusionConfigCollection> : ::std::true_type {};
|
|
template <>
|
|
inline const EnumDescriptor* GetEnumDescriptor< ::xla::HloModuleConfigProto_FusionConfigCollection>() {
|
|
return ::xla::HloModuleConfigProto_FusionConfigCollection_descriptor();
|
|
}
|
|
|
|
PROTOBUF_NAMESPACE_CLOSE
|
|
|
|
// @@protoc_insertion_point(global_scope)
|
|
|
|
#include <google/protobuf/port_undef.inc>
|
|
#endif // GOOGLE_PROTOBUF_INCLUDED_GOOGLE_PROTOBUF_INCLUDED_xla_2fxla_2eproto
|