| /external/tensorflow/tensorflow/lite/tools/optimize/ |
| D | quantize_weights_test.cc | 98 for (size_t subgraph_idx = 0; subgraph_idx < model_->subgraphs()->size(); in IsModelInputOrOutput() local
99 ++subgraph_idx) { in IsModelInputOrOutput()
100 const auto subgraph = model->subgraphs()->Get(subgraph_idx); in IsModelInputOrOutput()
116 bool GetProducerOpCode(const Model* model, uint32_t subgraph_idx, in GetProducerOpCode() argument
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() local
159 subgraph_idx++) { 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()
190 for (size_t subgraph_idx = 0; subgraph_idx < model_->subgraphs()->size(); in TEST_F() local
[all …]
|
| D | quantize_model.cc | 89 for (size_t subgraph_idx = 0; subgraph_idx < model->subgraphs.size(); in PopulateRealValueOpSet() local
90 subgraph_idx++) { in PopulateRealValueOpSet()
91 SubGraphT* subgraph = model->subgraphs.at(subgraph_idx).get(); in PopulateRealValueOpSet()
96 GetOperatorProperty(operator_names, model, subgraph_idx, op_idx, in PopulateRealValueOpSet()
390 for (int subgraph_idx = 0; subgraph_idx < model->subgraphs.size(); in SetInputAndOutputTypes() local
391 subgraph_idx++) { 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() local
443 subgraph_idx++) { in ApplyConstraints()
444 SubGraphT* subgraph = model->subgraphs.at(subgraph_idx).get(); in ApplyConstraints()
[all …]
|
| D | quantization_wrapper_utils.cc | 48 for (int subgraph_idx = 0; subgraph_idx < model->subgraphs.size(); in IntermediateTensorExists() local
49 ++subgraph_idx) { in IntermediateTensorExists()
50 SubGraphT* subgraph = model->subgraphs.at(subgraph_idx).get(); in IntermediateTensorExists()
86 for (int subgraph_idx = 0; subgraph_idx < model->subgraphs.size(); in AddIntermediateTensorsToFusedOp() local
87 ++subgraph_idx) { in AddIntermediateTensorsToFusedOp()
88 SubGraphT* subgraph = model->subgraphs.at(subgraph_idx).get(); in AddIntermediateTensorsToFusedOp()
93 operator_property::GetOperatorProperty(model, subgraph_idx, op_idx); in AddIntermediateTensorsToFusedOp()
|
| D | modify_model_interface.cc | 53 for (size_t subgraph_idx = 0; subgraph_idx < model->subgraphs.size(); in GetInputTensors() local
54 subgraph_idx++) { in GetInputTensors()
55 SubGraphT* subgraph = model->subgraphs.at(subgraph_idx).get(); in GetInputTensors()
114 result.push_back({subgraph_idx, op->inputs[0], op_idx, op->outputs[0], in GetInputTensors()
128 for (size_t subgraph_idx = 0; subgraph_idx < model->subgraphs.size(); in GetOutputTensors() local
129 subgraph_idx++) { in GetOutputTensors()
130 SubGraphT* subgraph = model->subgraphs.at(subgraph_idx).get(); in GetOutputTensors()
189 result.push_back({subgraph_idx, op->inputs[0], op_idx, op->outputs[0], in GetOutputTensors()
396 for (size_t subgraph_idx = 0; subgraph_idx < model->subgraphs.size(); in AddUint8Dequant() local
397 subgraph_idx++) { in AddUint8Dequant()
[all …]
|
| D | quantize_model_test.cc | 86 for (size_t subgraph_idx = 0; subgraph_idx < model.subgraphs.size(); in ExpectSameModels() local
87 subgraph_idx++) { 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()
156 for (size_t subgraph_idx = 0; subgraph_idx < model_.subgraphs.size(); in TEST_P() local
157 subgraph_idx++) { 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()
179 for (size_t subgraph_idx = 0; subgraph_idx < model_.subgraphs.size(); in TEST_P() local
180 subgraph_idx++) { in TEST_P()
[all …]
|
| D | model_utils.cc | 133 for (int subgraph_idx = 0, end = model->subgraphs.size(); subgraph_idx < end; in SetOperatorCodeVersion() local
134 subgraph_idx++) { in SetOperatorCodeVersion()
135 SubGraphT* subgraph = model->subgraphs.at(subgraph_idx).get(); in SetOperatorCodeVersion()
141 operator_property::GetOperatorProperty(model, subgraph_idx, op_idx); in SetOperatorCodeVersion()
|
| /external/tensorflow/tensorflow/lite/python/ |
| D | test_util.py | 31 for subgraph_idx in range(model.SubgraphsLength()):
32 subgraph = model.Subgraphs(subgraph_idx)
50 for subgraph_idx in range(model.SubgraphsLength()):
51 subgraph = model.Subgraphs(subgraph_idx)
|
| /external/tensorflow/tensorflow/lite/tools/ |
| D | visualize.py | 281 def GenerateGraph(subgraph_idx, g, opcode_mapper): argument
336 html = _D3_HTML_TEMPLATE % (graph_str, subgraph_idx)
462 for subgraph_idx, g in enumerate(data["subgraphs"]):
475 html += "<h2>Subgraph %d</h2>\n" % subgraph_idx
495 subgraph_idx,)
496 html += GenerateGraph(subgraph_idx, g, opcode_mapper)
|
| /external/tensorflow/tensorflow/lite/micro/ |
| D | micro_allocator.cc | 108 int subgraph_idx = metadata_buffer[1]; in CheckOfflinePlannedOffsets() local
118 version, subgraph_idx, nbr_offline_offsets); in CheckOfflinePlannedOffsets()
131 if (subgraph_idx != 0) { in CheckOfflinePlannedOffsets()
134 subgraph_idx); in CheckOfflinePlannedOffsets()
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/ |
| D | flatbuffer_export.cc | 1732 int subgraph_idx = 0; in TranslateInternal() local
1734 subgraph_index_map_[main_fn.getName().str()] = subgraph_idx++; in TranslateInternal()
1739 subgraph_index_map_[fn.getName().str()] = subgraph_idx++; in TranslateInternal()
|