Searched defs:expectedShape (Results 1 – 5 of 5) sorted by relevance
122 armnn::TensorShape expectedShape(4, expectedDimSizes.data()); in BatchToSpaceInferOutputShapeTest() local144 armnn::TensorShape expectedShape(4, expectedDimSizes.data()); in SpaceToDepthInferOutputShapeTest() local
76 const armnn::TensorShape& expectedShape, in CompareTensors()
34 armnn::TensorShape& expectedShape) in LstmUtilsVectorBatchVectorAddTestImpl()61 armnn::TensorShape& expectedShape) in LstmUtilsZeroVectorTestImpl()90 armnn::TensorShape& expectedShape) in LstmUtilsMeanStddevNormalizationTestImpl()118 armnn::TensorShape& expectedShape) in LstmUtilsVectorBatchVectorCwiseProductTestImpl()
82 int[] expectedShape = {2, 8, 8, 3}; in testBasic() local
674 final long[] expectedShape, in assertIValueTensor()