Searched refs:const_tensors_ (Results 1 – 6 of 6) sorted by relevance
387 } else if (const_tensors_.find(tensor_id) != const_tensors_.end()) { in GetTensor()388 return &const_tensors_[tensor_id]; in GetTensor()433 } else if (const_tensors_.find(id) != const_tensors_.end()) { in GetTensorMemoryType()471 RETURN_IF_ERROR(const_tensors_[description.first].CreateFromDescriptor( in AllocateMemoryForConstTensors()
214 std::map<ValueId, MetalSpatialTensor> const_tensors_; variable
474 if (const_tensors_.find(id) != const_tensors_.end()) { in GetTensorMemoryType()496 RETURN_IF_ERROR(const_tensors_[description.first].CreateFromDescriptor( in AllocateMemoryForConstTensors()701 if (const_tensors_.find(id) != const_tensors_.end()) { in GetTensor()702 return &const_tensors_[id]; in GetTensor()
221 std::map<ValueId, Tensor> const_tensors_; variable
156 std::vector<Tensor> const_tensors_; variable
153 const_tensors_.emplace_back(*const_tensor); in Initialize()