Searched refs:kFloat (Results 1 - 25 of 58) sorted by relevance

123

/external/tensorflow/tensorflow/contrib/lite/toco/graph_transformations/
H A Dlstm_utils.cc25 void CopyArrayData(const Buffer<ArrayDataType::kFloat>& src_buffer,
27 Buffer<ArrayDataType::kFloat>* dst_buffer, int dst_stride,
41 Buffer<ArrayDataType::kFloat>* CreateFloatArrayBuffer(Model* model,
46 array.data_type = ArrayDataType::kFloat;
48 Buffer<ArrayDataType::kFloat>* buffer =
49 &(array.GetMutableBuffer<ArrayDataType::kFloat>());
61 Buffer<ArrayDataType::kFloat>* buffer =
63 auto& orig_buffer = original_array.GetBuffer<ArrayDataType::kFloat>();
71 void CopyArrayToSubArray(Buffer<ArrayDataType::kFloat>& tensor_buffer,
78 auto& sub_buffer = sub_array.GetBuffer<ArrayDataType::kFloat>();
[all...]
H A Dlstm_utils.h62 Buffer<ArrayDataType::kFloat>* CreateFloatArrayBuffer(Model* model,
79 void CopyArrayData(const Buffer<ArrayDataType::kFloat>& src_buffer,
81 Buffer<ArrayDataType::kFloat>* dst_buffer, int dst_stride,
94 void CopyArrayToSubArray(Buffer<ArrayDataType::kFloat>& tensor_buffer,
H A Dresolve_batch_normalization.cc48 CHECK(mean_array.data_type == ArrayDataType::kFloat);
49 CHECK(multiplier_array.data_type == ArrayDataType::kFloat);
50 CHECK(offset_array.data_type == ArrayDataType::kFloat);
96 mul_param_array.data_type = ArrayDataType::kFloat;
97 add_param_array.data_type = ArrayDataType::kFloat;
99 mul_param_array.GetMutableBuffer<ArrayDataType::kFloat>().data;
101 add_param_array.GetMutableBuffer<ArrayDataType::kFloat>().data;
105 mean_array.GetBuffer<ArrayDataType::kFloat>().data;
107 multiplier_array.GetBuffer<ArrayDataType::kFloat>().data;
109 offset_array.GetBuffer<ArrayDataType::kFloat>()
[all...]
H A Dresolve_constant_fake_quant.cc49 CHECK(input_array.data_type == ArrayDataType::kFloat);
50 output_array.data_type = ArrayDataType::kFloat;
52 const auto& input_buffer = input_array.GetBuffer<ArrayDataType::kFloat>();
54 auto& output_buffer = output_array.GetMutableBuffer<ArrayDataType::kFloat>();
H A Ddequantize.cc34 array->data_type = ArrayDataType::kFloat;
35 auto& new_data = array->GetMutableBuffer<ArrayDataType::kFloat>().data;
95 CHECK(array->data_type == ArrayDataType::kFloat);
96 CHECK(array->buffer->type == ArrayDataType::kFloat);
102 array->data_type = ArrayDataType::kFloat;
157 new_array.data_type = ArrayDataType::kFloat;
194 if (input_array.data_type == ArrayDataType::kFloat) {
197 if (input_array.final_data_type != ArrayDataType::kFloat) {
200 input_array.data_type = ArrayDataType::kFloat;
203 output_array.data_type = ArrayDataType::kFloat;
[all...]
H A Dresolve_multiply_by_zero.cc94 case ArrayDataType::kFloat: {
96 constant_input_array.GetBuffer<ArrayDataType::kFloat>().data;
97 if (!AreAllBufferElementsZero<DataType<ArrayDataType::kFloat>>(
101 FillArrayWithZeros<ArrayDataType::kFloat>(&output_array);
H A Dresolve_constant_unary.cc83 std::vector<DataType<ArrayDataType::kFloat>> const* input_float_data;
86 if (cast_op->dst_data_type != ArrayDataType::kFloat) {
100 if (input_array.buffer->type != ArrayDataType::kFloat) {
103 input_float_data = &(input_array.GetBuffer<ArrayDataType::kFloat>().data);
111 output_array.GetMutableBuffer<ArrayDataType::kFloat>().data;
119 if (input_array.buffer->type == ArrayDataType::kFloat) {
121 input_array.GetBuffer<ArrayDataType::kFloat>().data[i]);
H A Densure_bias_vectors.cc38 bias_array.data_type = ArrayDataType::kFloat;
H A Dconvert_pure_conv_to_depthwise.cc41 if (weights_array.data_type != ArrayDataType::kFloat) {
80 weights_array.GetMutableBuffer<ArrayDataType::kFloat>();
H A Dmake_initial_dequantize_operator.cc55 if (input_array.data_type != ArrayDataType::kFloat) {
83 dequantized_input_array.data_type = ArrayDataType::kFloat;
H A Dread_fake_quant_min_max.cc72 minmax.min = min_array.GetBuffer<ArrayDataType::kFloat>().data[0];
73 minmax.max = max_array.GetBuffer<ArrayDataType::kFloat>().data[0];
H A Dremove_trivial_binary.cc86 if (constant_input_array.data_type != ArrayDataType::kFloat) {
90 constant_input_array.GetBuffer<ArrayDataType::kFloat>().data;
H A Dresolve_constant_stack.cc82 case ArrayDataType::kFloat:
83 Stack<ArrayDataType::kFloat>(model, *op);
H A Dresolve_reorder_axes.cc71 if (input_array.buffer->type == ArrayDataType::kFloat) {
72 ReorderAxes<float, ArrayDataType::kFloat>(reorder_op->input_axes_order,
H A Didentify_relu1.cc42 if (!op_array.buffer || op_array.buffer->type != ArrayDataType::kFloat ||
46 const auto& op_data = op_array.GetBuffer<ArrayDataType::kFloat>().data;
H A Dresolve_constant_fill.cc78 case ArrayDataType::kFloat:
79 if (!ComputeFillArray<ArrayDataType::kFloat>(model, op)) {
H A Didentify_lstm_merge_inputs.cc75 array.data_type = ArrayDataType::kFloat;
80 auto& buffer = array.GetMutableBuffer<ArrayDataType::kFloat>();
118 bias_array.data_type = ArrayDataType::kFloat;
120 auto& bias_buffer = bias_array.GetMutableBuffer<ArrayDataType::kFloat>();
H A Dfuse_binary_into_following_affine.cc55 operand.GetBuffer<ArrayDataType::kFloat>().data[0];
72 const auto& weights_buffer = weights.GetBuffer<ArrayDataType::kFloat>();
74 auto& bias_buffer = bias.GetMutableBuffer<ArrayDataType::kFloat>();
135 operand.GetBuffer<ArrayDataType::kFloat>().data[0];
138 weights.GetMutableBuffer<ArrayDataType::kFloat>().data.data();
H A Dfuse_binary_into_preceding_affine.cc46 auto& bias_buffer = bias.GetMutableBuffer<ArrayDataType::kFloat>();
48 const auto& operand_buffer = operand.GetBuffer<ArrayDataType::kFloat>();
103 auto& weights_buffer = weights.GetMutableBuffer<ArrayDataType::kFloat>();
105 auto& bias_buffer = bias.GetMutableBuffer<ArrayDataType::kFloat>();
107 const auto& operand_buffer = operand.GetBuffer<ArrayDataType::kFloat>();
H A Dquantize.cc60 CHECK(buffer.type == ArrayDataType::kFloat);
62 static_cast<const Buffer<ArrayDataType::kFloat>&>(buffer);
90 CHECK(array.data_type == ArrayDataType::kFloat);
148 CHECK(array.buffer->type == ArrayDataType::kFloat);
149 const auto& data = array.GetBuffer<ArrayDataType::kFloat>().data;
175 if (array.data_type != ArrayDataType::kFloat) {
314 if (array.data_type != ArrayDataType::kFloat) {
417 if (array.data_type == ArrayDataType::kFloat) {
515 dequantized_output_array.data_type = ArrayDataType::kFloat;
H A Didentify_l2_normalization.cc89 if (input_array.buffer->type != ArrayDataType::kFloat) {
96 input_array.GetBuffer<ArrayDataType::kFloat>().data;
/external/tensorflow/tensorflow/contrib/lite/toco/graph_transformations/tests/
H A Dresolve_constant_concatenation_test.cc127 in_array.data_type = ArrayDataType::kFloat;
136 in_array.GetMutableBuffer<toco::ArrayDataType::kFloat>();
139 in_array.GetMutableBuffer<toco::ArrayDataType::kFloat>().data.data();
148 out_array.data_type = ArrayDataType::kFloat;
175 EXPECT_THAT(concatenated_array->GetBuffer<toco::ArrayDataType::kFloat>().data,
194 EXPECT_THAT(concatenated_array->GetBuffer<toco::ArrayDataType::kFloat>().data,
213 EXPECT_THAT(concatenated_array->GetBuffer<toco::ArrayDataType::kFloat>().data,
H A Dlstm_utils_test.cc61 Buffer<ArrayDataType::kFloat>* GetSrcBuffer() { return src_buffer_; }
62 Buffer<ArrayDataType::kFloat>* GetDstBuffer() { return dst_buffer_; }
64 void PopulateBuffer(Buffer<ArrayDataType::kFloat>* buffer,
70 void UpdateBuffer(Buffer<ArrayDataType::kFloat>* buffer,
77 Buffer<ArrayDataType::kFloat>* src_buffer_;
78 Buffer<ArrayDataType::kFloat>* dst_buffer_;
/external/tensorflow/tensorflow/contrib/lite/toco/tflite/
H A Dtypes.cc50 case ArrayDataType::kFloat:
70 return ArrayDataType::kFloat;
89 case ArrayDataType::kFloat:
90 return CopyBuffer<ArrayDataType::kFloat>(array, builder);
109 return CopyBuffer<ArrayDataType::kFloat>(buffer, array);
H A Dtypes_test.cc74 {ArrayDataType::kFloat, ::tflite::TensorType_FLOAT32}};
115 array.GetMutableBuffer<ArrayDataType::kFloat>(); // This is OK.
137 Array recovered = ToFlatBufferAndBack<ArrayDataType::kFloat>({1.0f, 2.0f});
138 EXPECT_THAT(recovered.GetBuffer<ArrayDataType::kFloat>().data,

Completed in 252 milliseconds

123