com.pulumi.azurenative.machinelearningservices.kotlin.inputs.TensorFlowArgs.kt Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of pulumi-azure-native-kotlin Show documentation

Build cloud applications and infrastructure by combining the safety and reliability of infrastructure as code with the power of the Kotlin programming language.

There is a newer version: 2.82.0.0

Show newest version

@file:Suppress("NAME_SHADOWING", "DEPRECATION")

package com.pulumi.azurenative.machinelearningservices.kotlin.inputs

import com.pulumi.azurenative.machinelearningservices.inputs.TensorFlowArgs.builder
import com.pulumi.core.Output
import com.pulumi.core.Output.of
import com.pulumi.kotlin.ConvertibleToJava
import com.pulumi.kotlin.PulumiNullFieldException
import com.pulumi.kotlin.PulumiTagMarker
import kotlin.Int
import kotlin.String
import kotlin.Suppress
import kotlin.jvm.JvmName

/**
 * TensorFlow distribution configuration.
 * @property distributionType Enum to determine the job distribution type.
 * Expected value is 'TensorFlow'.
 * @property parameterServerCount Number of parameter server tasks.
 * @property workerCount Number of workers. If not specified, will default to the instance count.
 */
public data class TensorFlowArgs(
    public val distributionType: Output,
    public val parameterServerCount: Output? = null,
    public val workerCount: Output? = null,
) : ConvertibleToJava {
    override fun toJava(): com.pulumi.azurenative.machinelearningservices.inputs.TensorFlowArgs =
        com.pulumi.azurenative.machinelearningservices.inputs.TensorFlowArgs.builder()
            .distributionType(distributionType.applyValue({ args0 -> args0 }))
            .parameterServerCount(parameterServerCount?.applyValue({ args0 -> args0 }))
            .workerCount(workerCount?.applyValue({ args0 -> args0 })).build()
}

/**
 * Builder for [TensorFlowArgs].
 */
@PulumiTagMarker
public class TensorFlowArgsBuilder internal constructor() {
    private var distributionType: Output? = null

    private var parameterServerCount: Output? = null

    private var workerCount: Output? = null

    /**
     * @param value Enum to determine the job distribution type.
     * Expected value is 'TensorFlow'.
     */
    @JvmName("htxbleeasyhrxmew")
    public suspend fun distributionType(`value`: Output) {
        this.distributionType = value
    }

    /**
     * @param value Number of parameter server tasks.
     */
    @JvmName("pcnrjxbkjokshpun")
    public suspend fun parameterServerCount(`value`: Output) {
        this.parameterServerCount = value
    }

    /**
     * @param value Number of workers. If not specified, will default to the instance count.
     */
    @JvmName("ranscwyfyvbewemk")
    public suspend fun workerCount(`value`: Output) {
        this.workerCount = value
    }

    /**
     * @param value Enum to determine the job distribution type.
     * Expected value is 'TensorFlow'.
     */
    @JvmName("oesmkurdbbcvtsqu")
    public suspend fun distributionType(`value`: String) {
        val toBeMapped = value
        val mapped = toBeMapped.let({ args0 -> of(args0) })
        this.distributionType = mapped
    }

    /**
     * @param value Number of parameter server tasks.
     */
    @JvmName("nusljljdldyydqck")
    public suspend fun parameterServerCount(`value`: Int?) {
        val toBeMapped = value
        val mapped = toBeMapped?.let({ args0 -> of(args0) })
        this.parameterServerCount = mapped
    }

    /**
     * @param value Number of workers. If not specified, will default to the instance count.
     */
    @JvmName("hammdmesgacmamgt")
    public suspend fun workerCount(`value`: Int?) {
        val toBeMapped = value
        val mapped = toBeMapped?.let({ args0 -> of(args0) })
        this.workerCount = mapped
    }

    internal fun build(): TensorFlowArgs = TensorFlowArgs(
        distributionType = distributionType ?: throw PulumiNullFieldException("distributionType"),
        parameterServerCount = parameterServerCount,
        workerCount = workerCount,
    )
}