Home
last modified time | relevance | path

Searched defs:half_data (Results 1 – 7 of 7) sorted by relevance

/third_party/mindspore/mindspore/ccsrc/runtime/device/
Dconvert_tensor_utils.cc24 auto half_data = static_cast<const float16 *>(src); in HalfToFloat() local
37 auto half_data = static_cast<float16 *>(dst); in FloatToHalf() local
69 auto half_data = static_cast<const int16_t *>(src); in ShortToInt() local
81 auto half_data = static_cast<int16_t *>(dst); in IntToShort() local
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/ir_fission/
Dmax_pool3d_grad_grad_fission.cc58 std::vector<float16> half_data; in CreateTensor() local
Dspace_to_depth_split.cc54 std::vector<float16> half_data(dest_size); in CreateTensor() local
Dtopk_split.cc47 std::vector<float16> half_data; in CreateTensor() local
/third_party/mindspore/mindspore/ccsrc/runtime/device/ascend/
Dascend_device_address.cc122 std::vector<float16> half_data(elem_num); in FloatToHalfAndSyncHostToDevice() local
145 std::vector<float16> half_data(elem_num); in SyncDeviceToHostAndHalfToFloat() local
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/ir_fusion/
Dfused_batch_norm_fusion.cc74 auto *half_data = static_cast<const float16 *>(tensor_ptr->data_c()); in GetFactor() local
/third_party/mindspore/mindspore/ccsrc/common/
Dtrans.cc170 auto half_data = static_cast<float16 *>(dst); in TransDataSrc2Fp16() local