Class: Google::Apis::MlV1::GoogleCloudMlV1ReplicaConfig

Inherits:
Object
  • Object
show all
Includes:
Core::Hashable, Core::JsonObjectSupport
Defined in:
generated/google/apis/ml_v1/classes.rb,
generated/google/apis/ml_v1/representations.rb,
generated/google/apis/ml_v1/representations.rb

Overview

Represents the configuration for a replica in a cluster.

Instance Attribute Summary collapse

Instance Method Summary collapse

Methods included from Core::JsonObjectSupport

#to_json

Methods included from Core::Hashable

process_value, #to_h

Constructor Details

#initialize(**args) ⇒ GoogleCloudMlV1ReplicaConfig

Returns a new instance of GoogleCloudMlV1ReplicaConfig



1209
1210
1211
# File 'generated/google/apis/ml_v1/classes.rb', line 1209

def initialize(**args)
   update!(**args)
end

Instance Attribute Details

#accelerator_configGoogle::Apis::MlV1::GoogleCloudMlV1AcceleratorConfig

Represents a hardware accelerator request config. Note that the AcceleratorConfig could be used in both Jobs and Versions. Corresponds to the JSON property acceleratorConfig



1182
1183
1184
# File 'generated/google/apis/ml_v1/classes.rb', line 1182

def accelerator_config
  @accelerator_config
end

#image_uriString

The Docker image to run on the replica. This image must be in Container Registry. Learn more about configuring custom containers. Corresponds to the JSON property imageUri

Returns:

  • (String)


1189
1190
1191
# File 'generated/google/apis/ml_v1/classes.rb', line 1189

def image_uri
  @image_uri
end

#tpu_tf_versionString

The AI Platform runtime version that includes a TensorFlow version matching the one used in the custom container. This field is required if the replica is a TPU worker that uses a custom container. Otherwise, do not specify this field. This must be a runtime version that currently supports training with TPUs. Note that the version of TensorFlow included in a runtime version may differ from the numbering of the runtime version itself, because it may have a different patch version. In this field, you must specify the runtime version (TensorFlow minor version). For example, if your custom container runs TensorFlow 1.x.y, specify 1.x. Corresponds to the JSON property tpuTfVersion

Returns:

  • (String)


1207
1208
1209
# File 'generated/google/apis/ml_v1/classes.rb', line 1207

def tpu_tf_version
  @tpu_tf_version
end

Instance Method Details

#update!(**args) ⇒ Object

Update properties of this object



1214
1215
1216
1217
1218
# File 'generated/google/apis/ml_v1/classes.rb', line 1214

def update!(**args)
  @accelerator_config = args[:accelerator_config] if args.key?(:accelerator_config)
  @image_uri = args[:image_uri] if args.key?(:image_uri)
  @tpu_tf_version = args[:tpu_tf_version] if args.key?(:tpu_tf_version)
end