tensorflowSerDes.cpp File Reference

This file defines the TensorflowSerDes class, to support interfacing with Tensorflow AOT models via TFModelRunner. More...

#include "SerDes/tensorflowSerDes.h"
#include "SerDes/baseSerDes.h"

Include dependency graph for tensorflowSerDes.cpp:

Go to the source code of this file.

Namespaces
namespace	MLBridge

Macros
#define	SET_FEATURE(TYPE, _)

Detailed Description

This file defines the TensorflowSerDes class, to support interfacing with Tensorflow AOT models via TFModelRunner.

Definition in file tensorflowSerDes.cpp.

Macro Definition Documentation

SET_FEATURE

#define SET_FEATURE	(		TYPE,
			_ )

Value:

  void TensorflowSerDes::setFeature(const std::string &Name,                   \
                                    const TYPE Value) {                        \
    std::string prefix = "feed_";                                              \
    const int Index = CompiledModel->LookupArgIndex(prefix + Name);            \
    if (Index >= 0)                                                            \
      *reinterpret_cast<TYPE *>(CompiledModel->arg_data(Index)) = Value;       \
  }

Definition at line 20 of file tensorflowSerDes.cpp.