Home
last modified time | relevance | path

Searched refs:stridedSliceDescriptor (Results 1 – 7 of 7) sorted by relevance

/external/armnn/src/backends/backendsCommon/test/
DStridedSliceAsyncEndToEndTest.hpp260 armnn::StridedSliceDescriptor stridedSliceDescriptor; in CreateStridedSliceNetwork() local
261 stridedSliceDescriptor.m_Begin = beginData; in CreateStridedSliceNetwork()
262 stridedSliceDescriptor.m_End = endData; in CreateStridedSliceNetwork()
263 stridedSliceDescriptor.m_Stride = stridesData; in CreateStridedSliceNetwork()
264 stridedSliceDescriptor.m_BeginMask = beginMask; in CreateStridedSliceNetwork()
265 stridedSliceDescriptor.m_EndMask = endMask; in CreateStridedSliceNetwork()
266 stridedSliceDescriptor.m_ShrinkAxisMask = shrinkAxisMask; in CreateStridedSliceNetwork()
267 stridedSliceDescriptor.m_EllipsisMask = ellipsisMask; in CreateStridedSliceNetwork()
268 stridedSliceDescriptor.m_NewAxisMask = newAxisMask; in CreateStridedSliceNetwork()
271 IConnectableLayer* stridedSlice = net->AddStridedSliceLayer(stridedSliceDescriptor, "splitter"); in CreateStridedSliceNetwork()
/external/armnn/src/armnnSerializer/
DSerializer.cpp1384 … const armnn::StridedSliceDescriptor& stridedSliceDescriptor, in SerializeStridedSliceLayer() argument
1393 … m_flatBufferBuilder.CreateVector(stridedSliceDescriptor.m_Begin), in SerializeStridedSliceLayer()
1394 m_flatBufferBuilder.CreateVector(stridedSliceDescriptor.m_End), in SerializeStridedSliceLayer()
1395 … m_flatBufferBuilder.CreateVector(stridedSliceDescriptor.m_Stride), in SerializeStridedSliceLayer()
1396 stridedSliceDescriptor.m_BeginMask, in SerializeStridedSliceLayer()
1397 stridedSliceDescriptor.m_EndMask, in SerializeStridedSliceLayer()
1398 stridedSliceDescriptor.m_ShrinkAxisMask, in SerializeStridedSliceLayer()
1399 stridedSliceDescriptor.m_EllipsisMask, in SerializeStridedSliceLayer()
1400 stridedSliceDescriptor.m_NewAxisMask, in SerializeStridedSliceLayer()
1401 GetFlatBufferDataLayout(stridedSliceDescriptor.m_DataLayout)); in SerializeStridedSliceLayer()
DSerializer.hpp328 const armnn::StridedSliceDescriptor& stridedSliceDescriptor,
/external/armnn/src/armnn/
DNetwork.hpp213 IConnectableLayer* AddStridedSliceLayer(const StridedSliceDescriptor& stridedSliceDescriptor,
DNetwork.cpp538 …nectableLayer* INetwork::AddStridedSliceLayer(const StridedSliceDescriptor& stridedSliceDescriptor, in AddStridedSliceLayer() argument
541 return pNetworkImpl->AddStridedSliceLayer(stridedSliceDescriptor, name); in AddStridedSliceLayer()
2497 …tableLayer* NetworkImpl::AddStridedSliceLayer(const StridedSliceDescriptor& stridedSliceDescriptor, in AddStridedSliceLayer() argument
2500 return m_Graph->AddLayer<StridedSliceLayer>(stridedSliceDescriptor, name); in AddStridedSliceLayer()
/external/armnn/python/pyarmnn/src/pyarmnn/swig/modules/
Darmnn_network.i1068 …ConnectableLayer* AddStridedSliceLayer(const armnn::StridedSliceDescriptor& stridedSliceDescriptor,
/external/armnn/include/armnn/
DINetwork.hpp717 IConnectableLayer* AddStridedSliceLayer(const StridedSliceDescriptor& stridedSliceDescriptor,