Searched refs:dataLayoutIndex (Results 1 – 4 of 4) sorted by relevance
38 DataLayoutIndexed dataLayoutIndex(m_Param.m_DataLayout); in SerializeLayerParameters() local39 unsigned int filterWidth = filterShape[dataLayoutIndex.GetWidthIndex()]; in SerializeLayerParameters()40 unsigned int filterHeight = filterShape[dataLayoutIndex.GetHeightIndex()]; in SerializeLayerParameters()76 DataLayoutIndexed dataLayoutIndex(m_Param.m_DataLayout); in InferOutputShapes() local78 unsigned int inWidth = inputShape[dataLayoutIndex.GetWidthIndex()]; in InferOutputShapes()79 unsigned int inHeight = inputShape[dataLayoutIndex.GetHeightIndex()]; in InferOutputShapes()82 unsigned int filterWidth = filterShape[dataLayoutIndex.GetWidthIndex()]; in InferOutputShapes()87 unsigned int filterHeight = filterShape[dataLayoutIndex.GetHeightIndex()]; in InferOutputShapes()
66 DataLayoutIndexed dataLayoutIndex(m_Param.m_DataLayout); in InferOutputShapes() local70 const unsigned int wInput = inputShape[dataLayoutIndex.GetWidthIndex()]; in InferOutputShapes()71 const unsigned int hInput = inputShape[dataLayoutIndex.GetHeightIndex()]; in InferOutputShapes()73 const unsigned int wKernel = kernelShape[dataLayoutIndex.GetWidthIndex()]; in InferOutputShapes()74 const unsigned int hKernel = kernelShape[dataLayoutIndex.GetHeightIndex()]; in InferOutputShapes()
74 DataLayoutIndexed dataLayoutIndex(m_Param.m_DataLayout); in InferOutputShapes() local76 unsigned int inWidth = inputShape[dataLayoutIndex.GetWidthIndex()]; in InferOutputShapes()77 unsigned int inHeight = inputShape[dataLayoutIndex.GetHeightIndex()]; in InferOutputShapes()78 unsigned int inDepth = inputShape[dataLayoutIndex.GetDepthIndex()]; in InferOutputShapes()
76 DataLayoutIndexed dataLayoutIndex(m_Param.m_DataLayout); in InferOutputShapes() local79 unsigned int inputHeight = inputShape[dataLayoutIndex.GetHeightIndex()]; in InferOutputShapes()80 unsigned int inputWidth = inputShape[dataLayoutIndex.GetWidthIndex()]; in InferOutputShapes()