Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/lite/
Dinterpreter.cc163 TfLiteStatus Interpreter::SetCustomAllocationForTensor( in SetCustomAllocationForTensor() function in tflite::Interpreter
165 return primary_subgraph().SetCustomAllocationForTensor(tensor_index, in SetCustomAllocationForTensor()
Dinterpreter_test.cc1538 interpreter_->SetCustomAllocationForTensor(tensor_idx, tensor_alloc), in AssignCustomAllocForTensor()
1599 ASSERT_EQ(interpreter_->SetCustomAllocationForTensor( in TEST_F()
1610 ASSERT_EQ(interpreter_->SetCustomAllocationForTensor( in TEST_F()
1730 ASSERT_EQ(interpreter_->SetCustomAllocationForTensor( in TEST_F()
1737 ASSERT_EQ(interpreter_->SetCustomAllocationForTensor( in TEST_F()
Dinterpreter.h621 TfLiteStatus SetCustomAllocationForTensor(
/external/tensorflow/tensorflow/lite/core/
Dsubgraph.h347 TfLiteStatus SetCustomAllocationForTensor(
Dsubgraph.cc1593 TfLiteStatus Subgraph::SetCustomAllocationForTensor( in SetCustomAllocationForTensor() function in tflite::Subgraph
/external/tensorflow/tensorflow/lite/tools/serialization/
Dwriter_lib_test.cc453 interpreter_->SetCustomAllocationForTensor(interpreter_->inputs()[1], alloc); in TEST_F()