14 #include <boost/core/ignore_unused.hpp> 20 namespace optimizations
25 static void Func(std::unique_ptr<ScopedCpuTensorHandle>& handle)
46 static void Func(std::unique_ptr<ScopedCpuTensorHandle>& handle)
65 template<
typename Converter,
typename Predicate>
75 boost::ignore_unused(graph);
76 if (Predicate::Test(layer))
static void Func(std::unique_ptr< ScopedCpuTensorHandle > &handle)
void Run(Graph &graph, Layer &layer) const override
A tensor defined by a TensorInfo (shape and data type) and an immutable backing store.
DataType GetDataType() const
static bool Test(const Layer &layer)
static void ConvertFloat16To32(const void *srcFloat16Buffer, size_t numElements, float *dstFloat32Buffer)
static void ConvertFloat32To16(const float *srcFloat32Buffer, size_t numElements, void *dstFloat16Buffer)
static bool Test(const Layer &layer)
unsigned int GetNumElements() const
DataType GetDataType() const
void OperateOnConstantTensors(Op op)
static void Func(std::unique_ptr< ScopedCpuTensorHandle > &handle)
const TensorShape & GetShape() const