| /external/tensorflow/tensorflow/lite/tools/optimize/ |
| D | quantize_weights_test.cc | 74 for (size_t subgraph_idx = 0; subgraph_idx < model_->subgraphs()->size(); in IsModelInputOrOutput()
76 const auto subgraph = model->subgraphs()->Get(subgraph_idx); in IsModelInputOrOutput()
95 const auto subgraph = model->subgraphs()->Get(subgraph_idx); in GetProducerOpCode()
134 for (size_t subgraph_idx = 0; subgraph_idx < model_->subgraphs()->size(); in TEST_F()
136 const auto quantized_graph = output_model->subgraphs()->Get(subgraph_idx); in TEST_F()
137 const auto float_graph = model_->subgraphs()->Get(subgraph_idx); in TEST_F()
165 ASSERT_EQ(output_model->subgraphs()->size(), model_->subgraphs()->size()); in TEST_F()
166 for (size_t subgraph_idx = 0; subgraph_idx < model_->subgraphs()->size(); in TEST_F()
168 const auto quantized_graph = output_model->subgraphs()->Get(subgraph_idx); in TEST_F()
169 const auto float_graph = model_->subgraphs()->Get(subgraph_idx); in TEST_F()
[all …]
|
| D | quantize_model_test.cc | 76 ASSERT_EQ(model_.subgraphs.size(), readonly_model_->subgraphs()->size()); in TEST_F()
77 for (size_t subgraph_idx = 0; subgraph_idx < model_.subgraphs.size(); in TEST_F()
79 const auto quantized_graph = model_.subgraphs[subgraph_idx].get(); in TEST_F()
80 const auto float_graph = readonly_model_->subgraphs()->Get(subgraph_idx); in TEST_F()
105 ASSERT_EQ(model_.subgraphs.size(), readonly_model_->subgraphs()->size()); in TEST_F()
106 for (size_t subgraph_idx = 0; subgraph_idx < model_.subgraphs.size(); in TEST_F()
108 const auto quantized_graph = model_.subgraphs[subgraph_idx].get(); in TEST_F()
109 const auto float_graph = readonly_model_->subgraphs()->Get(subgraph_idx); in TEST_F()
125 for (const auto& subgraph : model_.subgraphs) { in TEST_F()
|
| D | subgraph_quantizer_test.cc | 67 ASSERT_TRUE(readonly_model->subgraphs()); in TEST()
68 ASSERT_GE(readonly_model->subgraphs()->size(), 1); in TEST()
71 auto subgraph = model.subgraphs[0].get(); in TEST()
166 ASSERT_TRUE(readonly_model->subgraphs()); in TEST()
167 ASSERT_GE(readonly_model->subgraphs()->size(), 1); in TEST()
170 auto subgraph = model.subgraphs[0].get(); in TEST()
278 ASSERT_TRUE(readonly_model->subgraphs()); in TEST()
279 ASSERT_GE(readonly_model->subgraphs()->size(), 1); in TEST()
282 auto subgraph = model.subgraphs[0].get(); in TEST()
296 auto float_graph = readonly_model->subgraphs()->Get(0); in TEST()
[all …]
|
| D | quantize_model.cc | 37 for (size_t subgraph_idx = 0; subgraph_idx < model->subgraphs.size(); in QuantizeModel()
39 SubGraphT* subgraph = model->subgraphs.at(subgraph_idx).get(); in QuantizeModel()
|
| D | quantization_utils_test.cc | 236 ASSERT_TRUE(readonly_model->subgraphs()); in TEST()
237 ASSERT_GE(readonly_model->subgraphs()->size(), 1); in TEST()
240 auto subgraph = model.subgraphs[0].get(); in TEST()
|
| D | quantize_weights.cc | 151 SubGraphT* subgraph = model->subgraphs.at(0).get(); in InsertQuantizableInputTensorsFromOperator()
261 if (model->subgraphs.size() != 1) { in QuantizeWeightsInternal()
267 SubGraphT* subgraph = model->subgraphs.at(0).get(); in QuantizeWeightsInternal()
|
| /external/tensorflow/tensorflow/lite/toco/tflite/ |
| D | import_test.cc | 128 auto subgraphs = BuildSubGraphs(tensors, operators); in BuildTestModel() local
133 opcodes, subgraphs, s, buffers)); in BuildTestModel()
190 auto subgraphs = BuildSubGraphs(tensors, operators); in TEST_F() local
194 subgraphs, comment, buffers)); in TEST_F()
204 auto subgraphs = BuildSubGraphs(tensors, operators); in TEST_F() local
208 subgraphs, comment, buffers)); in TEST_F()
218 auto subgraphs = BuildSubGraphs(tensors, operators); in TEST_F() local
222 subgraphs, comment, buffers)); in TEST_F()
233 auto subgraphs = BuildSubGraphs(tensors, operators); in TEST_F() local
237 subgraphs, comment, buffers)); in TEST_F()
[all …]
|
| D | import.cc | 33 auto tensors = (*input_model.subgraphs())[0]->tensors(); in LoadTensorsTable()
56 auto tensors = (*input_model.subgraphs())[0]->tensors(); in ImportTensors()
106 auto ops = (*input_model.subgraphs())[0]->operators(); in ImportOperators()
173 auto inputs = (*input_model.subgraphs())[0]->inputs(); in ImportIOTensors()
184 auto outputs = (*input_model.subgraphs())[0]->outputs(); in ImportIOTensors()
214 if (!input_model->subgraphs() || input_model->subgraphs()->size() != 1) { in Import()
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | partitioning_utils_test.cc | 115 std::unordered_map<string, std::unique_ptr<Graph>> subgraphs; in TEST_F() local
117 PartitionFunctionGraph(device_set_, std::move(graph), &subgraphs); in TEST_F()
126 std::unordered_map<string, std::unique_ptr<Graph>> subgraphs; in TEST_F() local
128 PartitionFunctionGraph(device_set_, std::move(graph), &subgraphs); in TEST_F()
131 ASSERT_EQ(1, subgraphs.size()); in TEST_F()
132 const auto& pair = *subgraphs.begin(); in TEST_F()
141 std::unordered_map<string, std::unique_ptr<Graph>> subgraphs; in TEST_F() local
143 PartitionFunctionGraph(device_set_, std::move(graph), &subgraphs); in TEST_F()
146 ASSERT_EQ(2, subgraphs.size()); in TEST_F()
148 const auto& part1 = subgraphs["/job:a/replica:0/task:0/device:CPU:0"]; in TEST_F()
[all …]
|
| D | partitioning_utils.cc | 27 std::unordered_map<string, std::unique_ptr<Graph>>* subgraphs) { in PartitionFunctionGraph() argument
66 subgraphs->emplace(device, std::move(subgraph)); in PartitionFunctionGraph()
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | if.cc | 64 auto* subgraphs = this_subgraph->GetSubgraphs(); in Prepare() local
65 TF_LITE_ENSURE(context, op_data->then_subgraph_index < subgraphs->size()); in Prepare()
66 TF_LITE_ENSURE(context, op_data->else_subgraph_index < subgraphs->size()); in Prepare()
68 Subgraph* then_subgraph = (*subgraphs)[op_data->then_subgraph_index].get(); in Prepare()
69 Subgraph* else_subgraph = (*subgraphs)[op_data->else_subgraph_index].get(); in Prepare()
135 auto* subgraphs = this_subgraph->GetSubgraphs(); in Eval() local
143 *(*subgraphs)[active_branch_subgraph_index]; in Eval()
|
| D | while.cc | 116 auto* subgraphs = this_subgraph->GetSubgraphs(); in Prepare() local
117 TF_LITE_ENSURE(context, op_data->cond_subgraph_index < subgraphs->size()); in Prepare()
118 TF_LITE_ENSURE(context, op_data->body_subgraph_index < subgraphs->size()); in Prepare()
120 Subgraph* cond_subgraph = (*subgraphs)[op_data->cond_subgraph_index].get(); in Prepare()
121 Subgraph* body_subgraph = (*subgraphs)[op_data->body_subgraph_index].get(); in Prepare()
196 auto* subgraphs = this_subgraph->GetSubgraphs(); in Eval() local
197 Subgraph* cond_subgraph = (*subgraphs)[op_data->cond_subgraph_index].get(); in Eval()
198 Subgraph* body_subgraph = (*subgraphs)[op_data->body_subgraph_index].get(); in Eval()
|
| D | test_util.cc | 104 std::vector<flatbuffers::Offset<SubGraph>> subgraphs; in BuildInterpreter() local
106 subgraphs.push_back(subgraph); in BuildInterpreter()
107 auto subgraphs_flatbuffer = builder_.CreateVector(subgraphs); in BuildInterpreter()
|
| /external/tensorflow/tensorflow/lite/tools/optimize/calibration/ |
| D | calibrator.cc | 264 auto subgraphs = tflite_model->subgraphs(); in BuildLoggingInterpreter() local
267 if (subgraphs->size() != 1) { in BuildLoggingInterpreter()
271 subgraphs->size()); in BuildLoggingInterpreter()
280 auto primary_subgraph = subgraphs->Get(0); in BuildLoggingInterpreter()
|
| D | calibration_reader.cc | 36 if (!model || model->subgraphs.empty()) { in AddCalibrationToModel()
39 const auto& subgraph = model->subgraphs[0]; in AddCalibrationToModel()
|
| /external/tensorflow/tensorflow/lite/ |
| D | model.cc | 510 auto* subgraphs = model_->subgraphs(); in operator ()() local
513 if (subgraphs->size() == 0) { in operator ()()
520 if (subgraphs->Length() > 1) { in operator ()()
521 (*interpreter)->AddSubgraphs(subgraphs->Length() - 1); in operator ()()
524 for (int subgraph_index = 0; subgraph_index < subgraphs->Length(); in operator ()()
526 const tflite::SubGraph* subgraph = (*subgraphs)[subgraph_index]; in operator ()()
|
| /external/tensorflow/tensorflow/lite/tools/ |
| D | verifier.cc | 264 if (!model.subgraphs()) { in VerifySubGraphs()
268 for (const auto& subgraph : *model.subgraphs()) { in VerifySubGraphs()
287 if (!model.subgraphs()) { in VerifyTensors()
295 for (const auto& subgraph : *model.subgraphs()) { in VerifyTensors()
|
| /external/tensorflow/tensorflow/lite/experimental/micro/ |
| D | micro_interpreter.cc | 72 auto* subgraphs = model->subgraphs(); in MicroInterpreter() local
73 if (subgraphs->size() != 1) { in MicroInterpreter()
78 subgraph_ = (*subgraphs)[0]; in MicroInterpreter()
|
| D | micro_interpreter_test.cc | 137 const Offset<SubGraph> subgraphs[subgraphs_size] = { in BuildMockModel() local
149 builder.CreateVector(subgraphs, subgraphs_size), in BuildMockModel()
|
| /external/tensorflow/tensorflow/lite/g3doc/performance/ |
| D | delegates.md | 31 …fic operations, then TensorFlow Lite will split the graph into multiple subgraphs where each subgr…
39 …ere delegated or multiple nodes handled the subgraphs. In general, you don't want to have multiple…
145 // This call split the graphs into subgraphs, for subgraphs that can be
|
| /external/tensorflow/tensorflow/lite/testdata/ |
| D | add.json | 7 subgraphs: [
|
| D | add_quantized.json | 7 subgraphs: [
|
| /external/tensorflow/tensorflow/core/grappler/graph_analyzer/ |
| D | graph_analyzer_test.cc | 105 auto& subgraphs = GetResult(); in TEST_F() local
106 EXPECT_THAT(subgraphs, SizeIs(0)); in TEST_F()
118 auto& subgraphs = GetResult(); in TEST_F() local
119 EXPECT_THAT(subgraphs, SizeIs(3)); in TEST_F()
|
| /external/tensorflow/tensorflow/lite/schema/ |
| D | schema_v2.fbs | 17 // Version 1: Add subgraphs to schema.
295 // All the subgraphs of the model. The 0th is assumed to be the main
297 subgraphs:[SubGraph];
|
| D | schema_v1.fbs | 17 // Version 1: Add subgraphs to schema.
287 // All the subgraphs of the model. The 0th is assumed to be the main
289 subgraphs:[SubGraph];
|