Searched refs:tensor_buf (Results 1 – 3 of 3) sorted by relevance
| /third_party/mindspore/mindspore/lite/src/ops/compat/ |
| D | attr_transfer_common.cc | 54 auto tensor_buf = reinterpret_cast<char *>(malloc(byte_num)); in AttrToTensor() local
55 if (tensor_buf == nullptr) { in AttrToTensor()
60 memcpy(tensor_buf, buf, fbb.GetSize()); in AttrToTensor()
61 auto tensor = flatbuffers::GetRoot<schema::Tensor>(tensor_buf); in AttrToTensor()
62 tensor_bufs->push_back(tensor_buf); in AttrToTensor()
|
| /third_party/mindspore/mindspore/core/load_mindir/ |
| D | anf_model_parser.cc | 496 const std::string &tensor_buf = attr_tensor.raw_data(); in ObtainCNodeAttrInTensorForm() local
498 …auto ret = memcpy_s(tensor_data_buf, tensor_info->data().nbytes(), tensor_buf.data(), tensor_buf.s… in ObtainCNodeAttrInTensorForm()
590 const std::string &tensor_buf = attr_tensor.raw_data(); in ObtainValueNodeInTensorForm() local
593 …_data_buf, static_cast<size_t>(tensor_info->data().nbytes()), tensor_buf.data(), tensor_buf.size()… in ObtainValueNodeInTensorForm()
619 const std::string &tensor_buf = attr_tensor.raw_data(); in ObtainValueNodeInTupleTensorForm() local
621 …t = memcpy_s(tensor_data_buf, static_cast<size_t>(tensor_info->data().nbytes()), tensor_buf.data(), in ObtainValueNodeInTupleTensorForm()
622 tensor_buf.size()); in ObtainValueNodeInTupleTensorForm()
|
| /third_party/mindspore/mindspore/lite/src/ |
| D | lite_model.cc | 111 for (auto &tensor_buf : attr_tensor_bufs_) { in Free() local
112 free(tensor_buf); in Free()
113 tensor_buf = nullptr; in Free()
|