| /external/tensorflow/tensorflow/lite/ |
| D | interpreter.cc | 104 context_ = primary_subgraph().context(); in Interpreter()
163 primary_subgraph().SetExternalContext(type, ctx); in SetExternalContext()
167 return primary_subgraph().SetInputs(std::move(inputs)); in SetInputs()
171 return primary_subgraph().SetOutputs(std::move(outputs)); in SetOutputs()
175 return primary_subgraph().SetVariables(std::move(variables)); in SetVariables()
185 return primary_subgraph().AllocateTensors(); in AllocateTensors()
206 return primary_subgraph().AddNodeWithParameters( in AddNodeWithParameters()
213 return primary_subgraph().ResizeInputTensor(tensor_index, dims); in ResizeInputTensor()
218 return primary_subgraph().ResizeInputTensorStrict(tensor_index, dims); in ResizeInputTensorStrict()
231 scoped_runtime_event, primary_subgraph().Invoke()); in Invoke()
[all …]
|
| D | interpreter_experimental.cc | 40 return primary_subgraph().SetCustomAllocationForTensor(tensor_index, in SetCustomAllocationForTensor()
48 return primary_subgraph().ReleaseNonPersistentMemory(); in ReleaseNonPersistentMemory()
73 bool Interpreter::IsCancelled() { return primary_subgraph().IsCancelled(); } in IsCancelled()
79 bool Interpreter::HasDelegates() { return primary_subgraph().HasDelegates(); } in HasDelegates()
85 TfLiteTensor* tensor = primary_subgraph().tensor(tensor_index); in SetBufferHandle()
104 TfLiteTensor* tensor = primary_subgraph().tensor(tensor_index); in GetBufferHandle()
135 return primary_subgraph().GetProfiler(); in GetProfiler()
|
| D | interpreter.h | 213 const std::vector<int>& inputs() const { return primary_subgraph().inputs(); } in inputs()
223 return primary_subgraph().outputs(); in outputs()
228 return primary_subgraph().variables(); in variables()
241 size_t nodes_size() const { return primary_subgraph().nodes_size(); } in nodes_size()
245 return primary_subgraph().execution_plan(); in execution_plan()
252 return primary_subgraph().tensor(tensor_index); in tensor()
257 return primary_subgraph().tensor(tensor_index); in tensor()
264 return primary_subgraph().node_and_registration(node_index); in node_and_registration()
579 return primary_subgraph().EnsureTensorDataIsReadable(tensor_index); in EnsureTensorDataIsReadable()
720 Subgraph& primary_subgraph() { in primary_subgraph() function
[all …]
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | subgraph_test_util_test.cc | 49 &interpreter_->primary_subgraph()); in TestAccumulateLoopBody()
72 builder_->BuildAddSubgraph(&interpreter_->primary_subgraph()); in TEST_F()
87 builder_->BuildMulSubgraph(&interpreter_->primary_subgraph()); in TEST_F()
102 builder_->BuildPadSubgraph(&interpreter_->primary_subgraph()); in TEST_F()
117 builder_->BuildPadSubgraph(&interpreter_->primary_subgraph()); in TEST_F()
133 builder_->BuildLessEqualCondSubgraph(&interpreter_->primary_subgraph(), 3); in TEST_F()
155 builder_->BuildPadLoopBodySubgraph(&interpreter_->primary_subgraph(), {1, 2}); in TEST_F()
|
| D | while_test.cc | 47 builder_->BuildWhileSubgraph(&interpreter_->primary_subgraph()); in TEST_F()
76 builder_->BuildWhileSubgraph(&interpreter_->primary_subgraph()); in TEST_F()
118 builder_->BuildWhileSubgraph(&interpreter_->primary_subgraph()); in TEST_F()
144 builder_->BuildWhileSubgraph(&interpreter_->primary_subgraph()); in TEST_F()
179 &interpreter_->primary_subgraph()); in TEST_F()
|
| D | test_util.cc | 238 tflite::Subgraph& primary_subgraph = interpreter_->primary_subgraph(); in BuildInterpreter() local
240 &primary_subgraph.context_, in BuildInterpreter()
241 std::unique_ptr<GraphInfo>(primary_subgraph.CreateGraphInfo())); in BuildInterpreter()
242 primary_subgraph.memory_planner_.reset(memory_planner); in BuildInterpreter()
|
| D | if_test.cc | 42 builder_->BuildIfSubgraph(&interpreter_->primary_subgraph()); in SetUp()
127 builder_->BuildIfSubgraph(&interpreter_->primary_subgraph()); in SetUp()
|
| D | call_once_test.cc | 37 &interpreter_->primary_subgraph()); in SetUp()
|
| D | hashtable_ops_test.cc | 649 return interpreter_->primary_subgraph().resources(); in GetResources()
770 return interpreter_->primary_subgraph().resources(); in GetResources()
|
| /external/tensorflow/tensorflow/lite/experimental/acceleration/mini_benchmark/ |
| D | call_test.cc | 193 CallTest::BuildCallSubgraph(&interpreter_->primary_subgraph(), 1, in TEST_F()
216 CallTest::BuildCallSubgraph(&interpreter_->primary_subgraph(), 1, loop_count, in TEST_F()
237 CallTest::BuildCallSubgraph(&interpreter_->primary_subgraph(), 1, kLoopCount, in TEST_F()
265 CallTest::BuildCallSubgraph(&interpreter_->primary_subgraph(), 1, in TEST_F()
295 CallTest::BuildCallSubgraph(&interpreter_->primary_subgraph(), in TEST_F()
298 const TfLiteNode* call_node = &interpreter_->primary_subgraph() in TEST_F()
309 CallTest::BuildMultiNodeGraph(&interpreter_->primary_subgraph()); in TEST_F()
346 CallTest::BuildCallSubgraph(&interpreter_->primary_subgraph(), 1, loop_count, in TEST_F()
361 CallTest::BuildCallSubgraph(&interpreter_->primary_subgraph(), 1, loop_count, in TEST_F()
380 CallTest::BuildCallSubgraph(&interpreter_->primary_subgraph(), 1, kLoopCount, in TEST_F()
[all …]
|
| /external/tensorflow/tensorflow/lite/core/ |
| D | subgraph_test.cc | 37 auto& subgraph = interpreter.primary_subgraph(); in TEST()
52 auto& subgraph = interpreter.primary_subgraph(); in TEST()
65 auto& subgraph = interpreter.primary_subgraph(); in TEST()
|
| /external/tensorflow/tensorflow/lite/tools/serialization/ |
| D | writer_lib_test.cc | 56 SubgraphWriter writer(&interpreter->primary_subgraph()); in WriteToFile()
90 SubgraphWriter writer(&interpreter.primary_subgraph()); in TEST_P()
100 SubgraphWriter writer(&interpreter.primary_subgraph()); in TEST_P()
174 SubgraphWriter writer(&interpreter.primary_subgraph()); in TEST_P()
224 SubgraphWriter writer(&interpreter.primary_subgraph()); in TEST_P()
269 SubgraphWriter writer(&interpreter.primary_subgraph()); in TEST_P()
458 SubgraphWriter writer(&interpreter.primary_subgraph()); in TEST_P()
527 builder_->BuildWhileSubgraph(&interpreter_->primary_subgraph()); in TEST_F()
583 builder_->BuildWhileSubgraph(&interpreter_->primary_subgraph()); in TEST_F()
|
| /external/tensorflow/tensorflow/lite/delegates/flex/ |
| D | kernel_test.cc | 139 const int node_size = interpreter_->primary_subgraph().nodes_size(); in TEST_F()
141 interpreter_->primary_subgraph().node_and_registration(node_size - 1); in TEST_F()
194 delegate_data_->GetBufferMap(interpreter_->primary_subgraph().context()); in TEST_F()
|
| /external/tensorflow/tensorflow/lite/delegates/ |
| D | delegate_test_util.cc | 125 SetUpSubgraph(&interpreter_->primary_subgraph()); in SetUp()
138 SetUpSubgraph(&interpreter_->primary_subgraph()); in SetUp()
|
| /external/tensorflow/tensorflow/lite/profiling/ |
| D | profile_summarizer_test.cc | 160 builder_->BuildIfSubgraph(&interpreter_->primary_subgraph()); in SetUp()
|
| /external/tensorflow/tensorflow/lite/tools/optimize/calibration/ |
| D | calibrator.cc | 307 *context = interpreter->primary_subgraph().context(); in GetNodeOpInfoMapAndContext()
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/common/ |
| D | model_builder_test.cc | 151 TfLiteContext* context() { return interpreter_.primary_subgraph().context(); } in context()
165 interpreter_.primary_subgraph().node_and_registration(index); in node()
170 interpreter_.primary_subgraph().node_and_registration(index); in registration()
|