protected override void Run(ModelProto model)
{
// Should this not be before loading input? Is the abstract base really that good?
var dimParamOrValue = int.TryParse(Dim, out var dimValue)
? DimParamOrValue.New(dimValue)
: DimParamOrValue.New(Dim);
_console.WriteLine($"Setting dimension at {Index} to '{dimParamOrValue}'");
model.Graph.SetDim(Index, dimParamOrValue);
}
public void SetDim()
{
// Arrange
var model = ModelProto.Parser.ParseFrom(m_createStream);
// Act
model.Graph.SetDim(dimIndex: 0, DimParamOrValue.New("N"));
// Assert
var graph = model.Graph;
Assert.AreEqual(9, graph.Input.Count);
Assert.AreEqual(1, graph.Output.Count);
var expectedName = $"mnist-8-expected-{nameof(SetDim)}.onnx";
AssertModelBytesEqualToEmbeddedExpected(model, expectedName);
}
