Home
last modified time | relevance | path

Searched refs:SetCustomAllocationForTensor (Results 1 – 7 of 7) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/lite/
H A Dinterpreter_test.cc1610 interpreter_->SetCustomAllocationForTensor(tensor_idx, tensor_alloc), in AssignCustomAllocForTensor()
1670 ASSERT_EQ(interpreter_->SetCustomAllocationForTensor( in TEST_F()
1684 ASSERT_EQ(interpreter_->SetCustomAllocationForTensor( in TEST_F()
1696 ASSERT_EQ(interpreter_->SetCustomAllocationForTensor( in TEST_F()
1814 ASSERT_EQ(interpreter_->SetCustomAllocationForTensor( in TEST_F()
1821 ASSERT_EQ(interpreter_->SetCustomAllocationForTensor( in TEST_F()
1880 ASSERT_EQ(interpreter_->SetCustomAllocationForTensor( in TEST_F()
1885 ASSERT_EQ(interpreter_->SetCustomAllocationForTensor( in TEST_F()
1890 ASSERT_EQ(interpreter_->SetCustomAllocationForTensor( in TEST_F()
1895 ASSERT_EQ(interpreter_->SetCustomAllocationForTensor( in TEST_F()
[all …]
H A Dinterpreter_experimental.cc38 TfLiteStatus Interpreter::SetCustomAllocationForTensor( in SetCustomAllocationForTensor() function in tflite::Interpreter
40 return primary_subgraph().SetCustomAllocationForTensor(tensor_index, in SetCustomAllocationForTensor()
H A Dinterpreter.h690 TfLiteStatus SetCustomAllocationForTensor(
/aosp_15_r20/external/tensorflow/tensorflow/lite/tools/serialization/
H A Dwriter_lib_test.cc541 interpreter_->SetCustomAllocationForTensor(interpreter_->inputs()[1], alloc); in TEST_F()
597 interpreter_->SetCustomAllocationForTensor(interpreter_->inputs()[1], alloc); in TEST_F()
/aosp_15_r20/external/tensorflow/tensorflow/lite/core/
H A Dsubgraph.h370 TfLiteStatus SetCustomAllocationForTensor(
H A Dsubgraph.cc1903 TfLiteStatus Subgraph::SetCustomAllocationForTensor( in SetCustomAllocationForTensor() function in tflite::Subgraph
/aosp_15_r20/packages/modules/NeuralNetworks/runtime/
DNeuralNetworksV2.cpp1410 context->interpreter->SetCustomAllocationForTensor(context->interpreter->outputs()[index], in ANeuralNetworksExecution_setOutput()