Home
last modified time | relevance | path

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

/external/XNNPACK/test/
Dsubgraph-tester.h37 subgraph_.reset(subgraph_ptr); in SubgraphTester()
66 xnn_define_tensor_value(subgraph_.get(), xnn_datatype_fp32, dims.size(), in add_tensor()
85 subgraph_.get(), input_padding_top, input_padding_right, in add_conv()
107 subgraph_.get(), input_padding_top, input_padding_right, in add_depthwise_conv()
122 xnn_define_add2(subgraph_.get(), -std::numeric_limits<float>::infinity(), in add_addition()
133 subgraph_.get(), -std::numeric_limits<float>::infinity(), in add_global_average_pooling()
141 const xnn_status status = xnn_subgraph_optimize(subgraph_.get(), 0 /* flags */); in optimize()
148 xnn_subgraph_rewrite_for_nchw(subgraph_.get()); in rewrite()
154 return subgraph_->values[value_id].layout; in get_layout()
159 …std::unique_ptr<xnn_subgraph, decltype(&xnn_delete_subgraph)> subgraph_{nullptr, xnn_delete_subgra…
/external/tensorflow/tensorflow/lite/tools/serialization/
Dwriter_lib.cc93 operator_to_opcode.resize(subgraph_->nodes_size(), -1); in ExportOperators()
96 subgraph_->node_and_registration(op_index); in ExportOperators()
109 subgraph_->node_and_registration(op_index); in ExportOperators()
132 custom_writer->second(fbb, subgraph_, op_index, &custom_options, in ExportOperators()
161 tensor_to_written_tensor_.resize(subgraph_->tensors_size(), -1); in ExportTensors()
166 std::vector<bool> tensor_is_temporary(subgraph_->tensors_size(), false); in ExportTensors()
167 for (int op_index = 0; op_index < subgraph_->nodes_size(); ++op_index) { in ExportTensors()
169 subgraph_->node_and_registration(op_index); in ExportTensors()
177 for (int tensor_index = 0; tensor_index < subgraph_->tensors_size(); in ExportTensors()
186 for (int tensor_index = 0; tensor_index < subgraph_->tensors_size(); in ExportTensors()
[all …]
Dwriter_lib.h89 : subgraph_(subgraph), in SubgraphWriter()
124 : subgraph_(subgraph), in SubgraphWriter()
185 Subgraph* subgraph_; variable
/external/tensorflow/tensorflow/lite/micro/
Dmicro_interpreter.h124 size_t inputs_size() const { return subgraph_->inputs()->Length(); } in inputs_size()
126 return *subgraph_->inputs(); in inputs()
140 size_t outputs_size() const { return subgraph_->outputs()->Length(); } in outputs_size()
142 return *subgraph_->outputs(); in outputs()
160 size_t operators_size() const { return subgraph_->operators()->size(); } in operators_size()
195 const SubGraph* subgraph_ = nullptr; variable
Dmicro_interpreter.cc147 for (size_t i = 0; i < subgraph_->operators()->size(); ++i) { in ~MicroInterpreter()
169 subgraph_ = (*subgraphs)[0]; in Init()
196 context_.tensors_size = subgraph_->tensors()->size(); in AllocateTensors()
203 for (size_t i = 0; i < subgraph_->operators()->size(); ++i) { in AllocateTensors()
225 for (size_t i = 0; i < subgraph_->operators()->size(); ++i) { in AllocateTensors()
320 for (size_t i = 0; i < subgraph_->operators()->size(); ++i) { in Invoke()
391 for (size_t i = 0; i < subgraph_->tensors()->size(); ++i) { in ResetVariableTensors()
392 auto* tensor = subgraph_->tensors()->Get(i); in ResetVariableTensors()
/external/tensorflow/tensorflow/lite/core/
Dsubgraph.cc178 explicit InterpreterInfo(Subgraph* subgraph) : subgraph_(subgraph) {} in InterpreterInfo()
180 size_t num_tensors() const override { return subgraph_->tensors_size(); } in num_tensors()
182 return subgraph_->tensor(index); in tensor()
185 return subgraph_->execution_plan().size(); in num_execution_nodes()
187 size_t num_total_nodes() const override { return subgraph_->nodes_size(); } in num_total_nodes()
189 int node_index = subgraph_->execution_plan()[index]; in node()
190 return subgraph_->nodes_and_registration()[node_index].first; in node()
193 return subgraph_->execution_plan()[index]; in node_index()
196 return subgraph_->inputs(); in inputs()
199 return subgraph_->outputs(); in outputs()
[all …]