/external/tensorflow/tensorflow/contrib/lite/toco/graph_transformations/ |
H A D | lstm_utils.cc | 25 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 D | lstm_utils.h | 62 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 D | resolve_batch_normalization.cc | 48 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 D | resolve_constant_fake_quant.cc | 49 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 D | dequantize.cc | 34 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 D | resolve_multiply_by_zero.cc | 94 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 D | resolve_constant_unary.cc | 83 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 D | ensure_bias_vectors.cc | 38 bias_array.data_type = ArrayDataType::kFloat;
|
H A D | convert_pure_conv_to_depthwise.cc | 41 if (weights_array.data_type != ArrayDataType::kFloat) { 80 weights_array.GetMutableBuffer<ArrayDataType::kFloat>();
|
H A D | make_initial_dequantize_operator.cc | 55 if (input_array.data_type != ArrayDataType::kFloat) { 83 dequantized_input_array.data_type = ArrayDataType::kFloat;
|
H A D | read_fake_quant_min_max.cc | 72 minmax.min = min_array.GetBuffer<ArrayDataType::kFloat>().data[0]; 73 minmax.max = max_array.GetBuffer<ArrayDataType::kFloat>().data[0];
|
H A D | remove_trivial_binary.cc | 86 if (constant_input_array.data_type != ArrayDataType::kFloat) { 90 constant_input_array.GetBuffer<ArrayDataType::kFloat>().data;
|
H A D | resolve_constant_stack.cc | 82 case ArrayDataType::kFloat: 83 Stack<ArrayDataType::kFloat>(model, *op);
|
H A D | resolve_reorder_axes.cc | 71 if (input_array.buffer->type == ArrayDataType::kFloat) { 72 ReorderAxes<float, ArrayDataType::kFloat>(reorder_op->input_axes_order,
|
H A D | identify_relu1.cc | 42 if (!op_array.buffer || op_array.buffer->type != ArrayDataType::kFloat || 46 const auto& op_data = op_array.GetBuffer<ArrayDataType::kFloat>().data;
|
H A D | resolve_constant_fill.cc | 78 case ArrayDataType::kFloat: 79 if (!ComputeFillArray<ArrayDataType::kFloat>(model, op)) {
|
H A D | identify_lstm_merge_inputs.cc | 75 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 D | fuse_binary_into_following_affine.cc | 55 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 D | fuse_binary_into_preceding_affine.cc | 46 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 D | quantize.cc | 60 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 D | identify_l2_normalization.cc | 89 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 D | resolve_constant_concatenation_test.cc | 127 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 D | lstm_utils_test.cc | 61 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 D | types.cc | 50 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 D | types_test.cc | 74 {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,
|