Searched refs:SqueezeDims (Results 1 – 4 of 4) sorted by relevance
53 std::vector<unsigned int> SqueezeDims(const armnn::TensorShape& tensorShape);
432 std::vector<unsigned int> outputDims = armnnUtils::SqueezeDims(outputShape); in ValidateAndCopyShape()433 std::vector<unsigned int> inferredDims = armnnUtils::SqueezeDims(inferredShape); in ValidateAndCopyShape()
193 std::vector<unsigned int> SqueezeDims(const TensorShape& tensorShape) in SqueezeDims() function
8669 * Rename `SqueezeDims` attribute to `Axis` in C++ API for Squeeze op.