Searched refs:preallocated_tensors_ (Results 1 – 2 of 2) sorted by relevance
375 nil, t.shape, t.descriptor, &preallocated_tensors_[tensor_id])); in AllocateTensors()385 if (preallocated_tensors_.find(tensor_id) != preallocated_tensors_.end()) { in GetTensor()386 return &preallocated_tensors_[tensor_id]; in GetTensor()431 if (preallocated_tensors_.find(id) != preallocated_tensors_.end()) { in GetTensorMemoryType()529 if (preallocated_tensors_.find(tensor_id) != preallocated_tensors_.end()) in AllocateMemoryForBuffers()659 auto status = preallocated_tensors_[it.first].SetBufferHandle(it.second); in UpdatePreallocatedTensors()665 const auto& it = preallocated_tensors_.find(src_ids[i]); in UpdatePreallocatedTensors()666 if (it != preallocated_tensors_.end()) { in UpdatePreallocatedTensors()672 const auto& it = preallocated_tensors_.find(dst_ids[i]); in UpdatePreallocatedTensors()673 if (it != preallocated_tensors_.end()) { in UpdatePreallocatedTensors()
211 std::map<ValueId, MetalSpatialTensor> preallocated_tensors_; variable