Searched refs:SetCustomAllocationForTensor (Results 1 – 6 of 6) sorted by relevance
38 TfLiteStatus Interpreter::SetCustomAllocationForTensor( in SetCustomAllocationForTensor() function in tflite::Interpreter40 return primary_subgraph().SetCustomAllocationForTensor(tensor_index, in SetCustomAllocationForTensor()
1610 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 …]
690 TfLiteStatus SetCustomAllocationForTensor(
541 interpreter_->SetCustomAllocationForTensor(interpreter_->inputs()[1], alloc); in TEST_F()597 interpreter_->SetCustomAllocationForTensor(interpreter_->inputs()[1], alloc); in TEST_F()
370 TfLiteStatus SetCustomAllocationForTensor(
1903 TfLiteStatus Subgraph::SetCustomAllocationForTensor( in SetCustomAllocationForTensor() function in tflite::Subgraph