Home
last modified time | relevance | path

Searched refs:PadKernelMaxDimensionCount (Results 1 – 2 of 2) sorted by relevance

/AliOS-Things-master/components/ai_agent/src/engine/tflite-micro/tensorflow/lite/kernels/internal/reference/
A Dpad.h28 constexpr int PadKernelMaxDimensionCount() { return 5; } in PadKernelMaxDimensionCount() function
43 RuntimeShape::ExtendedShape(PadKernelMaxDimensionCount(), input_shape); in PadImpl()
45 RuntimeShape::ExtendedShape(PadKernelMaxDimensionCount(), output_shape); in PadImpl()
46 TFLITE_DCHECK_LE(op_params.left_padding_count, PadKernelMaxDimensionCount()); in PadImpl()
47 TFLITE_DCHECK_LE(op_params.right_padding_count, PadKernelMaxDimensionCount()); in PadImpl()
51 int left_padding_copy[PadKernelMaxDimensionCount()]; in PadImpl()
52 for (int i = 0; i < PadKernelMaxDimensionCount(); i++) { in PadImpl()
56 left_padding_copy[i + PadKernelMaxDimensionCount() - in PadImpl()
59 int right_padding_copy[PadKernelMaxDimensionCount()]; in PadImpl()
60 for (int i = 0; i < PadKernelMaxDimensionCount(); i++) { in PadImpl()
[all …]
/AliOS-Things-master/components/ai_agent/src/engine/tflite-micro/tensorflow/lite/micro/kernels/
A Dpad.cc65 reference_ops::PadKernelMaxDimensionCount()); in Prepare()

Completed in 3 milliseconds