MCPcopy Create free account
hub / github.com/DeepRec-AI/DeepRec / BuildTensors

Method BuildTensors

tensorflow/lite/toco/tflite/import_test.cc:50–67  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

48 }
49
50 Offset<Vector<Offset<::tflite::Tensor>>> BuildTensors() {
51 auto q = ::tflite::CreateQuantizationParameters(
52 builder_,
53 /*min=*/builder_.CreateVector<float>({0.1f}),
54 /*max=*/builder_.CreateVector<float>({0.2f}),
55 /*scale=*/builder_.CreateVector<float>({0.3f}),
56 /*zero_point=*/builder_.CreateVector<int64_t>({100ll}));
57 auto t1 =
58 ::tflite::CreateTensor(builder_, builder_.CreateVector<int>({1, 2, 2}),
59 ::tflite::TensorType_FLOAT32, 1,
60 builder_.CreateString("tensor_one"), q);
61 auto t2 =
62 ::tflite::CreateTensor(builder_, builder_.CreateVector<int>({2, 1}),
63 ::tflite::TensorType_FLOAT32, 0,
64 builder_.CreateString("tensor_two"), q);
65 return builder_.CreateVector(
66 std::vector<Offset<::tflite::Tensor>>({t1, t2}));
67 }
68
69 Offset<Vector<Offset<::tflite::OperatorCode>>> BuildOpCodes(
70 std::initializer_list<::tflite::BuiltinOperator> op_codes) {

Callers

nothing calls this directly

Calls 3

CreateVectorMethod · 0.80
CreateTensorFunction · 0.50

Tested by

no test coverage detected