TensorVariant.cpp File Reference

#include "mir/TensorVariant.h"
#include <gtest/gtest.h>

Go to the source code of this file.

Functions
	TEST (TensorVariant, BasicTest)
	TEST (TensorVariant, ElementSizeDeductionTest)

Function Documentation

TEST() [1/2]

TEST	(	TensorVariant	,
		BasicTest
	)

Definition at line 23 of file TensorVariant.cpp.

{
  Shape shape{2, 2};
  TensorVariant t(DataType::FLOAT32, shape);
 
  ASSERT_EQ(t.getShape(), shape);
  ASSERT_EQ(t.getOffset({0, 0}), 0u);
}

References mir::TensorVariant::getOffset(), and mir::TensorVariant::getShape().

TEST() [2/2]

TEST	(	TensorVariant	,
		ElementSizeDeductionTest
	)

Definition at line 32 of file TensorVariant.cpp.

{
  Shape shape{2, 2, 2};
  TensorVariant t(DataType::FLOAT32, shape);
 
  ASSERT_EQ(t.getElementSize(), sizeof(float));
  ASSERT_EQ((float *)t.at({1, 1, 1}), (float *)t.at({0, 0, 0}) + 7);
}

References mir::TensorVariant::at(), and mir::TensorVariant::getElementSize().