Home
last modified time | relevance | path

Searched refs:GetDataLayout (Results 1 – 8 of 8) sorted by relevance

/external/armnn/src/armnnUtils/
DDataLayoutIndexed.cpp36 return dataLayout == indexed.GetDataLayout(); in operator ==()
41 return indexed.GetDataLayout() == dataLayout; in operator ==()
/external/armnn/src/backends/reference/workloads/
DConvImpl.cpp159 if (dataLayoutIndexed.GetDataLayout() == DataLayout::NHWC) in Convolve()
192 if (dataLayoutIndexed.GetDataLayout() == DataLayout::NHWC) in Convolve()
220 if (dataLayoutIndexed.GetDataLayout() == DataLayout::NHWC) in Convolve()
DPooling2d.cpp223 if(dataLayout.GetDataLayout() == DataLayout::NHWC) in Pooling2d()
258 if(dataLayout.GetDataLayout() == DataLayout::NHWC) in Pooling2d()
282 if(dataLayout.GetDataLayout() == DataLayout::NHWC) in Pooling2d()
DSpaceToDepth.cpp20 if (dataLayout.GetDataLayout() == armnn::DataLayout::NHWC) in GetOffset()
DBatchToSpaceNd.cpp22 if (dataLayout.GetDataLayout() == DataLayout::NHWC) in Offset()
DSpaceToBatchNd.cpp22 if (dataLayout.GetDataLayout() == DataLayout::NHWC) in GetOffset()
/external/armnn/include/armnnUtils/
DDataLayoutIndexed.hpp22 armnn::DataLayout GetDataLayout() const { return m_DataLayout; } in GetDataLayout() function in armnnUtils::DataLayoutIndexed
/external/armnn/src/backends/backendsCommon/test/layerTests/
DConv2dTestImpl.cpp2973 switch (layout.GetDataLayout()) in CompareDepthwiseConvolution2dTestImpl()
2985 … + std::to_string(static_cast<int>(layout.GetDataLayout())) + "]"); in CompareDepthwiseConvolution2dTestImpl()
3027 data.m_Parameters.m_DataLayout = layout.GetDataLayout(); in CompareDepthwiseConvolution2dTestImpl()