Lines Matching refs:dataLayout
196 static void NeonCreateBatchNormalizationWorkloadTest(DataLayout dataLayout) in NeonCreateBatchNormalizationWorkloadTest() argument
203 (factory, graph, dataLayout); in NeonCreateBatchNormalizationWorkloadTest()
210 …TensorShape inputShape = (dataLayout == DataLayout::NCHW) ? TensorShape{2, 3, 4, 4} : TensorShape… in NeonCreateBatchNormalizationWorkloadTest()
211 …TensorShape outputShape = (dataLayout == DataLayout::NCHW) ? TensorShape{2, 3, 4, 4} : TensorShape… in NeonCreateBatchNormalizationWorkloadTest()
240 static void NeonCreateConvolution2dWorkloadTest(DataLayout dataLayout = DataLayout::NCHW) in NeonCreateConvolution2dWorkloadTest() argument
246 …= CreateConvolution2dWorkloadTest<NeonConvolution2dWorkload, DataType>(factory, graph, dataLayout); in NeonCreateConvolution2dWorkloadTest()
248 …TensorShape inputShape = (dataLayout == DataLayout::NCHW) ? TensorShape{2, 3, 8, 16} : TensorShap… in NeonCreateConvolution2dWorkloadTest()
249 …TensorShape outputShape = (dataLayout == DataLayout::NCHW) ? TensorShape{2, 2, 2, 10} : TensorShap… in NeonCreateConvolution2dWorkloadTest()
308 static void NeonCreateDepthWiseConvolutionWorkloadTest(DataLayout dataLayout) in NeonCreateDepthWiseConvolutionWorkloadTest() argument
315 DataType>(factory, graph, dataLayout); in NeonCreateDepthWiseConvolutionWorkloadTest()
322 …TensorShape inputShape = (dataLayout == DataLayout::NCHW) ? std::initializer_list<unsigned int>({… in NeonCreateDepthWiseConvolutionWorkloadTest()
324 …TensorShape outputShape = (dataLayout == DataLayout::NCHW) ? std::initializer_list<unsigned int>({… in NeonCreateDepthWiseConvolutionWorkloadTest()
387 static void NeonCreateNormalizationWorkloadTest(DataLayout dataLayout) in NeonCreateNormalizationWorkloadTest() argument
393 …= CreateNormalizationWorkloadTest<NormalizationWorkloadType, DataType>(factory, graph, dataLayout); in NeonCreateNormalizationWorkloadTest()
400 …TensorShape inputShape = (dataLayout == DataLayout::NCHW) ? TensorShape{3, 5, 5, 1} : TensorShape… in NeonCreateNormalizationWorkloadTest()
401 …TensorShape outputShape = (dataLayout == DataLayout::NCHW) ? TensorShape{3, 5, 5, 1} : TensorShape… in NeonCreateNormalizationWorkloadTest()
431 static void NeonCreatePooling2dWorkloadTest(DataLayout dataLayout = DataLayout::NCHW) in NeonCreatePooling2dWorkloadTest() argument
437 …orkload = CreatePooling2dWorkloadTest<NeonPooling2dWorkload, DataType>(factory, graph, dataLayout); in NeonCreatePooling2dWorkloadTest()
439 …TensorShape inputShape = (dataLayout == DataLayout::NCHW) ? TensorShape{3, 2, 5, 5} : TensorShape… in NeonCreatePooling2dWorkloadTest()
440 …TensorShape outputShape = (dataLayout == DataLayout::NCHW) ? TensorShape{3, 2, 2, 4} : TensorShape… in NeonCreatePooling2dWorkloadTest()
555 static void NeonCreateResizeWorkloadTest(DataLayout dataLayout) in NeonCreateResizeWorkloadTest() argument
560 …kload = CreateResizeBilinearWorkloadTest<ResizeWorkloadType, DataType>(factory, graph, dataLayout); in NeonCreateResizeWorkloadTest()
567 switch (dataLayout) in NeonCreateResizeWorkloadTest()
799 static void NeonCreateL2NormalizationWorkloadTest(DataLayout dataLayout) in NeonCreateL2NormalizationWorkloadTest() argument
806 …eateL2NormalizationWorkloadTest<L2NormalizationWorkloadType, DataType>(factory, graph, dataLayout); in NeonCreateL2NormalizationWorkloadTest()
813 TensorShape inputShape = (dataLayout == DataLayout::NCHW) ? in NeonCreateL2NormalizationWorkloadTest()
815 TensorShape outputShape = (dataLayout == DataLayout::NCHW) ? in NeonCreateL2NormalizationWorkloadTest()