Home
last modified time | relevance | path

Searched refs:TFLITE_DCHECK_GE (Results 1 – 25 of 32) sorted by relevance

12

/external/tensorflow/tensorflow/lite/kernels/internal/reference/
Dgather.h35 TFLITE_DCHECK_GE(axis, 0); in Gather()
52 TFLITE_DCHECK_GE(coords_data[i], 0); in Gather()
Dfully_connected.h75 TFLITE_DCHECK_GE(filter_shape.DimensionsCount(), 2); in FullyConnected()
76 TFLITE_DCHECK_GE(output_shape.DimensionsCount(), 1); in FullyConnected()
176 TFLITE_DCHECK_GE(input_shape.DimensionsCount(), 1); in ShuffledFullyConnected()
177 TFLITE_DCHECK_GE(weights_shape.DimensionsCount(), 2); in ShuffledFullyConnected()
178 TFLITE_DCHECK_GE(output_shape.DimensionsCount(), 1); in ShuffledFullyConnected()
Dtanh.h55 TFLITE_DCHECK_GE(input_left_shift, 0); in Tanh()
118 TFLITE_DCHECK_GE(output_val_s32, 0); in Tanh()
Dsub.h418 TFLITE_DCHECK_GE(params.input1_offset, -1 * int8_max_value); in SubElementwise()
419 TFLITE_DCHECK_GE(params.input2_offset, -1 * int8_max_value); in SubElementwise()
473 TFLITE_DCHECK_GE(params.input1_offset, -1 * int8_max_value); in Sub()
474 TFLITE_DCHECK_GE(params.input2_offset, -1 * int8_max_value); in Sub()
Dbatch_to_space_nd.h48 TFLITE_DCHECK_GE(unextended_input1_shape.DimensionsCount(), 3); in BatchToSpaceND()
Dspace_to_batch_nd.h50 TFLITE_DCHECK_GE(unextended_input1_shape.DimensionsCount(), 3); in SpaceToBatchND()
Dpooling.h245 TFLITE_DCHECK_GE(params.quantized_activation_min, 0); in MaxPool()
/external/tensorflow/tensorflow/lite/kernels/internal/
Dcompatibility.h34 #ifndef TFLITE_DCHECK_GE
35 #define TFLITE_DCHECK_GE(x, y) ((x) >= (y)) ? (void)0 : TFLITE_ASSERT_FALSE macro
Dtypes.h214 TFLITE_DCHECK_GE(i, 0); in Dims()
219 TFLITE_DCHECK_GE(i, 0); in SetDim()
357 TFLITE_DCHECK_GE(dims[idx], current_val); in NextIndex()
Dcommon.h494 TFLITE_DCHECK_GE(multiplier_int32_t, 0); in DownScaleInt32ToInt16Multiplier()
643 TFLITE_DCHECK_GE(input, 0); in GetInvSqrtQuantizedMultiplierExp()
666 TFLITE_DCHECK_GE(input, (1 << 27)); in GetInvSqrtQuantizedMultiplierExp()
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/integer_ops/
Ddepthwise_conv_hybrid.h70 TFLITE_DCHECK_GE(kAccBufferMaxSize, output_depth); in DepthwiseConvHybridGeneral()
76 TFLITE_DCHECK_GE(kOutputPixelsInAccBuffer, 1); in DepthwiseConvHybridGeneral()
152 TFLITE_DCHECK_GE(thread_start, 0); in DepthwiseConvHybridGeneral()
159 TFLITE_DCHECK_GE(thread_start, 0); in DepthwiseConvHybridGeneral()
271 TFLITE_DCHECK_GE(dilation_width_factor, 1); in DepthwiseConvHybridWithRounding()
272 TFLITE_DCHECK_GE(dilation_height_factor, 1); in DepthwiseConvHybridWithRounding()
Dfully_connected.h45 TFLITE_DCHECK_GE(filter_shape.DimensionsCount(), 2); in FullyConnected()
46 TFLITE_DCHECK_GE(output_shape.DimensionsCount(), 1); in FullyConnected()
Ddepthwise_conv.h1641 TFLITE_DCHECK_GE(kAccBufferMaxSize, output_depth);
1647 TFLITE_DCHECK_GE(kOutputPixelsInAccBuffer, 1);
1723 TFLITE_DCHECK_GE(thread_start, 0);
1730 TFLITE_DCHECK_GE(thread_start, 0);
1806 TFLITE_DCHECK_GE(dilation_width_factor, 1);
1807 TFLITE_DCHECK_GE(dilation_height_factor, 1);
/external/tensorflow/tensorflow/lite/kernels/internal/reference/integer_ops/
Dfully_connected.h36 TFLITE_DCHECK_GE(filter_shape.DimensionsCount(), 2); in FullyConnected()
76 TFLITE_DCHECK_GE(filter_shape.DimensionsCount(), 2); in FullyConnected()
Dadd.h31 TFLITE_DCHECK_GE(-params.input1_offset, std::numeric_limits<int8_t>::min()); in CheckArithmeticParams()
32 TFLITE_DCHECK_GE(-params.input2_offset, std::numeric_limits<int8_t>::min()); in CheckArithmeticParams()
Dlogistic.h72 TFLITE_DCHECK_GE(input_left_shift, 0); in Logistic()
Dpooling.h87 TFLITE_DCHECK_GE(params.quantized_activation_min, in MaxPool()
203 TFLITE_DCHECK_GE(params.quantized_activation_min, in MaxPool()
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/
Ddepthwiseconv_uint8_transitional.h131 TFLITE_DCHECK_GE(input_offset, -255);
204 TFLITE_DCHECK_GE(input_offset, -255);
309 TFLITE_DCHECK_GE(input_offset, -255);
665 TFLITE_DCHECK_GE(workspace_height_stride, 4 * width_overall_micro_repeats);
720 TFLITE_DCHECK_GE(copy_size, input_height_stride - 1);
778 TFLITE_DCHECK_GE(depth_micro_repeats, 0);
960 TFLITE_DCHECK_GE(workspace_height_stride, 4 * width_overall_micro_repeats);
1014 TFLITE_DCHECK_GE(copy_size, input_height_stride - 1);
1273 TFLITE_DCHECK_GE(depth_micro_repeats, 0);
1914 TFLITE_DCHECK_GE(workspace_height_stride, 4 * width_overall_micro_repeats);
[all …]
Dlegacy_optimized_ops.h225 TFLITE_DCHECK_GE(dilation_width_factor, 1); in LegacyDepthwiseConvWithRounding()
226 TFLITE_DCHECK_GE(dilation_height_factor, 1); in LegacyDepthwiseConvWithRounding()
780 TFLITE_DCHECK_GE(input_shape.DimensionsCount(), 1); in LegacyFullyConnectedAsGEMVWorkerImpl()
781 TFLITE_DCHECK_GE(filter_shape.DimensionsCount(), 2); in LegacyFullyConnectedAsGEMVWorkerImpl()
782 TFLITE_DCHECK_GE(output_shape.DimensionsCount(), 1); in LegacyFullyConnectedAsGEMVWorkerImpl()
795 TFLITE_DCHECK_GE(row_end - row_start, kPeel); in LegacyFullyConnectedAsGEMVWorkerImpl()
1104 TFLITE_DCHECK_GE(filter_shape.DimensionsCount(), 2); in FullyConnected()
1105 TFLITE_DCHECK_GE(output_shape.DimensionsCount(), 1); in FullyConnected()
1163 TFLITE_DCHECK_GE(input_shape.DimensionsCount(), 1); in GEMVForLstmCell()
1164 TFLITE_DCHECK_GE(weights_shape.DimensionsCount(), 2); in GEMVForLstmCell()
[all …]
Ddepthwiseconv_multithread.h94 TFLITE_DCHECK_GE(thread_count, 2); in MultithreadAlongBatches()
Ddepthwiseconv_float.h937 TFLITE_DCHECK_GE(kAccBufferMaxSize, output_depth);
943 TFLITE_DCHECK_GE(kOutputPixelsInAccBuffer, 1);
1012 TFLITE_DCHECK_GE(thread_start, 0);
1020 TFLITE_DCHECK_GE(thread_start, 0);
Ddepthwiseconv_uint8_3x3_filter.h5684 TFLITE_DCHECK_GE(thread_start, 0);
5690 TFLITE_DCHECK_GE(thread_start, 0);
6081 TFLITE_DCHECK_GE(depth_micro_repeats, 0);
6751 TFLITE_DCHECK_GE(workspace_height_stride, 4 * width_overall_micro_repeats);
6805 TFLITE_DCHECK_GE(copy_size, input_height_stride - 1);
7157 TFLITE_DCHECK_GE(workspace_height_stride, 4 * width_overall_micro_repeats);
7175 TFLITE_DCHECK_GE(copy_size, input_height_stride - 1);
12953 TFLITE_DCHECK_GE(thread_start, 0);
12959 TFLITE_DCHECK_GE(thread_start, 0);
13203 TFLITE_DCHECK_GE(width_overall_macro_count, width_macro_count);
[all …]
Doptimized_ops.h382 TFLITE_DCHECK_GE(filter_shape.DimensionsCount(), 2); in FullyConnected()
383 TFLITE_DCHECK_GE(output_shape.DimensionsCount(), 1); in FullyConnected()
447 TFLITE_DCHECK_GE(filter_shape.DimensionsCount(), 2); in FullyConnected()
448 TFLITE_DCHECK_GE(output_shape.DimensionsCount(), 1); in FullyConnected()
825 TFLITE_DCHECK_GE(input_shape.DimensionsCount(), 1); in ShuffledFullyConnected()
826 TFLITE_DCHECK_GE(weights_shape.DimensionsCount(), 2); in ShuffledFullyConnected()
827 TFLITE_DCHECK_GE(output_shape.DimensionsCount(), 1); in ShuffledFullyConnected()
1570 TFLITE_DCHECK_GE(input_zero_point, 0); in Conv()
1579 TFLITE_DCHECK_GE(input_zero_point, 0); in Conv()
2811 TFLITE_DCHECK_GE(weights_shape.DimensionsCount(), 2); in LstmCell()
[all …]
/external/tensorflow/tensorflow/lite/kernels/
Dcpu_backend_gemm_custom_gemv.h163 TFLITE_DCHECK_GE(lhs_params.rows, Impl::kKernelRows); in CustomGemv()
330 TFLITE_DCHECK_GE(row_end - row_start, kKernelRows);
467 TFLITE_DCHECK_GE(back, 1);
642 TFLITE_DCHECK_GE(row_end - row_start, kKernelRows);
715 TFLITE_DCHECK_GE(back, 1);
/external/tensorflow/tensorflow/lite/micro/kernels/ceva/
Dtypes.h377 TFLITE_DCHECK_GE(i, 0);
382 TFLITE_DCHECK_GE(i, 0);
520 TFLITE_DCHECK_GE(dims[idx], current_val);

12