Searched refs:new_tensor_data (Results 1 – 3 of 3) sorted by relevance
| /third_party/mindspore/mindspore/lite/src/train/ |
| D | train_utils.cc | 164 auto new_tensor_data = tensor->data(); in CastTensor() local
165 MS_ASSERT(new_tensor_data != nullptr); in CastTensor()
172 … Float32ToFloat16_fp16_handler(origin_data, new_tensor_data, tensor->ElementsNum(), support_fp16); in CastTensor()
182 auto new_tensor_data = tensor->data(); in CastTensor() local
183 MS_ASSERT(new_tensor_data != nullptr); in CastTensor()
185 … Float16ToFloat32_fp16_handler(origin_data, new_tensor_data, tensor->ElementsNum(), support_fp16); in CastTensor()
|
| /third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/base/ |
| D | group_convolution_creator.cc | 78 …uint8_t *new_tensor_data = reinterpret_cast<uint8_t *>(tensor->data()) + index * new_tensor->Size(… in CreateConstTensor() local
79 memcpy(new_tensor->data(), new_tensor_data, new_tensor->Size()); in CreateConstTensor()
|
| /third_party/mindspore/mindspore/lite/src/ |
| D | scheduler.cc | 105 auto new_tensor_data = tensor->data(); in CastConstTensorData() local
106 MS_ASSERT(new_tensor_data != nullptr); in CastConstTensorData()
108 … Float16ToFloat32_fp16_handler(origin_data, new_tensor_data, tensor->ElementsNum(), support_fp16); in CastConstTensorData()
110 … Float32ToFloat16_fp16_handler(origin_data, new_tensor_data, tensor->ElementsNum(), support_fp16); in CastConstTensorData()
777 auto new_tensor_data = tensor->data(); in CastAndRestoreConstTensorData() local
778 MS_ASSERT(new_tensor_data != nullptr); in CastAndRestoreConstTensorData()
780 … Float16ToFloat32_fp16_handler(origin_data, new_tensor_data, tensor->ElementsNum(), support_fp16); in CastAndRestoreConstTensorData()
782 … Float32ToFloat16_fp16_handler(origin_data, new_tensor_data, tensor->ElementsNum(), support_fp16); in CastAndRestoreConstTensorData()
|