// Copyright 2025 Google LLC
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
//     http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.

// Code generated by protoc-gen-go. DO NOT EDIT.
// versions:
// 	protoc-gen-go v1.36.11
// 	protoc        v6.33.2
// source: google/cloud/aiplatform/v1/nas_job.proto

package aiplatformpb

import (
	reflect "reflect"
	sync "sync"
	unsafe "unsafe"

	_ "google.golang.org/genproto/googleapis/api/annotations"
	status "google.golang.org/genproto/googleapis/rpc/status"
	protoreflect "google.golang.org/protobuf/reflect/protoreflect"
	protoimpl "google.golang.org/protobuf/runtime/protoimpl"
	timestamppb "google.golang.org/protobuf/types/known/timestamppb"
)

const (
	// Verify that this generated code is sufficiently up-to-date.
	_ = protoimpl.EnforceVersion(20 - protoimpl.MinVersion)
	// Verify that runtime/protoimpl is sufficiently up-to-date.
	_ = protoimpl.EnforceVersion(protoimpl.MaxVersion - 20)
)

// The available types of multi-trial algorithms.
type NasJobSpec_MultiTrialAlgorithmSpec_MultiTrialAlgorithm int32

const (
	// Defaults to `REINFORCEMENT_LEARNING`.
	NasJobSpec_MultiTrialAlgorithmSpec_MULTI_TRIAL_ALGORITHM_UNSPECIFIED NasJobSpec_MultiTrialAlgorithmSpec_MultiTrialAlgorithm = 0
	// The Reinforcement Learning Algorithm for Multi-trial Neural
	// Architecture Search (NAS).
	NasJobSpec_MultiTrialAlgorithmSpec_REINFORCEMENT_LEARNING NasJobSpec_MultiTrialAlgorithmSpec_MultiTrialAlgorithm = 1
	// The Grid Search Algorithm for Multi-trial Neural
	// Architecture Search (NAS).
	NasJobSpec_MultiTrialAlgorithmSpec_GRID_SEARCH NasJobSpec_MultiTrialAlgorithmSpec_MultiTrialAlgorithm = 2
)

// Enum value maps for NasJobSpec_MultiTrialAlgorithmSpec_MultiTrialAlgorithm.
var (
	NasJobSpec_MultiTrialAlgorithmSpec_MultiTrialAlgorithm_name = map[int32]string{
		0: "MULTI_TRIAL_ALGORITHM_UNSPECIFIED",
		1: "REINFORCEMENT_LEARNING",
		2: "GRID_SEARCH",
	}
	NasJobSpec_MultiTrialAlgorithmSpec_MultiTrialAlgorithm_value = map[string]int32{
		"MULTI_TRIAL_ALGORITHM_UNSPECIFIED": 0,
		"REINFORCEMENT_LEARNING":            1,
		"GRID_SEARCH":                       2,
	}
)

func (x NasJobSpec_MultiTrialAlgorithmSpec_MultiTrialAlgorithm) Enum() *NasJobSpec_MultiTrialAlgorithmSpec_MultiTrialAlgorithm {
	p := new(NasJobSpec_MultiTrialAlgorithmSpec_MultiTrialAlgorithm)
	*p = x
	return p
}

func (x NasJobSpec_MultiTrialAlgorithmSpec_MultiTrialAlgorithm) String() string {
	return protoimpl.X.EnumStringOf(x.Descriptor(), protoreflect.EnumNumber(x))
}

func (NasJobSpec_MultiTrialAlgorithmSpec_MultiTrialAlgorithm) Descriptor() protoreflect.EnumDescriptor {
	return file_google_cloud_aiplatform_v1_nas_job_proto_enumTypes[0].Descriptor()
}

func (NasJobSpec_MultiTrialAlgorithmSpec_MultiTrialAlgorithm) Type() protoreflect.EnumType {
	return &file_google_cloud_aiplatform_v1_nas_job_proto_enumTypes[0]
}

func (x NasJobSpec_MultiTrialAlgorithmSpec_MultiTrialAlgorithm) Number() protoreflect.EnumNumber {
	return protoreflect.EnumNumber(x)
}

// Deprecated: Use NasJobSpec_MultiTrialAlgorithmSpec_MultiTrialAlgorithm.Descriptor instead.
func (NasJobSpec_MultiTrialAlgorithmSpec_MultiTrialAlgorithm) EnumDescriptor() ([]byte, []int) {
	return file_google_cloud_aiplatform_v1_nas_job_proto_rawDescGZIP(), []int{2, 0, 0}
}

// The available types of optimization goals.
type NasJobSpec_MultiTrialAlgorithmSpec_MetricSpec_GoalType int32

const (
	// Goal Type will default to maximize.
	NasJobSpec_MultiTrialAlgorithmSpec_MetricSpec_GOAL_TYPE_UNSPECIFIED NasJobSpec_MultiTrialAlgorithmSpec_MetricSpec_GoalType = 0
	// Maximize the goal metric.
	NasJobSpec_MultiTrialAlgorithmSpec_MetricSpec_MAXIMIZE NasJobSpec_MultiTrialAlgorithmSpec_MetricSpec_GoalType = 1
	// Minimize the goal metric.
	NasJobSpec_MultiTrialAlgorithmSpec_MetricSpec_MINIMIZE NasJobSpec_MultiTrialAlgorithmSpec_MetricSpec_GoalType = 2
)

// Enum value maps for NasJobSpec_MultiTrialAlgorithmSpec_MetricSpec_GoalType.
var (
	NasJobSpec_MultiTrialAlgorithmSpec_MetricSpec_GoalType_name = map[int32]string{
		0: "GOAL_TYPE_UNSPECIFIED",
		1: "MAXIMIZE",
		2: "MINIMIZE",
	}
	NasJobSpec_MultiTrialAlgorithmSpec_MetricSpec_GoalType_value = map[string]int32{
		"GOAL_TYPE_UNSPECIFIED": 0,
		"MAXIMIZE":              1,
		"MINIMIZE":              2,
	}
)

func (x NasJobSpec_MultiTrialAlgorithmSpec_MetricSpec_GoalType) Enum() *NasJobSpec_MultiTrialAlgorithmSpec_MetricSpec_GoalType {
	p := new(NasJobSpec_MultiTrialAlgorithmSpec_MetricSpec_GoalType)
	*p = x
	return p
}

func (x NasJobSpec_MultiTrialAlgorithmSpec_MetricSpec_GoalType) String() string {
	return protoimpl.X.EnumStringOf(x.Descriptor(), protoreflect.EnumNumber(x))
}

func (NasJobSpec_MultiTrialAlgorithmSpec_MetricSpec_GoalType) Descriptor() protoreflect.EnumDescriptor {
	return file_google_cloud_aiplatform_v1_nas_job_proto_enumTypes[1].Descriptor()
}

func (NasJobSpec_MultiTrialAlgorithmSpec_MetricSpec_GoalType) Type() protoreflect.EnumType {
	return &file_google_cloud_aiplatform_v1_nas_job_proto_enumTypes[1]
}

func (x NasJobSpec_MultiTrialAlgorithmSpec_MetricSpec_GoalType) Number() protoreflect.EnumNumber {
	return protoreflect.EnumNumber(x)
}

// Deprecated: Use NasJobSpec_MultiTrialAlgorithmSpec_MetricSpec_GoalType.Descriptor instead.
func (NasJobSpec_MultiTrialAlgorithmSpec_MetricSpec_GoalType) EnumDescriptor() ([]byte, []int) {
	return file_google_cloud_aiplatform_v1_nas_job_proto_rawDescGZIP(), []int{2, 0, 0, 0}
}

// Describes a NasTrial state.
type NasTrial_State int32

const (
	// The NasTrial state is unspecified.
	NasTrial_STATE_UNSPECIFIED NasTrial_State = 0
	// Indicates that a specific NasTrial has been requested, but it has not yet
	// been suggested by the service.
	NasTrial_REQUESTED NasTrial_State = 1
	// Indicates that the NasTrial has been suggested.
	NasTrial_ACTIVE NasTrial_State = 2
	// Indicates that the NasTrial should stop according to the service.
	NasTrial_STOPPING NasTrial_State = 3
	// Indicates that the NasTrial is completed successfully.
	NasTrial_SUCCEEDED NasTrial_State = 4
	// Indicates that the NasTrial should not be attempted again.
	// The service will set a NasTrial to INFEASIBLE when it's done but missing
	// the final_measurement.
	NasTrial_INFEASIBLE NasTrial_State = 5
)

// Enum value maps for NasTrial_State.
var (
	NasTrial_State_name = map[int32]string{
		0: "STATE_UNSPECIFIED",
		1: "REQUESTED",
		2: "ACTIVE",
		3: "STOPPING",
		4: "SUCCEEDED",
		5: "INFEASIBLE",
	}
	NasTrial_State_value = map[string]int32{
		"STATE_UNSPECIFIED": 0,
		"REQUESTED":         1,
		"ACTIVE":            2,
		"STOPPING":          3,
		"SUCCEEDED":         4,
		"INFEASIBLE":        5,
	}
)

func (x NasTrial_State) Enum() *NasTrial_State {
	p := new(NasTrial_State)
	*p = x
	return p
}

func (x NasTrial_State) String() string {
	return protoimpl.X.EnumStringOf(x.Descriptor(), protoreflect.EnumNumber(x))
}

func (NasTrial_State) Descriptor() protoreflect.EnumDescriptor {
	return file_google_cloud_aiplatform_v1_nas_job_proto_enumTypes[2].Descriptor()
}

func (NasTrial_State) Type() protoreflect.EnumType {
	return &file_google_cloud_aiplatform_v1_nas_job_proto_enumTypes[2]
}

func (x NasTrial_State) Number() protoreflect.EnumNumber {
	return protoreflect.EnumNumber(x)
}

// Deprecated: Use NasTrial_State.Descriptor instead.
func (NasTrial_State) EnumDescriptor() ([]byte, []int) {
	return file_google_cloud_aiplatform_v1_nas_job_proto_rawDescGZIP(), []int{4, 0}
}

// Represents a Neural Architecture Search (NAS) job.
type NasJob struct {
	state protoimpl.MessageState `protogen:"open.v1"`
	// Output only. Resource name of the NasJob.
	Name string `protobuf:"bytes,1,opt,name=name,proto3" json:"name,omitempty"`
	// Required. The display name of the NasJob.
	// The name can be up to 128 characters long and can consist of any UTF-8
	// characters.
	DisplayName string `protobuf:"bytes,2,opt,name=display_name,json=displayName,proto3" json:"display_name,omitempty"`
	// Required. The specification of a NasJob.
	NasJobSpec *NasJobSpec `protobuf:"bytes,4,opt,name=nas_job_spec,json=nasJobSpec,proto3" json:"nas_job_spec,omitempty"`
	// Output only. Output of the NasJob.
	NasJobOutput *NasJobOutput `protobuf:"bytes,5,opt,name=nas_job_output,json=nasJobOutput,proto3" json:"nas_job_output,omitempty"`
	// Output only. The detailed state of the job.
	State JobState `protobuf:"varint,6,opt,name=state,proto3,enum=google.cloud.aiplatform.v1.JobState" json:"state,omitempty"`
	// Output only. Time when the NasJob was created.
	CreateTime *timestamppb.Timestamp `protobuf:"bytes,7,opt,name=create_time,json=createTime,proto3" json:"create_time,omitempty"`
	// Output only. Time when the NasJob for the first time entered the
	// `JOB_STATE_RUNNING` state.
	StartTime *timestamppb.Timestamp `protobuf:"bytes,8,opt,name=start_time,json=startTime,proto3" json:"start_time,omitempty"`
	// Output only. Time when the NasJob entered any of the following states:
	// `JOB_STATE_SUCCEEDED`, `JOB_STATE_FAILED`, `JOB_STATE_CANCELLED`.
	EndTime *timestamppb.Timestamp `protobuf:"bytes,9,opt,name=end_time,json=endTime,proto3" json:"end_time,omitempty"`
	// Output only. Time when the NasJob was most recently updated.
	UpdateTime *timestamppb.Timestamp `protobuf:"bytes,10,opt,name=update_time,json=updateTime,proto3" json:"update_time,omitempty"`
	// Output only. Only populated when job's state is JOB_STATE_FAILED or
	// JOB_STATE_CANCELLED.
	Error *status.Status `protobuf:"bytes,11,opt,name=error,proto3" json:"error,omitempty"`
	// The labels with user-defined metadata to organize NasJobs.
	//
	// Label keys and values can be no longer than 64 characters
	// (Unicode codepoints), can only contain lowercase letters, numeric
	// characters, underscores and dashes. International characters are allowed.
	//
	// See https://goo.gl/xmQnxf for more information and examples of labels.
	Labels map[string]string `protobuf:"bytes,12,rep,name=labels,proto3" json:"labels,omitempty" protobuf_key:"bytes,1,opt,name=key" protobuf_val:"bytes,2,opt,name=value"`
	// Customer-managed encryption key options for a NasJob.
	// If this is set, then all resources created by the NasJob
	// will be encrypted with the provided encryption key.
	EncryptionSpec *EncryptionSpec `protobuf:"bytes,13,opt,name=encryption_spec,json=encryptionSpec,proto3" json:"encryption_spec,omitempty"`
	// Optional. Enable a separation of Custom model training
	// and restricted image training for tenant project.
	//
	// Deprecated: Marked as deprecated in google/cloud/aiplatform/v1/nas_job.proto.
	EnableRestrictedImageTraining bool `protobuf:"varint,14,opt,name=enable_restricted_image_training,json=enableRestrictedImageTraining,proto3" json:"enable_restricted_image_training,omitempty"`
	// Output only. Reserved for future use.
	SatisfiesPzs bool `protobuf:"varint,15,opt,name=satisfies_pzs,json=satisfiesPzs,proto3" json:"satisfies_pzs,omitempty"`
	// Output only. Reserved for future use.
	SatisfiesPzi  bool `protobuf:"varint,16,opt,name=satisfies_pzi,json=satisfiesPzi,proto3" json:"satisfies_pzi,omitempty"`
	unknownFields protoimpl.UnknownFields
	sizeCache     protoimpl.SizeCache
}

func (x *NasJob) Reset() {
	*x = NasJob{}
	mi := &file_google_cloud_aiplatform_v1_nas_job_proto_msgTypes[0]
	ms := protoimpl.X.MessageStateOf(protoimpl.Pointer(x))
	ms.StoreMessageInfo(mi)
}

func (x *NasJob) String() string {
	return protoimpl.X.MessageStringOf(x)
}

func (*NasJob) ProtoMessage() {}

func (x *NasJob) ProtoReflect() protoreflect.Message {
	mi := &file_google_cloud_aiplatform_v1_nas_job_proto_msgTypes[0]
	if x != nil {
		ms := protoimpl.X.MessageStateOf(protoimpl.Pointer(x))
		if ms.LoadMessageInfo() == nil {
			ms.StoreMessageInfo(mi)
		}
		return ms
	}
	return mi.MessageOf(x)
}

// Deprecated: Use NasJob.ProtoReflect.Descriptor instead.
func (*NasJob) Descriptor() ([]byte, []int) {
	return file_google_cloud_aiplatform_v1_nas_job_proto_rawDescGZIP(), []int{0}
}

func (x *NasJob) GetName() string {
	if x != nil {
		return x.Name
	}
	return ""
}

func (x *NasJob) GetDisplayName() string {
	if x != nil {
		return x.DisplayName
	}
	return ""
}

func (x *NasJob) GetNasJobSpec() *NasJobSpec {
	if x != nil {
		return x.NasJobSpec
	}
	return nil
}

func (x *NasJob) GetNasJobOutput() *NasJobOutput {
	if x != nil {
		return x.NasJobOutput
	}
	return nil
}

func (x *NasJob) GetState() JobState {
	if x != nil {
		return x.State
	}
	return JobState_JOB_STATE_UNSPECIFIED
}

func (x *NasJob) GetCreateTime() *timestamppb.Timestamp {
	if x != nil {
		return x.CreateTime
	}
	return nil
}

func (x *NasJob) GetStartTime() *timestamppb.Timestamp {
	if x != nil {
		return x.StartTime
	}
	return nil
}

func (x *NasJob) GetEndTime() *timestamppb.Timestamp {
	if x != nil {
		return x.EndTime
	}
	return nil
}

func (x *NasJob) GetUpdateTime() *timestamppb.Timestamp {
	if x != nil {
		return x.UpdateTime
	}
	return nil
}

func (x *NasJob) GetError() *status.Status {
	if x != nil {
		return x.Error
	}
	return nil
}

func (x *NasJob) GetLabels() map[string]string {
	if x != nil {
		return x.Labels
	}
	return nil
}

func (x *NasJob) GetEncryptionSpec() *EncryptionSpec {
	if x != nil {
		return x.EncryptionSpec
	}
	return nil
}

// Deprecated: Marked as deprecated in google/cloud/aiplatform/v1/nas_job.proto.
func (x *NasJob) GetEnableRestrictedImageTraining() bool {
	if x != nil {
		return x.EnableRestrictedImageTraining
	}
	return false
}

func (x *NasJob) GetSatisfiesPzs() bool {
	if x != nil {
		return x.SatisfiesPzs
	}
	return false
}

func (x *NasJob) GetSatisfiesPzi() bool {
	if x != nil {
		return x.SatisfiesPzi
	}
	return false
}

// Represents a NasTrial details along with its parameters. If there is a
// corresponding train NasTrial, the train NasTrial is also returned.
type NasTrialDetail struct {
	state protoimpl.MessageState `protogen:"open.v1"`
	// Output only. Resource name of the NasTrialDetail.
	Name string `protobuf:"bytes,1,opt,name=name,proto3" json:"name,omitempty"`
	// The parameters for the NasJob NasTrial.
	Parameters string `protobuf:"bytes,2,opt,name=parameters,proto3" json:"parameters,omitempty"`
	// The requested search NasTrial.
	SearchTrial *NasTrial `protobuf:"bytes,3,opt,name=search_trial,json=searchTrial,proto3" json:"search_trial,omitempty"`
	// The train NasTrial corresponding to
	// [search_trial][google.cloud.aiplatform.v1.NasTrialDetail.search_trial].
	// Only populated if
	// [search_trial][google.cloud.aiplatform.v1.NasTrialDetail.search_trial] is
	// used for training.
	TrainTrial    *NasTrial `protobuf:"bytes,4,opt,name=train_trial,json=trainTrial,proto3" json:"train_trial,omitempty"`
	unknownFields protoimpl.UnknownFields
	sizeCache     protoimpl.SizeCache
}

func (x *NasTrialDetail) Reset() {
	*x = NasTrialDetail{}
	mi := &file_google_cloud_aiplatform_v1_nas_job_proto_msgTypes[1]
	ms := protoimpl.X.MessageStateOf(protoimpl.Pointer(x))
	ms.StoreMessageInfo(mi)
}

func (x *NasTrialDetail) String() string {
	return protoimpl.X.MessageStringOf(x)
}

func (*NasTrialDetail) ProtoMessage() {}

func (x *NasTrialDetail) ProtoReflect() protoreflect.Message {
	mi := &file_google_cloud_aiplatform_v1_nas_job_proto_msgTypes[1]
	if x != nil {
		ms := protoimpl.X.MessageStateOf(protoimpl.Pointer(x))
		if ms.LoadMessageInfo() == nil {
			ms.StoreMessageInfo(mi)
		}
		return ms
	}
	return mi.MessageOf(x)
}

// Deprecated: Use NasTrialDetail.ProtoReflect.Descriptor instead.
func (*NasTrialDetail) Descriptor() ([]byte, []int) {
	return file_google_cloud_aiplatform_v1_nas_job_proto_rawDescGZIP(), []int{1}
}

func (x *NasTrialDetail) GetName() string {
	if x != nil {
		return x.Name
	}
	return ""
}

func (x *NasTrialDetail) GetParameters() string {
	if x != nil {
		return x.Parameters
	}
	return ""
}

func (x *NasTrialDetail) GetSearchTrial() *NasTrial {
	if x != nil {
		return x.SearchTrial
	}
	return nil
}

func (x *NasTrialDetail) GetTrainTrial() *NasTrial {
	if x != nil {
		return x.TrainTrial
	}
	return nil
}

// Represents the spec of a NasJob.
type NasJobSpec struct {
	state protoimpl.MessageState `protogen:"open.v1"`
	// The Neural Architecture Search (NAS) algorithm specification.
	//
	// Types that are valid to be assigned to NasAlgorithmSpec:
	//
	//	*NasJobSpec_MultiTrialAlgorithmSpec_
	NasAlgorithmSpec isNasJobSpec_NasAlgorithmSpec `protobuf_oneof:"nas_algorithm_spec"`
	// The ID of the existing NasJob in the same Project and Location
	// which will be used to resume search. search_space_spec and
	// nas_algorithm_spec are obtained from previous NasJob hence should not
	// provide them again for this NasJob.
	ResumeNasJobId string `protobuf:"bytes,3,opt,name=resume_nas_job_id,json=resumeNasJobId,proto3" json:"resume_nas_job_id,omitempty"`
	// It defines the search space for Neural Architecture Search (NAS).
	SearchSpaceSpec string `protobuf:"bytes,1,opt,name=search_space_spec,json=searchSpaceSpec,proto3" json:"search_space_spec,omitempty"`
	unknownFields   protoimpl.UnknownFields
	sizeCache       protoimpl.SizeCache
}

func (x *NasJobSpec) Reset() {
	*x = NasJobSpec{}
	mi := &file_google_cloud_aiplatform_v1_nas_job_proto_msgTypes[2]
	ms := protoimpl.X.MessageStateOf(protoimpl.Pointer(x))
	ms.StoreMessageInfo(mi)
}

func (x *NasJobSpec) String() string {
	return protoimpl.X.MessageStringOf(x)
}

func (*NasJobSpec) ProtoMessage() {}

func (x *NasJobSpec) ProtoReflect() protoreflect.Message {
	mi := &file_google_cloud_aiplatform_v1_nas_job_proto_msgTypes[2]
	if x != nil {
		ms := protoimpl.X.MessageStateOf(protoimpl.Pointer(x))
		if ms.LoadMessageInfo() == nil {
			ms.StoreMessageInfo(mi)
		}
		return ms
	}
	return mi.MessageOf(x)
}

// Deprecated: Use NasJobSpec.ProtoReflect.Descriptor instead.
func (*NasJobSpec) Descriptor() ([]byte, []int) {
	return file_google_cloud_aiplatform_v1_nas_job_proto_rawDescGZIP(), []int{2}
}

func (x *NasJobSpec) GetNasAlgorithmSpec() isNasJobSpec_NasAlgorithmSpec {
	if x != nil {
		return x.NasAlgorithmSpec
	}
	return nil
}

func (x *NasJobSpec) GetMultiTrialAlgorithmSpec() *NasJobSpec_MultiTrialAlgorithmSpec {
	if x != nil {
		if x, ok := x.NasAlgorithmSpec.(*NasJobSpec_MultiTrialAlgorithmSpec_); ok {
			return x.MultiTrialAlgorithmSpec
		}
	}
	return nil
}

func (x *NasJobSpec) GetResumeNasJobId() string {
	if x != nil {
		return x.ResumeNasJobId
	}
	return ""
}

func (x *NasJobSpec) GetSearchSpaceSpec() string {
	if x != nil {
		return x.SearchSpaceSpec
	}
	return ""
}

type isNasJobSpec_NasAlgorithmSpec interface {
	isNasJobSpec_NasAlgorithmSpec()
}

type NasJobSpec_MultiTrialAlgorithmSpec_ struct {
	// The spec of multi-trial algorithms.
	MultiTrialAlgorithmSpec *NasJobSpec_MultiTrialAlgorithmSpec `protobuf:"bytes,2,opt,name=multi_trial_algorithm_spec,json=multiTrialAlgorithmSpec,proto3,oneof"`
}

func (*NasJobSpec_MultiTrialAlgorithmSpec_) isNasJobSpec_NasAlgorithmSpec() {}

// Represents a uCAIP NasJob output.
type NasJobOutput struct {
	state protoimpl.MessageState `protogen:"open.v1"`
	// The output of this Neural Architecture Search (NAS) job.
	//
	// Types that are valid to be assigned to Output:
	//
	//	*NasJobOutput_MultiTrialJobOutput_
	Output        isNasJobOutput_Output `protobuf_oneof:"output"`
	unknownFields protoimpl.UnknownFields
	sizeCache     protoimpl.SizeCache
}

func (x *NasJobOutput) Reset() {
	*x = NasJobOutput{}
	mi := &file_google_cloud_aiplatform_v1_nas_job_proto_msgTypes[3]
	ms := protoimpl.X.MessageStateOf(protoimpl.Pointer(x))
	ms.StoreMessageInfo(mi)
}

func (x *NasJobOutput) String() string {
	return protoimpl.X.MessageStringOf(x)
}

func (*NasJobOutput) ProtoMessage() {}

func (x *NasJobOutput) ProtoReflect() protoreflect.Message {
	mi := &file_google_cloud_aiplatform_v1_nas_job_proto_msgTypes[3]
	if x != nil {
		ms := protoimpl.X.MessageStateOf(protoimpl.Pointer(x))
		if ms.LoadMessageInfo() == nil {
			ms.StoreMessageInfo(mi)
		}
		return ms
	}
	return mi.MessageOf(x)
}

// Deprecated: Use NasJobOutput.ProtoReflect.Descriptor instead.
func (*NasJobOutput) Descriptor() ([]byte, []int) {
	return file_google_cloud_aiplatform_v1_nas_job_proto_rawDescGZIP(), []int{3}
}

func (x *NasJobOutput) GetOutput() isNasJobOutput_Output {
	if x != nil {
		return x.Output
	}
	return nil
}

func (x *NasJobOutput) GetMultiTrialJobOutput() *NasJobOutput_MultiTrialJobOutput {
	if x != nil {
		if x, ok := x.Output.(*NasJobOutput_MultiTrialJobOutput_); ok {
			return x.MultiTrialJobOutput
		}
	}
	return nil
}

type isNasJobOutput_Output interface {
	isNasJobOutput_Output()
}

type NasJobOutput_MultiTrialJobOutput_ struct {
	// Output only. The output of this multi-trial Neural Architecture Search
	// (NAS) job.
	MultiTrialJobOutput *NasJobOutput_MultiTrialJobOutput `protobuf:"bytes,1,opt,name=multi_trial_job_output,json=multiTrialJobOutput,proto3,oneof"`
}

func (*NasJobOutput_MultiTrialJobOutput_) isNasJobOutput_Output() {}

// Represents a uCAIP NasJob trial.
type NasTrial struct {
	state protoimpl.MessageState `protogen:"open.v1"`
	// Output only. The identifier of the NasTrial assigned by the service.
	Id string `protobuf:"bytes,1,opt,name=id,proto3" json:"id,omitempty"`
	// Output only. The detailed state of the NasTrial.
	State NasTrial_State `protobuf:"varint,2,opt,name=state,proto3,enum=google.cloud.aiplatform.v1.NasTrial_State" json:"state,omitempty"`
	// Output only. The final measurement containing the objective value.
	FinalMeasurement *Measurement `protobuf:"bytes,3,opt,name=final_measurement,json=finalMeasurement,proto3" json:"final_measurement,omitempty"`
	// Output only. Time when the NasTrial was started.
	StartTime *timestamppb.Timestamp `protobuf:"bytes,4,opt,name=start_time,json=startTime,proto3" json:"start_time,omitempty"`
	// Output only. Time when the NasTrial's status changed to `SUCCEEDED` or
	// `INFEASIBLE`.
	EndTime       *timestamppb.Timestamp `protobuf:"bytes,5,opt,name=end_time,json=endTime,proto3" json:"end_time,omitempty"`
	unknownFields protoimpl.UnknownFields
	sizeCache     protoimpl.SizeCache
}

func (x *NasTrial) Reset() {
	*x = NasTrial{}
	mi := &file_google_cloud_aiplatform_v1_nas_job_proto_msgTypes[4]
	ms := protoimpl.X.MessageStateOf(protoimpl.Pointer(x))
	ms.StoreMessageInfo(mi)
}

func (x *NasTrial) String() string {
	return protoimpl.X.MessageStringOf(x)
}

func (*NasTrial) ProtoMessage() {}

func (x *NasTrial) ProtoReflect() protoreflect.Message {
	mi := &file_google_cloud_aiplatform_v1_nas_job_proto_msgTypes[4]
	if x != nil {
		ms := protoimpl.X.MessageStateOf(protoimpl.Pointer(x))
		if ms.LoadMessageInfo() == nil {
			ms.StoreMessageInfo(mi)
		}
		return ms
	}
	return mi.MessageOf(x)
}

// Deprecated: Use NasTrial.ProtoReflect.Descriptor instead.
func (*NasTrial) Descriptor() ([]byte, []int) {
	return file_google_cloud_aiplatform_v1_nas_job_proto_rawDescGZIP(), []int{4}
}

func (x *NasTrial) GetId() string {
	if x != nil {
		return x.Id
	}
	return ""
}

func (x *NasTrial) GetState() NasTrial_State {
	if x != nil {
		return x.State
	}
	return NasTrial_STATE_UNSPECIFIED
}

func (x *NasTrial) GetFinalMeasurement() *Measurement {
	if x != nil {
		return x.FinalMeasurement
	}
	return nil
}

func (x *NasTrial) GetStartTime() *timestamppb.Timestamp {
	if x != nil {
		return x.StartTime
	}
	return nil
}

func (x *NasTrial) GetEndTime() *timestamppb.Timestamp {
	if x != nil {
		return x.EndTime
	}
	return nil
}

// The spec of multi-trial Neural Architecture Search (NAS).
type NasJobSpec_MultiTrialAlgorithmSpec struct {
	state protoimpl.MessageState `protogen:"open.v1"`
	// The multi-trial Neural Architecture Search (NAS) algorithm
	// type. Defaults to `REINFORCEMENT_LEARNING`.
	MultiTrialAlgorithm NasJobSpec_MultiTrialAlgorithmSpec_MultiTrialAlgorithm `protobuf:"varint,1,opt,name=multi_trial_algorithm,json=multiTrialAlgorithm,proto3,enum=google.cloud.aiplatform.v1.NasJobSpec_MultiTrialAlgorithmSpec_MultiTrialAlgorithm" json:"multi_trial_algorithm,omitempty"`
	// Metric specs for the NAS job.
	// Validation for this field is done at `multi_trial_algorithm_spec` field.
	Metric *NasJobSpec_MultiTrialAlgorithmSpec_MetricSpec `protobuf:"bytes,2,opt,name=metric,proto3" json:"metric,omitempty"`
	// Required. Spec for search trials.
	SearchTrialSpec *NasJobSpec_MultiTrialAlgorithmSpec_SearchTrialSpec `protobuf:"bytes,3,opt,name=search_trial_spec,json=searchTrialSpec,proto3" json:"search_trial_spec,omitempty"`
	// Spec for train trials. Top N [TrainTrialSpec.max_parallel_trial_count]
	// search trials will be trained for every M
	// [TrainTrialSpec.frequency] trials searched.
	TrainTrialSpec *NasJobSpec_MultiTrialAlgorithmSpec_TrainTrialSpec `protobuf:"bytes,4,opt,name=train_trial_spec,json=trainTrialSpec,proto3" json:"train_trial_spec,omitempty"`
	unknownFields  protoimpl.UnknownFields
	sizeCache      protoimpl.SizeCache
}

func (x *NasJobSpec_MultiTrialAlgorithmSpec) Reset() {
	*x = NasJobSpec_MultiTrialAlgorithmSpec{}
	mi := &file_google_cloud_aiplatform_v1_nas_job_proto_msgTypes[6]
	ms := protoimpl.X.MessageStateOf(protoimpl.Pointer(x))
	ms.StoreMessageInfo(mi)
}

func (x *NasJobSpec_MultiTrialAlgorithmSpec) String() string {
	return protoimpl.X.MessageStringOf(x)
}

func (*NasJobSpec_MultiTrialAlgorithmSpec) ProtoMessage() {}

func (x *NasJobSpec_MultiTrialAlgorithmSpec) ProtoReflect() protoreflect.Message {
	mi := &file_google_cloud_aiplatform_v1_nas_job_proto_msgTypes[6]
	if x != nil {
		ms := protoimpl.X.MessageStateOf(protoimpl.Pointer(x))
		if ms.LoadMessageInfo() == nil {
			ms.StoreMessageInfo(mi)
		}
		return ms
	}
	return mi.MessageOf(x)
}

// Deprecated: Use NasJobSpec_MultiTrialAlgorithmSpec.ProtoReflect.Descriptor instead.
func (*NasJobSpec_MultiTrialAlgorithmSpec) Descriptor() ([]byte, []int) {
	return file_google_cloud_aiplatform_v1_nas_job_proto_rawDescGZIP(), []int{2, 0}
}

func (x *NasJobSpec_MultiTrialAlgorithmSpec) GetMultiTrialAlgorithm() NasJobSpec_MultiTrialAlgorithmSpec_MultiTrialAlgorithm {
	if x != nil {
		return x.MultiTrialAlgorithm
	}
	return NasJobSpec_MultiTrialAlgorithmSpec_MULTI_TRIAL_ALGORITHM_UNSPECIFIED
}

func (x *NasJobSpec_MultiTrialAlgorithmSpec) GetMetric() *NasJobSpec_MultiTrialAlgorithmSpec_MetricSpec {
	if x != nil {
		return x.Metric
	}
	return nil
}

func (x *NasJobSpec_MultiTrialAlgorithmSpec) GetSearchTrialSpec() *NasJobSpec_MultiTrialAlgorithmSpec_SearchTrialSpec {
	if x != nil {
		return x.SearchTrialSpec
	}
	return nil
}

func (x *NasJobSpec_MultiTrialAlgorithmSpec) GetTrainTrialSpec() *NasJobSpec_MultiTrialAlgorithmSpec_TrainTrialSpec {
	if x != nil {
		return x.TrainTrialSpec
	}
	return nil
}

// Represents a metric to optimize.
type NasJobSpec_MultiTrialAlgorithmSpec_MetricSpec struct {
	state protoimpl.MessageState `protogen:"open.v1"`
	// Required. The ID of the metric. Must not contain whitespaces.
	MetricId string `protobuf:"bytes,1,opt,name=metric_id,json=metricId,proto3" json:"metric_id,omitempty"`
	// Required. The optimization goal of the metric.
	Goal          NasJobSpec_MultiTrialAlgorithmSpec_MetricSpec_GoalType `protobuf:"varint,2,opt,name=goal,proto3,enum=google.cloud.aiplatform.v1.NasJobSpec_MultiTrialAlgorithmSpec_MetricSpec_GoalType" json:"goal,omitempty"`
	unknownFields protoimpl.UnknownFields
	sizeCache     protoimpl.SizeCache
}

func (x *NasJobSpec_MultiTrialAlgorithmSpec_MetricSpec) Reset() {
	*x = NasJobSpec_MultiTrialAlgorithmSpec_MetricSpec{}
	mi := &file_google_cloud_aiplatform_v1_nas_job_proto_msgTypes[7]
	ms := protoimpl.X.MessageStateOf(protoimpl.Pointer(x))
	ms.StoreMessageInfo(mi)
}

func (x *NasJobSpec_MultiTrialAlgorithmSpec_MetricSpec) String() string {
	return protoimpl.X.MessageStringOf(x)
}

func (*NasJobSpec_MultiTrialAlgorithmSpec_MetricSpec) ProtoMessage() {}

func (x *NasJobSpec_MultiTrialAlgorithmSpec_MetricSpec) ProtoReflect() protoreflect.Message {
	mi := &file_google_cloud_aiplatform_v1_nas_job_proto_msgTypes[7]
	if x != nil {
		ms := protoimpl.X.MessageStateOf(protoimpl.Pointer(x))
		if ms.LoadMessageInfo() == nil {
			ms.StoreMessageInfo(mi)
		}
		return ms
	}
	return mi.MessageOf(x)
}

// Deprecated: Use NasJobSpec_MultiTrialAlgorithmSpec_MetricSpec.ProtoReflect.Descriptor instead.
func (*NasJobSpec_MultiTrialAlgorithmSpec_MetricSpec) Descriptor() ([]byte, []int) {
	return file_google_cloud_aiplatform_v1_nas_job_proto_rawDescGZIP(), []int{2, 0, 0}
}

func (x *NasJobSpec_MultiTrialAlgorithmSpec_MetricSpec) GetMetricId() string {
	if x != nil {
		return x.MetricId
	}
	return ""
}

func (x *NasJobSpec_MultiTrialAlgorithmSpec_MetricSpec) GetGoal() NasJobSpec_MultiTrialAlgorithmSpec_MetricSpec_GoalType {
	if x != nil {
		return x.Goal
	}
	return NasJobSpec_MultiTrialAlgorithmSpec_MetricSpec_GOAL_TYPE_UNSPECIFIED
}

// Represent spec for search trials.
type NasJobSpec_MultiTrialAlgorithmSpec_SearchTrialSpec struct {
	state protoimpl.MessageState `protogen:"open.v1"`
	// Required. The spec of a search trial job. The same spec applies to
	// all search trials.
	SearchTrialJobSpec *CustomJobSpec `protobuf:"bytes,1,opt,name=search_trial_job_spec,json=searchTrialJobSpec,proto3" json:"search_trial_job_spec,omitempty"`
	// Required. The maximum number of Neural Architecture Search (NAS) trials
	// to run.
	MaxTrialCount int32 `protobuf:"varint,2,opt,name=max_trial_count,json=maxTrialCount,proto3" json:"max_trial_count,omitempty"`
	// Required. The maximum number of trials to run in parallel.
	MaxParallelTrialCount int32 `protobuf:"varint,3,opt,name=max_parallel_trial_count,json=maxParallelTrialCount,proto3" json:"max_parallel_trial_count,omitempty"`
	// The number of failed trials that need to be seen before failing
	// the NasJob.
	//
	// If set to 0, Vertex AI decides how many trials must fail
	// before the whole job fails.
	MaxFailedTrialCount int32 `protobuf:"varint,4,opt,name=max_failed_trial_count,json=maxFailedTrialCount,proto3" json:"max_failed_trial_count,omitempty"`
	unknownFields       protoimpl.UnknownFields
	sizeCache           protoimpl.SizeCache
}

func (x *NasJobSpec_MultiTrialAlgorithmSpec_SearchTrialSpec) Reset() {
	*x = NasJobSpec_MultiTrialAlgorithmSpec_SearchTrialSpec{}
	mi := &file_google_cloud_aiplatform_v1_nas_job_proto_msgTypes[8]
	ms := protoimpl.X.MessageStateOf(protoimpl.Pointer(x))
	ms.StoreMessageInfo(mi)
}

func (x *NasJobSpec_MultiTrialAlgorithmSpec_SearchTrialSpec) String() string {
	return protoimpl.X.MessageStringOf(x)
}

func (*NasJobSpec_MultiTrialAlgorithmSpec_SearchTrialSpec) ProtoMessage() {}

func (x *NasJobSpec_MultiTrialAlgorithmSpec_SearchTrialSpec) ProtoReflect() protoreflect.Message {
	mi := &file_google_cloud_aiplatform_v1_nas_job_proto_msgTypes[8]
	if x != nil {
		ms := protoimpl.X.MessageStateOf(protoimpl.Pointer(x))
		if ms.LoadMessageInfo() == nil {
			ms.StoreMessageInfo(mi)
		}
		return ms
	}
	return mi.MessageOf(x)
}

// Deprecated: Use NasJobSpec_MultiTrialAlgorithmSpec_SearchTrialSpec.ProtoReflect.Descriptor instead.
func (*NasJobSpec_MultiTrialAlgorithmSpec_SearchTrialSpec) Descriptor() ([]byte, []int) {
	return file_google_cloud_aiplatform_v1_nas_job_proto_rawDescGZIP(), []int{2, 0, 1}
}

func (x *NasJobSpec_MultiTrialAlgorithmSpec_SearchTrialSpec) GetSearchTrialJobSpec() *CustomJobSpec {
	if x != nil {
		return x.SearchTrialJobSpec
	}
	return nil
}

func (x *NasJobSpec_MultiTrialAlgorithmSpec_SearchTrialSpec) GetMaxTrialCount() int32 {
	if x != nil {
		return x.MaxTrialCount
	}
	return 0
}

func (x *NasJobSpec_MultiTrialAlgorithmSpec_SearchTrialSpec) GetMaxParallelTrialCount() int32 {
	if x != nil {
		return x.MaxParallelTrialCount
	}
	return 0
}

func (x *NasJobSpec_MultiTrialAlgorithmSpec_SearchTrialSpec) GetMaxFailedTrialCount() int32 {
	if x != nil {
		return x.MaxFailedTrialCount
	}
	return 0
}

// Represent spec for train trials.
type NasJobSpec_MultiTrialAlgorithmSpec_TrainTrialSpec struct {
	state protoimpl.MessageState `protogen:"open.v1"`
	// Required. The spec of a train trial job. The same spec applies to
	// all train trials.
	TrainTrialJobSpec *CustomJobSpec `protobuf:"bytes,1,opt,name=train_trial_job_spec,json=trainTrialJobSpec,proto3" json:"train_trial_job_spec,omitempty"`
	// Required. The maximum number of trials to run in parallel.
	MaxParallelTrialCount int32 `protobuf:"varint,2,opt,name=max_parallel_trial_count,json=maxParallelTrialCount,proto3" json:"max_parallel_trial_count,omitempty"`
	// Required. Frequency of search trials to start train stage. Top N
	// [TrainTrialSpec.max_parallel_trial_count]
	// search trials will be trained for every M
	// [TrainTrialSpec.frequency] trials searched.
	Frequency     int32 `protobuf:"varint,3,opt,name=frequency,proto3" json:"frequency,omitempty"`
	unknownFields protoimpl.UnknownFields
	sizeCache     protoimpl.SizeCache
}

func (x *NasJobSpec_MultiTrialAlgorithmSpec_TrainTrialSpec) Reset() {
	*x = NasJobSpec_MultiTrialAlgorithmSpec_TrainTrialSpec{}
	mi := &file_google_cloud_aiplatform_v1_nas_job_proto_msgTypes[9]
	ms := protoimpl.X.MessageStateOf(protoimpl.Pointer(x))
	ms.StoreMessageInfo(mi)
}

func (x *NasJobSpec_MultiTrialAlgorithmSpec_TrainTrialSpec) String() string {
	return protoimpl.X.MessageStringOf(x)
}

func (*NasJobSpec_MultiTrialAlgorithmSpec_TrainTrialSpec) ProtoMessage() {}

func (x *NasJobSpec_MultiTrialAlgorithmSpec_TrainTrialSpec) ProtoReflect() protoreflect.Message {
	mi := &file_google_cloud_aiplatform_v1_nas_job_proto_msgTypes[9]
	if x != nil {
		ms := protoimpl.X.MessageStateOf(protoimpl.Pointer(x))
		if ms.LoadMessageInfo() == nil {
			ms.StoreMessageInfo(mi)
		}
		return ms
	}
	return mi.MessageOf(x)
}

// Deprecated: Use NasJobSpec_MultiTrialAlgorithmSpec_TrainTrialSpec.ProtoReflect.Descriptor instead.
func (*NasJobSpec_MultiTrialAlgorithmSpec_TrainTrialSpec) Descriptor() ([]byte, []int) {
	return file_google_cloud_aiplatform_v1_nas_job_proto_rawDescGZIP(), []int{2, 0, 2}
}

func (x *NasJobSpec_MultiTrialAlgorithmSpec_TrainTrialSpec) GetTrainTrialJobSpec() *CustomJobSpec {
	if x != nil {
		return x.TrainTrialJobSpec
	}
	return nil
}

func (x *NasJobSpec_MultiTrialAlgorithmSpec_TrainTrialSpec) GetMaxParallelTrialCount() int32 {
	if x != nil {
		return x.MaxParallelTrialCount
	}
	return 0
}

func (x *NasJobSpec_MultiTrialAlgorithmSpec_TrainTrialSpec) GetFrequency() int32 {
	if x != nil {
		return x.Frequency
	}
	return 0
}

// The output of a multi-trial Neural Architecture Search (NAS) jobs.
type NasJobOutput_MultiTrialJobOutput struct {
	state protoimpl.MessageState `protogen:"open.v1"`
	// Output only. List of NasTrials that were started as part of search stage.
	SearchTrials []*NasTrial `protobuf:"bytes,1,rep,name=search_trials,json=searchTrials,proto3" json:"search_trials,omitempty"`
	// Output only. List of NasTrials that were started as part of train stage.
	TrainTrials   []*NasTrial `protobuf:"bytes,2,rep,name=train_trials,json=trainTrials,proto3" json:"train_trials,omitempty"`
	unknownFields protoimpl.UnknownFields
	sizeCache     protoimpl.SizeCache
}

func (x *NasJobOutput_MultiTrialJobOutput) Reset() {
	*x = NasJobOutput_MultiTrialJobOutput{}
	mi := &file_google_cloud_aiplatform_v1_nas_job_proto_msgTypes[10]
	ms := protoimpl.X.MessageStateOf(protoimpl.Pointer(x))
	ms.StoreMessageInfo(mi)
}

func (x *NasJobOutput_MultiTrialJobOutput) String() string {
	return protoimpl.X.MessageStringOf(x)
}

func (*NasJobOutput_MultiTrialJobOutput) ProtoMessage() {}

func (x *NasJobOutput_MultiTrialJobOutput) ProtoReflect() protoreflect.Message {
	mi := &file_google_cloud_aiplatform_v1_nas_job_proto_msgTypes[10]
	if x != nil {
		ms := protoimpl.X.MessageStateOf(protoimpl.Pointer(x))
		if ms.LoadMessageInfo() == nil {
			ms.StoreMessageInfo(mi)
		}
		return ms
	}
	return mi.MessageOf(x)
}

// Deprecated: Use NasJobOutput_MultiTrialJobOutput.ProtoReflect.Descriptor instead.
func (*NasJobOutput_MultiTrialJobOutput) Descriptor() ([]byte, []int) {
	return file_google_cloud_aiplatform_v1_nas_job_proto_rawDescGZIP(), []int{3, 0}
}

func (x *NasJobOutput_MultiTrialJobOutput) GetSearchTrials() []*NasTrial {
	if x != nil {
		return x.SearchTrials
	}
	return nil
}

func (x *NasJobOutput_MultiTrialJobOutput) GetTrainTrials() []*NasTrial {
	if x != nil {
		return x.TrainTrials
	}
	return nil
}

var File_google_cloud_aiplatform_v1_nas_job_proto protoreflect.FileDescriptor

const file_google_cloud_aiplatform_v1_nas_job_proto_rawDesc = "" +
	"\n" +
	"(google/cloud/aiplatform/v1/nas_job.proto\x12\x1agoogle.cloud.aiplatform.v1\x1a\x1fgoogle/api/field_behavior.proto\x1a\x19google/api/resource.proto\x1a+google/cloud/aiplatform/v1/custom_job.proto\x1a0google/cloud/aiplatform/v1/encryption_spec.proto\x1a*google/cloud/aiplatform/v1/job_state.proto\x1a&google/cloud/aiplatform/v1/study.proto\x1a\x1fgoogle/protobuf/timestamp.proto\x1a\x17google/rpc/status.proto\"\xbb\b\n" +
	"\x06NasJob\x12\x17\n" +
	"\x04name\x18\x01 \x01(\tB\x03\xe0A\x03R\x04name\x12&\n" +
	"\fdisplay_name\x18\x02 \x01(\tB\x03\xe0A\x02R\vdisplayName\x12M\n" +
	"\fnas_job_spec\x18\x04 \x01(\v2&.google.cloud.aiplatform.v1.NasJobSpecB\x03\xe0A\x02R\n" +
	"nasJobSpec\x12S\n" +
	"\x0enas_job_output\x18\x05 \x01(\v2(.google.cloud.aiplatform.v1.NasJobOutputB\x03\xe0A\x03R\fnasJobOutput\x12?\n" +
	"\x05state\x18\x06 \x01(\x0e2$.google.cloud.aiplatform.v1.JobStateB\x03\xe0A\x03R\x05state\x12@\n" +
	"\vcreate_time\x18\a \x01(\v2\x1a.google.protobuf.TimestampB\x03\xe0A\x03R\n" +
	"createTime\x12>\n" +
	"\n" +
	"start_time\x18\b \x01(\v2\x1a.google.protobuf.TimestampB\x03\xe0A\x03R\tstartTime\x12:\n" +
	"\bend_time\x18\t \x01(\v2\x1a.google.protobuf.TimestampB\x03\xe0A\x03R\aendTime\x12@\n" +
	"\vupdate_time\x18\n" +
	" \x01(\v2\x1a.google.protobuf.TimestampB\x03\xe0A\x03R\n" +
	"updateTime\x12-\n" +
	"\x05error\x18\v \x01(\v2\x12.google.rpc.StatusB\x03\xe0A\x03R\x05error\x12F\n" +
	"\x06labels\x18\f \x03(\v2..google.cloud.aiplatform.v1.NasJob.LabelsEntryR\x06labels\x12S\n" +
	"\x0fencryption_spec\x18\r \x01(\v2*.google.cloud.aiplatform.v1.EncryptionSpecR\x0eencryptionSpec\x12N\n" +
	" enable_restricted_image_training\x18\x0e \x01(\bB\x05\xe0A\x01\x18\x01R\x1denableRestrictedImageTraining\x12(\n" +
	"\rsatisfies_pzs\x18\x0f \x01(\bB\x03\xe0A\x03R\fsatisfiesPzs\x12(\n" +
	"\rsatisfies_pzi\x18\x10 \x01(\bB\x03\xe0A\x03R\fsatisfiesPzi\x1a9\n" +
	"\vLabelsEntry\x12\x10\n" +
	"\x03key\x18\x01 \x01(\tR\x03key\x12\x14\n" +
	"\x05value\x18\x02 \x01(\tR\x05value:\x028\x01:`\xeaA]\n" +
	" aiplatform.googleapis.com/NasJob\x129projects/{project}/locations/{location}/nasJobs/{nas_job}\"\xe8\x02\n" +
	"\x0eNasTrialDetail\x12\x17\n" +
	"\x04name\x18\x01 \x01(\tB\x03\xe0A\x03R\x04name\x12\x1e\n" +
	"\n" +
	"parameters\x18\x02 \x01(\tR\n" +
	"parameters\x12G\n" +
	"\fsearch_trial\x18\x03 \x01(\v2$.google.cloud.aiplatform.v1.NasTrialR\vsearchTrial\x12E\n" +
	"\vtrain_trial\x18\x04 \x01(\v2$.google.cloud.aiplatform.v1.NasTrialR\n" +
	"trainTrial:\x8c\x01\xeaA\x88\x01\n" +
	"(aiplatform.googleapis.com/NasTrialDetail\x12\\projects/{project}/locations/{location}/nasJobs/{nas_job}/nasTrialDetails/{nas_trial_detail}\"\xb2\f\n" +
	"\n" +
	"NasJobSpec\x12}\n" +
	"\x1amulti_trial_algorithm_spec\x18\x02 \x01(\v2>.google.cloud.aiplatform.v1.NasJobSpec.MultiTrialAlgorithmSpecH\x00R\x17multiTrialAlgorithmSpec\x12)\n" +
	"\x11resume_nas_job_id\x18\x03 \x01(\tR\x0eresumeNasJobId\x12*\n" +
	"\x11search_space_spec\x18\x01 \x01(\tR\x0fsearchSpaceSpec\x1a\xb7\n" +
	"\n" +
	"\x17MultiTrialAlgorithmSpec\x12\x86\x01\n" +
	"\x15multi_trial_algorithm\x18\x01 \x01(\x0e2R.google.cloud.aiplatform.v1.NasJobSpec.MultiTrialAlgorithmSpec.MultiTrialAlgorithmR\x13multiTrialAlgorithm\x12a\n" +
	"\x06metric\x18\x02 \x01(\v2I.google.cloud.aiplatform.v1.NasJobSpec.MultiTrialAlgorithmSpec.MetricSpecR\x06metric\x12\x7f\n" +
	"\x11search_trial_spec\x18\x03 \x01(\v2N.google.cloud.aiplatform.v1.NasJobSpec.MultiTrialAlgorithmSpec.SearchTrialSpecB\x03\xe0A\x02R\x0fsearchTrialSpec\x12w\n" +
	"\x10train_trial_spec\x18\x04 \x01(\v2M.google.cloud.aiplatform.v1.NasJobSpec.MultiTrialAlgorithmSpec.TrainTrialSpecR\x0etrainTrialSpec\x1a\xde\x01\n" +
	"\n" +
	"MetricSpec\x12 \n" +
	"\tmetric_id\x18\x01 \x01(\tB\x03\xe0A\x02R\bmetricId\x12k\n" +
	"\x04goal\x18\x02 \x01(\x0e2R.google.cloud.aiplatform.v1.NasJobSpec.MultiTrialAlgorithmSpec.MetricSpec.GoalTypeB\x03\xe0A\x02R\x04goal\"A\n" +
	"\bGoalType\x12\x19\n" +
	"\x15GOAL_TYPE_UNSPECIFIED\x10\x00\x12\f\n" +
	"\bMAXIMIZE\x10\x01\x12\f\n" +
	"\bMINIMIZE\x10\x02\x1a\x94\x02\n" +
	"\x0fSearchTrialSpec\x12a\n" +
	"\x15search_trial_job_spec\x18\x01 \x01(\v2).google.cloud.aiplatform.v1.CustomJobSpecB\x03\xe0A\x02R\x12searchTrialJobSpec\x12+\n" +
	"\x0fmax_trial_count\x18\x02 \x01(\x05B\x03\xe0A\x02R\rmaxTrialCount\x12<\n" +
	"\x18max_parallel_trial_count\x18\x03 \x01(\x05B\x03\xe0A\x02R\x15maxParallelTrialCount\x123\n" +
	"\x16max_failed_trial_count\x18\x04 \x01(\x05R\x13maxFailedTrialCount\x1a\xd2\x01\n" +
	"\x0eTrainTrialSpec\x12_\n" +
	"\x14train_trial_job_spec\x18\x01 \x01(\v2).google.cloud.aiplatform.v1.CustomJobSpecB\x03\xe0A\x02R\x11trainTrialJobSpec\x12<\n" +
	"\x18max_parallel_trial_count\x18\x02 \x01(\x05B\x03\xe0A\x02R\x15maxParallelTrialCount\x12!\n" +
	"\tfrequency\x18\x03 \x01(\x05B\x03\xe0A\x02R\tfrequency\"i\n" +
	"\x13MultiTrialAlgorithm\x12%\n" +
	"!MULTI_TRIAL_ALGORITHM_UNSPECIFIED\x10\x00\x12\x1a\n" +
	"\x16REINFORCEMENT_LEARNING\x10\x01\x12\x0f\n" +
	"\vGRID_SEARCH\x10\x02B\x14\n" +
	"\x12nas_algorithm_spec\"\xc8\x02\n" +
	"\fNasJobOutput\x12x\n" +
	"\x16multi_trial_job_output\x18\x01 \x01(\v2<.google.cloud.aiplatform.v1.NasJobOutput.MultiTrialJobOutputB\x03\xe0A\x03H\x00R\x13multiTrialJobOutput\x1a\xb3\x01\n" +
	"\x13MultiTrialJobOutput\x12N\n" +
	"\rsearch_trials\x18\x01 \x03(\v2$.google.cloud.aiplatform.v1.NasTrialB\x03\xe0A\x03R\fsearchTrials\x12L\n" +
	"\ftrain_trials\x18\x02 \x03(\v2$.google.cloud.aiplatform.v1.NasTrialB\x03\xe0A\x03R\vtrainTrialsB\b\n" +
	"\x06output\"\xa5\x03\n" +
	"\bNasTrial\x12\x13\n" +
	"\x02id\x18\x01 \x01(\tB\x03\xe0A\x03R\x02id\x12E\n" +
	"\x05state\x18\x02 \x01(\x0e2*.google.cloud.aiplatform.v1.NasTrial.StateB\x03\xe0A\x03R\x05state\x12Y\n" +
	"\x11final_measurement\x18\x03 \x01(\v2'.google.cloud.aiplatform.v1.MeasurementB\x03\xe0A\x03R\x10finalMeasurement\x12>\n" +
	"\n" +
	"start_time\x18\x04 \x01(\v2\x1a.google.protobuf.TimestampB\x03\xe0A\x03R\tstartTime\x12:\n" +
	"\bend_time\x18\x05 \x01(\v2\x1a.google.protobuf.TimestampB\x03\xe0A\x03R\aendTime\"f\n" +
	"\x05State\x12\x15\n" +
	"\x11STATE_UNSPECIFIED\x10\x00\x12\r\n" +
	"\tREQUESTED\x10\x01\x12\n" +
	"\n" +
	"\x06ACTIVE\x10\x02\x12\f\n" +
	"\bSTOPPING\x10\x03\x12\r\n" +
	"\tSUCCEEDED\x10\x04\x12\x0e\n" +
	"\n" +
	"INFEASIBLE\x10\x05B\xc9\x01\n" +
	"\x1ecom.google.cloud.aiplatform.v1B\vNasJobProtoP\x01Z>cloud.google.com/go/aiplatform/apiv1/aiplatformpb;aiplatformpb\xaa\x02\x1aGoogle.Cloud.AIPlatform.V1\xca\x02\x1aGoogle\\Cloud\\AIPlatform\\V1\xea\x02\x1dGoogle::Cloud::AIPlatform::V1b\x06proto3"

var (
	file_google_cloud_aiplatform_v1_nas_job_proto_rawDescOnce sync.Once
	file_google_cloud_aiplatform_v1_nas_job_proto_rawDescData []byte
)

func file_google_cloud_aiplatform_v1_nas_job_proto_rawDescGZIP() []byte {
	file_google_cloud_aiplatform_v1_nas_job_proto_rawDescOnce.Do(func() {
		file_google_cloud_aiplatform_v1_nas_job_proto_rawDescData = protoimpl.X.CompressGZIP(unsafe.Slice(unsafe.StringData(file_google_cloud_aiplatform_v1_nas_job_proto_rawDesc), len(file_google_cloud_aiplatform_v1_nas_job_proto_rawDesc)))
	})
	return file_google_cloud_aiplatform_v1_nas_job_proto_rawDescData
}

var file_google_cloud_aiplatform_v1_nas_job_proto_enumTypes = make([]protoimpl.EnumInfo, 3)
var file_google_cloud_aiplatform_v1_nas_job_proto_msgTypes = make([]protoimpl.MessageInfo, 11)
var file_google_cloud_aiplatform_v1_nas_job_proto_goTypes = []any{
	(NasJobSpec_MultiTrialAlgorithmSpec_MultiTrialAlgorithm)(0), // 0: google.cloud.aiplatform.v1.NasJobSpec.MultiTrialAlgorithmSpec.MultiTrialAlgorithm
	(NasJobSpec_MultiTrialAlgorithmSpec_MetricSpec_GoalType)(0), // 1: google.cloud.aiplatform.v1.NasJobSpec.MultiTrialAlgorithmSpec.MetricSpec.GoalType
	(NasTrial_State)(0),    // 2: google.cloud.aiplatform.v1.NasTrial.State
	(*NasJob)(nil),         // 3: google.cloud.aiplatform.v1.NasJob
	(*NasTrialDetail)(nil), // 4: google.cloud.aiplatform.v1.NasTrialDetail
	(*NasJobSpec)(nil),     // 5: google.cloud.aiplatform.v1.NasJobSpec
	(*NasJobOutput)(nil),   // 6: google.cloud.aiplatform.v1.NasJobOutput
	(*NasTrial)(nil),       // 7: google.cloud.aiplatform.v1.NasTrial
	nil,                    // 8: google.cloud.aiplatform.v1.NasJob.LabelsEntry
	(*NasJobSpec_MultiTrialAlgorithmSpec)(nil),                 // 9: google.cloud.aiplatform.v1.NasJobSpec.MultiTrialAlgorithmSpec
	(*NasJobSpec_MultiTrialAlgorithmSpec_MetricSpec)(nil),      // 10: google.cloud.aiplatform.v1.NasJobSpec.MultiTrialAlgorithmSpec.MetricSpec
	(*NasJobSpec_MultiTrialAlgorithmSpec_SearchTrialSpec)(nil), // 11: google.cloud.aiplatform.v1.NasJobSpec.MultiTrialAlgorithmSpec.SearchTrialSpec
	(*NasJobSpec_MultiTrialAlgorithmSpec_TrainTrialSpec)(nil),  // 12: google.cloud.aiplatform.v1.NasJobSpec.MultiTrialAlgorithmSpec.TrainTrialSpec
	(*NasJobOutput_MultiTrialJobOutput)(nil),                   // 13: google.cloud.aiplatform.v1.NasJobOutput.MultiTrialJobOutput
	(JobState)(0),                                              // 14: google.cloud.aiplatform.v1.JobState
	(*timestamppb.Timestamp)(nil),                              // 15: google.protobuf.Timestamp
	(*status.Status)(nil),                                      // 16: google.rpc.Status
	(*EncryptionSpec)(nil),                                     // 17: google.cloud.aiplatform.v1.EncryptionSpec
	(*Measurement)(nil),                                        // 18: google.cloud.aiplatform.v1.Measurement
	(*CustomJobSpec)(nil),                                      // 19: google.cloud.aiplatform.v1.CustomJobSpec
}
var file_google_cloud_aiplatform_v1_nas_job_proto_depIdxs = []int32{
	5,  // 0: google.cloud.aiplatform.v1.NasJob.nas_job_spec:type_name -> google.cloud.aiplatform.v1.NasJobSpec
	6,  // 1: google.cloud.aiplatform.v1.NasJob.nas_job_output:type_name -> google.cloud.aiplatform.v1.NasJobOutput
	14, // 2: google.cloud.aiplatform.v1.NasJob.state:type_name -> google.cloud.aiplatform.v1.JobState
	15, // 3: google.cloud.aiplatform.v1.NasJob.create_time:type_name -> google.protobuf.Timestamp
	15, // 4: google.cloud.aiplatform.v1.NasJob.start_time:type_name -> google.protobuf.Timestamp
	15, // 5: google.cloud.aiplatform.v1.NasJob.end_time:type_name -> google.protobuf.Timestamp
	15, // 6: google.cloud.aiplatform.v1.NasJob.update_time:type_name -> google.protobuf.Timestamp
	16, // 7: google.cloud.aiplatform.v1.NasJob.error:type_name -> google.rpc.Status
	8,  // 8: google.cloud.aiplatform.v1.NasJob.labels:type_name -> google.cloud.aiplatform.v1.NasJob.LabelsEntry
	17, // 9: google.cloud.aiplatform.v1.NasJob.encryption_spec:type_name -> google.cloud.aiplatform.v1.EncryptionSpec
	7,  // 10: google.cloud.aiplatform.v1.NasTrialDetail.search_trial:type_name -> google.cloud.aiplatform.v1.NasTrial
	7,  // 11: google.cloud.aiplatform.v1.NasTrialDetail.train_trial:type_name -> google.cloud.aiplatform.v1.NasTrial
	9,  // 12: google.cloud.aiplatform.v1.NasJobSpec.multi_trial_algorithm_spec:type_name -> google.cloud.aiplatform.v1.NasJobSpec.MultiTrialAlgorithmSpec
	13, // 13: google.cloud.aiplatform.v1.NasJobOutput.multi_trial_job_output:type_name -> google.cloud.aiplatform.v1.NasJobOutput.MultiTrialJobOutput
	2,  // 14: google.cloud.aiplatform.v1.NasTrial.state:type_name -> google.cloud.aiplatform.v1.NasTrial.State
	18, // 15: google.cloud.aiplatform.v1.NasTrial.final_measurement:type_name -> google.cloud.aiplatform.v1.Measurement
	15, // 16: google.cloud.aiplatform.v1.NasTrial.start_time:type_name -> google.protobuf.Timestamp
	15, // 17: google.cloud.aiplatform.v1.NasTrial.end_time:type_name -> google.protobuf.Timestamp
	0,  // 18: google.cloud.aiplatform.v1.NasJobSpec.MultiTrialAlgorithmSpec.multi_trial_algorithm:type_name -> google.cloud.aiplatform.v1.NasJobSpec.MultiTrialAlgorithmSpec.MultiTrialAlgorithm
	10, // 19: google.cloud.aiplatform.v1.NasJobSpec.MultiTrialAlgorithmSpec.metric:type_name -> google.cloud.aiplatform.v1.NasJobSpec.MultiTrialAlgorithmSpec.MetricSpec
	11, // 20: google.cloud.aiplatform.v1.NasJobSpec.MultiTrialAlgorithmSpec.search_trial_spec:type_name -> google.cloud.aiplatform.v1.NasJobSpec.MultiTrialAlgorithmSpec.SearchTrialSpec
	12, // 21: google.cloud.aiplatform.v1.NasJobSpec.MultiTrialAlgorithmSpec.train_trial_spec:type_name -> google.cloud.aiplatform.v1.NasJobSpec.MultiTrialAlgorithmSpec.TrainTrialSpec
	1,  // 22: google.cloud.aiplatform.v1.NasJobSpec.MultiTrialAlgorithmSpec.MetricSpec.goal:type_name -> google.cloud.aiplatform.v1.NasJobSpec.MultiTrialAlgorithmSpec.MetricSpec.GoalType
	19, // 23: google.cloud.aiplatform.v1.NasJobSpec.MultiTrialAlgorithmSpec.SearchTrialSpec.search_trial_job_spec:type_name -> google.cloud.aiplatform.v1.CustomJobSpec
	19, // 24: google.cloud.aiplatform.v1.NasJobSpec.MultiTrialAlgorithmSpec.TrainTrialSpec.train_trial_job_spec:type_name -> google.cloud.aiplatform.v1.CustomJobSpec
	7,  // 25: google.cloud.aiplatform.v1.NasJobOutput.MultiTrialJobOutput.search_trials:type_name -> google.cloud.aiplatform.v1.NasTrial
	7,  // 26: google.cloud.aiplatform.v1.NasJobOutput.MultiTrialJobOutput.train_trials:type_name -> google.cloud.aiplatform.v1.NasTrial
	27, // [27:27] is the sub-list for method output_type
	27, // [27:27] is the sub-list for method input_type
	27, // [27:27] is the sub-list for extension type_name
	27, // [27:27] is the sub-list for extension extendee
	0,  // [0:27] is the sub-list for field type_name
}

func init() { file_google_cloud_aiplatform_v1_nas_job_proto_init() }
func file_google_cloud_aiplatform_v1_nas_job_proto_init() {
	if File_google_cloud_aiplatform_v1_nas_job_proto != nil {
		return
	}
	file_google_cloud_aiplatform_v1_custom_job_proto_init()
	file_google_cloud_aiplatform_v1_encryption_spec_proto_init()
	file_google_cloud_aiplatform_v1_job_state_proto_init()
	file_google_cloud_aiplatform_v1_study_proto_init()
	file_google_cloud_aiplatform_v1_nas_job_proto_msgTypes[2].OneofWrappers = []any{
		(*NasJobSpec_MultiTrialAlgorithmSpec_)(nil),
	}
	file_google_cloud_aiplatform_v1_nas_job_proto_msgTypes[3].OneofWrappers = []any{
		(*NasJobOutput_MultiTrialJobOutput_)(nil),
	}
	type x struct{}
	out := protoimpl.TypeBuilder{
		File: protoimpl.DescBuilder{
			GoPackagePath: reflect.TypeOf(x{}).PkgPath(),
			RawDescriptor: unsafe.Slice(unsafe.StringData(file_google_cloud_aiplatform_v1_nas_job_proto_rawDesc), len(file_google_cloud_aiplatform_v1_nas_job_proto_rawDesc)),
			NumEnums:      3,
			NumMessages:   11,
			NumExtensions: 0,
			NumServices:   0,
		},
		GoTypes:           file_google_cloud_aiplatform_v1_nas_job_proto_goTypes,
		DependencyIndexes: file_google_cloud_aiplatform_v1_nas_job_proto_depIdxs,
		EnumInfos:         file_google_cloud_aiplatform_v1_nas_job_proto_enumTypes,
		MessageInfos:      file_google_cloud_aiplatform_v1_nas_job_proto_msgTypes,
	}.Build()
	File_google_cloud_aiplatform_v1_nas_job_proto = out.File
	file_google_cloud_aiplatform_v1_nas_job_proto_goTypes = nil
	file_google_cloud_aiplatform_v1_nas_job_proto_depIdxs = nil
}
