Searched refs:output_channel (Results 1 – 5 of 5) sorted by relevance
| /tflite-micro-3.4.0-2.7.6/tensorflow/lite/kernels/internal/reference/integer_ops/ |
| D | depthwise_conv.h | 66 const int output_channel = m + in_channel * depth_multiplier; in DepthwiseConvPerChannel() local
83 filter_shape, 0, filter_y, filter_x, output_channel)]; in DepthwiseConvPerChannel()
105 acc += bias_data[output_channel]; in DepthwiseConvPerChannel()
108 acc, output_multiplier[output_channel], in DepthwiseConvPerChannel()
109 output_shift[output_channel]); in DepthwiseConvPerChannel()
114 output_channel)] = static_cast<int8_t>(acc); in DepthwiseConvPerChannel()
163 const int output_channel = m + in_channel * depth_multiplier; in DepthwiseConvPerChannel() local
180 filter_shape, 0, filter_y, filter_x, output_channel)]; in DepthwiseConvPerChannel()
191 acc += bias_data[output_channel]; in DepthwiseConvPerChannel()
194 acc, output_multiplier[output_channel], in DepthwiseConvPerChannel()
[all …]
|
| /tflite-micro-3.4.0-2.7.6/tensorflow/lite/micro/kernels/vexriscv/ |
| D | depthwise_conv.cc | 106 const int output_channel = m + in_channel * depth_multiplier; in DepthwiseConvPerChannel() local
111 filter_shape, 0, filter_y, filter_x, output_channel)]; in DepthwiseConvPerChannel()
136 const int output_channel = m + in_channel * depth_multiplier; in DepthwiseConvPerChannel() local
140 value += bias_data[output_channel]; in DepthwiseConvPerChannel()
144 value, output_multiplier[output_channel], in DepthwiseConvPerChannel()
145 output_shift[output_channel]); in DepthwiseConvPerChannel()
151 output_channel)] = static_cast<int8_t>(value); in DepthwiseConvPerChannel()
229 const int output_channel = m + in_channel * depth_multiplier; in DepthwiseConv() local
234 filter_shape, 0, filter_y, filter_x, output_channel)]; in DepthwiseConv()
244 const int output_channel = m + in_channel * depth_multiplier; in DepthwiseConv() local
[all …]
|
| /tflite-micro-3.4.0-2.7.6/tensorflow/lite/micro/kernels/xtensa/ |
| D | depthwise_conv_hifimini.cc | 71 const int output_channel = m + in_channel * depth_multiplier; in DepthwiseConvEvalHifiMini() local
97 (output_channel); in DepthwiseConvEvalHifiMini()
125 ae_q56s bias_56 = AE_CVTQ48A32S(bias_data[output_channel]); in DepthwiseConvEvalHifiMini()
137 acc_24x2, output_multiplier[output_channel], in DepthwiseConvEvalHifiMini()
138 output_shift[output_channel]); in DepthwiseConvEvalHifiMini()
148 output_channel; in DepthwiseConvEvalHifiMini()
|
| /tflite-micro-3.4.0-2.7.6/tensorflow/lite/kernels/internal/reference/ |
| D | depthwiseconv_uint8.h | 221 const int output_channel = m + in_channel * depth_multiplier; in RunPerChannel() local
239 filter_shape, 0, filter_y, filter_x, output_channel)]; in RunPerChannel()
260 acc += bias_data[output_channel]; in RunPerChannel()
263 acc, output_multiplier[output_channel], in RunPerChannel()
264 output_shift[output_channel]); in RunPerChannel()
269 output_channel)] = static_cast<int8_t>(acc); in RunPerChannel()
|
| /tflite-micro-3.4.0-2.7.6/tensorflow/lite/micro/kernels/vexriscv/doc/ |
| D | DepthwiseConv2D_int8.md | 173 create an int32 buffer of size output_channel as accumulators
|
