Home
last modified time | relevance | path

Searched refs:TF_LITE_ENSURE (Results 1 – 25 of 58) sorted by relevance

123

/external/tensorflow/tensorflow/lite/kernels/
Dsub.cc78 TF_LITE_ENSURE(context, in Prepare8BitSubOp()
94 TF_LITE_ENSURE(context, in Prepare8BitSubOp()
96 TF_LITE_ENSURE(context, in Prepare8BitSubOp()
98 TF_LITE_ENSURE(context, in Prepare8BitSubOp()
100 TF_LITE_ENSURE(context, in Prepare8BitSubOp()
102 TF_LITE_ENSURE(context, in Prepare8BitSubOp()
104 TF_LITE_ENSURE(context, in Prepare8BitSubOp()
163 TF_LITE_ENSURE(context, input1_scale_is_pot); in PrepareInt16SubOp()
168 TF_LITE_ENSURE(context, input2_scale_is_pot); in PrepareInt16SubOp()
173 TF_LITE_ENSURE(context, output_scale_is_pot); in PrepareInt16SubOp()
[all …]
Dreverse_sequence.cc75 TF_LITE_ENSURE(context, seq_dim >= 0); in ReverseSequenceImpl()
76 TF_LITE_ENSURE(context, batch_dim >= 0); in ReverseSequenceImpl()
77 TF_LITE_ENSURE(context, seq_dim != batch_dim); in ReverseSequenceImpl()
78 TF_LITE_ENSURE(context, seq_dim < NumDimensions(input)); in ReverseSequenceImpl()
79 TF_LITE_ENSURE(context, batch_dim < NumDimensions(input)); in ReverseSequenceImpl()
83 TF_LITE_ENSURE(context, seq_lengths[i] <= SizeOfDimension(input, seq_dim)); in ReverseSequenceImpl()
Dpad.cc97 TF_LITE_ENSURE(context, NumInputs(node) == 2 || NumInputs(node) == 3); in Prepare()
108 TF_LITE_ENSURE(context, op_context.dims <= 4); in Prepare()
150 TF_LITE_ENSURE(context, before_padding.size() <= 4); \ in Eval()
151 TF_LITE_ENSURE(context, after_padding.size() <= 4); \ in Eval()
189 TF_LITE_ENSURE(context, op_context.output->params.zero_point >= in Eval()
191 TF_LITE_ENSURE(context, op_context.output->params.zero_point <= in Eval()
222 TF_LITE_ENSURE(context, op_context.output->params.zero_point >= in Eval()
224 TF_LITE_ENSURE(context, op_context.output->params.zero_point <= in Eval()
Dkernel_util.cc58 TF_LITE_ENSURE(context, affine_quantization); in PopulateConvolutionQuantizationParams()
59 TF_LITE_ENSURE(context, affine_quantization->scale); in PopulateConvolutionQuantizationParams()
118 TF_LITE_ENSURE(context, std::abs(input_product_scale - bias_scale) <= in GetQuantizedConvolutionMultipler()
120 TF_LITE_ENSURE(context, input_product_scale >= 0); in GetQuantizedConvolutionMultipler()
172 TF_LITE_ENSURE(context, false); in CalculateActivationRangeQuantized()
223 TF_LITE_ENSURE(context, d1 == d2 || d1 == 1 || d2 == 1); in CalculateShapeForBroadcast()
Daudio_spectrogram.cc83 TF_LITE_ENSURE(context, params->spectrogram->Initialize(params->window_size, in Prepare()
108 TF_LITE_ENSURE(context, params->spectrogram->Initialize(params->window_size, in Eval()
128 TF_LITE_ENSURE(context, in Eval()
133 TF_LITE_ENSURE(context, spectrogram_output.empty() || in Eval()
Dbatch_to_space_nd.cc69 TF_LITE_ENSURE(context, crops[0] >= 0); in ResizeOutputTensor()
70 TF_LITE_ENSURE(context, crops[1] >= 0); in ResizeOutputTensor()
71 TF_LITE_ENSURE(context, crops[2] >= 0); in ResizeOutputTensor()
72 TF_LITE_ENSURE(context, crops[3] >= 0); in ResizeOutputTensor()
Dsparse_to_dense.cc106 TF_LITE_ENSURE(context, true_dimensions <= kMaxDimensions); in GetIndicesVector()
161 TF_LITE_ENSURE(context, NumDimensions(indices) < 3); in Prepare()
166 TF_LITE_ENSURE(context, NumDimensions(values) < 2); in Prepare()
170 TF_LITE_ENSURE( in Prepare()
172 TF_LITE_ENSURE(context, output_shape->type == kTfLiteInt32 || in Prepare()
Dadd.cc146 TF_LITE_ENSURE(context, input1_scale_is_pot); in Prepare()
151 TF_LITE_ENSURE(context, input2_scale_is_pot); in Prepare()
156 TF_LITE_ENSURE(context, output_scale_is_pot); in Prepare()
163 TF_LITE_ENSURE(context, data->input1_shift == 0 || data->input2_shift == 0); in Prepare()
164 TF_LITE_ENSURE(context, data->input1_shift <= 0); in Prepare()
165 TF_LITE_ENSURE(context, data->input2_shift <= 0); in Prepare()
Dselect.cc43 TF_LITE_ENSURE(context, input_condition->type == kTfLiteBool); in SelectPrepare()
47 TF_LITE_ENSURE(context, HaveSameShapes(input_x, input_y)); in SelectPrepare()
58 TF_LITE_ENSURE(context, same_shape); in SelectPrepare()
Dconcatenation.cc52 TF_LITE_ENSURE(context, axis >= 0); in Prepare()
53 TF_LITE_ENSURE(context, axis < t0->dims->size); in Prepare()
57 TF_LITE_ENSURE(context, t0->dims->size <= 4); in Prepare()
59 TF_LITE_ENSURE(context, in Prepare()
Ddetection_postprocess.cc386 TF_LITE_ENSURE(context, (max_detections >= 0)); in NonMaxSuppressionSingleClassHelper()
389 TF_LITE_ENSURE(context, (intersection_over_union_threshold > 0.0f) && in NonMaxSuppressionSingleClassHelper()
392 TF_LITE_ENSURE(context, ValidateBoxes(decoded_boxes, num_boxes)); in NonMaxSuppressionSingleClassHelper()
412 TF_LITE_ENSURE(context, (active_candidate->dims->data[0]) == num_boxes); in NonMaxSuppressionSingleClassHelper()
475 TF_LITE_ENSURE(context, label_offset != -1); in NonMaxSuppressionMultiClassRegularHelper()
476 TF_LITE_ENSURE(context, num_detections_per_class > 0); in NonMaxSuppressionMultiClassRegularHelper()
599 TF_LITE_ENSURE(context, (label_offset != -1)); in NonMaxSuppressionMultiClassFastHelper()
600 TF_LITE_ENSURE(context, (max_categories_per_anchor > 0)); in NonMaxSuppressionMultiClassFastHelper()
674 TF_LITE_ENSURE(context, (num_classes_with_background == num_classes + 1)); in NonMaxSuppressionMultiClass()
703 TF_LITE_ENSURE(context, (kBatchSize == 1)); in Eval()
Dunpack.cc53 TF_LITE_ENSURE(context, NumDimensions(input) <= 4); in Prepare()
54 TF_LITE_ENSURE(context, NumDimensions(input) > 1); in Prepare()
59 TF_LITE_ENSURE(context, 0 <= axis && axis < NumDimensions(input)); in Prepare()
Dlstm.cc125 TF_LITE_ENSURE(context, params->cell_clip >= 0); in CheckInputTensorDimensions()
126 TF_LITE_ENSURE(context, params->proj_clip >= 0); in CheckInputTensorDimensions()
181 TF_LITE_ENSURE(context, cifg_weights_all_or_none == true); in CheckInputTensorDimensions()
212 TF_LITE_ENSURE(context, peephole_weights_all_or_none == true); in CheckInputTensorDimensions()
260 TF_LITE_ENSURE(context, projection_tensors_consistent == true); in CheckInputTensorDimensions()
268 TF_LITE_ENSURE(context, input_layer_norm_coefficients != nullptr); in CheckInputTensorDimensions()
276 TF_LITE_ENSURE(context, forget_layer_norm_coefficients != nullptr); in CheckInputTensorDimensions()
283 TF_LITE_ENSURE(context, cell_layer_norm_coefficients != nullptr); in CheckInputTensorDimensions()
290 TF_LITE_ENSURE(context, output_layer_norm_coefficients != nullptr); in CheckInputTensorDimensions()
340 TF_LITE_ENSURE(context, input->dims->size > 1); in Prepare()
[all …]
Dpack.cc38 TF_LITE_ENSURE(context, NumDimensions(input0) >= data->axis); in Prepare()
40 TF_LITE_ENSURE(context, data->axis >= 0); in Prepare()
51 TF_LITE_ENSURE(context, HaveSameShapes(input0, input)); in Prepare()
Dlsh_projection.cc76 TF_LITE_ENSURE(context, NumInputs(node) == 2 || NumInputs(node) == 3); in Resize()
82 TF_LITE_ENSURE(context, SizeOfDimension(hash, 1) <= 32); in Resize()
85 TF_LITE_ENSURE(context, NumDimensions(input) >= 1); in Resize()
Dreduce.cc126 TF_LITE_ENSURE(context, current >= 0 && current < input_num_dims); in ResizeOutputTensor()
358 TF_LITE_ENSURE(context, TF_LITE_MEAN(reference_ops, float, float)); in EvalMean()
362 TF_LITE_ENSURE(context, TF_LITE_MEAN(reference_ops, int, int64_t)); in EvalMean()
365 TF_LITE_ENSURE(context, TF_LITE_MEAN(reference_ops, int64_t, int64_t)); in EvalMean()
371 TF_LITE_ENSURE(context, TF_LITE_MEAN(reference_ops, uint8_t, int)); in EvalMean()
373 TF_LITE_ENSURE( in EvalMean()
419 TF_LITE_ENSURE( in EvalLogic()
543 TF_LITE_ENSURE( in EvalSum()
Drange.cc37 TF_LITE_ENSURE(context, !std::equal_to<T>()(delta, 0)); in GetSize()
38 TF_LITE_ENSURE(context, in GetSize()
91 TF_LITE_ENSURE(context, dtype == kTfLiteInt32 || dtype == kTfLiteFloat32); in Prepare()
Done_hot.cc106 TF_LITE_ENSURE(context, *op_context.depth->data.i32 >= 0); in ResizeOutputTensor()
140 TF_LITE_ENSURE(context, op_context.indices->type == kTfLiteInt32 || in Prepare()
142 TF_LITE_ENSURE(context, op_context.axis >= 0 && in Prepare()
Dadd_n.cc31 TF_LITE_ENSURE(context, num_inputs >= 2); in Prepare()
41 TF_LITE_ENSURE(context, HaveSameShapes(input1, input)); in Prepare()
Dsqueeze.cc49 TF_LITE_ENSURE(context, input_num_dims <= 8); in Prepare()
63 TF_LITE_ENSURE(context, current >= 0 && current < input_num_dims && in Prepare()
/external/tensorflow/tensorflow/lite/
Dsimple_memory_arena.cc38 TF_LITE_ENSURE(context, alignment <= arena_alignment_); in Allocate()
134 TF_LITE_ENSURE(context, committed_); in ResolveAlloc()
135 TF_LITE_ENSURE(context, output_ptr != nullptr); in ResolveAlloc()
Dinterpreter.cc244 TF_LITE_ENSURE(context_, tensor_index < tensors_size()); in SetBufferHandle()
248 TF_LITE_ENSURE(context_, in SetBufferHandle()
252 TF_LITE_ENSURE(context_, tensor->delegate->FreeBufferHandle != nullptr); in SetBufferHandle()
264 TF_LITE_ENSURE(context_, tensor_index < tensors_size()); in GetBufferHandle()
/external/tensorflow/tensorflow/lite/core/
Dsubgraph.cc308 TF_LITE_ENSURE(context_, tensor->delegate == nullptr || in ReplaceNodeSubsetsWithDelegateKernels()
431 TF_LITE_ENSURE(context_, bytes != nullptr); in BytesRequired()
511 TF_LITE_ENSURE(context_, tensor.data.raw != nullptr); in ResetVariableTensors()
589 TF_LITE_ENSURE(context_, in ResizeInputTensor()
691 TF_LITE_ENSURE(context_, next_execution_plan_index_to_prepare_ >= in Invoke()
802 TF_LITE_ENSURE(context_, node_index >= 0); in GetNodeAndRegistration()
804 TF_LITE_ENSURE(context_, static_cast<size_t>(node_index) < nodes_size); in GetNodeAndRegistration()
805 TF_LITE_ENSURE(context_, node != nullptr && registration != nullptr); in GetNodeAndRegistration()
829 TF_LITE_ENSURE(context_, in SetTensorParametersReadOnly()
878 TF_LITE_ENSURE(context_, in SetTensorParametersReadWrite()
[all …]
Dsubgraph.h250 TF_LITE_ENSURE(context_, t != nullptr); in EnsureTensorDataIsReadable()
252 TF_LITE_ENSURE(context_, t->delegate != nullptr); in EnsureTensorDataIsReadable()
253 TF_LITE_ENSURE(context_, t->buffer_handle != kTfLiteNullBufferHandle); in EnsureTensorDataIsReadable()
254 TF_LITE_ENSURE(context_, t->delegate->CopyFromBufferHandle != nullptr); in EnsureTensorDataIsReadable()
/external/tensorflow/tensorflow/contrib/lite/kernels/
Dlsh_projection.cc76 TF_LITE_ENSURE(context, NumInputs(node) == 2 || NumInputs(node) == 3); in Resize()
82 TF_LITE_ENSURE(context, SizeOfDimension(hash, 1) <= 32); in Resize()
85 TF_LITE_ENSURE(context, NumDimensions(input) >= 1); in Resize()

123