4 instantiations of InputWithCustomShape
Microsoft.ML.OnnxTransformerTest (4)
OnnxTransformTests.cs (4)
802new InputWithCustomShape(){input = new float[] { 1.1f, 1.3f, 1.2f, 1.9f, 1.3f, 1.2f, 1.1f, 1.3f, 1.8f } }, 807new InputWithCustomShape(){input = new float[] { 0f, 0f, 1f, 1f, 0f, 0f, 1f, 0f, 0f } } 972new InputWithCustomShape(){input = new float[] { 1.1f, 1.3f, 1.2f, 1.9f, 1.3f, 1.2f, 1.1f, 1.3f, 1.8f } }, 977new InputWithCustomShape(){input = new float[] { 0f, 0f, 1f, 1f, 0f, 0f, 1f, 0f, 0f } }
7 references to InputWithCustomShape
Microsoft.ML.OnnxTransformerTest (7)
OnnxTransformTests.cs (7)
796var dataPoints = new InputWithCustomShape[] 810var shapeDictionary = new Dictionary<string, int[]>() { { nameof(InputWithCustomShape.input), new int[] { 3, 3 } } }; 822new[] { nameof(PredictionWithCustomShape.argmax) }, new[] { nameof(InputWithCustomShape.input) }, 829nameof(PredictionWithCustomShape.argmax), nameof(InputWithCustomShape.input), 966var dataPoints = new InputWithCustomShape[] 980var shapeDictionary = new Dictionary<string, int[]>() { { nameof(InputWithCustomShape.input), new int[] { 3, 3 } } }; 985nameof(InputWithCustomShape.input), modelFile, shapeDictionary, gpuDeviceId: _gpuDeviceId, fallbackToCpu: _fallbackToCpu);