Home
last modified time | relevance | path

Searched refs:kFloat (Results 1 – 25 of 92) sorted by relevance

1234

/external/tensorflow/tensorflow/lite/toco/graph_transformations/
Dlstm_utils.cc25 void CopyArrayData(const Buffer<ArrayDataType::kFloat>& src_buffer, in CopyArrayData()
27 Buffer<ArrayDataType::kFloat>* dst_buffer, int dst_stride, in CopyArrayData()
41 Buffer<ArrayDataType::kFloat>* CreateFloatArrayBuffer(Model* model, in CreateFloatArrayBuffer()
46 array.data_type = ArrayDataType::kFloat; in CreateFloatArrayBuffer()
48 Buffer<ArrayDataType::kFloat>* buffer = in CreateFloatArrayBuffer()
49 &(array.GetMutableBuffer<ArrayDataType::kFloat>()); in CreateFloatArrayBuffer()
61 Buffer<ArrayDataType::kFloat>* buffer = in CopySubArrayToArray()
63 auto& orig_buffer = original_array.GetBuffer<ArrayDataType::kFloat>(); in CopySubArrayToArray()
71 void CopyArrayToSubArray(Buffer<ArrayDataType::kFloat>& tensor_buffer, in CopyArrayToSubArray()
78 auto& sub_buffer = sub_array.GetBuffer<ArrayDataType::kFloat>(); in CopyArrayToSubArray()
Dresolve_batch_normalization.cc51 CHECK(mean_array.data_type == ArrayDataType::kFloat); in Run()
52 CHECK(multiplier_array.data_type == ArrayDataType::kFloat); in Run()
53 CHECK(offset_array.data_type == ArrayDataType::kFloat); in Run()
98 auto& data = mean_array.GetMutableBuffer<ArrayDataType::kFloat>().data; in Run()
112 mul_param_array.data_type = ArrayDataType::kFloat; in Run()
113 add_param_array.data_type = ArrayDataType::kFloat; in Run()
115 mul_param_array.GetMutableBuffer<ArrayDataType::kFloat>().data; in Run()
117 add_param_array.GetMutableBuffer<ArrayDataType::kFloat>().data; in Run()
121 mean_array.GetBuffer<ArrayDataType::kFloat>().data; in Run()
123 multiplier_array.GetBuffer<ArrayDataType::kFloat>().data; in Run()
[all …]
Ddequantize.cc34 array->data_type = ArrayDataType::kFloat; in DequantizeBuffer()
35 auto& new_data = array->GetMutableBuffer<ArrayDataType::kFloat>().data; in DequantizeBuffer()
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()
157 new_array.data_type = ArrayDataType::kFloat; in DequantizeArray()
197 if (input_array.data_type == ArrayDataType::kFloat) { in Run()
200 if (input_array.final_data_type != ArrayDataType::kFloat) { in Run()
203 input_array.data_type = ArrayDataType::kFloat; in Run()
206 output_array.data_type = ArrayDataType::kFloat; in Run()
Dlstm_utils.h69 Buffer<ArrayDataType::kFloat>* CreateFloatArrayBuffer(Model* model,
86 void CopyArrayData(const Buffer<ArrayDataType::kFloat>& src_buffer,
88 Buffer<ArrayDataType::kFloat>* dst_buffer, int dst_stride,
101 void CopyArrayToSubArray(Buffer<ArrayDataType::kFloat>& tensor_buffer,
Dresolve_constant_fake_quant.cc86 CHECK(input_array.data_type == ArrayDataType::kFloat); in Run()
99 CHECK(input_array.data_type == ArrayDataType::kFloat); in Run()
100 output_array.data_type = ArrayDataType::kFloat; in Run()
110 const auto& input_buffer = input_array.GetBuffer<ArrayDataType::kFloat>(); in Run()
112 auto& output_buffer = output_array.GetMutableBuffer<ArrayDataType::kFloat>(); in Run()
Dquantization_util.cc68 case ArrayDataType::kFloat: in GetQuantizedDataType()
120 case ArrayDataType::kFloat: in ChooseQuantizationParamsForArrayAndQuantizedDataType()
136 CHECK(buffer.type == ArrayDataType::kFloat); in QuantizeBuffer()
138 static_cast<const Buffer<ArrayDataType::kFloat>&>(buffer); in QuantizeBuffer()
170 CHECK(array.data_type == ArrayDataType::kFloat); in QuantizeArray()
216 quantized_data_type == ArrayDataType::kFloat) { in IsArrayQuantizedRangeSubset()
Dresolve_multiply_by_zero.cc106 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()
Dfuse_binary_into_preceding_affine.cc46 auto& bias_buffer = bias.GetMutableBuffer<ArrayDataType::kFloat>(); in FuseAddOrSubParamsIntoPrecedingAffine()
48 const auto& operand_buffer = operand.GetBuffer<ArrayDataType::kFloat>(); in FuseAddOrSubParamsIntoPrecedingAffine()
115 auto& weights_buffer = weights.GetMutableBuffer<ArrayDataType::kFloat>(); in FuseMulOrDivParamsIntoPrecedingAffine()
117 auto& bias_buffer = bias.GetMutableBuffer<ArrayDataType::kFloat>(); in FuseMulOrDivParamsIntoPrecedingAffine()
119 const auto& operand_buffer = operand.GetBuffer<ArrayDataType::kFloat>(); in FuseMulOrDivParamsIntoPrecedingAffine()
285 if (weights.data_type != ArrayDataType::kFloat || in Run()
286 bias.data_type != ArrayDataType::kFloat) { in Run()
Dresolve_constant_unary.cc185 std::vector<DataType<ArrayDataType::kFloat>> const* input_float_data; in Run()
188 if (cast_op->dst_data_type != ArrayDataType::kFloat) { in Run()
202 if (input_array.buffer->type != ArrayDataType::kFloat) { in Run()
205 input_float_data = &(input_array.GetBuffer<ArrayDataType::kFloat>().data); in Run()
213 output_array.GetMutableBuffer<ArrayDataType::kFloat>().data; in Run()
221 if (input_array.buffer->type == ArrayDataType::kFloat) { in Run()
223 input_array.GetBuffer<ArrayDataType::kFloat>().data[i]); in Run()
Dpropagate_array_data_types.cc60 SetDataTypeForAllOutputs(model, op, ArrayDataType::kFloat); in Run()
237 ArrayDataType::kFloat; in Run()
251 if (data_type != ArrayDataType::kFloat) return ::tensorflow::Status::OK(); in Run()
257 if (data_type != ArrayDataType::kFloat) return ::tensorflow::Status::OK(); in Run()
271 if (data_type != ArrayDataType::kFloat) return ::tensorflow::Status::OK(); in Run()
277 if (data_type != ArrayDataType::kFloat) return ::tensorflow::Status::OK(); in Run()
Dfuse_binary_into_following_affine.cc55 operand.GetBuffer<ArrayDataType::kFloat>().data[0]; in FuseAddOrSubParamsIntoFollowingAffine()
72 const auto& weights_buffer = weights.GetBuffer<ArrayDataType::kFloat>(); in FuseAddOrSubParamsIntoFollowingAffine()
74 auto& bias_buffer = bias.GetMutableBuffer<ArrayDataType::kFloat>(); in FuseAddOrSubParamsIntoFollowingAffine()
135 operand.GetBuffer<ArrayDataType::kFloat>().data[0]; in FuseMulOrDivParamsIntoFollowingAffine()
138 weights.GetMutableBuffer<ArrayDataType::kFloat>().data.data(); in FuseMulOrDivParamsIntoFollowingAffine()
Didentify_lstm_merge_inputs.cc79 array.data_type = ArrayDataType::kFloat; in Run()
84 auto& buffer = array.GetMutableBuffer<ArrayDataType::kFloat>(); in Run()
122 bias_array.data_type = ArrayDataType::kFloat; in Run()
124 auto& bias_buffer = bias_array.GetMutableBuffer<ArrayDataType::kFloat>(); in Run()
Dresolve_fake_quant_args_from_vars.cc60 minmax.min = min_array.GetBuffer<ArrayDataType::kFloat>().data[0]; in Run()
61 minmax.max = max_array.GetBuffer<ArrayDataType::kFloat>().data[0]; in Run()
Densure_bias_vectors.cc57 bias_array.data_type = ArrayDataType::kFloat; in ProcessLinearOperator()
59 auto& bias_buffer = bias_array.GetMutableBuffer<ArrayDataType::kFloat>(); in ProcessLinearOperator()
Dquantize.cc104 CHECK(array.buffer->type == ArrayDataType::kFloat); in GetOrComputeMinMax()
105 const auto& data = array.GetBuffer<ArrayDataType::kFloat>().data; in GetOrComputeMinMax()
189 if (array.data_type != ArrayDataType::kFloat) { in ChooseQuantizationForOperatorInput()
345 if (array.data_type != ArrayDataType::kFloat) { in ChooseQuantizationForOperatorOutput()
486 input_array.data_type == ArrayDataType::kFloat) { in Run()
502 if (array.data_type == ArrayDataType::kFloat) { in Run()
648 dequantized_output_array.data_type = ArrayDataType::kFloat; in Run()
Dmake_initial_dequantize_operator.cc56 if (input_array.data_type != ArrayDataType::kFloat) { in AddDequantizeOperatorToInput()
82 dequantized_input_array.data_type = ArrayDataType::kFloat; in AddDequantizeOperatorToInput()
Dresolve_constant_concatenation.cc112 CHECK(concatenated_array->data_type == ArrayDataType::kFloat); in SetMinMaxForConcatenedArray()
180 case ArrayDataType::kFloat: in Run()
181 ConcatenateTensorBuffers<ArrayDataType::kFloat>( in Run()
Dresolve_constant_random_uniform.cc96 case ArrayDataType::kFloat: in Run()
97 if (!ComputeRandomUniformArray<ArrayDataType::kFloat>(model, op)) { in Run()
/external/tensorflow/tensorflow/lite/toco/tflite/
Dexport_test.cc49 input_array.data_type = ArrayDataType::kFloat; in AddOperatorsByName()
50 filter_array.data_type = ArrayDataType::kFloat; in AddOperatorsByName()
51 output_array.data_type = ArrayDataType::kFloat; in AddOperatorsByName()
60 input1_array.data_type = ArrayDataType::kFloat; in AddOperatorsByName()
61 input2_array.data_type = ArrayDataType::kFloat; in AddOperatorsByName()
62 output_array.data_type = ArrayDataType::kFloat; in AddOperatorsByName()
71 input1_array.data_type = ArrayDataType::kFloat; in AddOperatorsByName()
72 input2_array.data_type = ArrayDataType::kFloat; in AddOperatorsByName()
73 output_array.data_type = ArrayDataType::kFloat; in AddOperatorsByName()
108 weight_array.data_type = ArrayDataType::kFloat; in BuildQuantizableTestModel()
[all …]
Dtypes.cc89 case ArrayDataType::kFloat: in Serialize()
115 return ArrayDataType::kFloat; in Deserialize()
140 case ArrayDataType::kFloat: in Serialize()
141 return CopyBuffer<ArrayDataType::kFloat>(array, builder); in Serialize()
168 return CopyBuffer<ArrayDataType::kFloat>(buffer, array); in Deserialize()
Dtypes_test.cc75 {ArrayDataType::kFloat, ::tflite::TensorType_FLOAT32}, in TEST()
118 array.GetMutableBuffer<ArrayDataType::kFloat>(); // This is OK. in TEST()
140 Array recovered = ToFlatBufferAndBack<ArrayDataType::kFloat>({1.0f, 2.0f}); in TEST()
141 EXPECT_THAT(recovered.GetBuffer<ArrayDataType::kFloat>().data, in TEST()
/external/tensorflow/tensorflow/lite/toco/graph_transformations/tests/
Dresolve_constant_concatenation_test.cc125 in_array.data_type = ArrayDataType::kFloat; in PrepareModel()
134 in_array.GetMutableBuffer<toco::ArrayDataType::kFloat>(); in PrepareModel()
137 in_array.GetMutableBuffer<toco::ArrayDataType::kFloat>().data.data(); in PrepareModel()
146 out_array.data_type = ArrayDataType::kFloat; in PrepareModel()
176 EXPECT_THAT(concatenated_array->GetBuffer<toco::ArrayDataType::kFloat>().data, in TEST_F()
198 EXPECT_THAT(concatenated_array->GetBuffer<toco::ArrayDataType::kFloat>().data, in TEST_F()
220 EXPECT_THAT(concatenated_array->GetBuffer<toco::ArrayDataType::kFloat>().data, in TEST_F()
Dlstm_utils_test.cc61 Buffer<ArrayDataType::kFloat>* GetSrcBuffer() { return src_buffer_; } in GetSrcBuffer()
62 Buffer<ArrayDataType::kFloat>* GetDstBuffer() { return dst_buffer_; } in GetDstBuffer()
64 void PopulateBuffer(Buffer<ArrayDataType::kFloat>* buffer, in PopulateBuffer()
70 void UpdateBuffer(Buffer<ArrayDataType::kFloat>* buffer, in UpdateBuffer()
77 Buffer<ArrayDataType::kFloat>* src_buffer_;
78 Buffer<ArrayDataType::kFloat>* dst_buffer_;
Dresolve_constant_unary_test.cc39 input0.data_type = ArrayDataType::kFloat; in RunResolveSum()
40 input0.GetMutableBuffer<ArrayDataType::kFloat>().data = input; in RunResolveSum()
55 EXPECT_EQ(model.GetArray("output").GetBuffer<ArrayDataType::kFloat>().data, in RunResolveSum()
Dfuse_binary_into_following_affine_test.cc48 array.data_type = ArrayDataType::kFloat; in CreateArray()
57 auto& array_buffer = array.GetMutableBuffer<ArrayDataType::kFloat>(); in CreateConstantArray()
104 EXPECT_THAT(weights_array.GetBuffer<toco::ArrayDataType::kFloat>().data, in TEST_F()
108 EXPECT_THAT(bias_array.GetBuffer<toco::ArrayDataType::kFloat>().data, in TEST_F()

1234