/external/tensorflow/tensorflow/lite/tools/optimize/ |
D | quantization_wrapper_utils_custom_test.cc | 46 model->subgraphs.push_back(std::move(subgraph)); in TEST() 56 model->subgraphs[0]->tensors.push_back(std::move(tensor)); in TEST() 58 model->subgraphs[0]->operators.push_back(std::move(lstm_op)); in TEST() 68 EXPECT_EQ(model->subgraphs.size(), 1); in TEST() 69 EXPECT_EQ(model->subgraphs[0]->operators.size(), 1); in TEST() 70 EXPECT_EQ(model->subgraphs[0]->tensors.size(), 33); in TEST() 75 EXPECT_EQ(model->subgraphs[0]->tensors[0]->name, "lstm_tensor0"); in TEST() 76 EXPECT_EQ(model->subgraphs[0]->tensors[21]->name, "intermediate_0_0"); in TEST() 77 EXPECT_EQ(model->subgraphs[0]->tensors[22]->name, "intermediate_0_1"); in TEST() 78 EXPECT_EQ(model->subgraphs[0]->tensors[23]->name, "intermediate_0_2"); in TEST() [all …]
|
D | modify_model_interface_test.cc | 43 model->subgraphs.push_back(std::move(subgraph)); in CreateQuantizedModelSingleInputOutput() 74 model->subgraphs[0]->operators.push_back(std::move(quant_op)); in CreateQuantizedModelSingleInputOutput() 75 model->subgraphs[0]->operators.push_back(std::move(fc_op)); in CreateQuantizedModelSingleInputOutput() 76 model->subgraphs[0]->operators.push_back(std::move(dequant_op)); in CreateQuantizedModelSingleInputOutput() 83 model->subgraphs[0]->inputs = {0}; in CreateQuantizedModelSingleInputOutput() 84 model->subgraphs[0]->outputs = {3}; in CreateQuantizedModelSingleInputOutput() 113 model->subgraphs[0]->tensors.push_back(std::move(tensor_0)); in CreateQuantizedModelSingleInputOutput() 114 model->subgraphs[0]->tensors.push_back(std::move(tensor_1)); in CreateQuantizedModelSingleInputOutput() 115 model->subgraphs[0]->tensors.push_back(std::move(tensor_2)); in CreateQuantizedModelSingleInputOutput() 116 model->subgraphs[0]->tensors.push_back(std::move(tensor_3)); in CreateQuantizedModelSingleInputOutput() [all …]
|
D | quantization_wrapper_utils_test.cc | 47 model->subgraphs.push_back(std::move(subgraph)); in TEST() 57 model->subgraphs[0]->tensors.push_back(std::move(tensor)); in TEST() 59 model->subgraphs[0]->operators.push_back(std::move(lstm_op)); in TEST() 69 EXPECT_EQ(model->subgraphs.size(), 1); in TEST() 70 EXPECT_EQ(model->subgraphs[0]->operators.size(), 1); in TEST() 71 EXPECT_EQ(model->subgraphs[0]->tensors.size(), 26); in TEST() 76 EXPECT_EQ(model->subgraphs[0]->tensors[0]->name, "lstm_tensor0"); in TEST() 77 EXPECT_EQ(model->subgraphs[0]->tensors[21]->name, "intermediate_0_0"); in TEST() 78 EXPECT_EQ(model->subgraphs[0]->tensors[22]->name, "intermediate_0_1"); in TEST() 79 EXPECT_EQ(model->subgraphs[0]->tensors[23]->name, "intermediate_0_2"); in TEST() [all …]
|
D | quantize_weights_test.cc | 98 for (size_t subgraph_idx = 0; subgraph_idx < model_->subgraphs()->size(); in IsModelInputOrOutput() 100 const auto subgraph = model->subgraphs()->Get(subgraph_idx); in IsModelInputOrOutput() 119 const auto subgraph = model->subgraphs()->Get(subgraph_idx); in GetProducerOpCode() 158 for (size_t subgraph_idx = 0; subgraph_idx < model_->subgraphs()->size(); in TEST_F() 160 const auto quantized_graph = output_model->subgraphs()->Get(subgraph_idx); in TEST_F() 161 const auto float_graph = model_->subgraphs()->Get(subgraph_idx); in TEST_F() 189 ASSERT_EQ(output_model->subgraphs()->size(), model_->subgraphs()->size()); in TEST_F() 190 for (size_t subgraph_idx = 0; subgraph_idx < model_->subgraphs()->size(); in TEST_F() 192 const auto quantized_graph = output_model->subgraphs()->Get(subgraph_idx); in TEST_F() 193 const auto float_graph = model_->subgraphs()->Get(subgraph_idx); in TEST_F() [all …]
|
D | quantization_utils_test.cc | 469 ASSERT_TRUE(readonly_model->subgraphs()); in TEST_F() 470 ASSERT_GE(readonly_model->subgraphs()->size(), 1); in TEST_F() 473 auto subgraph = model.subgraphs[0].get(); in TEST_F() 519 ASSERT_TRUE(readonly_model->subgraphs()); in TEST_F() 520 ASSERT_GE(readonly_model->subgraphs()->size(), 1); in TEST_F() 523 auto subgraph = model.subgraphs[0].get(); in TEST_F() 554 ASSERT_TRUE(readonly_model->subgraphs()); in TEST_F() 555 ASSERT_GE(readonly_model->subgraphs()->size(), 1); in TEST_F() 558 auto subgraph = model.subgraphs[0].get(); in TEST_F() 594 model->subgraphs.push_back(std::move(subgraph)); in TEST_F() [all …]
|
D | quantize_model_test.cc | 85 ASSERT_EQ(model.subgraphs.size(), expected_model.subgraphs.size()); in ExpectSameModels() 86 for (size_t subgraph_idx = 0; subgraph_idx < model.subgraphs.size(); in ExpectSameModels() 88 const auto graph = model.subgraphs[subgraph_idx].get(); in ExpectSameModels() 89 const auto expected_graph = expected_model.subgraphs[subgraph_idx].get(); in ExpectSameModels() 153 ASSERT_EQ(model_.subgraphs.size(), readonly_model_->subgraphs()->size()); in TEST_P() 155 ASSERT_EQ(model_.subgraphs.size(), readonly_model_->subgraphs()->size()); in TEST_P() 156 for (size_t subgraph_idx = 0; subgraph_idx < model_.subgraphs.size(); in TEST_P() 158 const auto quantized_graph = model_.subgraphs[subgraph_idx].get(); in TEST_P() 159 const auto float_graph = readonly_model_->subgraphs()->Get(subgraph_idx); in TEST_P() 178 ASSERT_EQ(model_.subgraphs.size(), readonly_model_->subgraphs()->size()); in TEST_P() [all …]
|
D | quantization_wrapper_utils.cc | 48 for (int subgraph_idx = 0; subgraph_idx < model->subgraphs.size(); in IntermediateTensorExists() 50 SubGraphT* subgraph = model->subgraphs.at(subgraph_idx).get(); in IntermediateTensorExists() 78 if (model->subgraphs.size() == 1 && model->subgraphs[0]->operators.empty()) { in AddIntermediateTensorsToFusedOp() 86 for (int subgraph_idx = 0; subgraph_idx < model->subgraphs.size(); in AddIntermediateTensorsToFusedOp() 88 SubGraphT* subgraph = model->subgraphs.at(subgraph_idx).get(); in AddIntermediateTensorsToFusedOp()
|
D | modify_model_interface.cc | 53 for (size_t subgraph_idx = 0; subgraph_idx < model->subgraphs.size(); in GetInputTensors() 55 SubGraphT* subgraph = model->subgraphs.at(subgraph_idx).get(); in GetInputTensors() 128 for (size_t subgraph_idx = 0; subgraph_idx < model->subgraphs.size(); in GetOutputTensors() 130 SubGraphT* subgraph = model->subgraphs.at(subgraph_idx).get(); in GetOutputTensors() 200 SubGraphT* subgraph = model->subgraphs.at(tot.subgraph_index).get(); in SetInputTypeToUINT8() 227 SubGraphT* subgraph = model->subgraphs.at(tot.subgraph_index).get(); in SetOutputTypeToUINT8() 260 SubGraphT* subgraph = model->subgraphs.at(tot.subgraph_index).get(); in RemoveInputTensor() 286 SubGraphT* subgraph = model->subgraphs.at(tot.subgraph_index).get(); in RemoveOutputTensor() 306 return model->subgraphs[0]->tensors.size() - outputs.size() - inputs.size(); in GetOriginalNumberOfTensors() 396 for (size_t subgraph_idx = 0; subgraph_idx < model->subgraphs.size(); in AddUint8Dequant() [all …]
|
D | model_utils_test.cc | 50 model.subgraphs.push_back(std::move(subgraph)); in TEST() 53 EXPECT_FALSE(HasBuffer(&model, model.subgraphs[0].get(), 0)); in TEST() 55 EXPECT_TRUE(HasBuffer(&model, model.subgraphs[0].get(), 0)); in TEST()
|
D | quantize_model.cc | 60 const SubGraphT* subgraph = model->subgraphs[subgraph_index].get(); in GetOperatorProperty() 89 for (size_t subgraph_idx = 0; subgraph_idx < model->subgraphs.size(); in PopulateRealValueOpSet() 91 SubGraphT* subgraph = model->subgraphs.at(subgraph_idx).get(); in PopulateRealValueOpSet() 390 for (int subgraph_idx = 0; subgraph_idx < model->subgraphs.size(); in SetInputAndOutputTypes() 392 SubGraphT* subgraph = model->subgraphs.at(subgraph_idx).get(); in SetInputAndOutputTypes() 442 for (int subgraph_idx = 0; subgraph_idx < model->subgraphs.size(); in ApplyConstraints() 444 SubGraphT* subgraph = model->subgraphs.at(subgraph_idx).get(); in ApplyConstraints() 524 for (int subgraph_idx = 0, end = model->subgraphs.size(); subgraph_idx < end; in SetOperatorPropertyADDSUBOperator() 526 SubGraphT* subgraph = model->subgraphs.at(subgraph_idx).get(); in SetOperatorPropertyADDSUBOperator() 602 SubGraphT* subgraph = model->subgraphs.at(subgraph_idx).get(); in QuantizeOpInput() [all …]
|
/external/dagger2/java/dagger/internal/codegen/binding/ |
D | LegacyBindingGraph.java | 36 private final ImmutableList<LegacyBindingGraph> subgraphs; field in LegacyBindingGraph 42 ImmutableList<LegacyBindingGraph> subgraphs) { in LegacyBindingGraph() argument 46 this.subgraphs = checkForDuplicates(subgraphs); in LegacyBindingGraph() 66 ImmutableList<LegacyBindingGraph> subgraphs() { in subgraphs() method in LegacyBindingGraph 67 return subgraphs; in subgraphs()
|
/external/tensorflow/tensorflow/lite/toco/tflite/ |
D | import_test.cc | 129 auto subgraphs = BuildSubGraphs(tensors, operators); in BuildTestModel() local 134 opcodes, subgraphs, s, buffers)); in BuildTestModel() 191 auto subgraphs = BuildSubGraphs(tensors, operators); in TEST_F() local 195 subgraphs, comment, buffers)); in TEST_F() 205 auto subgraphs = BuildSubGraphs(tensors, operators); in TEST_F() local 209 subgraphs, comment, buffers)); in TEST_F() 219 auto subgraphs = BuildSubGraphs(tensors, operators); in TEST_F() local 223 subgraphs, comment, buffers)); in TEST_F() 234 auto subgraphs = BuildSubGraphs(tensors, operators); in TEST_F() local 238 subgraphs, comment, buffers)); in TEST_F() [all …]
|
D | import.cc | 34 auto tensors = (*input_model.subgraphs())[0]->tensors(); in LoadTensorsTable() 57 auto tensors = (*input_model.subgraphs())[0]->tensors(); in ImportTensors() 107 auto ops = (*input_model.subgraphs())[0]->operators(); in ImportOperators() 174 auto inputs = (*input_model.subgraphs())[0]->inputs(); in ImportIOTensors() 185 auto outputs = (*input_model.subgraphs())[0]->outputs(); in ImportIOTensors() 215 if (!input_model->subgraphs() || input_model->subgraphs()->size() != 1) { in Import()
|
/external/tensorflow/tensorflow/lite/tools/ |
D | list_flex_ops_no_kernel.cc | 31 auto* subgraphs = model->subgraphs(); in AddFlexOpsFromModel() local 32 if (!subgraphs) return; in AddFlexOpsFromModel() 34 for (int subgraph_index = 0; subgraph_index < subgraphs->size(); in AddFlexOpsFromModel() 36 const tflite::SubGraph* subgraph = subgraphs->Get(subgraph_index); in AddFlexOpsFromModel()
|
D | list_flex_ops.cc | 80 auto* subgraphs = model->subgraphs(); in AddFlexOpsFromModel() local 81 if (!subgraphs) return; in AddFlexOpsFromModel() 82 for (int subgraph_index = 0; subgraph_index < subgraphs->size(); in AddFlexOpsFromModel() 84 const tflite::SubGraph* subgraph = subgraphs->Get(subgraph_index); in AddFlexOpsFromModel()
|
D | flatbuffer_utils_test.py | 50 initial_subgraph = initial_model.subgraphs[0] 51 final_subgraph = final_model.subgraphs[0] 96 initial_subgraph = initial_model.subgraphs[0] 97 final_subgraph = final_model.subgraphs[0] 98 self.assertIsNotNone(initial_model.subgraphs[0].name) 99 self.assertIsNone(final_model.subgraphs[0].name) 141 initial_subgraph = initial_model.subgraphs[0] 142 final_subgraph = final_model.subgraphs[0]
|
/external/tensorflow/tensorflow/lite/kernels/ |
D | call_once.cc | 69 auto* subgraphs = this_subgraph->GetSubgraphs(); in Prepare() local 70 TF_LITE_ENSURE(context, op_data->init_subgraph_index < subgraphs->size()); in Prepare() 73 Subgraph* init_subgraph = (*subgraphs)[op_data->init_subgraph_index].get(); in Prepare() 86 auto* subgraphs = this_subgraph->GetSubgraphs(); in Eval() local 87 Subgraph& init_subgraph = *(*subgraphs)[op_data->init_subgraph_index]; in Eval()
|
D | if.cc | 71 auto* subgraphs = this_subgraph->GetSubgraphs(); in Prepare() local 72 TF_LITE_ENSURE(context, op_data->then_subgraph_index < subgraphs->size()); in Prepare() 73 TF_LITE_ENSURE(context, op_data->else_subgraph_index < subgraphs->size()); in Prepare() 75 Subgraph* then_subgraph = (*subgraphs)[op_data->then_subgraph_index].get(); in Prepare() 76 Subgraph* else_subgraph = (*subgraphs)[op_data->else_subgraph_index].get(); in Prepare() 145 auto* subgraphs = this_subgraph->GetSubgraphs(); in Eval() local 153 *(*subgraphs)[active_branch_subgraph_index]; in Eval()
|
D | while.cc | 138 auto* subgraphs = this_subgraph->GetSubgraphs(); in Prepare() local 139 TF_LITE_ENSURE(context, op_data->cond_subgraph_index < subgraphs->size()); in Prepare() 140 TF_LITE_ENSURE(context, op_data->body_subgraph_index < subgraphs->size()); in Prepare() 142 Subgraph* cond_subgraph = (*subgraphs)[op_data->cond_subgraph_index].get(); in Prepare() 143 Subgraph* body_subgraph = (*subgraphs)[op_data->body_subgraph_index].get(); in Prepare() 217 auto* subgraphs = this_subgraph->GetSubgraphs(); in Eval() local 218 Subgraph* cond_subgraph = (*subgraphs)[op_data->cond_subgraph_index].get(); in Eval() 219 Subgraph* body_subgraph = (*subgraphs)[op_data->body_subgraph_index].get(); in Eval()
|
/external/tensorflow/tensorflow/core/common_runtime/ |
D | partitioning_utils_test.cc | 122 std::unordered_map<string, std::unique_ptr<Graph>> subgraphs; in TEST_F() local 124 PartitionFunctionGraph(device_set_, std::move(graph), &subgraphs); in TEST_F() 133 std::unordered_map<string, std::unique_ptr<Graph>> subgraphs; in TEST_F() local 135 PartitionFunctionGraph(device_set_, std::move(graph), &subgraphs); in TEST_F() 138 ASSERT_EQ(1, subgraphs.size()); in TEST_F() 139 const auto& pair = *subgraphs.begin(); in TEST_F() 148 std::unordered_map<string, std::unique_ptr<Graph>> subgraphs; in TEST_F() local 150 PartitionFunctionGraph(device_set_, std::move(graph), &subgraphs); in TEST_F() 153 ASSERT_EQ(2, subgraphs.size()); in TEST_F() 155 const auto& part1 = subgraphs["/job:a/replica:0/task:0/device:CPU:0"]; in TEST_F() [all …]
|
/external/tensorflow/tensorflow/lite/python/ |
D | util.py | 586 model_object.subgraphs[0].inputs[0] = model_object.subgraphs[0].inputs[0] 603 if len(model.subgraphs) > 1: 605 "{} subgraphs.".format(len(model.subgraphs))) 606 subgraph = model.subgraphs[0] 648 subgraph = model.subgraphs[0] 741 subgraph = model.subgraphs[0] 875 if len(model_object.subgraphs) > 1: 877 "{} subgraphs.".format(len(model_object.subgraphs)))
|
/external/tensorflow/tensorflow/lite/tools/optimize/calibration/ |
D | calibration_reader.cc | 38 if (!model || model->subgraphs.empty()) { in AddCalibrationToModel() 41 const auto& subgraph = model->subgraphs[0]; in AddCalibrationToModel()
|
D | calibrator.cc | 336 auto subgraphs = tflite_model->subgraphs(); in BuildLoggingInterpreter() local 339 if (subgraphs->size() != 1) { in BuildLoggingInterpreter() 343 subgraphs->size()); in BuildLoggingInterpreter() 352 auto primary_subgraph = subgraphs->Get(0); in BuildLoggingInterpreter()
|
/external/tensorflow/tensorflow/lite/tools/versioning/ |
D | runtime_version.cc | 362 auto subgraphs = model->subgraphs(); in UpdateMinimumRuntimeVersionForModel() local 363 for (int i = 0; i < subgraphs->Length(); ++i) { in UpdateMinimumRuntimeVersionForModel() 364 const SubGraph* subgraph = subgraphs->Get(i); in UpdateMinimumRuntimeVersionForModel()
|
/external/tensorflow/tensorflow/lite/micro/ |
D | test_helpers.cc | 211 const flatbuffers::Offset<SubGraph> subgraphs[subgraphs_size] = { in BuildModel() local 224 builder_->CreateVector(subgraphs, subgraphs_size), in BuildModel() 233 builder_->CreateVector(subgraphs, subgraphs_size), in BuildModel() 399 const Offset<SubGraph> subgraphs[subgraphs_size] = { in BuildSimpleMockModel() local 412 builder->CreateVector(subgraphs, subgraphs_size), in BuildSimpleMockModel() 548 const Offset<SubGraph> subgraphs[subgraphs_size] = { in BuildComplexMockModel() local 563 builder->CreateVector(subgraphs, subgraphs_size), in BuildComplexMockModel() 619 const Offset<SubGraph> subgraphs[subgraphs_size] = { in BuildSimpleMultipleInputsModel() local 632 builder->CreateVector(subgraphs, subgraphs_size), in BuildSimpleMultipleInputsModel() 1071 auto* subgraphs = model->subgraphs(); in GetModelTensorCount() local [all …]
|