PluginTensorDesc has four members
Dims dims
DataType type
TensorFormat format
float *scale
How can I print them to check?
I did this
cout <<"pos " << pos << inOut[pos].dims << inOut[pos].type << inOut[pos].format << inOut[pos].scale << endl;
and I have compile error.
They are from Plugin’s method
bool supportsFormatCombination(int pos, const PluginTensorDesc* inOut, int nbInputs, int nbOutputs)
{
cout <<"pos " << pos << inOut[pos].dims << inOut[pos].type << inOut[pos].format << inOut[pos].scale << endl;
assert(nbInputs == 2 && nbOutputs == 1 && pos < nbInputs + nbOutputs);
bool condition = inOut[pos].format == TensorFormat::kLINEAR;
condition = inOut[pos].type != DataType::kINT32;
condition &= inOut[pos].type == inOut[0].type;
cout << "condition " << condition << endl;
return condition;
}