Searched refs:subgraph_ (Results 1 – 5 of 5) sorted by relevance
198 TensorT* tensor = subgraph_->tensors[tensor_idx].get(); in AsymmetricQuantizeTensor()241 TensorT* weights_tensor = subgraph_->tensors[weights_tensor_idx].get(); in QuantizeOpWithBias()251 const TensorT* input_tensor = subgraph_->tensors[input_tensor_idx].get(); in QuantizeOpWithBias()252 TensorT* bias_tensor = subgraph_->tensors[bias_tensor_idx].get(); in QuantizeOpWithBias()275 auto output_tensor = subgraph_->tensors[op->outputs[0]].get(); in PropagateMinMaxForAvgAndMaxPool()279 auto input_tensor = subgraph_->tensors[op->inputs[0]].get(); in PropagateMinMaxForAvgAndMaxPool()319 auto output_tensor = subgraph_->tensors[op->outputs[0]].get(); in AsymmetricQuantizeSoftmax()336 auto input_tensor = subgraph_->tensors[op->inputs[input_idx]].get(); in AsymmetricQuantizeInputsAndOutputs()345 auto output_tensor = subgraph_->tensors[op->outputs[output_idx]].get(); in AsymmetricQuantizeInputsAndOutputs()355 return std::find(subgraph_->inputs.begin(), subgraph_->inputs.end(), in IsSubgraphInput()[all …]
35 : model_(model), subgraph_(subgraph), error_reporter_(error_reporter) {} in SubgraphQuantizer()66 SubGraphT* subgraph_; variable
46 size_t inputs_size() const { return subgraph_->inputs()->Length(); } in inputs_size()49 size_t outputs_size() const { return subgraph_->outputs()->Length(); } in outputs_size()66 const SubGraph* subgraph_; variable
78 subgraph_ = (*subgraphs)[0]; in MicroInterpreter()79 tensors_ = subgraph_->tensors(); in MicroInterpreter()80 operators_ = subgraph_->operators(); in MicroInterpreter()86 for (int i = 0; i < subgraph_->inputs()->Length(); ++i) { in MicroInterpreter()87 const int tensor_index = subgraph_->inputs()->Get(i); in MicroInterpreter()289 const flatbuffers::Vector<int32_t>* inputs = subgraph_->inputs(); in input()300 const flatbuffers::Vector<int32_t>* outputs = subgraph_->outputs(); in output()
112 explicit InterpreterInfo(Subgraph* subgraph) : subgraph_(subgraph) {} in InterpreterInfo()114 size_t num_tensors() const override { return subgraph_->tensors().size(); } in num_tensors()116 return &subgraph_->tensors()[index]; in tensor()119 return subgraph_->execution_plan().size(); in num_nodes()122 int node_index = subgraph_->execution_plan()[index]; in node()123 return subgraph_->nodes_and_registration()[node_index].first; in node()126 return subgraph_->inputs(); in inputs()129 return subgraph_->outputs(); in outputs()132 return subgraph_->variables(); in variables()136 Subgraph* subgraph_; member in tflite::InterpreterInfo