Home
last modified time | relevance | path

Searched defs:subgraph_idx (Results 1 – 12 of 12) sorted by relevance

/external/tensorflow/tensorflow/lite/tools/optimize/
Dquantize_weights_test.cc98 for (size_t subgraph_idx = 0; subgraph_idx < model_->subgraphs()->size(); in IsModelInputOrOutput() local
116 bool GetProducerOpCode(const Model* model, uint32_t subgraph_idx, in GetProducerOpCode()
160 for (size_t subgraph_idx = 0; subgraph_idx < model_->subgraphs()->size(); in TEST_F() local
193 for (size_t subgraph_idx = 0; subgraph_idx < model_->subgraphs()->size(); in TEST_F() local
249 for (size_t subgraph_idx = 0; subgraph_idx < model_->subgraphs()->size(); in TEST_F() local
309 for (size_t subgraph_idx = 0; subgraph_idx < model_->subgraphs()->size(); in TEST_F() local
369 for (size_t subgraph_idx = 0; subgraph_idx < model_->subgraphs()->size(); in TEST_F() local
404 for (size_t subgraph_idx = 0; subgraph_idx < model_->subgraphs()->size(); in TEST_F() local
445 for (size_t subgraph_idx = 0; subgraph_idx < model_->subgraphs()->size(); in TEST_F() local
567 for (size_t subgraph_idx = 0; subgraph_idx < model_->subgraphs()->size(); in TEST_F() local
[all …]
Dquantize_model.cc50 for (size_t subgraph_idx = 0; subgraph_idx < model->subgraphs.size(); in DuplicateBiasesWithMultipleUses() local
76 for (size_t subgraph_idx = 0; subgraph_idx < model->subgraphs.size(); in DuplicateBiasesWithMultipleUses() local
216 for (size_t subgraph_idx = 0; subgraph_idx < model->subgraphs.size(); in PopulateRealValueOpSet() local
541 for (int subgraph_idx = 0; subgraph_idx < model->subgraphs.size(); in SetInputAndOutputTypes() local
691 for (int subgraph_idx = 0; subgraph_idx < model->subgraphs.size(); in ApplyConstraints() local
800 for (int subgraph_idx = 0, end = model->subgraphs.size(); subgraph_idx < end; in SetOperatorPropertyADDSUBOperator() local
872 ModelT* model, int32_t subgraph_idx, size_t* op_idx, in QuantizeOpInput()
1065 ModelT* model, int32_t subgraph_idx, int32_t op_idx, in QuantizeOpOutput()
1166 for (size_t subgraph_idx = 0; subgraph_idx < model->subgraphs.size(); in QuantizeIntermediateTensors() local
1239 for (size_t subgraph_idx = 0; subgraph_idx < model->subgraphs.size(); in QuantizeSharedRange() local
[all …]
Dquantization_wrapper_utils.cc50 for (int subgraph_idx = 0; subgraph_idx < model->subgraphs.size(); in IntermediateTensorExists() local
88 for (int subgraph_idx = 0; subgraph_idx < model->subgraphs.size(); in AddIntermediateTensorsToFusedOp() local
Dmodify_model_interface.cc56 for (size_t subgraph_idx = 0; subgraph_idx < model->subgraphs.size(); in GetInputTensors() local
131 for (size_t subgraph_idx = 0; subgraph_idx < model->subgraphs.size(); in GetOutputTensors() local
399 for (size_t subgraph_idx = 0; subgraph_idx < model->subgraphs.size(); in AddUint8Dequant() local
439 for (size_t subgraph_idx = 0; subgraph_idx < model->subgraphs.size(); in AddUint8Quant() local
Dquantize_model_test.cc94 for (size_t subgraph_idx = 0; subgraph_idx < model.subgraphs.size(); in ExpectSameModels() local
167 for (size_t subgraph_idx = 0; subgraph_idx < model_.subgraphs.size(); in TEST_P() local
190 for (size_t subgraph_idx = 0; subgraph_idx < model_.subgraphs.size(); in TEST_P() local
231 for (size_t subgraph_idx = 0; subgraph_idx < model_.subgraphs.size(); in TEST_P() local
272 for (int32_t subgraph_idx = 0; subgraph_idx < model_.subgraphs.size(); in TEST_P() local
337 for (int32_t subgraph_idx = 0; subgraph_idx < model_.subgraphs.size(); in TEST_P() local
456 const int32_t subgraph_idx = 0; in TEST_P() local
558 const int32_t subgraph_idx = 0; in TEST_F() local
1380 for (size_t subgraph_idx = 0; subgraph_idx < model_.subgraphs.size(); in TEST_F() local
1866 const int32_t subgraph_idx = 0; in TEST_P() local
[all …]
Dmodel_utils.cc134 for (int subgraph_idx = 0, end = model->subgraphs.size(); subgraph_idx < end; in SetOperatorCodeVersion() local
/external/tensorflow/tensorflow/compiler/mlir/lite/quantization/lite/
Dquantize_weights_test.cc108 for (size_t subgraph_idx = 0; subgraph_idx < model_->subgraphs()->size(); in IsModelInputOrOutput() local
126 bool GetProducerOpCode(const Model* model, uint32_t subgraph_idx, in GetProducerOpCode()
230 for (size_t subgraph_idx = 0; subgraph_idx < model_->subgraphs()->size(); in TEST_F() local
262 for (size_t subgraph_idx = 0; subgraph_idx < model_->subgraphs()->size(); in TEST_F() local
319 for (size_t subgraph_idx = 0; subgraph_idx < model_->subgraphs()->size(); in TEST_F() local
378 for (size_t subgraph_idx = 0; subgraph_idx < model_->subgraphs()->size(); in TEST_F() local
440 for (size_t subgraph_idx = 0; subgraph_idx < model_->subgraphs()->size(); in TEST_F() local
474 for (size_t subgraph_idx = 0; subgraph_idx < model_->subgraphs()->size(); in TEST_F() local
514 for (size_t subgraph_idx = 0; subgraph_idx < model_->subgraphs()->size(); in TEST_F() local
632 for (size_t subgraph_idx = 0; subgraph_idx < model_->subgraphs()->size(); in TEST_F() local
[all …]
Dquantize_model_test.cc197 for (size_t subgraph_idx = 0; subgraph_idx < model.subgraphs.size(); in ExpectSameModels() local
339 const int32_t subgraph_idx = 0; in TEST_F() local
1359 const int32_t subgraph_idx = 0; in TEST_P() local
1426 const int32_t subgraph_idx = 0; in TEST_P() local
1483 const int32_t subgraph_idx = 0; in TEST_F() local
/external/tensorflow/tensorflow/lite/python/analyzer_wrapper/
Dmodel_analyzer.cc69 const std::string tensor_str(const int tensor_idx, const int subgraph_idx, in tensor_str()
89 const std::string subgraph_str(const int subgraph_idx) { in subgraph_str()
103 const int subgraph_idx, const tflite::Model* model, in dump_tensor_detail()
150 const int subgraph_idx, in dump_tensor_list()
/external/pytorch/torch/ao/ns/fx/
Dn_shadows_utils.py37 def _get_attr_name(subgraph_idx, subgraph_candidate_idx): argument
41 def _get_attr_wrapper_name(subgraph_idx, subgraph_candidate_idx): argument
/external/tensorflow/tensorflow/lite/tools/
Dvisualize.py287 def GenerateGraph(subgraph_idx, g, opcode_mapper): argument
/external/tensorflow/tensorflow/compiler/mlir/lite/
Dflatbuffer_export.cc1932 int subgraph_idx = 0; in TranslateInternal() local