Home
last modified time | relevance | path

Searched defs:input_depth (Results 1 – 19 of 19) sorted by relevance

/tflite-micro-3.4.0-2.7.6/tensorflow/lite/micro/kernels/xtensa/
Dconv_hifi.cc49 const int input_depth = MatchingDim(input_shape, 3, filter_shape, 3); in ConvPrepareHifi() local
101 const int input_depth = MatchingDim(input_shape, 3, filter_shape, 3); in ConvEvalHifi() local
Ddepthwise_conv_hifi.cc57 const int input_depth = input_shape.Dims(3); in DepthwiseConvPrepareHifi() local
119 const int input_depth = input_shape.Dims(3); in DepthwiseConvEvalHifi() local
Dconv_hifimini.cc51 const int input_depth = input_shape.Dims(3); in ConvEvalHifiMini() local
Ddepthwise_conv_hifimini.cc49 const int input_depth = input_shape.Dims(3); in DepthwiseConvEvalHifiMini() local
/tflite-micro-3.4.0-2.7.6/tensorflow/lite/kernels/internal/reference/
Dspace_to_depth.h38 const int input_depth = input_shape.Dims(3); in SpaceToDepth() local
Ddepth_to_space.h36 const int input_depth = input_shape.Dims(3); in DepthToSpace() local
Dconv.h46 const int input_depth = MatchingDim(input_shape, 3, filter_shape, 3); in Conv() local
129 const int input_depth = MatchingDim(input_shape, 3, filter_shape, 3); in Conv() local
209 const int input_depth = MatchingDim(input_shape, 3, filter_shape, 3); in HybridConvPerChannel() local
Dtranspose_conv.h42 const int input_depth = MatchingDim(input_shape, 3, filter_shape, 3); in TransposeConv() local
132 const int input_depth = MatchingDim(input_shape, 3, filter_shape, 3); in TransposeConv() local
Ddepthwiseconv_float.h48 const int input_depth = input_shape.Dims(3); in DepthwiseConv() local
Ddepthwiseconv_uint8.h121 const int input_depth = input_shape.Dims(3); in Run() local
208 const int input_depth = input_shape.Dims(3); in RunPerChannel() local
/tflite-micro-3.4.0-2.7.6/tensorflow/lite/kernels/internal/reference/integer_ops/
Dconv.h51 const int input_depth = MatchingDim(input_shape, 3, filter_shape, 3); in ConvPerChannel() local
154 const int input_depth = MatchingDim(input_shape, 3, filter_shape, 3); in ConvPerChannel() local
Ddepthwise_conv.h53 const int input_depth = input_shape.Dims(3); in DepthwiseConvPerChannel() local
150 const int input_depth = input_shape.Dims(3); in DepthwiseConvPerChannel() local
233 const int input_depth = input_shape.Dims(3); in DepthwiseConvHybridPerChannel() local
Dtranspose_conv.h43 const int input_depth = MatchingDim(input_shape, 3, filter_shape, 3); in TransposeConv() local
142 const int input_depth = MatchingDim(input_shape, 3, filter_shape, 3); in TransposeConv() local
/tflite-micro-3.4.0-2.7.6/tensorflow/lite/micro/kernels/ceva/
Ddepthwise_conv.cc68 const int input_depth = input_shape.Dims(3); in EvalFloat() local
144 const int input_depth = input_shape.Dims(3); in EvalQuantizedPerChannel() local
Dconv.cc76 const int input_depth = MatchingDim(input_shape, 3, filter_shape, 3); in EvalQuantizedPerChannel() local
163 const int input_depth = MatchingDim(input_shape, 3, filter_shape, 3); in EvalFloat() local
/tflite-micro-3.4.0-2.7.6/tensorflow/lite/micro/kernels/cmsis_nn/
Dconv.cc168 const int input_depth = MatchingDim(input_shape, 3, filter_shape, 3); in EvalQuantizedPerChannel() local
/tflite-micro-3.4.0-2.7.6/tensorflow/lite/micro/kernels/vexriscv/
Ddepthwise_conv.cc66 const int input_depth = input_shape.Dims(3); in DepthwiseConvPerChannel() local
189 const int input_depth = input_shape.Dims(3); in DepthwiseConv() local
/tflite-micro-3.4.0-2.7.6/tensorflow/lite/micro/kernels/arc_mli/
Ddepthwise_conv_slicing_test.cc63 int input_depth = tensors[0].dims->data[3]; in ValidateDepthwiseConvGoldens() local
/tflite-micro-3.4.0-2.7.6/tensorflow/lite/micro/kernels/
Ddepthwise_conv_test.cc57 int input_depth = tensors[0].dims->data[3]; in ValidateDepthwiseConvGoldens() local