Home
last modified time | relevance | path

Searched refs:subgraph_ (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/lite/tools/optimize/
Dsubgraph_quantizer.cc198 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 …]
Dsubgraph_quantizer.h35 : model_(model), subgraph_(subgraph), error_reporter_(error_reporter) {} in SubgraphQuantizer()
66 SubGraphT* subgraph_; variable
/external/tensorflow/tensorflow/lite/experimental/micro/
Dmicro_interpreter.h46 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
Dmicro_interpreter.cc78 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()
/external/tensorflow/tensorflow/lite/core/
Dsubgraph.cc112 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