Home
last modified time | relevance | path

Searched refs:TFLITE_DCHECK_EQ (Results 1 – 25 of 28) sorted by relevance

12

/external/tensorflow/tensorflow/contrib/lite/kernels/internal/
Dtypes.h358 TFLITE_DCHECK_EQ(shape.DimensionsCount(), 4); in Offset()
398 TFLITE_DCHECK_EQ(ArraySize(array1, index1), ArraySize(array2, index2)); in MatchingArraySize()
405 TFLITE_DCHECK_EQ(ArraySize(array1, index1), ArraySize(array2, index2)); in MatchingArraySize()
412 TFLITE_DCHECK_EQ(shape1.Dims(index1), shape2.Dims(index2)); in MatchingDim()
419 TFLITE_DCHECK_EQ(shape1.Dims(index1), shape2.Dims(index2)); in MatchingDim()
442 TFLITE_DCHECK_EQ(shape.DimensionsCount(), check_shape_0.DimensionsCount()); in MatchingFlatSize()
445 TFLITE_DCHECK_EQ(shape.Dims(i), check_shape_0.Dims(i)); in MatchingFlatSize()
453 TFLITE_DCHECK_EQ(shape.DimensionsCount(), check_shape_0.DimensionsCount()); in MatchingFlatSize()
456 TFLITE_DCHECK_EQ(shape.Dims(i), check_shape_0.Dims(i)); in MatchingFlatSize()
465 TFLITE_DCHECK_EQ(shape.DimensionsCount(), check_shape_0.DimensionsCount()); in MatchingFlatSize()
[all …]
Dcommon.h220 TFLITE_DCHECK_EQ(extent1, 1); in NdArrayDescsForElementwiseBroadcast()
261 TFLITE_DCHECK_EQ(extent1, 1); in NdArrayDescsForElementwiseBroadcast()
/external/tensorflow/tensorflow/lite/kernels/internal/
Dtypes.h374 TFLITE_DCHECK_EQ(shape.DimensionsCount(), 4); in Offset()
414 TFLITE_DCHECK_EQ(ArraySize(array1, index1), ArraySize(array2, index2)); in MatchingArraySize()
421 TFLITE_DCHECK_EQ(ArraySize(array1, index1), ArraySize(array2, index2)); in MatchingArraySize()
428 TFLITE_DCHECK_EQ(shape1.Dims(index1), shape2.Dims(index2)); in MatchingDim()
435 TFLITE_DCHECK_EQ(shape1.Dims(index1), shape2.Dims(index2)); in MatchingDim()
458 TFLITE_DCHECK_EQ(shape.DimensionsCount(), check_shape_0.DimensionsCount()); in MatchingFlatSize()
461 TFLITE_DCHECK_EQ(shape.Dims(i), check_shape_0.Dims(i)); in MatchingFlatSize()
469 TFLITE_DCHECK_EQ(shape.DimensionsCount(), check_shape_0.DimensionsCount()); in MatchingFlatSize()
472 TFLITE_DCHECK_EQ(shape.Dims(i), check_shape_0.Dims(i)); in MatchingFlatSize()
481 TFLITE_DCHECK_EQ(shape.DimensionsCount(), check_shape_0.DimensionsCount()); in MatchingFlatSize()
[all …]
Dcompatibility.h26 #ifndef TFLITE_DCHECK_EQ
27 #define TFLITE_DCHECK_EQ(x, y) ((x) == (y)) ? (void)0 : TFLITE_ASSERT_FALSE macro
/external/tensorflow/tensorflow/lite/kernels/internal/reference/
Ddepthwiseconv_float.h40 TFLITE_DCHECK_EQ(input_shape.DimensionsCount(), 4); in DepthwiseConv()
41 TFLITE_DCHECK_EQ(filter_shape.DimensionsCount(), 4); in DepthwiseConv()
42 TFLITE_DCHECK_EQ(output_shape.DimensionsCount(), 4); in DepthwiseConv()
53 TFLITE_DCHECK_EQ(output_depth, input_depth * depth_multiplier); in DepthwiseConv()
54 TFLITE_DCHECK_EQ(bias_shape.FlatSize(), output_depth); in DepthwiseConv()
Ddepthwiseconv_uint8.h113 TFLITE_DCHECK_EQ(input_shape.DimensionsCount(), 4); in Run()
114 TFLITE_DCHECK_EQ(filter_shape.DimensionsCount(), 4); in Run()
115 TFLITE_DCHECK_EQ(output_shape.DimensionsCount(), 4); in Run()
127 TFLITE_DCHECK_EQ(output_depth, input_depth * depth_multiplier); in Run()
128 TFLITE_DCHECK_EQ(bias_shape.FlatSize(), output_depth); in Run()
Dreference_ops.h147 TFLITE_DCHECK_EQ(input_shape.DimensionsCount(), 4); in Conv()
148 TFLITE_DCHECK_EQ(filter_shape.DimensionsCount(), 4); in Conv()
149 TFLITE_DCHECK_EQ(output_shape.DimensionsCount(), 4); in Conv()
157 TFLITE_DCHECK_EQ(bias_shape.FlatSize(), output_depth); in Conv()
230 TFLITE_DCHECK_EQ(input_shape.DimensionsCount(), 4); in Conv()
231 TFLITE_DCHECK_EQ(filter_shape.DimensionsCount(), 4); in Conv()
232 TFLITE_DCHECK_EQ(output_shape.DimensionsCount(), 4); in Conv()
237 TFLITE_DCHECK_EQ(bias_shape.FlatSize(), output_depth); in Conv()
314 TFLITE_DCHECK_EQ(input_width * block_size, output_width); in DepthToSpace()
315 TFLITE_DCHECK_EQ(input_height * block_size, output_height); in DepthToSpace()
[all …]
Dlegacy_reference_ops.h173 TFLITE_DCHECK_EQ(output_activation_min, 0); in DepthwiseConv()
174 TFLITE_DCHECK_EQ(output_activation_max, 255); in DepthwiseConv()
345 TFLITE_DCHECK_EQ(output_activation_min, 0); in Conv()
346 TFLITE_DCHECK_EQ(output_activation_max, 255); in Conv()
510 TFLITE_DCHECK_EQ(output_activation_min, 0); in FullyConnected()
511 TFLITE_DCHECK_EQ(output_activation_max, 255); in FullyConnected()
602 TFLITE_DCHECK_EQ(Ac, FusedActivationFunctionType::kNone); in Concatenation()
648 TFLITE_DCHECK_EQ(Ac, FusedActivationFunctionType::kNone); in DepthConcatenation()
692 TFLITE_DCHECK_EQ(Ac, FusedActivationFunctionType::kNone); in TensorFlowSplit()
851 TFLITE_DCHECK_EQ(start_indices.size(), 4); in StridedSlice()
[all …]
Dportable_tensor_utils.cc117 TFLITE_DCHECK_EQ( // NOLINT in PortableSparseMatrixBatchVectorMultiplyAccumulate()
149 TFLITE_DCHECK_EQ( // NOLINT in PortableSparseMatrixBatchVectorMultiplyAccumulate()
/external/tensorflow/tensorflow/lite/kernels/internal/reference/integer_ops/
Ddepthwise_conv.h45 TFLITE_DCHECK_EQ(input_shape.DimensionsCount(), 4); in DepthwiseConvPerChannel()
46 TFLITE_DCHECK_EQ(filter_shape.DimensionsCount(), 4); in DepthwiseConvPerChannel()
47 TFLITE_DCHECK_EQ(output_shape.DimensionsCount(), 4); in DepthwiseConvPerChannel()
59 TFLITE_DCHECK_EQ(output_depth, input_depth * depth_multiplier); in DepthwiseConvPerChannel()
60 TFLITE_DCHECK_EQ(bias_shape.FlatSize(), output_depth); in DepthwiseConvPerChannel()
Dconv.h47 TFLITE_DCHECK_EQ(input_shape.DimensionsCount(), 4); in ConvPerChannel()
48 TFLITE_DCHECK_EQ(filter_shape.DimensionsCount(), 4); in ConvPerChannel()
49 TFLITE_DCHECK_EQ(output_shape.DimensionsCount(), 4); in ConvPerChannel()
54 TFLITE_DCHECK_EQ(bias_shape.FlatSize(), output_depth); in ConvPerChannel()
Dmean.h44 TFLITE_DCHECK_EQ(op_params.axis_count, 2); in Mean()
47 TFLITE_DCHECK_EQ(output_height, 1); in Mean()
48 TFLITE_DCHECK_EQ(output_width, 1); in Mean()
Dpooling.h29 TFLITE_DCHECK_EQ(input_shape.DimensionsCount(), 4); in AveragePool()
30 TFLITE_DCHECK_EQ(output_shape.DimensionsCount(), 4); in AveragePool()
90 TFLITE_DCHECK_EQ(input_shape.DimensionsCount(), 4); in MaxPool()
91 TFLITE_DCHECK_EQ(output_shape.DimensionsCount(), 4); in MaxPool()
Dfully_connected.h38 TFLITE_DCHECK_EQ(output_shape.DimensionsCount(), 2); in FullyConnected()
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/
Ddepthwiseconv_uint8_3x3_filter.h43 TFLITE_DCHECK_EQ(reinterpret_cast<std::uintptr_t>(dst) % 4, 0); \
46 TFLITE_DCHECK_EQ(reinterpret_cast<std::uintptr_t>(dst) % 4, 0); \
3917 TFLITE_DCHECK_EQ(function_params->padding_bottom, 0);
3918 TFLITE_DCHECK_EQ(function_params->padding_top, 0);
3919 TFLITE_DCHECK_EQ(function_params->padding_left, 0);
3920 TFLITE_DCHECK_EQ(function_params->padding_right, 0);
4042 TFLITE_DCHECK_EQ(width_overall_micro_repeats,
4082 TFLITE_DCHECK_EQ(
4338 TFLITE_DCHECK_EQ(
4477 TFLITE_DCHECK_EQ((start_width + copy_done) % 16, 0);
[all …]
Ddepthwiseconv_uint8_transitional.h45 TFLITE_DCHECK_EQ(reinterpret_cast<std::uintptr_t>(dst) % 4, 0); \
48 TFLITE_DCHECK_EQ(reinterpret_cast<std::uintptr_t>(dst) % 4, 0); \
804 TFLITE_DCHECK_EQ(width_overall_micro_repeats,
858 TFLITE_DCHECK_EQ(scratch_data, scratch_block_data +
995 TFLITE_DCHECK_EQ((start_width + copy_done) % 16, 0);
1137 TFLITE_DCHECK_EQ(width_overall_micro_repeats, 1);
1182 TFLITE_DCHECK_EQ(
1200 TFLITE_DCHECK_EQ(function_params->padding_bottom, 0);
1201 TFLITE_DCHECK_EQ(function_params->padding_top, 0);
1202 TFLITE_DCHECK_EQ(function_params->padding_left, 0);
[all …]
Doptimized_ops.h182 TFLITE_DCHECK_EQ(flatsize % rows, 0); in MapAsMatrixWithGivenNumberOfRows()
197 TFLITE_DCHECK_EQ((array_size % bias_size), 0); in AddBiasAndEvalActivationFunction()
248 TFLITE_DCHECK_EQ((array_size % bias_size), 0); in AddBiasAndEvalActivationFunction()
307 TFLITE_DCHECK_EQ(FlatSizeSkipDim(output_shape, output_dim_count - 1), 1); in GEMVForLstmCell()
494 TFLITE_DCHECK_EQ(FlatSizeSkipDim(output_shape, output_dim_count - 1), 1); in GEMVForLstmCellWithSymmetricRange()
829 TFLITE_DCHECK_EQ(FlatSizeSkipDim(output_shape, output_dim_count - 1), 1); in FullyConnectedAsGEMVWorkerImpl()
1133 TFLITE_DCHECK_EQ(row_start, output_rows); in FullyConnectedAsGEMV()
1205 TFLITE_DCHECK_EQ(filter_shape.FlatSize(), filter_rows * filter_cols); in FullyConnected()
1207 TFLITE_DCHECK_EQ(output_rows, filter_rows); in FullyConnected()
1208 TFLITE_DCHECK_EQ(bias_shape.FlatSize(), output_rows); in FullyConnected()
[all …]
Dmultithreaded_conv.h149 TFLITE_DCHECK_EQ(input_shape.DimensionsCount(), 4); in Conv()
150 TFLITE_DCHECK_EQ(filter_shape.DimensionsCount(), 4); in Conv()
151 TFLITE_DCHECK_EQ(output_shape.DimensionsCount(), 4); in Conv()
Dlegacy_optimized_ops.h272 TFLITE_DCHECK_EQ(output_activation_min, 0); in DepthwiseConv()
273 TFLITE_DCHECK_EQ(output_activation_max, 255); in DepthwiseConv()
675 TFLITE_DCHECK_EQ(output_activation_min, 0); in Conv()
676 TFLITE_DCHECK_EQ(output_activation_max, 255); in Conv()
760 TFLITE_DCHECK_EQ(output_rows, filter_rows); in ConvAsGemm()
761 TFLITE_DCHECK_EQ(output_cols, input_cols); in ConvAsGemm()
762 TFLITE_DCHECK_EQ(filter_cols, input_rows); in ConvAsGemm()
763 TFLITE_DCHECK_EQ(bias_dims.sizes[0], output_rows); in ConvAsGemm()
764 TFLITE_DCHECK_EQ(bias_dims.sizes[1], 1); in ConvAsGemm()
765 TFLITE_DCHECK_EQ(bias_dims.sizes[2], 1); in ConvAsGemm()
[all …]
Ddepthwiseconv_float.h781 TFLITE_DCHECK_EQ(input_depth, kFixedInputDepth);
784 TFLITE_DCHECK_EQ(depth_multiplier, kFixedDepthMultiplier);
786 TFLITE_DCHECK_EQ(output_depth, input_depth * depth_multiplier);
909 TFLITE_DCHECK_EQ(input_shape.DimensionsCount(), 4);
910 TFLITE_DCHECK_EQ(filter_shape.DimensionsCount(), 4);
911 TFLITE_DCHECK_EQ(output_shape.DimensionsCount(), 4);
922 TFLITE_DCHECK_EQ(output_depth, input_depth * depth_multiplier);
923 TFLITE_DCHECK_EQ(bias_shape.FlatSize(), output_depth);
Dneon_tensor_utils.cc453 TFLITE_DCHECK_EQ( // NOLINT in NeonMatrixBatchVectorMultiplyAccumulate()
482 TFLITE_DCHECK_EQ( // NOLINT in NeonMatrixBatchVectorMultiplyAccumulate()
515 TFLITE_DCHECK_EQ( // NOLINT in NeonSparseMatrixBatchVectorMultiplyAccumulate()
570 TFLITE_DCHECK_EQ( // NOLINT in NeonSparseMatrixBatchVectorMultiplyAccumulate()
598 TFLITE_DCHECK_EQ( // NOLINT in NeonSparseMatrixBatchVectorMultiplyAccumulate()
/external/tensorflow/tensorflow/lite/kernels/
Dshape.cc67 TFLITE_DCHECK_EQ(NumDimensions(output), 1); in Eval()
68 TFLITE_DCHECK_EQ(SizeOfDimension(output, 0), NumDimensions(input)); in Eval()
/external/tensorflow/tensorflow/contrib/lite/kernels/internal/optimized/
Doptimized_ops.h229 TFLITE_DCHECK_EQ(flatsize % rows, 0); in MapAsMatrixWithGivenNumberOfRows()
292 TFLITE_DCHECK_EQ((array_size % bias_size), 0); in AddBiasAndEvalActivationFunction()
343 TFLITE_DCHECK_EQ((array_size % bias_size), 0); in AddBiasAndEvalActivationFunction()
429 TFLITE_DCHECK_EQ(FlatSizeSkipDim(output_shape, output_dim_count - 1), 1); in GEMVForLstmCell()
616 TFLITE_DCHECK_EQ(FlatSizeSkipDim(output_shape, output_dim_count - 1), 1); in GEMVForLstmCellWithSymmetricRange()
975 TFLITE_DCHECK_EQ(FlatSizeSkipDim(output_shape, output_dim_count - 1), 1); in FullyConnectedAsGEMV()
1186 TFLITE_DCHECK_EQ(filter_shape.FlatSize(), filter_rows * filter_cols); in FullyConnected()
1188 TFLITE_DCHECK_EQ(output_rows, filter_rows); in FullyConnected()
1189 TFLITE_DCHECK_EQ(bias_shape.FlatSize(), output_rows); in FullyConnected()
1251 TFLITE_DCHECK_EQ(output_offset, 0); in FullyConnected()
[all …]
/external/tensorflow/tensorflow/lite/experimental/kernels/
Dctc_beam_search.h201 TFLITE_DCHECK_EQ(top_n, beam_log_probabilities.size()); in Decode()
202 TFLITE_DCHECK_EQ(beams.size(), beam_log_probabilities.size()); in Decode()
220 TFLITE_DCHECK_EQ(num_classes_, input.size()); in GetTopK()
275 TFLITE_DCHECK_EQ(num_classes_, raw_input.size()); in Step()
/external/tensorflow/tensorflow/lite/experimental/micro/kernels/portable_optimized/
Ddepthwise_conv.cc113 TFLITE_DCHECK_EQ(input_shape.DimensionsCount(), 4); in DepthwiseConvOptimizedForFilterWidthEight()
114 TFLITE_DCHECK_EQ(filter_shape.DimensionsCount(), 4); in DepthwiseConvOptimizedForFilterWidthEight()
115 TFLITE_DCHECK_EQ(output_shape.DimensionsCount(), 4); in DepthwiseConvOptimizedForFilterWidthEight()
127 TFLITE_DCHECK_EQ(output_depth, input_depth * depth_multiplier); in DepthwiseConvOptimizedForFilterWidthEight()
128 TFLITE_DCHECK_EQ(bias_shape.FlatSize(), output_depth); in DepthwiseConvOptimizedForFilterWidthEight()

12