| /external/tensorflow/tensorflow/lite/python/interpreter_wrapper/ |
| D | interpreter_wrapper_pybind11.cc | 94 [](InterpreterWrapper& self, int subgraph_index) { in PYBIND11_MODULE()
100 [](InterpreterWrapper& self, int subgraph_index) { in PYBIND11_MODULE()
115 int subgraph_index) { in PYBIND11_MODULE()
155 int subgraph_index) { in PYBIND11_MODULE()
163 int subgraph_index) { in PYBIND11_MODULE()
194 int tensor_index, int subgraph_index) { in PYBIND11_MODULE()
|
| D | interpreter_wrapper.cc | 56 #define TFLITE_PY_SUBGRAPH_TENSOR_BOUNDS_CHECK(i, subgraph_index) \ argument
259 PyObject* InterpreterWrapper::AllocateTensors(int subgraph_index) { in AllocateTensors()
266 PyObject* InterpreterWrapper::Invoke(int subgraph_index) { in Invoke()
336 int subgraph_index) { in ResizeInputTensor()
481 int subgraph_index) { in SetTensor()
606 int subgraph_index) { in CheckGetTensorArgs()
665 int32_t subgraph_index = in GetSubgraphIndexFromSignature() local
750 int subgraph_index) { in tensor()
|
| /external/tensorflow/tensorflow/lite/profiling/ |
| D | profile_summarizer.cc | 29 uint32_t subgraph_index; member
71 uint32_t subgraph_index, in GetOperatorDetails()
131 const auto subgraph_index = event->extra_event_metadata; in ProcessProfiles() local
209 uint32_t subgraph_index) { in GetStatsCalculator()
|
| D | profile_summary_formatter.cc | 49 auto subgraph_index = stats_calc.first; in GenerateReport() local
|
| /external/tensorflow/tensorflow/lite/ |
| D | interpreter.h | 269 int subgraph_index, int node_index) const { in node_and_registration()
368 const int subgraph_index = GetSubgraphIndexFromSignature(signature_key); in input_tensor_by_signature() local
382 const int subgraph_index = GetSubgraphIndexFromSignature(signature_key); in output_tensor_by_signature() local
679 const Subgraph* subgraph(int subgraph_index) const { in subgraph()
688 Subgraph* subgraph(int subgraph_index) { in subgraph()
|
| D | interpreter_experimental.cc | 150 for (int subgraph_index = 0; subgraph_index < subgraphs_.size(); in SetSubgraphProfiler() local
162 for (int subgraph_index = 0; subgraph_index < subgraphs_.size(); in PreserveAllTensorsExperimental() local
|
| /external/tensorflow/tensorflow/lite/tools/optimize/calibration/builtin_logging_ops/ |
| D | lstm.cc | 50 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()
|
| /external/tensorflow/tensorflow/lite/experimental/acceleration/mini_benchmark/ |
| D | call_register.h | 27 int subgraph_index; member
|
| D | call.cc | 168 int subgraph_index; member
|
| D | grafter.cc | 97 int subgraph_index = 0; in SubGraphs() local
|
| /external/tensorflow/tensorflow/lite/internal/ |
| D | signature_def.h | 33 uint32_t subgraph_index; member
|
| /external/tensorflow/tensorflow/lite/tools/optimize/calibration/ |
| D | calibration_logger.h | 52 TfLiteStatus LogTensorValue(int subgraph_index, int tensor_index, in LogTensorValue()
|
| D | calibration_reader.cc | 42 int subgraph_index, tensor_index; in AddCalibrationToModel() local
|
| D | calibration_common.h | 57 int subgraph_index; member
|
| D | calibrator.cc | 316 int subgraph_index, op_index; in GetNodeOpInfoMapAndContext() local
395 for (size_t subgraph_index = 0; subgraph_index < subgraphs->size(); in BuildLoggingInterpreter() local
|
| /external/tensorflow/tensorflow/lite/tools/optimize/calibration/custom_logging_ops/ |
| D | lstm.cc | 40 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()
|
| /external/tensorflow/tensorflow/lite/tools/ |
| D | list_flex_ops_no_kernel.cc | 34 for (int subgraph_index = 0; subgraph_index < subgraphs->size(); in AddFlexOpsFromModel() local
|
| D | list_flex_ops.cc | 82 for (int subgraph_index = 0; subgraph_index < subgraphs->size(); in AddFlexOpsFromModel() local
|
| /external/tensorflow/tensorflow/lite/examples/label_image/ |
| D | label_image.cc | 176 uint32_t subgraph_index, uint32_t op_index, in PrintProfilingInfo()
340 auto subgraph_index = profile_events[i]->extra_event_metadata; in RunInference() local
|
| /external/tensorflow/tensorflow/lite/tools/optimize/ |
| D | operator_property.cc | 25 const OpVariant GetOperatorVariant(const ModelT* model, int subgraph_index, in GetOperatorVariant()
55 OperatorProperty GetOperatorProperty(const ModelT* model, int subgraph_index, in GetOperatorProperty()
|
| D | quantize_weights.cc | 200 int subgraph_index, bool use_updated_hybrid_scheme) { in InsertQuantizableInputTensorsFromOperator()
435 for (int subgraph_index = 0, end = model->subgraphs.size(); local
560 for (int subgraph_index = 0, end = model->subgraphs.size(); in QuantizeWeightsFloat16() local
|
| /external/tensorflow/tensorflow/lite/tools/serialization/ |
| D | writer_lib.cc | 477 void ModelWriter::SetUnusedTensors(int subgraph_index, in SetUnusedTensors()
483 int subgraph_index, const std::vector<int>& inputs, in SetCustomInputOutput()
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/ |
| D | flatbuffer_export.cc | 472 uint32_t subgraph_index; member
1365 const std::string& tensor_name) { in BuildSubGraph()
1606 const uint32_t subgraph_index, tensorflow::OpOrArgNameMapper& name_mapper) { in BuildSignaturedef()
1683 const int subgraph_index, const std::map<std::string, std::string>& items) { in GetList()
1704 int subgraph_index = 0; in CreateSignatureDefs() local
1825 int subgraph_index = 0; in TranslateInternal() local
|
| /external/tensorflow/tensorflow/lite/java/src/main/native/ |
| D | tensor_jni.cc | 44 TensorHandle(Interpreter* interpreter, int tensor_index, int subgraph_index) in TensorHandle()
442 jint subgraph_index) { in Java_org_tensorflow_lite_Tensor_create()
|
| /external/tensorflow/tensorflow/lite/core/ |
| D | subgraph.h | 382 SubgraphAwareProfiler(Profiler* profiler, int64_t subgraph_index) in SubgraphAwareProfiler()
|