Searched refs:PadKernelMaxDimensionCount (Results 1 – 3 of 3) sorted by relevance
28 constexpr int PadKernelMaxDimensionCount() { return 4; } in PadKernelMaxDimensionCount() function43 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 …]
111 context, op_context.dims <= reference_ops::PadKernelMaxDimensionCount()); in Prepare()140 context, op_context.dims <= reference_ops::PadKernelMaxDimensionCount()); in Eval()
96 context, op_context.dims <= reference_ops::PadKernelMaxDimensionCount()); in Prepare()