Home
last modified time | relevance | path

Searched refs:TFLITE_DCHECK (Results 1 – 19 of 19) sorted by relevance

/external/tensorflow/tensorflow/contrib/lite/kernels/internal/
Dcommon.h163 TFLITE_DCHECK(i0 >= 0 && i0 < desc.extents[0]); in SubscriptToIndex()
164 TFLITE_DCHECK(i1 >= 0 && i1 < desc.extents[1]); in SubscriptToIndex()
165 TFLITE_DCHECK(i2 >= 0 && i2 < desc.extents[2]); in SubscriptToIndex()
166 TFLITE_DCHECK(i3 >= 0 && i3 < desc.extents[3]); in SubscriptToIndex()
198 TFLITE_DCHECK(desc0_out != nullptr); in NdArrayDescsForElementwiseBroadcast()
199 TFLITE_DCHECK(desc1_out != nullptr); in NdArrayDescsForElementwiseBroadcast()
232 TFLITE_DCHECK(desc0_out != nullptr); in NdArrayDescsForElementwiseBroadcast()
233 TFLITE_DCHECK(desc1_out != nullptr); in NdArrayDescsForElementwiseBroadcast()
Dtypes.h306 TFLITE_DCHECK(dims != nullptr); in NextIndex()
307 TFLITE_DCHECK(current != nullptr); in NextIndex()
335 TFLITE_DCHECK(dims != nullptr); in ReducedOutputOffset()
336 TFLITE_DCHECK(index != nullptr); in ReducedOutputOffset()
360 TFLITE_DCHECK(i0 >= 0 && i0 < dims_data[0]); in Offset()
361 TFLITE_DCHECK(i1 >= 0 && i1 < dims_data[1]); in Offset()
362 TFLITE_DCHECK(i2 >= 0 && i2 < dims_data[2]); in Offset()
363 TFLITE_DCHECK(i3 >= 0 && i3 < dims_data[3]); in Offset()
368 TFLITE_DCHECK(i0 >= 0 && i0 < dims.sizes[0]); in Offset()
369 TFLITE_DCHECK(i1 >= 0 && i1 < dims.sizes[1]); in Offset()
[all …]
/external/tensorflow/tensorflow/lite/kernels/internal/
Dtypes.h322 TFLITE_DCHECK(dims != nullptr); in NextIndex()
323 TFLITE_DCHECK(current != nullptr); in NextIndex()
351 TFLITE_DCHECK(dims != nullptr); in ReducedOutputOffset()
352 TFLITE_DCHECK(index != nullptr); in ReducedOutputOffset()
376 TFLITE_DCHECK(i0 >= 0 && i0 < dims_data[0]); in Offset()
377 TFLITE_DCHECK(i1 >= 0 && i1 < dims_data[1]); in Offset()
378 TFLITE_DCHECK(i2 >= 0 && i2 < dims_data[2]); in Offset()
379 TFLITE_DCHECK(i3 >= 0 && i3 < dims_data[3]); in Offset()
384 TFLITE_DCHECK(i0 >= 0 && i0 < dims.sizes[0]); in Offset()
385 TFLITE_DCHECK(i1 >= 0 && i1 < dims.sizes[1]); in Offset()
[all …]
Dcommon.h444 TFLITE_DCHECK(i0 >= 0 && i0 < desc.extents[0]); in SubscriptToIndex()
445 TFLITE_DCHECK(i1 >= 0 && i1 < desc.extents[1]); in SubscriptToIndex()
446 TFLITE_DCHECK(i2 >= 0 && i2 < desc.extents[2]); in SubscriptToIndex()
447 TFLITE_DCHECK(i3 >= 0 && i3 < desc.extents[3]); in SubscriptToIndex()
479 TFLITE_DCHECK(desc0_out != nullptr); in NdArrayDescsForElementwiseBroadcast()
480 TFLITE_DCHECK(desc1_out != nullptr); in NdArrayDescsForElementwiseBroadcast()
513 TFLITE_DCHECK(desc0_out != nullptr); in NdArrayDescsForElementwiseBroadcast()
514 TFLITE_DCHECK(desc1_out != nullptr); in NdArrayDescsForElementwiseBroadcast()
Dcompatibility.h22 #ifndef TFLITE_DCHECK
23 #define TFLITE_DCHECK(condition) (condition) ? (void)0 : TFLITE_ASSERT_FALSE macro
Dstrided_slice_logic.h29 TFLITE_DCHECK(!(hi < lo)); in Clamp()
/external/tensorflow/tensorflow/lite/kernels/internal/reference/
Dfully_connected.h196 TFLITE_DCHECK((accum_depth % 16) == 0); in ShuffledFullyConnected()
197 TFLITE_DCHECK((output_depth % 4) == 0); in ShuffledFullyConnected()
220 TFLITE_DCHECK(false); in ShuffledFullyConnected()
318 TFLITE_DCHECK(false); in ShuffledFullyConnected()
Dreference_ops.h662 TFLITE_DCHECK(input1_shift == 0 || params.input2_shift == 0); in Add()
1563 TFLITE_DCHECK(input1_shift == 0 || params.input2_shift == 0); in Sub16()
3467 TFLITE_DCHECK(current >= 0 && current < num_dims); in ResolveAxis()
3628 TFLITE_DCHECK((op_params.axis[0] == 1 && op_params.axis[1] == 2) || in Mean()
3672 TFLITE_DCHECK((op_params.axis[0] == 1 && op_params.axis[1] == 2) || in Mean()
4553 TFLITE_DCHECK(!op_params.align_corners); in ResizeNearestNeighbor()
Dlegacy_reference_ops.h1208 TFLITE_DCHECK(Ac == FusedActivationFunctionType::kNone); in Add()
/external/tensorflow/tensorflow/lite/experimental/kernels/
Dtop_n.h252 TFLITE_DCHECK(!empty()); in peek_bottom()
305 TFLITE_DCHECK(output); in ExtractNondestructive()
324 TFLITE_DCHECK(output); in ExtractUnsortedNondestructive()
Dctc_beam_search.h400 TFLITE_DCHECK(paths); in TopPaths()
401 TFLITE_DCHECK(log_probs); in TopPaths()
/external/tensorflow/tensorflow/lite/kernels/internal/reference/integer_ops/
Dmean.h45 TFLITE_DCHECK((op_params.axis[0] == 1 && op_params.axis[1] == 2) || in Mean()
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/
Ddepthwiseconv_uint8_3x3_filter.h3285 TFLITE_DCHECK(
3288 TFLITE_DCHECK(
3580 TFLITE_DCHECK(params.output_depth == params.input_depth * depth_multiplier);
3581 TFLITE_DCHECK(depth_multiplier == 1);
3582 TFLITE_DCHECK(filter_height == 3);
3583 TFLITE_DCHECK(filter_width == 3);
3584 TFLITE_DCHECK(stride_height == 1 || stride_height == 2);
3585 TFLITE_DCHECK(stride_width == 1 || stride_width == 2);
3586 TFLITE_DCHECK(stride_width == stride_height);
3587 TFLITE_DCHECK(pad_height == 0 || pad_height == 1);
[all …]
Doptimized_ops.h316 TFLITE_DCHECK(!(output_size % 4)); in GEMVForLstmCell()
317 TFLITE_DCHECK(!(input_size % 8)); in GEMVForLstmCell()
503 TFLITE_DCHECK(!(output_size % 4)); in GEMVForLstmCellWithSymmetricRange()
504 TFLITE_DCHECK(!(input_size % 64)); in GEMVForLstmCellWithSymmetricRange()
1488 TFLITE_DCHECK(false); in ShuffledFullyConnectedWorkerImpl()
1585 TFLITE_DCHECK(false); in ShuffledFullyConnectedWorkerImpl()
1660 TFLITE_DCHECK((accum_depth % 16) == 0); in ShuffledFullyConnected()
1661 TFLITE_DCHECK((output_depth % 4) == 0); in ShuffledFullyConnected()
1720 TFLITE_DCHECK(false); in ShuffledFullyConnected()
1774 TFLITE_DCHECK((op_params.axis[0] == 1 && op_params.axis[1] == 2) || in MeanImpl()
[all …]
Ddepthwiseconv_uint8_transitional.h1895 TFLITE_DCHECK(leading_width_padding);
1896 TFLITE_DCHECK(trailing_width_padding);
2343 TFLITE_DCHECK(depth_micro_repeats > 0);
2507 TFLITE_DCHECK(depth_micro_repeats > 0);
2593 TFLITE_DCHECK(depth_micro_repeats > 0);
2813 TFLITE_DCHECK(depth_micro_repeats > 0);
2976 TFLITE_DCHECK(depth_micro_repeats > 0);
3600 TFLITE_DCHECK(depth_micro_repeats > 0);
3937 TFLITE_DCHECK(depth_micro_repeats > 0);
Dlegacy_optimized_ops.h124 TFLITE_DCHECK((flatsize % rows) == 0); in MapAsMatrixWithGivenNumberOfRows()
988 TFLITE_DCHECK(Ac == FusedActivationFunctionType::kNone); in Add()
1651 TFLITE_DCHECK(Ac == FusedActivationFunctionType::kNone); in Mul()
Ddepthwiseconv_float.h779 TFLITE_DCHECK(stride == 1 || kAllowStrided);
Ddepthwiseconv_uint8.h1487 TFLITE_DCHECK(stride == 1 || kAllowStrided);
/external/tensorflow/tensorflow/contrib/lite/kernels/internal/optimized/
Doptimized_ops.h217 TFLITE_DCHECK((flatsize % rows) == 0); in MapAsMatrixWithGivenNumberOfRows()
438 TFLITE_DCHECK(!(output_size % 4)); in GEMVForLstmCell()
439 TFLITE_DCHECK(!(input_size % 8)); in GEMVForLstmCell()
625 TFLITE_DCHECK(!(output_size % 4)); in GEMVForLstmCellWithSymmetricRange()
626 TFLITE_DCHECK(!(input_size % 64)); in GEMVForLstmCellWithSymmetricRange()
987 TFLITE_DCHECK(!(output_size % kPeel)); in FullyConnectedAsGEMV()
1544 TFLITE_DCHECK(false); in ShuffledFullyConnectedWorkerImpl()
1641 TFLITE_DCHECK(false); in ShuffledFullyConnectedWorkerImpl()
1716 TFLITE_DCHECK((accum_depth % 16) == 0); in ShuffledFullyConnected()
1717 TFLITE_DCHECK((output_depth % 4) == 0); in ShuffledFullyConnected()
[all …]