Searched refs:primary_subgraph (Results 1 – 7 of 7) sorted by relevance
69 context_ = primary_subgraph().context(); in Interpreter()83 primary_subgraph().SetExternalContext(type, ctx); in SetExternalContext()87 return primary_subgraph().SetInputs(inputs); in SetInputs()91 return primary_subgraph().SetOutputs(outputs); in SetOutputs()95 return primary_subgraph().SetVariables(variables); in SetVariables()99 return primary_subgraph().AllocateTensors(); in AllocateTensors()103 primary_subgraph().ReserveNodes(count); in ReserveNodes()123 return primary_subgraph().AddNodeWithParameters(inputs, outputs, init_data, in AddNodeWithParameters()130 return primary_subgraph().ResizeInputTensor(tensor_index, dims); in ResizeInputTensor()134 TF_LITE_ENSURE_STATUS(primary_subgraph().Invoke()); in Invoke()[all …]
210 const std::vector<int>& inputs() const { return primary_subgraph().inputs(); } in inputs()220 return primary_subgraph().outputs(); in outputs()225 return primary_subgraph().variables(); in variables()238 size_t nodes_size() const { return primary_subgraph().nodes_size(); } in nodes_size()242 return primary_subgraph().execution_plan(); in execution_plan()253 return primary_subgraph().tensor(tensor_index); in tensor()258 return primary_subgraph().tensor(tensor_index); in tensor()264 return primary_subgraph().node_and_registration(node_index); in node_and_registration()382 return primary_subgraph().EnsureTensorDataIsReadable(tensor_index); in EnsureTensorDataIsReadable()469 Subgraph& primary_subgraph() { in primary_subgraph() function[all …]
42 &interpreter_->primary_subgraph()); in TestAccumelateLoopBody()65 builder_->BuildAddSubgraph(&interpreter_->primary_subgraph()); in TEST_F()80 builder_->BuildMulSubgraph(&interpreter_->primary_subgraph()); in TEST_F()95 builder_->BuildPadSubgraph(&interpreter_->primary_subgraph()); in TEST_F()110 builder_->BuildLessEqualCondSubgraph(&interpreter_->primary_subgraph(), 3); in TEST_F()132 builder_->BuildPadLoopBodySubgraph(&interpreter_->primary_subgraph(), {1, 2}); in TEST_F()
46 builder_->BuildWhileSubgraph(&interpreter_->primary_subgraph()); in TEST_F()69 builder_->BuildWhileSubgraph(&interpreter_->primary_subgraph()); in TEST_F()
41 builder_->BuildIfSubgraph(&interpreter_->primary_subgraph()); in SetUp()75 builder_->BuildIfSubgraph(&interpreter_->primary_subgraph()); in SetUp()
103 auto primary_subgraph = model->GetModel()->subgraphs()->Get(0); in TEST() local104 auto operators = primary_subgraph->operators(); in TEST()105 auto subgraph_tensors = primary_subgraph->tensors(); in TEST()
280 auto primary_subgraph = subgraphs->Get(0); in BuildLoggingInterpreter() local282 auto operators = primary_subgraph->operators(); in BuildLoggingInterpreter()283 auto tensors = primary_subgraph->tensors(); in BuildLoggingInterpreter()