Lines Matching refs:dataLayout
286 static void RefCreateBatchNormalizationWorkloadTest(DataLayout dataLayout) in RefCreateBatchNormalizationWorkloadTest() argument
292 … dataLayout); in RefCreateBatchNormalizationWorkloadTest()
297 switch (dataLayout) in RefCreateBatchNormalizationWorkloadTest()
402 static void RefCreateConvolution2dWorkloadTest(DataLayout dataLayout = DataLayout::NCHW) in RefCreateConvolution2dWorkloadTest() argument
407 (factory, graph, dataLayout); in RefCreateConvolution2dWorkloadTest()
409 …TensorShape inputShape = (dataLayout == DataLayout::NCHW) ? std::initializer_list<unsigned int>({… in RefCreateConvolution2dWorkloadTest()
411 …TensorShape outputShape = (dataLayout == DataLayout::NCHW) ? std::initializer_list<unsigned int>({… in RefCreateConvolution2dWorkloadTest()
432 DataLayout dataLayout = DataLayout::NHWC; in BOOST_AUTO_TEST_CASE() local
436 (factory, graph, dataLayout); in BOOST_AUTO_TEST_CASE()
438 …TensorShape inputShape = (dataLayout == DataLayout::NCHW) ? std::initializer_list<unsigned int>({… in BOOST_AUTO_TEST_CASE()
440 …TensorShape outputShape = (dataLayout == DataLayout::NCHW) ? std::initializer_list<unsigned int>({… in BOOST_AUTO_TEST_CASE()
449 static void RefCreateDepthwiseConvolutionWorkloadTest(DataLayout dataLayout) in RefCreateDepthwiseConvolutionWorkloadTest() argument
454 (factory, graph, dataLayout); in RefCreateDepthwiseConvolutionWorkloadTest()
456 …TensorShape inputShape = (dataLayout == DataLayout::NCHW) ? std::initializer_list<unsigned int>({… in RefCreateDepthwiseConvolutionWorkloadTest()
458 …TensorShape outputShape = (dataLayout == DataLayout::NCHW) ? std::initializer_list<unsigned int>({… in RefCreateDepthwiseConvolutionWorkloadTest()
518 static void RefCreateNormalizationWorkloadTest(DataLayout dataLayout) in RefCreateNormalizationWorkloadTest() argument
522 …= CreateNormalizationWorkloadTest<NormalizationWorkloadType, DataType>(factory, graph, dataLayout); in RefCreateNormalizationWorkloadTest()
527 switch (dataLayout) in RefCreateNormalizationWorkloadTest()
575 static void RefCreatePooling2dWorkloadTest(DataLayout dataLayout) in RefCreatePooling2dWorkloadTest() argument
579 …orkload = CreatePooling2dWorkloadTest<Pooling2dWorkloadType, DataType>(factory, graph, dataLayout); in RefCreatePooling2dWorkloadTest()
584 switch (dataLayout) in RefCreatePooling2dWorkloadTest()
814 static void RefCreateResizeBilinearTest(DataLayout dataLayout) in RefCreateResizeBilinearTest() argument
818 …CreateResizeBilinearWorkloadTest<ResizeBilinearWorkloadType, DataType>(factory, graph, dataLayout); in RefCreateResizeBilinearTest()
823 switch (dataLayout) in RefCreateResizeBilinearTest()
900 static void RefCreateL2NormalizationTest(DataLayout dataLayout) in RefCreateL2NormalizationTest() argument
905 …eateL2NormalizationWorkloadTest<L2NormalizationWorkloadType, DataType>(factory, graph, dataLayout); in RefCreateL2NormalizationTest()
910 switch (dataLayout) in RefCreateL2NormalizationTest()