Searched refs:preallocated_ids (Results 1 – 2 of 2) sorted by relevance
137 std::set<ValueId> preallocated_ids; in InitFromGraph() local141 preallocated_ids.insert(input->id); in InitFromGraph()147 preallocated_ids.insert(output->id); in InitFromGraph()153 preallocated_ids); in InitFromGraph()157 RETURN_IF_ERROR(AllocateTensors(&metal_device, preallocated_ids)); in InitFromGraph()166 const GraphFloat32& graph, const std::set<ValueId>& preallocated_ids) { in ReserveGraphTensors() argument172 if (preallocated_ids.find(t->id) != preallocated_ids.end()) { in ReserveGraphTensors()363 MetalDevice* device, const std::set<ValueId>& preallocated_ids) { in AllocateTensors() argument366 if (HasIntersection(node.inputs, preallocated_ids) || in AllocateTensors()367 HasIntersection(node.outputs, preallocated_ids)) { in AllocateTensors()[all …]
133 const std::set<ValueId>& preallocated_ids);139 const std::set<ValueId>& preallocated_ids);