Searched refs:half_data (Results 1 – 7 of 7) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/runtime/device/ |
| D | convert_tensor_utils.cc | 24 auto half_data = static_cast<const float16 *>(src); in HalfToFloat() local
27 float tmp = half_to_float(half_data[i]); in HalfToFloat()
37 auto half_data = static_cast<float16 *>(dst); in FloatToHalf() local
39 half_data[i] = float16(float_data[i]); in FloatToHalf()
69 auto half_data = static_cast<const int16_t *>(src); in ShortToInt() local
72 int_data[i] = static_cast<int>(half_data[i]); in ShortToInt()
81 auto half_data = static_cast<int16_t *>(dst); in IntToShort() local
83 half_data[i] = static_cast<int16_t>(int_data[i]); in IntToShort()
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/ir_fission/ |
| D | topk_split.cc | 47 std::vector<float16> half_data; in CreateTensor() local
49 half_data.emplace_back(float16(static_cast<float>(i))); in CreateTensor()
53 half_data.emplace_back(float16(static_cast<float>(gap))); in CreateTensor()
57 reinterpret_cast<void *>(half_data.data()), elem_num); in CreateTensor()
|
| D | max_pool3d_grad_grad_fission.cc | 58 std::vector<float16> half_data; in CreateTensor() local
62 half_data.emplace_back(float16(static_cast<float>(counter))); in CreateTensor()
68 reinterpret_cast<void *>(half_data.data()), elem_num); in CreateTensor()
|
| D | space_to_depth_split.cc | 54 std::vector<float16> half_data(dest_size); in CreateTensor() local
60 half_data[static_cast<size_t>(fill_assist_idx)] = assist_value; in CreateTensor()
65 reinterpret_cast<void *>(half_data.data()), elem_num); in CreateTensor()
|
| /third_party/mindspore/mindspore/ccsrc/runtime/device/ascend/ |
| D | ascend_device_address.cc | 122 std::vector<float16> half_data(elem_num); in FloatToHalfAndSyncHostToDevice() local
123 FloatToHalf(half_data.data(), src, elem_num); in FloatToHalfAndSyncHostToDevice()
124 SyncMemory(dst, half_data.data(), dst_size, RT_MEMCPY_HOST_TO_DEVICE); in FloatToHalfAndSyncHostToDevice()
145 std::vector<float16> half_data(elem_num); in SyncDeviceToHostAndHalfToFloat() local
146 SyncMemory(half_data.data(), src, src_size, RT_MEMCPY_DEVICE_TO_HOST); in SyncDeviceToHostAndHalfToFloat()
147 HalfToFloat(dst, half_data.data(), elem_num); in SyncDeviceToHostAndHalfToFloat()
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/ir_fusion/ |
| D | fused_batch_norm_fusion.cc | 74 auto *half_data = static_cast<const float16 *>(tensor_ptr->data_c()); in GetFactor() local
75 MS_EXCEPTION_IF_NULL(half_data); in GetFactor()
76 float float_data = half_to_float(half_data[0]); in GetFactor()
|
| /third_party/mindspore/mindspore/ccsrc/common/ |
| D | trans.cc | 170 auto half_data = static_cast<float16 *>(dst); in TransDataSrc2Fp16() local
172 half_data[i] = float16(src_data[i]); in TransDataSrc2Fp16()
|