Searched refs:channel_first (Results 1 – 2 of 2) sorted by relevance
| /third_party/mindspore/mindspore/lite/src/ |
| D | weight_decoder.cc | 191 int WeightDecoder::DequantWeight(lite::Tensor *input_tensor, bool channel_first, TypeId dst_data_ty… in DequantWeight() argument
202 auto new_const_data = DequantData<int16_t, float>(input_tensor, channel_first); in DequantWeight()
209 auto new_const_data = DequantData<int16_t, float16_t>(input_tensor, channel_first); in DequantWeight()
219 auto new_const_data = DequantData<int8_t, float>(input_tensor, channel_first); in DequantWeight()
226 auto new_const_data = DequantData<int8_t, float16_t>(input_tensor, channel_first); in DequantWeight()
331 auto channel_first = IsChannelFirst(index++, op_parameter); in DequantNode() local
332 auto ret = WeightDecoder::DequantTensor(tensor, channel_first, dst_data_type); in DequantNode()
341 int WeightDecoder::DequantTensor(Tensor *tensor, bool channel_first, TypeId dst_data_type) { in DequantTensor() argument
352 auto ret = WeightDecoder::DequantWeight(tensor, channel_first, dst_data_type); in DequantTensor()
|
| D | weight_decoder.h | 136 …static int DequantTensor(Tensor *tensor, bool channel_first = true, TypeId dst_data_type = kNumber…
143 static DT *DequantData(lite::Tensor *input_tensor, bool channel_first = true) {
158 if (!channel_first) {
179 if (!channel_first) {
230 …static int DequantWeight(lite::Tensor *input_tensor, bool channel_first, TypeId dst_data_type = kN…
|