Home
last modified time | relevance | path

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

12345

/external/tensorflow/tensorflow/lite/delegates/flex/
Ddelegate_test.cc52 AddTensors(9, {0, 3}, {8}, kTfLiteFloat32, {3}); in TEST_F()
73 ASSERT_EQ(GetType(8), kTfLiteFloat32); in TEST_F()
115 AddTensors(9, {0, 3}, {8}, kTfLiteFloat32, {3}); in TEST_F()
137 AddTensors(10, {0}, {9}, kTfLiteFloat32, {3}); in TEST_F()
161 AddTensors(10, {0, 1}, {2}, kTfLiteFloat32, {3}); in TEST_F()
179 AddTensors(10, {0, 1}, {2}, kTfLiteFloat32, {3}); in TEST_F()
208 AddTensors(9, {0, 3}, {8}, kTfLiteFloat32, {3}); in TEST_F()
226 AddTensors(10, {0}, {9}, kTfLiteFloat32, {3}); in TEST_F()
256 AddTensors(9, {0, 3}, {8}, kTfLiteFloat32, {3}); in TEST_F()
277 ASSERT_EQ(GetType(8), kTfLiteFloat32); in TEST_F()
[all …]
Dkernel_test.cc103 AddTensors(9, {0, 3}, {8}, kTfLiteFloat32, {3}); in TEST_F()
140 AddTensors(2, {0}, {1}, kTfLiteFloat32, {3}); in TEST_F()
153 AddTensors(3, {0}, {1, 2}, kTfLiteFloat32, {3}); in TEST_F()
169 AddTensors(2, {0}, {1}, kTfLiteFloat32, {3}); in TEST_F()
187 AddTensors(4, {0}, {3}, kTfLiteFloat32, {3}); in TEST_F()
202 AddTensors(9, {0, 3}, {8}, kTfLiteFloat32, {3}); in TEST_F()
233 AddTensors(18, {0, 1}, {17}, kTfLiteFloat32, {3}); in TEST_F()
317 AddTensors(kMaxTensors, {kInput, kOnes, kTwos}, {12}, kTfLiteFloat32, {3}); in TEST_F()
350 AddTensors(kMaxTensors, {kInput, kOnes, kTwos}, {12}, kTfLiteFloat32, {3}); in TEST_F()
386 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()
122 EXPECT_EQ(kTfLiteFloat32, GetTensorFlowLiteType(TF_FLOAT)); in TEST()
Dutil.cc61 case kTfLiteFloat32: in GetTensorFlowDataType()
85 return kTfLiteFloat32; in GetTensorFlowLiteType()
/external/tensorflow/tensorflow/lite/kernels/
Drange.cc60 case kTfLiteFloat32: { in ResizeOutput()
91 TF_LITE_ENSURE(context, dtype == kTfLiteInt32 || dtype == kTfLiteFloat32); in Prepare()
100 case kTfLiteFloat32: in Prepare()
101 output->type = kTfLiteFloat32; in Prepare()
149 case kTfLiteFloat32: { in Eval()
Darg_min_max.cc81 case kTfLiteFloat32: in Prepare()
133 case kTfLiteFloat32: in Eval()
151 case kTfLiteFloat32: in Eval()
174 case kTfLiteFloat32: in Eval()
189 case kTfLiteFloat32: in Eval()
Dlstm.cc339 TF_LITE_ENSURE_EQ(context, input->type, kTfLiteFloat32); in Prepare()
387 input->type == kTfLiteFloat32); in Prepare()
462 scaling_factors->type = kTfLiteFloat32; in Prepare()
474 prod_scaling_factors->type = kTfLiteFloat32; in Prepare()
490 recovered_cell_weights->type = kTfLiteFloat32; in Prepare()
579 case kTfLiteFloat32: { in Eval()
747 if (input->type == kTfLiteFloat32 && in Eval()
748 prev_activation->type == kTfLiteFloat32 && in Eval()
749 weights->type == kTfLiteFloat32 && bias->type == kTfLiteFloat32 && in Eval()
750 prev_state->type == kTfLiteFloat32 && state_out->type == kTfLiteFloat32 && in Eval()
[all …]
Dtranspose_conv.cc96 context->tensors[data->im2col_id].type = kTfLiteFloat32; in AllocateIm2colTensorIfRequired()
153 TF_LITE_ENSURE_EQ(context, input->type, kTfLiteFloat32); in Prepare()
154 TF_LITE_ENSURE_EQ(context, weights->type, kTfLiteFloat32); in Prepare()
155 TF_LITE_ENSURE_EQ(context, output->type, kTfLiteFloat32); in Prepare()
212 case kTfLiteFloat32: { in Eval()
Dfully_connected.cc120 if (data_type != kTfLiteFloat32 && data_type != kTfLiteInt32) { in Prepare()
135 if (input->type == kTfLiteFloat32 && in Prepare()
153 scaling_factors->type = kTfLiteFloat32; in Prepare()
212 TF_LITE_ENSURE_EQ(context, input->type, kTfLiteFloat32); in EvalHybrid()
216 TF_LITE_ENSURE_EQ(context, bias->type, kTfLiteFloat32); in EvalHybrid()
218 TF_LITE_ENSURE_EQ(context, output->type, kTfLiteFloat32); in EvalHybrid()
341 if (input->type == kTfLiteFloat32) { in EvalQuantized()
474 case kTfLiteFloat32: in Eval()
Dcomparisons.cc135 case kTfLiteFloat32: in EqualEval()
171 case kTfLiteFloat32: in NotEqualEval()
203 case kTfLiteFloat32: in GreaterEval()
235 case kTfLiteFloat32: in GreaterEqualEval()
267 case kTfLiteFloat32: in LessEval()
299 case kTfLiteFloat32: in LessEqualEval()
Dbasic_rnn.cc83 TF_LITE_ENSURE_EQ(context, input->type, kTfLiteFloat32); in Prepare()
99 input->type == kTfLiteFloat32 && (input_weights->type == kTfLiteUInt8 || in Prepare()
132 scaling_factors->type = kTfLiteFloat32; in Prepare()
232 case kTfLiteFloat32: in Eval()
Dbidirectional_sequence_lstm.cc183 TF_LITE_ENSURE(context, (input_to_forget_weights->type == kTfLiteFloat32) || in CheckLstmTensorDimensionsAndTypes()
298 TF_LITE_ENSURE_EQ(context, input_gate_bias->type, kTfLiteFloat32); in CheckLstmTensorDimensionsAndTypes()
305 TF_LITE_ENSURE_EQ(context, forget_gate_bias->type, kTfLiteFloat32); in CheckLstmTensorDimensionsAndTypes()
311 TF_LITE_ENSURE_EQ(context, cell_bias->type, kTfLiteFloat32); in CheckLstmTensorDimensionsAndTypes()
317 TF_LITE_ENSURE_EQ(context, output_gate_bias->type, kTfLiteFloat32); in CheckLstmTensorDimensionsAndTypes()
334 TF_LITE_ENSURE_EQ(context, projection_bias->type, kTfLiteFloat32); in CheckLstmTensorDimensionsAndTypes()
399 TF_LITE_ENSURE_EQ(context, input->type, kTfLiteFloat32); in Prepare()
685 scaling_factors->type = kTfLiteFloat32; in Prepare()
698 prod_scaling_factors->type = kTfLiteFloat32; in Prepare()
715 recovered_cell_weights->type = kTfLiteFloat32; in Prepare()
[all …]
Ddetection_postprocess.cc163 detection_boxes->type = kTfLiteFloat32; in Prepare()
170 detection_classes->type = kTfLiteFloat32; in Prepare()
176 detection_scores->type = kTfLiteFloat32; in Prepare()
182 num_detections->type = kTfLiteFloat32; in Prepare()
195 decoded_boxes->type = kTfLiteFloat32; in Prepare()
202 scores->type = kTfLiteFloat32; in Prepare()
288 case kTfLiteFloat32: in DecodeCenterSizeBoxes()
684 case kTfLiteFloat32: in NonMaxSuppressionMultiClass()
Dgather.cc58 case kTfLiteFloat32: in Prepare()
135 case kTfLiteFloat32: in Eval()
155 case kTfLiteFloat32: in Eval()
Dsvdf.cc142 TF_LITE_ENSURE_EQ(context, input->type, kTfLiteFloat32); in Prepare()
179 const bool is_hybrid_op = (input->type == kTfLiteFloat32 && in Prepare()
218 scaling_factors->type = kTfLiteFloat32; in Prepare()
232 float_weights_time->type = kTfLiteFloat32; in Prepare()
381 case kTfLiteFloat32: { in Eval()
Dlocal_response_norm.cc46 TF_LITE_ENSURE_EQ(context, output->type, kTfLiteFloat32); in Prepare()
66 if (output->type == kTfLiteFloat32) { in Eval()
/external/tensorflow/tensorflow/lite/experimental/writer/
Dwriter_lib_test.cc30 interpreter.SetTensorParametersReadWrite(0, kTfLiteFloat32, "a", {3}, in TEST()
33 1, kTfLiteFloat32, "b", {3}, TfLiteQuantizationParams(), in TEST()
35 interpreter.SetTensorParametersReadWrite(2, kTfLiteFloat32, "c", {3}, in TEST()
/external/tensorflow/tensorflow/lite/
Dinterpreter_test.cc90 interpreter.SetTensorParametersReadWrite(tensor_index, kTfLiteFloat32, "", in TEST()
165 {kTfLiteFloat32, sizeof(float)}, {kTfLiteInt32, sizeof(int32_t)}, in TEST()
246 {kTfLiteFloat32, sizeof(float), reinterpret_cast<const char*>(floats)}, in TEST()
286 {kTfLiteFloat32}, {kTfLiteInt32}, {kTfLiteUInt8}, in TEST()
349 0, kTfLiteFloat32, "", {3}, TfLiteQuantizationParams()), in TEST()
367 0, kTfLiteFloat32, "", {3}, TfLiteQuantizationParams()), in TEST()
382 0, kTfLiteFloat32, "", {3}, TfLiteQuantizationParams()), in TEST()
403 0, kTfLiteFloat32, "", {3}, TfLiteQuantizationParams()), in TEST()
406 1, kTfLiteFloat32, "", {3}, TfLiteQuantizationParams()), in TEST()
434 0, kTfLiteFloat32, "", {3}, TfLiteQuantizationParams()), in TEST()
[all …]
Dmodel_test.cc144 ASSERT_EQ(i0->type, kTfLiteFloat32); in TEST()
148 ASSERT_EQ(i1->type, kTfLiteFloat32); in TEST()
152 ASSERT_EQ(o0->type, kTfLiteFloat32); in TEST()
156 ASSERT_EQ(o1->type, kTfLiteFloat32); in TEST()
/external/tensorflow/tensorflow/lite/delegates/nnapi/
Dnnapi_delegate.cc57 case kTfLiteFloat32: in IsFloat()
431 case kTfLiteFloat32: in AddTensor()
779 context->tensors[node->inputs->data[0]].type == kTfLiteFloat32) { in Map()
786 context->tensors[node->inputs->data[0]].type == kTfLiteFloat32) { in Map()
799 context->tensors[node->inputs->data[0]].type == kTfLiteFloat32) { in Map()
813 (context->tensors[node->inputs->data[0]].type == kTfLiteFloat32 || in Map()
857 kTfLiteFloat32) { in Map()
881 .type == kTfLiteFloat32) { in Map()
912 .type == kTfLiteFloat32) { in Map()
950 context->tensors[node->inputs->data[0]].type == kTfLiteFloat32 && in Map()
[all …]
/external/tensorflow/tensorflow/lite/examples/label_image/
Dbitmap_helpers_impl.h50 0, kTfLiteFloat32, "input", in resize()
55 2, kTfLiteFloat32, "output", in resize()
/external/tensorflow/tensorflow/lite/testing/nnapi_tflite_zip_tests/
Dtflite_driver.cc179 case kTfLiteFloat32: in Check()
275 case kTfLiteFloat32: { in SetInput()
323 case kTfLiteFloat32: in SetExpectation()
393 case kTfLiteFloat32: in ReadOutput()
/external/tensorflow/tensorflow/lite/testing/
Dtflite_driver.cc179 case kTfLiteFloat32: in Check()
274 case kTfLiteFloat32: { in SetInput()
329 case kTfLiteFloat32: in SetExpectation()
396 case kTfLiteFloat32: in ReadOutput()
/external/libtextclassifier/utils/tflite/
Dencoder_common.cc61 case kTfLiteFloat32: { in CopyValuesToTensorAndPadOrTruncate()
81 case kTfLiteFloat32: { in CopyValuesToTensorAndPadOrTruncate()
/external/tensorflow/tensorflow/lite/python/interpreter_wrapper/
Dpython_utils.cc33 case kTfLiteFloat32: in TfLiteTypeToPyArrayType()
62 return kTfLiteFloat32; in TfLiteTypeFromPyArray()

12345