Home
last modified time | relevance | path

Searched refs:kTfLiteFloat32 (Results 1 – 25 of 261) sorted by relevance

1234567891011

/external/tensorflow/tensorflow/lite/delegates/flex/
Ddelegate_test.cc55 AddTensors(9, {0, 3}, {8}, kTfLiteFloat32, {3}); in TEST_F()
76 ASSERT_EQ(GetType(8), kTfLiteFloat32); in TEST_F()
118 AddTensors(9, {0, 3}, {8}, kTfLiteFloat32, {3}); in TEST_F()
140 AddTensors(10, {0}, {9}, kTfLiteFloat32, {3}); in TEST_F()
164 AddTensors(10, {0, 1}, {2}, kTfLiteFloat32, {3}); in TEST_F()
182 AddTensors(10, {0, 1}, {2}, kTfLiteFloat32, {3}); in TEST_F()
211 AddTensors(9, {0, 3}, {8}, kTfLiteFloat32, {3}); in TEST_F()
229 AddTensors(10, {0}, {9}, kTfLiteFloat32, {3}); in TEST_F()
259 AddTensors(9, {0, 3}, {8}, kTfLiteFloat32, {3}); in TEST_F()
280 ASSERT_EQ(GetType(8), kTfLiteFloat32); in TEST_F()
[all …]
Dkernel_test.cc62 AddTensors(9, {0, 3}, {8}, kTfLiteFloat32, {3}); in TEST_F()
96 AddTensors(2, {0}, {1}, kTfLiteFloat32, {3}); in TEST_F()
107 AddTensors(3, {0}, {1, 2}, kTfLiteFloat32, {3}); in TEST_F()
121 AddTensors(2, {0}, {1}, kTfLiteFloat32, {3}); in TEST_F()
137 AddTensors(4, {0}, {3}, kTfLiteFloat32, {3}); in TEST_F()
152 AddTensors(9, {0, 3}, {8}, kTfLiteFloat32, {3}); in TEST_F()
181 AddTensors(18, {0, 1}, {17}, kTfLiteFloat32, {3}); in TEST_F()
257 AddTensors(kMaxTensors, {kInput, kOnes, kTwos}, {12}, kTfLiteFloat32, {3}); in TEST_F()
290 AddTensors(kMaxTensors, {kInput, kOnes, kTwos}, {12}, kTfLiteFloat32, {3}); in TEST_F()
326 AddTensors(kMaxTensors, {kInput, kOnes, kTwos}, {12}, kTfLiteFloat32, {3}); in TEST_F()
Dutil_test.cc84 EXPECT_EQ(dst.type, kTfLiteFloat32); in TEST()
89 EXPECT_EQ(dst.type, kTfLiteFloat32); in TEST()
111 EXPECT_EQ(TF_FLOAT, GetTensorFlowDataType(kTfLiteFloat32)); in TEST()
129 EXPECT_EQ(kTfLiteFloat32, GetTensorFlowLiteType(TF_FLOAT)); in TEST()
/external/tensorflow/tensorflow/lite/tools/serialization/
Dwriter_lib_test.cc63 interpreter.SetTensorParametersReadWrite(0, kTfLiteFloat32, "a", {3}, in TEST_P()
66 1, kTfLiteFloat32, "b", {3}, TfLiteQuantization(), in TEST_P()
68 interpreter.SetTensorParametersReadWrite(2, kTfLiteFloat32, "c", {3}, in TEST_P()
106 interpreter.SetTensorParametersReadWrite(0, kTfLiteFloat32, "a", {3}, in TEST_P()
109 1, kTfLiteFloat32, "b", {3}, TfLiteQuantization(), in TEST_P()
111 interpreter.SetTensorParametersReadWrite(2, kTfLiteFloat32, "c", {3}, in TEST_P()
140 interpreter.SetTensorParametersReadWrite(0, kTfLiteFloat32, "a", {3}, in TEST_P()
143 1, kTfLiteFloat32, "b", {3}, TfLiteQuantization(), in TEST_P()
145 interpreter.SetTensorParametersReadWrite(2, kTfLiteFloat32, "c", {3}, in TEST_P()
147 interpreter.SetTensorParametersReadWrite(3, kTfLiteFloat32, "d", {3}, in TEST_P()
[all …]
/external/tensorflow/tensorflow/lite/experimental/resource/
Dresource_variable_test.cc36 TfLiteTensorReset(TfLiteType::kTfLiteFloat32, nullptr, dims, {}, in InitTensor()
55 EXPECT_EQ(kTfLiteFloat32, value->type); in TEST()
81 EXPECT_EQ(kTfLiteFloat32, value->type); in TEST()
108 EXPECT_EQ(kTfLiteFloat32, value->type); in TEST()
120 EXPECT_EQ(kTfLiteFloat32, value->type); in TEST()
149 EXPECT_EQ(kTfLiteFloat32, value->type); in TEST()
161 EXPECT_EQ(kTfLiteFloat32, value->type); in TEST()
/external/tensorflow/tensorflow/lite/delegates/
Ddelegate_test_util.cc103 subgraph->SetTensorParametersReadWrite(0, kTfLiteFloat32, "", dims.size(), in SetUpSubgraph()
105 subgraph->SetTensorParametersReadWrite(1, kTfLiteFloat32, "", dims.size(), in SetUpSubgraph()
107 subgraph->SetTensorParametersReadWrite(2, kTfLiteFloat32, "", dims.size(), in SetUpSubgraph()
109 subgraph->SetTensorParametersReadWrite(3, kTfLiteFloat32, "", dims.size(), in SetUpSubgraph()
111 subgraph->SetTensorParametersReadWrite(4, kTfLiteFloat32, "", dims.size(), in SetUpSubgraph()
342 interpreter_->SetTensorParametersReadWrite(0, kTfLiteFloat32, "", {1}, quant); in SetUp()
347 interpreter_->SetTensorParametersReadWrite(2, kTfLiteFloat32, "", {1}, quant); in SetUp()
348 interpreter_->SetTensorParametersReadWrite(3, kTfLiteFloat32, "", {1}, quant); in SetUp()
353 interpreter_->SetTensorParametersReadWrite(5, kTfLiteFloat32, "", {1}, quant); in SetUp()
354 interpreter_->SetTensorParametersReadWrite(6, kTfLiteFloat32, "", {1}, quant); in SetUp()
[all …]
/external/tensorflow/tensorflow/lite/
Dtype_to_tflitetype_test.cc35 EXPECT_EQ(kTfLiteFloat32, in TEST()
36 typeToTfLiteType<TfLiteTypeToType<kTfLiteFloat32>::Type>()); in TEST()
56 EXPECT_EQ(kTfLiteFloat32, typeToTfLiteType<float>()); in TEST()
60 std::is_same<float, TfLiteTypeToType<kTfLiteFloat32>::Type>::value, in TEST()
Dinterpreter_test.cc96 interpreter.SetTensorParametersReadWrite(tensor_index, kTfLiteFloat32, "", in TEST()
192 {kTfLiteFloat32, sizeof(float)}, {kTfLiteInt32, sizeof(int32_t)}, in TEST()
277 {kTfLiteFloat32, sizeof(float), reinterpret_cast<const char*>(floats)}, in TEST()
319 } cases[] = {{kTfLiteFloat32}, {kTfLiteInt32}, {kTfLiteUInt32}, in TEST()
381 0, kTfLiteFloat32, "", {3}, TfLiteQuantizationParams()), in TEST()
399 0, kTfLiteFloat32, "", {3}, TfLiteQuantizationParams()), in TEST()
414 0, kTfLiteFloat32, "", {3}, TfLiteQuantizationParams()), in TEST()
435 0, kTfLiteFloat32, "", {3}, TfLiteQuantizationParams()), in TEST()
438 1, kTfLiteFloat32, "", {3}, TfLiteQuantizationParams()), in TEST()
466 0, kTfLiteFloat32, "", {3}, TfLiteQuantizationParams()), in TEST()
[all …]
/external/tensorflow/tensorflow/lite/delegates/gpu/common/
Dmodel_builder_test.cc41 tflite_tensor.type = TfLiteType::kTfLiteFloat32; in TEST()
107 tflite_tensor.type = TfLiteType::kTfLiteFloat32; in TEST()
120 tflite_tensor.type = TfLiteType::kTfLiteFloat32; in TEST()
259 1, TfLiteType::kTfLiteFloat32, "t1", dims, quantization, false), in InterpreterFp16()
263 3, TfLiteType::kTfLiteFloat32, "t3", dims, quantization, false), in InterpreterFp16()
403 TfLiteType::kTfLiteFloat32); in TEST()
405 TfLiteType::kTfLiteFloat32); in TEST()
457 TfLiteType::kTfLiteFloat32); in TEST()
459 TfLiteType::kTfLiteFloat32); in TEST()
511 1, TfLiteType::kTfLiteFloat32, "t1", dims, quantization, false), in InterpreterFp32()
[all …]
/external/tensorflow/tensorflow/lite/delegates/utils/
Dsimple_delegate_test.cc35 interpreter_->SetTensorParametersReadWrite(0, kTfLiteFloat32, "", {3}, in SetUp()
37 interpreter_->SetTensorParametersReadWrite(1, kTfLiteFloat32, "", {3}, in SetUp()
39 interpreter_->SetTensorParametersReadWrite(2, kTfLiteFloat32, "", {3}, in SetUp()
41 interpreter_->SetTensorParametersReadWrite(3, kTfLiteFloat32, "", {3}, in SetUp()
43 interpreter_->SetTensorParametersReadWrite(4, kTfLiteFloat32, "", {3}, in SetUp()
/external/tensorflow/tensorflow/lite/tools/evaluation/stages/
Dtopk_accuracy_eval_stage_test.cc78 TfLiteType model_output_type = kTfLiteFloat32; in TEST()
94 TfLiteType model_output_type = kTfLiteFloat32; in TEST()
112 TfLiteType model_output_type = kTfLiteFloat32; in TEST()
128 TfLiteType model_output_type = kTfLiteFloat32; in TEST()
159 TfLiteType model_output_type = kTfLiteFloat32; in TEST()
176 TfLiteType model_output_type = kTfLiteFloat32; in TEST()
197 TfLiteType model_output_type = kTfLiteFloat32; in TEST()
297 kTfLiteFloat32); in TEST_F()
Dimage_preprocessing_stage_test.cc36 kTfLiteFloat32); in TEST()
45 kTfLiteFloat32); in TEST()
53 kTfLiteFloat32); in TEST()
65 kTfLiteFloat32); in TEST()
106 kTfLiteFloat32); in TEST()
272 kTfLiteFloat32); in TEST()
Dinference_profiler_stage.cc99 model_input_type == kTfLiteFloat32) { in Init()
119 model_output_type == kTfLiteFloat32) { in Init()
153 } else if (model_input_type == kTfLiteFloat32) { in Run()
180 } else if (model_output_type == kTfLiteFloat32) { in Run()
/external/tensorflow/tensorflow/lite/kernels/
Ddetection_postprocess.cc175 detection_boxes->type = kTfLiteFloat32; in Prepare()
184 detection_classes->type = kTfLiteFloat32; in Prepare()
192 detection_scores->type = kTfLiteFloat32; in Prepare()
200 num_detections->type = kTfLiteFloat32; in Prepare()
212 decoded_boxes->type = kTfLiteFloat32; in Prepare()
219 scores->type = kTfLiteFloat32; in Prepare()
313 case kTfLiteFloat32: { in DecodeCenterSizeBoxes()
319 TF_LITE_ENSURE_EQ(context, input_anchors->type, kTfLiteFloat32); in DecodeCenterSizeBoxes()
352 TF_LITE_ENSURE_EQ(context, decoded_boxes->type, kTfLiteFloat32); in DecodeCenterSizeBoxes()
440 TF_LITE_ENSURE_EQ(context, decoded_boxes->type, kTfLiteFloat32); in NonMaxSuppressionSingleClassHelper()
[all …]
Darg_min_max.cc97 case kTfLiteFloat32: in Prepare()
153 case kTfLiteFloat32: in Eval()
175 case kTfLiteFloat32: in Eval()
205 case kTfLiteFloat32: in Eval()
227 case kTfLiteFloat32: in Eval()
Dnon_max_suppression.cc85 TF_LITE_ENSURE_EQ(context, input_boxes->type, kTfLiteFloat32); in Prepare()
92 TF_LITE_ENSURE_EQ(context, input_scores->type, kTfLiteFloat32); in Prepare()
115 TF_LITE_ENSURE_EQ(context, input_iou_threshold->type, kTfLiteFloat32); in Prepare()
121 TF_LITE_ENSURE_EQ(context, input_iou_threshold->type, kTfLiteFloat32); in Prepare()
128 TF_LITE_ENSURE_EQ(context, input_sigma->type, kTfLiteFloat32); in Prepare()
142 output_selected_scores->type = kTfLiteFloat32; in Prepare()
Dsvdf.cc95 input->type == kTfLiteFloat32 || input->type == kTfLiteInt8); in Prepare()
161 scratch_tensor->type = kTfLiteFloat32; in Prepare()
188 scaling_factors->type = kTfLiteFloat32; in Prepare()
204 float_weights_time->type = kTfLiteFloat32; in Prepare()
219 zero_points->type = kTfLiteFloat32; in Prepare()
233 row_sums->type = kTfLiteFloat32; in Prepare()
306 case kTfLiteFloat32: { in Eval()
320 if (input->type == kTfLiteFloat32) { in Eval()
Delementwise.cc54 return type == kTfLiteFloat32; in IsNumericSupportedType()
62 return type == kTfLiteFloat32 || type == kTfLiteInt8 || type == kTfLiteInt16; in IsAbsSupportedType()
66 return type == kTfLiteFloat32 || type == kTfLiteInt8; in IsRsqrtSupportedType()
167 return EvalImpl<float>(context, node, float_func, kTfLiteFloat32); in EvalNumeric()
209 case kTfLiteFloat32: in AbsEval()
275 case kTfLiteFloat32: in RsqrtEval()
/external/tensorflow/tensorflow/lite/examples/label_image/
Dbitmap_helpers_impl.h49 0, kTfLiteFloat32, "input", in resize()
54 2, kTfLiteFloat32, "output", in resize()
87 case kTfLiteFloat32: in resize()
/external/tensorflow/tensorflow/lite/micro/kernels/
Dsvdf.cc231 input->type == kTfLiteFloat32 || input->type == kTfLiteInt8); in Prepare()
315 TF_LITE_ENSURE_EQ(context, weights_feature->type, kTfLiteFloat32); in Prepare()
316 TF_LITE_ENSURE_EQ(context, weights_time->type, kTfLiteFloat32); in Prepare()
317 TF_LITE_ENSURE_EQ(context, activation_state->type, kTfLiteFloat32); in Prepare()
319 TF_LITE_ENSURE_EQ(context, bias->type, kTfLiteFloat32); in Prepare()
321 TF_LITE_ENSURE_TYPES_EQ(context, output->type, kTfLiteFloat32); in Prepare()
354 case kTfLiteFloat32: { in Eval()
Dl2norm.cc59 TF_LITE_ENSURE(context, output->type == kTfLiteFloat32 || in Prepare()
66 } else if (output->type == kTfLiteFloat32) { in Prepare()
107 if (output->type == kTfLiteFloat32) { in Eval()
/external/tensorflow/tensorflow/lite/micro/examples/magic_wand/
Dmagic_wand_test.cc78 TF_LITE_MICRO_EXPECT_EQ(kTfLiteFloat32, input->type); in TF_LITE_MICRO_TEST()
100 TF_LITE_MICRO_EXPECT_EQ(kTfLiteFloat32, output->type); in TF_LITE_MICRO_TEST()
137 TF_LITE_MICRO_EXPECT_EQ(kTfLiteFloat32, output->type); in TF_LITE_MICRO_TEST()
/external/tensorflow/tensorflow/lite/micro/kernels/cmsis_nn/
Dsvdf.cc318 input->type == kTfLiteFloat32 || input->type == kTfLiteInt8); in Prepare()
401 TF_LITE_ENSURE_EQ(context, weights_feature->type, kTfLiteFloat32); in Prepare()
402 TF_LITE_ENSURE_EQ(context, weights_time->type, kTfLiteFloat32); in Prepare()
403 TF_LITE_ENSURE_EQ(context, activation_state->type, kTfLiteFloat32); in Prepare()
405 TF_LITE_ENSURE_EQ(context, bias->type, kTfLiteFloat32); in Prepare()
407 TF_LITE_ENSURE_TYPES_EQ(context, output->type, kTfLiteFloat32); in Prepare()
440 case kTfLiteFloat32: { in Eval()
/external/tensorflow/tensorflow/lite/delegates/nnapi/
Dnnapi_delegate.cc140 case kTfLiteFloat32: in IsFloat()
149 case kTfLiteFloat32: in IsFloatOrUInt8()
179 case kTfLiteFloat32: in IsFloatOrQuantized()
190 case kTfLiteFloat32: in IsFloatOrInt32()
200 case kTfLiteFloat32: in IsFloatQuantizedOrInt32()
399 case kTfLiteFloat32: in ConvertTensorTypeToNNType()
1163 *type = kTfLiteFloat32; in GetEquivalentToANNType()
1293 case kTfLiteFloat32: in AddTensor()
1708 EXPECT_INPUT_TYPE_IN(input_type, kTfLiteFloat16, kTfLiteFloat32, in Validate()
1882 ExpectTypeIn(output.type, {kTfLiteFloat32, kTfLiteUInt8, kTfLiteInt8}, in Validate()
[all …]
/external/tensorflow/tensorflow/lite/delegates/coreml/builders/
Dfully_connected_op_builder.cc53 if (weights_->type == kTfLiteFloat32) { in FillCoreMLWeights()
71 if (bias_->type == kTfLiteFloat32) { in FillCoreMLBias()
139 return type == kTfLiteFloat32 || type == kTfLiteFloat16; in IsFloatType()

1234567891011