Home
last modified time | relevance | path

Searched refs:TFLITE_CHECK_EQ (Results 1 – 11 of 11) sorted by relevance

/AliOS-Things-master/components/ai_agent/src/engine/tflite-micro/tensorflow/lite/kernels/internal/reference/integer_ops/
A Dmean.h31 TFLITE_CHECK_EQ(unextended_input_shape.DimensionsCount(), 4); in Mean()
45 TFLITE_CHECK_EQ(op_params.axis_count, 2); in Mean()
48 TFLITE_CHECK_EQ(output_height, 1); in Mean()
49 TFLITE_CHECK_EQ(output_width, 1); in Mean()
/AliOS-Things-master/components/ai_agent/src/engine/tflite-micro/tensorflow/lite/kernels/internal/optimized/integer_ops/
A Dmean.h41 TFLITE_CHECK_EQ(op_params.axis_count, 2); in MeanImpl()
44 TFLITE_CHECK_EQ(output_height, 1); in MeanImpl()
45 TFLITE_CHECK_EQ(output_width, 1); in MeanImpl()
187 TFLITE_CHECK_EQ(unextended_input_shape.DimensionsCount(), 4); in Mean()
197 TFLITE_CHECK_EQ(op_params.axis_count, 2); in Mean()
200 TFLITE_CHECK_EQ(output_height, 1); in Mean()
201 TFLITE_CHECK_EQ(output_width, 1); in Mean()
/AliOS-Things-master/components/ai_agent/src/engine/tflite-micro/tensorflow/lite/kernels/internal/reference/
A Dreduce.h225 TFLITE_CHECK_EQ(unextended_input_shape.DimensionsCount(), 4); in Mean()
240 TFLITE_CHECK_EQ(op_params.axis_count, 2); in Mean()
243 TFLITE_CHECK_EQ(output_height, 1); in Mean()
244 TFLITE_CHECK_EQ(output_width, 1); in Mean()
270 TFLITE_CHECK_EQ(unextended_input_shape.DimensionsCount(), 4); in Mean()
284 TFLITE_CHECK_EQ(op_params.axis_count, 2); in Mean()
287 TFLITE_CHECK_EQ(output_height, 1); in Mean()
288 TFLITE_CHECK_EQ(output_width, 1); in Mean()
A Dlegacy_reference_ops.h914 TFLITE_CHECK_EQ(p->start_indices_count, p->stop_indices_count); in StridedSliceReverseIndices()
915 TFLITE_CHECK_EQ(p->stop_indices_count, p->strides_count); in StridedSliceReverseIndices()
/AliOS-Things-master/components/ai_agent/src/engine/tflite-micro/tensorflow/lite/kernels/internal/
A Dcompatibility.h55 #ifndef TFLITE_CHECK_EQ
56 #define TFLITE_CHECK_EQ(x, y) ((x) == (y)) ? (void)0 : TFLITE_ABORT macro
A Dquantization_util.h43 TFLITE_CHECK_EQ(rmin, 0.); in ChooseQuantizationParams()
44 TFLITE_CHECK_EQ(rmax, 0.); in ChooseQuantizationParams()
A Dstrided_slice_logic.h41 TFLITE_CHECK_EQ(p->start_indices_count, p->stop_indices_count); in StridedSlicePadIndices()
42 TFLITE_CHECK_EQ(p->stop_indices_count, p->strides_count); in StridedSlicePadIndices()
A Dquantization_util.cc114 TFLITE_CHECK_EQ(8, sizeof(double)); in IntegerFrExp()
A Dtypes.h502 TFLITE_CHECK_EQ(size_1, size_2); in MatchingElementsSize()
512 TFLITE_CHECK_EQ(size_1, size_2); in MatchingElementsSize()
513 TFLITE_CHECK_EQ(size_2, size_3); in MatchingElementsSize()
/AliOS-Things-master/components/ai_agent/src/engine/tflite-micro/tensorflow/lite/kernels/internal/optimized/
A Doptimized_ops.h984 TFLITE_CHECK_EQ(op_params.axis_count, 2); in MeanImpl()
987 TFLITE_CHECK_EQ(output_height, 1); in MeanImpl()
988 TFLITE_CHECK_EQ(output_width, 1); in MeanImpl()
1135 TFLITE_CHECK_EQ(unextended_input_shape.DimensionsCount(), 4); in Mean()
1145 TFLITE_CHECK_EQ(op_params.axis_count, 2); in Mean()
1148 TFLITE_CHECK_EQ(output_height, 1); in Mean()
1149 TFLITE_CHECK_EQ(output_width, 1); in Mean()
A Dneon_tensor_utils.cc2163 TFLITE_CHECK_EQ((intptr_t)(&vector[0]) & (kNeonVectorAlignment - 1), 0); in NeonVectorScalarMultiply()

Completed in 54 milliseconds