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 = tensors.at(inputs.at(0));
35 assert(tensor !=
nullptr);
37 switch (tensor->type())
39 case circle::TensorType_FLOAT32:
40 case circle::TensorType_FLOAT64:
41 case circle::TensorType_INT32:
42 case circle::TensorType_UINT8:
43 case circle::TensorType_INT16:
44 case circle::TensorType_INT8:
45 case circle::TensorType_INT64:
46 case circle::TensorType_FLOAT16:
52 assert(tensors[inputs.at(1)] !=
nullptr);
53 if (tensors[inputs.at(1)]->type() != tensor->type())
58 assert(tensors[outputs[0]] !=
nullptr);
59 return tensors[outputs[0]]->type() == circle::TensorType_BOOL;