| /external/tensorflow/tensorflow/lite/kernels/internal/reference/integer_ops/ |
| D | mean.h | 33 TFLITE_CHECK_EQ(unextended_input_shape.DimensionsCount(), 4); in Mean()
47 TFLITE_CHECK_EQ(op_params.axis_count, 2); in Mean()
50 TFLITE_CHECK_EQ(output_height, 1); in Mean()
51 TFLITE_CHECK_EQ(output_width, 1); in Mean()
|
| /external/tensorflow/tensorflow/lite/kernels/internal/optimized/integer_ops/ |
| D | mean.h | 43 TFLITE_CHECK_EQ(op_params.axis_count, 2); in MeanImpl()
46 TFLITE_CHECK_EQ(output_height, 1); in MeanImpl()
47 TFLITE_CHECK_EQ(output_width, 1); in MeanImpl()
189 TFLITE_CHECK_EQ(unextended_input_shape.DimensionsCount(), 4); in Mean()
199 TFLITE_CHECK_EQ(op_params.axis_count, 2); in Mean()
202 TFLITE_CHECK_EQ(output_height, 1); in Mean()
203 TFLITE_CHECK_EQ(output_width, 1); in Mean()
|
| /external/tensorflow/tensorflow/lite/kernels/perception/ |
| D | max_pool_with_argmax.cc | 136 TFLITE_CHECK_EQ(pool_size.size(), 4); in Init()
137 TFLITE_CHECK_EQ(pool_size[0].AsInt32(), 1); in Init()
138 TFLITE_CHECK_EQ(pool_size[3].AsInt32(), 1); in Init()
144 TFLITE_CHECK_EQ(strides.size(), 4); in Init()
145 TFLITE_CHECK_EQ(strides[0].AsInt32(), 1); in Init()
146 TFLITE_CHECK_EQ(strides[3].AsInt32(), 1); in Init()
|
| D | dense_image_warp.cc | 43 TFLITE_CHECK_EQ(flow_shape.Dims(3), 2); in DenseImageWarp()
|
| /external/tensorflow/tensorflow/lite/delegates/ |
| D | delegate_test_util.cc | 175 TFLITE_CHECK_EQ(reg->builtin_code, tflite::BuiltinOperator_CUSTOM); in SimpleDelegate()
176 TFLITE_CHECK_EQ(strcmp(reg->custom_name, "my_add"), 0); in SimpleDelegate()
178 TFLITE_CHECK_EQ(reg->builtin_code, tflite::BuiltinOperator_ADD); in SimpleDelegate()
192 TFLITE_CHECK_EQ(reg->builtin_code, tflite::BuiltinOperator_CUSTOM); in SimpleDelegate()
193 TFLITE_CHECK_EQ(strcmp(reg->custom_name, "my_add"), 0); in SimpleDelegate()
195 TFLITE_CHECK_EQ(reg->builtin_code, tflite::BuiltinOperator_ADD); in SimpleDelegate()
203 TFLITE_CHECK_EQ( in SimpleDelegate()
229 TFLITE_CHECK_EQ( in SimpleDelegate()
251 TFLITE_CHECK_EQ(output->buffer_handle, buffer_handle); in SimpleDelegate()
|
| /external/tensorflow/tensorflow/lite/kernels/internal/reference/ |
| D | reduce.h | 280 TFLITE_CHECK_EQ(unextended_input_shape.DimensionsCount(), 4); in Mean()
295 TFLITE_CHECK_EQ(op_params.axis_count, 2); in Mean()
298 TFLITE_CHECK_EQ(output_height, 1); in Mean()
299 TFLITE_CHECK_EQ(output_width, 1); in Mean()
325 TFLITE_CHECK_EQ(unextended_input_shape.DimensionsCount(), 4); in Mean()
339 TFLITE_CHECK_EQ(op_params.axis_count, 2); in Mean()
342 TFLITE_CHECK_EQ(output_height, 1); in Mean()
343 TFLITE_CHECK_EQ(output_width, 1); in Mean()
|
| D | broadcast_args.h | 47 TFLITE_CHECK_EQ(shape1_i, shape2_i); in BroadcastArgs()
|
| D | legacy_reference_ops.h | 916 TFLITE_CHECK_EQ(p->start_indices_count, p->stop_indices_count); in StridedSliceReverseIndices()
917 TFLITE_CHECK_EQ(p->stop_indices_count, p->strides_count); in StridedSliceReverseIndices()
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | pooling3d.cc | 252 TFLITE_CHECK_EQ(data_format, "NDHWC"); in Init()
265 TFLITE_CHECK_EQ(pool_size.size(), 5); in Init()
266 TFLITE_CHECK_EQ(pool_size[0].AsInt32(), 1); in Init()
267 TFLITE_CHECK_EQ(pool_size[4].AsInt32(), 1); in Init()
274 TFLITE_CHECK_EQ(strides.size(), 5); in Init()
275 TFLITE_CHECK_EQ(strides[0].AsInt32(), 1); in Init()
276 TFLITE_CHECK_EQ(strides[4].AsInt32(), 1); in Init()
|
| /external/tensorflow/tensorflow/lite/kernels/internal/ |
| D | compatibility.h | 55 #ifndef TFLITE_CHECK_EQ
56 #define TFLITE_CHECK_EQ(x, y) ((x) == (y)) ? (void)0 : TFLITE_ABORT macro
|
| D | quantization_util.h | 43 TFLITE_CHECK_EQ(rmin, 0.); in ChooseQuantizationParams()
44 TFLITE_CHECK_EQ(rmax, 0.); in ChooseQuantizationParams()
|
| D | strided_slice_logic.h | 41 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()
|
| D | types.h | 295 TFLITE_CHECK_EQ(size_1, size_2); in MatchingElementsSize()
305 TFLITE_CHECK_EQ(size_1, size_2); in MatchingElementsSize()
306 TFLITE_CHECK_EQ(size_2, size_3); in MatchingElementsSize()
|
| D | quantization_util.cc | 127 TFLITE_CHECK_EQ(8, sizeof(double)); in IntegerFrExp()
|
| /external/tensorflow/tensorflow/lite/kernels/internal/optimized/ |
| D | reduce.h | 51 TFLITE_CHECK_EQ(op_params.axis_count, 2); in MeanImpl()
54 TFLITE_CHECK_EQ(output_height, 1); in MeanImpl()
55 TFLITE_CHECK_EQ(output_width, 1); in MeanImpl()
202 TFLITE_CHECK_EQ(unextended_input_shape.DimensionsCount(), 4); in Mean()
212 TFLITE_CHECK_EQ(op_params.axis_count, 2); in Mean()
215 TFLITE_CHECK_EQ(output_height, 1); in Mean()
216 TFLITE_CHECK_EQ(output_width, 1); in Mean()
|
| D | neon_tensor_utils.cc | 2264 TFLITE_CHECK_EQ((intptr_t)(&vector[0]) & (kNeonVectorAlignment - 1), 0); in NeonVectorScalarMultiply()
|