Home
last modified time | relevance | path

Searched refs:ArrayDataType (Results 1 – 25 of 97) sorted by relevance

1234

/aosp_15_r20/external/tensorflow/tensorflow/lite/toco/graph_transformations/
H A Dquantization_util.cc28 const FakeQuantOperator& op, ArrayDataType* out_quantized_data_type) { in InferQuantizedDataTypeFromFakeQuant()
30 *out_quantized_data_type = ArrayDataType::kUint8; in InferQuantizedDataTypeFromFakeQuant()
33 *out_quantized_data_type = ArrayDataType::kInt16; in InferQuantizedDataTypeFromFakeQuant()
36 *out_quantized_data_type = ArrayDataType::kNone; in InferQuantizedDataTypeFromFakeQuant()
41 bool GetQuantizedDataTypeNumericalRange(ArrayDataType data_type, in GetQuantizedDataTypeNumericalRange()
45 case ArrayDataType::kUint8: in GetQuantizedDataTypeNumericalRange()
49 case ArrayDataType::kInt16: in GetQuantizedDataTypeNumericalRange()
58 ArrayDataType GetQuantizedDataType(const Array& array, in GetQuantizedDataType()
59 ArrayDataType default_type) { in GetQuantizedDataType()
61 case ArrayDataType::kInt8: in GetQuantizedDataType()
[all …]
H A Dresolve_constant_reshape.cc46 if (output_array.data_type == ArrayDataType::kNone) { in Run()
65 case ArrayDataType::kBool: in Run()
66 CopyArrayBuffer<ArrayDataType::kBool>(input_array, &output_array); in Run()
68 case ArrayDataType::kFloat: in Run()
69 CopyArrayBuffer<ArrayDataType::kFloat>(input_array, &output_array); in Run()
71 case ArrayDataType::kInt8: in Run()
72 CopyArrayBuffer<ArrayDataType::kInt8>(input_array, &output_array); in Run()
74 case ArrayDataType::kUint8: in Run()
75 CopyArrayBuffer<ArrayDataType::kUint8>(input_array, &output_array); in Run()
77 case ArrayDataType::kInt16: in Run()
[all …]
H A Dpropagate_array_data_types.cc28 ArrayDataType data_type) { in SetDataTypeForAllOutputs()
45 model->GetArray(input).data_type == ArrayDataType::kNone) { in Run()
51 std::unordered_map<std::string, ArrayDataType> old_output_data_types; in Run()
59 SetDataTypeForAllOutputs(model, op, ArrayDataType::kFloat); in Run()
72 SetDataTypeForAllOutputs(model, op, ArrayDataType::kBool); in Run()
76 SetDataTypeForAllOutputs(model, op, ArrayDataType::kInt32); in Run()
90 const ArrayDataType data_type = model->GetArray(op->inputs[1]).data_type; in Run()
97 const ArrayDataType data_type = model->GetArray(op->inputs[0]).data_type; in Run()
104 const ArrayDataType data_type = model->GetArray(op->inputs[2]).data_type; in Run()
132 ArrayDataType data_type; in Run()
[all …]
H A Dresolve_constant_fake_quant.cc28 template <ArrayDataType A>
35 void GetBoundsForQuantizedDataType(ArrayDataType quantized_data_type, in GetBoundsForQuantizedDataType()
41 case ArrayDataType::kUint8: in GetBoundsForQuantizedDataType()
42 return GetBoundsForQuantizedDataType<ArrayDataType::kUint8>(min, max); in GetBoundsForQuantizedDataType()
43 case ArrayDataType::kInt8: in GetBoundsForQuantizedDataType()
44 return GetBoundsForQuantizedDataType<ArrayDataType::kInt8>(min, max); in GetBoundsForQuantizedDataType()
45 case ArrayDataType::kUint16: in GetBoundsForQuantizedDataType()
46 return GetBoundsForQuantizedDataType<ArrayDataType::kUint16>(min, max); in GetBoundsForQuantizedDataType()
47 case ArrayDataType::kInt16: in GetBoundsForQuantizedDataType()
48 return GetBoundsForQuantizedDataType<ArrayDataType::kInt16>(min, max); in GetBoundsForQuantizedDataType()
[all …]
H A Dresolve_multiply_by_zero.cc39 template <ArrayDataType Type>
70 if (output_array.data_type == ArrayDataType::kNone) { in Run()
106 case ArrayDataType::kFloat: { in Run()
108 constant_input_array.GetBuffer<ArrayDataType::kFloat>().data; in Run()
109 if (!AreAllBufferElementsZero<DataType<ArrayDataType::kFloat>>( in Run()
113 FillArrayWithZeros<ArrayDataType::kFloat>(&output_array); in Run()
115 case ArrayDataType::kUint8: { in Run()
117 constant_input_array.GetBuffer<ArrayDataType::kUint8>().data; in Run()
118 if (!AreAllBufferElementsZero<DataType<ArrayDataType::kUint8>>( in Run()
122 FillArrayWithZeros<ArrayDataType::kUint8>(&output_array); in Run()
[all …]
H A Dresolve_constant_tile.cc72 template <ArrayDataType Type>
80 case ArrayDataType::kInt32: in Tile()
83 multiples_array.GetBuffer<ArrayDataType::kInt32>().data.data(), in Tile()
86 case ArrayDataType::kInt64: in Tile()
89 multiples_array.GetBuffer<ArrayDataType::kInt64>().data.data(), in Tile()
115 if (output_array.data_type == ArrayDataType::kNone) { in Run()
131 CHECK(multiples_array.data_type == ArrayDataType::kInt32 || in Run()
132 multiples_array.data_type == ArrayDataType::kInt64) in Run()
139 case ArrayDataType::kFloat: in Run()
140 Tile<ArrayDataType::kFloat>(input_array, multiples_array, &output_array); in Run()
[all …]
H A Dquantize.cc43 return (array.data_type != ArrayDataType::kFloat && in SupportsQuantization()
44 array.data_type != ArrayDataType::kFloat16); in SupportsQuantization()
145 CHECK(array.buffer->type == ArrayDataType::kFloat); in GetOrComputeMinMax()
146 const auto& data = array.GetBuffer<ArrayDataType::kFloat>().data; in GetOrComputeMinMax()
198 template <ArrayDataType A>
211 QuantizationPoints GetQuantizationPoints(ArrayDataType data_type) { in GetQuantizationPoints()
213 case ArrayDataType::kUint8: in GetQuantizationPoints()
214 return GetQuantizationPoints<ArrayDataType::kUint8>(); in GetQuantizationPoints()
215 case ArrayDataType::kInt16: in GetQuantizationPoints()
216 return GetQuantizationPoints<ArrayDataType::kInt16>(); in GetQuantizationPoints()
[all …]
H A Dresolve_constant_gather.cc27 template <ArrayDataType Type>
35 coords_array.GetBuffer<ArrayDataType::kInt32>().data; in Gather()
81 if (output_array.data_type == ArrayDataType::kNone) { in Run()
108 CHECK(coords_array.data_type == ArrayDataType::kInt32) in Run()
123 case ArrayDataType::kFloat: in Run()
124 Gather<ArrayDataType::kFloat>(input_array, coords_array, &output_array); in Run()
126 case ArrayDataType::kUint8: in Run()
127 Gather<ArrayDataType::kUint8>(input_array, coords_array, &output_array); in Run()
129 case ArrayDataType::kInt32: in Run()
130 Gather<ArrayDataType::kInt32>(input_array, coords_array, &output_array); in Run()
[all …]
H A Dresolve_constant_concatenation.cc36 template <ArrayDataType A, typename T>
79 template <ArrayDataType A>
113 CHECK(concatenated_array->data_type == ArrayDataType::kFloat); in SetMinMaxForConcatenedArray()
182 case ArrayDataType::kFloat: in Run()
183 ConcatenateTensorBuffers<ArrayDataType::kFloat>( in Run()
187 case ArrayDataType::kUint8: in Run()
188 ConcatenateTensorBuffers<ArrayDataType::kUint8>( in Run()
191 case ArrayDataType::kInt32: in Run()
192 ConcatenateTensorBuffers<ArrayDataType::kInt32>( in Run()
195 case ArrayDataType::kInt64: in Run()
[all …]
H A Dresolve_constant_fill.cc24 template <ArrayDataType Type>
59 if (output_array.data_type == ArrayDataType::kNone) { in Run()
81 case ArrayDataType::kFloat: in Run()
82 if (!ComputeFillArray<ArrayDataType::kFloat>(model, op)) { in Run()
86 case ArrayDataType::kUint8: in Run()
87 if (!ComputeFillArray<ArrayDataType::kUint8>(model, op)) { in Run()
91 case ArrayDataType::kInt32: in Run()
92 if (!ComputeFillArray<ArrayDataType::kInt32>(model, op)) { in Run()
96 case ArrayDataType::kInt64: in Run()
97 if (!ComputeFillArray<ArrayDataType::kInt64>(model, op)) { in Run()
[all …]
H A Dresolve_constant_pack.cc26 template <ArrayDataType Type>
66 if (output_array.data_type == ArrayDataType::kNone) { in Run()
92 case ArrayDataType::kFloat: in Run()
93 Pack<ArrayDataType::kFloat>(model, *op); in Run()
95 case ArrayDataType::kUint8: in Run()
96 Pack<ArrayDataType::kUint8>(model, *op); in Run()
98 case ArrayDataType::kInt32: in Run()
99 Pack<ArrayDataType::kInt32>(model, *op); in Run()
101 case ArrayDataType::kInt64: in Run()
102 Pack<ArrayDataType::kInt64>(model, *op); in Run()
[all …]
H A Ddequantize.cc30 template <ArrayDataType A>
34 array->data_type = ArrayDataType::kFloat; in DequantizeBuffer()
35 auto& new_data = array->GetMutableBuffer<ArrayDataType::kFloat>().data; in DequantizeBuffer()
88 if (array->data_type == ArrayDataType::kUint8) { in DequantizeArray()
89 DequantizeBuffer<ArrayDataType::kUint8>(array); in DequantizeArray()
90 } else if (array->data_type == ArrayDataType::kInt32) { in DequantizeArray()
91 DequantizeBuffer<ArrayDataType::kInt32>(array); in DequantizeArray()
95 CHECK(array->data_type == ArrayDataType::kFloat); in DequantizeArray()
96 CHECK(array->buffer->type == ArrayDataType::kFloat); in DequantizeArray()
102 array->data_type = ArrayDataType::kFloat; in DequantizeArray()
[all …]
H A Dpropagate_fixed_sizes.cc183 CHECK(specified_output_shape_array.data_type == ArrayDataType::kInt32) in ProcessTransposeConvOperator()
238 specified_output_shape_array.GetBuffer<ArrayDataType::kInt32>().data; in ProcessTransposeConvOperator()
358 CHECK(dims_array.data_type == ArrayDataType::kInt32) << "dims must be int32"; in ProcessOpWithShapeInput()
363 dims_array.GetBuffer<ArrayDataType::kInt32>().data; in ProcessOpWithShapeInput()
420 CHECK(shape_array.data_type == ArrayDataType::kInt32) in ProcessTensorFlowReshapeOperator()
429 shape_array.GetBuffer<ArrayDataType::kInt32>().data; in ProcessTensorFlowReshapeOperator()
571 CHECK(reduction_indices_array.buffer->type == ArrayDataType::kInt32); in ProcessTensorFlowReductionOperator()
576 reduction_indices_array.GetBuffer<ArrayDataType::kInt32>().data; in ProcessTensorFlowReductionOperator()
723 const ArrayDataType& start_dtype = start_array.data_type; in ProcessRangeOperator()
724 CHECK(start_dtype == ArrayDataType::kInt32 || in ProcessRangeOperator()
[all …]
H A Dresolve_constant_unary.cc187 std::vector<DataType<ArrayDataType::kFloat>> const* input_float_data = in Run()
191 if (cast_op->dst_data_type != ArrayDataType::kFloat) { in Run()
205 if (input_array.buffer->type != ArrayDataType::kFloat) { in Run()
208 input_float_data = &(input_array.GetBuffer<ArrayDataType::kFloat>().data); in Run()
216 output_array.GetMutableBuffer<ArrayDataType::kFloat>().data; in Run()
224 if (input_array.buffer->type == ArrayDataType::kFloat) { in Run()
226 input_array.GetBuffer<ArrayDataType::kFloat>().data[i]); in Run()
227 } else if (input_array.buffer->type == ArrayDataType::kUint8) { in Run()
229 input_array.GetBuffer<ArrayDataType::kUint8>().data[i]); in Run()
230 } else if (input_array.buffer->type == ArrayDataType::kInt32) { in Run()
[all …]
H A Dresolve_constant_strided_slice.cc27 template <ArrayDataType Type>
116 if (output_array.data_type == ArrayDataType::kNone) { in Run()
144 case ArrayDataType::kFloat: in Run()
145 StridedSlice<ArrayDataType::kFloat>(*op, input_array, &output_array); in Run()
147 case ArrayDataType::kUint8: in Run()
148 StridedSlice<ArrayDataType::kUint8>(*op, input_array, &output_array); in Run()
150 case ArrayDataType::kInt32: in Run()
151 StridedSlice<ArrayDataType::kInt32>(*op, input_array, &output_array); in Run()
153 case ArrayDataType::kInt64: in Run()
154 StridedSlice<ArrayDataType::kInt64>(*op, input_array, &output_array); in Run()
[all …]
H A Dresolve_constant_slice.cc26 template <ArrayDataType Type>
103 if (output_array.data_type == ArrayDataType::kNone) { in Run()
130 case ArrayDataType::kFloat: in Run()
131 if (!Slice<ArrayDataType::kFloat>(*op, input_array, &output_array)) { in Run()
135 case ArrayDataType::kUint8: in Run()
136 if (!Slice<ArrayDataType::kUint8>(*op, input_array, &output_array)) { in Run()
140 case ArrayDataType::kInt32: in Run()
141 if (!Slice<ArrayDataType::kInt32>(*op, input_array, &output_array)) { in Run()
145 case ArrayDataType::kInt64: in Run()
146 if (!Slice<ArrayDataType::kInt64>(*op, input_array, &output_array)) { in Run()
[all …]
H A Dresolve_constant_transpose.cc28 template <ArrayDataType Type>
118 if (output_array.data_type == ArrayDataType::kNone) { in Run()
146 case ArrayDataType::kFloat: in Run()
147 Transpose<ArrayDataType::kFloat>(model, input_array, op->perm, in Run()
150 case ArrayDataType::kUint8: in Run()
151 Transpose<ArrayDataType::kUint8>(model, input_array, op->perm, in Run()
154 case ArrayDataType::kInt32: in Run()
155 Transpose<ArrayDataType::kInt32>(model, input_array, op->perm, in Run()
158 case ArrayDataType::kInt64: in Run()
159 Transpose<ArrayDataType::kInt64>(model, input_array, op->perm, in Run()
[all …]
H A Dresolve_constant_binary.cc64 template <ArrayDataType InputsDataType, ArrayDataType OutputDataType>
180 TOCO_HANDLE_CASE(ArrayDataType::kFloat, ArrayDataType::kFloat) in EvaluateBinaryOperatorOnConstantInputs()
181 TOCO_HANDLE_CASE(ArrayDataType::kFloat, ArrayDataType::kBool) in EvaluateBinaryOperatorOnConstantInputs()
182 TOCO_HANDLE_CASE(ArrayDataType::kInt32, ArrayDataType::kInt32) in EvaluateBinaryOperatorOnConstantInputs()
183 TOCO_HANDLE_CASE(ArrayDataType::kInt32, ArrayDataType::kBool) in EvaluateBinaryOperatorOnConstantInputs()
184 TOCO_HANDLE_CASE(ArrayDataType::kInt64, ArrayDataType::kInt64) in EvaluateBinaryOperatorOnConstantInputs()
185 TOCO_HANDLE_CASE(ArrayDataType::kInt64, ArrayDataType::kBool) in EvaluateBinaryOperatorOnConstantInputs()
H A Dlstm_utils.cc27 void CopyArrayData(const Buffer<ArrayDataType::kFloat>& src_buffer, in CopyArrayData()
29 Buffer<ArrayDataType::kFloat>* dst_buffer, int dst_stride, in CopyArrayData()
43 Buffer<ArrayDataType::kFloat>* CreateFloatArrayBuffer(Model* model, in CreateFloatArrayBuffer()
48 array.data_type = ArrayDataType::kFloat; in CreateFloatArrayBuffer()
50 Buffer<ArrayDataType::kFloat>* buffer = in CreateFloatArrayBuffer()
51 &(array.GetMutableBuffer<ArrayDataType::kFloat>()); in CreateFloatArrayBuffer()
63 Buffer<ArrayDataType::kFloat>* buffer = in CopySubArrayToArray()
65 auto& orig_buffer = original_array.GetBuffer<ArrayDataType::kFloat>(); in CopySubArrayToArray()
73 void CopyArrayToSubArray(Buffer<ArrayDataType::kFloat>& tensor_buffer, in CopyArrayToSubArray()
80 auto& sub_buffer = sub_array.GetBuffer<ArrayDataType::kFloat>(); in CopyArrayToSubArray()
/aosp_15_r20/external/tensorflow/tensorflow/lite/toco/tflite/
H A Dtypes.cc29 const auto& src_data = array.GetBuffer<ArrayDataType::kString>().data; in CopyStringToBuffer()
48 const auto& src_data = array.GetBuffer<ArrayDataType::kBool>().data; in CopyBoolToBuffer()
52 template <ArrayDataType T>
65 &array->GetMutableBuffer<ArrayDataType::kString>().data; in CopyStringFromBuffer()
74 template <ArrayDataType T>
90 ::tflite::TensorType DataType::Serialize(ArrayDataType array_data_type) { in Serialize()
92 case ArrayDataType::kFloat: in Serialize()
94 case ArrayDataType::kInt16: in Serialize()
96 case ArrayDataType::kInt32: in Serialize()
98 case ArrayDataType::kUint32: in Serialize()
[all …]
H A Dtypes_test.cc32 static const ArrayDataType kUnsupportedTocoTypes[] = {ArrayDataType::kNone};
43 template <ArrayDataType T>
70 std::vector<std::pair<ArrayDataType, ::tflite::TensorType>> testdata = { in TEST()
71 {ArrayDataType::kUint8, ::tflite::TensorType_UINT8}, in TEST()
72 {ArrayDataType::kInt32, ::tflite::TensorType_INT32}, in TEST()
73 {ArrayDataType::kUint32, ::tflite::TensorType_UINT32}, in TEST()
74 {ArrayDataType::kInt64, ::tflite::TensorType_INT64}, in TEST()
75 {ArrayDataType::kFloat, ::tflite::TensorType_FLOAT32}, in TEST()
76 {ArrayDataType::kBool, ::tflite::TensorType_BOOL}, in TEST()
77 {ArrayDataType::kComplex64, ::tflite::TensorType_COMPLEX64}}; in TEST()
[all …]
H A Doperator_test.cc193 op.src_data_type = ArrayDataType::kFloat; in TEST_F()
194 op.dst_data_type = ArrayDataType::kUint8; in TEST_F()
545 op.output_data_type = ArrayDataType::kInt64; in TEST_F()
568 int32_array.data_type = ArrayDataType::kInt32; in TEST_F()
575 int64_array.data_type = ArrayDataType::kInt64; in TEST_F()
582 int8_array.data_type = ArrayDataType::kInt8; in TEST_F()
588 uint8_array.data_type = ArrayDataType::kUint8; in TEST_F()
756 op.idx_out_type = ArrayDataType::kInt64; in TEST_F()
806 uint8_array.data_type = ArrayDataType::kUint8; in SimpleVersioningTest()
812 int8_array.data_type = ArrayDataType::kInt8; in SimpleVersioningTest()
[all …]
/aosp_15_r20/external/tensorflow/tensorflow/lite/toco/
H A Dtooling_util.cc68 std::string ArrayDataTypeName(ArrayDataType data_type) { in ArrayDataTypeName()
70 case ArrayDataType::kFloat: in ArrayDataTypeName()
72 case ArrayDataType::kInt8: in ArrayDataTypeName()
74 case ArrayDataType::kUint8: in ArrayDataTypeName()
76 case ArrayDataType::kInt16: in ArrayDataTypeName()
78 case ArrayDataType::kUint16: in ArrayDataTypeName()
80 case ArrayDataType::kInt32: in ArrayDataTypeName()
82 case ArrayDataType::kUint32: in ArrayDataTypeName()
84 case ArrayDataType::kInt64: in ArrayDataTypeName()
86 case ArrayDataType::kUint64: in ArrayDataTypeName()
[all …]
H A Dimport_tensorflow_test.cc306 template <ArrayDataType T>
329 constexpr ArrayDataType kType = ArrayDataType::kInt32; in TEST_F()
344 constexpr ArrayDataType kType = ArrayDataType::kInt64; in TEST_F()
359 constexpr ArrayDataType kType = ArrayDataType::kUint8; in TEST_F()
374 constexpr ArrayDataType kType = ArrayDataType::kBool; in TEST_F()
389 constexpr ArrayDataType kType = ArrayDataType::kFloat; in TEST_F()
408 constexpr ArrayDataType kType = ArrayDataType::kComplex64; in TEST_F()
441 std::vector<std::pair<tensorflow::DataType, ArrayDataType>> UnaryTestTypes() { in UnaryTestTypes()
442 return {{DT_FLOAT, ArrayDataType::kFloat}, in UnaryTestTypes()
443 {DT_INT32, ArrayDataType::kInt32}, in UnaryTestTypes()
[all …]
H A Dmodel.h225 enum class ArrayDataType : uint8 { enum
245 template <ArrayDataType A>
248 struct DataTypeImpl<ArrayDataType::kNone> {
252 struct DataTypeImpl<ArrayDataType::kBool> {
256 struct DataTypeImpl<ArrayDataType::kFloat> {
260 struct DataTypeImpl<ArrayDataType::kInt8> {
264 struct DataTypeImpl<ArrayDataType::kUint8> {
268 struct DataTypeImpl<ArrayDataType::kInt16> {
272 struct DataTypeImpl<ArrayDataType::kUint16> {
276 struct DataTypeImpl<ArrayDataType::kInt32> {
[all …]

1234