-
Notifications
You must be signed in to change notification settings - Fork 4.5k
/
TensorFlowDistributionConfiguration.cs
35 lines (30 loc) · 1.62 KB
/
TensorFlowDistributionConfiguration.cs
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
// Copyright (c) Microsoft Corporation. All rights reserved.
// Licensed under the MIT License.
// <auto-generated/>
#nullable disable
namespace Azure.ResourceManager.MachineLearning.Models
{
/// <summary> TensorFlow distribution configuration. </summary>
public partial class TensorFlowDistributionConfiguration : MachineLearningDistributionConfiguration
{
/// <summary> Initializes a new instance of <see cref="TensorFlowDistributionConfiguration"/>. </summary>
public TensorFlowDistributionConfiguration()
{
DistributionType = DistributionType.TensorFlow;
}
/// <summary> Initializes a new instance of <see cref="TensorFlowDistributionConfiguration"/>. </summary>
/// <param name="distributionType"> [Required] Specifies the type of distribution framework. </param>
/// <param name="parameterServerCount"> Number of parameter server tasks. </param>
/// <param name="workerCount"> Number of workers. If not specified, will default to the instance count. </param>
internal TensorFlowDistributionConfiguration(DistributionType distributionType, int? parameterServerCount, int? workerCount) : base(distributionType)
{
ParameterServerCount = parameterServerCount;
WorkerCount = workerCount;
DistributionType = distributionType;
}
/// <summary> Number of parameter server tasks. </summary>
public int? ParameterServerCount { get; set; }
/// <summary> Number of workers. If not specified, will default to the instance count. </summary>
public int? WorkerCount { get; set; }
}
}