TensorFlow.NET/test/TensorFlowNET.UnitTest/CApiTest.cs at master · rajeevyadav/TensorFlow.NET · GitHub

History

66 lines (54 loc) · 1.83 KB

Raw

using Microsoft.VisualStudio.TestTools.UnitTesting;

using Tensorflow;

namespace TensorFlowNET.UnitTest

{

public class CApiTest

{

protected TF_Code TF_OK = TF_Code.TF_OK;

protected TF_DataType TF_FLOAT = TF_DataType.TF_FLOAT;

protected void EXPECT_TRUE(bool expected)

{

Assert.IsTrue(expected);

}

protected void EXPECT_EQ(object expected, object actual)

{

Assert.AreEqual(expected, actual);

}

protected void ASSERT_EQ(object expected, object actual)

{

Assert.AreEqual(expected, actual);

}

protected void ASSERT_TRUE(bool condition)

{

Assert.IsTrue(condition);

}

protected OperationDescription TF_NewOperation(Graph graph, string opType, string opName)

{

return c_api.TF_NewOperation(graph, opType, opName);

}

protected void TF_AddInput(OperationDescription desc, TF_Output input)

{

c_api.TF_AddInput(desc, input);

}

protected Operation TF_FinishOperation(OperationDescription desc, Status s)

{

return c_api.TF_FinishOperation(desc, s);

}

protected void TF_SetAttrTensor(OperationDescription desc, string attrName, Tensor value, Status s)

{

c_api.TF_SetAttrTensor(desc, attrName, value, s);

}

protected void TF_SetAttrType(OperationDescription desc, string attrName, TF_DataType dtype)

{

c_api.TF_SetAttrType(desc, attrName, dtype);

}

protected void TF_SetAttrBool(OperationDescription desc, string attrName, bool value)

{

c_api.TF_SetAttrBool(desc, attrName, value);

}

protected TF_Code TF_GetCode(Status s)

{

return s.Code;

}

TensorFlow.NET/test/TensorFlowNET.UnitTest/CApiTest.cs at master · rajeevyadav/TensorFlow.NET · GitHub

Expand file tree

Breadcrumbs

CApiTest.cs

Latest commit

History

Breadcrumbs

CApiTest.cs

File metadata and controls