Home
last modified time | relevance | path

Searched refs:kTfLiteUInt8 (Results 1 – 25 of 147) sorted by relevance

123456

/external/tensorflow/tensorflow/lite/delegates/hexagon/
Dutils.cc145 {{kTfLiteUInt8, kTfLiteInt8}, {kTfLiteUInt8, kTfLiteInt8}})) in IsNodeSupportedByHexagon()
154 {{kTfLiteUInt8, kTfLiteInt8}, {kTfLiteUInt8, kTfLiteInt8}})) in IsNodeSupportedByHexagon()
164 {{kTfLiteUInt8, kTfLiteInt8}, {kTfLiteUInt8, kTfLiteInt8}})) in IsNodeSupportedByHexagon()
178 {{kTfLiteUInt8, kTfLiteInt8}, {kTfLiteInt32}}) && in IsNodeSupportedByHexagon()
183 node, context, {{kTfLiteUInt8, kTfLiteInt8}, {kTfLiteInt32}}) || in IsNodeSupportedByHexagon()
198 node, context, {{kTfLiteUInt8, kTfLiteInt8}, {kTfLiteInt32}}) && in IsNodeSupportedByHexagon()
203 {{kTfLiteUInt8, kTfLiteInt8}, in IsNodeSupportedByHexagon()
204 {kTfLiteUInt8, kTfLiteInt8}, in IsNodeSupportedByHexagon()
227 if (!TensorTypeMatch(node->inputs->data[i], context, kTfLiteUInt8) && in IsNodeSupportedByHexagon()
234 if (!InputsWithCorrectTypes(node, context, {{kTfLiteUInt8, kTfLiteInt8}})) in IsNodeSupportedByHexagon()
[all …]
/external/tensorflow/tensorflow/lite/tools/versioning/
Dop_version_test.cc113 std::vector<TfLiteType>{kTfLiteUInt8, kTfLiteUInt8, kTfLiteUInt8}), in TEST()
143 .inputs = CreateOpSignatureTensorSpecs(kTfLiteUInt8), in SimpleVersioningTest()
172 .outputs = CreateOpSignatureTensorSpecs(kTfLiteUInt8), in SimpleOutputVersioningTest()
232 .inputs = CreateOpSignatureTensorSpecs(kTfLiteUInt8), in TEST()
258 .inputs = CreateOpSignatureTensorSpecs(kTfLiteUInt8), in TEST()
281 fake_op_sig.inputs = CreateOpSignatureTensorSpecs(kTfLiteUInt8, 3); in TEST()
283 fake_op_sig.inputs = CreateOpSignatureTensorSpecs(kTfLiteUInt8, 4); in TEST()
310 fake_op_sig.inputs = CreateOpSignatureTensorSpecs(kTfLiteUInt8); in TEST()
348 fake_op_sig.inputs = CreateOpSignatureTensorSpecs(kTfLiteUInt8, 4); in TEST()
375 fake_op_sig.inputs = CreateOpSignatureTensorSpecs(kTfLiteUInt8, 4, 5); in TEST()
[all …]
/external/tensorflow/tensorflow/lite/kernels/
Dquantize.cc121 TF_LITE_ENSURE(context, output->type == kTfLiteUInt8 || in Prepare()
135 input->type == kTfLiteInt8 || input->type == kTfLiteUInt8); in Prepare()
137 context, output->type == kTfLiteUInt8 || output->type == kTfLiteInt8); in Prepare()
190 case kTfLiteUInt8: in Eval()
216 case kTfLiteUInt8: in Eval()
305 case kTfLiteUInt8: in Eval()
316 case kTfLiteUInt8: { in Eval()
327 case kTfLiteUInt8: in Eval()
Dl2norm.cc56 output->type == kTfLiteUInt8 || in Prepare()
60 if (output->type == kTfLiteUInt8 || output->type == kTfLiteInt8) { in Prepare()
62 if (output->type == kTfLiteUInt8) { in Prepare()
115 } else if (output->type == kTfLiteUInt8) { in Eval()
Darg_min_max.cc102 case kTfLiteUInt8: in Prepare()
151 case kTfLiteUInt8: in Eval()
176 case kTfLiteUInt8: in Eval()
209 case kTfLiteUInt8: in Eval()
234 case kTfLiteUInt8: in Eval()
Dactivations.cc273 if (input->type == kTfLiteInt8 || input->type == kTfLiteUInt8 || in ReluPrepare()
306 if (output->type == kTfLiteUInt8 || output->type == kTfLiteInt8) { in HardSwishPrepare()
350 if (output->type == kTfLiteUInt8 || output->type == kTfLiteInt8 || in LeakyReluPrepare()
386 if (input->type == kTfLiteUInt8 || input->type == kTfLiteInt8) { in TanhPrepare()
405 if (input->type == kTfLiteUInt8) { in TanhPrepare()
481 if (input->type == kTfLiteUInt8 || input->type == kTfLiteInt8) { in SigmoidPrepare()
482 if (input->type == kTfLiteUInt8) { in SigmoidPrepare()
510 if (input->type == kTfLiteUInt8) { in SigmoidPrepare()
595 if (input->type == kTfLiteUInt8 || input->type == kTfLiteInt8) { in SoftmaxPrepare()
609 case kTfLiteUInt8: in SoftmaxPrepare()
[all …]
Ddiv.cc100 if (output->type == kTfLiteUInt8) { in Prepare()
165 if (input1->type == kTfLiteUInt8 && input2->type == kTfLiteUInt8 && in EvalQuantized()
166 output->type == kTfLiteUInt8) { in EvalQuantized()
237 } else if (output->type == kTfLiteUInt8) { in Eval()
Dcomparisons.cc98 if (input1->type == kTfLiteUInt8 || input1->type == kTfLiteInt8) { in ComparisonQuantized()
190 case kTfLiteUInt8: in EqualEval()
240 case kTfLiteUInt8: in NotEqualEval()
286 case kTfLiteUInt8: in GreaterEval()
327 case kTfLiteUInt8: in GreaterEqualEval()
368 case kTfLiteUInt8: in LessEval()
409 case kTfLiteUInt8: in LessEqualEval()
Dkernel_util_test.cc466 input.type = kTfLiteUInt8; in TEST_F()
483 filter.type = kTfLiteUInt8; in TEST_F()
505 bias.type = kTfLiteUInt8; in TEST_F()
525 output.type = kTfLiteUInt8; in TEST_F()
678 input.type = kTfLiteUInt8; in TEST_F()
695 filter.type = kTfLiteUInt8; in TEST_F()
733 output.type = kTfLiteUInt8; in TEST_F()
776 input.type = kTfLiteUInt8; in TEST_F()
793 filter.type = kTfLiteUInt8; in TEST_F()
815 output.type = kTfLiteUInt8; in TEST_F()
[all …]
Dwhere.cc96 case kTfLiteUInt8: in Prepare()
138 case kTfLiteUInt8: in Eval()
186 case kTfLiteUInt8: in Eval()
Dmul.cc102 if (output->type == kTfLiteUInt8 || output->type == kTfLiteInt8 || in Prepare()
192 (input1->type == kTfLiteUInt8 || input1->type == kTfLiteInt8 || in EvalQuantized()
257 (output->type == kTfLiteUInt8 || output->type == kTfLiteInt8)) { in EvalQuantized()
307 } else if (output->type == kTfLiteUInt8 || output->type == kTfLiteInt8 || in Eval()
Dsub.cc97 TF_LITE_ENSURE(context, output->type == kTfLiteUInt8 || in PrepareGeneralSubOp()
105 if (output->type == kTfLiteUInt8) { in PrepareGeneralSubOp()
287 if (output->type == kTfLiteUInt8 || output->type == kTfLiteInt8 || in Prepare()
414 } else if (output->type == kTfLiteUInt8) { in EvalQuantized()
456 } else if (output->type == kTfLiteUInt8 || output->type == kTfLiteInt8 || in Eval()
Ddequantize.h64 case kTfLiteUInt8: in PerChannelDequantizeImpl()
94 case kTfLiteUInt8: in DequantizeImpl()
/external/tensorflow/tensorflow/lite/tools/evaluation/stages/
Dtflite_inference_stage_test.cc105 EXPECT_EQ(tensor->type, kTfLiteUInt8); in TEST()
115 EXPECT_EQ(tensor->type, kTfLiteUInt8); in TEST()
139 EXPECT_EQ(tensor->type, kTfLiteUInt8); in TEST()
149 EXPECT_EQ(tensor->type, kTfLiteUInt8); in TEST()
Dinference_profiler_stage.cc99 if (model_input_type == kTfLiteUInt8 || model_input_type == kTfLiteInt8 || in Init()
119 if (model_output_type == kTfLiteUInt8 || model_output_type == kTfLiteInt8 || in Init()
144 if (model_input_type == kTfLiteUInt8) { in Run()
173 if (model_output_type == kTfLiteUInt8) { in Run()
Dtopk_accuracy_eval_stage.cc71 model_output_type_ != kTfLiteUInt8 && model_output_type_ != kTfLiteInt8) { in Init()
96 } else if (model_output_type_ == kTfLiteUInt8) { in Run()
/external/tensorflow/tensorflow/lite/examples/label_image/
Dlabel_image_test.cc34 s.input_type = kTfLiteUInt8; in TEST()
51 get_top_n<uint8_t>(in, 10, 5, 0.025, &top_results, kTfLiteUInt8); in TEST()
/external/libtextclassifier/native/tensorflow_models/seq_flow_lite/tflite_ops/
Dlayer_norm.cc59 TF_LITE_ENSURE_EQ(context, input->type, kTfLiteUInt8); in Resize()
62 TF_LITE_ENSURE_EQ(context, offset->type, kTfLiteUInt8); in Resize()
65 TF_LITE_ENSURE_EQ(context, scale->type, kTfLiteUInt8); in Resize()
72 TF_LITE_ENSURE_EQ(context, output->type, kTfLiteUInt8); in Resize()
398 if (input->type == kTfLiteUInt8) { in Eval()
412 if (input->type == kTfLiteUInt8) { in Eval()
/external/tensorflow/tensorflow/lite/
Dtype_to_tflitetype_test.cc39 EXPECT_EQ(kTfLiteUInt8, in TEST()
40 typeToTfLiteType<TfLiteTypeToType<kTfLiteUInt8>::Type>()); in TEST()
/external/tensorflow/tensorflow/lite/delegates/nnapi/
Dnnapi_delegate.cc176 case kTfLiteUInt8: in IsFloatOrUInt8()
185 case kTfLiteUInt8: in IsQuantized()
206 case kTfLiteUInt8: in IsFloatOrQuantized()
227 case kTfLiteUInt8: in IsFloatQuantizedOrInt32()
273 weights_tensor.type == kTfLiteUInt8) && in NeedInt8Conversion()
465 case kTfLiteUInt8: in ConvertTensorTypeToNNType()
482 } else if (ann_type_equivalent == kTfLiteUInt8) { in ConvertTensorTypeToNNType()
1013 if (tensor.type == kTfLiteInt8 || tensor.type == kTfLiteUInt8) { in TransformHardSwishIntoSupportedOps()
1210 int nn_type = (output_tensor.type == kTfLiteUInt8 || need_int8_conversion) in AppendRequantize()
1404 case kTfLiteUInt8: in TransformSquaredDifferenceIntoSupportedOps()
[all …]
/external/tensorflow/tensorflow/lite/delegates/flex/
Dutil.cc87 case kTfLiteUInt8: in GetTensorFlowDataType()
123 return kTfLiteUInt8; in GetTensorFlowLiteType()
162 case kTfLiteUInt8: in TfLiteTypeToTfTypeName()
/external/tensorflow/tensorflow/lite/testing/nnapi_tflite_zip_tests/
Dtflite_driver.cc185 case kTfLiteUInt8: in Check()
293 case kTfLiteUInt8: { in SetInput()
332 case kTfLiteUInt8: in SetExpectation()
399 case kTfLiteUInt8: in ReadOutput()
/external/tensorflow/tensorflow/lite/tools/
Dutils.cc119 case kTfLiteUInt8: { in CreateRandomTensorData()
160 } else if (type == kTfLiteUInt8) { in GetDataRangesForType()
/external/tensorflow/tensorflow/lite/python/interpreter_wrapper/
Dnumpy.cc51 case kTfLiteUInt8: in TfLiteTypeToPyArrayType()
92 return kTfLiteUInt8; in TfLiteTypeFromPyType()
/external/tensorflow/tensorflow/lite/delegates/gpu/common/
Dquantization_util.cc51 } else if (original_tflite_tensor.type == kTfLiteUInt8) { in DequantizeInput()
79 } else if (original_tflite_tensor.type == kTfLiteUInt8) { in QuantizeOutput()

123456