38constexpr int kWeightTensorIdx = 1;
39constexpr int kInputTensorIdx = 2;
40constexpr int kBiasTensorIdx = 3;
41constexpr int kOutputTensorIdx = 0;
46onert_micro::import::configure_kernel_CircleTransposeConv(
const OMConfigureArgs &config_args)
49 uint16_t op_index = config_args.kernel_index;
52 runtime_kernel.
readKernel(op_index, runtime_context);
54 const circle::Tensor *
input = runtime_kernel.
inputs[kInputTensorIdx];
55 const circle::Tensor *weight = runtime_kernel.
inputs[kWeightTensorIdx];
56 const circle::Tensor *
output = runtime_kernel.
outputs[kOutputTensorIdx];
58 assert(input !=
nullptr);
59 assert(weight !=
nullptr);
61 assert(output !=
nullptr);
65 status = utils::checkCondition(
input->type() ==
output->type());
69 status = utils::checkCondition(
input->type() == weight->type());
76 status = utils::checkCondition(input_shape.dimensionsCount() == 4);
80 status = utils::checkCondition(input_shape.dimensionsCount() == weight_shape.dimensionsCount());
OMStatus readKernel(uint16_t op_index, core::OMRuntimeContext &runtime_context)
const circle::Tensor * outputs[maxOutputSize]
const circle::Tensor * inputs[maxInputSize]