Home
last modified time | relevance | path

Searched refs:TfLiteType (Results 1 – 25 of 109) sorted by relevance

12345

/external/tensorflow/tensorflow/lite/experimental/resource/
Dstatic_hashtable.h36 explicit StaticHashtable(TfLiteType key_type, TfLiteType value_type) in StaticHashtable()
54 TfLiteType GetKeyType() const override { return key_type_; } in GetKeyType()
55 TfLiteType GetValueType() const override { return value_type_; } in GetValueType()
69 TfLiteType key_type_;
70 TfLiteType value_type_;
77 TfLiteType key_type, TfLiteType value_type);
Dlookup_interfaces.h42 virtual TfLiteType GetKeyType() const = 0;
43 virtual TfLiteType GetValueType() const = 0;
54 TfLiteType key_dtype,
55 TfLiteType value_dtype);
Dstatic_hashtable.cc83 LookupInterface* CreateStaticHashtable(TfLiteType key_type, in CreateStaticHashtable()
84 TfLiteType value_type) { in CreateStaticHashtable()
97 TfLiteType key_dtype, in CreateHashtableResourceIfNotAvailable()
98 TfLiteType value_dtype) { in CreateHashtableResourceIfNotAvailable()
/external/tensorflow/tensorflow/lite/delegates/gpu/common/
Dmodel_builder_test.cc41 tflite_tensor.type = TfLiteType::kTfLiteFloat32; in TEST()
56 tflite_tensor.type = TfLiteType::kTfLiteInt32; in TEST()
72 tflite_tensor.type = TfLiteType::kTfLiteInt64; in TEST()
89 tflite_tensor.type = TfLiteType::kTfLiteUInt8; in TEST()
107 tflite_tensor.type = TfLiteType::kTfLiteFloat32; in TEST()
120 tflite_tensor.type = TfLiteType::kTfLiteFloat32; in TEST()
242 0, TfLiteType::kTfLiteFloat16, "t0", dims, quantization, false), in InterpreterFp16()
246 2, TfLiteType::kTfLiteFloat16, "t2", dims, quantization, false), in InterpreterFp16()
259 1, TfLiteType::kTfLiteFloat32, "t1", dims, quantization, false), in InterpreterFp16()
263 3, TfLiteType::kTfLiteFloat32, "t3", dims, quantization, false), in InterpreterFp16()
[all …]
/external/tensorflow/tensorflow/lite/examples/label_image/
Dget_top_n.h27 TfLiteType input_type);
31 std::vector<std::pair<float, int>>*, TfLiteType);
34 TfLiteType);
37 TfLiteType);
/external/tensorflow/tensorflow/lite/c/
Dbuiltin_op_data.h352 TfLiteType in_data_type;
353 TfLiteType out_data_type;
402 TfLiteType output_type;
406 TfLiteType output_type;
420 TfLiteType out_type;
456 TfLiteType index_out_type;
Dcommon.h259 const char* TfLiteTypeGetName(TfLiteType type);
384 TfLiteType type;
523 TfLiteType type;
579 TfLiteType type;
596 void TfLiteTensorReset(TfLiteType type, const char* name, TfLiteIntArray* dims,
/external/tensorflow/tensorflow/lite/python/interpreter_wrapper/
Dnumpy.h65 int TfLiteTypeToPyArrayType(TfLiteType tf_lite_type);
67 TfLiteType TfLiteTypeFromPyType(int py_type);
69 TfLiteType TfLiteTypeFromPyArray(PyArrayObject* array);
Dnumpy.cc35 int TfLiteTypeToPyArrayType(TfLiteType tf_lite_type) { in TfLiteTypeToPyArrayType()
75 TfLiteType TfLiteTypeFromPyType(int py_type) { in TfLiteTypeFromPyType()
111 TfLiteType TfLiteTypeFromPyArray(PyArrayObject* array) { in TfLiteTypeFromPyArray()
/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()
144 TfLiteType model_output_type = kTfLiteComplex64; 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()
245 TfLiteType model_output_type) { in VerifyCorrectBehaviorForType()
Dtopk_accuracy_eval_stage.h48 TfLiteType model_output_type, in SetTaskInfo()
70 TfLiteType model_output_type_ = kTfLiteNoType;
Dinference_profiler_stage.cc97 const TfLiteType model_input_type = model_info_->inputs[i]->type; in Init()
117 const TfLiteType model_output_type = model_info_->outputs[i]->type; in Init()
142 const TfLiteType model_input_type = model_info_->inputs[i]->type; in Run()
168 const TfLiteType model_output_type = model_info_->outputs[i]->type; in Run()
/external/tensorflow/tensorflow/lite/kernels/
Delementwise.cc53 bool IsNumericSupportedType(const TfLiteType type) { in IsNumericSupportedType()
57 bool IsLogicalSupportedType(const TfLiteType type) { in IsLogicalSupportedType()
61 bool IsAbsSupportedType(const TfLiteType type) { in IsAbsSupportedType()
65 bool IsRsqrtSupportedType(const TfLiteType type) { in IsRsqrtSupportedType()
82 typedef bool (*IsSupportedType)(TfLiteType);
139 TfLiteType expected_type) { in EvalImpl()
160 TfLiteType expected_type) { in EvalImpl()
186 TfLiteType type) { in AbsEvalQuantized()
207 const TfLiteType type = GetInput(context, node, 0)->type; in AbsEval()
239 TfLiteType type) { in RsqrtEvalQuantized()
[all …]
Dmatrix_diag_test.cc48 TfLiteType GetOutputType() { in GetOutputType()
97 EXPECT_THAT(model.GetOutputType(), TfLiteType::kTfLiteInt32); in TEST()
106 EXPECT_THAT(model.GetOutputType(), TfLiteType::kTfLiteUInt8); in TEST()
Dmatrix_set_diag_test.cc52 TfLiteType GetOutputType() { in GetOutputType()
103 EXPECT_THAT(model.GetOutputType(), TfLiteType::kTfLiteInt32); in TEST()
128 EXPECT_THAT(model.GetOutputType(), TfLiteType::kTfLiteInt32); in TEST()
Dtest_util.h945 static constexpr TfLiteType tflite_type = TfLiteType::kTfLiteFloat32;
955 static constexpr TfLiteType tflite_type = TfLiteType::kTfLiteInt32;
965 static constexpr TfLiteType tflite_type = TfLiteType::kTfLiteUInt32;
975 static constexpr TfLiteType tflite_type = TfLiteType::kTfLiteInt16;
985 static constexpr TfLiteType tflite_type = TfLiteType::kTfLiteInt8;
995 static constexpr TfLiteType tflite_type = TfLiteType::kTfLiteUInt8;
/external/tensorflow/tensorflow/lite/
Dportable_type_to_tflitetype.h39 constexpr TfLiteType typeToTfLiteType() { in typeToTfLiteType()
45 template <TfLiteType TFLITE_TYPE_ENUM>
51 constexpr TfLiteType typeToTfLiteType<CPP_TYPE>() { \
Dinterpreter.h151 int tensor_index, TfLiteType type, const char* name,
157 int tensor_index, TfLiteType type, const char* name,
167 int tensor_index, TfLiteType type, const char* name, const size_t rank,
175 TfLiteStatus SetTensorParametersReadWrite(int tensor_index, TfLiteType type,
183 int tensor_index, TfLiteType type, const char* name,
198 int tensor_index, TfLiteType type, const char* name, const size_t rank,
/external/tensorflow/tensorflow/lite/delegates/flex/
Dutil.h39 TF_DataType GetTensorFlowDataType(TfLiteType type);
42 TfLiteType GetTensorFlowLiteType(TF_DataType);
Dtest_util.h83 TfLiteType GetType(int tensor_index);
94 const std::vector<int>& outputs, TfLiteType type,
100 TfLiteType type, const char* buffer, size_t bytes);
Dutil.cc57 TF_DataType GetTensorFlowDataType(TfLiteType type) { in GetTensorFlowDataType()
96 TfLiteType GetTensorFlowLiteType(TF_DataType type) { in GetTensorFlowLiteType()
/external/tensorflow/tensorflow/lite/micro/kernels/
Delementwise.cc30 bool IsNumericSupportedType(const TfLiteType type) { in IsNumericSupportedType()
34 bool IsLogicalSupportedType(const TfLiteType type) { in IsLogicalSupportedType()
38 typedef bool (*IsSupportedType)(TfLiteType);
58 T func(T), TfLiteType expected_type) { in EvalImpl()
/external/tensorflow/tensorflow/lite/kernels/hashtable/
Dhashtable.cc40 TfLiteType key_dtype;
41 TfLiteType value_dtype;
51 TfLiteType key_dtype, value_dtype; in InitHashtable()
Dhashtable_ops_test.cc90 TensorType ConvertTfLiteType(TfLiteType type) { in ConvertTfLiteType()
114 HashtableGraph(TfLiteType key_type, TfLiteType value_type) in HashtableGraph()
481 TfLiteType key_type_;
482 TfLiteType value_type_;
512 HashtableDefaultGraphTest(TfLiteType key_type, TfLiteType value_type, in HashtableDefaultGraphTest()
557 TfLiteType value_type_;
684 TfLiteTensor CreateTensor(TfLiteType type, const std::vector<T>& vec) { in CreateTensor()
706 TfLiteTensor CreateTensor(TfLiteType type, in CreateTensor()
729 TfLiteType key_type, TfLiteType value_type, in InitHashtableResource()
/external/tensorflow/tensorflow/lite/python/optimize/
Dcalibration_wrapper.cc64 inline TensorType TfLiteTypeToSchemaType(TfLiteType type) { in TfLiteTypeToSchemaType()
335 TfLiteType input_type = python_utils::TfLiteTypeFromPyType(input_py_type); in QuantizeModel()
336 TfLiteType output_type = python_utils::TfLiteTypeFromPyType(output_py_type); in QuantizeModel()
337 TfLiteType activations_type = in QuantizeModel()
371 TfLiteType input_type = python_utils::TfLiteTypeFromPyType(input_py_type); in QuantizeModel()
372 TfLiteType output_type = python_utils::TfLiteTypeFromPyType(output_py_type); in QuantizeModel()

12345