TensorFlow.NET/src/TensorFlowNet.Benchmarks/Program.cs at master · lvyitian/TensorFlow.NET · GitHub

History

34 lines (31 loc) · 1005 Bytes

Raw

using BenchmarkDotNet.Configs;

using BenchmarkDotNet.Running;

using System;

using System.Reflection;

using Tensorflow.Benchmark.Leak;

namespace TensorFlowBenchmark

{

class Program

{

static void Main(string[] args)

{

new GpuLeakByCNN().Run();

if (args?.Length > 0)

{

for (int i = 0; i < args.Length; i++)

{

string name = $"TensorFlowBenchmark.{args[i]}";

var type = Type.GetType(name);

BenchmarkRunner.Run(type);

}

else

{

#pragma warning disable CS0618 // Type or member is obsolete

BenchmarkSwitcher.FromAssembly(Assembly.GetExecutingAssembly()).Run(args, ManualConfig.Create(DefaultConfig.Instance).With(ConfigOptions.DisableOptimizationsValidator));

#pragma warning restore CS0618 // Type or member is obsolete

}

Console.ReadLine();

}

TensorFlow.NET/src/TensorFlowNet.Benchmarks/Program.cs at master · lvyitian/TensorFlow.NET · GitHub

Expand file tree

Breadcrumbs

Program.cs

Latest commit

History

Breadcrumbs

Program.cs

File metadata and controls