35constexpr uint32_t inputTensorIdx = 0;
43 const circle::Tensor *
input =
nullptr;
44 const circle::Tensor *
output =
nullptr;
49 SISOHeader(execute_args, &input, &output, &input_data, &output_data);
53 switch (
input->type())
56 case circle::TensorType_FLOAT32:
60 case circle::TensorType_INT32:
67 case circle::TensorType_INT8:
74 case circle::TensorType_INT16:
84 assert(
false &&
"Unsupported type.");
94 assert(
false &&
"Unsupported type.");
constexpr uint32_t outputTensorIdx
OMStatus Cast(const core::OMRuntimeShape &input_shape, const FromT *input_data, const core::OMRuntimeShape &output_shape, ToT *output_data)
OMStatus SISOHeader(const OMExecuteArgs &execute_args, const circle::Tensor **input, const circle::Tensor **output, uint8_t **input_data, uint8_t **output_data)