Searched defs:tensor_alloc (Results 1 – 2 of 2) sorted by relevance
291 const auto& tensor_alloc = tensor_alloc_it.second; in GenerateGraphTimeline() local
1536 auto tensor_alloc = NewCustomAlloc(tensor->bytes, required_alignment); in AssignCustomAllocForTensor() local