Home
last modified time | relevance | path

Searched defs:subgraph_index (Results 1 – 25 of 42) sorted by relevance

12

/aosp_15_r20/external/tensorflow/tensorflow/lite/
H A Dtensorflow_profiler_logger_shim.cc29 int subgraph_index, in OnTfLiteOpPrepare()
35 const char* subgraph_name, int subgraph_index) { in OnTfLiteSubgraphInvoke()
47 const char* op_name, int subgraph_index, int node_index) { in OnTfLiteOpInvoke()
67 TFLITE_ATTRIBUTE_WEAK void OnTfLiteArenaAlloc(int subgraph_index, int arena_id, in OnTfLiteArenaAlloc()
72 TFLITE_ATTRIBUTE_WEAK void OnTfLiteArenaDealloc(int subgraph_index, in OnTfLiteArenaDealloc()
H A Dtensorflow_profiler_logger.cc112 void OnTfLiteOpPrepare(const char* op_name, int subgraph_index, in OnTfLiteOpPrepare()
138 int subgraph_index, in OnTfLiteOpInvoke()
174 void AddArenaTrace(bool is_allocating, int subgraph_index, int arena_id, in AddArenaTrace()
192 void OnTfLiteArenaAlloc(int subgraph_index, int arena_id, size_t num_bytes) { in OnTfLiteArenaAlloc()
197 void OnTfLiteArenaDealloc(int subgraph_index, int arena_id, size_t num_bytes) { in OnTfLiteArenaDealloc()
H A Dinterpreter.h270 int subgraph_index, int node_index) const { in node_and_registration()
369 const int subgraph_index = GetSubgraphIndexFromSignature(signature_key); in input_tensor_by_signature() local
383 const int subgraph_index = GetSubgraphIndexFromSignature(signature_key); in output_tensor_by_signature() local
705 const Subgraph* subgraph(int subgraph_index) const { in subgraph()
714 Subgraph* subgraph(int subgraph_index) { in subgraph()
H A Dinterpreter.cc417 for (int subgraph_index = 0; subgraph_index < subgraphs_.size(); in SetMetadata() local
444 for (int subgraph_index = 0; subgraph_index < subgraphs_.size(); in SetSubgraphProfiler() local
H A Darena_planner_test.cc39 void OnTfLiteArenaAlloc(int subgraph_index, int arena_id, size_t num_bytes) { in OnTfLiteArenaAlloc()
45 void OnTfLiteArenaDealloc(int subgraph_index, int arena_id, size_t num_bytes) { in OnTfLiteArenaDealloc()
/aosp_15_r20/external/tensorflow/tensorflow/lite/python/interpreter_wrapper/
H A Dinterpreter_wrapper_pybind11.cc97 [](InterpreterWrapper& self, int subgraph_index) { in PYBIND11_MODULE()
105 [](InterpreterWrapper& self, int subgraph_index) { in PYBIND11_MODULE()
120 int subgraph_index) { in PYBIND11_MODULE()
160 int subgraph_index) { in PYBIND11_MODULE()
168 int subgraph_index) { in PYBIND11_MODULE()
199 int tensor_index, int subgraph_index) { in PYBIND11_MODULE()
H A Dinterpreter_wrapper.cc57 #define TFLITE_PY_SUBGRAPH_TENSOR_BOUNDS_CHECK(i, subgraph_index) \ argument
263 PyObject* InterpreterWrapper::AllocateTensors(int subgraph_index) { in AllocateTensors()
274 PyObject* InterpreterWrapper::Invoke(int subgraph_index) { in Invoke()
344 int subgraph_index) { in ResizeInputTensor()
489 int subgraph_index) { in SetTensor()
615 int subgraph_index) { in CheckGetTensorArgs()
677 int32_t subgraph_index = in GetSubgraphIndexFromSignature() local
762 int subgraph_index) { in tensor()
/aosp_15_r20/external/tensorflow/tensorflow/lite/tools/optimize/calibration/builtin_logging_ops/
H A Dlstm.cc50 Logger* logger, int intermediate_tensor_index, const int subgraph_index, in CalculateLstmGateFloat()
128 int intermediate_tensor_index, const int subgraph_index, in CalculateLstmOutputCalibration()
158 inline void LstmStepCalibration( in LstmStepCalibration()
264 TfLiteStatus EvalCalibration( in EvalCalibration()
465 TfLiteStatus lstm_eval(TfLiteContext* context, int subgraph_index, in lstm_eval()
639 const int subgraph_index, TfLiteNode* node, in lstm_logging_kernel()
647 TfLiteContext* context, const int subgraph_index, TfLiteNode* node, in unidirectional_sequence_lstm_logging_kernel()
/aosp_15_r20/external/tensorflow/tensorflow/lite/profiling/
H A Dprofile_summarizer.cc30 uint32_t subgraph_index; member
72 uint32_t subgraph_index, in GetOperatorDetails()
117 const auto subgraph_index = event->extra_event_metadata; in ProcessProfiles() local
192 uint32_t subgraph_index) { in GetStatsCalculator()
H A Dprofile_summary_formatter.cc50 auto subgraph_index = stats_calc.first; in GenerateReport() local
/aosp_15_r20/external/tensorflow/tensorflow/lite/tools/optimize/calibration/custom_logging_ops/
H A Dlstm.cc40 inline void LstmStepWithAuxInput( in LstmStepWithAuxInput()
340 TfLiteStatus EvalFloat( in EvalFloat()
543 TfLiteStatus lstm_eval(TfLiteContext* context, const int subgraph_index, in lstm_eval()
654 const int subgraph_index, TfLiteNode* node, in lstm_logging_kernel()
/aosp_15_r20/external/tensorflow/tensorflow/lite/python/
H A Dutil.py658 def _modify_model_input_type_per_subgraph(model, subgraph_index, argument
765 def _modify_model_output_type_per_subgraph(model, subgraph_index, argument
885 def _remove_redundant_quantize_ops_per_subgraph(model, subgraph_index, argument
/aosp_15_r20/external/tensorflow/tensorflow/lite/python/optimize/
H A Dcalibration_wrapper.cc241 const int subgraph_index = in Prepare() local
314 const int subgraph_index = in FeedTensor() local
391 const int subgraph_index = in SetTensor() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/lite/
H A Dflatbuffer_export.cc484 uint32_t subgraph_index; member
1465 const std::string& tensor_name) { in BuildSubGraph()
1741 const uint32_t subgraph_index, tensorflow::OpOrArgNameMapper& name_mapper) { in BuildSignaturedef()
1818 const int subgraph_index, const std::map<std::string, std::string>& items) { in GetList()
1839 int subgraph_index = 0; in CreateSignatureDefs() local
1983 int subgraph_index = 0; in TranslateInternal() local
/aosp_15_r20/external/tensorflow/tensorflow/lite/tools/optimize/
H A Dquantize_weights.cc214 int subgraph_index, bool use_updated_hybrid_scheme) { in InsertQuantizableInputTensorsFromOperator()
403 for (int subgraph_index = 0, end = model->subgraphs.size(); local
528 for (int subgraph_index = 0, end = model->subgraphs.size(); in QuantizeWeightsFloat16() local
H A Dquantize_weights_portable.cc200 int subgraph_index, bool use_updated_hybrid_scheme) { in InsertQuantizableInputTensorsFromOperator()
389 for (int subgraph_index = 0, end = model->subgraphs.size(); local
514 for (int subgraph_index = 0, end = model->subgraphs.size(); in QuantizeWeightsFloat16() local
H A Doperator_property.cc25 const OpVariant GetOperatorVariant(const ModelT* model, int subgraph_index, in GetOperatorVariant()
55 OperatorProperty GetOperatorProperty(const ModelT* model, int subgraph_index, in GetOperatorProperty()
/aosp_15_r20/external/tensorflow/tensorflow/lite/examples/label_image/
H A Dlabel_image.cc183 uint32_t subgraph_index, uint32_t op_index, in PrintProfilingInfo()
345 auto subgraph_index = profile_events[i]->extra_event_metadata; in RunInference() local
/aosp_15_r20/external/tensorflow/tensorflow/lite/tools/optimize/calibration/
H A Dcalibrator.cc318 int subgraph_index, op_index; in GetNodeOpInfoMapAndContext() local
401 for (size_t subgraph_index = 0; subgraph_index < subgraphs->size(); in BuildLoggingInterpreter() local
H A Dcalibration_logger.h52 TfLiteStatus LogTensorValue(int subgraph_index, int tensor_index, in LogTensorValue()
H A Dcalibration_reader.cc45 int subgraph_index, tensor_index; in AddCalibrationToModel() local
/aosp_15_r20/external/tensorflow/tensorflow/lite/tools/serialization/
H A Dwriter_lib.cc497 void ModelWriter::SetUnusedTensors(int subgraph_index, in SetUnusedTensors()
503 int subgraph_index, const std::vector<int>& inputs, in SetCustomInputOutput()
/aosp_15_r20/external/tensorflow/tensorflow/lite/experimental/acceleration/mini_benchmark/
H A Dcall_register.h27 int subgraph_index; member
/aosp_15_r20/external/tensorflow/tensorflow/lite/internal/
H A Dsignature_def.h33 uint32_t subgraph_index; member
/aosp_15_r20/external/tensorflow/tensorflow/lite/tools/
H A Dlist_flex_ops_no_kernel.cc36 for (int subgraph_index = 0; subgraph_index < subgraphs->size(); in AddFlexOpsFromModel() local

12