Searched defs:newShape (Results 1 – 9 of 9) sorted by relevance
64 auto newShape = targetShapeType.getElementType().isIndex() in matchAndRewrite() local131 auto newShape = targetShapeType.getElementType().isIndex() in matchAndRewrite() local
2074 int64_t reshapedDim, int64_t factor) { in matchAndRewrite()2097 llvm::SmallVector<int64_t> newShape; in matchAndRewrite() local2112 llvm::SmallVector<int64_t> newShape; in matchAndRewrite() local2140 llvm::SmallVector<int64_t> newShape; in matchAndRewrite() local2155 llvm::SmallVector<int64_t> newShape; in matchAndRewrite() local
70 TensorShape newShape = permuteLayer->GetOutputSlot().GetTensorInfo().GetShape(); in PermuteDepthwiseConv2dWeightsTestRunner() local
112 std::vector<unsigned int> newShape; in ReduceDims() local176 std::vector<unsigned int> newShape; in ExpandDimsToRank() local
55 …const armnn::TensorShape& newShape = armnnUtils::ExpandDimsToRank(smallInfo.GetShape(), biggerInpu… in ExpandTensorRankToEqual() local
192 int[] newShape = tensor.getInputShapeIfDifferent(input.getValue()); in runSignature() local194 signatureRunnerWrapper.resizeInput(input.getKey(), newShape); in runSignature() local234 int[] newShape = tensor.getInputShapeIfDifferent(inputs[i]); in run() local
193 void SetShape(const TensorShape& newShape) { m_Shape = newShape; } in SetShape()
151 std::vector<unsigned int> newShape; in ReduceDimsForACL() local
1066 std::vector<uint32_t> newShape; in To1DTensor() local2346 std::vector<uint32_t> newShape; in PrependForBroadcast() local