MCPcopy Create free account
hub / github.com/ARM-software/armnn / TensorInfoAsString

Function TensorInfoAsString

src/armnnOnnxParser/OnnxParser.cpp:361–377  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

359}
360
361std::string TensorInfoAsString(const TensorInfo& info,
362 const std::string& name,
363 const onnx::TensorProto::DataType& type)
364{
365 const TensorShape shape = info.GetShape();
366 std::stringstream ss;
367 ss << "tensor '" << name << "' contains "
368 << onnx::TensorProto::DataType_Name(type)
369 << " and has shape [";
370
371 for (uint32_t i = 0; i < shape.GetNumDimensions() - 1; ++i)
372 {
373 ss << shape[i] << ", ";
374 }
375 ss << shape[shape.GetNumDimensions() - 1] << "]";
376 return ss.str();
377}
378
379void CalcPadding(uint32_t inputSize,
380 uint32_t filterSize,

Callers 4

To1DTensorMethod · 0.85
AddFullyConnectedMethod · 0.85
ParseAddMethod · 0.85
ParseConvMethod · 0.85

Calls 3

GetShapeMethod · 0.45
GetNumDimensionsMethod · 0.45
strMethod · 0.45

Tested by

no test coverage detected