31 const auto &inputs = args.op.inputs;
32 const auto &outputs = args.op.outputs;
33 const auto tensors = args.reader.tensors();
34 const auto tensor_in = tensors.at(inputs.at(0));
35 const auto tensor_axis = tensors.at(inputs.at(1));
36 const auto tensor_out = tensors.at(outputs[0]);
37 assert(tensor_in !=
nullptr);
38 assert(tensor_axis !=
nullptr);
39 assert(tensor_out !=
nullptr);
41 switch (tensor_axis->type())
43 case circle::TensorType_INT32:
44 case circle::TensorType_INT64:
50 if (tensor_out->type() != tensor_in->type())