| @@ -1562,8 +1562,15 @@ int main(int argc, char** argv) | |||||
| else | else | ||||
| { | { | ||||
| const onnx::TensorProto& scales_tp = weights[node.input(1)]; | const onnx::TensorProto& scales_tp = weights[node.input(1)]; | ||||
| const float* shape_data = scales_tp.float_data().data(); | |||||
| for (int j=0; j<scales_tp.float_data_size(); j++) | |||||
| const float* shape_data = scales_tp.has_raw_data() ? (const float*)scales_tp.raw_data().data() : scales_tp.float_data().data(); | |||||
| int float_data_size = scales_tp.float_data_size(); | |||||
| //float data is None, use raw data instead | |||||
| if (float_data_size == 0) { | |||||
| float_data_size = scales_tp.dims().Get(0); | |||||
| } | |||||
| for (int j=0; j<float_data_size; j++) | |||||
| { | { | ||||
| scales.push_back(shape_data[j]); | scales.push_back(shape_data[j]); | ||||
| } | } | ||||