Searched refs:SetCustomAllocationForTensor (Results 1 – 6 of 6) sorted by relevance
39 TfLiteStatus Interpreter::SetCustomAllocationForTensor( in SetCustomAllocationForTensor() function in tflite::Interpreter41 return primary_subgraph().SetCustomAllocationForTensor(tensor_index, in SetCustomAllocationForTensor()
1541 interpreter_->SetCustomAllocationForTensor(tensor_idx, tensor_alloc), in AssignCustomAllocForTensor()1602 ASSERT_EQ(interpreter_->SetCustomAllocationForTensor( in TEST_F()1616 ASSERT_EQ(interpreter_->SetCustomAllocationForTensor( in TEST_F()1628 ASSERT_EQ(interpreter_->SetCustomAllocationForTensor( in TEST_F()1746 ASSERT_EQ(interpreter_->SetCustomAllocationForTensor( in TEST_F()1753 ASSERT_EQ(interpreter_->SetCustomAllocationForTensor( in TEST_F()1812 ASSERT_EQ(interpreter_->SetCustomAllocationForTensor( in TEST_F()1817 ASSERT_EQ(interpreter_->SetCustomAllocationForTensor( in TEST_F()1822 ASSERT_EQ(interpreter_->SetCustomAllocationForTensor( in TEST_F()1827 ASSERT_EQ(interpreter_->SetCustomAllocationForTensor( in TEST_F()
660 TfLiteStatus SetCustomAllocationForTensor(
455 interpreter_->SetCustomAllocationForTensor(interpreter_->inputs()[1], alloc); in TEST_F()511 interpreter_->SetCustomAllocationForTensor(interpreter_->inputs()[1], alloc); in TEST_F()
355 TfLiteStatus SetCustomAllocationForTensor(
1708 TfLiteStatus Subgraph::SetCustomAllocationForTensor( in SetCustomAllocationForTensor() function in tflite::Subgraph