TensorFlowArgs

data class TensorFlowArgs(val distributionType: Output<String>, val parameterServerCount: Output<Int>? = null, val workerCount: Output<Int>? = null) : ConvertibleToJava<TensorFlowArgs>

TensorFlow distribution configuration.

Constructors

Link copied to clipboard
constructor(distributionType: Output<String>, parameterServerCount: Output<Int>? = null, workerCount: Output<Int>? = null)

Properties

Link copied to clipboard

Enum to determine the job distribution type. Expected value is 'TensorFlow'.

Link copied to clipboard
val parameterServerCount: Output<Int>? = null

Number of parameter server tasks.

Link copied to clipboard
val workerCount: Output<Int>? = null

Number of workers. Overwrites the node count in compute binding.

Functions

Link copied to clipboard
open override fun toJava(): TensorFlowArgs