SplitV.cpp File Reference

#include "SplitV.h"
#include "Utils.h"
#include <tensorflow/lite/kernels/internal/optimized/optimized_ops.h>

Go to the source code of this file.

Namespaces
namespace	luci_interpreter
namespace	luci_interpreter::kernels

Macros
#define	TF_LITE_SPLIT(scalar)

Macro Definition Documentation

TF_LITE_SPLIT

#define TF_LITE_SPLIT

(

scalar

)

Value:

  {                                                                                               \
    VectorOfTensors<scalar, false> all_outputs(_outputs);                                         \
    tflite::optimized_ops::Split(params, getTensorShape(input()), getTensorData<scalar>(input()), \
                                 all_outputs.shapes(), all_outputs.data());                       \
  }