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

Go to download
// *** WARNING: this file was generated by pulumi-java-gen. ***
// *** Do not edit by hand unless you're certain you know what you are doing! ***

package com.pulumi.azurenative.machinelearningservices.inputs;

import com.pulumi.core.Output;
import com.pulumi.core.annotations.Import;
import com.pulumi.core.internal.Codegen;
import com.pulumi.exceptions.MissingRequiredPropertyException;
import java.lang.Integer;
import java.lang.String;
import java.util.Objects;
import java.util.Optional;
import javax.annotation.Nullable;


/**
 * TensorFlow distribution configuration.
 * 
 */
public final class TensorFlowArgs extends com.pulumi.resources.ResourceArgs {

    public static final TensorFlowArgs Empty = new TensorFlowArgs();

    /**
     * Enum to determine the job distribution type.
     * Expected value is 'TensorFlow'.
     * 
     */
    @Import(name="distributionType", required=true)
    private Output distributionType;

    /**
     * @return Enum to determine the job distribution type.
     * Expected value is 'TensorFlow'.
     * 
     */
    public Output distributionType() {
        return this.distributionType;
    }

    /**
     * Number of parameter server tasks.
     * 
     */
    @Import(name="parameterServerCount")
    private @Nullable Output parameterServerCount;

    /**
     * @return Number of parameter server tasks.
     * 
     */
    public Optional> parameterServerCount() {
        return Optional.ofNullable(this.parameterServerCount);
    }

    /**
     * Number of workers. If not specified, will default to the instance count.
     * 
     */
    @Import(name="workerCount")
    private @Nullable Output workerCount;

    /**
     * @return Number of workers. If not specified, will default to the instance count.
     * 
     */
    public Optional> workerCount() {
        return Optional.ofNullable(this.workerCount);
    }

    private TensorFlowArgs() {}

    private TensorFlowArgs(TensorFlowArgs $) {
        this.distributionType = $.distributionType;
        this.parameterServerCount = $.parameterServerCount;
        this.workerCount = $.workerCount;
    }

    public static Builder builder() {
        return new Builder();
    }
    public static Builder builder(TensorFlowArgs defaults) {
        return new Builder(defaults);
    }

    public static final class Builder {
        private TensorFlowArgs $;

        public Builder() {
            $ = new TensorFlowArgs();
        }

        public Builder(TensorFlowArgs defaults) {
            $ = new TensorFlowArgs(Objects.requireNonNull(defaults));
        }

        /**
         * @param distributionType Enum to determine the job distribution type.
         * Expected value is 'TensorFlow'.
         * 
         * @return builder
         * 
         */
        public Builder distributionType(Output distributionType) {
            $.distributionType = distributionType;
            return this;
        }

        /**
         * @param distributionType Enum to determine the job distribution type.
         * Expected value is 'TensorFlow'.
         * 
         * @return builder
         * 
         */
        public Builder distributionType(String distributionType) {
            return distributionType(Output.of(distributionType));
        }

        /**
         * @param parameterServerCount Number of parameter server tasks.
         * 
         * @return builder
         * 
         */
        public Builder parameterServerCount(@Nullable Output parameterServerCount) {
            $.parameterServerCount = parameterServerCount;
            return this;
        }

        /**
         * @param parameterServerCount Number of parameter server tasks.
         * 
         * @return builder
         * 
         */
        public Builder parameterServerCount(Integer parameterServerCount) {
            return parameterServerCount(Output.of(parameterServerCount));
        }

        /**
         * @param workerCount Number of workers. If not specified, will default to the instance count.
         * 
         * @return builder
         * 
         */
        public Builder workerCount(@Nullable Output workerCount) {
            $.workerCount = workerCount;
            return this;
        }

        /**
         * @param workerCount Number of workers. If not specified, will default to the instance count.
         * 
         * @return builder
         * 
         */
        public Builder workerCount(Integer workerCount) {
            return workerCount(Output.of(workerCount));
        }

        public TensorFlowArgs build() {
            $.distributionType = Codegen.stringProp("distributionType").output().arg($.distributionType).require();
            $.parameterServerCount = Codegen.integerProp("parameterServerCount").output().arg($.parameterServerCount).def(0).getNullable();
            return $;
        }
    }

}