Home
last modified time | relevance | path

Searched refs:output_depth (Results 1 – 25 of 29) sorted by relevance

12

/tflite-micro-3.4.0-2.7.6/tensorflow/lite/kernels/internal/reference/integer_ops/
Dfully_connected.h42 const int output_depth = output_shape.Dims(1); in FullyConnected() local
43 TFLITE_DCHECK_LE(output_depth, filter_shape.Dims(filter_dim_count - 2)); in FullyConnected()
46 for (int out_c = 0; out_c < output_depth; ++out_c) { in FullyConnected()
60 output_data[out_c + output_depth * b] = static_cast<int8_t>(acc); in FullyConnected()
82 const int output_depth = output_shape.Dims(1); in FullyConnected() local
83 TFLITE_DCHECK_LE(output_depth, filter_shape.Dims(filter_dim_count - 2)); in FullyConnected()
86 for (int out_c = 0; out_c < output_depth; ++out_c) { in FullyConnected()
100 output_data[out_c + output_depth * b] = static_cast<int16_t>(acc_scaled); in FullyConnected()
Dtranspose_conv.h44 const int output_depth = MatchingDim(filter_shape, 0, output_shape, 3); in TransposeConv() local
46 TFLITE_DCHECK_EQ(bias_shape.FlatSize(), output_depth); in TransposeConv()
75 for (int out_channel = 0; out_channel < output_depth; in TransposeConv()
103 for (int out_channel = 0; out_channel < output_depth; ++out_channel) { in TransposeConv()
143 const int output_depth = MatchingDim(filter_shape, 0, output_shape, 3); in TransposeConv() local
145 TFLITE_DCHECK_EQ(bias_shape.FlatSize(), output_depth); in TransposeConv()
172 for (int out_channel = 0; out_channel < output_depth; in TransposeConv()
200 for (int out_channel = 0; out_channel < output_depth; ++out_channel) { in TransposeConv()
Ddepthwise_conv.h50 const int output_depth = MatchingDim(filter_shape, 3, output_shape, 3); in DepthwiseConvPerChannel() local
58 TFLITE_DCHECK_EQ(output_depth, input_depth * depth_multiplier); in DepthwiseConvPerChannel()
59 TFLITE_DCHECK_EQ(bias_shape.FlatSize(), output_depth); in DepthwiseConvPerChannel()
147 const int output_depth = MatchingDim(filter_shape, 3, output_shape, 3); in DepthwiseConvPerChannel() local
155 TFLITE_DCHECK_EQ(output_depth, input_depth * depth_multiplier); in DepthwiseConvPerChannel()
156 TFLITE_DCHECK_EQ(bias_shape.FlatSize(), output_depth); in DepthwiseConvPerChannel()
230 const int output_depth = MatchingDim(filter_shape, 3, output_shape, 3); in DepthwiseConvHybridPerChannel() local
239 TFLITE_DCHECK_EQ(output_depth, input_depth * depth_multiplier); in DepthwiseConvHybridPerChannel()
240 TFLITE_DCHECK_EQ(bias_depth, output_depth); in DepthwiseConvHybridPerChannel()
Dconv.h52 const int output_depth = MatchingDim(filter_shape, 0, output_shape, 3); in ConvPerChannel() local
54 TFLITE_DCHECK_EQ(bias_shape.FlatSize(), output_depth); in ConvPerChannel()
69 for (int out_channel = 0; out_channel < output_depth; ++out_channel) { in ConvPerChannel()
155 const int output_depth = MatchingDim(filter_shape, 0, output_shape, 3); in ConvPerChannel() local
157 TFLITE_DCHECK_EQ(bias_shape.FlatSize(), output_depth); in ConvPerChannel()
172 for (int out_channel = 0; out_channel < output_depth; ++out_channel) { in ConvPerChannel()
Dmean.h40 const int output_depth = output_shape.Dims(3); in Mean() local
55 for (int out_d = 0; out_d < output_depth; ++out_d) { in Mean()
/tflite-micro-3.4.0-2.7.6/tensorflow/lite/kernels/internal/reference/
Dfully_connected.h42 const int output_depth = MatchingDim(weights_shape, weights_dims_count - 2, in FullyConnected() local
46 for (int out_c = 0; out_c < output_depth; ++out_c) { in FullyConnected()
56 output_data[out_c + output_depth * b] = ActivationFunctionWithMinMax( in FullyConnected()
87 const int output_depth = MatchingDim(filter_shape, filter_dim_count - 2, in FullyConnected() local
91 for (int out_c = 0; out_c < output_depth; ++out_c) { in FullyConnected()
105 output_data[out_c + output_depth * b] = static_cast<uint8_t>(acc); in FullyConnected()
134 const int output_depth = MatchingDim(filter_shape, filter_dim_count - 2, in FullyConnected() local
138 for (int out_c = 0; out_c < output_depth; ++out_c) { in FullyConnected()
159 output_data[out_c + output_depth * b] = accum; in FullyConnected()
187 const int output_depth = MatchingDim(weights_shape, weights_dim_count - 2, in ShuffledFullyConnected() local
[all …]
Ddepth_to_space.h41 const int output_depth = output_shape.Dims(3); in DepthToSpace() local
50 TFLITE_DCHECK_EQ(input_depth, output_depth * block_size * block_size); in DepthToSpace()
56 for (int out_d = 0; out_d < output_depth; ++out_d) { in DepthToSpace()
59 output_depth; in DepthToSpace()
Dtranspose_conv.h43 const int output_depth = MatchingDim(filter_shape, 0, output_shape, 3); in TransposeConv() local
51 TFLITE_DCHECK_EQ(bias_shape.FlatSize(), output_depth); in TransposeConv()
76 for (int out_channel = 0; out_channel < output_depth; in TransposeConv()
104 for (int out_channel = 0; out_channel < output_depth; ++out_channel) { in TransposeConv()
133 const int output_depth = MatchingDim(filter_shape, 0, output_shape, 3); in TransposeConv() local
149 TFLITE_DCHECK_EQ(bias_shape.FlatSize(), output_depth); in TransposeConv()
167 for (int out_channel = 0; out_channel < output_depth; in TransposeConv()
195 for (int out_channel = 0; out_channel < output_depth; ++out_channel) { in TransposeConv()
Dconv.h47 const int output_depth = MatchingDim(filter_shape, 0, output_shape, 3); in Conv() local
49 TFLITE_DCHECK_EQ(bias_shape.FlatSize(), output_depth); in Conv()
62 for (int out_channel = 0; out_channel < output_depth; ++out_channel) { in Conv()
130 const int output_depth = MatchingDim(filter_shape, 0, output_shape, 3); in Conv() local
132 TFLITE_DCHECK_EQ(bias_shape.FlatSize(), output_depth); in Conv()
145 for (int out_channel = 0; out_channel < output_depth; ++out_channel) { in Conv()
210 const int output_depth = MatchingDim(filter_shape, 0, output_shape, 3); in HybridConvPerChannel() local
212 TFLITE_DCHECK_EQ(bias_shape.FlatSize(), output_depth); in HybridConvPerChannel()
223 for (int out_channel = 0; out_channel < output_depth; ++out_channel) { in HybridConvPerChannel()
Ddepthwiseconv_float.h45 const int output_depth = MatchingDim(filter_shape, 3, output_shape, 3); in DepthwiseConv() local
53 TFLITE_DCHECK_EQ(output_depth, input_depth * depth_multiplier); in DepthwiseConv()
54 TFLITE_DCHECK_EQ(bias_shape.FlatSize(), output_depth); in DepthwiseConv()
Ddepthwiseconv_uint8.h118 const int output_depth = MatchingDim(filter_shape, 3, output_shape, 3); in Run() local
126 TFLITE_DCHECK_EQ(output_depth, input_depth * depth_multiplier); in Run()
127 TFLITE_DCHECK_EQ(bias_shape.FlatSize(), output_depth); in Run()
205 const int output_depth = MatchingDim(filter_shape, 3, output_shape, 3); in RunPerChannel() local
213 TFLITE_DCHECK_EQ(output_depth, input_depth * depth_multiplier); in RunPerChannel()
214 TFLITE_DCHECK_EQ(bias_shape.FlatSize(), output_depth); in RunPerChannel()
Dspace_to_depth.h43 const int output_depth = output_shape.Dims(3); in SpaceToDepth() local
52 TFLITE_DCHECK_EQ(input_depth * block_size * block_size, output_depth); in SpaceToDepth()
Dpad.h73 const int output_depth = ext_output_shape.Dims(4); in PadImpl() local
95 for (int out_d = 0; out_d < output_depth; ++out_d) { in PadImpl()
105 out_d >= output_depth - right_d_padding) { in PadImpl()
Dreduce.h288 const int output_depth = output_shape.Dims(3); in Mean() local
300 for (int out_d = 0; out_d < output_depth; ++out_d) { in Mean()
332 const int output_depth = output_shape.Dims(3); in Mean() local
356 for (int out_d = 0; out_d < output_depth; ++out_d) { in Mean()
/tflite-micro-3.4.0-2.7.6/tensorflow/lite/micro/kernels/ceva/
Ddepthwise_conv.cc76 const int output_depth = output_shape.Dims(3); in EvalFloat() local
97 output_height, output_width, output_depth, in EvalFloat()
98 &output_data[k * output_height * output_width * output_depth], in EvalFloat()
109 output_depth, input_depth); in EvalFloat()
152 const int output_depth = output_shape.Dims(3); in EvalQuantizedPerChannel() local
179 output_height, output_width, output_depth, in EvalQuantizedPerChannel()
180 &output_data[k * output_height * output_width * output_depth], in EvalQuantizedPerChannel()
192 output_depth, input_depth); in EvalQuantizedPerChannel()
Dconv.cc77 const int output_depth = MatchingDim(filter_shape, 0, output_shape, 3); in EvalQuantizedPerChannel() local
115 output_height, output_width, output_depth_Dims3, output_depth, in EvalQuantizedPerChannel()
127 output_depth, input_depth); in EvalQuantizedPerChannel()
164 const int output_depth = MatchingDim(filter_shape, 0, output_shape, 3); in EvalFloat() local
190 output_height, output_width, output_depth_Dims3, output_depth, in EvalFloat()
202 output_depth, input_depth); in EvalFloat()
Dfully_connected.cc106 int output_depth = in EvalQuantizedInt8CEVA() local
129 batches, output_depth, accum_depth); in EvalQuantizedInt8CEVA()
177 int output_depth = in EvalFloatCEVA() local
196 batches, output_depth, accum_depth); in EvalFloatCEVA()
/tflite-micro-3.4.0-2.7.6/tensorflow/lite/micro/kernels/xtensa/
Dfully_connected.cc50 const int output_depth = output_shape.Dims(1); in FullyConnected() local
61 for (int out_c = 0; out_c < output_depth; ++out_c) { in FullyConnected()
112 output_data[out_c + output_depth * b] = in FullyConnected()
208 const int output_depth = output_shape.Dims(1); in EvalQuantizedInt8()
219 (tflite::micro::GetTensorData<int8_t>(output) + b * output_depth), in EvalQuantizedInt8()
223 output_depth, op_params.input_offset, op_params.output_multiplier, in EvalQuantizedInt8()
232 data.output_activation_max, num_batches * output_depth), in EvalQuantizedInt8()
Ddepthwise_conv_hifi.cc116 const int output_depth = MatchingDim(filter_shape, 3, output_shape, 3); in DepthwiseConvEvalHifi() local
124 TFLITE_DCHECK_EQ(output_depth, input_depth * depth_multiplier); in DepthwiseConvEvalHifi()
125 TFLITE_DCHECK_EQ(bias_shape.FlatSize(), output_depth); in DepthwiseConvEvalHifi()
142 &output_data[i * output_height * output_width * output_depth], in DepthwiseConvEvalHifi()
156 int out_length = batches * output_height * output_width * output_depth; in DepthwiseConvEvalHifi()
Dconv_hifi.cc102 const int output_depth = MatchingDim(filter_shape, 0, output_shape, 3); in ConvEvalHifi() local
116 int out_length = output_height * output_width * output_depth; in ConvEvalHifi()
131 input_depth, output_depth, input_offset, in ConvEvalHifi()
159 filter_height, filter_width, output_depth, stride_width, in ConvEvalHifi()
Dconv_hifimini.cc59 const int output_depth = output_shape.Dims(3); in ConvEvalHifiMini() local
71 for (int out_channel = 0; out_channel < output_depth; ++out_channel) { in ConvEvalHifiMini()
153 output_depth + in ConvEvalHifiMini()
/tflite-micro-3.4.0-2.7.6/tensorflow/lite/micro/kernels/cmsis_nn/
Ddepthwise_conv.cc115 const int output_depth = MatchingDim(output_shape, 3, filter_shape, 3); in Prepare() local
128 filter_dims.c = output_depth; in Prepare()
134 output_dims.c = output_depth; in Prepare()
190 const int output_depth = MatchingDim(filter_shape, 3, output_shape, 3); in EvalQuantizedPerChannel() local
193 TFLITE_DCHECK_EQ(bias_shape.FlatSize(), output_depth); in EvalQuantizedPerChannel()
206 filter_dims.c = output_depth; in EvalQuantizedPerChannel()
212 bias_dims.c = output_depth; in EvalQuantizedPerChannel()
218 output_dims.c = output_depth; in EvalQuantizedPerChannel()
Dfully_connected.cc112 const int output_depth = output_shape.Dims(1); in EvalQuantizedInt8() local
139 filter_dims.c = output_depth; in EvalQuantizedInt8()
145 bias_dims.c = output_depth; in EvalQuantizedInt8()
151 output_dims.c = output_depth; in EvalQuantizedInt8()
Dconv.cc169 const int output_depth = MatchingDim(filter_shape, 0, output_shape, 3); in EvalQuantizedPerChannel() local
171 TFLITE_DCHECK_EQ(bias_shape.FlatSize(), output_depth); in EvalQuantizedPerChannel()
184 filter_dims.n = output_depth; in EvalQuantizedPerChannel()
194 bias_dims.c = output_depth; in EvalQuantizedPerChannel()
201 output_dims.c = output_depth; in EvalQuantizedPerChannel()
/tflite-micro-3.4.0-2.7.6/tensorflow/lite/micro/kernels/vexriscv/
Ddepthwise_conv.cc63 const int output_depth = MatchingDim(filter_shape, 3, output_shape, 3); in DepthwiseConvPerChannel() local
71 TFLITE_DCHECK_EQ(output_depth, input_depth * depth_multiplier); in DepthwiseConvPerChannel()
72 TFLITE_DCHECK_EQ(bias_shape.FlatSize(), output_depth); in DepthwiseConvPerChannel()
186 const int output_depth = MatchingDim(filter_shape, 3, output_shape, 3); in DepthwiseConv() local
194 TFLITE_DCHECK_EQ(output_depth, input_depth * depth_multiplier); in DepthwiseConv()
195 TFLITE_DCHECK_EQ(bias_shape.FlatSize(), output_depth); in DepthwiseConv()

12