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