| /tflite-micro-3.4.0-2.7.6/tensorflow/lite/kernels/internal/reference/ |
| D | space_to_depth.h | 38 const int input_depth = input_shape.Dims(3); in SpaceToDepth() local
52 TFLITE_DCHECK_EQ(input_depth * block_size * block_size, output_depth); in SpaceToDepth()
58 for (int in_d = 0; in_d < input_depth; ++in_d) { in SpaceToDepth()
61 input_depth; in SpaceToDepth()
|
| D | depthwiseconv_float.h | 48 const int input_depth = input_shape.Dims(3); in DepthwiseConv() local
53 TFLITE_DCHECK_EQ(output_depth, input_depth * depth_multiplier); in DepthwiseConv()
59 for (int ic = 0; ic < input_depth; ++ic) { in DepthwiseConv()
|
| D | depthwiseconv_uint8.h | 121 const int input_depth = input_shape.Dims(3); in Run() local
126 TFLITE_DCHECK_EQ(output_depth, input_depth * depth_multiplier); in Run()
132 for (int ic = 0; ic < input_depth; ++ic) { in Run()
208 const int input_depth = input_shape.Dims(3); in RunPerChannel() local
213 TFLITE_DCHECK_EQ(output_depth, input_depth * depth_multiplier); in RunPerChannel()
219 for (int in_channel = 0; in_channel < input_depth; ++in_channel) { in RunPerChannel()
|
| D | conv.h | 46 const int input_depth = MatchingDim(input_shape, 3, filter_shape, 3); in Conv() local
78 for (int in_channel = 0; in_channel < input_depth; ++in_channel) { in Conv()
129 const int input_depth = MatchingDim(input_shape, 3, filter_shape, 3); in Conv() local
161 for (int in_channel = 0; in_channel < input_depth; ++in_channel) { in Conv()
209 const int input_depth = MatchingDim(input_shape, 3, filter_shape, 3); in HybridConvPerChannel() local
229 for (int in_channel = 0; in_channel < input_depth; ++in_channel) { in HybridConvPerChannel()
|
| D | depth_to_space.h | 36 const int input_depth = input_shape.Dims(3); in DepthToSpace() local
50 TFLITE_DCHECK_EQ(input_depth, output_depth * block_size * block_size); in DepthToSpace()
|
| D | transpose_conv.h | 42 const int input_depth = MatchingDim(input_shape, 3, filter_shape, 3); in TransposeConv() local
70 for (int in_channel = 0; in_channel < input_depth; ++in_channel) { in TransposeConv()
132 const int input_depth = MatchingDim(input_shape, 3, filter_shape, 3); in TransposeConv() local
161 for (int in_channel = 0; in_channel < input_depth; ++in_channel) { in TransposeConv()
|
| /tflite-micro-3.4.0-2.7.6/tensorflow/lite/micro/kernels/ceva/ |
| D | depthwise_conv.cc | 68 const int input_depth = input_shape.Dims(3); in EvalFloat() local
94 input_height, input_width, input_depth, in EvalFloat()
95 &input_data[k * input_height * input_width * input_depth], in EvalFloat()
109 output_depth, input_depth); in EvalFloat()
144 const int input_depth = input_shape.Dims(3); in EvalQuantizedPerChannel() local
163 if ((input_depth * 4) > CEVA_TFLM_KERNELS_SCRATCH_SIZE_VAL) { in EvalQuantizedPerChannel()
165 CEVA_TFLM_KERNELS_SCRATCH_SIZE_VAL, (input_depth * 4)); in EvalQuantizedPerChannel()
176 data.per_channel_output_shift, input_height, input_width, input_depth, in EvalQuantizedPerChannel()
177 &input_data[k * input_height * input_width * input_depth], in EvalQuantizedPerChannel()
192 output_depth, input_depth); in EvalQuantizedPerChannel()
|
| D | conv.cc | 76 const int input_depth = MatchingDim(input_shape, 3, filter_shape, 3); in EvalQuantizedPerChannel() local
111 input_depth_Dims3, input_depth, in EvalQuantizedPerChannel()
127 output_depth, input_depth); in EvalQuantizedPerChannel()
163 const int input_depth = MatchingDim(input_shape, 3, filter_shape, 3); in EvalFloat() local
187 input_width, input_depth_Dims3, input_depth, in EvalFloat()
202 output_depth, input_depth); in EvalFloat()
|
| D | ceva_tflm_lib.h | 142 const int input_width, const int input_depth, const float* input_data,
173 const int input_depth, const int8_t* input_data, const int filter_height,
186 const int input_depth, const float* input_data, const int filter_height,
|
| /tflite-micro-3.4.0-2.7.6/tensorflow/lite/micro/kernels/xtensa/ |
| D | conv_hifi.cc | 49 const int input_depth = MatchingDim(input_shape, 3, filter_shape, 3); in ConvPrepareHifi() local
64 input_height, input_depth, filter_height, filter_width, stride_height, in ConvPrepareHifi()
101 const int input_depth = MatchingDim(input_shape, 3, filter_shape, 3); in ConvEvalHifi() local
129 input_width * input_depth]), in ConvEvalHifi()
131 input_depth, output_depth, input_offset, in ConvEvalHifi()
156 &input_data[batch * input_height * input_width * input_depth], in ConvEvalHifi()
158 bias_data, input_height, input_width, input_depth, in ConvEvalHifi()
|
| D | depthwise_conv_hifi.cc | 57 const int input_depth = input_shape.Dims(3); in DepthwiseConvPrepareHifi() local
74 input_height, input_width, input_depth, filter_height, filter_width, in DepthwiseConvPrepareHifi()
119 const int input_depth = input_shape.Dims(3); in DepthwiseConvEvalHifi() local
124 TFLITE_DCHECK_EQ(output_depth, input_depth * depth_multiplier); in DepthwiseConvEvalHifi()
144 &input_data[i * input_height * input_width * input_depth], in DepthwiseConvEvalHifi()
145 bias_data, input_height, input_width, input_depth, filter_height, in DepthwiseConvEvalHifi()
|
| D | conv_hifimini.cc | 51 const int input_depth = input_shape.Dims(3); in ConvEvalHifiMini() local
88 input_depth * 2 - in ConvEvalHifiMini()
91 for (int i = 0; i < input_depth; i += 2) { in ConvEvalHifiMini()
|
| D | depthwise_conv_hifimini.cc | 49 const int input_depth = input_shape.Dims(3); in DepthwiseConvEvalHifiMini() local
69 for (int in_channel = 0; in_channel < input_depth; ++in_channel) { in DepthwiseConvEvalHifiMini()
89 input_depth + in DepthwiseConvEvalHifiMini()
|
| /tflite-micro-3.4.0-2.7.6/tensorflow/lite/kernels/internal/reference/integer_ops/ |
| D | depthwise_conv.h | 53 const int input_depth = input_shape.Dims(3); in DepthwiseConvPerChannel() local
58 TFLITE_DCHECK_EQ(output_depth, input_depth * depth_multiplier); in DepthwiseConvPerChannel()
64 for (int in_channel = 0; in_channel < input_depth; ++in_channel) { in DepthwiseConvPerChannel()
150 const int input_depth = input_shape.Dims(3); in DepthwiseConvPerChannel() local
155 TFLITE_DCHECK_EQ(output_depth, input_depth * depth_multiplier); in DepthwiseConvPerChannel()
161 for (int in_channel = 0; in_channel < input_depth; ++in_channel) { in DepthwiseConvPerChannel()
233 const int input_depth = input_shape.Dims(3); in DepthwiseConvHybridPerChannel() local
239 TFLITE_DCHECK_EQ(output_depth, input_depth * depth_multiplier); in DepthwiseConvHybridPerChannel()
245 for (int in_channel = 0; in_channel < input_depth; ++in_channel) { in DepthwiseConvHybridPerChannel()
|
| D | conv.h | 51 const int input_depth = MatchingDim(input_shape, 3, filter_shape, 3); in ConvPerChannel() local
85 for (int in_channel = 0; in_channel < input_depth; ++in_channel) { in ConvPerChannel()
154 const int input_depth = MatchingDim(input_shape, 3, filter_shape, 3); in ConvPerChannel() local
188 for (int in_channel = 0; in_channel < input_depth; ++in_channel) { in ConvPerChannel()
|
| D | transpose_conv.h | 43 const int input_depth = MatchingDim(input_shape, 3, filter_shape, 3); in TransposeConv() local
69 for (int in_channel = 0; in_channel < input_depth; ++in_channel) { in TransposeConv()
142 const int input_depth = MatchingDim(input_shape, 3, filter_shape, 3); in TransposeConv() local
166 for (int in_channel = 0; in_channel < input_depth; ++in_channel) { in TransposeConv()
|
| /tflite-micro-3.4.0-2.7.6/tensorflow/lite/micro/kernels/vexriscv/ |
| D | depthwise_conv.cc | 66 const int input_depth = input_shape.Dims(3); in DepthwiseConvPerChannel() local
71 TFLITE_DCHECK_EQ(output_depth, input_depth * depth_multiplier); in DepthwiseConvPerChannel()
81 for (int begin_ch = 0; begin_ch < input_depth; in DepthwiseConvPerChannel()
88 std::min(input_depth, begin_ch + kChannelStep) - begin_ch; in DepthwiseConvPerChannel()
189 const int input_depth = input_shape.Dims(3); in DepthwiseConv() local
194 TFLITE_DCHECK_EQ(output_depth, input_depth * depth_multiplier); in DepthwiseConv()
204 for (int begin_ch = 0; begin_ch < input_depth; in DepthwiseConv()
211 std::min(input_depth, begin_ch + kChannelStep) - begin_ch; in DepthwiseConv()
|
| /tflite-micro-3.4.0-2.7.6/tensorflow/lite/micro/kernels/cmsis_nn/ |
| D | conv.cc | 168 const int input_depth = MatchingDim(input_shape, 3, filter_shape, 3); in EvalQuantizedPerChannel() local
180 input_dims.c = input_depth; in EvalQuantizedPerChannel()
187 filter_dims.c = input_depth; in EvalQuantizedPerChannel()
|
| /tflite-micro-3.4.0-2.7.6/tensorflow/lite/micro/kernels/arc_mli/ |
| D | depthwise_conv_slicing_test.cc | 63 int input_depth = tensors[0].dims->data[3]; in ValidateDepthwiseConvGoldens() local
65 int depth_mul = output_depth / input_depth; in ValidateDepthwiseConvGoldens()
|
| /tflite-micro-3.4.0-2.7.6/tensorflow/lite/micro/kernels/ |
| D | depthwise_conv_test.cc | 57 int input_depth = tensors[0].dims->data[3]; in ValidateDepthwiseConvGoldens() local
59 int depth_mul = output_depth / input_depth; in ValidateDepthwiseConvGoldens()
|
