Home
last modified time | relevance | path

Searched refs:subgraph_index (Results 1 – 25 of 46) sorted by relevance

12

/external/tensorflow/tensorflow/lite/
Dinterpreter_experimental.cc150 for (int subgraph_index = 0; subgraph_index < subgraphs_.size(); in SetSubgraphProfiler() local
151 ++subgraph_index) { in SetSubgraphProfiler()
152 subgraphs_[subgraph_index]->SetProfiler(installed_profiler_, in SetSubgraphProfiler()
153 subgraph_index); in SetSubgraphProfiler()
162 for (int subgraph_index = 0; subgraph_index < subgraphs_.size(); in PreserveAllTensorsExperimental() local
163 ++subgraph_index) { in PreserveAllTensorsExperimental()
165 subgraphs_[subgraph_index]->PreserveAllTensorsExperimental()); in PreserveAllTensorsExperimental()
Dinterpreter.h269 int subgraph_index, int node_index) const { in node_and_registration() argument
270 return subgraph(subgraph_index)->node_and_registration(node_index); in node_and_registration()
328 return signature.subgraph_index; in GetSubgraphIndexFromSignature()
368 const int subgraph_index = GetSubgraphIndexFromSignature(signature_key); in input_tensor_by_signature() local
369 if (subgraph_index == -1) return nullptr; in input_tensor_by_signature()
373 return subgraph(subgraph_index)->tensor(tensor_index); in input_tensor_by_signature()
382 const int subgraph_index = GetSubgraphIndexFromSignature(signature_key); in output_tensor_by_signature() local
383 if (subgraph_index == -1) return nullptr; in output_tensor_by_signature()
387 return subgraph(subgraph_index)->tensor(tensor_index); in output_tensor_by_signature()
679 const Subgraph* subgraph(int subgraph_index) const { in subgraph() argument
[all …]
Dgraph_info_test.cc113 for (size_t subgraph_index = 0; subgraph_index < generated_subgraphs.size(); in CheckPartitionSubgraphs() local
114 subgraph_index++) { in CheckPartitionSubgraphs()
115 EXPECT_EQ(generated_subgraphs[subgraph_index].nodes, in CheckPartitionSubgraphs()
116 expected_subgraphs[subgraph_index].nodes); in CheckPartitionSubgraphs()
117 EXPECT_EQ(generated_subgraphs[subgraph_index].input_tensors, in CheckPartitionSubgraphs()
118 expected_subgraphs[subgraph_index].input_tensors); in CheckPartitionSubgraphs()
119 EXPECT_EQ(generated_subgraphs[subgraph_index].output_tensors, in CheckPartitionSubgraphs()
120 expected_subgraphs[subgraph_index].output_tensors); in CheckPartitionSubgraphs()
Dinterpreter.cc381 SignatureRunner(&signature, subgraph(signature.subgraph_index))}); in GetSignatureRunner()
391 for (int subgraph_index = 0; subgraph_index < subgraphs_.size(); in SetMetadata() local
392 ++subgraph_index) { in SetMetadata()
393 TF_LITE_ENSURE_STATUS(subgraphs_[subgraph_index]->SetMetadata(&metadata_)); in SetMetadata()
Dinterpreter_builder.cc540 signature_def.subgraph_index = fb_signature_def->subgraph_index(); in ParseSignatureDefs()
755 for (int subgraph_index = 0; subgraph_index < subgraphs->size(); in operator ()() local
756 ++subgraph_index) { in operator ()()
757 const tflite::SubGraph* subgraph = (*subgraphs)[subgraph_index]; in operator ()()
759 (*interpreter)->subgraph(subgraph_index); in operator ()()
765 subgraph_index); in operator ()()
/external/tensorflow/tensorflow/lite/python/interpreter_wrapper/
Dinterpreter_wrapper_pybind11.cc94 [](InterpreterWrapper& self, int subgraph_index) { in PYBIND11_MODULE() argument
95 return tensorflow::PyoOrThrow(self.AllocateTensors(subgraph_index)); in PYBIND11_MODULE()
100 [](InterpreterWrapper& self, int subgraph_index) { in PYBIND11_MODULE() argument
101 return tensorflow::PyoOrThrow(self.Invoke(subgraph_index)); in PYBIND11_MODULE()
115 int subgraph_index) { in PYBIND11_MODULE() argument
117 self.ResizeInputTensor(i, value.ptr(), strict, subgraph_index)); in PYBIND11_MODULE()
155 int subgraph_index) { in PYBIND11_MODULE() argument
157 self.SetTensor(i, value.ptr(), subgraph_index)); in PYBIND11_MODULE()
163 int subgraph_index) { in PYBIND11_MODULE() argument
165 self.GetTensor(tensor_index, subgraph_index)); in PYBIND11_MODULE()
[all …]
Dinterpreter_wrapper.cc56 #define TFLITE_PY_SUBGRAPH_TENSOR_BOUNDS_CHECK(i, subgraph_index) \ argument
57 if (i >= interpreter_->subgraph(subgraph_index)->tensors_size() || i < 0) { \
60 interpreter_->subgraph(subgraph_index)->tensors_size()); \
259 PyObject* InterpreterWrapper::AllocateTensors(int subgraph_index) { in AllocateTensors() argument
261 TFLITE_PY_SUBGRAPH_BOUNDS_CHECK(subgraph_index); in AllocateTensors()
262 TFLITE_PY_CHECK(interpreter_->subgraph(subgraph_index)->AllocateTensors()); in AllocateTensors()
266 PyObject* InterpreterWrapper::Invoke(int subgraph_index) { in Invoke() argument
268 TFLITE_PY_SUBGRAPH_BOUNDS_CHECK(subgraph_index); in Invoke()
273 tflite::Subgraph* subgraph = interpreter_->subgraph(subgraph_index); in Invoke()
336 int subgraph_index) { in ResizeInputTensor() argument
[all …]
Dinterpreter_wrapper.h70 PyObject* AllocateTensors(int subgraph_index);
71 PyObject* Invoke(int subgraph_index);
76 int subgraph_index);
87 PyObject* SetTensor(int i, PyObject* value, int subgraph_index);
88 PyObject* GetTensor(int i, int subgraph_index) const;
100 PyObject* tensor(PyObject* base_object, int tensor_index, int subgraph_index);
/external/tensorflow/tensorflow/lite/profiling/
Dprofile_summarizer.cc29 uint32_t subgraph_index; member
71 uint32_t subgraph_index, in GetOperatorDetails() argument
74 const_cast<tflite::Interpreter&>(interpreter).subgraph(subgraph_index); in GetOperatorDetails()
131 const auto subgraph_index = event->extra_event_metadata; in ProcessProfiles() local
132 auto stats_calculator = GetStatsCalculator(subgraph_index); in ProcessProfiles()
144 GetOperatorDetails(interpreter, subgraph_index, node_index); in ProcessProfiles()
191 total_us_per_subgraph_map[subgraph_index] += node_exec_time; in ProcessProfiles()
209 uint32_t subgraph_index) { in GetStatsCalculator() argument
210 if (stats_calculator_map_.count(subgraph_index) == 0) { in GetStatsCalculator()
211 stats_calculator_map_[subgraph_index] = in GetStatsCalculator()
[all …]
Dprofile_summary_formatter.cc49 auto subgraph_index = stats_calc.first; in GenerateReport() local
52 if (subgraph_index == 0) { in GenerateReport()
55 stream << "Subgraph (index: " << subgraph_index << ") " << tag << ":" in GenerateReport()
62 if (subgraph_index != 0) { in GenerateReport()
63 stream << "Subgraph (index: " << subgraph_index << ") "; in GenerateReport()
/external/tensorflow/tensorflow/lite/tools/optimize/calibration/
Dcalibrator.cc232 logger->LogTensorValue(op_info.subgraph_index, i, tensor.data.f, in LoggingEval()
242 kernel_invoke_intermediate(context, op_info.subgraph_index, node, in LoggingEval()
256 logger->LogTensorValue(op_info.subgraph_index, i, tensor.data.f, in LoggingEval()
263 logger->LogTensorValue(op_info.subgraph_index, i, tensor.data.f, in LoggingEval()
316 int subgraph_index, op_index; in GetNodeOpInfoMapAndContext() local
317 std::tie(subgraph_index, op_index) = entry.first; in GetNodeOpInfoMapAndContext()
319 interpreter->node_and_registration(subgraph_index, op_index); in GetNodeOpInfoMapAndContext()
395 for (size_t subgraph_index = 0; subgraph_index < subgraphs->size(); in BuildLoggingInterpreter() local
396 subgraph_index++) { in BuildLoggingInterpreter()
397 auto subgraph = subgraphs->Get(subgraph_index); in BuildLoggingInterpreter()
[all …]
Dcalibration_reader.cc42 int subgraph_index, tensor_index; in AddCalibrationToModel() local
43 std::tie(subgraph_index, tensor_index) = tensorid_stat.first; in AddCalibrationToModel()
44 const auto& subgraph = model->subgraphs[subgraph_index]; in AddCalibrationToModel()
Dcalibration_logger.h52 TfLiteStatus LogTensorValue(int subgraph_index, int tensor_index, in LogTensorValue() argument
55 std::tuple<int, int> key{subgraph_index, tensor_index}; in LogTensorValue()
/external/tensorflow/tensorflow/lite/tools/optimize/calibration/custom_logging_ops/
Dlstm.cc69 const int subgraph_index, ErrorReporter* error_reporter) { in LstmStepWithAuxInput()
118 logger->LogTensorValue(subgraph_index, intemediate_tensor_indexes[1], in LstmStepWithAuxInput()
122 logger->LogTensorValue(subgraph_index, intemediate_tensor_indexes[4], in LstmStepWithAuxInput()
125 logger->LogTensorValue(subgraph_index, intemediate_tensor_indexes[7], in LstmStepWithAuxInput()
127 logger->LogTensorValue(subgraph_index, intemediate_tensor_indexes[10], in LstmStepWithAuxInput()
173 logger->LogTensorValue(subgraph_index, intemediate_tensor_indexes[2], in LstmStepWithAuxInput()
181 logger->LogTensorValue(subgraph_index, intemediate_tensor_indexes[5], in LstmStepWithAuxInput()
190 logger->LogTensorValue(subgraph_index, intemediate_tensor_indexes[8], in LstmStepWithAuxInput()
197 logger->LogTensorValue(subgraph_index, intemediate_tensor_indexes[11], in LstmStepWithAuxInput()
210 logger->LogTensorValue(subgraph_index, intemediate_tensor_indexes[0], in LstmStepWithAuxInput()
[all …]
/external/tensorflow/tensorflow/lite/tools/optimize/calibration/builtin_logging_ops/
Dlstm.cc50 Logger* logger, int intermediate_tensor_index, const int subgraph_index, in CalculateLstmGateFloat() argument
85 logger->LogTensorValue(subgraph_index, intermediate_tensor_index, gate, in CalculateLstmGateFloat()
128 int intermediate_tensor_index, const int subgraph_index, in CalculateLstmOutputCalibration() argument
135 logger->LogTensorValue(subgraph_index, intermediate_tensor_index, scratch, in CalculateLstmOutputCalibration()
186 const int subgraph_index, ErrorReporter* error_reporter) { in LstmStepCalibration()
214 intermediate_tensor_indexes[0], subgraph_index, error_reporter); in LstmStepCalibration()
225 subgraph_index, error_reporter); in LstmStepCalibration()
235 subgraph_index, error_reporter); in LstmStepCalibration()
249 subgraph_index, error_reporter); in LstmStepCalibration()
255 intermediate_tensor_indexes[4], subgraph_index, error_reporter); in LstmStepCalibration()
[all …]
Dlstm.h32 const int subgraph_index, TfLiteNode* node,
36 TfLiteContext* context, const int subgraph_index, TfLiteNode* node,
/external/tensorflow/tensorflow/lite/tools/
Dlist_flex_ops_no_kernel.cc34 for (int subgraph_index = 0; subgraph_index < subgraphs->size(); in AddFlexOpsFromModel() local
35 ++subgraph_index) { in AddFlexOpsFromModel()
36 const tflite::SubGraph* subgraph = subgraphs->Get(subgraph_index); in AddFlexOpsFromModel()
Dlist_flex_ops.cc82 for (int subgraph_index = 0; subgraph_index < subgraphs->size(); in AddFlexOpsFromModel() local
83 ++subgraph_index) { in AddFlexOpsFromModel()
84 const tflite::SubGraph* subgraph = subgraphs->Get(subgraph_index); in AddFlexOpsFromModel()
/external/tensorflow/tensorflow/lite/experimental/acceleration/mini_benchmark/
Dcall.cc168 int subgraph_index; member
185 op_data->subgraph_index = map["subgraph_index"].AsInt32(); in Init()
202 (op_data->subgraph_index < subgraphs->size()) && in Prepare()
203 (op_data->subgraph_index >= 0), in Prepare()
205 Subgraph* subgraph = (*subgraphs)[op_data->subgraph_index].get(); in Prepare()
231 Subgraph* subgraph = (*subgraphs)[op_data->subgraph_index].get(); in Eval()
Dcall_register.h27 int subgraph_index; member
/external/tensorflow/tensorflow/lite/tools/optimize/
Dquantize_weights.cc200 int subgraph_index, bool use_updated_hybrid_scheme) { in InsertQuantizableInputTensorsFromOperator() argument
201 SubGraphT* subgraph = model->subgraphs.at(subgraph_index).get(); in InsertQuantizableInputTensorsFromOperator()
435 for (int subgraph_index = 0, end = model->subgraphs.size(); local
436 subgraph_index < end; ++subgraph_index) {
437 SubGraphT* subgraph = model->subgraphs.at(subgraph_index).get();
444 subgraph_index, use_updated_hybrid_scheme));
560 for (int subgraph_index = 0, end = model->subgraphs.size(); in QuantizeWeightsFloat16() local
561 subgraph_index < end; ++subgraph_index) { in QuantizeWeightsFloat16()
562 SubGraphT* subgraph = model->subgraphs.at(subgraph_index).get(); in QuantizeWeightsFloat16()
Dmodify_model_interface.cc42 size_t subgraph_index; // index of the subgraph. member
203 SubGraphT* subgraph = model->subgraphs.at(tot.subgraph_index).get(); in SetInputTypeToUINT8()
230 SubGraphT* subgraph = model->subgraphs.at(tot.subgraph_index).get(); in SetOutputTypeToUINT8()
263 SubGraphT* subgraph = model->subgraphs.at(tot.subgraph_index).get(); in RemoveInputTensor()
289 SubGraphT* subgraph = model->subgraphs.at(tot.subgraph_index).get(); in RemoveOutputTensor()
/external/tensorflow/tensorflow/lite/examples/label_image/
Dlabel_image.cc176 uint32_t subgraph_index, uint32_t op_index, in PrintProfilingInfo() argument
185 << subgraph_index << ", Node " << std::setw(3) in PrintProfilingInfo()
340 auto subgraph_index = profile_events[i]->extra_event_metadata; in RunInference() local
342 const auto subgraph = interpreter->subgraph(subgraph_index); in RunInference()
346 PrintProfilingInfo(profile_events[i], subgraph_index, op_index, in RunInference()
/external/tensorflow/tensorflow/compiler/mlir/lite/
Dflatbuffer_export.cc472 uint32_t subgraph_index; member
619 const int subgraph_index,
1364 auto build_tensor_and_buffer = [&](Value value, const int subgraph_index, in BuildSubGraph()
1372 tensor_index_map_[subgraph_index][tensor_name] = tensors.size(); in BuildSubGraph()
1606 const uint32_t subgraph_index, tensorflow::OpOrArgNameMapper& name_mapper) { in BuildSignaturedef() argument
1678 result[0].subgraph_index = subgraph_index; in BuildSignaturedef()
1683 const int subgraph_index, const std::map<std::string, std::string>& items) { in GetList() argument
1690 tensor_index_map_[subgraph_index][item.second]); in GetList()
1704 int subgraph_index = 0; in CreateSignatureDefs() local
1706 auto inputs = GetList(subgraph_index, signature_def_data.inputs); in CreateSignatureDefs()
[all …]
/external/tensorflow/tensorflow/lite/tools/serialization/
DREADME.md31 // Note that 0 <= subgraph_index < num_subgraphs
32 tflite::SubgraphWriter writer(&interpreter->subgraph(subgraph_index));

12