Home
last modified time | relevance | path

Searched refs:input_model_ (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/lite/toco/tflite/
Dexport_test.cc33 void ResetOperators() { input_model_.operators.clear(); } in ResetOperators()
36 input_model_.GetOrCreateArray(name); in AddTensorsByName()
46 Array& input_array = input_model_.GetOrCreateArray(op->inputs[0]); in AddOperatorsByName()
47 Array& filter_array = input_model_.GetOrCreateArray(op->inputs[1]); in AddOperatorsByName()
48 Array& output_array = input_model_.GetOrCreateArray(op->outputs[0]); in AddOperatorsByName()
52 input_model_.operators.emplace_back(op); in AddOperatorsByName()
57 Array& input1_array = input_model_.GetOrCreateArray(op->inputs[0]); in AddOperatorsByName()
58 Array& input2_array = input_model_.GetOrCreateArray(op->inputs[1]); in AddOperatorsByName()
59 Array& output_array = input_model_.GetOrCreateArray(op->outputs[0]); in AddOperatorsByName()
63 input_model_.operators.emplace_back(op); in AddOperatorsByName()
[all …]
Dimport_test.cc135 input_model_ = ::tflite::GetModel(builder_.GetBufferPointer()); in BuildTestModel()
142 const ::tflite::Model* input_model_ = nullptr; member in toco::tflite::__anon5b60e9db0111::ImportTest
149 details::LoadTensorsTable(*input_model_, &tensors); in TEST_F()
157 details::LoadOperatorsTable(*input_model_, &operators); in TEST_F()
253 input_model_ = ::tflite::GetModel(builder_.GetBufferPointer()); in TEST_F()
/external/tensorflow/tensorflow/lite/tools/optimize/
Dquantize_model_test.cc53 input_model_ = ReadTestModel(); in QuantizeModelTest()
54 readonly_model_ = input_model_->GetModel(); in QuantizeModelTest()
58 std::unique_ptr<FlatBufferModel> input_model_; member in tflite::optimize::__anon2d3b55260211::QuantizeModelTest
Dquantize_weights_test.cc61 input_model_ = ReadTestModel(); in LoadBasicModel()
62 model_ = input_model_->GetModel(); in LoadBasicModel()
66 input_model_ = ReadSharedWeightsTestModel(); in LoadSharedWeightsModel()
67 model_ = input_model_->GetModel(); in LoadSharedWeightsModel()
70 std::unique_ptr<FlatBufferModel> input_model_; member in tflite::optimize::__anond8b8c2d00211::QuantizeWeightsTest