Home
last modified time | relevance | path

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

/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/runtime/device/
Dconvert_tensor_utils.cc25 auto half_data = static_cast<const float16 *>(src); in HalfToFloat() local
38 auto half_data = static_cast<float16 *>(dst); in FloatToHalf() local
70 auto half_data = static_cast<const int16_t *>(src); in ShortToInt() local
82 auto half_data = static_cast<int16_t *>(dst); in IntToShort() local
Dms_device_shape_transfer.cc350 auto half_data = static_cast<float16 *>(dst); in TransDataSrc2Fp16() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/optimizer/ir_fission/
Dmax_pool3d_grad_grad_fission.cc64 std::vector<uint16_t> half_data(dims); in CreateTensor() local
Dspace_to_depth_split.cc56 std::vector<float16> half_data(dest_size); in CreateTensor() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/optimizer/ir_fusion/
Dfused_batch_norm_fusion.cc79 auto *half_data = static_cast<const float16 *>(tensor_ptr->data_c()); in GetFactor() local