• Home
  • Raw
  • Download

Lines Matching refs:dataLayout

37 void ReshapeWeightsForAcl(TensorInfo& weightInfo, DataLayout dataLayout)  in ReshapeWeightsForAcl()  argument
41 switch (dataLayout) in ReshapeWeightsForAcl()
63 ConstTensor ReorderWeightChannelsForAcl(const ConstTensor& weightHandle, DataLayout dataLayout, voi… in ReorderWeightChannelsForAcl() argument
71 switch (dataLayout) in ReorderWeightChannelsForAcl()
110 …nsorInfo ConvertWeightTensorInfoFromArmnnToAcl(const TensorInfo& weightInfo, DataLayout dataLayout) in ConvertWeightTensorInfoFromArmnnToAcl() argument
119 if (dataLayout == DataLayout::NHWC) in ConvertWeightTensorInfoFromArmnnToAcl()
127 ReshapeWeightsForAcl(weightPermutedInfo, dataLayout); in ConvertWeightTensorInfoFromArmnnToAcl()
134 DataLayout dataLayout, in ConvertWeightTensorFromArmnnToAcl() argument
151 if (dataLayout == DataLayout::NHWC) in ConvertWeightTensorFromArmnnToAcl()
159 if (multiplier > 1 && inputChannels > 1 && dataLayout == DataLayout::NCHW) in ConvertWeightTensorFromArmnnToAcl()
164 … weightPermuted = ReorderWeightChannelsForAcl<float>(weightPermuted, dataLayout, permuteBuffer); in ConvertWeightTensorFromArmnnToAcl()
168 … ReorderWeightChannelsForAcl<half_float::half>(weightPermuted, dataLayout, permuteBuffer); in ConvertWeightTensorFromArmnnToAcl()
172 … weightPermuted = ReorderWeightChannelsForAcl<uint8_t>(weightPermuted, dataLayout, permuteBuffer); in ConvertWeightTensorFromArmnnToAcl()
178 … weightPermuted = ReorderWeightChannelsForAcl<int8_t>(weightPermuted, dataLayout, permuteBuffer); in ConvertWeightTensorFromArmnnToAcl()
187 ReshapeWeightsForAcl(weightPermuted.GetInfo(), dataLayout); in ConvertWeightTensorFromArmnnToAcl()