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



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

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

Instance Attribute Details

#accelerator_configGoogle::Apis::MlV1::GoogleCloudMlV1AcceleratorConfig

Represents a hardware accelerator request config. Corresponds to the JSON property acceleratorConfig



1180
1181
1182
# File 'generated/google/apis/ml_v1/classes.rb', line 1180

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)


1187
1188
1189
# File 'generated/google/apis/ml_v1/classes.rb', line 1187

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)


1205
1206
1207
# File 'generated/google/apis/ml_v1/classes.rb', line 1205

def tpu_tf_version
  @tpu_tf_version
end

Instance Method Details

#update!(**args) ⇒ Object

Update properties of this object



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

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