Searched refs:tensor_alloc (Results 1 – 2 of 2) sorted by relevance
294 const auto& tensor_alloc = tensor_alloc_it.second; in GenerateGraphTimeline() local295 auto it = tensor_alloc.lower_bound(ts); in GenerateGraphTimeline()296 if (it != tensor_alloc.begin()) { in GenerateGraphTimeline()
1608 auto tensor_alloc = NewCustomAlloc(tensor->bytes, required_alignment); in AssignCustomAllocForTensor() local1610 interpreter_->SetCustomAllocationForTensor(tensor_idx, tensor_alloc), in AssignCustomAllocForTensor()1683 auto tensor_alloc = NewCustomAlloc(input_tensor->bytes, required_alignment); in TEST_F() local1685 interpreter_->inputs()[0], tensor_alloc, in TEST_F()