Tensor.cpp File Reference

#include "Tensor.h"
#include <luci/IR/CircleNodeDecl.h>
#include <cassert>

Go to the source code of this file.

Namespaces
namespace	circle_eval_diff

Macros
#define	THROW_UNLESS(COND, MSG)
#define	INSTANTIATE(DT)

Functions
	circle_eval_diff::INSTANTIATE (loco::DataType::S64)
	circle_eval_diff::INSTANTIATE (loco::DataType::S32)
	circle_eval_diff::INSTANTIATE (loco::DataType::S16)
	circle_eval_diff::INSTANTIATE (loco::DataType::U8)
	circle_eval_diff::INSTANTIATE (loco::DataType::FLOAT32)
std::shared_ptr< Tensor >	circle_eval_diff::createEmptyTensor (const luci::CircleNode *node)

Macro Definition Documentation

INSTANTIATE

#define INSTANTIATE

(

DT

)

Value:

  template uint32_t Tensor::size<DT>(void) const;                                       \
  template void Tensor::size<DT>(uint32_t);                                             \
  template const typename loco::DataTypeImpl<DT>::Type &Tensor::at<DT>(uint32_t) const; \
  template typename loco::DataTypeImpl<DT>::Type &Tensor::at<DT>(uint32_t);

Definition at line 74 of file Tensor.cpp.

{
  auto tensor = std::make_shared<Tensor>();
  {
    tensor->dtype(node->dtype());
    tensor->rank(node->rank());
    for (uint32_t i = 0; i < node->rank(); i++)
      tensor->dim(i) = node->dim(i);
 
    switch (node->dtype())
    {
      case loco::DataType::FLOAT32:
        tensor->size<loco::DataType::FLOAT32>(numElements(node));
        break;
      case loco::DataType::U8:
        tensor->size<loco::DataType::U8>(numElements(node));
        break;
      case loco::DataType::S16:
        tensor->size<loco::DataType::S16>(numElements(node));
        break;
      case loco::DataType::S32:
        tensor->size<loco::DataType::S32>(numElements(node));
        break;
      case loco::DataType::S64:
        tensor->size<loco::DataType::S64>(numElements(node));
        break;
      default:
        throw std::runtime_error("Unsupported input tensor dtype for " + node->name());
    }
  }
 
  return tensor;
}
 
} // namespace circle_eval_diff

THROW_UNLESS

#define THROW_UNLESS	(		COND,
			MSG
	)

Value:

  if (not(COND))                \
    throw std::runtime_error(MSG);

Definition at line 40 of file Tensor.cpp.